Add find-in function

weavejester · Oct 19, 2024 · 7c36bc6 · 7c36bc6
1 parent b2dd5d0
commit 7c36bc6
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/medley/core.cljc b/src/medley/core.cljc
@@ -667,3 +667,13 @@
   (when (some? coll)
     (let [index (.indexOf coll item)]
       (when-not (neg? index) index))))
+
+(defn find-in
+  "Similar to `clojure.core/find`, except that it finds a key/value pair in an
+  nested associate structure `m`, given a sequence of keys `ks`. See also:
+  `clojure.core/get-in`."
+  {:added "1.9.0"}
+  [m ks]
+  (if (next ks)
+    (-> (get-in m (butlast ks)) (find (last ks)))
+    (find m (first ks))))
diff --git a/test/medley/core_test.cljc b/test/medley/core_test.cljc
@@ -516,3 +516,12 @@
 
   (is (= 1 (m/index-of (range 0 10) 1)))
   (is (= 1 (m/index-of (map str [:a :b]) ":b"))))
+
+(deftest test-find-in
+  (is (nil? (m/find-in {} [:a])))
+  (is (nil? (m/find-in {} [:a :b])))
+  (is (nil? (m/find-in {:a {}} [:a :b])))
+  (is (= [:a 1] (m/find-in {:a 1} [:a])))
+  (is (= [:b 2] (m/find-in {:a {:b 2}} [:a :b])))
+  (is (= [:b {:c 3}] (m/find-in {:a {:b {:c 3}}} [:a :b])))
+  (is (= [:c 3] (m/find-in {:a {:b {:c 3}}} [:a :b :c]))))