local guide = require 'parser.guide' local util = require 'utility' local config = require 'config' local rpath = require 'workspace.require-path' local files = require 'files' ---@class vm local vm = require 'vm.vm' ---@class parser.object ---@field _compiledNodes boolean ---@field _node vm.node ---@field package _globalBase table ---@field cindex integer ---@field func parser.object -- 该函数有副作用,会给source绑定node! ---@param source parser.object ---@return boolean local function bindDocs(source) local docs = source.bindDocs if not docs then return false end for i = #docs, 1, -1 do local doc = docs[i] if doc.type == 'doc.type' then vm.setNode(source, vm.compileNode(doc)) return true end if doc.type == 'doc.class' then vm.setNode(source, vm.compileNode(doc)) return true end if doc.type == 'doc.param' then local node = vm.compileNode(doc) if doc.optional then node:addOptional() end vm.setNode(source, node) return true end if doc.type == 'doc.module' then local name = doc.module if not name then return true end local uri = rpath.findUrisByRequireName(guide.getUri(source), name)[1] if not uri then return true end local state = files.getState(uri) local ast = state and state.ast if not ast then return true end vm.setNode(source, vm.compileNode(ast)) return true end end return false end ---@param source parser.object ---@param key any ---@param ref boolean ---@param pushResult fun(res: parser.object, markDoc?: boolean) local function searchFieldByLocalID(source, key, ref, pushResult) local fields if key then fields = vm.getLocalSourcesSets(source, key) if ref then local gets = vm.getLocalSourcesGets(source, key) if gets then fields = fields or {} for _, src in ipairs(gets) do fields[#fields+1] = src end end end else fields = vm.getLocalFields(source, false) end if not fields then return end local hasMarkDoc = {} for _, src in ipairs(fields) do if src.bindDocs then if bindDocs(src) then local skey = guide.getKeyName(src) if skey then hasMarkDoc[skey] = true end pushResult(src, true) end end end for _, src in ipairs(fields) do local skey = guide.getKeyName(src) if not hasMarkDoc[skey] then pushResult(src) end end end ---@param suri uri ---@param source parser.object ---@param key any ---@param ref boolean ---@param pushResult fun(res: parser.object, markDoc?: boolean) local function searchFieldByGlobalID(suri, source, key, ref, pushResult) local node = vm.getGlobalNode(source) if not node then return end if node.cate == 'variable' then if key then if type(key) ~= 'string' then return end local global = vm.getGlobal('variable', node.name, key) if global then for _, set in ipairs(global:getSets(suri)) do pushResult(set) end for _, get in ipairs(global:getGets(suri)) do pushResult(get) end end else local globals = vm.getGlobalFields('variable', node.name) for _, global in ipairs(globals) do for _, set in ipairs(global:getSets(suri)) do pushResult(set) end for _, get in ipairs(global:getGets(suri)) do pushResult(get) end end end end if node.cate == 'type' then vm.getClassFields(suri, node, key, ref, pushResult) end end local searchFieldSwitch = util.switch() : case 'table' : call(function (suri, source, key, ref, pushResult) local hasFiled = false for _, field in ipairs(source) do if field.type == 'tablefield' or field.type == 'tableindex' then local fieldKey = guide.getKeyName(field) if key == nil or key == fieldKey then hasFiled = true pushResult(field) end end if field.type == 'tableexp' then if key == nil or key == field.tindex then hasFiled = true pushResult(field) end end if field.type == 'varargs' then if not hasFiled and type(key) == 'number' and key >= 1 and math.tointeger(key) then hasFiled = true pushResult(field) end if key == nil then pushResult(field) end end end end) : case 'string' : case 'doc.type.string' : call(function (suri, source, key, ref, pushResult) -- change to `string: stringlib` ? local stringlib = vm.getGlobal('type', 'stringlib') if stringlib then vm.getClassFields(suri, stringlib, key, ref, pushResult) end end) : case 'doc.type.array' : call(function (suri, source, key, ref, pushResult) if type(key) == 'number' then if key < 1 or not math.tointeger(key) then return end pushResult(source.node) end if type(key) == 'table' then if vm.isSubType(suri, key, 'integer') then pushResult(source.node) end end end) : case 'doc.type.table' : call(function (suri, source, key, ref, pushResult) if type(key) == 'string' and key:find(vm.ID_SPLITE) then return end for _, field in ipairs(source.fields) do local fieldKey = field.name if fieldKey.type == 'doc.type' then local fieldNode = vm.compileNode(fieldKey) for fn in fieldNode:eachObject() do if fn.type == 'global' and fn.cate == 'type' then if key == nil or fn.name == 'any' or (fn.name == 'boolean' and type(key) == 'boolean') or (fn.name == 'number' and type(key) == 'number') or (fn.name == 'integer' and math.tointeger(key)) or (fn.name == 'string' and type(key) == 'string') then pushResult(field) end elseif fn.type == 'doc.type.string' or fn.type == 'doc.type.integer' or fn.type == 'doc.type.boolean' then if key == nil or fn[1] == key then pushResult(field) end end end end if fieldKey.type == 'doc.field.name' then if key == nil or fieldKey[1] == key then pushResult(field) end end end end) : case 'global' : call(function (suri, node, key, ref, pushResult) if node.cate == 'variable' then if key then if type(key) ~= 'string' then return end local global = vm.getGlobal('variable', node.name, key) if global then for _, set in ipairs(global:getSets(suri)) do pushResult(set) end if ref then for _, get in ipairs(global:getGets(suri)) do pushResult(get) end end end else local globals = vm.getGlobalFields('variable', node.name) for _, global in ipairs(globals) do for _, set in ipairs(global:getSets(suri)) do pushResult(set) end if ref then for _, get in ipairs(global:getGets(suri)) do pushResult(get) end end end end end if node.cate == 'type' then vm.getClassFields(suri, node, key, ref, pushResult) end end) : default(function (suri, source, key, ref, pushResult) searchFieldByLocalID(source, key, ref, pushResult) searchFieldByGlobalID(suri, source, key, ref, pushResult) end) ---@param suri uri ---@param object vm.global ---@param key? string|number|integer|boolean|vm.global ---@param ref boolean ---@param pushResult fun(field: vm.object, isMark?: boolean) function vm.getClassFields(suri, object, key, ref, pushResult) local mark = {} local function searchClass(class, searchedFields) local name = class.name if mark[name] then return end mark[name] = true searchedFields = searchedFields or {} local hasFounded = {} local function copyToSearched() for fieldKey in pairs(hasFounded) do searchedFields[fieldKey] = true hasFounded[fieldKey] = nil end end local sets = class:getSets(suri) for _, set in ipairs(sets) do if set.type == 'doc.class' then -- check ---@field for _, field in ipairs(set.fields) do local fieldKey = guide.getKeyName(field) if fieldKey then -- ---@field x boolean -> class.x if key == nil or fieldKey == key then if not searchedFields[fieldKey] then pushResult(field, true) hasFounded[fieldKey] = true end end goto CONTINUE end if key == nil then pushResult(field, true) goto CONTINUE end if hasFounded[key] then goto CONTINUE end local keyType = type(key) if keyType == 'table' then -- ---@field [integer] boolean -> class[integer] local fieldNode = vm.compileNode(field.field) if vm.isSubType(suri, key.name, fieldNode) then local nkey = '|' .. key.name if not searchedFields[nkey] then pushResult(field, true) hasFounded[nkey] = true end end else local keyObject if keyType == 'number' then if math.tointeger(key) then keyObject = { type = 'integer', [1] = key } else keyObject = { type = 'number', [1] = key } end elseif keyType == 'boolean' or keyType == 'string' then keyObject = { type = keyType, [1] = key } end if keyObject and field.field.type ~= 'doc.field.name' then -- ---@field [integer] boolean -> class[1] local fieldNode = vm.compileNode(field.field) if vm.isSubType(suri, keyObject, fieldNode) then local nkey = '|' .. keyType if not searchedFields[nkey] then pushResult(field, true) hasFounded[nkey] = true end end end end ::CONTINUE:: end end end copyToSearched() for _, set in ipairs(sets) do if set.type == 'doc.class' then -- check local field and global field if not searchedFields[key] and set.bindSource then local src = set.bindSource if src.value and src.value.type == 'table' then searchFieldSwitch('table', suri, src.value, key, ref, function (field) local fieldKey = guide.getKeyName(field) if fieldKey then if not searchedFields[fieldKey] and guide.isSet(field) then hasFounded[fieldKey] = true pushResult(field, true) end end end) end if src.value and src.value.type == 'select' and src.value.vararg.type == 'call' then local func = src.value.vararg.node local args = src.value.vararg.args if func.special == 'setmetatable' and args and args[1] and args[1].type == 'table' then searchFieldSwitch('table', suri, args[1], key, ref, function (field) local fieldKey = guide.getKeyName(field) if fieldKey then if not searchedFields[fieldKey] and guide.isSet(field) then hasFounded[fieldKey] = true pushResult(field, true) end end end) end end end end end copyToSearched() for _, set in ipairs(sets) do if set.type == 'doc.class' then if not searchedFields[key] and set.bindSource then local src = set.bindSource searchFieldSwitch(src.type, suri, src, key, ref, function (field) local fieldKey = guide.getKeyName(field) if fieldKey and not searchedFields[fieldKey] then if not searchedFields[fieldKey] and guide.isSet(field) and field.value then if vm.getLocalID(field) and vm.getLocalID(field) == vm.getLocalID(field.value) then elseif vm.getGlobalNode(src) and vm.getGlobalNode(src) == vm.getGlobalNode(field.value) then else hasFounded[fieldKey] = true end pushResult(field, true) end end end) end end end copyToSearched() for _, set in ipairs(sets) do if set.type == 'doc.class' then -- look into extends(if field not found) if not searchedFields[key] and set.extends then for _, extend in ipairs(set.extends) do if extend.type == 'doc.extends.name' then local extendType = vm.getGlobal('type', extend[1]) if extendType then searchClass(extendType, searchedFields) end end end end end end copyToSearched() end local function searchGlobal(class) if class.cate == 'type' and class.name == '_G' then if key == nil then local sets = vm.getGlobalSets(suri, 'variable') for _, set in ipairs(sets) do pushResult(set) end elseif type(key) == 'string' then local global = vm.getGlobal('variable', key) if global then for _, set in ipairs(global:getSets(suri)) do pushResult(set) end end end end end searchClass(object) searchGlobal(object) end ---@param func parser.object ---@param index integer ---@return (parser.object|vm.generic)? function vm.getReturnOfFunction(func, index) if func.type == 'function' then if not func._returns then func._returns = {} end if not func._returns[index] then func._returns[index] = { type = 'function.return', parent = func, returnIndex = index, } end return func._returns[index] end if func.type == 'doc.type.function' then local rtn = func.returns[index] if not rtn then local lastReturn = func.returns[#func.returns] if lastReturn and lastReturn.name and lastReturn.name[1] == '...' then rtn = lastReturn else return nil end end local sign = vm.getSign(func) if not sign then return rtn end return vm.createGeneric(rtn, sign) end return nil end ---@param args parser.object[] ---@return vm.node local function getReturnOfSetMetaTable(args) local tbl = args[1] local mt = args[2] local node = vm.createNode() if tbl then node:merge(vm.compileNode(tbl)) end if mt then vm.compileByParentNode(mt, '__index', false, function (src) for n in vm.compileNode(src):eachObject() do if n.type == 'global' or n.type == 'local' or n.type == 'table' or n.type == 'doc.type.table' then node:merge(n) end end end) end return node end ---@param source parser.object local function matchCall(source) local call = source.parent if not call or call.type ~= 'call' or call.node ~= source then return end local funcs = vm.getMatchedFunctions(source, call.args) local myNode = vm.getNode(source) if not myNode then return end local needRemove for n in myNode:eachObject() do if n.type == 'function' or n.type == 'doc.type.function' then if not util.arrayHas(funcs, n) then if not needRemove then needRemove = vm.createNode() end needRemove:merge(n) end end end if needRemove then local newNode = myNode:copy() newNode:removeNode(needRemove) newNode:setData('originNode', myNode) vm.setNode(source, newNode, true) end end ---@param func parser.object ---@param index integer ---@param args parser.object[] ---@return vm.node local function getReturn(func, index, args) if not func._callReturns then func._callReturns = {} end if not func._callReturns[index] then local call = func.parent func._callReturns[index] = { type = 'call.return', parent = call, func = func, cindex = index, args = args, start = call.start, finish = call.finish, } end return vm.compileNode(func._callReturns[index]) end ---@param source parser.object ---@return boolean local function bindAs(source) local root = guide.getRoot(source) local docs = root.docs if not docs then return false end local ases = docs._asCache if not ases then ases = {} docs._asCache = ases for _, doc in ipairs(docs) do if doc.type == 'doc.as' and doc.as and doc.touch then ases[#ases+1] = doc end end table.sort(ases, function (a, b) return a.touch < b.touch end) end if #ases == 0 then return false end local max = #ases local index local left = 1 local right = max for _ = 1, 1000 do if left == right then index = left break end index = left + (right - left) // 2 local doc = ases[index] if doc.touch < source.finish then left = index + 1 else right = index end end local doc = ases[index] if doc and doc.touch == source.finish then local asNode = vm.compileNode(doc.as) asNode.resolved = true vm.setNode(source, asNode, true) return true end return false end ---@param source parser.object ---@param key? string|vm.global ---@param pushResult fun(source: parser.object) function vm.compileByParentNode(source, key, ref, pushResult) local parentNode = vm.compileNode(source) local docedResults = {} local commonResults = {} local mark = {} local suri = guide.getUri(source) local hasClass for node in parentNode:eachObject() do if node.type == 'global' and node.cate == 'type' ---@cast node vm.global and not guide.isBasicType(node.name) then hasClass = true break end end for node in parentNode:eachObject() do if not hasClass or ( node.type == 'global' and node.cate == 'type' ---@cast node vm.global and not guide.isBasicType(node.name) ) or guide.isLiteral(node) then searchFieldSwitch(node.type, suri, node, key, ref, function (res, markDoc) if mark[res] then return end mark[res] = true if markDoc then docedResults[#docedResults+1] = res else commonResults[#commonResults+1] = res end end) end end if not next(mark) then searchFieldByLocalID(source, key, ref, function (res, markDoc) if mark[res] then return end mark[res] = true if markDoc then docedResults[#docedResults+1] = res else commonResults[#commonResults+1] = res end end) end if #docedResults > 0 then for _, res in ipairs(docedResults) do pushResult(res) end end if #docedResults == 0 or key == nil then for _, res in ipairs(commonResults) do pushResult(res) end end end ---@param list parser.object[] ---@param index integer ---@return vm.node ---@return parser.object? function vm.selectNode(list, index) local exp if list[index] then exp = list[index] index = 1 else for i = index, 1, -1 do if list[i] then local last = list[i] if last.type == 'call' or last.type == 'varargs' then index = index - i + 1 exp = last end break end end end if not exp then return vm.createNode(vm.declareGlobal('type', 'nil')), nil end ---@type vm.node? local result if exp.type == 'call' then result = getReturn(exp.node, index, exp.args) if result:isEmpty() then result:merge(vm.declareGlobal('type', 'unknown')) end else ---@type vm.node result = vm.compileNode(exp) if result:isEmpty() then result:merge(vm.declareGlobal('type', 'unknown')) end end return result, exp end ---@param source parser.object ---@param list parser.object[] ---@param index integer ---@return vm.node local function selectNode(source, list, index) local result = vm.selectNode(list, index) if source.type == 'function.return' then -- remove any for returns local rtnNode = vm.createNode() for n in result:eachObject() do if guide.isLiteral(n) then rtnNode:merge(n) end if n.type == 'global' and n.cate == 'type' then if n.name ~= 'any' then rtnNode:merge(n) end else rtnNode:merge(n) end end vm.setNode(source, rtnNode) return rtnNode end vm.setNode(source, result) return result end ---@param source parser.object ---@param node vm.node.object ---@return boolean local function isValidCallArgNode(source, node) if source.type == 'function' then return node.type == 'doc.type.function' end if source.type == 'table' then return node.type == 'doc.type.table' or node.type == 'doc.type.array' or ( node.type == 'global' and node.cate == 'type' ---@cast node vm.global and not guide.isBasicType(node.name) ) end if source.type == 'dummyarg' then return true end return false end ---@param func parser.object ---@param index integer ---@return parser.object? local function getFuncArg(func, index) local args = func.args if not args then return nil end if args[index] then return args[index] end local lastArg = args[#args] if lastArg and lastArg.type == '...' then return lastArg end return nil end ---@param arg parser.object ---@param call parser.object ---@param callNode vm.node ---@param fixIndex integer ---@param myIndex integer local function compileCallArgNode(arg, call, callNode, fixIndex, myIndex) ---@type integer?, table? local eventIndex, eventMap if call.args then for i = 1, 2 do local eventArg = call.args[i + fixIndex] if not eventArg then break end eventMap = vm.getLiterals(eventArg) if eventMap then eventIndex = i break end end end for n in callNode:eachObject() do if n.type == 'function' then ---@cast n parser.object local sign = vm.getSign(n) local farg = getFuncArg(n, myIndex) if farg then for fn in vm.compileNode(farg):eachObject() do if isValidCallArgNode(arg, fn) then if fn.type == 'doc.type.function' then ---@cast fn parser.object if sign then local generic = vm.createGeneric(fn, sign) local args = {} for i = fixIndex + 1, myIndex - 1 do args[#args+1] = call.args[i] end fn = generic:resolve(guide.getUri(call), args) end end vm.setNode(arg, fn) end end end end if n.type == 'doc.type.function' then ---@cast n parser.object local myEvent if n.args[eventIndex] then local argNode = vm.compileNode(n.args[eventIndex]) myEvent = argNode:get(1) end if not myEvent or not eventMap or myIndex <= eventIndex or myEvent.type ~= 'doc.type.string' or eventMap[myEvent[1]] then local farg = getFuncArg(n, myIndex) if farg then for fn in vm.compileNode(farg):eachObject() do if isValidCallArgNode(arg, fn) then vm.setNode(arg, fn) end end end end end end end ---@param arg parser.object ---@param call parser.object ---@param index? integer ---@return vm.node? function vm.compileCallArg(arg, call, index) if not index then for i, carg in ipairs(call.args) do if carg == arg then index = i break end end if not index then return nil end end local callNode = vm.compileNode(call.node) compileCallArgNode(arg, call, callNode, 0, index) if call.node.special == 'pcall' or call.node.special == 'xpcall' then local fixIndex = call.node.special == 'pcall' and 1 or 2 if call.args and call.args[1] then callNode = vm.compileNode(call.args[1]) compileCallArgNode(arg, call, callNode, fixIndex, index - fixIndex) end end return vm.getNode(arg) end ---@class parser.object ---@field package _iterator? table ---@field package _iterArgs? table ---@field package _iterVars? table ---@param source parser.object ---@param target parser.object local function compileForVars(source, target) if not source.exps then return end -- for k, v in pairs(t) do --> for k, v in iterator, status, initValue do --> local k, v = iterator(status, initValue) if not source._iterator then source._iterator = { type = 'dummyfunc', parent = source, } source._iterArgs = {{},{}} source._iterVars = {} end -- iterator if not vm.getNode(source._iterator) then selectNode(source._iterator, source.exps, 1) end -- status if not vm.getNode(source._iterArgs[1]) then selectNode(source._iterArgs[1], source.exps, 2) end -- initValue if not vm.getNode(source._iterArgs[2]) then selectNode(source._iterArgs[2], source.exps, 3) end if source.keys then for i, loc in ipairs(source.keys) do if loc == target then local node = getReturn(source._iterator, i, source._iterArgs) node:removeOptional() vm.setNode(loc, node) end end end end ---@param source parser.object local function compileLocal(source) local myNode = vm.setNode(source, source) local hasMarkDoc if source.bindDocs then hasMarkDoc = bindDocs(source) end local hasMarkParam if not hasMarkDoc then local selfNode = guide.getSelfNode(source) if selfNode then hasMarkParam = true vm.setNode(source, vm.compileNode(selfNode)) myNode:remove 'function' end end local hasMarkValue if (not hasMarkDoc and source.value) or (source.value and source.value.type == 'table') then hasMarkValue = true if source.value.type == 'table' then vm.setNode(source, source.value) elseif source.value.type ~= 'nil' then vm.setNode(source, vm.compileNode(source.value)) end end if not hasMarkValue and not hasMarkValue then if source.ref then for _, ref in ipairs(source.ref) do if ref.type == 'setlocal' and ref.value and ref.value.type == 'function' then vm.setNode(source, vm.compileNode(ref.value)) end end end end -- function x.y(self, ...) --> function x:y(...) if source[1] == 'self' and not hasMarkDoc and source.parent.type == 'funcargs' and source.parent[1] == source then local setfield = source.parent.parent.parent if setfield.type == 'setfield' then hasMarkParam = true vm.setNode(source, vm.compileNode(setfield.node)) end end if source.parent.type == 'funcargs' and not hasMarkDoc and not hasMarkParam then local func = source.parent.parent -- local call ---@type fun(f: fun(x: number));call(function (x) end) --> x -> number local funcNode = vm.compileNode(func) local hasDocArg for n in funcNode:eachObject() do if n.type == 'doc.type.function' then for index, arg in ipairs(n.args) do if func.args[index] == source then local argNode = vm.compileNode(arg) for an in argNode:eachObject() do if an.type ~= 'doc.generic.name' then vm.setNode(source, an) end end hasDocArg = true end end end end if not hasDocArg then vm.setNode(source, vm.declareGlobal('type', 'any')) end end -- for x in ... do if source.parent.type == 'in' then compileForVars(source.parent, source) end -- for x = ... do if source.parent.type == 'loop' then if source.parent.loc == source then if bindDocs(source) then return end vm.setNode(source, vm.declareGlobal('type', 'integer')) end end myNode:setData('hasDefined', hasMarkDoc or hasMarkParam or hasMarkValue) end ---@param source parser.object ---@param mfunc parser.object ---@param index integer ---@param args parser.object[] local function bindReturnOfFunction(source, mfunc, index, args) local returnObject = vm.getReturnOfFunction(mfunc, index) if not returnObject then return end local returnNode = vm.compileNode(returnObject) for rnode in returnNode:eachObject() do if rnode.type == 'generic' then returnNode = rnode:resolve(guide.getUri(source), args) break end end if returnNode then for rnode in returnNode:eachObject() do -- TODO: narrow type if rnode.type ~= 'doc.generic.name' then vm.setNode(source, rnode) end end if returnNode:isOptional() then vm.getNode(source):addOptional() end end end local compilerSwitch = util.switch() : case 'nil' : case 'boolean' : case 'integer' : case 'number' : case 'string' : case 'doc.type.function' : case 'doc.type.table' : case 'doc.type.array' : call(function (source) vm.setNode(source, source) end) : case 'table' : call(function (source) vm.setNode(source, source) if source.parent.type == 'callargs' then local call = source.parent.parent vm.compileCallArg(source, call) end if source.parent.type == 'setglobal' or source.parent.type == 'local' or source.parent.type == 'setlocal' or source.parent.type == 'tablefield' or source.parent.type == 'tableindex' or source.parent.type == 'tableexp' or source.parent.type == 'setfield' or source.parent.type == 'setindex' then local parentNode = vm.compileNode(source.parent) for _, pn in ipairs(parentNode) do if pn.type == 'global' and pn.cate == 'type' then ---@cast pn vm.global if not guide.isBasicType(pn.name) then vm.setNode(source, pn) end elseif pn.type == 'doc.type.table' or pn.type == 'doc.type.array' then vm.setNode(source, pn) end end end end) : case 'function' : call(function (source) vm.setNode(source, source) local parent = source.parent if source.bindDocs then for _, doc in ipairs(source.bindDocs) do if doc.type == 'doc.overload' then vm.setNode(source, vm.compileNode(doc)) end end end -- table.sort(string[], function () end) if parent.type == 'callargs' then local call = parent.parent vm.compileCallArg(source, call) end -- function f() return function () end end if parent.type == 'return' then for i, ret in ipairs(parent) do if ret == source then local func = guide.getParentFunction(parent) if func then local returnObj = vm.getReturnOfFunction(func, i) if returnObj then vm.setNode(source, vm.compileNode(returnObj)) end end break end end end -- { f = function () end } if guide.isSet(parent) then vm.setNode(source, vm.compileNode(parent)) end end) : case 'paren' : call(function (source) if bindAs(source) then return end if source.exp then vm.setNode(source, vm.compileNode(source.exp)) end end) : case 'local' : case 'self' ---@async ---@param source parser.object : call(function (source) vm.launchRunner(source, function () local myNode = vm.getNode(source) ---@cast myNode -? myNode:setData('resolving', true) if source.ref then for _, ref in ipairs(source.ref) do if ref.type == 'getlocal' or ref.type == 'setlocal' then vm.setNode(ref, myNode, true) end end end compileLocal(source) myNode.resolved = true end, function () local myNode = vm.getNode(source) ---@cast myNode -? myNode:setData('resolving', nil) local hasMark = vm.getNode(source):getData 'hasDefined' if source.ref and not hasMark then local parentFunc = guide.getParentFunction(source) for _, ref in ipairs(source.ref) do if ref.type == 'setlocal' and guide.getParentFunction(ref) == parentFunc then local refNode = vm.getNode(ref) if refNode then vm.setNode(source, refNode) end end end end end, function (src, node) if src.type == 'setlocal' then if src.bindDocs then for _, doc in ipairs(src.bindDocs) do if doc.type == 'doc.type' then vm.setNode(src, vm.compileNode(doc), true) return vm.getNode(src) end end end if src.value then if src.value.type == 'table' then vm.setNode(src, vm.createNode(src.value), true) vm.setNode(src, node:copy():asTable()) else vm.setNode(src, vm.compileNode(src.value), true) end else vm.setNode(src, node, true) end return vm.getNode(src) elseif src.type == 'getlocal' then if bindAs(src) then return end vm.setNode(src, node, true) node.resolved = true matchCall(src) end end) vm.waitResolveRunner(source) end) : case 'setlocal' : call(function (source) vm.compileNode(source.node) end) : case 'getlocal' ---@async : call(function (source) if bindAs(source) then return end vm.compileNode(source.node) vm.waitResolveRunner(source) end) : case 'setfield' : case 'setmethod' : case 'setindex' : case 'getfield' : case 'getmethod' : case 'getindex' : call(function (source) if guide.isGet(source) and bindAs(source) then return end ---@type (string|vm.node)? local key = guide.getKeyName(source) if key == nil and source.index then key = vm.compileNode(source.index) end if key == nil then return end if type(key) == 'table' then ---@cast key vm.node local uri = guide.getUri(source) local value = vm.getTableValue(uri, vm.compileNode(source.node), key) if value then vm.setNode(source, value) end for k in key:eachObject() do if k.type == 'global' and k.cate == 'type' then ---@cast k vm.global vm.compileByParentNode(source.node, k, false, function (src) vm.setNode(source, vm.compileNode(src)) if src.value then vm.setNode(source, vm.compileNode(src.value)) end end) end end else ---@cast key string vm.compileByParentNode(source.node, key, false, function (src) if src.value then if bindDocs(src) then vm.setNode(source, vm.compileNode(src)) elseif src.value.type ~= 'nil' then vm.setNode(source, vm.compileNode(src.value)) local node = vm.getNode(src) if node then vm.setNode(source, node) end end else vm.setNode(source, vm.compileNode(src)) end end) end end) : case 'setglobal' : call(function (source) if source.node[1] ~= '_ENV' then return end local key = guide.getKeyName(source) vm.compileByParentNode(source.node, key, false, function (src) if src.type == 'doc.type.field' or src.type == 'doc.field' then vm.setNode(source, vm.compileNode(src)) end end) end) : case 'getglobal' : call(function (source) if bindAs(source) then return end if source.node[1] ~= '_ENV' then return end local key = guide.getKeyName(source) vm.compileByParentNode(source.node, key, false, function (src) vm.setNode(source, vm.compileNode(src)) end) end) : case 'tablefield' : case 'tableindex' : call(function (source) local hasMarkDoc if source.bindDocs then hasMarkDoc = bindDocs(source) end if not hasMarkDoc then vm.compileByParentNode(source.node, guide.getKeyName(source), false, function (src) if src.type == 'doc.field' or src.type == 'doc.type.field' or src.type == 'doc.type.name' then hasMarkDoc = true vm.setNode(source, vm.compileNode(src)) end end) end if not hasMarkDoc and source.value then vm.setNode(source, vm.compileNode(source.value)) end end) : case 'field' : case 'method' : call(function (source) vm.setNode(source, vm.compileNode(source.parent)) end) : case 'tableexp' : call(function (source) local hasMarkDoc vm.compileByParentNode(source.parent, source.tindex, false, function (src) if src.type == 'doc.field' or src.type == 'doc.type.field' or src.type == 'doc.type.name' or guide.isLiteral(src) then hasMarkDoc = true vm.setNode(source, vm.compileNode(src)) end end) if not hasMarkDoc then vm.setNode(source, vm.compileNode(source.value)) end end) : case 'function.return' ---@param source parser.object : call(function (source) local func = source.parent local index = source.returnIndex local hasMarkDoc if func.bindDocs then local sign = vm.getSign(func) local lastReturn for _, doc in ipairs(func.bindDocs) do if doc.type == 'doc.return' then for _, rtn in ipairs(doc.returns) do lastReturn = rtn if rtn.returnIndex == index then hasMarkDoc = true local hasGeneric if sign then guide.eachSourceType(rtn, 'doc.generic.name', function (src) hasGeneric = true end) end if hasGeneric then ---@cast sign -? vm.setNode(source, vm.createGeneric(rtn, sign)) else vm.setNode(source, vm.compileNode(rtn)) end end end end end if lastReturn and not hasMarkDoc then if lastReturn.name and lastReturn.name[1] == '...' then hasMarkDoc = true vm.setNode(source, vm.compileNode(lastReturn)) end end end local hasReturn if func.returns and not hasMarkDoc then for _, rtn in ipairs(func.returns) do if selectNode(source, rtn, index) then hasReturn = true end end if hasReturn then local hasKnownType local hasUnknownType for n in vm.getNode(source):eachObject() do if guide.isLiteral(n) then if n.type ~= 'nil' then hasKnownType = true break end goto CONTINUE end if n.type == 'global' and n.cate == 'type' then if n.name ~= 'nil' then hasKnownType = true break end goto CONTINUE end hasUnknownType = true ::CONTINUE:: end if not hasKnownType and hasUnknownType then vm.setNode(source, vm.declareGlobal('type', 'unknown')) end end end if not hasMarkDoc and not hasReturn then vm.setNode(source, vm.declareGlobal('type', 'nil')) end end) : case 'call.return' ---@param source parser.object : call(function (source) if bindAs(source) then return end local func = source.func local args = source.args local index = source.cindex if func.special == 'setmetatable' then if not args then return end vm.setNode(source, getReturnOfSetMetaTable(args)) return end if func.special == 'pcall' and index > 1 then if not args then return end local newArgs = {} for i = 2, #args do newArgs[#newArgs+1] = args[i] end local node = getReturn(args[1], index - 1, newArgs) if node then vm.setNode(source, node) end return end if func.special == 'xpcall' and index > 1 then if not args then return end local newArgs = {} for i = 3, #args do newArgs[#newArgs+1] = args[i] end local node = getReturn(args[1], index - 1, newArgs) if node then vm.setNode(source, node) end return end if func.special == 'require' then if not args then return end local nameArg = args[1] if not nameArg or nameArg.type ~= 'string' then return end local name = nameArg[1] if not name or type(name) ~= 'string' then return end local uri = rpath.findUrisByRequireName(guide.getUri(func), name)[1] if not uri then return end local state = files.getState(uri) local ast = state and state.ast if not ast then return end vm.setNode(source, vm.compileNode(ast)) return end local funcNode = vm.compileNode(func) ---@type vm.node? for nd in funcNode:eachObject() do if nd.type == 'function' or nd.type == 'doc.type.function' then ---@cast nd parser.object bindReturnOfFunction(source, nd, index, args) elseif nd.type == 'global' and nd.cate == 'type' then ---@cast nd vm.global for _, set in ipairs(nd:getSets(guide.getUri(source))) do if set.type == 'doc.class' then for _, overload in ipairs(set.calls) do bindReturnOfFunction(source, overload.overload, index, args) end end end end end end) : case 'main' : call(function (source) if source.returns then for _, rtn in ipairs(source.returns) do if rtn[1] then vm.setNode(source, vm.compileNode(rtn[1])) end end end end) : case 'select' : call(function (source) local vararg = source.vararg if vararg.type == 'call' then local node = getReturn(vararg.node, source.sindex, vararg.args) if not node then return end if node:isEmpty() then node = vm.runOperator('call', vararg.node) or node end vm.setNode(source, node) end if vararg.type == 'varargs' then vm.setNode(source, vm.compileNode(vararg)) end end) : case 'varargs' : call(function (source) if source.node then vm.setNode(source, vm.compileNode(source.node)) end end) : case 'call' : call(function (source) local node = getReturn(source.node, 1, source.args) if not node then return end if node:isEmpty() then node = vm.runOperator('call', source.node) or node end vm.setNode(source, node) end) : case 'doc.type' : call(function (source) for _, typeUnit in ipairs(source.types) do vm.setNode(source, vm.compileNode(typeUnit)) end if source.optional then vm.getNode(source):addOptional() end end) : case 'doc.type.integer' : case 'doc.type.string' : case 'doc.type.boolean' : case 'doc.type.code' : call(function (source) vm.setNode(source, source) end) : case 'doc.type.name' : call(function (source) if source[1] == 'self' then local state = guide.getDocState(source) if state.type == 'doc.return' or state.type == 'doc.param' then local func = state.bindSource if func.type == 'function' then local node = guide.getFunctionSelfNode(func) if node then vm.setNode(source, vm.compileNode(node)) return end end elseif state.type == 'doc.field' or state.type == 'doc.overload' then local class = state.class if class then vm.setNode(source, vm.compileNode(class)) return end end end end) : case 'doc.generic.name' : call(function (source) vm.setNode(source, source) end) : case 'doc.type.sign' : call(function (source) local uri = guide.getUri(source) vm.setNode(source, source) if not source.node[1] then return end local global = vm.getGlobal('type', source.node[1]) if not global then return end for _, set in ipairs(global:getSets(uri)) do if set.type == 'doc.class' then if set.extends then for _, ext in ipairs(set.extends) do if ext.type == 'doc.type.table' then if vm.getGeneric(ext) then local resolved = vm.getGeneric(ext):resolve(uri, source.signs) vm.setNode(source, resolved) end end end end end if set.type == 'doc.alias' then if vm.getGeneric(set.extends) then local resolved = vm.getGeneric(set.extends):resolve(uri, source.signs) vm.setNode(source, resolved) end end end end) : case 'doc.class.name' : call(function (source) vm.setNode(source, vm.compileNode(source.parent)) end) : case 'doc.enum.name' : call(function (source) vm.setNode(source, vm.compileNode(source.parent)) end) : case 'doc.field' : call(function (source) if not source.extends then return end local fieldNode = vm.compileNode(source.extends) if source.optional then fieldNode:addOptional() end vm.setNode(source, fieldNode) end) : case 'doc.field.name' : call(function (source) vm.setNode(source, source) end) : case 'doc.type.field' : call(function (source) if not source.extends then return end local fieldNode = vm.compileNode(source.extends) if source.optional then fieldNode:addOptional() end vm.setNode(source, fieldNode) end) : case 'doc.param' : call(function (source) if not source.extends then return end vm.setNode(source, vm.compileNode(source.extends)) end) : case 'doc.vararg' : call(function (source) if not source.vararg then return end vm.setNode(source, vm.compileNode(source.vararg)) end) : case '...' : call(function (source) if not source.bindDocs then return end for _, doc in ipairs(source.bindDocs) do if doc.type == 'doc.vararg' then vm.setNode(source, vm.compileNode(doc)) end if doc.type == 'doc.param' then vm.setNode(source, vm.compileNode(doc)) end end end) : case 'doc.overload' : call(function (source) vm.setNode(source, vm.compileNode(source.overload)) end) : case 'doc.type.arg' : call(function (source) if source.extends then vm.setNode(source, vm.compileNode(source.extends)) else vm.setNode(source, vm.declareGlobal('type', 'any')) end if source.optional then vm.getNode(source):addOptional() end end) : case 'unary' : call(function (source) if bindAs(source) then return end if not source[1] then return end vm.unarySwich(source.op.type, source) end) : case 'binary' : call(function (source) if bindAs(source) then return end if not source[1] or not source[2] then return end vm.binarySwitch(source.op.type, source) end) ---@param source parser.object local function compileByNode(source) compilerSwitch(source.type, source) end ---@param source parser.object local function compileByGlobal(source) local global = vm.getGlobalNode(source) if not global then return end ---@cast source parser.object local root = guide.getRoot(source) local uri = guide.getUri(source) if not root._globalBase then root._globalBase = {} end local name = global:asKeyName() if not root._globalBase[name] then root._globalBase[name] = { type = 'globalbase', parent = root, } end local globalNode = vm.getNode(root._globalBase[name]) if globalNode then vm.setNode(source, globalNode, true) return end ---@type vm.node globalNode = vm.createNode(global) vm.setNode(root._globalBase[name], globalNode, true) vm.setNode(source, globalNode, true) -- TODO:don't mix --local sets = global.links[uri].sets or {} --local gets = global.links[uri].gets or {} --for _, set in ipairs(sets) do -- vm.setNode(set, globalNode, true) --end --for _, get in ipairs(gets) do -- vm.setNode(get, globalNode, true) --end if global.cate == 'variable' then local hasMarkDoc for _, set in ipairs(global:getSets(uri)) do if set.bindDocs and set.parent.type == 'main' then if bindDocs(set) then globalNode:merge(vm.compileNode(set)) hasMarkDoc = true end if vm.getNode(set) then globalNode:merge(vm.compileNode(set)) end end end -- Set all globals node first to avoid recursive for _, set in ipairs(global:getSets(uri)) do vm.setNode(set, globalNode, true) end for _, set in ipairs(global:getSets(uri)) do if set.value and set.value.type ~= 'nil' and set.parent.type == 'main' then if not hasMarkDoc or guide.isLiteral(set.value) then globalNode:merge(vm.compileNode(set.value)) end end end for _, set in ipairs(global:getSets(uri)) do vm.setNode(set, globalNode, true) end end if global.cate == 'type' then for _, set in ipairs(global:getSets(uri)) do if set.type == 'doc.class' then if set.extends then for _, ext in ipairs(set.extends) do if ext.type == 'doc.type.table' then if not vm.getGeneric(ext) then globalNode:merge(vm.compileNode(ext)) end end end end end if set.type == 'doc.alias' then if not vm.getGeneric(set.extends) then globalNode:merge(vm.compileNode(set.extends)) end end end end end local nodeSwitch;nodeSwitch = util.switch() : case 'field' : case 'method' : call(function (source, lastKey, pushResult) return nodeSwitch(source.parent.type, source.parent, lastKey, pushResult) end) : case 'getfield' : case 'setfield' : case 'getmethod' : case 'setmethod' : case 'getindex' : case 'setindex' : call(function (source, lastKey, pushResult) local parentNode = vm.compileNode(source.node) local uri = guide.getUri(source) local key = guide.getKeyName(source) if type(key) ~= 'string' then return end if lastKey then key = key .. vm.ID_SPLITE .. lastKey end for pn in parentNode:eachObject() do searchFieldSwitch(pn.type, uri, pn, key, false, pushResult) end return key, source.node end) : case 'tableindex' : case 'tablefield' : call(function (source, lastKey, pushResult) if lastKey then return end local key = guide.getKeyName(source) if type(key) ~= 'string' then return end local uri = guide.getUri(source) local parentNode = vm.compileNode(source.node) for pn in parentNode:eachObject() do searchFieldSwitch(pn.type, uri, pn, key, false, pushResult) end end) function vm.compileByNodeChain(source, pushResult) local lastKey local src = source while true do local key, node = nodeSwitch(src.type, src, lastKey, pushResult) if not key then break end src = node lastKey = key end end ---@param source vm.object local function compileByParentNode(source) if not vm.getNode(source):isEmpty() then return end vm.compileByNodeChain(source, function (result) vm.setNode(source, vm.compileNode(result)) end) end ---@param source vm.object ---@return vm.node function vm.compileNode(source) if not source then if TEST then error('Can not compile nil source') else log.error('Can not compile nil source') end end if source.type == 'getlocal' then ---@cast source parser.object vm.storeWaitingRunner(source) ---@diagnostic disable-next-line: await-in-sync vm.waitResolveRunner(source) end local cache = vm.getNode(source) if cache ~= nil then return cache end if source.type == 'generic' then vm.setNode(source, source) local node = vm.getNode(source) ---@cast node -? return node end ---@cast source parser.object vm.setNode(source, vm.createNode(), true) compileByGlobal(source) compileByNode(source) compileByParentNode(source) matchCall(source) local node = vm.getNode(source) ---@cast node -? node.resolved = true return node end