feat: add mappings api (#28)

* feat: add mappings api

* fix(mappings): create lazily

* feat(mappings): some fixes and don't create lazily

lua/neocomplete/init.lua

@@ -8,8 +8,30 @@ local function on_insert_enter()
     neocomplete.core:setup()
 end
 
+neocomplete.mappings = {
+    get_fallback = function(key)
+        return require("neocomplete.mappings").get_fallback(key)
+    end,
+    is_open = function()
+        return neocomplete.core and neocomplete.core.menu:is_open()
+    end,
+    confirm = function()
+        neocomplete.core.menu:confirm()
+    end,
+    close = function()
+        neocomplete.core.menu:close()
+    end,
+    select_prev = function(count)
+        neocomplete.core.menu:select_prev(count)
+    end,
+    select_next = function(count)
+        neocomplete.core.menu:select_next(count)
+    end,
+}
+
 --- Sets up neocomplete
 function neocomplete.setup()
+    require("neocomplete.mappings").setup()
     require("neocomplete.config").setup()
     require("neocomplete.highlights")
 

lua/neocomplete/mappings.lua

@@ -0,0 +1,49 @@
+local mappings = {}
+
+function mappings.get_fallback(key)
+    local lhs = ("<Plug>(NeocompleteFallback.%s)"):format(key)
+    vim.keymap.set("i", lhs, key, { noremap = false })
+    return function()
+        vim.api.nvim_feedkeys(vim.keycode(lhs), "im", false)
+    end
+end
+
+local function get_mapping(rhs)
+    for _, map in pairs(vim.api.nvim_get_keymap("i")) do
+        ---@diagnostic disable-next-line: undefined-field
+        if type(map.rhs) == "string" and map.rhs == rhs then
+            ---@diagnostic disable-next-line: undefined-field
+            return map.lhs
+        end
+    end
+end
+
+local function map(plug, callback)
+    vim.keymap.set("i", plug, function()
+        if require("neocomplete").mappings.is_open() then
+            callback()
+        else
+            mappings.get_fallback(get_mapping(plug))()
+        end
+    end)
+end
+
+function mappings.setup()
+    map("<Plug>(NeocompleteConfirm)", function()
+        require("neocomplete").mappings.confirm()
+    end)
+
+    map("<Plug>(NeocompleteSelectNext)", function()
+        require("neocomplete").mappings.select_next(1)
+    end)
+
+    map("<Plug>(NeocompleteSelectPrev)", function()
+        require("neocomplete").mappings.select_prev(1)
+    end)
+
+    map("<Plug>(NeocompleteClose)", function()
+        require("neocomplete").mappings.close()
+    end)
+end
+
+return mappings

lua/neocomplete/menu/draw.lua

@@ -115,6 +115,7 @@ return function(self)
             virt_text_pos = "overlay",
         })
     end
+
     local line = vim.api.nvim_get_current_line()
     local cursor = vim.api.nvim_win_get_cursor(0)
     local cursor_col = cursor[2]

lua/neocomplete/menu/init.lua

@@ -113,6 +113,7 @@ function Menu:set_scroll(direction)
 end
 
 function Menu:select_next(count)
+    count = count or 1
     self.index = self.index + count
     if self.index > #self.entries then
         self.index = self.index - #self.entries - 1
@@ -122,6 +123,7 @@ function Menu:select_next(count)
 end
 
 function Menu:select_prev(count)
+    count = count or 1
     self.index = self.index - count
     if self.index < 0 then
         self.index = #self.entries + self.index + 1