From 0d242853e699226e3a9df1105645506bfb7b31a8 Mon Sep 17 00:00:00 2001
From: Sam Gammon <sam@elide.ventures>
Date: Fri, 12 Jan 2024 23:00:07 -0800
Subject: [PATCH] feat: makevar expansion for `extra_args`

Adds Makefile variable expansion support to the `extra_args` flags
passed to `native-image`.

Signed-off-by: Sam Gammon <sam@elide.ventures>
---
 internal/native_image/builder.bzl |  2 +-
 internal/native_image/rules.bzl   | 16 ++++++++++++++--
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/internal/native_image/builder.bzl b/internal/native_image/builder.bzl
index fc2a2068..a95ec8be 100644
--- a/internal/native_image/builder.bzl
+++ b/internal/native_image/builder.bzl
@@ -231,4 +231,4 @@ def assemble_native_build_options(
 
     # append extra arguments last
     for arg in ctx.attr.extra_args:
-        args.add(arg)
+        args.add(ctx.expand_location(arg))
diff --git a/internal/native_image/rules.bzl b/internal/native_image/rules.bzl
index 1b18f62e..5ca59aa8 100644
--- a/internal/native_image/rules.bzl
+++ b/internal/native_image/rules.bzl
@@ -43,6 +43,14 @@ def _build_action_message(ctx):
     }
     return (_mode_label[ctx.attr.optimization_mode or "default"])
 
+def _prepare_execution_env(ctx, base_env = {}):
+    effective = {}
+    effective.update(base_env)
+    effective.update({
+        "GRAALVM_BAZEL": "true",
+    })
+    return effective
+
 def _graal_binary_implementation(ctx):
     graal_attr = ctx.attr.native_image_tool
     extra_tool_deps = []
@@ -115,7 +123,7 @@ def _graal_binary_implementation(ctx):
     elif (not is_windows and not is_macos) and ctx.attr.shared_library:
         bin_postfix = _BIN_POSTFIX_SO
 
-    args = ctx.actions.args().use_param_file("@%s", use_always=False)
+    args = ctx.actions.args().use_param_file("@%s", use_always=True)
     all_outputs = _prepare_native_image_rule_context(
         ctx,
         args,
@@ -126,6 +134,10 @@ def _graal_binary_implementation(ctx):
         bin_postfix = bin_postfix,
     )
     binary = all_outputs[0]
+    execution_env = _prepare_execution_env(
+        ctx,
+        native_toolchain.env,
+    )
 
     # assemble final inputs
     inputs = depset(
@@ -137,7 +149,7 @@ def _graal_binary_implementation(ctx):
         "executable": graal,
         "inputs": inputs,
         "mnemonic": "NativeImage",
-        "env": native_toolchain.env,
+        "env": execution_env,
         "execution_requirements": {k: "" for k in native_toolchain.execution_requirements},
         "progress_message": "Native Image __target__ (__mode__) %{label}"
             .replace("__mode__", _build_action_message(ctx))