Switch to Figwheel REPL

Having trouble getting the built-in browser REPL to pick everything up
correctly.

This figwheel REPL seems to work pretty well, though it relies on you
having a preload to do the figwheel connection.

Author: ajoberstar

build.gradle.kts

@@ -34,6 +34,14 @@ dependencies {
   compileOnly("nrepl:nrepl:0.9.0")
   compileOnly("cider:piggieback:0.5.3")
   devImplementation("cider:piggieback:0.5.3")
+
+  compileOnly("com.bhauman:figwheel-repl:0.2.18")
+  devImplementation("com.bhauman:figwheel-repl:0.2.18")
+
+  compileOnly("ring:ring-jetty-adapter:1.9.5")
+  devImplementation("ring:ring-jetty-adapter:1.9.5")
+  compileOnly("org.eclipse.jetty.websocket:websocket-server:9.4.7.v20180619")
+  devImplementation("org.eclipse.jetty.websocket:websocket-server:9.4.7.v20180619")
 }
 
 tasks.withType<Test>() {

src/main/clojure/dev/clojurephant/tooling/api.clj

@@ -1,34 +1,46 @@
 (ns dev.clojurephant.tooling.api
   (:require [dev.clojurephant.tooling.core :as core]
+            [clojure.edn :as edn]
             [clojure.string :as string]))
 
 (defonce db (atom {}))
 
-(defn connect [dir]
-  (swap! db update :con
-         (fn [existing-con]
-           (if existing-con
-             (throw (ex-info "Must close existing connection" {}))
-             (core/connect dir)))))
-
-(defn close []
-  (swap! db update :con
-         (fn [existing-con]
-           (when existing-con
-             (.close existing-con)
-             nil))))
-
-(defn model []
+(defn connect! [project-dir]
+  (swap! db (fn [{:keys [dir connection] :as current-db}]
+              (cond
+                (nil? connection)
+                (merge current-db
+                       {:dir project-dir
+                        :connection (core/connect project-dir)})
+
+                (= dir project-dir)
+                current-db
+
+                :else
+                (throw (ex-info "Must close existing connection" {:dir dir})))))
+  nil)
+
+(defn close! []
+  (swap! db (fn [{:keys [connection]}]
+              (when connection
+                (.close connection))
+              nil))
+  nil)
+
+(defn reload-model! []
   (swap! db (fn [current-db]
-              (if-let [con (:con current-db)]
+              (if-let [con (:connection current-db)]
                 (let [m (core/wait (core/clojurephant-model con))]
                   (if (= :success (:result m))
-                    (assoc current-db :model (read-string (.getEdn (:value m))))
+                    (assoc current-db :model (edn/read-string (.getEdn (:value m))))
                     (throw (ex-info "Failed to get Clojurephant model"
                                     (select-keys m [:error])))))
                 (throw (ex-info "Cannot get model until connect" {}))))))
 
-(defn reroot-cljs-build [build new-dir]
+(defn repl-output-dir []
+  (-> @db :model :repl :dir))
+
+(defn ^:private reroot-cljs-build [build new-dir]
   (let [old-dir (:output-dir build)
         replacer (fn [dir]
                    (cond
@@ -40,13 +52,13 @@
         (update-in [:output-to] replacer)
         (update-in [:source-map] replacer))))
 
-(defn repl-cljs-builds []
+(defn cljs-all-build-opts []
   (let [m (:model @db)
-        repl-dir (-> m :repl :dir)
+        repl-dir (repl-output-dir)
         builds (:clojurescript m)]
     (into {} (map (fn [[id build]]
                  [id (reroot-cljs-build build repl-dir)]))
           builds)))
 
-(defn repl-cljs-build [id]
-  (get (repl-cljs-builds) id))
+(defn cljs-build-opts [id]
+  (get (cljs-all-build-opts) id))

src/main/clojure/dev/clojurephant/tooling/cljs.clj

@@ -0,0 +1,53 @@
+(ns dev.clojurephant.tooling.cljs
+  (:require [dev.clojurephant.tooling.api :as api]
+            [cljs.analyzer.api :as ana]
+            [cljs.build.api :as build]
+            [cljs.closure :as closure]
+            [cljs.repl :as repl]))
+
+(def registry (atom {}))
+
+(defn build-opts [id]
+  (api/cljs-build-opts id))
+
+(defn compiler-env! [id]
+  (let [opts (build-opts id)]
+    (-> registry
+        (swap! update-in [id :compiler-env]
+               (fn [env] (or env (ana/empty-state opts))))
+        (get-in [id :compiler-env]))))
+
+(defn build! [id]
+  (let [opts (build-opts id)
+        compiler-env (compiler-env! id)]
+    (build/build nil opts compiler-env)))
+
+(defn watch! [id]
+  (let [opts (build-opts id)
+        compiler-env (compiler-env! id)
+        stop (promise)]
+    (swap! registry assoc-in [id :watch-stop] stop)
+    (build/watch nil opts compiler-env stop)))
+
+(defn stop-watch! [id]
+  (swap! registry update-in [id :watch-stop]
+         (fn [stop] (when stop (deliver stop true))))
+  nil)
+
+(defn clean! [id]
+  (let [opts (build-opts id)]
+    (stop-watch! id)
+    (swap! registry dissoc id)
+    ;; TODO delete output-dir and output-to
+    ))
+
+(defn repl-env!
+  ([id]
+   (get-in registry [id :repl-env]))
+  ([id env]
+   (swap! registry update-in [id :repl-env]
+          (fn [old-env]
+            (when old-env
+              (repl/tear-down old-env))
+            env))
+   env))

src/main/clojure/dev/clojurephant/tooling/figwheel_repl.clj

@@ -0,0 +1,21 @@
+(ns dev.clojurephant.tooling.figwheel-repl
+  (:require [dev.clojurephant.tooling.api :as api]
+            [dev.clojurephant.tooling.cljs :as cljs]
+            [cider.piggieback :as piggie]
+            [figwheel.repl :as repl]))
+
+(defn repl-env [id]
+    (let [opts (cljs/build-opts id)
+          env (repl/repl-env* opts)]
+      (cljs/repl-env! id env)))
+
+(defn cljs-repl [id]
+  (piggie/cljs-repl (repl-env id)))
+
+(defn start
+  ([] (start :dev))
+  ([id]
+   (api/connect! ".")
+   (api/reload-model!)
+   (cljs/build! id)
+   (cljs-repl id)))