1 files changed, 86 insertions, 0 deletions
diff --git a/script/src/method/textDocument/references.lua b/script/src/method/textDocument/references.lua
new file mode 100644
index 00000000..0a198323
--- /dev/null
+++ b/script/src/method/textDocument/references.lua
@@ -0,0 +1,86 @@
+local core = require 'core'
+local LastTask
+
+local function findReferences(lsp, uri, position)
+    local vm = lsp:getVM(uri)
+
+    local positions, isGlobal = core.definition(vm, position, 'reference')
+    if not positions then
+        return nil, isGlobal
+    end
+
+    local locations = {}
+    for i, position in ipairs(positions) do
+        local start, finish, valueUri = position[1], position[2], (position[3] or uri)
+        local vm, valueLines = lsp:getVM(valueUri)
+        if valueLines then
+            local start_row,  start_col  = valueLines:rowcol(start)
+            local finish_row, finish_col = valueLines:rowcol(finish)
+            locations[#locations+1] = {
+                uri =  valueUri,
+                range = {
+                    start = {
+                        line = start_row - 1,
+                        character = start_col - 1,
+                    },
+                    ['end'] = {
+                        line = finish_row - 1,
+                        -- 这里不用-1，因为前端期待的是匹配完成后的位置
+                        character = finish_col,
+                    },
+                }
+            }
+        elseif vm then
+            locations[#locations+1] = {
+                uri =  valueUri,
+                range = {
+                    start = {
+                        line = 0,
+                        character = 0,
+                    },
+                    ['end'] = {
+                        line = 0,
+                        character = 0,
+                    },
+                }
+            }
+        end
+    end
+
+    if #locations == 0 then
+        return nil, isGlobal
+    end
+
+    return locations, isGlobal
+end
+
+return function (lsp, params)
+    local uri = params.textDocument.uri
+    local declarat = params.context.includeDeclaration
+    local vm, lines = lsp:loadVM(uri)
+    if not vm then
+        return nil
+    end
+
+    if LastTask then
+        LastTask:remove()
+        LastTask = nil
+    end
+
+    -- lua是从1开始的，因此都要+1
+    local position = lines:positionAsChar(params.position.line + 1, params.position.character)
+
+    return function (response)
+        local clock = os.clock()
+        LastTask = ac.loop(0.1, function ()
+            local positions, isGlobal = findReferences(lsp, uri, position)
+            if isGlobal and lsp:isWaitingCompile() and os.clock() - clock < 5 then
+                return
+            end
+            response(positions)
+            LastTask:remove()
+            LastTask = nil
+        end)
+        LastTask:onTimer()
+    end
+end