Fix enum definition name hover

kralicky · Jan 4, 2024 · 56baafc · 56baafc
1 parent 21c25f7
commit 56baafc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/lsp/search.go b/pkg/lsp/search.go
@@ -352,7 +352,12 @@ func deepPathSearch(path []ast.Node, parseRes parser.Result, linkRes linker.Resu
 				case *ast.ReservedNode:
 				}
 			case ast.IdentValueNode:
-				want.desc = haveDesc.Values().ByName(protoreflect.Name(wantNode.AsIdentifier()))
+				// this could be either the enum name itself or a value name
+				if haveNode, ok := have.node.(*ast.EnumNode); ok && haveNode.Name == wantNode {
+					want.desc = haveDesc
+				} else {
+					want.desc = haveDesc.Values().ByName(protoreflect.Name(wantNode.AsIdentifier()))
+				}
 			}
 		case protoreflect.EnumValueDescriptor:
 			switch wantNode := want.node.(type) {