summaryrefslogtreecommitdiff
path: root/script/capability/completion.lua
blob: e302f30d5d6fbf358489f89bd6eed287d8797d21 (plain)
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
local rpc = require 'rpc'
local nonil = require 'without-check-nil'

local isEnable = false

local function allWords()
    local str = [[abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789.:('"[,#*@| ]]
    local list = {}
    for c in str:gmatch '.' do
        list[#list+1] = c
    end
    return list
end

local function enable(lsp)
    if isEnable then
        return
    end

    nonil.enable()
    if not lsp.client.capabilities.textDocument.completion.dynamicRegistration then
        return
    end
    nonil.disable()

    isEnable = true
    log.debug('Enable completion.')
    rpc:request('client/registerCapability', {
        registrations = {
            {
                id = 'completion',
                method = 'textDocument/completion',
                registerOptions = {
                    resolveProvider = true,
                    triggerCharacters = allWords(),
                },
            },
        }
    })
end

local function disable(lsp)
    if not isEnable then
        return
    end

    nonil.enable()
    if not lsp.client.capabilities.textDocument.completion.dynamicRegistration then
        return
    end
    nonil.disable()

    isEnable = false
    log.debug('Disable completion.')
    rpc:request('client/unregisterCapability', {
        unregisterations = {
            {
                id = 'completion',
                method = 'textDocument/completion',
            },
        }
    })
end

return {
    enable = enable,
    disable = disable,
}