1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
-- Credit goes to @hrsh7th for the code that this was based on
-- https://github.com/hrsh7th/cmp-cmdline
-- License: MIT
local async = require('blink.cmp.lib.async')
local constants = require('blink.cmp.sources.cmdline.constants')
--- @class blink.cmp.Source
local cmdline = {}
function cmdline.new()
local self = setmetatable({}, { __index = cmdline })
self.before_line = ''
self.offset = -1
self.ctype = ''
self.items = {}
return self
end
function cmdline:get_trigger_characters() return { ' ', '.', '#', '-', '=', '/', ':' } end
function cmdline:get_completions(context, callback)
local arguments = vim.split(context.line, ' ', { plain = true })
local arg_number = #vim.split(context.line:sub(1, context.cursor[2] + 1), ' ', { plain = true })
local text_before_argument = table.concat(require('blink.cmp.lib.utils').slice(arguments, 1, arg_number - 1), ' ')
.. (arg_number > 1 and ' ' or '')
local current_arg = arguments[arg_number]
local keyword_config = require('blink.cmp.config').completion.keyword
local keyword = context.get_bounds(keyword_config.range)
local current_arg_prefix = current_arg:sub(1, keyword.start_col - #text_before_argument - 1)
local task = async.task
.empty()
:map(function()
-- Special case for help where we read all the tags ourselves
if vim.tbl_contains(constants.help_commands, arguments[1] or '') then
return require('blink.cmp.sources.cmdline.help').get_completions(current_arg_prefix)
end
local completions = {}
local completion_type = vim.fn.getcmdcompltype()
-- Handle custom completions explicitly, since otherwise they won't work in input() mode (getcmdtype() == '@')
if vim.startswith(completion_type, 'custom,') or vim.startswith(completion_type, 'customlist,') then
local fun = completion_type:gsub('custom,', ''):gsub('customlist,', '')
completions = vim.fn.call(fun, { current_arg_prefix, vim.fn.getcmdline(), vim.fn.getcmdpos() })
-- `custom,` type returns a string, delimited by newlines
if type(completions) == 'string' then completions = vim.split(completions, '\n') end
else
local query = (text_before_argument .. current_arg_prefix):gsub([[\\]], [[\\\\]])
completions = vim.fn.getcompletion(query, 'cmdline')
end
-- Special case for files, escape special characters
if vim.tbl_contains(constants.file_commands, arguments[1] or '') then
completions = vim.tbl_map(function(completion) return vim.fn.fnameescape(completion) end, completions)
end
return completions
end)
:map(function(completions)
local items = {}
for _, completion in ipairs(completions) do
local has_prefix = string.find(completion, current_arg_prefix, 1, true) == 1
-- remove prefix from the filter text
local filter_text = completion
if has_prefix then filter_text = completion:sub(#current_arg_prefix + 1) end
-- for lua, use the filter text as the label since it doesn't include the prefix
local label = arguments[1] == 'lua' and filter_text or completion
-- add prefix to the newText
local new_text = completion
if not has_prefix then new_text = current_arg_prefix .. completion end
table.insert(items, {
label = label,
filterText = filter_text,
-- move items starting with special characters to the end of the list
sortText = label:lower():gsub('^([!-@\\[-`])', '~%1'),
textEdit = {
newText = new_text,
range = {
start = { line = 0, character = #text_before_argument },
['end'] = { line = 0, character = #text_before_argument + #current_arg },
},
},
kind = require('blink.cmp.types').CompletionItemKind.Property,
})
end
callback({
is_incomplete_backward = true,
is_incomplete_forward = false,
items = items,
})
end)
:catch(function(err)
vim.notify('Error while fetching completions: ' .. err, vim.log.levels.ERROR, { title = 'blink.cmp' })
callback({ is_incomplete_backward = false, is_incomplete_forward = false, items = {} })
end)
return function() task:cancel() end
end
return cmdline
|
