Add update ep ids on letsc-select-start

KGOH · Mar 6, 2024 · 9dff56d · 9dff56d
1 parent 786530c
commit 9dff56d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/src/ps/sc.cljc b/src/ps/sc.cljc
@@ -61,14 +61,13 @@
    (letsc-select-start!* @letsc-select-state ep-ids body))
 
   ([select-state ep-ids body]
-   (let [{:as state
-          saved-body     :body
+   (let [{saved-body     :body
           selected-ep-id :selected-ep-id}
          select-state]
      (mark-selected-ep-id-value
        (cond
          (= saved-body body)
-         state
+         (swap! letsc-select-state assoc :ep-ids ep-ids)
 
          (contains? (into #{} (map first) ep-ids)
                     selected-ep-id)
@@ -88,16 +87,15 @@
    (letsc-select-start-no-mark!* @letsc-select-state ep-ids body))
 
   ([select-state ep-ids body]
-   (let [{:as state
-          saved-body     :body
+   (let [{saved-body     :body
           selected-ep-id :selected-ep-id}
          select-state
 
          {:keys [ep-ids selected-ep-id]}
          (->>
            (cond
              (= saved-body body)
-             state
+             (swap! letsc-select-state assoc :ep-ids ep-ids)
 
              (contains? (into #{} (map first) ep-ids)
                         selected-ep-id)
@@ -106,7 +104,7 @@
              :else
              (init-letsc-select! ep-ids body)))
 
-         selected-idx (some (fn [[idx[id v]]] (when (= selected-ep-id id) idx)) (map-indexed vector ep-ids))]
+         selected-idx (some (fn [[idx [id _]]] (when (= selected-ep-id id) idx)) (map-indexed vector ep-ids))]
      [selected-idx (map second ep-ids)])))
 
 

diff --git a/test/ps/sc_test.cljc b/test/ps/sc_test.cljc
@@ -172,7 +172,12 @@
       (sut/undefsc-lastdef)
 
       (t/is (undefined-sym? `x))
-      (t/is (undefined-sym? `i)))))
+      (t/is (undefined-sym? `i)))
+
+    (t/testing "starting on the same expr updates the list"
+      (collatz 2)
+      (t/is (= [5 16 [8] 4 2 1 2 1]
+              (sut/letsc-select-start! 'x))))))
 
 
 (t/deftest undefsc-restore-test