refactor(lsp): split completion module by concern

lua/lsp/completion/init.lua

@@ -0,0 +1,116 @@
+local popup = require("lsp.completion.popup")
+local request = require("lsp.completion.request")
+
+local GROUP = vim.api.nvim_create_augroup("ow.lsp.completion", { clear = true })
+
+local M = {}
+
+---@param capabilities lsp.ClientCapabilities
+function M.apply_capabilities(capabilities)
+    capabilities.textDocument.completion.completionItem.snippetSupport = false
+end
+
+---@param client vim.lsp.Client
+---@param buf integer
+function M.on_attach(client, buf)
+    if
+        not client:supports_method(
+            vim.lsp.protocol.Methods.textDocument_completion
+        )
+    then
+        return
+    end
+
+    vim.bo[buf].omnifunc = "v:lua.ow_lsp_completion.omnifunc"
+
+    vim.api.nvim_clear_autocmds({ group = GROUP, buffer = buf })
+    vim.api.nvim_create_autocmd("InsertCharPre", {
+        buffer = buf,
+        group = GROUP,
+        callback = request.on_insert_char_pre,
+    })
+end
+
+function M.setup()
+    vim.api.nvim_create_autocmd("CompleteChanged", {
+        group = GROUP,
+        callback = function(ev)
+            local completed = vim.v.event.completed_item or {}
+            local lsp_item = vim.tbl_get(
+                completed,
+                "user_data",
+                "nvim",
+                "lsp",
+                "completion_item"
+            )
+            local client_id =
+                vim.tbl_get(completed, "user_data", "nvim", "lsp", "client_id")
+            local client = client_id and vim.lsp.get_client_by_id(client_id)
+
+            if not lsp_item or not client then
+                vim.schedule(popup.close)
+                return
+            end
+
+            local ft = vim.bo[ev.buf].filetype
+            ---@type ow.lsp.completion.Pum
+            local pum = {
+                row = vim.v.event.row,
+                col = vim.v.event.col,
+                width = vim.v.event.width,
+                height = vim.v.event.height,
+                scrollbar = vim.v.event.scrollbar,
+            }
+
+            if
+                client:supports_method(
+                    vim.lsp.protocol.Methods.completionItem_resolve
+                )
+            then
+                popup.dispatch_resolve(
+                    client,
+                    lsp_item,
+                    ft,
+                    pum,
+                    completed.word,
+                    ev.buf
+                )
+            else
+                vim.schedule(function()
+                    popup.show_for(lsp_item, ft, pum)
+                end)
+            end
+        end,
+    })
+
+    vim.api.nvim_create_autocmd({ "CompleteDonePre", "InsertLeave" }, {
+        group = GROUP,
+        callback = popup.close,
+    })
+
+    ---@param key string
+    ---@param action fun()
+    local function scroll_map(key, action)
+        vim.keymap.set("i", key, function()
+            if popup.is_visible() then
+                vim.schedule(action)
+                return ""
+            end
+            return key
+        end, { expr = true, replace_keycodes = true })
+    end
+    scroll_map("<C-d>", function()
+        popup.scroll_half(vim.keycode("<C-e>"))
+    end)
+    scroll_map("<C-u>", function()
+        popup.scroll_half(vim.keycode("<C-y>"))
+    end)
+    scroll_map("<C-j>", function()
+        popup.scroll_line(vim.keycode("<C-e>"))
+    end)
+    scroll_map("<C-k>", function()
+        popup.scroll_line(vim.keycode("<C-y>"))
+    end)
+end
+
+return M