diff options
Diffstat (limited to 'script/src/method/textDocument/completion.lua')
-rw-r--r-- | script/src/method/textDocument/completion.lua | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/script/src/method/textDocument/completion.lua b/script/src/method/textDocument/completion.lua new file mode 100644 index 00000000..4c7581df --- /dev/null +++ b/script/src/method/textDocument/completion.lua @@ -0,0 +1,104 @@ +local core = require 'core' +local parser = require 'parser' + +local function posToRange(lines, start, finish) + local start_row, start_col = lines:rowcol(start) + local finish_row, finish_col = lines:rowcol(finish) + return { + start = { + line = start_row - 1, + character = start_col - 1, + }, + ['end'] = { + line = finish_row - 1, + character = finish_col, + }, + } +end + +local function fastCompletion(lsp, params, lines) + local uri = params.textDocument.uri + local text, oldText = lsp:getText(uri) + -- lua是从1开始的,因此都要+1 + local position = lines:positionAsChar(params.position.line + 1, params.position.character) + + local vm = lsp:getVM(uri) + if not vm then + vm = lsp:loadVM(uri) + if not vm then + return nil + end + end + + local items = core.completion(vm, text, position, oldText) + if not items or #items == 0 then + vm = lsp:loadVM(uri) + if not vm then + return nil + end + items = core.completion(vm, text, position) + if not items or #items == 0 then + return nil + end + end + + return items +end + +local function finishCompletion(lsp, params, lines) + local uri = params.textDocument.uri + local text = lsp:getText(uri) + -- lua是从1开始的,因此都要+1 + local position = lines:positionAsChar(params.position.line + 1, params.position.character) + + local vm = lsp:loadVM(uri) + if not vm then + return nil + end + + local items = core.completion(vm, text, position) + if not items or #items == 0 then + return nil + end + + return items +end + +return function (lsp, params) + local uri = params.textDocument.uri + local text, oldText = lsp:getText(uri) + if not text then + return nil + end + + local lines = parser:lines(text, 'utf8') + local items = fastCompletion(lsp, params, lines) + --local items = finishCompletion(lsp, params, lines) + if not items then + return nil + end + + for i, item in ipairs(items) do + item.sortText = ('%04d'):format(i) + item.insertTextFormat = 2 + item.insertText = item.insertText or item.label + if item.textEdit then + item.textEdit.range = posToRange(lines, item.textEdit.start, item.textEdit.finish) + item.textEdit.start = nil + item.textEdit.finish = nil + end + if item.additionalTextEdits then + for _, textEdit in ipairs(item.additionalTextEdits) do + textEdit.range = posToRange(lines, textEdit.start, textEdit.finish) + textEdit.start = nil + textEdit.finish = nil + end + end + end + + local response = { + isIncomplete = true, + items = items, + } + return response +end |