Merge pull request #9373 from PumasAI/jk/custom-julia-project

Allow using a custom env for loading QuartoNotebookRunner

Author: cscheid

src/execute/julia.ts

@@ -239,7 +239,43 @@ async function startOrReuseJuliaServer(
       `Transport file ${transportFile} doesn't exist`,
     );
     info("Starting julia control server process. This might take a while...");
-    await ensureQuartoNotebookRunnerEnvironment(options);
+
+    let juliaProject = Deno.env.get("QUARTO_JULIA_PROJECT");
+
+    if (juliaProject === undefined) {
+      await ensureQuartoNotebookRunnerEnvironment(options);
+      juliaProject = juliaRuntimeDir();
+    } else {
+      trace(
+        options,
+        `Custom julia project set via QUARTO_JULIA_PROJECT="${juliaProject}". Checking if QuartoNotebookRunner can be loaded.`,
+      );
+      const qnrTestCommand = new Deno.Command(options.julia_cmd, {
+        args: [
+          "--startup-file=no",
+          `--project=${juliaProject}`,
+          "-e",
+          "using QuartoNotebookRunner",
+        ],
+        env: {
+          // ignore the main env
+          "JULIA_LOAD_PATH": Deno.build.os === "windows"
+            ? "@;@stdlib"
+            : "@:@stdlib",
+        },
+      });
+      const qnrTestProc = qnrTestCommand.spawn();
+      const result = await qnrTestProc.output();
+      if (!result.success) {
+        throw Error(
+          `Executing \`using QuartoNotebookRunner\` failed with QUARTO_JULIA_PROJECT="${juliaProject}". Ensure that this project exists, has QuartoNotebookRunner installed and is instantiated correctly.`,
+        );
+      }
+      trace(
+        options,
+        `QuartoNotebookRunner could be loaded successfully.`,
+      );
+    }
 
     // We need to spawn the julia server in its own process that can outlive quarto.
     // Apparently, `run(detach(cmd))` in julia does not do that reliably on Windows,
@@ -260,7 +296,7 @@ async function startOrReuseJuliaServer(
             // string array argument list, each element but the last must have a "," element after
             "--startup-file=no",
             ",",
-            `--project=${juliaRuntimeDir()}`,
+            `--project=${juliaProject}`,
             ",",
             resourcePath("julia/quartonotebookrunner.jl"),
             ",",
@@ -269,6 +305,9 @@ async function startOrReuseJuliaServer(
             "-WindowStyle",
             "Hidden",
           ],
+          env: {
+            "JULIA_LOAD_PATH": "@;@stdlib", // ignore the main env
+          },
         },
       );
       trace(
@@ -285,10 +324,13 @@ async function startOrReuseJuliaServer(
           "--startup-file=no",
           resourcePath("julia/start_quartonotebookrunner_detached.jl"),
           options.julia_cmd,
-          juliaRuntimeDir(),
+          juliaProject,
           resourcePath("julia/quartonotebookrunner.jl"),
           transportFile,
         ],
+        env: {
+          "JULIA_LOAD_PATH": "@:@stdlib", // ignore the main env
+        },
       });
       trace(
         options,

src/resources/julia/start_quartonotebookrunner_detached.jl

@@ -9,5 +9,7 @@ if length(ARGS) > 4
   error("Too many arguments")
 end
 
+env = copy(ENV)
+env["JULIA_LOAD_PATH"] = "@:@stdlib" # ignore the main env
 cmd = `$julia_bin --startup-file=no --project=$project $julia_file $transport_file`
-run(detach(cmd), wait = false)
+run(detach(setenv(cmd, env)), wait = false)