submap sub map should maintain properties of parent map redplanetla…

…bs#235

TODO: figure out how to handle transients (i.e. submap!)

Adding new SubMap protocol to handle the different implementations of key selection (including one that maintains order to be used for sorted-map)

Adding test to ensure order preserved for a large map

src/clj/com/rpl/specter.cljc

@@ -905,10 +905,10 @@
   submap
   [m-keys]
   (select* [this structure next-fn]
-    (next-fn (select-keys structure m-keys)))
+           (next-fn (n/select-submap structure m-keys)))
 
   (transform* [this structure next-fn]
-    (let [submap (select-keys structure m-keys)
+    (let [submap (n/select-submap structure m-keys)
           newmap (next-fn submap)]
       (merge (reduce dissoc structure m-keys)
              newmap))))

src/clj/com/rpl/specter/navs.cljc

@@ -691,3 +691,30 @@
     ((:end-fn end-fn) structure start)
     (end-fn structure)
     ))
+
+(defn- maintain-sorted-submap [structure m-keys]
+  (into (empty structure) (select-keys structure m-keys)))
+
+(defprotocol SubMap
+  (select-submap [structure m-keys]))
+
+(extend-protocol SubMap
+  nil
+  (select-submap [structure m-keys] nil)
+
+  #?(:clj clojure.lang.PersistentArrayMap :cljs cljs.core/PersistentArrayMap)
+  (select-submap [structure m-keys]
+    (maintain-sorted-submap structure m-keys))
+
+  #?(:clj clojure.lang.PersistentHashMap :cljs cljs.core/PersistentHashMap)
+  (select-submap [structure m-keys]
+    (select-keys structure m-keys))
+
+  #?(:clj clojure.lang.PersistentStructMap)
+  #?(:clj
+    (select-submap [structure m-keys]
+      (select-keys structure m-keys)))
+
+  #?(:clj clojure.lang.PersistentTreeMap :cljs cljs.core/PersistentTreeMap)
+  (select-submap [structure m-keys]
+    (maintain-sorted-submap structure m-keys)))

test/com/rpl/specter/core_test.cljc

@@ -611,7 +611,8 @@
         (= (transform (s/subset s3) identity combined) combined)
         (= (setval (s/subset s3) #{} combined) (set/difference combined s2))
         (= (setval (s/subset s3) s4 combined) (-> combined (set/difference s2) (set/union s4)))))))
-
+#?(:clj
+  (defstruct test-struct :x :y :z))
 
 (deftest submap-test
   (is (= [{:foo 1}]
@@ -623,7 +624,16 @@
   (is (= {:a {:new 1}
           :c {:new 1
               :old 1}}
-         (setval [s/ALL s/LAST (s/submap [])] {:new 1} {:a nil, :c {:old 1}}))))
+         (setval [s/ALL s/LAST (s/submap [])] {:new 1} {:a nil, :c {:old 1}})))
+  #?(:clj
+    ; make sure struct-map works
+    (is (= [{:z 3, :y 2}]
+           (select (s/submap [:z :y]) (into (struct-map test-struct) {:x 1 :y 2 :z 3})))))
+  ; ensure order is preserved for larger map (i.e. TreeMap)
+  (let [range50 (range 50)]
+    (is (= (take-nth 2 range50)
+           (select [(s/submap (take-nth 2 range50)) s/MAP-KEYS] (apply sorted-map (interleave range50 range50))))))
+  )
 
 (deftest nil->val-test
   (is (= {:a #{:b}}