add set-elem navigator and change semantics of map-key to only navigate if the key exists (for consistency with set-elem)

CHANGES.md

@@ -2,6 +2,7 @@
 
 * Added `pred=`, `pred<`, `pred>`, `pred<=`, `pred>=` for filtering using common comparisons
 * Add `map-key` navigator
+* Add `set-elem` navigator
 
 ## 1.0.1
 

src/clj/com/rpl/specter.cljc

@@ -869,17 +869,37 @@
                             structure)))))
 
 (defrichnav
-  ^{:doc "Navigates to the given key in the map (not to the value). Navigates regardless
+  ^{:doc "Navigates to the given key in the map (not to the value). Navigates only if the
-          of whether the key currently exists in the map. For transforms, will assoc nil
+          key currently exists in the map."}
-          value if key doesn't currently exist in map."}
   map-key
   [key]
   (select* [this vals structure next-fn]
-    (next-fn vals key))
+    (if (contains? structure key)
+      (next-fn vals key)
+      NONE
+      ))
   (transform* [this vals structure next-fn]
-    (let [newkey (next-fn vals key)
+    (if (contains? structure key)
-          oldval (get structure key)]
+      (let [newkey (next-fn vals key)
-      (-> structure (dissoc key) (assoc newkey oldval))
+            oldval (get structure key)]
+        (-> structure (dissoc key) (assoc newkey oldval))
+        )
+      structure
+      )))
+
+(defrichnav
+  ^{:doc "Navigates to the given element in the set only if it exists in the set."}
+  set-elem
+  [elem]
+  (select* [this vals structure next-fn]
+    (if (contains? structure elem)
+      (next-fn vals elem)
+      NONE
+      ))
+  (transform* [this vals structure next-fn]
+    (if (contains? structure elem)
+      (-> structure (disj elem) (conj (next-fn vals elem)))
+      structure
       )))
 
 (def ^{:doc "Navigate to the specified keys one after another. If navigate to NONE,

test/com/rpl/specter/core_test.cljc

@@ -1528,11 +1528,17 @@
     ))
 
 (deftest map-key-test
-  (is (= {:b nil :c 3} (setval (s/map-key :a) :b {:c 3})))
+  (is (= {:c 3} (setval (s/map-key :a) :b {:c 3})))
   (is (= {:b 2} (setval (s/map-key :a) :b {:a 2})))
   (is (= {:b 2} (setval (s/map-key :a) :b {:a 2 :b 1})))
   )
 
+(deftest set-elem-test
+  (is (= #{:b :d} (setval (s/set-elem :a) :x #{:b :d})))
+  (is (= #{:x :a} (setval (s/set-elem :b) :x #{:b :a})))
+  (is (= #{:a} (setval (s/set-elem :b) :a #{:b :a})))
+  )
+
 #?(:clj
   (do
     (defprotocolpath FooPP)