diff options
Diffstat (limited to 'server/src/vm/vm.lua')
| -rw-r--r-- | server/src/vm/vm.lua | 46 |
1 files changed, 7 insertions, 39 deletions
diff --git a/server/src/vm/vm.lua b/server/src/vm/vm.lua index c519ad72..940e822f 100644 --- a/server/src/vm/vm.lua +++ b/server/src/vm/vm.lua @@ -153,31 +153,6 @@ function mt:callSetMetaTable(func, values, source) values[1]:setMetaTable(values[2]) end -function mt:getRequire(strValue, destVM) - -- 取出对方的主函数 - local main = destVM.main - -- 获取主函数返回值 - local mainValue = main:getFunction():getReturn(1) - if mainValue then - mainValue:markGlobal() - else - mainValue = self:createValue('any', self:getDefaultSource()) - mainValue.uri = destVM.uri - end - - return mainValue -end - -function mt:getLoadFile(strValue, destVM) - -- 取出对方的主函数 - local main = destVM.main - -- loadfile 的返回值就是对方的主函数 - local mainValue = main - mainValue:markGlobal() - - return mainValue -end - function mt:tryRequireOne(strValue, mode) if not self.lsp or not self.lsp.workspace then return nil @@ -203,18 +178,8 @@ function mt:tryRequireOne(strValue, mode) end strSource:set('target uri', uri) - -- 如果取不到VM（不编译），则做个标记，之后再取一次 - local destVM = self.lsp:getVM(uri) self.lsp:compileChain(self:getUri(), uri) - if destVM then - if mode == 'require' then - return self:getRequire(strValue, destVM) - elseif mode == 'loadfile' then - return self:getLoadFile(strValue, destVM) - elseif mode == 'dofile' then - return self:getRequire(strValue, destVM) - end - end + return self.lsp.chain:get(uri) end return nil end @@ -236,7 +201,7 @@ function mt:callRequire(func, values) else local requireValue = self:tryRequireOne(values[1], 'require') if not requireValue then - requireValue = self:createValue('boolean', self:getDefaultSource()) + requireValue = self:createValue('any', self:getDefaultSource()) requireValue:set('cross file', true) end func:setReturn(1, requireValue) @@ -330,7 +295,7 @@ function mt:call(value, values, source) self:runFunction(func) end else - func:setReturn(1, self:createValue('any', source)) + func:mergeReturn(1, self:createValue('any', source)) end end @@ -734,7 +699,7 @@ function mt:doReturn(action) local func = self:getCurrentFunction() values:eachValue(function (n, value) value.uri = self:getUri() - func:setReturn(n, value) + func:mergeReturn(n, value) local source = action[n] or value:getSource() if not source or source.start == 0 then source = self:getDefaultSource() @@ -1147,6 +1112,9 @@ function mt:createEnvironment(ast) -- 整个文件是一个函数 self.main = self:createFunction(ast) self:setCurrentFunction(self.main:getFunction()) + if self.lsp then + self.main:getFunction():mergeReturn(1, self.lsp.chain:get(self.uri)) + end -- 全局变量`_G` local global = buildGlobal(self.lsp) -- 隐藏的上值`_ENV` |