completion: support delegate

Author: sumneko

changelog.md

@@ -2,6 +2,7 @@
 
 ## 1.6.0
 * `NEW` completion: auto require local modules
+* `NEW` completion: support delegate
 * `NEW` hover: show function by keyword `function`
 * `NEW` code action: swap params
 * `CHG` standalone: unbind the relative path between binaries and scripts

script/core/completion.lua

@@ -6,6 +6,7 @@ local vm         = require 'vm'
 local getLabel   = require 'core.hover.label'
 local getName    = require 'core.hover.name'
 local getArg     = require 'core.hover.arg'
+local getReturn  = require 'core.hover.return'
 local getDesc    = require 'core.hover.description'
 local getHover   = require 'core.hover'
 local config     = require 'config'
@@ -939,8 +940,9 @@ local function mergeEnums(a, b, source)
             mark[label] = true
             local result = {
                 label       = label,
-                kind        = define.CompletionItemKind.EnumMember,
+                kind        = enum.kind,
                 description = enum.description,
+                insertText  = enum.insertText,
                 textEdit    = source and {
                     start   = source.start,
                     finish  = source.finish,
@@ -1140,6 +1142,17 @@ local function trySymbol(ast, text, offset, results)
     end
 end
 
+local function buildInsertDocFunction(doc)
+    local args = {}
+    for i, arg in ipairs(doc.args) do
+        args[i] = ('${%d:%s}'):format(i, arg.name[1])
+    end
+    return ([[
+function (%s)
+    $0
+end]]):format(table.concat(args, ', '))
+end
+
 local function getCallEnums(source, index)
     if source.type == 'function' and source.bindDocs then
         if not source.args then
@@ -1167,12 +1180,14 @@ local function getCallEnums(source, index)
                         kind        = define.CompletionItemKind.EnumMember,
                     }
                 end
-                for _, unit in ipairs(vm.getDocTypes(doc.extends)) do
+                for _, unit in ipairs(vm.getDocTypeUnits(doc.extends)) do
                     if unit.type == 'doc.type.function' then
+                        local text = files.getText(guide.getUri(unit))
                         enums[#enums+1] = {
-                            label       = guide.getDocTypeUnitName(nil, unit),
+                            label       = text:sub(unit.start, unit.finish),
                             description = doc.comment,
                             kind        = define.CompletionItemKind.Function,
+                            insertText  = buildInsertDocFunction(unit),
                         }
                     end
                 end

script/vm/getDocs.lua

@@ -74,6 +74,35 @@ function vm.getDocEnums(doc, mark, results)
     return results
 end
 
+function vm.getDocTypeUnits(doc, mark, results)
+    mark = mark or {}
+    if mark[doc] then
+        return nil
+    end
+    mark[doc] = true
+    results = results or {}
+    for _, enum in ipairs(doc.enums) do
+        results[#results+1] = enum
+    end
+    for _, resume in ipairs(doc.resumes) do
+        results[#results+1] = resume
+    end
+    for _, unit in ipairs(doc.types) do
+        if unit.type == 'doc.type.name' then
+            for _, other in ipairs(vm.getDocTypes(unit[1])) do
+                if other.type == 'doc.alias.name' then
+                    vm.getDocTypeUnits(other.parent.extends, mark, results)
+                elseif other.type == 'doc.class.name' then
+                    results[#results+1] = other
+                end
+            end
+        else
+            results[#results+1] = unit
+        end
+    end
+    return results
+end
+
 function vm.getDocTypes(name)
     local cache = vm.getCache('getDocTypes')[name]
     if cache ~= nil then

test/completion/init.lua

@@ -1953,17 +1953,21 @@ vvv$
     },
 }
 
---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,
---    },
---}
+Cared['insertText'] = true
+TEST [[
+---@param callback fun(x: number, y: number):string
+local function f(callback) end
+
+f($)
+]]
+{
+    {
+        label  = 'fun(x: number, y: number):string',
+        kind   = define.CompletionItemKind.Function,
+        insertText = [[
+function (${1:x}, ${2:y})
+    $0
+end]],
+    },
+}
+Cared['insertText'] = nil