stash

Author: sumneko

script/core/completion.lua

@@ -1167,6 +1167,15 @@ local function getCallEnums(source, index)
                         kind        = define.CompletionItemKind.EnumMember,
                     }
                 end
+                for _, unit in ipairs(vm.getDocTypes(doc.extends)) do
+                    if unit.type == 'doc.type.function' then
+                        enums[#enums+1] = {
+                            label       = guide.getDocTypeUnitName(nil, unit),
+                            description = doc.comment,
+                            kind        = define.CompletionItemKind.Function,
+                        }
+                    end
+                end
                 return enums
             elseif doc.type == 'doc.vararg'
             and    arg.type == '...' then

script/parser/guide.lua

@@ -2941,17 +2941,17 @@ local function getDocAliasExtends(status, typeUnit)
     return nil
 end
 
-local function getDocTypeUnitName(status, unit)
+function m.getDocTypeUnitName(status, unit)
     local typeName
     if unit.type == 'doc.type.name' then
         typeName = unit[1]
     elseif unit.type == 'doc.type.function' then
         typeName = 'function'
     elseif unit.type == 'doc.type.array' then
-        typeName = getDocTypeUnitName(status, unit.node) .. '[]'
+        typeName = m.getDocTypeUnitName(status, unit.node) .. '[]'
     elseif unit.type == 'doc.type.generic' then
         typeName = ('%s<%s, %s>'):format(
-            getDocTypeUnitName(status, unit.node),
+            m.getDocTypeUnitName(status, unit.node),
             m.viewInferType(m.getDocTypeNames(status, unit.key)),
             m.viewInferType(m.getDocTypeNames(status, unit.value))
         )
@@ -2975,7 +2975,7 @@ function m.getDocTypeNames(status, doc)
                 results[#results+1] = res
             end
         else
-            local typeName = getDocTypeUnitName(status, unit)
+            local typeName = m.getDocTypeUnitName(status, unit)
             results[#results+1] = {
                 type   = typeName,
                 source = unit,

test/completion/init.lua

@@ -1952,3 +1952,18 @@ vvv$
         description = EXISTS,
     },
 }
+
+--TEST [[
+-----@param callback fun(x: number, y: number):string
+--local function f(callback)
+--
+--f($)
+--]]
+--{
+--    {
+--        label  = 'fun(x: number, y: number):string',
+--        detail = 'function',
+--        kind   = define.CompletionItemKind.Function,
+--        description = EXISTS,
+--    },
+--}