FIX Autocompletion for enum values ​​is not available in some cases

Author: sumneko

changelog.md

@@ -2,9 +2,9 @@
 
 ## Unreleased
 <!-- Add all new changes here. They will be moved under a version at release -->
-* `FIX` Don't do diagnostics when the workspace is not ready
 * `NEW` Reference workspace symbols in comments using `[some text](lua://symbolName)` syntax
-
+* `FIX` Don't do diagnostics when the workspace is not ready
+* `FIX` Autocompletion for enum values ​​is not available in some cases
 ## 3.9.1
 `2024-5-14`
 * revert extension runtime

script/core/completion/completion.lua

@@ -1357,18 +1357,21 @@ local function insertEnum(state, pos, src, enums, isInArray, mark)
             kind        = define.CompletionItemKind.Function,
             insertText  = insertText,
         }
+    elseif src.type == 'doc.enum' then
+        ---@cast src parser.object
+        if vm.docHasAttr(src, 'key') then
+            insertDocEnumKey(state, pos, src, enums)
+        else
+            insertDocEnum(state, pos, src, enums)
+        end
     elseif isInArray and src.type == 'doc.type.array' then
         for i, d in ipairs(vm.getDefs(src.node)) do
             insertEnum(state, pos, d, enums, isInArray, mark)
         end
     elseif src.type == 'global' and src.cate == 'type' then
         for _, set in ipairs(src:getSets(state.uri)) do
             if set.type == 'doc.enum' then
-                if vm.docHasAttr(set, 'key') then
-                    insertDocEnumKey(state, pos, set, enums)
-                else
-                    insertDocEnum(state, pos, set, enums)
-                end
+                insertEnum(state, pos, set, enums, isInArray, mark)
             end
         end
     end

test/completion/common.lua

@@ -3841,6 +3841,35 @@ f(<??>)
     },
 }
 
+TEST [[
+---@class optional
+---@field enum enum
+
+---@enum(key) enum
+local t = {
+    a = 1,
+    b = 2,
+}
+
+---@param a optional
+local function f(a)
+end
+
+f {
+    enum = <??>
+}
+]]
+{
+    {
+        label    = '"a"',
+        kind     = define.CompletionItemKind.EnumMember,
+    },
+    {
+        label    = '"b"',
+        kind     = define.CompletionItemKind.EnumMember,
+    },
+}
+
 TEST [[
 --
 <??>
@@ -4433,4 +4462,3 @@ new 'A' {
         kind  = define.CompletionItemKind.Property,
     }
 }
-