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

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,
  }
 }
-