Add SymbolMapValue to selector resolve in the index case

2 files changed, 22 insertions, 2 deletions

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index 7170935..0a2fe1d 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -1842,8 +1842,11 @@ resolve_implicit_selector :: proc(
 			}
 		}
 
-		if array, ok := symbol.value.(SymbolFixedArrayValue); ok {
-			return resolve_type_expression(ast_context, array.len)
+		#partial switch value in symbol.value {
+		case SymbolFixedArrayValue:
+			return resolve_type_expression(ast_context, value.len)
+		case SymbolMapValue:
+			return resolve_type_expression(ast_context, value.key)
 		}
 	}
 
diff --git a/tests/hover_test.odin b/tests/hover_test.odin
index 18c2ced..5e6a66f 100644
--- a/tests/hover_test.odin
+++ b/tests/hover_test.odin
@@ -1961,6 +1961,23 @@ ast_hover_enum_field_definition_with_type :: proc(t: ^testing.T) {
 	}
 	test.expect_hover(t, &source, "test.Foo: .A")
 }
+
+@(test)
+ast_hover_enum_map_key :: proc(t: ^testing.T) {
+	source :=test.Source {
+		main = `package test
+		Foo :: enum {
+			A = 1,
+			B,
+		}
+		main :: proc() {
+			m: map[Foo]int
+			m[.A{*}] = 2
+		}
+		`
+	}
+	test.expect_hover(t, &source, "test.Foo: .A")
+}
 /*
 
 Waiting for odin fix