Add external reader support in pkl-gradle

Author: HT154

pkl-gradle/src/main/java/org/pkl/gradle/PklPlugin.java

@@ -470,6 +470,8 @@ private <T extends BasePklTask, S extends BasePklSpec> void configureBaseTask(T
     task.getTestPort().set(spec.getTestPort());
     task.getHttpProxy().set(spec.getHttpProxy());
     task.getHttpNoProxy().set(spec.getHttpNoProxy());
+    task.getExternalModuleReaders().set(spec.getExternalModuleReaders());
+    task.getExternalResourceReaders().set(spec.getExternalResourceReaders());
   }
 
   private List<File> getTransitiveModules(AnalyzeImportsTask analyzeTask) {

pkl-gradle/src/main/java/org/pkl/gradle/spec/BasePklSpec.java

@@ -1,5 +1,5 @@
 /*
- * Copyright © 2024 Apple Inc. and the Pkl project authors. All rights reserved.
+ * Copyright © 2024-2025 Apple Inc. and the Pkl project authors. All rights reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -57,4 +57,8 @@ public interface BasePklSpec {
   Property<URI> getHttpProxy();
 
   ListProperty<String> getHttpNoProxy();
+
+  MapProperty<String, String> getExternalModuleReaders();
+
+  MapProperty<String, String> getExternalResourceReaders();
 }

pkl-gradle/src/main/java/org/pkl/gradle/task/BasePklTask.java

@@ -42,8 +42,10 @@
 import org.gradle.api.tasks.Internal;
 import org.gradle.api.tasks.Optional;
 import org.gradle.api.tasks.TaskAction;
+import org.pkl.commons.StringsKt;
 import org.pkl.commons.cli.CliBaseOptions;
 import org.pkl.core.evaluatorSettings.Color;
+import org.pkl.core.evaluatorSettings.PklEvaluatorSettings;
 import org.pkl.core.util.LateInit;
 import org.pkl.core.util.Nullable;
 import org.pkl.gradle.utils.PluginUtils;
@@ -142,6 +144,14 @@ public Provider<String> getEvalRootDirPath() {
   @Optional
   public abstract ListProperty<String> getHttpNoProxy();
 
+  @Input
+  @Optional
+  public abstract MapProperty<String, String> getExternalModuleReaders();
+
+  @Input
+  @Optional
+  public abstract MapProperty<String, String> getExternalResourceReaders();
+
   /**
    * There are issues with using native libraries in Gradle plugins. As a workaround for now, make
    * Truffle use an un-optimized runtime.
@@ -195,8 +205,8 @@ protected CliBaseOptions getCliBaseOptions() {
               Collections.emptyList(),
               getHttpProxy().getOrNull(),
               getHttpNoProxy().getOrElse(List.of()),
-              Map.of(),
-              Map.of());
+              parseExternalReaders(getExternalModuleReaders()),
+              parseExternalReaders(getExternalResourceReaders()));
     }
     return cachedOptions;
   }
@@ -231,4 +241,17 @@ protected List<Pattern> patternsFromStrings(List<String> patterns) {
     @Nullable T value = provider.getOrNull();
     return value == null ? null : f.apply(value);
   }
+
+  protected Map<String, PklEvaluatorSettings.ExternalReader> parseExternalReaders(
+      Provider<Map<String, String>> provider) {
+    return provider.getOrElse(Map.of()).entrySet().stream()
+        .collect(
+            Collectors.toMap(
+                Map.Entry::getKey,
+                (it) -> {
+                  var cmd = StringsKt.shlex(it.getValue());
+                  return new PklEvaluatorSettings.ExternalReader(
+                      cmd.getFirst(), cmd.subList(1, cmd.size()));
+                }));
+  }
 }

pkl-gradle/src/main/java/org/pkl/gradle/task/ModulesTask.java

@@ -163,8 +163,8 @@ protected CliBaseOptions getCliBaseOptions() {
               Collections.emptyList(),
               null,
               List.of(),
-              Map.of(),
-              Map.of());
+              parseExternalReaders(getExternalModuleReaders()),
+              parseExternalReaders(getExternalResourceReaders()));
     }
     return cachedOptions;
   }