From dbb28b86e14a9372bfc85f7ce8aa36e453030e5d Mon Sep 17 00:00:00 2001
From: Benni Schwerdtner <Benjamin.Schwerdtner@gmail.com>
Date: Fri, 7 Apr 2023 12:02:25 +0200
Subject: [PATCH] Add middleware arg #1529

---
 src/babashka/main.clj       | 16 ++++++++++++----
 test/babashka/main_test.clj |  2 ++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/babashka/main.clj b/src/babashka/main.clj
index 31b04037..cc82b370 100644
--- a/src/babashka/main.clj
+++ b/src/babashka/main.clj
@@ -313,8 +313,11 @@ Use bb run --help to show this help output.
 (defn start-socket-repl! [address ctx]
   (socket-repl/start-repl! address ctx))
 
-(defn start-nrepl! [address]
-  (let [opts (nrepl-server/parse-opt address)]
+(defn start-nrepl! [user-middleware address]
+  (let [user-middleware (when-not (str/blank? user-middleware)
+                     (read-string user-middleware))
+        opts (nrepl-server/parse-opt address)
+        opts (assoc opts :user-middleware user-middleware)]
     (babashka.impl.nrepl-server/start-server! opts))
   (binding [*out* *err*]
     (println "For more info visit: https://book.babashka.org/#_nrepl"))
@@ -645,6 +648,11 @@ Use bb run --help to show this help output.
             (recur options
                    (assoc opts-map
                           :nrepl (or opt "1667"))))
+          ("--middlware")
+          (let [options (next options)]
+            (recur (next options)
+                   (assoc opts-map
+                          :middleware (first options))))
           ("--eval", "-e")
           (let [options (next options)
                 opts-map (assoc opts-map :prn true)]
@@ -839,7 +847,7 @@ Use bb run --help to show this help output.
                     :main :uberscript :describe?
                     :jar :uberjar :clojure
                     :doc :run :list-tasks
-                    :print-deps :prepare]
+                    :print-deps :prepare :middleware]
              exec-fn :exec}
             cli-opts
             print-result? (:prn cli-opts)
@@ -1018,7 +1026,7 @@ Use bb run --help to show this help output.
                        describe?
                        [(print-describe) 0]
                        repl [(repl/start-repl! sci-ctx) 0]
-                       nrepl [(start-nrepl! nrepl) 0]
+                       nrepl [(start-nrepl! middleware nrepl) 0]
                        uberjar [nil 0]
                        list-tasks [(tasks/list-tasks sci-ctx) 0]
                        print-deps [(print-deps/print-deps (:print-deps-format cli-opts)) 0]
diff --git a/test/babashka/main_test.clj b/test/babashka/main_test.clj
index 5057b659..ae1b6815 100644
--- a/test/babashka/main_test.clj
+++ b/test/babashka/main_test.clj
@@ -25,6 +25,8 @@
 (deftest parse-opts-test
   (is (= "1667"
          (:nrepl (parse-opts ["--nrepl-server"]))))
+  (is (= "[foo]"
+         (:middleware (parse-opts ["--middlware" "[foo]"]))))
   (is (= "1666"
          (:socket-repl (parse-opts ["--socket-repl"]))))
   (is (= {:nrepl "1667", :classpath "src"}