Add Map.filterMap & fixed darklang#151

pm5 · Apr 9, 2022 · 7837b49 · 7837b49
1 parent 66b820f
commit 7837b49
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 4 deletions.
diff --git a/native/src/TableclothMap.ml b/native/src/TableclothMap.ml
@@ -94,6 +94,12 @@ let map_with_index = mapWithIndex
 
 let filter = Base.Map.filter
 
+let filterMap m ~f =
+  Base.Map.filter_mapi m ~f:(fun ~key ~data:value -> f ~key ~value)
+
+
+let filter_map = filterMap
+
 let partition m ~f =
   Base.Map.partitioni_tf m ~f:(fun ~key ~data -> f ~key ~value:data)
 

diff --git a/native/src/TableclothMap.mli b/native/src/TableclothMap.mli
@@ -18,12 +18,12 @@
         end)
       end
 
-      type animal = 
+      type animal =
         | Cow
         | Pig
         | Alpacca
 
-      let pointToAnimal : animal Map.Of(Point).t = 
+      let pointToAnimal : animal Map.Of(Point).t =
         Map.fromList (module Points) [((0, 0), Alpacca); ((3, 4), Cow); ((6, 7), Sheep)]
     ]}
 
@@ -409,6 +409,16 @@ val filter :
     ]}
 *)
 
+val filterMap :
+     ('key, 'value, 'id) t
+  -> f:(key:'key -> value:'value -> 'b option)
+  -> ('key, 'b, 'id) t
+
+val filter_map :
+     ('key, 'value, 'id) t
+  -> f:(key:'key -> value:'value -> 'b option)
+  -> ('key, 'b, 'id) t
+
 val partition :
      ('key, 'value, 'id) t
   -> f:(key:'key -> value:'value -> bool)

diff --git a/rescript/src/TableclothMap.ml b/rescript/src/TableclothMap.ml
@@ -57,6 +57,15 @@ let map_with_index = mapWithIndex
 
 let filter m ~f = Belt.Map.keep m (fun _ value -> f value)
 
+let filterMap m ~f =
+  let id = Belt.Map.getId m in
+  Belt.Map.(
+    reduce m (make ~id) (fun m' key value ->
+        Belt.Map.update m' key (fun _ -> f ~key ~value) ))
+
+
+let filter_map = filterMap
+
 let partition m ~f = Belt.Map.partition m (fun key value -> f ~key ~value)
 
 let find m ~f = Belt.Map.findFirstBy m (fun key value -> f ~key ~value)

diff --git a/rescript/src/TableclothMap.mli b/rescript/src/TableclothMap.mli
@@ -18,12 +18,12 @@
         end)
       end
 
-      type animal = 
+      type animal =
         | Cow
         | Pig
         | Alpacca
 
-      let pointToAnimal : animal Map.Of(Point).t = 
+      let pointToAnimal : animal Map.Of(Point).t =
         Map.fromList (module Points) [((0, 0), Alpacca); ((3, 4), Cow); ((6, 7), Sheep)]
     ]}
 
@@ -409,6 +409,16 @@ val filter :
     ]}
 *)
 
+val filterMap :
+     ('key, 'value, 'id) t
+  -> f:(key:'key -> value:'value -> 'b option)
+  -> ('key, 'b, 'id) t
+
+val filter_map :
+     ('key, 'value, 'id) t
+  -> f:(key:'key -> value:'value -> 'b option)
+  -> ('key, 'b, 'id) t
+
 val partition :
      ('key, 'value, 'id) t
   -> f:(key:'key -> value:'value -> bool)