Execute *main-cli-fn* if set and no -m (fixes #235)

anmonteiro · Aug 22, 2017 · 9e83cde · 9e83cde
1 parent ad5a87d
commit 9e83cde
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/cljs/snapshot/lumo/repl.cljs b/src/cljs/snapshot/lumo/repl.cljs
@@ -1149,6 +1149,11 @@
                     (handle-error e true)))))))))
     nil))
 
+(defn- ^:export run-main-cli-fn
+  []
+  (when (fn? *main-cli-fn*)
+    (apply *main-cli-fn* *command-line-args*)))
+
 (defn- ^:export get-current-ns [session-id]
   (let [{:keys [ns]} (get @session-states session-id @default-session-state)]
     (str ns)))

diff --git a/src/js/cljs.js b/src/js/cljs.js
@@ -460,6 +460,17 @@ async function startClojureScriptEngine(opts: CLIOptsType): Promise<mixed> {
 
     startREPL(opts);
   }
+
+  if (!mainNsName && !repl) {
+    initClojureScriptEngine(opts);
+    execute(
+      "(require 'lumo.repl) (lumo.repl/run-main-cli-fn)",
+      'text',
+      true,
+      false,
+      0,
+    );
+  }
 }
 
 export default startClojureScriptEngine;