加个自动完成

2 files changed, 23 insertions, 0 deletions

diff --git a/server/src/core/completion.lua b/server/src/core/completion.lua
index eb4e9a1d..1ae88c8f 100644
--- a/server/src/core/completion.lua
+++ b/server/src/core/completion.lua
@@ -436,6 +436,21 @@ local function searchEmmyFunctionParam(vm, source, word, callback)
     if not func.args then
         return
     end
+    if word == '' then
+        local list = {}
+        local args = {}
+        for i, arg in ipairs(func.args) do
+            args[i] = arg.name
+            if i == 1 then
+                list[i] = ('%s any'):format(arg.name)
+            else
+                list[i] = ('---@param %s any'):format(arg.name)
+            end
+        end
+        callback(('(%s)'):format(table.concat(args, ', ')), nil, CompletionItemKind.Snippet, {
+            insertText = table.concat(list, '\n')
+        })
+    end
     for _, arg in ipairs(func.args) do
         if matchKey(word, arg.name) then
             callback(arg.name, arg, CompletionItemKind.Interface)
diff --git a/server/test/completion/init.lua b/server/test/completion/init.lua
index f8d1bef0..60e197bc 100644
--- a/server/test/completion/init.lua
+++ b/server/test/completion/init.lua
@@ -850,6 +850,14 @@ end
 ]]
 {
     {
+        label = '(a, b, c)',
+        kind = CompletionItemKind.Snippet,
+        insertText = [[
+a any
+---@param b any
+---@param c any]]
+    },
+    {
         label = 'a',
         kind = CompletionItemKind.Interface,
     },