Pass gh_token into script runner

Author: gaggle

lib/command_line.ex

@@ -15,12 +15,12 @@ defmodule ElixirScript.CommandLine do
     defstruct debug?: false, gh_token: nil, help?: false, script: nil
   end
 
-  def main(args, opts \\ []) do
+  def main(args) do
     parsed_args = parse_args!(args)
     log_level = if parsed_args.debug?, do: :debug, else: :info
     Logger.configure(level: log_level)
 
-    runner = Keyword.get(opts, :runner, &ScriptRunner.run/1)
+    runner = script_runner()
     Logger.debug("Running in debug mode, using runner: #{inspect(runner)}")
 
     Logger.debug("Environment Variables: #{inf_inspect(System.get_env())}")
@@ -32,7 +32,7 @@ defmodule ElixirScript.CommandLine do
       System.halt(0)
     else
       Logger.debug("Script input: #{inf_inspect(parsed_args.script)}")
-      result = runner.(parsed_args.script)
+      result = runner.run(parsed_args.script, github_token: parsed_args.gh_token)
       Core.set_output(result, "result")
       Logger.debug("Result output: #{inspect(result, pretty: true)}")
     end
@@ -77,4 +77,7 @@ defmodule ElixirScript.CommandLine do
   def inf_inspect(exec) do
     inspect(exec, pretty: true, limit: :infinity, printable_limit: :infinity)
   end
+
+  defp script_runner,
+    do: Application.get_env(:command_line, :script_runner, ScriptRunner)
 end

test/command_line_test.exs

@@ -1,8 +1,52 @@
 defmodule ElixirScript.CommandLineTest do
   use ExUnit.Case, async: false
+  import ExUnit.CaptureIO
+  import Mox
 
   alias ElixirScript.CommandLine
 
+  setup :verify_on_exit!
+
+  describe "main/1" do
+    test "passes script and github_token to script runner" do
+      expect(ElixirScript.ScriptRunnerMock, :run, fn "foo", [github_token: "token"] -> "bar" end)
+
+      capture_io(fn ->
+        CommandLine.main(["--script", "foo", "--gh-token", "token"])
+      end)
+    end
+
+    test "outputs the result of script runner to GitHub command" do
+      stub(ElixirScript.ScriptRunnerMock, :run, fn _, _ -> "bar" end)
+      safe_delete_env("GITHUB_OUTPUT")
+
+      io =
+        capture_io(fn ->
+          CommandLine.main(["--script", "foo", "--gh-token", "token"])
+        end)
+
+      assert io == "::set-output name=result::bar\n"
+    end
+
+    test "outputs result to GitHub environment commands if available" do
+      {:ok, file_path} = safe_temp_path()
+      safe_put_env("GITHUB_OUTPUT", file_path)
+
+      stub(ElixirScript.ScriptRunnerMock, :run, fn _, _ -> "bar" end)
+
+      capture_io(fn ->
+        CommandLine.main(["--script", "foo", "--gh-token", "token"])
+      end)
+
+      regex =
+        ~r/result<<ghadelimiter_#Ref<(?:\d|\.)+>\n(?<result>.*)\nghadelimiter_#Ref<(?:\d|\.)+>\n/
+
+      output = File.read!(file_path)
+      assert [_, result] = Regex.run(regex, output), "Regex failed on: " <> output
+      assert result == "bar"
+    end
+  end
+
   describe "parse_args!/1" do
     @script "IO.puts('Hello, world!')"
 
@@ -84,4 +128,17 @@ defmodule ElixirScript.CommandLineTest do
         else: System.delete_env(varname)
     end)
   end
+
+  def safe_delete_env(varname) do
+    original_value = System.get_env(varname)
+    System.delete_env(varname)
+
+    on_exit(fn -> if original_value, do: System.put_env(varname, original_value) end)
+  end
+
+  def safe_temp_path do
+    {:ok, file_path} = Temp.open("my-file", &IO.write(&1, ""))
+    on_exit(fn -> File.rm(file_path) end)
+    {:ok, file_path}
+  end
 end