[ base ] Add update functions to sorted maps

CHANGELOG.md

@@ -203,6 +203,8 @@
 
 * Adds `infixOfBy` and `isInfixOfBy` into `Data.List`.
 
+* Adds updating functions to `SortedMap` and `SortedDMap`.
+
 #### System
 
 * Changes `getNProcessors` to return the number of online processors rather than

libs/base/Data/SortedMap.idr

@@ -37,6 +37,19 @@ export
 delete : k -> SortedMap k v -> SortedMap k v
 delete k = M . delete k . unM
 
+export
+update : (Maybe v -> Maybe v) -> k -> SortedMap k v -> SortedMap k v
+update f k m = case f $ lookup k m of
+  Just v  => insert k v m
+  Nothing => delete k m
+
+||| Updates existing value, if it is present, and does nothing otherwise
+export
+updateExisting : (v -> v) -> k -> SortedMap k v -> SortedMap k v
+updateExisting f k m = case lookup k m of
+  Just v  => insert k (f v) m
+  Nothing => m
+
 export
 fromList : Ord k => List (k, v) -> SortedMap k v
 fromList = flip insertFrom empty

libs/base/Data/SortedMap/Dependent.idr

@@ -254,6 +254,19 @@ delete k (M (S n) t) =
     Left t' => (M _ t')
     Right t' => (M _ t')
 
+export
+update : DecEq k => (x : k) -> (Maybe (v x) -> Maybe (v x)) -> SortedDMap k v -> SortedDMap k v
+update k f m = case f $ lookupPrecise k m of
+  Just v  => insert k v m
+  Nothing => delete k m
+
+||| Updates existing value, if it is present, and does nothing otherwise
+export
+updateExisting : DecEq k => (x : k) -> (v x -> v x) -> SortedDMap k v -> SortedDMap k v
+updateExisting k f m = case lookupPrecise k m of
+  Just v  => insert k (f v) m
+  Nothing => m
+
 export
 fromList : Ord k => List (x : k ** v x) -> SortedDMap k v
 fromList = foldl (flip (uncurry insert)) empty