improve watch-ci task

Author: escherize

bb.edn

@@ -83,21 +83,11 @@
                                                  :choices-doc "list of branches"
                                                  :prompt :select})
                     branch (if (= "current" branch)
-                             (let [current (str/trim
-                                             (:out (shell {:dir (t/env "MB_DIR" (fn []
-                                                                                  (println "Please set MB_DIR to your metabase repo!")
-                                                                                  (System/exit 1))) :out :string}
-                                                          "git rev-parse --abbrev-ref HEAD")))]
-                               (do (println (c/green (c/bold "Checking current branch: " current)))
+                             (let [current (t/current-branch)]
+                               (do (println (c/green (c/bold "Using current branch: " (t/current-branch))))
                                    current))
                              branch)]
-                (pp/pprint dl/pretty)
-                (loop []
-                  (let [check (dl/checks-for-branch branch)]
-                    (print (str "\n[ " (.format (java.text.SimpleDateFormat. "hh:mm:ss a") (java.util.Date.)) " ]"))
-                    (doseq [[status count] check] (print (str/join (repeat count (dl/pretty status)))) (print "|")) (flush)
-                    (when-not (= (keys check) [:success])
-                      (Thread/sleep 10000) (recur))))))}
+                (bb.watch-ci/branch branch)))}
 
          install-autotab {:doc "Prints shell code to autocomplete tasks using bb.
                         Note: for fish shell please make sure ~/.config/fish/completions exists."

bb/tasks.clj

@@ -61,7 +61,7 @@
        (print (c/white "="))
        (println (c/cyan value))))))
 
-(defn mb-env []
+(defn mb-dir []
   (env "MB_DIR" (fn []
                   (println (c/red "Please put the path of your metabase repository into the MB_DIR env variable like so:"))
                   (println (c/white "export MB_DIR=path/to/metabase"))
@@ -99,3 +99,10 @@
         _ (bencode/write-bencode out {"op" "eval" "code" expr})
         bytes (get (bencode/read-bencode in) "value")]
     (String. bytes)))
+
+(defn current-branch
+  "Returns current metabase repo branch, as given by MB_DIR"
+  []
+  (str/trim
+    (:out (shell {:dir (mb-dir) :out :string}
+                 "git rev-parse --abbrev-ref HEAD"))))

bb/watch_ci.clj

@@ -0,0 +1,49 @@
+(ns bb.watch-ci
+  (:require [babashka.process :refer [shell]]
+            [bask.colors :as c]
+            [bb.cli :as cli]
+            [bb.tasks :as t]
+            [clojure.string :as str]
+            [bb.dl-and-run :as dl]
+            [clojure.pprint :as pp]
+            [table.core :as table]))
+
+(defn checks-for-branch [branch]
+  (->> (str "https://api.github.com/repos/metabase/metabase/commits/" branch "/check-runs")
+       dl/gh-get
+       :check_runs
+       (mapv #(select-keys % [:conclusion :name :html_url]))
+       (mapv #(update % :conclusion (fnil keyword "in-progress")))
+       (sort-by :conclusion)))
+
+(defn tc [color & s]
+  (let [cm {:bold 1
+            :gray 30 :grey 30 :red 31 :green 32 :yellow 33
+            :blue 34 :magenta 35 :cyan 36 :white 37
+            :on-gray 40 :on-grey 40 :on-red 41 :on-green 42 :on-yellow 43
+            :on-blue 44 :on-magenta 45 :on-cyan 46 :on-white 47}]
+    (if-let [c (get cm color)]
+      (str "[" c "m" (str/join s) "\033[0m")
+      (str/join s))))
+
+(defn colorize-line [line]
+  (or (first (for [[re color] [[#"success" :green] [#"in-progress" :cyan] [#"failure" :red] [#"skipped" :yellow]]
+                   :when (re-find re line)]
+               (tc color line)))
+      line))
+
+(defn branch [branch]
+  (loop []
+    (let [checks (checks-for-branch branch)]
+      (println (tc :red (tc :bold branch)) (str "[ " (.format (java.text.SimpleDateFormat. "hh:mm:ss a") (java.util.Date.)) " ]"))
+      (->> (table/table-str checks)
+           str/split-lines
+           (map colorize-line)
+           (str/join "\n")
+           println)
+      (if (= #{:success} (set (map :conclusion checks)))
+        (do
+          (println (tc :green branch "passed."))
+          (System/exit 0))
+        (do (Thread/sleep 10000)
+            (recur))))))