feat: add feed_with_count method

Wansmer · Sep 14, 2024 · cc89712 · cc89712
1 parent 62cc28b
commit cc89712
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lua/langmapper/utils.lua b/lua/langmapper/utils.lua
@@ -171,6 +171,12 @@ function M.trans_list(list)
   return trans_list
 end
 
+function M.feed_with_count(keys)
+  local modes = { 'c', 'o', 'i' } -- ignore count on this modes
+  local count = (vim.v.count == 0 or vim.tbl_contains(modes, vim.fn.mode())) and '' or vim.v.count
+  return count .. keys
+end
+
 ---Remapping each CTRL+ sequence
 function M._map_translated_ctrls()
   local function remap_ctrl(list, from, to)