summaryrefslogtreecommitdiff
path: root/script-beta/vm
diff options
context:
space:
mode:
author最萌小汐 <sumneko@hotmail.com>2019-12-16 22:02:14 +0800
committer最萌小汐 <sumneko@hotmail.com>2019-12-16 22:02:14 +0800
commitdad5101542e0a0a84a3b66df88faa9b6179210a6 (patch)
treee7c7ebe3c95a3e6f446a0990c26c3a743b672e76 /script-beta/vm
parent1fcdba8cba71e6db67272550071c4e37aceb1994 (diff)
downloadlua-language-server-dad5101542e0a0a84a3b66df88faa9b6179210a6.zip
整理代码
Diffstat (limited to 'script-beta/vm')
-rw-r--r--script-beta/vm/eachRef.lua577
-rw-r--r--script-beta/vm/init.lua1
-rw-r--r--script-beta/vm/refOf.lua466
3 files changed, 269 insertions, 775 deletions
diff --git a/script-beta/vm/eachRef.lua b/script-beta/vm/eachRef.lua
index f08423a5..dc97c04d 100644
--- a/script-beta/vm/eachRef.lua
+++ b/script-beta/vm/eachRef.lua
@@ -3,8 +3,153 @@ local files = require 'files'
local vm = require 'vm.vm'
local library = require 'library'
local await = require 'await'
+local vm = require 'vm.vm'
+local guide = require 'parser.guide'
+local files = require 'files'
+local library = require 'library'
+
+local function ofSelf(state, loc, callback)
+ -- self 的2个特殊引用位置:
+ -- 1. 当前方法定义时的对象(mt)
+ local method = loc.method
+ local node = method.node
+ callback(node)
+ -- 2. 调用该方法时传入的对象
+end
+
+local function ofLocal(state, loc, callback)
+ if state[loc] then
+ return
+ end
+ state[loc] = true
+ -- 方法中的 self 使用了一个虚拟的定义位置
+ if loc.tag ~= 'self' then
+ callback(loc, 'declare')
+ end
+ local refs = loc.ref
+ if refs then
+ for i = 1, #refs do
+ local ref = refs[i]
+ if ref.type == 'getlocal' then
+ callback(ref, 'get')
+ if loc.tag == '_ENV' then
+ local parent = ref.parent
+ if parent.type == 'getfield'
+ or parent.type == 'getindex' then
+ if guide.getKeyName(parent) == '_G' then
+ callback(parent, 'declare')
+ end
+ end
+ end
+ elseif ref.type == 'setlocal' then
+ callback(ref, 'set')
+ elseif ref.type == 'getglobal' then
+ if loc.tag == '_ENV' then
+ if guide.getName(ref) == '_G' then
+ callback(ref, 'get')
+ end
+ end
+ end
+ end
+ end
+ if loc.tag == 'self' then
+ ofSelf(state, loc, callback)
+ end
+end
+
+local function ofGlobal(state, source, callback)
+ if state[source] then
+ return
+ end
+ local key = guide.getKeyName(source)
+ local node = source.node
+ if node.tag == '_ENV' then
+ local uris = files.findGlobals(key)
+ for i = 1, #uris do
+ local uri = uris[i]
+ local ast = files.getAst(uri)
+ local globals = vm.getGlobals(ast.ast)
+ if globals and globals[key] then
+ for _, info in ipairs(globals[key]) do
+ state[info.source] = true
+ callback(info)
+ end
+ end
+ end
+ else
+ vm.eachField(node, function (info)
+ if key == info.key then
+ state[info.source] = true
+ callback(info)
+ end
+ end)
+ end
+end
+
+local function ofField(state, source, callback)
+ if state[source] then
+ return
+ end
+ local parent = source.parent
+ local key = guide.getKeyName(source)
+ if parent.type == 'tablefield'
+ or parent.type == 'tableindex' then
+ local tbl = parent.parent
+ vm.eachField(tbl, function (info)
+ if key == info.key then
+ state[info.source] = true
+ callback(info)
+ end
+ end)
+ else
+ local node = parent.node
+ vm.eachField(node, function (info)
+ if key == info.key then
+ state[info.source] = true
+ callback(info)
+ end
+ end)
+ end
+end
+
+local function ofLabel(state, source, callback)
+ if state[source] then
+ return
+ end
+ state[source] = true
+ callback(source, 'set')
+ if source.ref then
+ for _, ref in ipairs(source.ref) do
+ callback(ref, 'get')
+ end
+ end
+end
+
+local function ofGoTo(state, source, callback)
+ local name = source[1]
+ local label = guide.getLabel(source, name)
+ if label then
+ ofLabel(state, label, callback)
+ end
+end
+
+local function ofValue(state, source, callback)
+ callback(source, 'value')
+end
-local function ofCall(func, index, callback, offset)
+local function ofIndex(state, source, callback)
+ local parent = source.parent
+ if not parent then
+ return
+ end
+ if parent.type == 'setindex'
+ or parent.type == 'getindex'
+ or parent.type == 'tableindex' then
+ ofField(state, source, callback)
+ end
+end
+
+local function ofCall(state, func, index, callback, offset)
offset = offset or 0
vm.eachRef(func, function (info)
local src = info.source
@@ -14,53 +159,30 @@ local function ofCall(func, index, callback, offset)
end
if returns then
-- 搜索函数第 index 个返回值
- for _, rtn in ipairs(returns) do
+ for i = 1, #returns do
+ local rtn = returns[i]
local val = rtn[index-offset]
if val then
- callback {
- source = val,
- mode = 'return',
- }
- vm.eachRef(val, callback)
+ callback(val)
end
end
end
end)
end
-local function ofCallSelect(call, index, callback)
- local slc = call.parent
- if slc.index == index then
- vm.eachRef(slc.parent, callback)
- return
- end
- if call.extParent then
- for i = 1, #call.extParent do
- slc = call.extParent[i]
- if slc.index == index then
- vm.eachRef(slc.parent, callback)
- return
- end
- end
- end
-end
-
-local function ofSpecialCall(call, func, index, callback, offset)
+local function ofSpecialCall(state, call, func, index, callback, offset)
local name = func.special
offset = offset or 0
if name == 'setmetatable' then
if index == 1 + offset then
local args = call.args
if args[1+offset] then
- vm.eachRef(args[1+offset], callback)
+ callback(args[1+offset])
end
if args[2+offset] then
vm.eachField(args[2+offset], function (info)
if info.key == 's|__index' then
- vm.eachRef(info.source, callback)
- if info.value then
- vm.eachRef(info.value, callback)
- end
+ callback(info.source)
end
end)
end
@@ -71,11 +193,12 @@ local function ofSpecialCall(call, func, index, callback, offset)
local result = vm.getLinkUris(call)
if result then
local myUri = guide.getRoot(call).uri
- for _, uri in ipairs(result) do
+ for i = 1, #result do
+ local uri = result[i]
if not files.eq(uri, myUri) then
local ast = files.getAst(uri)
if ast then
- ofCall(ast.ast, 1, callback)
+ ofCall(state, ast.ast, 1, callback)
end
end
end
@@ -87,10 +210,7 @@ local function ofSpecialCall(call, func, index, callback, offset)
local objName = args[1+offset][1]
local lib = library.library[objName]
if lib then
- callback {
- source = lib,
- mode = 'value',
- }
+ callback(lib)
end
end
end
@@ -103,8 +223,8 @@ local function ofSpecialCall(call, func, index, callback, offset)
vm.eachRef(args[1+offset], function (info)
local src = info.source
if src.type == 'function' then
- ofCall(src, index, callback, 1+offset)
- ofSpecialCall(call, src, index, callback, 1+offset)
+ ofCall(state, src, index, callback, 1+offset)
+ ofSpecialCall(state, call, src, index, callback, 1+offset)
end
end)
end
@@ -112,59 +232,17 @@ local function ofSpecialCall(call, func, index, callback, offset)
end
end
-local function asSetValue(value, callback)
- if value.type == 'field'
- or value.type == 'method' then
- value = value.parent
- end
- local parent = value.parent
- if not parent then
- return
- end
- if parent.type == 'local'
- or parent.type == 'setglobal'
- or parent.type == 'setlocal'
- or parent.type == 'setfield'
- or parent.type == 'setmethod'
- or parent.type == 'setindex'
- or parent.type == 'tablefield'
- or parent.type == 'tableindex' then
- if parent.value == value then
- vm.eachRef(parent, callback)
- if guide.getName(parent) == '__index' then
- if parent.type == 'tablefield'
- or parent.type == 'tableindex' then
- local t = parent.parent
- local args = t.parent
- if args[2] == t then
- local call = args.parent
- local func = call.node
- if func.special == 'setmetatable' then
- vm.eachRef(args[1], callback)
- end
- end
- end
- end
- end
- end
-end
-
-local function ofSelect(source, callback)
+local function ofSelect(state, source, callback)
-- 检查函数返回值
local call = source.vararg
if call.type == 'call' then
- ofCall(call.node, source.index, callback)
- ofSpecialCall(call, call.node, source.index, callback)
+ ofCall(state, call.node, source.index, callback)
+ ofSpecialCall(state, call, call.node, source.index, callback)
end
end
-local function ofSelf(loc, callback)
- -- self 的2个特殊引用位置:
- -- 1. 当前方法定义时的对象(mt)
- local method = loc.method
- local node = method.node
- vm.eachRef(node, callback)
- -- 2. 调用该方法时传入的对象
+local function ofMain(state, source, callback)
+ callback(source, 'main')
end
local function getCallRecvs(call)
@@ -176,7 +254,8 @@ local function getCallRecvs(call)
local recvs = {}
recvs[1] = parent.parent
if extParent then
- for _, p in ipairs(extParent) do
+ for i = 1, #extParent do
+ local p = extParent[i]
recvs[#recvs+1] = p.parent
end
end
@@ -184,7 +263,7 @@ local function getCallRecvs(call)
end
--- 自己作为函数的参数
-local function asArg(source, callback)
+local function checkAsArg(state, source, callback)
local parent = source.parent
if not parent then
return
@@ -198,16 +277,13 @@ local function asArg(source, callback)
if parent[2] then
vm.eachField(parent[2], function (info)
if info.key == 's|__index' then
- vm.eachRef(info.source, callback)
- if info.value then
- vm.eachRef(info.value, callback)
- end
+ callback(info)
end
end)
end
local recvs = getCallRecvs(call)
if recvs and recvs[1] then
- vm.eachRef(recvs[1], callback)
+ callback(recvs[1])
end
vm.setMeta(source, parent[2])
end
@@ -215,8 +291,25 @@ local function asArg(source, callback)
end
end
+local function ofCallSelect(state, call, index, callback)
+ local slc = call.parent
+ if slc.index == index then
+ callback(slc.parent)
+ return
+ end
+ if call.extParent then
+ for i = 1, #call.extParent do
+ slc = call.extParent[i]
+ if slc.index == index then
+ callback(slc.parent)
+ return
+ end
+ end
+ end
+end
+
--- 自己作为函数的返回值
-local function asReturn(source, callback)
+local function checkAsReturn(state, source, callback)
local parent = source.parent
if source.type == 'field'
or source.type == 'method' then
@@ -232,15 +325,16 @@ local function asReturn(source, callback)
if not uris then
return
end
- for _, uri in ipairs(uris) do
+ for i = 1, #uris do
+ local uri = uris[i]
local ast = files.getAst(uri)
if ast then
local links = vm.getLinks(ast.ast)
if links then
for linkUri, calls in pairs(links) do
if files.eq(linkUri, myUri) then
- for i = 1, #calls do
- ofCallSelect(calls[i], 1, callback)
+ for j = 1, #calls do
+ ofCallSelect(state, calls[j], 1, callback)
end
end
end
@@ -266,262 +360,114 @@ local function asReturn(source, callback)
end
local recvs = getCallRecvs(call)
if recvs and recvs[index] then
- vm.eachRef(recvs[index], callback)
+ callback(recvs[index])
elseif index == 1 then
- callback {
- type = 'call',
- source = call,
- }
+ callback(call, 'call')
end
end)
end
end
-local function ofLocal(loc, callback)
- -- 方法中的 self 使用了一个虚拟的定义位置
- if loc.tag ~= 'self' then
- callback {
- source = loc,
- mode = 'declare',
- }
- end
- if loc.ref then
- for _, ref in ipairs(loc.ref) do
- if ref.type == 'getlocal' then
- callback {
- source = ref,
- mode = 'get',
- }
- vm.eachRef(ref, callback)
- elseif ref.type == 'setlocal' then
- callback {
- source = ref,
- mode = 'set',
- }
- vm.eachRef(ref, callback)
- if ref.value then
- vm.eachRef(ref.value, callback)
- end
- end
- end
- end
- if loc.tag == 'self' then
- ofSelf(loc, callback)
- end
- if loc.value then
- vm.eachRef(loc.value, callback)
+local function checkAsParen(state, source, callback)
+ if state[source] then
+ return
end
- if loc.tag == '_ENV' and loc.ref then
- for _, ref in ipairs(loc.ref) do
- if ref.type == 'getlocal' then
- local parent = ref.parent
- if parent.type == 'getfield'
- or parent.type == 'getindex' then
- if guide.getKeyName(parent) == '_G' then
- callback {
- source = parent,
- mode = 'get',
- }
- end
- end
- elseif ref.type == 'getglobal' then
- if guide.getName(ref) == '_G' then
- callback {
- source = ref,
- mode = 'get',
- }
- end
- end
- end
+ state[source] = true
+ if source.parent and source.parent.type == 'paren' then
+ vm.refOf(state, source.parent, callback)
end
end
-local function ofGlobal(source, callback)
- local key = guide.getKeyName(source)
- local node = source.node
- if node.tag == '_ENV' then
- local uris = files.findGlobals(key)
- for _, uri in ipairs(uris) do
- local ast = files.getAst(uri)
- local globals = vm.getGlobals(ast.ast)
- if globals and globals[key] then
- for _, info in ipairs(globals[key]) do
- callback(info)
- if info.value then
- vm.eachRef(info.value, callback)
- end
- end
- end
- end
- else
- vm.eachField(node, function (info)
- if key == info.key then
- callback {
- source = info.source,
- mode = info.mode,
- }
- if info.value then
- vm.eachRef(info.value, callback)
- end
- end
- end)
+local function checkValue(state, source, callback)
+ if source.value then
+ callback(source.value)
end
end
-local function ofField(source, callback)
- if not source then
- return
- end
- local parent = source.parent
- local key = guide.getKeyName(source)
- if parent.type == 'tablefield'
- or parent.type == 'tableindex' then
- local tbl = parent.parent
- vm.eachField(tbl, function (info)
- if key == info.key then
- callback {
- source = info.source,
- mode = info.mode,
- }
- vm.eachRef(info.source, callback)
- if info.value then
- vm.eachRef(info.value, callback)
- end
- end
- end)
- else
- local node = parent.node
- vm.eachField(node, function (info)
- if key == info.key then
- callback {
- source = info.source,
- mode = info.mode,
- }
- vm.eachRef(info.source, callback)
- if info.value then
- vm.eachRef(info.value, callback)
- end
- end
- end)
+local function checkSetValue(state, value, callback)
+ if value.type == 'field'
+ or value.type == 'method' then
+ value = value.parent
end
-end
-
-local function ofLiteral(source, callback)
- local parent = source.parent
+ local parent = value.parent
if not parent then
return
end
- if parent.type == 'setindex'
- or parent.type == 'getindex'
+ if parent.type == 'local'
+ or parent.type == 'setglobal'
+ or parent.type == 'setlocal'
+ or parent.type == 'setfield'
+ or parent.type == 'setmethod'
+ or parent.type == 'setindex'
+ or parent.type == 'tablefield'
or parent.type == 'tableindex' then
- ofField(source, callback)
- end
-end
-
-local function ofLabel(source, callback)
- callback {
- source = source,
- mode = 'set',
- }
- if source.ref then
- for _, ref in ipairs(source.ref) do
- callback {
- source = ref,
- mode = 'get',
- }
+ if parent.value == value then
+ callback(parent)
+ if guide.getName(parent) == '__index' then
+ if parent.type == 'tablefield'
+ or parent.type == 'tableindex' then
+ local t = parent.parent
+ local args = t.parent
+ if args[2] == t then
+ local call = args.parent
+ local func = call.node
+ if func.special == 'setmetatable' then
+ callback(args[1])
+ end
+ end
+ end
+ end
end
end
end
-local function ofGoTo(source, callback)
- local name = source[1]
- local label = guide.getLabel(source, name)
- if label then
- ofLabel(label, callback)
- end
-end
-
-local function ofMain(source, callback)
- callback {
- source = source,
- mode = 'main',
- }
-end
-
-local function asParen(source, callback)
- if source.parent and source.parent.type == 'paren' then
- vm.eachRef(source.parent, callback)
- end
-end
-
-local function ofSelfValue(source, callback)
- callback {
- source = source,
- mode = 'value',
- }
-end
-
-local function eachRef(source, callback)
+function vm.refOf(state, source, callback)
local stype = source.type
- if stype == 'local' then
- ofLocal(source, callback)
+ if stype == 'local' then
+ ofLocal(state, source, callback)
elseif stype == 'getlocal'
or stype == 'setlocal' then
- ofLocal(source.node, callback)
+ ofLocal(state, source.node, callback)
elseif stype == 'setglobal'
or stype == 'getglobal' then
- ofGlobal(source, callback)
+ ofGlobal(state, source, callback)
elseif stype == 'field'
or stype == 'method' then
- ofField(source, callback)
+ ofField(state, source, callback)
elseif stype == 'setfield'
or stype == 'getfield'
or stype == 'tablefield' then
- ofField(source.field, callback)
+ ofField(state, source.field, callback)
elseif stype == 'setmethod'
or stype == 'getmethod' then
- ofField(source.method, callback)
+ ofField(state, source.method, callback)
+ elseif stype == 'goto' then
+ ofGoTo(state, source, callback)
+ elseif stype == 'label' then
+ ofLabel(state, source, callback)
elseif stype == 'number'
or stype == 'boolean'
or stype == 'string' then
- ofLiteral(source, callback)
- ofSelfValue(source, callback)
- elseif stype == 'goto' then
- ofGoTo(source, callback)
- elseif stype == 'label' then
- ofLabel(source, callback)
+ ofIndex(state, source, callback)
+ ofValue(state, source, callback)
elseif stype == 'table'
or stype == 'function' then
- ofSelfValue(source, callback)
+ ofValue(state, source, callback)
elseif stype == 'select' then
- ofSelect(source, callback)
+ ofSelect(state, source, callback)
elseif stype == 'call' then
- ofCall(source.node, 1, callback)
- ofSpecialCall(source, source.node, 1, callback)
+ ofCall(state, source.node, 1, callback)
+ ofSpecialCall(state, source, source.node, 1, callback)
elseif stype == 'main' then
- ofMain(source, callback)
+ ofMain(state, source, callback)
elseif stype == 'paren' then
- eachRef(source.exp, callback)
- end
- asArg(source, callback)
- asReturn(source, callback)
- asParen(source, callback)
- asSetValue(source, callback)
-end
-
---- 判断2个对象是否拥有相同的引用
-function vm.isSameRef(a, b)
- local cache = vm.cache.eachRef[a]
- if cache then
- -- 相同引用的source共享同一份cache
- return cache == vm.cache.eachRef[b]
- else
- return vm.eachRef(a, function (info)
- if info.source == b then
- return true
- end
- end) or false
+ vm.refOf(state, source.exp, callback)
end
+ checkValue(state, source, callback)
+ checkAsArg(state, source, callback)
+ checkAsReturn(state, source, callback)
+ checkAsParen(state, source, callback)
+ checkSetValue(state, source, callback)
end
local function applyCache(cache, callback, max)
@@ -578,6 +524,21 @@ local function eachRef(source, callback)
return result
end
+--- 判断2个对象是否拥有相同的引用
+function vm.isSameRef(a, b)
+ local cache = vm.cache.eachRef[a]
+ if cache then
+ -- 相同引用的source共享同一份cache
+ return cache == vm.cache.eachRef[b]
+ else
+ return vm.eachRef(a, function (info)
+ if info.source == b then
+ return true
+ end
+ end) or false
+ end
+end
+
--- 获取所有的引用
function vm.eachRef(source, callback, max)
local cache = vm.cache.eachRef[source]
diff --git a/script-beta/vm/init.lua b/script-beta/vm/init.lua
index 3645e77a..a4f81d07 100644
--- a/script-beta/vm/init.lua
+++ b/script-beta/vm/init.lua
@@ -1,6 +1,5 @@
local vm = require 'vm.vm'
require 'vm.eachField'
-require 'vm.refOf'
require 'vm.eachRef'
require 'vm.eachDef'
require 'vm.getGlobals'
diff --git a/script-beta/vm/refOf.lua b/script-beta/vm/refOf.lua
deleted file mode 100644
index f31c0ec8..00000000
--- a/script-beta/vm/refOf.lua
+++ /dev/null
@@ -1,466 +0,0 @@
-local vm = require 'vm.vm'
-local guide = require 'parser.guide'
-local files = require 'files'
-local library = require 'library'
-
-local function ofSelf(state, loc, callback)
- -- self 的2个特殊引用位置:
- -- 1. 当前方法定义时的对象(mt)
- local method = loc.method
- local node = method.node
- callback(node)
- -- 2. 调用该方法时传入的对象
-end
-
-local function ofLocal(state, loc, callback)
- if state[loc] then
- return
- end
- state[loc] = true
- -- 方法中的 self 使用了一个虚拟的定义位置
- if loc.tag ~= 'self' then
- callback(loc, 'declare')
- end
- local refs = loc.ref
- if refs then
- for i = 1, #refs do
- local ref = refs[i]
- if ref.type == 'getlocal' then
- callback(ref, 'get')
- if loc.tag == '_ENV' then
- local parent = ref.parent
- if parent.type == 'getfield'
- or parent.type == 'getindex' then
- if guide.getKeyName(parent) == '_G' then
- callback(parent, 'declare')
- end
- end
- end
- elseif ref.type == 'setlocal' then
- callback(ref, 'set')
- elseif ref.type == 'getglobal' then
- if loc.tag == '_ENV' then
- if guide.getName(ref) == '_G' then
- callback(ref, 'get')
- end
- end
- end
- end
- end
- if loc.tag == 'self' then
- ofSelf(state, loc, callback)
- end
-end
-
-local function ofGlobal(state, source, callback)
- if state[source] then
- return
- end
- local key = guide.getKeyName(source)
- local node = source.node
- if node.tag == '_ENV' then
- local uris = files.findGlobals(key)
- for i = 1, #uris do
- local uri = uris[i]
- local ast = files.getAst(uri)
- local globals = vm.getGlobals(ast.ast)
- if globals and globals[key] then
- for _, info in ipairs(globals[key]) do
- state[info.source] = true
- callback(info)
- end
- end
- end
- else
- vm.eachField(node, function (info)
- if key == info.key then
- state[info.source] = true
- callback(info)
- end
- end)
- end
-end
-
-local function ofField(state, source, callback)
- if state[source] then
- return
- end
- local parent = source.parent
- local key = guide.getKeyName(source)
- if parent.type == 'tablefield'
- or parent.type == 'tableindex' then
- local tbl = parent.parent
- vm.eachField(tbl, function (info)
- if key == info.key then
- state[info.source] = true
- callback(info)
- end
- end)
- else
- local node = parent.node
- vm.eachField(node, function (info)
- if key == info.key then
- state[info.source] = true
- callback(info)
- end
- end)
- end
-end
-
-local function ofLabel(state, source, callback)
- if state[source] then
- return
- end
- state[source] = true
- callback(source, 'set')
- if source.ref then
- for _, ref in ipairs(source.ref) do
- callback(ref, 'get')
- end
- end
-end
-
-local function ofGoTo(state, source, callback)
- local name = source[1]
- local label = guide.getLabel(source, name)
- if label then
- ofLabel(state, label, callback)
- end
-end
-
-local function ofValue(state, source, callback)
- callback(source, 'value')
-end
-
-local function ofIndex(state, source, callback)
- local parent = source.parent
- if not parent then
- return
- end
- if parent.type == 'setindex'
- or parent.type == 'getindex'
- or parent.type == 'tableindex' then
- ofField(state, source, callback)
- end
-end
-
-local function ofCall(state, func, index, callback, offset)
- offset = offset or 0
- vm.eachRef(func, function (info)
- local src = info.source
- local returns
- if src.type == 'main' or src.type == 'function' then
- returns = src.returns
- end
- if returns then
- -- 搜索函数第 index 个返回值
- for i = 1, #returns do
- local rtn = returns[i]
- local val = rtn[index-offset]
- if val then
- callback(val)
- end
- end
- end
- end)
-end
-
-local function ofSpecialCall(state, call, func, index, callback, offset)
- local name = func.special
- offset = offset or 0
- if name == 'setmetatable' then
- if index == 1 + offset then
- local args = call.args
- if args[1+offset] then
- callback(args[1+offset])
- end
- if args[2+offset] then
- vm.eachField(args[2+offset], function (info)
- if info.key == 's|__index' then
- callback(info.source)
- end
- end)
- end
- vm.setMeta(args[1+offset], args[2+offset])
- end
- elseif name == 'require' then
- if index == 1 + offset then
- local result = vm.getLinkUris(call)
- if result then
- local myUri = guide.getRoot(call).uri
- for i = 1, #result do
- local uri = result[i]
- if not files.eq(uri, myUri) then
- local ast = files.getAst(uri)
- if ast then
- ofCall(state, ast.ast, 1, callback)
- end
- end
- end
- end
-
- local args = call.args
- if args[1+offset] then
- if args[1+offset].type == 'string' then
- local objName = args[1+offset][1]
- local lib = library.library[objName]
- if lib then
- callback(lib)
- end
- end
- end
- end
- elseif name == 'pcall'
- or name == 'xpcall' then
- if index >= 2-offset then
- local args = call.args
- if args[1+offset] then
- vm.eachRef(args[1+offset], function (info)
- local src = info.source
- if src.type == 'function' then
- ofCall(state, src, index, callback, 1+offset)
- ofSpecialCall(state, call, src, index, callback, 1+offset)
- end
- end)
- end
- end
- end
-end
-
-local function ofSelect(state, source, callback)
- -- 检查函数返回值
- local call = source.vararg
- if call.type == 'call' then
- ofCall(state, call.node, source.index, callback)
- ofSpecialCall(state, call, call.node, source.index, callback)
- end
-end
-
-local function ofMain(state, source, callback)
- callback(source, 'main')
-end
-
-local function getCallRecvs(call)
- local parent = call.parent
- if parent.type ~= 'select' then
- return nil
- end
- local extParent = call.extParent
- local recvs = {}
- recvs[1] = parent.parent
- if extParent then
- for i = 1, #extParent do
- local p = extParent[i]
- recvs[#recvs+1] = p.parent
- end
- end
- return recvs
-end
-
---- 自己作为函数的参数
-local function checkAsArg(state, source, callback)
- local parent = source.parent
- if not parent then
- return
- end
- if parent.type == 'callargs' then
- local call = parent.parent
- local func = call.node
- local name = func.special
- if name == 'setmetatable' then
- if parent[1] == source then
- if parent[2] then
- vm.eachField(parent[2], function (info)
- if info.key == 's|__index' then
- callback(info)
- end
- end)
- end
- local recvs = getCallRecvs(call)
- if recvs and recvs[1] then
- callback(recvs[1])
- end
- vm.setMeta(source, parent[2])
- end
- end
- end
-end
-
-local function ofCallSelect(state, call, index, callback)
- local slc = call.parent
- if slc.index == index then
- callback(slc.parent)
- return
- end
- if call.extParent then
- for i = 1, #call.extParent do
- slc = call.extParent[i]
- if slc.index == index then
- callback(slc.parent)
- return
- end
- end
- end
-end
-
---- 自己作为函数的返回值
-local function checkAsReturn(state, source, callback)
- local parent = source.parent
- if source.type == 'field'
- or source.type == 'method' then
- parent = parent.parent
- end
- if not parent or parent.type ~= 'return' then
- return
- end
- local func = guide.getParentFunction(source)
- if func.type == 'main' then
- local myUri = func.uri
- local uris = files.findLinkTo(myUri)
- if not uris then
- return
- end
- for i = 1, #uris do
- local uri = uris[i]
- local ast = files.getAst(uri)
- if ast then
- local links = vm.getLinks(ast.ast)
- if links then
- for linkUri, calls in pairs(links) do
- if files.eq(linkUri, myUri) then
- for j = 1, #calls do
- ofCallSelect(state, calls[j], 1, callback)
- end
- end
- end
- end
- end
- end
- else
- local index
- for i = 1, #parent do
- if parent[i] == source then
- index = i
- break
- end
- end
- if not index then
- return
- end
- vm.eachRef(func, function (info)
- local src = info.source
- local call = src.parent
- if not call or call.type ~= 'call' then
- return
- end
- local recvs = getCallRecvs(call)
- if recvs and recvs[index] then
- callback(recvs[index])
- elseif index == 1 then
- callback(call, 'call')
- end
- end)
- end
-end
-
-local function checkAsParen(state, source, callback)
- if state[source] then
- return
- end
- state[source] = true
- if source.parent and source.parent.type == 'paren' then
- vm.refOf(state, source.parent, callback)
- end
-end
-
-local function checkValue(state, source, callback)
- if source.value then
- callback(source.value)
- end
-end
-
-local function checkSetValue(state, value, callback)
- if value.type == 'field'
- or value.type == 'method' then
- value = value.parent
- end
- local parent = value.parent
- if not parent then
- return
- end
- if parent.type == 'local'
- or parent.type == 'setglobal'
- or parent.type == 'setlocal'
- or parent.type == 'setfield'
- or parent.type == 'setmethod'
- or parent.type == 'setindex'
- or parent.type == 'tablefield'
- or parent.type == 'tableindex' then
- if parent.value == value then
- callback(parent)
- if guide.getName(parent) == '__index' then
- if parent.type == 'tablefield'
- or parent.type == 'tableindex' then
- local t = parent.parent
- local args = t.parent
- if args[2] == t then
- local call = args.parent
- local func = call.node
- if func.special == 'setmetatable' then
- callback(args[1])
- end
- end
- end
- end
- end
- end
-end
-
-function vm.refOf(state, source, callback)
- local stype = source.type
- if stype == 'local' then
- ofLocal(state, source, callback)
- elseif stype == 'getlocal'
- or stype == 'setlocal' then
- ofLocal(state, source.node, callback)
- elseif stype == 'setglobal'
- or stype == 'getglobal' then
- ofGlobal(state, source, callback)
- elseif stype == 'field'
- or stype == 'method' then
- ofField(state, source, callback)
- elseif stype == 'setfield'
- or stype == 'getfield'
- or stype == 'tablefield' then
- ofField(state, source.field, callback)
- elseif stype == 'setmethod'
- or stype == 'getmethod' then
- ofField(state, source.method, callback)
- elseif stype == 'goto' then
- ofGoTo(state, source, callback)
- elseif stype == 'label' then
- ofLabel(state, source, callback)
- elseif stype == 'number'
- or stype == 'boolean'
- or stype == 'string' then
- ofIndex(state, source, callback)
- ofValue(state, source, callback)
- elseif stype == 'table'
- or stype == 'function' then
- ofValue(state, source, callback)
- elseif stype == 'select' then
- ofSelect(state, source, callback)
- elseif stype == 'call' then
- ofCall(state, source.node, 1, callback)
- ofSpecialCall(state, source, source.node, 1, callback)
- elseif stype == 'main' then
- ofMain(state, source, callback)
- elseif stype == 'paren' then
- vm.refOf(state, source.exp, callback)
- end
- checkValue(state, source, callback)
- checkAsArg(state, source, callback)
- checkAsReturn(state, source, callback)
- checkAsParen(state, source, callback)
- checkSetValue(state, source, callback)
-end