fixed jsx case handling

Author: johnfav03

internal/ls/documenthighlights.go

@@ -19,18 +19,23 @@ func (l *LanguageService) ProvideDocumentHighlights(ctx context.Context, documen
 	position := int(l.converters.LineAndCharacterToPosition(sourceFile, documentPosition))
 	node := astnav.GetTouchingPropertyName(sourceFile, position)
 
-	if node.Parent.Kind == ast.KindJsxOpeningElement || (node.Parent.Kind == ast.KindJsxClosingElement && node.Parent.TagName() == node) {
+	if node.Parent != nil && (node.Parent.Kind == ast.KindJsxClosingElement || (node.Parent.Kind == ast.KindJsxOpeningElement && node.Parent.TagName() == node)) {
 		var documentHighlights []*lsproto.DocumentHighlight
+		var openingElement, closingElement *ast.Node
 		kind := lsproto.DocumentHighlightKindRead
-		if node.Parent.Kind == ast.KindJsxOpeningElement {
+		if ast.IsJsxElement(node.Parent.Parent) {
+			openingElement = node.Parent.Parent.AsJsxElement().OpeningElement
+			closingElement = node.Parent.Parent.AsJsxElement().ClosingElement
+		}
+		if openingElement != nil {
 			documentHighlights = append(documentHighlights, &lsproto.DocumentHighlight{
-				Range: *l.createLspRangeFromNode(node.Parent, sourceFile),
+				Range: *l.createLspRangeFromNode(openingElement, sourceFile),
 				Kind:  &kind,
 			})
 		}
-		if node.Parent.Kind == ast.KindJsxClosingElement {
+		if closingElement != nil {
 			documentHighlights = append(documentHighlights, &lsproto.DocumentHighlight{
-				Range: *l.createLspRangeFromNode(node.Parent, sourceFile),
+				Range: *l.createLspRangeFromNode(closingElement, sourceFile),
 				Kind:  &kind,
 			})
 		}

testdata/baselines/reference/fourslash/documentHighlights/findReferencesJSXTagName3.baseline.jsonc

@@ -28,8 +28,11 @@
 //     [|</*HIGHLIGHTS*/div>|]
 //         Some content
 //         <div>More content</div>
-//     </div>;
-// // --- (line: 13) skipped ---
+//     [|</div>|];
+// 
+// const x = <Comp>
+//     Content
+// </Comp>;
 
 
 
@@ -39,7 +42,7 @@
 // const Comp = () =>
 //     <div>
 //         Some content
-//         [|</*HIGHLIGHTS*/div>|]More content</div>
+//         [|</*HIGHLIGHTS*/div>|]More content[|</div>|]
 //     </div>;
 // 
 // const x = <Comp>
@@ -54,7 +57,7 @@
 // const Comp = () =>
 //     <div>
 //         Some content
-//         <div>More content[|<//*HIGHLIGHTS*/div>|]
+//         [|<div>|]More content[|<//*HIGHLIGHTS*/div>|]
 //     </div>;
 // 
 // const x = <Comp>
@@ -65,8 +68,11 @@
 
 // === documentHighlights ===
 // === /a.tsx ===
-// --- (line: 8) skipped ---
-//     <div>
+// --- (line: 5) skipped ---
+// }
+// 
+// const Comp = () =>
+//     [|<div>|]
 //         Some content
 //         <div>More content</div>
 //     [|<//*HIGHLIGHTS*/div>|];
@@ -103,14 +109,16 @@
 // 
 // const x = [|</*HIGHLIGHTS*/Comp>|]
 //     Content
-// </Comp>;
+// [|</Comp>|];
 
 
 
 // === documentHighlights ===
 // === /a.tsx ===
-// --- (line: 12) skipped ---
+// --- (line: 10) skipped ---
+//         <div>More content</div>
+//     </div>;
 // 
-// const x = <Comp>
+// const x = [|<Comp>|]
 //     Content
 // [|<//*HIGHLIGHTS*/Comp>|];