summaryrefslogtreecommitdiff
path: root/script/provider/completion.lua
blob: 3c0c82d748442425e79554174ff7e48b15bde08f (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
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
local proto  = require 'proto'
local nonil  = require 'without-check-nil'
local client = require 'client'
local config = require 'config'
local ws     = require 'workspace'

local isEnable = false

local function allWords()
    local str = '\t\n.:(\'"[,#*@|=-{/\\ +?'
    local mark = {}
    local list = {}
    for c in str:gmatch '.' do
        list[#list+1] = c
        mark[c] = true
    end
    for _, scp in ipairs(ws.folders) do
        local postfix = config.get(scp.uri, 'Lua.completion.postfix')
        if postfix ~= '' and not mark[postfix] then
            list[#list+1] = postfix
            mark[postfix] = true
        end
    end
    return list
end

local function enable(uri)
    if isEnable then
        return
    end
    nonil.enable()
    if not client.info.capabilities.textDocument.completion.dynamicRegistration then
        nonil.disable()
        return
    end
    nonil.disable()
    isEnable = true
    log.debug('Enable completion.')
    proto.request('client/registerCapability', {
        registrations = {
            {
                id = 'completion',
                method = 'textDocument/completion',
                registerOptions = {
                    resolveProvider = true,
                    triggerCharacters = allWords(),
                },
            },
        }
    })
end

local function disable(uri)
    if not isEnable then
        return
    end
    nonil.enable()
    if not client.info.capabilities.textDocument.completion.dynamicRegistration then
        nonil.disable()
        return
    end
    nonil.disable()
    isEnable = false
    log.debug('Disable completion.')
    proto.request('client/unregisterCapability', {
        unregisterations = {
            {
                id = 'completion',
                method = 'textDocument/completion',
            },
        }
    })
end

config.watch(function (uri, key, value)
    if key == '' then
        key   = 'Lua.completion.enable'
        value = config.get(uri, key)
    end
    if key == 'Lua.completion.enable' then
        if value == true then
            enable(uri)
        else
            disable(uri)
        end
    end
    if key == 'Lua.completion.postfix' then
        if config.get(uri, 'Lua.completion.enable') then
            disable(uri)
            enable(uri)
        end
    end
end)

return {
    enable   = enable,
    disable  = disable,
    allWords = allWords,
}