Fix native dot replay bridge

1 files changed, 128 insertions, 1 deletions

diff --git a/lua/clever_f/host_adapter.lua b/lua/clever_f/host_adapter.lua
index 17c29ef..a81a44c 100644
--- a/lua/clever_f/host_adapter.lua
+++ b/lua/clever_f/host_adapter.lua
@@ -48,6 +48,7 @@ function HostAdapter.new(options)
     cursor_presentations = {},
     events = {},
     dot_repeat = nil,
+    dot_bridge = nil,
     event_order = {},
     augroup = nil,
   }
@@ -124,6 +125,8 @@ function HostAdapter:set_operator_inclusive(enabled)
   end
 end
 
+local install_dot_bridge
+
 function HostAdapter:register_dot_repeat(payload, callback)
   if not domain.DotPayload.is(payload) then
     fail("dot-repeat payload must be a DotPayload", 2)
@@ -131,10 +134,18 @@ function HostAdapter:register_dot_repeat(payload, callback)
   if callback ~= nil and type(callback) ~= "function" then
     fail("dot-repeat callback must be a function", 2)
   end
-  adapter_records[self].dot_repeat = {
+  local record = adapter_records[self]
+  record.dot_repeat = {
     payload = payload,
     callback = callback,
+    operator = self:read_pending_operator(),
   }
+  if install_dot_bridge ~= nil then
+    install_dot_bridge(self)
+  end
+  if record.dot_bridge ~= nil then
+    record.dot_bridge.awaiting_change = true
+  end
   return payload
 end
 
@@ -479,6 +490,122 @@ function HostAdapter:remove_event_registration(identity)
   return true
 end
 
+local DOT_MOTION_MAPPING = "<Plug>(clever-f-dot-motion)"
+
+local function dot_bridge_supported(runtime)
+  return type(runtime.keymap) == "table"
+    and type(runtime.keymap.set) == "function"
+    and type(runtime.keymap.del) == "function"
+    and type(runtime.fn.maparg) == "function"
+    and type(runtime.fn.mapset) == "function"
+    and type(runtime.api.nvim_feedkeys) == "function"
+end
+
+local function restore_dot_mapping(record)
+  local bridge = record.dot_bridge
+  if bridge == nil or not bridge.active then
+    return false
+  end
+  bridge.active = false
+  pcall(record.runtime.keymap.del, "n", ".")
+  if type(bridge.previous_mapping) == "table"
+    and next(bridge.previous_mapping) ~= nil
+  then
+    record.runtime.fn.mapset("n", false, bridge.previous_mapping)
+  end
+  return true
+end
+
+local function dot_replay_keys(runtime, count, operator)
+  local prefix = count > 0 and tostring(count) or ""
+  local keys = prefix .. operator .. DOT_MOTION_MAPPING
+  if type(runtime.keycode) == "function" then
+    return runtime.keycode(keys)
+  end
+  return runtime.api.nvim_replace_termcodes(keys, true, false, true)
+end
+
+install_dot_bridge = function(adapter)
+  local record = adapter_records[adapter]
+  local runtime = record.runtime
+  if not dot_bridge_supported(runtime) then
+    return nil
+  end
+  local bridge = record.dot_bridge
+  if bridge == nil then
+    bridge = {
+      active = false,
+      awaiting_change = false,
+      previous_mapping = nil,
+    }
+    record.dot_bridge = bridge
+    runtime.keymap.set("o", DOT_MOTION_MAPPING, function()
+      local registration = record.dot_repeat
+      if registration == nil or registration.callback == nil then
+        return
+      end
+      local count = runtime.v.count1
+      if type(count) ~= "number" or count < 1 then
+        count = 1
+      end
+      local outcome = registration.callback(
+        registration.payload,
+        domain.Count.new(count)
+      )
+      if domain.ActionOutcome.is(outcome) then
+        adapter:translate_action_outcome(outcome)
+      end
+    end, {
+      silent = true,
+      remap = false,
+      desc = "clever-f dot motion",
+    })
+    local has_cmd_atom = type(runtime.fn.exists) == "function"
+      and runtime.fn.exists("##CmdAtom") == 1
+    local ownership_events = has_cmd_atom
+        and "CmdAtom"
+      or { "TextChanged", "TextChangedI", "TextChangedP" }
+    runtime.api.nvim_create_autocmd(ownership_events, {
+      group = event_augroup(record),
+      desc = "clever-f dot ownership",
+      callback = function(event)
+        if has_cmd_atom and not (event.data and event.data.changed) then
+          return
+        end
+        if bridge.awaiting_change then
+          bridge.awaiting_change = false
+          return
+        end
+        restore_dot_mapping(record)
+      end,
+    })
+  end
+  if not bridge.active then
+    bridge.previous_mapping = runtime.fn.maparg(".", "n", false, true)
+    runtime.keymap.set("n", ".", function()
+      local registration = record.dot_repeat
+      if registration == nil or registration.operator == "" then
+        restore_dot_mapping(record)
+        runtime.api.nvim_feedkeys(".", "n", false)
+        return
+      end
+      bridge.awaiting_change = true
+      local count = runtime.v.count or 0
+      runtime.api.nvim_feedkeys(
+        dot_replay_keys(runtime, count, registration.operator),
+        "n",
+        false
+      )
+    end, {
+      silent = true,
+      remap = false,
+      desc = "clever-f dot repeat",
+    })
+    bridge.active = true
+  end
+  return bridge
+end
+
 function HostAdapter:deliver_event(name, payload)
   if type(name) ~= "string" or name == "" then
     fail("event name must be a nonempty string", 2)