From d0ff66c9abe9d6abbca12fd811e0c3cb69c1033a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= <sumneko@hotmail.com>
Date: Fri, 22 Nov 2019 23:26:32 +0800
Subject: =?UTF-8?q?=E6=95=B4=E7=90=86=E4=B8=80=E4=B8=8B=E7=9B=AE=E5=BD=95?=
 =?UTF-8?q?=E7=BB=93=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script/src/method/textDocument/completion.lua | 104 ++++++++++++++++++++++++++
 1 file changed, 104 insertions(+)
 create mode 100644 script/src/method/textDocument/completion.lua

(limited to 'script/src/method/textDocument/completion.lua')

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
-- 
cgit v1.2.3