Add babashka CLI explorations

Author: davehughes

bb/README.md

@@ -0,0 +1,11 @@
+Basic scripting
+* Entrypoints are defined as tasks in bb.edn.
+
+```
+bb nrepl-server 1667
+```
+
+Nice walkthrough of conjure:
+```
+:ConjureSchool
+```

bb/bb.edn

@@ -1,6 +1,3 @@
 {:paths [""]
- :tasks {
-    default (prn "Hello, World!")
-    findenv hello/venv-test
-  }
- }
+ :tasks {venv-find-activate-script dot.cli/venv-find-activate-script
+         venv-find dot.cli/venv-find}}

bb/bb.sh

@@ -0,0 +1,10 @@
+function source-env {
+  local from_dir=${1:-$PWD}
+  local activate_file="$(bb venv-find-activate-script ${from_dir})"
+  if [[ -f $activate_file ]]; then
+    echo "Sourcing ${activate_file}"
+    source $activate_file
+  else
+    echo "No virtualenv found"
+  fi
+}

bb/dot/cli.clj

@@ -0,0 +1,32 @@
+(ns dot.cli
+  (:require [clojure.tools.cli :refer [parse-opts]]
+            [clojure.pprint]
+            [dot.venv]))
+
+;; CLI adapters
+(defn venv-find-activate-script [& args]
+  (let [path (first args)]
+    (-> path dot.venv/find-env :activate-path print)))
+
+(defn venv-find [& args]
+  (let [path (first args)]
+    (-> path dot.venv/find-env clojure.pprint/pprint)))
+
+(def cli-options
+  ;; An option with a required argument
+  [["-p" "--port PORT" "Port number"
+    :default 80
+    :parse-fn #(Integer/parseInt %)
+    :validate [#(< 0 % 0x10000) "Must be a number between 0 and 65536"]]
+   ;; A non-idempotent option (:default is applied first)
+   ["-v" nil "Verbosity level"
+    :id :verbosity
+    :default 0
+    :update-fn inc] ; Prior to 0.4.1, you would have to use:
+                   ;; :assoc-fn (fn [m k _] (update-in m [k] inc))
+   ;; A boolean option defaulting to nil
+   ["-h" "--help"]])
+
+(defn -main [& args]
+  (let [opts (parse-opts args cli-options)]
+    (print opts)))

bb/dot/util.clj

@@ -0,0 +1,10 @@
+(ns dot.util
+  (:require [babashka.fs :as fs]))
+
+(defn pwd []
+  (System/getProperty "user.dir"))
+
+(defn ancestors-and-self [path]
+  (if (nil? path)
+    []
+    (lazy-seq (cons (str path) (ancestors-and-self (fs/parent path))))))

bb/dot/venv.clj

@@ -0,0 +1,38 @@
+(ns dot.venv
+  (:require [dot.util :refer [pwd ancestors-and-self]]
+            [babashka.fs :as fs]
+            [babashka.process :refer [shell]]))
+
+(defn- resolve-path [path]
+  (if (nil? path) (pwd) path))
+
+(defn dir-candidates [dirseq]
+  (for [dir dirseq
+        env-dir-name [".venv" "venv" "env"]]
+    (str dir "/" env-dir-name)))
+
+(defn env-binary-path [env-path binary]
+  (str env-path "/bin/" binary))
+
+(defn python-site-packages-dir [python-exe]
+  (let [python-exe (if (nil? python-exe) "python" python-exe)
+        python-code "import site; print(site.getsitepackages()[-1])"
+        process (shell {:out :string :err :string} python-exe "-c" python-code)]
+    (:out process)))
+
+(defn find-env [path]
+  (let [path (resolve-path path)
+        dirseq (ancestors-and-self path)
+        env (->> dirseq
+                 (dir-candidates)
+                 (filter #(fs/exists? (env-binary-path % "activate")))
+                 first)]
+    (if env
+      {:root (str (fs/parent env))
+       :venv-path env
+       :activate-path (env-binary-path env "activate")
+       :python-exe-path (env-binary-path env "python")
+       :site-packages-dir (python-site-packages-dir (env-binary-path env "python"))}
+      nil)))
+
+;; (python-site-packages-dir nil)