once for onSafeState

Author: jake-stewart

lua/multicursor-nvim/core.lua

@@ -21,8 +21,9 @@ end
 --- Registers a safe state callback.
 --- It will be called whenever the input manager is in a safe state.
 --- @param callback fun(info: mc.SafeStateInfo)
-function core.onSafeState(callback)
-    inputManager:onSafeState(callback)
+--- @param opts? { once?: boolean }
+function core.onSafeState(callback, opts)
+    inputManager:onSafeState(callback, opts)
 end
 
 --- Registers a keymap layer callback.

lua/multicursor-nvim/input-manager.lua

@@ -121,10 +121,18 @@ end
 --- @field wasMode string
 
 --- @param callback fun(info: mc.SafeStateInfo)
-function InputManager:onSafeState(callback)
+--- @param opts? { once?: boolean }
+function InputManager:onSafeState(callback, opts)
     if not self._safeStateCallbacks then
         self._safeStateCallbacks = {}
     end
+    if opts and opts.once then
+        local wrapped = callback
+        callback = function(info)
+            wrapped(info)
+            tbl.remove(self._safeStateCallbacks, callback)
+        end
+    end
     self._safeStateCallbacks[#self._safeStateCallbacks + 1] = callback
 end
 

lua/multicursor-nvim/tbl.lua

@@ -65,6 +65,16 @@ function tbl.filter(t, callback)
     return result
 end
 
+--- @generic T
+--- @param t T[]
+--- @param value T
+function tbl.remove(t, value)
+    local idx = tbl.indexOf(t, value)
+    if idx then
+        table.remove(t, idx)
+    end
+end
+
 --- @generic T
 --- @param t T[]
 --- @callback fun(v: T, i: integer, t: T[]): any