暂时先这样实现吧

4 files changed, 77 insertions, 48 deletions

diff --git a/server/src/core/document_symbol.lua b/server/src/core/document_symbol.lua
index 8e6fd01c..3902f8a3 100644
--- a/server/src/core/document_symbol.lua
+++ b/server/src/core/document_symbol.lua
@@ -30,7 +30,7 @@ local SymbolKind = {
     TypeParameter = 26,
 }
 
-local function buildFunc(vm, func)
+local function buildFunction(vm, func)
     local source = func.source
     local declarat = func.declarat
     local name
@@ -75,10 +75,23 @@ local function buildFunc(vm, func)
     }
 end
 
+local function buildChunk(vm, chunk)
+    local symbol = buildFunction(vm, chunk.func)
+    local childs = {}
+    for i, child in vm.chunk:childs(chunk) do
+        childs[i] = buildChunk(vm, child)
+    end
+    if #childs ~= 0 then
+        symbol.children = childs
+    end
+    return symbol
+end
+
 return function (vm)
     local symbols = {}
-    for _, func in ipairs(vm.results.funcs) do
-        symbols[#symbols+1] = buildFunc(vm, func)
+
+    for i, chunk in vm.chunk:childs() do
+        symbols[i] = buildChunk(vm, chunk)
     end
 
     return symbols
diff --git a/server/src/core/env.lua b/server/src/core/env.lua
index ada26145..bee90012 100644
--- a/server/src/core/env.lua
+++ b/server/src/core/env.lua
@@ -16,7 +16,13 @@ return function (root)
 
     local mt = { _env = env }
     function mt:push()
-        env[#env+1] = { _next = env[#env], _cut = {} }
+        local current = env[#env]
+        if not current._child then
+            current._child = {}
+        end
+        local new = { _next = env[#env], _cut = {} }
+        current._child[#current._child+1] = new
+        env[#env+1] = new
     end
     function mt:pop()
         env[#env] = nil
@@ -24,6 +30,15 @@ return function (root)
     function mt:cut(key)
         env[#env]._cut[key] = true
     end
+    function mt:childs(obj)
+        if not obj then
+            obj = env[#env]
+        end
+        if not obj._child then
+            return function () end
+        end
+        return ipairs(obj._child)
+    end
     function mt:__index(key)
         local origin = env[#env]
         if is_table[key] then
@@ -99,7 +114,7 @@ return function (root)
                 cuted[key] = true
             end
             for key, value in pairs(o) do
-                if key == '_cut' or key == '_next' then
+                if key == '_cut' or key == '_next' or key == '_child' then
                     goto CONTINUE
                 end
                 if cuted[key] then
diff --git a/server/src/core/vm.lua b/server/src/core/vm.lua
index fc00b581..f60fcb57 100644
--- a/server/src/core/vm.lua
+++ b/server/src/core/vm.lua
@@ -401,6 +401,7 @@ function mt:buildFunction(exp, object)
     self.chunk:cut 'dots'
     self.chunk:cut 'labels'
     self.chunk.func = func
+    self.chunk.locals = {}
 
     if object then
         local var = self:createArg('self', object.source, self:getValue(object))
@@ -1435,6 +1436,7 @@ function mt:createEnvironment()
     self.scope.block = { start = 0, finish = math.maxinteger }
     -- 整个文件是一个函数
     self.chunk.func = self:buildFunction()
+    self.chunk.locals = {}
     self.results.main = self.chunk.func
     -- 隐藏的上值`_ENV`
     local parent = self:createLocal('_ENV')
@@ -1464,7 +1466,6 @@ local function compile(ast, lsp, uri)
         },
         chunk   = env {
             labels = {},
-            locals = {},
         },
         results = {
             locals = {},
diff --git a/server/test/document_symbol/init.lua b/server/test/document_symbol/init.lua
index c8f16fbd..7aaef322 100644
--- a/server/test/document_symbol/init.lua
+++ b/server/test/document_symbol/init.lua
@@ -154,45 +154,45 @@ end
     }
 }
 
---TEST [[
---function A()
---    function A1()
---    end
---    function A2()
---    end
---end
---function B()
---end
---]]
---{
---    [1] = {
---        name = 'A',
---        detail = 'function A()',
---        kind = SymbolKind.Function,
---        range = {1, 68},
---        selectionRange = {10, 10},
---        children = {
---            [1] = {
---                name = 'A1',
---                detail = 'function A1()',
---                kind = SymbolKind.Function,
---                range = {18, 38},
---                selectionRange = {27, 28},
---            },
---            [2] = {
---                name = 'A2',
---                detail = 'function A2()',
---                kind = SymbolKind.Function,
---                range = {44, 64},
---                selectionRange = {53, 54},
---            },
---        },
---    },
---    [2] = {
---        name = 'B',
---        detail = 'function B()',
---        kind = SymbolKind.Function,
---        range = {70, 85},
---        selectionRange = {79, 79},
---    },
---}
+TEST [[
+function A()
+    function A1()
+    end
+    function A2()
+    end
+end
+function B()
+end
+]]
+{
+    [1] = {
+        name = 'A',
+        detail = 'function A()',
+        kind = SymbolKind.Function,
+        range = {1, 68},
+        selectionRange = {10, 10},
+        children = {
+            [1] = {
+                name = 'A1',
+                detail = 'function A1()',
+                kind = SymbolKind.Function,
+                range = {18, 38},
+                selectionRange = {27, 28},
+            },
+            [2] = {
+                name = 'A2',
+                detail = 'function A2()',
+                kind = SymbolKind.Function,
+                range = {44, 64},
+                selectionRange = {53, 54},
+            },
+        },
+    },
+    [2] = {
+        name = 'B',
+        detail = 'function B()',
+        kind = SymbolKind.Function,
+        range = {70, 85},
+        selectionRange = {79, 79},
+    },
+}