Support overrides on coerce-structure

wilkerlucio · Aug 3, 2017 · ca513c2 · ca513c2
1 parent 2710118
commit ca513c2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 7 deletions.
diff --git a/src/spec_coerce/core.cljc b/src/spec_coerce/core.cljc
@@ -273,14 +273,18 @@
                :coercion any?)
   :ret qualified-keyword?)
 
-(defn coerce-structure [x]
+(defn coerce-structure
   "Recursively coerce map values on a structure."
-  (walk/prewalk (fn [x]
-                  (if (map? x)
-                    (with-meta (into {} (map (fn [[k v]] [k (coerce k v)])) x)
-                               (meta x))
-                    x))
-                x))
+  ([x] (coerce-structure x {}))
+  ([x {::keys [overrides]}]
+    (walk/prewalk (fn [x]
+                    (if (map? x)
+                      (with-meta (into {} (map (fn [[k v]]
+                                                 (let [coercion (get overrides k k)]
+                                                   [k (coerce coercion v)]))) x)
+                                 (meta x))
+                      x))
+                  x)))
 
 (s/fdef coerce-structure
   :args (s/cat :x any?)

diff --git a/test/spec_coerce/core_test.cljc b/test/spec_coerce/core_test.cljc
@@ -134,4 +134,14 @@
                                :sub            {::infer-int "42"}})
          {::some-coercion 321
           ::not-defined   "bla"
+          :sub            {::infer-int 42}}))
+  (is (= (sc/coerce-structure {::some-coercion "321"
+                               ::not-defined   "bla"
+                               :unqualified    "12"
+                               :sub            {::infer-int "42"}}
+                              {::sc/overrides {::not-defined `keyword?
+                                               :unqualified  ::infer-int}})
+         {::some-coercion 321
+          ::not-defined   :bla
+          :unqualified    12
           :sub            {::infer-int 42}})))