Should fix repl

2024-02-08 14:01:53 -08:00
parent fcefbd5d6d
commit f8573f58e8
2 changed files with 127 additions and 25 deletions
--- a/src/clj/auto_ap/server.clj
+++ b/src/clj/auto_ap/server.clj
@@ -1,13 +1,40 @@
 (ns auto-ap.server
-  (:require [auto-ap.handler :refer [app]]
-            [auto-ap.logging :as alog]
-            [com.unbounce.dogstatsd.core :as statsd]
-            [config.core :refer [env]]
-            [mount.core :as mount]
-            [ring.adapter.jetty :refer [run-jetty]]
-            [yang.scheduler :as scheduler])
-  (:import (org.eclipse.jetty.server.handler StatisticsHandler)
-           (org.eclipse.jetty.server.handler.gzip GzipHandler)))
+  (:gen-class)
+  (:require
+   [auto-ap.handler :refer [app]]
+   [auto-ap.jobs.restore-from-backup :as job-restore-from-backup]
+   [auto-ap.jobs.bulk-journal-import :as job-bulk-journal-import]
+   [auto-ap.jobs.close-auto-invoices :as job-close-auto-invoices]
+   [auto-ap.jobs.current-balance-cache :as job-current-balance-cache]
+   [auto-ap.jobs.ezcater-upsert :as job-ezcater-upsert]
+   [auto-ap.jobs.import-uploaded-invoices :as job-import-uploaded-invoices]
+   [auto-ap.jobs.intuit :as job-intuit]
+   [auto-ap.jobs.ntg :as job-ntg]
+   #_[auto-ap.backup :as backup]
+   [auto-ap.jobs.ledger-reconcile :as job-reconcile-ledger]
+   [auto-ap.jobs.load-historical-sales :as job-load-historical-sales]
+   [auto-ap.jobs.plaid :as job-plaid]
+   [auto-ap.jobs.register-invoice-import :as job-register-invoice-import]
+   [auto-ap.jobs.square :as job-square]
+   [auto-ap.jobs.sysco :as job-sysco]
+   [auto-ap.jobs.vendor-usages :as job-vendor-usages]
+   [auto-ap.jobs.yodlee2 :as job-yodlee2]
+   [auto-ap.logging :as alog]
+   [com.unbounce.dogstatsd.core :as statsd]
+   [config.core :refer [env]]
+   [mount.core :as mount]
+   [nrepl.server :refer [start-server]]
+   [ring.adapter.jetty :refer [run-jetty]]
+   [yang.scheduler :as scheduler]
+   [auto-ap.jobs.insight-outcome-recommendation :as insight-outcome-recommendation])
+  (:import
+   (org.eclipse.jetty.server.handler StatisticsHandler)
+   (org.eclipse.jetty.server.handler.gzip GzipHandler)))
+
+(defn add-shutdown-hook! [^Runnable f]
+  (.addShutdownHook (Runtime/getRuntime)
+                    (Thread. f)))
+

 (defn gzip-handler []
  (doto (GzipHandler.)
@@ -61,3 +88,73 @@
  :start (scheduler/every (* 1000 10) collect-jetty-stats)
  :stop (scheduler/stop jetty-stats))

+(defn shutdown-mount []
+  (mount/stop))
+
+(defn -main [& _]
+  (let [job (System/getenv "INTEGREAT_JOB")]
+    (println "JOB is" job)
+    (cond (= job "square-import-job")
+          (job-square/-main)
+
+          (= job "reconcile-ledger")
+          (job-reconcile-ledger/-main)
+
+          (= job "current-balance-cache")
+          (job-current-balance-cache/-main)
+
+          (= job "yodlee2")
+          (job-yodlee2/-main)
+
+          (= job "yodlee2-accounts")
+          (job-yodlee2/accounts-only)
+
+          (= job "plaid")
+          (job-plaid/-main)
+
+          (= job "intuit")
+          (job-intuit/-main)
+
+          (= job "vendor-usages")
+          (job-vendor-usages/-main)
+
+          (= job "import-uploaded-invoices")
+          (job-import-uploaded-invoices/-main)
+
+          (= job "sysco")
+          (job-sysco/-main)
+
+          (= job "close-auto-invoices")
+          (job-close-auto-invoices/-main)
+
+          (= job "ezcater-upsert")
+          (job-ezcater-upsert/-main)
+
+          (= job "register-invoice-import")
+          (job-register-invoice-import/-main)
+
+          (= job "load-historical-sales")
+          (job-load-historical-sales/-main)
+
+          (= job "bulk-journal-import")
+          (job-bulk-journal-import/-main)
+
+          (= job "restore-from-backup")
+          (job-restore-from-backup/-main)
+
+          (= job "insight-outcome-recommendation")
+          (insight-outcome-recommendation/-main)
+
+           ;; (= job "export-backup")
+           ;; (backup/-main)
+
+          (= job "ntg")
+          (job-ntg/-main)
+
+          :else
+          (do
+            (add-shutdown-hook! shutdown-mount)
+            (start-server :port 9000 :bind "0.0.0.0" #_#_:handler (cider-nrepl-handler))
+            (mount/start)
+            #_(alter-var-root #'nrepl.middleware.print/*print-fn* (constantly clojure.pprint/pprint))))))
+