feat: Fix TypeScript inheritance extraction by properly handling class_heritage wrapper nodes

colbymchenry · colbymchenry · commit ce7b7684dbe3 · 2026-04-06T21:13:57.000-05:00
Addresses TypeScript's AST structure where class_heritage nodes wrap extends_clause and implements_clause rather than directly indicating inheritance relationships. Moves class_heritage from direct inheritance extraction to recursive container processing to properly traverse the wrapped inheritance syntax.
diff --git a/src/extraction/tree-sitter.ts b/src/extraction/tree-sitter.ts
@@ -1214,7 +1214,6 @@ export class TreeSitterExtractor {
 
       if (
         child.type === 'extends_clause' ||
-        child.type === 'class_heritage' ||
         child.type === 'superclass' ||
         child.type === 'extends_interfaces' // Java interface extends
       ) {
@@ -1329,8 +1328,9 @@ export class TreeSitterExtractor {
         }
       }
 
-      // Recurse into container nodes (e.g. field_declaration_list in Go structs)
-      if (child.type === 'field_declaration_list') {
+      // Recurse into container nodes (e.g. field_declaration_list in Go structs,
+      // class_heritage in TypeScript which wraps extends_clause/implements_clause)
+      if (child.type === 'field_declaration_list' || child.type === 'class_heritage') {
         this.extractInheritance(child, classId);
       }
     }

Original file line number	Diff line number	Diff line change
`@@ -1214,7 +1214,6 @@ export class TreeSitterExtractor {`
`1214`	`1214`
`1215`	`1215`	`if (`
`1216`	`1216`	`child.type === 'extends_clause' \|\|`
`1217`		`- child.type === 'class_heritage' \|\|`
`1218`	`1217`	`child.type === 'superclass' \|\|`
`1219`	`1218`	`child.type === 'extends_interfaces' // Java interface extends`
`1220`	`1219`	`) {`
`@@ -1329,8 +1328,9 @@ export class TreeSitterExtractor {`
`1329`	`1328`	`}`
`1330`	`1329`	`}`
`1331`	`1330`
`1332`		`- // Recurse into container nodes (e.g. field_declaration_list in Go structs)`
`1333`		`- if (child.type === 'field_declaration_list') {`
	`1331`	`+ // Recurse into container nodes (e.g. field_declaration_list in Go structs,`
	`1332`	`+ // class_heritage in TypeScript which wraps extends_clause/implements_clause)`
	`1333`	`+ if (child.type === 'field_declaration_list' \|\| child.type === 'class_heritage') {`
`1334`	`1334`	`this.extractInheritance(child, classId);`
`1335`	`1335`	`}`
`1336`	`1336`	`}`