From b04ad0cbe77602f739cb35d7ba838a1f62d298dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= <sumneko@hotmail.com>
Date: Tue, 5 Apr 2022 01:30:20 +0800
Subject: update

---
 script/vm/compiler.lua | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'script/vm/compiler.lua')

diff --git a/script/vm/compiler.lua b/script/vm/compiler.lua
index 6f542367..1473b985 100644
--- a/script/vm/compiler.lua
+++ b/script/vm/compiler.lua
@@ -7,6 +7,8 @@ local signMgr    = require 'vm.sign'
 local config     = require 'config'
 local union      = require 'vm.union'
 local genericMgr = require 'vm.generic'
+local rpath      = require 'workspace.require-path'
+local files      = require 'files'
 
 ---@class parser.object
 ---@field _compiledNodes  boolean
@@ -275,6 +277,26 @@ local function getReturn(func, index, args)
         end
         return getReturn(args[1], index - 1, newArgs)
     end
+    if func.special == 'require' then
+        local nameArg = args[1]
+        if not nameArg or nameArg.type ~= 'string' then
+            return nil
+        end
+        local name = nameArg[1]
+        if not name or type(name) ~= 'string' then
+            return nil
+        end
+        local uri = rpath.findUrisByRequirePath(guide.getUri(func), name)[1]
+        if not uri then
+            return nil
+        end
+        local state = files.getState(uri)
+        local ast   = state and state.ast
+        if not ast then
+            return nil
+        end
+        return m.compileNode(ast)
+    end
     local node = m.compileNode(func)
     ---@type vm.node.union
     local result, hasCalled
@@ -674,6 +696,14 @@ local compilerSwitch = util.switch()
             end
         end
     end)
+    : case 'main'
+    : call(function (source)
+        if source.returns then
+            for _, rtn in ipairs(source.returns) do
+                nodeMgr.setNode(source, m.compileNode(rtn[1]))
+            end
+        end
+    end)
     : case 'select'
     : call(function (source)
         local vararg = source.vararg
-- 
cgit v1.2.3