diff options
| -rw-r--r-- | script/vm/compiler.lua | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/script/vm/compiler.lua b/script/vm/compiler.lua index 35fefd01..dba97932 100644 --- a/script/vm/compiler.lua +++ b/script/vm/compiler.lua @@ -178,6 +178,17 @@ local function selectNode(source, list, index) return m.compileNode(exp) end +local function bindDocs(source) + local isParam = source.parent.type == 'funcargs' + for _, doc in ipairs(source.bindDocs) do + if doc.type == 'doc.type' then + if not isParam then + m.setNode(source, m.compileNode(doc)) + end + end + end +end + local compilerMap = util.switch() : case 'boolean' : case 'table' @@ -205,6 +216,9 @@ local compilerMap = util.switch() if source.dummy then m.setNode(source, m.compileNode(source.method.node)) end + if source.bindDocs then + bindDocs(source) + end -- function x.y(self, ...) --> function x:y(...) if source[1] == 'self' and source.parent.type == 'funcargs' @@ -263,6 +277,20 @@ local compilerMap = util.switch() m.setNode(source, getReturn(vararg.node, source.sindex, source, vararg.args)) end end) + : case 'doc.type' + : call(function (source) + for _, typeUnit in ipairs(source.types) do + m.setNode(source, m.compileNode(typeUnit)) + end + end) + : case 'doc.type.name' + : call(function (source) + local name = source[1] + local uri = guide.getUri(source) + local type = globalMgr.declareGlobal('type', name, uri) + type:addGet(uri, source) + m.setNode(source, type) + end) : getMap() ---@param source parser.object |