✨ Add improved REPL support

backend/src/app/main.clj

@@ -37,8 +37,12 @@
    [app.storage.s3 :as-alias sto.s3]
    [app.util.time :as dt]
    [app.worker :as-alias wrk]
+   [cider.nrepl :refer [cider-nrepl-handler]]
+   [clojure.test :as test]
+   [clojure.tools.namespace.repl :as repl]
    [cuerdas.core :as str]
    [integrant.core :as ig]
+   [nrepl.server :as nrepl]
    [promesa.exec :as px])
   (:gen-class))
 
@@ -527,22 +531,65 @@
                                    (merge worker-config))
                                  (ig/prep)
                                  (ig/init))))
-  (l/info :hint "welcome to penpot"
-          :flags (str/join "," (map name cf/flags))
-          :worker? (contains? cf/flags :backend-worker)
-          :version (:full cf/version)))
+  (l/inf :hint "welcome to penpot"
+         :flags (str/join "," (map name cf/flags))
+         :worker? (contains? cf/flags :backend-worker)
+         :version (:full cf/version)))
 
 (defn stop
   []
   (alter-var-root #'system (fn [sys]
                              (when sys (ig/halt! sys))
                              nil)))
+(defn restart
+  []
+  (stop)
+  (repl/refresh :after 'app.main/start))
+
+(defn restart-all
+  []
+  (stop)
+  (repl/refresh-all :after 'app.main/start))
+
+(defmacro run-bench
+  [& exprs]
+  `(do
+     (require 'criterium.core)
+     (criterium.core/with-progress-reporting (crit/quick-bench (do ~@exprs) :verbose))))
+
+(defn run-tests
+  ([] (run-tests #"^backend-tests.*-test$"))
+  ([o]
+   (repl/refresh)
+   (cond
+     (instance? java.util.regex.Pattern o)
+     (test/run-all-tests o)
+
+     (symbol? o)
+     (if-let [sns (namespace o)]
+       (do (require (symbol sns))
+           (test/test-vars [(resolve o)]))
+       (test/test-ns o)))))
+
+(repl/disable-reload! (find-ns 'integrant.core))
 
 (defn -main
   [& _args]
   (try
-    (start)
+    (let [p (promise)]
+      (when (contains? cf/flags :nrepl-server)
+        (l/inf :hint "start nrepl server" :port 6064)
+        (nrepl/start-server :bind "0.0.0.0" :port 6064 :handler cider-nrepl-handler))
+
+      (start)
+      (deref p))
     (catch Throwable cause
-      (l/error :hint (ex-message cause)
-               :cause cause)
+      (binding [*out* *err*]
+        (println "==== ERROR ===="))
+      (.printStackTrace cause)
+      (when-let [cause' (ex-cause cause)]
+        (binding [*out* *err*]
+          (println "==== CAUSE ===="))
+        (.printStackTrace cause'))
+      (px/sleep 500)
       (System/exit -1))))

backend/src/app/srepl.clj

@@ -36,7 +36,9 @@
         lock (locks/create)]
     (ccs/prepl *in*
                (fn [m]
-                 (binding [*out* out, *flush-on-newline* true, *print-readably* true]
+                 (binding [*out* out,
+                           *flush-on-newline* true,
+                           *print-readably* true]
                    (locks/locking lock
                      (println (json/encode-str m))))))))
 
@@ -44,13 +46,10 @@
 
 (s/def ::port ::us/integer)
 (s/def ::host ::us/not-empty-string)
-(s/def ::flag #{:urepl-server :prepl-server})
-(s/def ::type #{::prepl ::urepl})
-(s/def ::key (s/tuple ::type ::us/keyword))
 
 (defmethod ig/pre-init-spec ::server
   [_]
-  (s/keys :req [::flag ::host ::port]))
+  (s/keys :req [::host ::port]))
 
 (defmethod ig/prep-key ::server
   [[type _] cfg]
@@ -59,6 +58,12 @@
 (defmethod ig/init-key ::server
   [[type _] {:keys [::flag ::port ::host] :as cfg}]
   (when (contains? cf/flags flag)
+
+    (l/inf :hint "initializing repl server"
+           :name (name type)
+           :port port
+           :host host)
+
     (let [accept (case type
                    ::prepl 'app.srepl/json-repl
                    ::urepl 'app.srepl/user-repl)
@@ -67,14 +72,8 @@
                   :name (name type)
                   :accept accept}]
 
-      (l/info :msg "initializing repl server"
-              :name (name type)
-              :port port
-              :host host)
-
       (ccs/start-server params)
-
-      params)))
+      (assoc params :type type))))
 
 (defmethod ig/halt-key! ::server
   [_ params]

backend/src/app/srepl/main.clj

@@ -31,6 +31,7 @@
    [app.util.time :as dt]
    [app.worker :as wrk]
    [clojure.pprint :refer [pprint print-table]]
+   [clojure.tools.namespace.repl :as repl]
    [cuerdas.core :as str]))
 
 (defn print-available-tasks