unixify v2

[babashka-scripts.git] / lib.clj

diff --git a/lib.clj b/lib.clj
index 1f88070ba456a7b9e516567a17180a9c3a8cf26c..d0f3b404032016665bab2b80cdb2b24a7fe4b64c 100644 (file)
--- a/lib.clj
+++ b/lib.clj
@@ -1,10 +1,11 @@
 ; Common utility functions used by scripts
 ; Common utility functions used by scripts

+; Unlike the functions in utils.clj, these are not directly useful

 
 (ns lib
   (:require [babashka.process :as p]
             [org.httpkit.client :as http]
             [clojure.java.io :as io]
 
 (ns lib
   (:require [babashka.process :as p]
             [org.httpkit.client :as http]
             [clojure.java.io :as io]

-            [clojure.string :refer [replace-first]])
+            [clojure.string :as str])

   (:import [java.nio.file Files FileSystems CopyOption StandardCopyOption]))
 
 (defn run-cmd
   (:import [java.nio.file Files FileSystems CopyOption StandardCopyOption]))
 
 (defn run-cmd


@@ -27,9 +28,35 @@
 (defn expand-home
   [s]
   (if (.startsWith s "~")
 (defn expand-home
   [s]
   (if (.startsWith s "~")

-    (replace-first s "~" (System/getProperty "user.home"))
+    (str/replace-first s "~" (System/getProperty "user.home"))

     s))
 
 (defn download-binary
   [url destination]
   (io/copy (:body @(http/get url)) (io/file destination)))
     s))
 
 (defn download-binary
   [url destination]
   (io/copy (:body @(http/get url)) (io/file destination)))

+
+(defn unixify
+  "Make a function `f` behave like a UNIX shell command.
+
+  Examples:
+
+  # Multiple arguments
+  $ install-deb url1 url2 url3
+
+  # Read from file with one url on each line
+  $ install-deb `cat urls.txt`
+
+  # Piping from another process
+  $ ls | install-deb  # Note: install-deb doesn't work with ls
+  "
+  [f]
+  (when (= *file* (System/getProperty "babashka.file"))
+    (if (> (.available System/in) 0)
+      (->> (slurp *in*)
+           (str/split-lines)
+           (filter seq)
+           (run! f))
+      (->> (vec *command-line-args*)
+           (map #(str/split % #" "))
+           (flatten)
+           (run! f)))))