Start adding support for enumerated arrays for completion.

3 files changed, 42 insertions, 0 deletions

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index c482206..d6592ae 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -35,6 +35,8 @@ DocumentPositionContext :: struct {
 	identifier:             ^ast.Node,
 	label:                  ^ast.Ident,
 	implicit_context:       ^ast.Implicit,
+	index:                  ^ast.Index_Expr,
+	previous_index:         ^ast.Index_Expr,
 	tag:                    ^ast.Node,
 	field:                  ^ast.Expr, //used for completion
 	call:                   ^ast.Expr, //used for signature help
@@ -4957,6 +4959,8 @@ get_document_position_node :: proc(
 			get_document_position(n.field, position_context)
 		}
 	case ^Index_Expr:
+		position_context.previous_index = position_context.index
+		position_context.index = n
 		get_document_position(n.expr, position_context)
 		get_document_position(n.index, position_context)
 	case ^Deref_Expr:
diff --git a/src/server/completion.odin b/src/server/completion.odin
index 2e5689f..a863d5c 100644
--- a/src/server/completion.odin
+++ b/src/server/completion.odin
@@ -1283,6 +1283,42 @@ get_implicit_completion :: proc(
 			}
 		}
 	}
+
+	if position_context.index != nil {
+		symbol: Symbol
+		ok := false
+		if position_context.previous_index != nil {
+			symbol, ok = resolve_type_expression(
+				ast_context,
+				position_context.previous_index,
+			)
+			if !ok {
+				return
+			}
+		} else {
+			symbol, ok = resolve_type_expression(
+				ast_context,
+				position_context.index.expr,
+			)
+		}
+
+		if array, ok := symbol.value.(SymbolFixedArrayValue); ok {
+			if enum_value, ok := unwrap_enum(ast_context, array.len); ok {
+				for name in enum_value.names {
+					item := CompletionItem {
+						label  = name,
+						kind   = .EnumMember,
+						detail = name,
+					}
+
+					append(&items, item)
+				}
+
+				list.items = items[:]
+				return
+			}
+		}
+	}
 }
 
 get_identifier_completion :: proc(
diff --git a/src/server/file_resolve.odin b/src/server/file_resolve.odin
index 64bdd56..39186de 100644
--- a/src/server/file_resolve.odin
+++ b/src/server/file_resolve.odin
@@ -32,6 +32,8 @@ reset_position_context :: proc(position_context: ^DocumentPositionContext) {
 	position_context.call = nil
 	position_context.binary = nil
 	position_context.parent_binary = nil
+	position_context.previous_index = nil
+	position_context.index = nil
 }
 
 resolve_entire_file :: proc(