From 09166ba1651cb72d869b36fb2a5273a43f03aa28 Mon Sep 17 00:00:00 2001
From: Daniel Chao <dan.chao@apple.com>
Date: Mon, 3 Mar 2025 08:49:44 -0800
Subject: [PATCH] Allow native-image flags to be passed through from Gradle
 properties (#1001)

Allow other build environments (e.g. homebrew) to configure the native-image build without requiring source code changes.
---
 pkl-cli/pkl-cli.gradle.kts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pkl-cli/pkl-cli.gradle.kts b/pkl-cli/pkl-cli.gradle.kts
index 87679e3c..facfeabb 100644
--- a/pkl-cli/pkl-cli.gradle.kts
+++ b/pkl-cli/pkl-cli.gradle.kts
@@ -237,6 +237,16 @@ fun Exec.configureExecutable(
   workingDir(outputFile.map { it.asFile.parentFile })
   executable = "${graalVm.baseDir}/bin/$nativeImageCommandName"
 
+  // For any system properties starting with `pkl.native`, strip off that prefix and pass the rest
+  // through as arguments to native-image.
+  //
+  // Allow setting args using flags like
+  // (-Dpkl.native-Dpolyglot.engine.userResourceCache=/my/cache/dir) when building through Gradle.
+  val extraArgsFromProperties =
+    System.getProperties()
+      .filter { it.key.toString().startsWith("pkl.native") }
+      .map { "${it.key}=${it.value}".substring("pkl.native".length) }
+
   // JARs to exclude from the class path for the native-image build.
   val exclusions = listOf(libs.graalSdk).map { it.get().module.name }
   // https://www.graalvm.org/22.0/reference-manual/native-image/Options/
@@ -294,6 +304,7 @@ fun Exec.configureExecutable(
         // tools.
         addAll(environment.keys.filter { it.startsWith("HOMEBREW_") }.map { "-E$it" })
         addAll(extraArgs)
+        addAll(extraArgsFromProperties)
       }
     }
   )