feat(aot): support command line args in main fn

Author: Samy-33

compiler+runtime/include/cpp/jank/c_api.h

@@ -300,6 +300,8 @@ extern "C"
                 jank_bool init_default_ctx,
                 int (*fn)(int const, char const **));
 
+  jank_object_ref jank_parse_command_line_args(int const argc, char const **argv);
+
 #ifdef __cplusplus
 }
 #endif

compiler+runtime/include/cpp/jank/util/cli.hpp

@@ -54,5 +54,12 @@ namespace jank::util::cli
     command command{ command::repl };
   };
 
+  struct empty_options
+  {
+    /* TODO: Use a native_persistent_vector instead. */
+    std::vector<native_transient_string> opts;
+  };
+
   jtl::result<options, int> parse(int const argc, char const **argv);
+  empty_options parse_empty(int const argc, char const **argv);
 }

compiler+runtime/src/cpp/jank/aot/processor.cpp

@@ -72,7 +72,8 @@ extern "C" jank_object_ref jank_load_jank_compiler_native();
 extern "C" jank_object_ref jank_load_clojure_core();
 extern "C" jank_object_ref jank_var_intern_c(char const *, char const *);
 extern "C" jank_object_ref jank_deref(jank_object_ref);
-extern "C" jank_object_ref jank_call0(jank_object_ref);
+extern "C" jank_object_ref jank_call2(jank_object_ref, jank_object_ref, jank_object_ref);
+extern "C" jank_object_ref jank_parse_command_line_args(int, char const **);
 )");
 
     for(auto const &m : __rt_ctx->loaded_modules_in_order)
@@ -99,9 +100,14 @@ int main(int argc, const char** argv)
       util::format_to(sb, "{}();\n", module::module_to_load_function(m));
     }
 
+    sb(R"(auto const apply{ jank_var_intern_c("clojure.core", "apply") };)");
+    sb("\n");
+    sb(R"(auto const command_line_args{ jank_parse_command_line_args(argc, argv) };)");
+    sb("\n");
+
     util::format_to(sb, R"(auto const fn(jank_var_intern_c("{}", "-main"));)", module);
     sb("\n");
-    sb(R"(jank_call0(jank_deref(fn));
+    sb(R"(jank_call2(jank_deref(apply), jank_deref(fn), command_line_args);
 
     return 0;
 
@@ -150,8 +156,7 @@ int main(int argc, const char** argv)
     auto const clang_inferred_path{ llvm::sys::findProgramByName("clang++") };
     if(!clang_inferred_path)
     {
-      return jtl::err(
-        compiler_err{ 1, "clang++ executable not found. Ensure it exists on the path!" });
+      return compiler_err{ 1, "clang++ executable not found. Ensure it exists on the path!" };
     }
     clang::driver::Driver driver(clang_inferred_path.get(),
                                  target_triple,
@@ -231,7 +236,7 @@ int main(int argc, const char** argv)
 
     if(!C || C->containsError())
     {
-      return jtl::err(compiler_err{ 1, "Failed to build compilation steps." });
+      return compiler_err{ 1, "Failed to build compilation steps." };
     }
 
     /* Execute the compilation jobs (preprocess, compile, assemble)
@@ -260,7 +265,7 @@ int main(int argc, const char** argv)
 
     if(diags.hasErrorOccurred())
     {
-      return jtl::err(compiler_err{ 1, "Compilation failed with errors.\n" });
+      return compiler_err{ 1, "Compilation failed with errors.\n" };
     }
 
     if(Result == 0)
@@ -271,7 +276,7 @@ int main(int argc, const char** argv)
     }
     else
     {
-      return jtl::err(compiler_err{ Result, "Compilation command execution failed.\n" });
+      return compiler_err{ Result, "Compilation command execution failed.\n" };
     }
 
     return jtl::ok();

compiler+runtime/src/cpp/jank/c_api.cpp

@@ -995,4 +995,17 @@ extern "C"
 
     return 0;
   }
+
+  jank_object_ref jank_parse_command_line_args(int const argc, char const **argv)
+  {
+    obj::transient_vector trans;
+    auto const opts{ util::cli::parse_empty(argc, argv).opts };
+
+    for(auto &arg : opts)
+    {
+      trans.conj_in_place(make_box(arg));
+    }
+
+    return trans.to_persistent().erase();
+  }
 }

compiler+runtime/src/cpp/jank/util/cli.cpp

@@ -131,4 +131,20 @@ namespace jank::util::cli
 
     return ok(opts);
   }
+
+  empty_options parse_empty(int const argc, char const **argv)
+  {
+    CLI::App cli{ "jank default cli" };
+    empty_options opts;
+    cli.allow_extras();
+
+    cli.parse(argc, argv);
+
+    if(cli.remaining_size() > 0)
+    {
+      opts.opts = cli.remaining();
+    }
+
+    return opts;
+  }
 }