1 files changed, 104 insertions, 0 deletions
diff --git a/script/method/textDocument/completion.lua b/script/method/textDocument/completion.lua
new file mode 100644
index 00000000..4c7581df
--- /dev/null
+++ b/script/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