支持转到实现

3 files changed, 48 insertions, 1 deletions

diff --git a/src/method/init.lua b/src/method/init.lua
index 214997d7..46b79cc4 100644
--- a/src/method/init.lua
+++ b/src/method/init.lua
@@ -7,6 +7,7 @@ end
 init 'initialize'
 init 'initialized'
 init 'shutdown'
+init 'textDocument/implementation'
 init 'textDocument/definition'
 init 'textDocument/didOpen'
 init 'textDocument/didChange'
diff --git a/src/method/initialize.lua b/src/method/initialize.lua
index ea5bb3c4..866ded66 100644
--- a/src/method/initialize.lua
+++ b/src/method/initialize.lua
@@ -5,7 +5,7 @@ return function (lsp, data)
             -- 支持“转到定义”
             definitionProvider = true,
             -- 支持“转到实现”
-            --implementationProvider = true,
+            implementationProvider = true,
             -- 文本同步方式
             textDocumentSync = {
                 -- 打开关闭文本时通知
diff --git a/src/method/textDocument/implementation.lua b/src/method/textDocument/implementation.lua
new file mode 100644
index 00000000..95bd0c20
--- /dev/null
+++ b/src/method/textDocument/implementation.lua
@@ -0,0 +1,46 @@
+local parser = require 'parser'
+local matcher = require 'matcher'
+
+return function (lsp, params)
+    local uri = params.textDocument.uri
+    local text = lsp:loadText(uri)
+    if not text then
+        return nil, '找不到文件：' .. uri
+    end
+    local start_clock = os.clock()
+    -- lua是从1开始的，因此都要+1
+    local pos = parser.calcline.position_utf8(text, params.position.line + 1, params.position.character + 1)
+    local suc, start, finish = matcher.implementation(text, pos)
+    if not suc then
+        if finish then
+            log.debug(start, uri)
+            finish.lua = nil
+            log.debug(table.dump(finish))
+        end
+        return {}
+    end
+
+    local start_row,  start_col  = parser.calcline.rowcol_utf8(text, start)
+    local finish_row, finish_col = parser.calcline.rowcol_utf8(text, finish)
+
+    local response = {
+        uri = uri,
+        range = {
+            start = {
+                line = start_row - 1,
+                character = start_col - 1,
+            },
+            ['end'] = {
+                line = finish_row - 1,
+                -- 这里不用-1，因为前端期待的是匹配完成后的位置
+                character = finish_col,
+            },
+        },
+    }
+    local passed_clock = os.clock() - start_clock
+    if passed_clock >= 0.01 then
+        log.warn(('[转到实现]耗时[%.3f]秒，文件大小[%s]字节'):format(passed_clock, #text))
+    end
+
+    return response
+end