支持全局符号

3 files changed, 87 insertions, 1 deletions

diff --git a/script/method/init.lua b/script/method/init.lua
index cd9010bb..742590f6 100644
--- a/script/method/init.lua
+++ b/script/method/init.lua
@@ -30,5 +30,6 @@ init 'workspace/didChangeConfiguration'
 init 'workspace/didChangeWatchedFiles'
 init 'workspace/didChangeWorkspaceFolders'
 init 'workspace/executeCommand'
+init 'workspace/symbol'
 
 return method
diff --git a/script/method/initialize.lua b/script/method/initialize.lua
index 0f986dc1..5038a27f 100644
--- a/script/method/initialize.lua
+++ b/script/method/initialize.lua
@@ -38,6 +38,7 @@ return function (lsp, params)
             documentHighlightProvider = true,
             codeActionProvider = true,
             foldingRangeProvider = true,
+            workspaceSymbolProvider = true,
             signatureHelpProvider = {
                 triggerCharacters = { '(', ',' },
             },
@@ -52,7 +53,7 @@ return function (lsp, params)
                 workspaceFolders = {
                     supported = true,
                     changeNotifications = true,
-                }
+                },
             },
             documentOnTypeFormattingProvider = {
                 firstTriggerCharacter = '}',
diff --git a/script/method/workspace/symbol.lua b/script/method/workspace/symbol.lua
new file mode 100644
index 00000000..6ec2d025
--- /dev/null
+++ b/script/method/workspace/symbol.lua
@@ -0,0 +1,84 @@
+local matchKey = require 'core.matchKey'
+
+local SymbolKind = {
+    File = 1,
+    Module = 2,
+    Namespace = 3,
+    Package = 4,
+    Class = 5,
+    Method = 6,
+    Property = 7,
+    Field = 8,
+    Constructor = 9,
+    Enum = 10,
+    Interface = 11,
+    Function = 12,
+    Variable = 13,
+    Constant = 14,
+    String = 15,
+    Number = 16,
+    Boolean = 17,
+    Array = 18,
+    Object = 19,
+    Key = 20,
+    Null = 21,
+    EnumMember = 22,
+    Struct = 23,
+    Event = 24,
+    Operator = 25,
+    TypeParameter = 26,
+}
+
+local function convertRange(lines, range)
+    local start_row,  start_col  = lines:rowcol(range.start)
+    local finish_row, finish_col = lines:rowcol(range.finish)
+    local result = {
+        start = {
+            line = start_row - 1,
+            character = start_col - 1,
+        },
+        ['end'] = {
+            line = finish_row - 1,
+            -- 这里不用-1，因为前端期待的是匹配完成后的位置
+            character = finish_col,
+        },
+    }
+    return result
+end
+
+local function searchVM(lsp, results, query, uri)
+    local vm, lines = lsp:getVM(uri)
+    if not vm then
+        return
+    end
+    vm:eachSource(function (src)
+        if src.type == 'name' then
+            if matchKey(query, src[1]) then
+                results[#results+1] = {
+                    name = src[1],
+                    kind = SymbolKind.Variable,
+                    location = {
+                        uri = uri,
+                        range = convertRange(lines, src),
+                    }
+                }
+            end
+        end
+    end)
+end
+
+--- @param lsp LSP
+--- @param params table
+return function (lsp, params)
+    local query = params.query
+    if #query <= 0 then
+        return {}
+    end
+    local results = {}
+
+    for uri in lsp:eachFile() do
+        searchVM(lsp, results, query, uri)
+    end
+
+    return results
+end