refactor(template): additional env functions & simplify methods

Author: devnote-dev

src/template/runner.cr

@@ -1,40 +1,63 @@
-module Lua
+module Geode
   class Runner
     @script : String
     @output : IO
     @state : Lua::State
+    getter? uses_shell : Bool = false
 
     def initialize(@script, @output)
       @state = Lua::State.new
-      @state.open :all
+      @state.open Lua::Library[:string, :table, :math]
     end
 
-    def load_normal_env : Nil
+    def run_normal_env : Nil
+      Lua.create_function @state, "print" do |args|
+        @output.puts args
+      end
+
       Lua.create_function @state, "current_dir" do |_|
         Dir.current
       end
 
-      Lua.create_function @state, "run_command" do |args|
-        command, *rest = args.map(&.as_s) rescue raise "expected all string arguments to 'run_command'"
-        Process.run(command, rest).exit_code
+      Lua.create_function @state, "dir_exists" do |path|
+        Dir.exists? path.as_s
+      end
+
+      Lua.create_function @state, "dir_list" do |path|
+        Dir.children path.as_s
+      end
+
+      Lua.create_function @state, "dir_make" do |path|
+        Dir.mkdir_p path.as_s
+        nil
+      end
+
+      Lua.create_function @state, "dir_remove" do |path, recurse|
+        if recurse.as_bool?
+          FileUtils.rm_rf path.as_s
+        else
+          Dir.delete path.as_s
+        end
       end
+
+      Lua.create_function @state, "run_command" do |command|
+        Process.run(command.as_s).exit_code
+      end
+
+      @state.run_string @script
     end
 
-    def load_test_env : Nil
-      Lua.create_function @state, "current_dir" do |_|
+    def run_test_env : Nil
+      Lua.create_function @state, "dir_current" do |_|
         Dir.current
       end
 
       Lua.create_function @state, "run_command" do |_|
+        @uses_shell = true
         0
       end
-    end
-
-    def load_checks_env : Nil
-      # TODO
-    end
 
-    def run : Nil
+      @state.run_string @script
     end
   end
 end

src/template/template.cr

@@ -60,11 +60,14 @@ module Geode
 
     def run_script(output : IO) : Nil
       script = File.read Config::TEMPLATES / name / "control.lua"
-      runner = Runner.new script, output
-      runner.load_standard_functions
+      runner = Geode::Runner.new script, output
+      runner.run_normal_env
     end
 
     def test_script(output : IO) : Nil
+      script = File.read Config::TEMPLATES / name / "control.lua"
+      runner = Geode::Runner.new script, output
+      runner.run_test_env
     end
   end
 end