check-stage-match.clj

!/usr/bin/env bb

(require '[clojure.java.shell :refer [sh]])


(def expected (get (System/getenv "STAGE") 
                { "dev" "12345" 
                "prod" "123456"}))

(def curr-account
  ((json/parse-string 
    ((sh "aws" "sts" "get-caller-identity") :out) true) :Account))

(if (= expected curr-account)
      (println "stage matches account")
      ((fn [] (println (format "error: wrong stage for account %s" curr-account))
        (System/exit -1))))

foo.clj

(require '[babashka.process :as p]
         '[cheshire.core :as json])

(defn main
  []
  (let [accounts     {"dev"  "12345"
                      "prod" "123456"}
        expected     (get (System/getenv "STAGE") accounts)
        curr-account (-> (p/sh "aws sts get-caller-identity")
                         p/check ; aborts if non-zero exit
                         :out
                         (json/parse-string true)
                         :Account)]
    (if (= expected curr-account)
      (println "stage matches account")
      (do
        (println (str "error: wrong stage for account " curr-account))
        (System/exit -1)))))

(when (= *file* (System/getProperty "babashka.file")) ; same as python's if __name__ == "__main__"
  (main))