664 lines
20 KiB
Lua
664 lines
20 KiB
Lua
local kind = require("lsp.kind")
|
|
local util = require("util")
|
|
|
|
local MAX_WIDTH = 80
|
|
local MAX_HEIGHT = 20
|
|
local REQUEST_DEBOUNCE_MS = 100
|
|
|
|
local GROUP = vim.api.nvim_create_augroup("ow.lsp.completion", { clear = true })
|
|
local NS = vim.api.nvim_create_namespace("ow.lsp.completion")
|
|
|
|
---@class ow.lsp.completion.Pum
|
|
---@field row integer
|
|
---@field col integer
|
|
---@field width integer
|
|
---@field height integer
|
|
---@field scrollbar boolean
|
|
|
|
---@class ow.lsp.completion.PendingResolve
|
|
---@field client vim.lsp.Client
|
|
---@field id integer
|
|
|
|
---@class ow.lsp.completion.Popup
|
|
---@field winid integer?
|
|
---@field bufnr integer?
|
|
---@field pending ow.lsp.completion.PendingResolve?
|
|
|
|
---@param ft string
|
|
---@param text string
|
|
---@return string
|
|
local function fence(ft, text)
|
|
return string.format("```%s\n%s\n```", ft, text)
|
|
end
|
|
|
|
---@param item lsp.CompletionItem
|
|
---@return string
|
|
local function signature_of(item)
|
|
return item.detail or vim.tbl_get(item, "labelDetails", "description") or ""
|
|
end
|
|
|
|
---@type ow.lsp.completion.Popup
|
|
local popup = {
|
|
winid = nil,
|
|
bufnr = nil,
|
|
pending = nil,
|
|
}
|
|
|
|
local function cancel_pending()
|
|
if popup.pending then
|
|
pcall(
|
|
popup.pending.client.cancel_request,
|
|
popup.pending.client,
|
|
popup.pending.id
|
|
)
|
|
popup.pending = nil
|
|
end
|
|
end
|
|
|
|
---@return integer bufnr
|
|
local function ensure_buffer()
|
|
if not popup.bufnr or not vim.api.nvim_buf_is_valid(popup.bufnr) then
|
|
popup.bufnr = vim.api.nvim_create_buf(false, true)
|
|
vim.bo[popup.bufnr].buftype = "nofile"
|
|
vim.bo[popup.bufnr].bufhidden = "hide"
|
|
vim.bo[popup.bufnr].swapfile = false
|
|
pcall(vim.treesitter.start, popup.bufnr, "markdown")
|
|
end
|
|
return popup.bufnr
|
|
end
|
|
|
|
local function update_indicators()
|
|
if not popup.winid or not vim.api.nvim_win_is_valid(popup.winid) then
|
|
return
|
|
end
|
|
local visible = vim.api.nvim_win_get_height(popup.winid)
|
|
local topline, botline =
|
|
unpack(vim.api.nvim_win_call(popup.winid, function()
|
|
return { vim.fn.line("w0"), vim.fn.line("w$") }
|
|
end))
|
|
local ok_total, info_total =
|
|
pcall(vim.api.nvim_win_text_height, popup.winid, {})
|
|
local total = ok_total and info_total.all or visible
|
|
|
|
local display_topline = 1
|
|
if topline > 1 then
|
|
local ok, info = pcall(vim.api.nvim_win_text_height, popup.winid, {
|
|
end_row = topline - 2,
|
|
})
|
|
if ok then
|
|
display_topline = (info.all or 0) + 1
|
|
end
|
|
end
|
|
local display_bot = display_topline + visible - 1
|
|
|
|
local has_above = display_topline > 1
|
|
local has_below = display_bot < total
|
|
and botline
|
|
< vim.api.nvim_buf_line_count(vim.api.nvim_win_get_buf(popup.winid))
|
|
pcall(vim.api.nvim_win_set_config, popup.winid, {
|
|
title = has_above and { { "▲ ", "FloatBorder" } } or "",
|
|
title_pos = has_above and "right" or nil,
|
|
footer = has_below and { { "▼ ", "FloatBorder" } } or "",
|
|
footer_pos = has_below and "right" or nil,
|
|
})
|
|
end
|
|
|
|
local function close()
|
|
cancel_pending()
|
|
if popup.winid and vim.api.nvim_win_is_valid(popup.winid) then
|
|
pcall(vim.api.nvim_win_close, popup.winid, true)
|
|
end
|
|
popup.winid = nil
|
|
end
|
|
|
|
---@param content string
|
|
---@param pum ow.lsp.completion.Pum
|
|
---@param width integer
|
|
local function show(content, pum, width)
|
|
local bufnr = ensure_buffer()
|
|
local lines = vim.split(content, "\n", { plain = true })
|
|
vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, lines)
|
|
|
|
vim.api.nvim_buf_clear_namespace(bufnr, NS, 0, -1)
|
|
for i, line in ipairs(lines) do
|
|
if line ~= "" and line:gsub("─", "") == "" then
|
|
vim.api.nvim_buf_set_extmark(bufnr, NS, i - 1, 0, {
|
|
end_row = i - 1,
|
|
end_col = #line,
|
|
hl_group = "FloatBorder",
|
|
priority = 200,
|
|
})
|
|
end
|
|
end
|
|
|
|
width = math.min(width, MAX_WIDTH)
|
|
local height = math.min(#lines, MAX_HEIGHT)
|
|
|
|
local right_edge = pum.col + pum.width + (pum.scrollbar and 1 or 0)
|
|
local right_space = vim.o.columns - right_edge
|
|
local col
|
|
if right_space >= width + 2 then
|
|
col = right_edge
|
|
else
|
|
col = math.max(pum.col - width - 2, 0)
|
|
end
|
|
|
|
local cfg = {
|
|
relative = "editor",
|
|
row = pum.row,
|
|
col = col,
|
|
width = width,
|
|
height = height,
|
|
border = "rounded",
|
|
focusable = false,
|
|
style = "minimal",
|
|
}
|
|
|
|
if popup.winid and vim.api.nvim_win_is_valid(popup.winid) then
|
|
pcall(vim.api.nvim_win_set_config, popup.winid, cfg)
|
|
else
|
|
cfg.noautocmd = true
|
|
popup.winid = vim.api.nvim_open_win(bufnr, false, cfg)
|
|
vim.wo[popup.winid].wrap = true
|
|
vim.wo[popup.winid].linebreak = true
|
|
vim.wo[popup.winid].conceallevel = 2
|
|
end
|
|
pcall(vim.api.nvim_win_set_cursor, popup.winid, { 1, 0 })
|
|
|
|
local ok, info = pcall(vim.api.nvim_win_text_height, popup.winid, {})
|
|
if ok and info.all and info.all ~= height then
|
|
pcall(
|
|
vim.api.nvim_win_set_height,
|
|
popup.winid,
|
|
math.min(info.all, MAX_HEIGHT)
|
|
)
|
|
end
|
|
|
|
update_indicators()
|
|
end
|
|
|
|
---@param item lsp.CompletionItem
|
|
---@param ft string
|
|
---@return string? content
|
|
---@return integer? width
|
|
local function build_content(item, ft)
|
|
local signature = signature_of(item)
|
|
local doc = item.documentation
|
|
if type(doc) == "table" then
|
|
doc = doc.value
|
|
end
|
|
doc = doc or ""
|
|
|
|
local code_parts = {}
|
|
if item.additionalTextEdits then
|
|
for _, edit in ipairs(item.additionalTextEdits) do
|
|
local text = (edit.newText or ""):gsub("%s+$", "")
|
|
if text ~= "" then
|
|
table.insert(code_parts, fence(ft, text))
|
|
end
|
|
end
|
|
end
|
|
if signature ~= "" then
|
|
table.insert(code_parts, fence(ft, signature))
|
|
end
|
|
|
|
local sections = {}
|
|
if #code_parts > 0 then
|
|
table.insert(sections, table.concat(code_parts, "\n\n"))
|
|
end
|
|
if doc ~= "" then
|
|
table.insert(sections, doc)
|
|
end
|
|
if #sections == 0 then
|
|
return nil, nil
|
|
end
|
|
|
|
local max_w = 0
|
|
for _, s in ipairs(sections) do
|
|
for _, line in ipairs(vim.split(s, "\n", { plain = true })) do
|
|
max_w = math.max(max_w, vim.fn.strdisplaywidth(line))
|
|
end
|
|
end
|
|
local sep = "\n" .. string.rep("─", math.min(max_w, MAX_WIDTH)) .. "\n"
|
|
return table.concat(sections, sep), max_w
|
|
end
|
|
|
|
---@class ow.lsp.completion.Request
|
|
---@field generation integer
|
|
---@field phase ('awaiting' | 'ready' | 'done')?
|
|
---@field result table?
|
|
---@field cancel function?
|
|
|
|
---@type ow.lsp.completion.Request
|
|
local request = {
|
|
generation = 0,
|
|
phase = nil,
|
|
result = nil,
|
|
cancel = nil,
|
|
}
|
|
|
|
local function discard_request()
|
|
if request.cancel then
|
|
pcall(request.cancel)
|
|
end
|
|
request.generation = request.generation + 1
|
|
request.phase = nil
|
|
request.result = nil
|
|
request.cancel = nil
|
|
end
|
|
|
|
local function buffer_has_completion_client()
|
|
return #vim.lsp.get_clients({
|
|
bufnr = 0,
|
|
method = vim.lsp.protocol.Methods.textDocument_completion,
|
|
}) > 0
|
|
end
|
|
|
|
--- Normalize an `itemDefaults.editRange` to a plain Range, since it may be
|
|
--- either `Range` or `{ insert: Range, replace: Range }`.
|
|
---@param edit_range table?
|
|
---@return table?
|
|
local function default_range(edit_range)
|
|
if type(edit_range) ~= "table" then
|
|
return nil
|
|
end
|
|
return edit_range.insert or (edit_range.start and edit_range) or nil
|
|
end
|
|
|
|
--- Normalize an item's `textEdit`, which may be `TextEdit` (has `range`) or
|
|
--- `InsertReplaceEdit` (has `insert`/`replace`).
|
|
---@param item lsp.CompletionItem
|
|
---@return table?
|
|
local function item_range(item)
|
|
local te = item.textEdit
|
|
if type(te) ~= "table" then
|
|
return nil
|
|
end
|
|
return te.range or te.insert
|
|
end
|
|
|
|
---@param response table LSP response entry (as returned by buf_request_all)
|
|
---@return integer? character 0-indexed UTF-16 character where edit starts
|
|
local function response_edit_start(response)
|
|
if not response.result or response.err or response.error then
|
|
return nil
|
|
end
|
|
local defaults =
|
|
default_range(vim.tbl_get(response.result, "itemDefaults", "editRange"))
|
|
if defaults then
|
|
return defaults.start.character
|
|
end
|
|
local items = response.result.items or response.result
|
|
if type(items) ~= "table" then
|
|
return nil
|
|
end
|
|
for _, item in ipairs(items) do
|
|
local range = item_range(item)
|
|
if range then
|
|
return range.start.character
|
|
end
|
|
end
|
|
return nil
|
|
end
|
|
|
|
---@return integer byte_col 0-indexed byte column for completefunc findstart
|
|
local function resolve_start_col()
|
|
for _, response in pairs(request.result or {}) do
|
|
local col = response_edit_start(response)
|
|
if col then
|
|
return col
|
|
end
|
|
end
|
|
local line = vim.api.nvim_get_current_line()
|
|
local cursor_col = vim.fn.col(".") - 1
|
|
return vim.fn.match(line:sub(1, cursor_col), "\\k*$")
|
|
end
|
|
|
|
---@param item lsp.CompletionItem
|
|
---@return string
|
|
local function item_word(item)
|
|
local edit = item.textEdit
|
|
if type(edit) == "table" and edit.newText then
|
|
return edit.newText
|
|
end
|
|
return item.insertText or item.label
|
|
end
|
|
|
|
---@param items lsp.CompletionItem[]
|
|
---@param defaults? table
|
|
local function apply_defaults(items, defaults)
|
|
if type(defaults) ~= "table" then
|
|
return
|
|
end
|
|
local range = default_range(defaults.editRange)
|
|
for _, item in ipairs(items) do
|
|
item.commitCharacters = item.commitCharacters
|
|
or defaults.commitCharacters
|
|
item.data = item.data or defaults.data
|
|
item.insertTextFormat = item.insertTextFormat
|
|
or defaults.insertTextFormat
|
|
item.insertTextMode = item.insertTextMode or defaults.insertTextMode
|
|
if range and type(item.textEdit) ~= "table" then
|
|
item.textEdit = {
|
|
range = range,
|
|
newText = item.textEditText or item.insertText or item.label,
|
|
}
|
|
end
|
|
end
|
|
end
|
|
|
|
---@param base string
|
|
---@return table[]
|
|
local function to_complete_items(base)
|
|
local entries = {}
|
|
for client_id, response in pairs(request.result or {}) do
|
|
if response.result and not (response.err or response.error) then
|
|
local raw = response.result.items or response.result
|
|
if type(raw) == "table" then
|
|
apply_defaults(raw, response.result.itemDefaults)
|
|
for _, item in ipairs(raw) do
|
|
entries[#entries + 1] =
|
|
{ item = item, client_id = client_id }
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
if base ~= "" then
|
|
entries = vim.tbl_filter(function(entry)
|
|
local key = entry.item.filterText or entry.item.label
|
|
return vim.startswith(key, base)
|
|
end, entries)
|
|
end
|
|
|
|
table.sort(entries, function(a, b)
|
|
local ka = a.item.sortText or a.item.label
|
|
local kb = b.item.sortText or b.item.label
|
|
return ka < kb
|
|
end)
|
|
|
|
local result = {}
|
|
for _, entry in ipairs(entries) do
|
|
local item = entry.item
|
|
local kind_icon, kind_hl = kind.get(item.kind)
|
|
result[#result + 1] = {
|
|
word = item_word(item),
|
|
abbr = item.label:match("[^(]+") or item.label,
|
|
menu = "",
|
|
kind = kind_icon,
|
|
kind_hlgroup = kind_hl,
|
|
-- non-empty so our CompleteChanged handler triggers resolve
|
|
info = " ",
|
|
icase = 1,
|
|
dup = 1,
|
|
empty = 1,
|
|
user_data = {
|
|
nvim = {
|
|
lsp = {
|
|
completion_item = item,
|
|
client_id = entry.client_id,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
end
|
|
return result
|
|
end
|
|
|
|
local function feed_user_completion()
|
|
if vim.fn.mode() ~= "i" or vim.fn.pumvisible() ~= 0 then
|
|
return
|
|
end
|
|
vim.api.nvim_feedkeys(vim.keycode("<C-x><C-o>"), "n", false)
|
|
end
|
|
|
|
---@param trigger_kind integer
|
|
---@param trigger_char string?
|
|
local function dispatch_request(trigger_kind, trigger_char)
|
|
discard_request()
|
|
request.phase = "awaiting"
|
|
local gen = request.generation
|
|
local buf = vim.api.nvim_get_current_buf()
|
|
local params = vim.lsp.util.make_position_params(0, "utf-16")
|
|
---@cast params lsp.CompletionParams
|
|
params.context = {
|
|
triggerKind = trigger_kind,
|
|
triggerCharacter = trigger_char,
|
|
}
|
|
request.cancel = vim.lsp.buf_request_all(
|
|
buf,
|
|
vim.lsp.protocol.Methods.textDocument_completion,
|
|
params,
|
|
function(result)
|
|
if request.generation ~= gen or request.phase ~= "awaiting" then
|
|
return
|
|
end
|
|
request.phase = "ready"
|
|
request.result = result
|
|
feed_user_completion()
|
|
end
|
|
)
|
|
end
|
|
|
|
_G.ow_lsp_completion = _G.ow_lsp_completion or {}
|
|
|
|
--- Returns cached items without re-requesting, which is what lets Vim filter
|
|
--- the popup in-place on backspace.
|
|
---@param findstart 0 | 1
|
|
---@param base string
|
|
---@return integer | table
|
|
function _G.ow_lsp_completion.omnifunc(findstart, base)
|
|
if not buffer_has_completion_client() or request.phase == "awaiting" then
|
|
return findstart == 1 and -3 or {}
|
|
end
|
|
|
|
if request.phase ~= "ready" then
|
|
dispatch_request(vim.lsp.protocol.CompletionTriggerKind.Invoked)
|
|
return findstart == 1 and -3 or {}
|
|
end
|
|
|
|
if findstart == 1 then
|
|
return resolve_start_col()
|
|
end
|
|
|
|
local items = to_complete_items(base)
|
|
request.phase = "done"
|
|
return items
|
|
end
|
|
|
|
---@param char string
|
|
---@return boolean
|
|
local function is_trigger_char(char)
|
|
local clients = vim.lsp.get_clients({
|
|
bufnr = 0,
|
|
method = vim.lsp.protocol.Methods.textDocument_completion,
|
|
})
|
|
for _, client in ipairs(clients) do
|
|
local chars = vim.tbl_get(
|
|
client,
|
|
"server_capabilities",
|
|
"completionProvider",
|
|
"triggerCharacters"
|
|
)
|
|
if chars and vim.list_contains(chars, char) then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
local debounce = util.debounce(function(_, trigger_kind, char)
|
|
if vim.fn.mode() ~= "i" then
|
|
return
|
|
end
|
|
dispatch_request(trigger_kind, char)
|
|
end, REQUEST_DEBOUNCE_MS)
|
|
|
|
local function on_insert_char_pre()
|
|
local char = vim.v.char
|
|
if char == "" then
|
|
return
|
|
end
|
|
local is_word = char:match("[%w_]") ~= nil
|
|
local is_trigger = not is_word and is_trigger_char(char)
|
|
if not (is_word or is_trigger) then
|
|
return
|
|
end
|
|
-- For word chars, let Vim's native in-place filtering handle things while
|
|
-- the pum is already showing. Trigger chars always schedule a request
|
|
-- because they change the completion context (e.g. member access).
|
|
if is_word and vim.fn.pumvisible() ~= 0 then
|
|
return
|
|
end
|
|
|
|
local kind_num = is_trigger
|
|
and vim.lsp.protocol.CompletionTriggerKind.TriggerCharacter
|
|
or vim.lsp.protocol.CompletionTriggerKind.Invoked
|
|
debounce:call(nil, kind_num, is_trigger and char or nil)
|
|
end
|
|
|
|
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 = 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(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,
|
|
}
|
|
|
|
local will_resolve = client:supports_method(
|
|
vim.lsp.protocol.Methods.completionItem_resolve
|
|
)
|
|
|
|
if not will_resolve then
|
|
local initial, width = build_content(lsp_item, ft)
|
|
vim.schedule(function()
|
|
if initial and width then
|
|
show(initial, pum, width)
|
|
else
|
|
close()
|
|
end
|
|
end)
|
|
return
|
|
end
|
|
|
|
cancel_pending()
|
|
local word = completed.word
|
|
local _, request_id = client:request(
|
|
vim.lsp.protocol.Methods.completionItem_resolve,
|
|
lsp_item,
|
|
function(err, result)
|
|
popup.pending = nil
|
|
if err or not result then
|
|
return
|
|
end
|
|
local cur = vim.fn.complete_info({ "completed" })
|
|
if
|
|
(vim.tbl_get(cur, "completed", "word") or "") ~= word
|
|
then
|
|
return
|
|
end
|
|
local content, width = build_content(result, ft)
|
|
if content and width then
|
|
show(content, pum, width)
|
|
else
|
|
close()
|
|
end
|
|
end,
|
|
ev.buf
|
|
)
|
|
if request_id then
|
|
popup.pending = { client = client, id = request_id }
|
|
end
|
|
end,
|
|
})
|
|
|
|
vim.api.nvim_create_autocmd({ "CompleteDonePre", "InsertLeave" }, {
|
|
group = GROUP,
|
|
callback = close,
|
|
})
|
|
|
|
---@param key string
|
|
---@param direction string single-line scroll keycode (<C-e> or <C-y>)
|
|
---@param count integer
|
|
local function scroll_map(key, direction, count)
|
|
vim.keymap.set("i", key, function()
|
|
if popup.winid and vim.api.nvim_win_is_valid(popup.winid) then
|
|
vim.schedule(function()
|
|
if
|
|
popup.winid
|
|
and vim.api.nvim_win_is_valid(popup.winid)
|
|
then
|
|
vim.api.nvim_win_call(popup.winid, function()
|
|
vim.cmd.normal({
|
|
args = { count .. direction },
|
|
bang = true,
|
|
})
|
|
end)
|
|
update_indicators()
|
|
end
|
|
end)
|
|
return ""
|
|
end
|
|
return key
|
|
end, { expr = true, replace_keycodes = true })
|
|
end
|
|
local half = math.floor(MAX_HEIGHT / 2)
|
|
scroll_map("<C-d>", vim.keycode("<C-e>"), half)
|
|
scroll_map("<C-u>", vim.keycode("<C-y>"), half)
|
|
scroll_map("<C-j>", vim.keycode("<C-e>"), 1)
|
|
scroll_map("<C-k>", vim.keycode("<C-y>"), 1)
|
|
end
|
|
|
|
return M
|