doc.enum

1 files changed, 9 insertions, 0 deletions

diff --git a/script/core/completion/completion.lua b/script/core/completion/completion.lua
index 908398e7..01b5fcfe 100644
--- a/script/core/completion/completion.lua
+++ b/script/core/completion/completion.lua
@@ -1648,6 +1648,7 @@ local function tryluaDocBySource(state, position, source, results)
         for _, doc in ipairs(vm.getDocSets(state.uri)) do
             local name = (doc.type == 'doc.class' and doc.class[1])
                     or   (doc.type == 'doc.alias' and doc.alias[1])
+                    or   (doc.type == 'doc.enum'  and doc.enum[1])
             if  name
             and not used[name]
             and matchKey(source[1], name) then
@@ -1803,6 +1804,14 @@ local function tryluaDocByErr(state, position, err, docState, results)
                     kind        = define.CompletionItemKind.Class,
                 }
             end
+            if  doc.type == 'doc.enum'
+            and not used[doc.enum[1]] then
+                used[doc.enum[1]] = true
+                results[#results+1] = {
+                    label       = doc.enum[1],
+                    kind        = define.CompletionItemKind.Enum,
+                }
+            end
         end
     elseif err.type == 'LUADOC_MISS_PARAM_NAME' then
         local funcs = {}