feat(zipper): add function to map on cursor focus with default

1 files changed, 5 insertions, 0 deletions

diff --git a/lib/zipper.ml b/lib/zipper.ml
index 30be8eb..9313518 100644
--- a/lib/zipper.ml
+++ b/lib/zipper.ml
@@ -107,6 +107,11 @@ let map_focus f z =
   | None -> z
   | Some (h, t) -> { z with after = Sequence.shift_right t (f h) }
 
+let map_focus_or ~default f z =
+  match Sequence.next z.after with
+  | None -> { z with after = Sequence.singleton default }
+  | Some (h, t) -> { z with after = Sequence.shift_right t (f h) }
+
 let map f z =
   { z with before = Sequence.map ~f z.before; after = Sequence.map ~f z.after }