#477

LuaLS · Sep 28, 2021 · f1a974b · f1a974b
1 parent 05555be
commit f1a974b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 6 deletions.
diff --git a/script/core/noder.lua b/script/core/noder.lua
@@ -1176,6 +1176,8 @@ compileNodeMap = util.switch()
             end
         end
         if source.args then
+            local parent = source.parent
+            local parentID = guide.isSet(parent) and getID(parent)
             for i, arg in ipairs(source.args) do
                 if arg[1] == 'self' then
                     goto CONTINUE
@@ -1192,12 +1194,26 @@ compileNodeMap = util.switch()
                         , PARAM_NAME
                         , arg[1]
                     ))
+                    if parentID then
+                        pushForward(noders, getID(arg), sformat('%s%s%s'
+                            , parentID
+                            , PARAM_NAME
+                            , arg[1]
+                        ))
+                    end
                 else
                     pushForward(noders, getID(arg), sformat('%s%s%s'
                         , id
                         , PARAM_NAME
                         , '...'
                     ))
+                    if parentID then
+                        pushForward(noders, getID(arg), sformat('%s%s%s'
+                            , parentID
+                            , PARAM_NAME
+                            , '...'
+                        ))
+                    end
                 end
                 ::CONTINUE::
             end

diff --git a/test/type_inference/init.lua b/test/type_inference/init.lua
@@ -872,9 +872,9 @@ for _, a in ipairs(v) do
 end
 ]]
 
---TEST 'number' [[
------@param x number
---local f
---
---f = function (<?x?>) end
---]]
+TEST 'number' [[
+---@param x number
+local f
+
+f = function (<?x?>) end
+]]