reitit/test/cljc/ctrl/merge.cljc
2023-05-21 18:15:44 +03:00

120 lines
4 KiB
Clojure

(ns ctrl.merge
(:refer-clojure :exclude [merge])
(:require [clojure.core :as c]
[clojure.set :as set]))
(defn- meta* [obj]
(if #?(:clj (instance? clojure.lang.IObj obj)
:cljs (satisfies? IMeta obj))
(meta obj)))
(defn- with-meta* [obj m]
(if #?(:clj (instance? clojure.lang.IObj obj)
:cljs (satisfies? IWithMeta obj))
(with-meta obj m)
obj))
(defn- displace? [obj]
(-> obj meta* :displace))
(defn- replace? [obj]
(-> obj meta* :replace))
(defn- top-displace? [obj]
(-> obj meta* :top-displace))
(defn- different-priority? [left right]
(boolean (or (some (some-fn nil? displace? replace?) [left right])
(top-displace? left))))
(defn- remove-top-displace [obj {:keys [::replace-nil]}]
(cond replace-nil nil
(top-displace? obj) obj
:else (vary-meta obj dissoc :top-displace)))
(defn- pick-prioritized [left right options]
(cond (nil? left) right
(nil? right) (remove-top-displace left options)
(top-displace? left) right
(and (displace? left) ;; Pick the rightmost
(displace? right)) ;; if both are marked as displaceable
(with-meta* right
(c/merge (meta* left) (meta* right)))
(and (replace? left) ;; Pick the rightmost
(replace? right)) ;; if both are marked as replaceable
(with-meta* right
(c/merge (meta* left) (meta* right)))
(or (displace? left)
(replace? right))
(with-meta* right
(c/merge (-> left meta* (dissoc :displace))
(-> right meta* (dissoc :replace))))
(or (replace? left)
(displace? right))
(with-meta* left
(c/merge (-> right meta* (dissoc :displace))
(-> left meta* (dissoc :replace))))))
(defn find-custom-merge [path path-map]
(letfn [(match [x f] (cond (keyword? f) (= x f) (or (fn? f) (ifn? f)) (f x)))]
(reduce (fn [_ [ps f]] (let [match (loop [[p & pr] path, [pp & ppr] ps]
(cond (and p pp (match p pp)) (recur pr ppr)
(= nil p pp) true))]
(when match (reduced f)))) nil path-map)))
(defrecord Acc [data])
(defn accumulate? [x] (instance? Acc x))
(defn unaccumulate [x] (if (accumulate? x) (:data x) x))
(defn accumulate
([x] (if (accumulate? x) x (->Acc [x])))
([x y] (update (accumulate x) :data into (unaccumulate y))))
;;
;; public api
;;
(defn merge
([] {})
([left] left)
([left right] (merge left right nil))
([left right {:keys [::path ::path-map] :as options}]
(let [custom-merge (find-custom-merge path path-map)]
(cond
(different-priority? left right)
(pick-prioritized left right options)
(accumulate? left)
(accumulate left right)
custom-merge
(custom-merge left right options)
(and (map? left) (map? right))
(let [merge-entry (fn [m e]
(let [k (key e) v (val e)]
(if (contains? m k)
(assoc m k (merge (get m k) v (-> options
(update ::path (fnil conj []) k)
(update ::acc assoc (or path []) m))))
(assoc m k v))))
merge2 (fn [m1 m2]
(reduce merge-entry (or m1 {}) (seq m2)))]
(reduce merge2 [left right]))
(and (set? left) (set? right))
(set/union right left)
(and (coll? left) (coll? right))
(if (or (-> left meta :prepend)
(-> right meta :prepend))
(-> (into (empty left) (concat right left))
(with-meta (c/merge (meta left)
(select-keys (meta right) [:displace]))))
(into (empty left) (concat left right)))
:else right))))