map-traverse_tree-seq.clj

(ns io.djui.misc)

;; Use case example for tree-seq
(defn- map-traverse
  "Traverse a map with vectors in values as branches and return a list of key
  x's values. Example:
  {:a nil :x {:foo 1} :b [{:c nil :x 42} \"test\"]} => [{:foo 1} 42]"
  [map-tree x]
  (let [map-vals #(when (map? %) (vals %))
        branch? (comp (partial some sequential?) map-vals)
        children (comp flatten (partial filter sequential?) vals)
        nodes (tree-seq branch? children map-tree)]
    (keep x nodes)))