feat: extends

tomastrajan · tomastrajan · commit 9fdac1c8d06f · 2026-01-05T11:48:58.000+01:00
diff --git a/src/converters/component/component.converter.spec.ts b/src/converters/component/component.converter.spec.ts
@@ -17,6 +17,7 @@ describe('Component converter', () => {
         implementation: '',
         template: '',
         styles: '',
+        extends: [],
         methods: [],
         methodsPublicExplicit: [],
         fieldsPublicExplicit: [],
@@ -34,6 +35,7 @@ describe('Component converter', () => {
         implementation: '',
         template: '',
         styles: '',
+        extends: [],
         methods: [],
         methodsPublicExplicit: [],
         fieldsPublicExplicit: [],
@@ -51,6 +53,7 @@ describe('Component converter', () => {
         implementation: '',
         template: '',
         styles: '',
+        extends: [],
         methods: [],
         methodsPublicExplicit: [],
         fieldsPublicExplicit: [],
diff --git a/src/converters/directive/directive.converter.spec.ts b/src/converters/directive/directive.converter.spec.ts
@@ -14,6 +14,7 @@ describe('Directive converter', () => {
         inputs: [],
         outputs: [],
         implementation: '',
+        extends: [],
         methods: [],
         methodsPublicExplicit: [],
         fieldsPublicExplicit: [],
@@ -28,6 +29,7 @@ describe('Directive converter', () => {
         inputs: [],
         outputs: [],
         implementation: '',
+        extends: [],
         methods: [],
         methodsPublicExplicit: [],
         fieldsPublicExplicit: [],
@@ -42,6 +44,7 @@ describe('Directive converter', () => {
         inputs: [],
         outputs: [],
         implementation: '',
+        extends: [],
         methods: [],
         methodsPublicExplicit: [],
         fieldsPublicExplicit: [],
diff --git a/src/converters/service/service.converter.component.spec.ts b/src/converters/service/service.converter.component.spec.ts
@@ -9,6 +9,7 @@ describe('Service stats converter', () => {
         className: 'first',
         type: NgParselOutputType.SERVICE,
         filePath: '',
+        extends: [],
         fieldsPublicExplicit: [],
         methods: [],
         methodsPublicExplicit: [],
@@ -17,6 +18,7 @@ describe('Service stats converter', () => {
         className: 'second',
         type: NgParselOutputType.SERVICE,
         filePath: '',
+        extends: [],
         fieldsPublicExplicit: [],
         methods: [],
         methodsPublicExplicit: [],
@@ -25,6 +27,7 @@ describe('Service stats converter', () => {
         className: 'third',
         type: NgParselOutputType.SERVICE,
         filePath: '',
+        extends: [],
         fieldsPublicExplicit: [],
         methods: [],
         methodsPublicExplicit: [],
diff --git a/src/parser/component/component.model.ts b/src/parser/component/component.model.ts
@@ -8,6 +8,7 @@ export interface NgParselComponent extends NgParselOutput {
   selector: string;
   standalone: boolean;
   cva: boolean;
+  extends: string[];
   onPush: boolean;
   inputs: NgParselFieldDecorator[];
   outputs: NgParselFieldDecorator[];
diff --git a/src/parser/component/component.parser.spec.ts b/src/parser/component/component.parser.spec.ts
@@ -40,6 +40,7 @@ describe('ComponentParser', () => {
       selector: 'my-test-comp',
       standalone: false,
       cva: false,
+      extends: [],
       onPush: false,
       template: inlineTemplate,
       styles: [styles],
@@ -133,6 +134,7 @@ describe('ComponentParser', () => {
       selector: 'my-test-comp',
       standalone: true,
       cva: false,
+      extends: [],
       onPush: false,
       template: inlineTemplate,
       styles: [styles],
diff --git a/src/parser/component/component.parser.ts b/src/parser/component/component.parser.ts
@@ -4,6 +4,7 @@ import * as ts from 'typescript';
 import { tsquery } from '@phenomnomnominal/tsquery';
 
 import { isCva } from '../shared/parser/cva.parser.js';
+import { parseExtends } from '../shared/parser/extends.parser.js';
 import { NgParselOutputType } from '../shared/model/types.model.js';
 import { parseInputsAndOutputs } from '../shared/parser/field-decorator.parser.js';
 import { getDecoratorProperties } from '../shared/parser/decorator.parser.js';
@@ -33,6 +34,7 @@ export function parseComponent(ast: ts.SourceFile, componentFilePath: string): N
     filePath: componentFilePath,
     selector: componentDecorators.selector as string,
     standalone: componentDecorators.standalone || false,
+    extends: parseExtends(ast),
     template: template,
     cva: isCva(ast),
     onPush: isOnPushChangeDetection(ast),
diff --git a/src/parser/directive/directive.model.ts b/src/parser/directive/directive.model.ts
@@ -7,6 +7,7 @@ export interface NgParselDirective extends NgParselOutput {
   className: string;
   selector: string;
   standalone: boolean;
+  extends: string[];
   cva: boolean;
   implementation: string;
   inputs: NgParselFieldDecorator[];
diff --git a/src/parser/directive/directive.parser.spec.ts b/src/parser/directive/directive.parser.spec.ts
@@ -32,6 +32,7 @@ describe('DirectiveParser', () => {
       selector: '[myTestDirective]',
       standalone: false,
       cva: false,
+      extends: [],
       inputs: [
         {
           decorator: '@Input()',
@@ -90,6 +91,7 @@ describe('DirectiveParser', () => {
       selector: '[myTestDirective]',
       standalone: true,
       cva: false,
+      extends: [],
       inputs: [
         {
           decorator: '@Input()',
diff --git a/src/parser/directive/directive.parser.ts b/src/parser/directive/directive.parser.ts
@@ -10,6 +10,7 @@ import { parseClassName, parseClassJsDoc } from '../shared/parser/class.parser.j
 import { parseExplicitPublicMethods, parseMethods } from '../shared/parser/method.parser.js';
 
 import { NgParselDirective } from './directive.model.js';
+import { parseExtends } from '../shared/parser/extends.parser.js';
 
 export function parseDirective(ast: ts.SourceFile, directiveFilePath: string): NgParselDirective {
   const directiveDecorators = getDecoratorProperties(ast);
@@ -24,6 +25,7 @@ export function parseDirective(ast: ts.SourceFile, directiveFilePath: string): N
     filePath: directiveFilePath,
     selector: directiveDecorators.selector as string,
     standalone: directiveDecorators.standalone || false,
+    extends: parseExtends(ast),
     cva: isCva(ast),
     implementation: directiveImplementation,
     inputs: inputsAndOutputs.inputs,
diff --git a/src/parser/services/service.model.ts b/src/parser/services/service.model.ts
@@ -3,6 +3,7 @@ import { NgParselMethod } from '../shared/model/method.model.js';
 import { NgParselField } from '../shared/model/field.model.js';
 
 export interface NgParselService extends NgParselOutput {
+  extends: string[];
   fieldsPublicExplicit: NgParselField[];
   methods: NgParselMethod[];
   methodsPublicExplicit: NgParselMethod[];
diff --git a/src/parser/services/service.parser.spec.ts b/src/parser/services/service.parser.spec.ts
@@ -24,6 +24,7 @@ describe('ServiceParser', () => {
       type: NgParselOutputType.SERVICE,
       className,
       filePath,
+      extends: [],
       fieldsPublicExplicit: [],
       methods: [
         {
@@ -89,6 +90,7 @@ describe('ServiceParser', () => {
       className,
       filePath,
       fieldsPublicExplicit: [],
+      extends: [],
       methods: [
         {
           name: 'foo',
@@ -146,6 +148,7 @@ describe('ServiceParser', () => {
       type: NgParselOutputType.SERVICE,
       className,
       filePath,
+      extends: [],
       fieldsPublicExplicit: [
         {
           name: 'counter',
diff --git a/src/parser/services/service.parser.ts b/src/parser/services/service.parser.ts
@@ -1,16 +1,18 @@
 import * as ts from 'typescript';
 
+import { parseExtends } from '../shared/parser/extends.parser.js';
 import { NgParselOutputType } from '../shared/model/types.model.js';
+import { parseExplicitPublicFields } from '../shared/parser/field.parser.js';
 import { parseClassName, parseClassJsDoc } from '../shared/parser/class.parser.js';
 import { parseExplicitPublicMethods, parseMethods } from '../shared/parser/method.parser.js';
 
 import { NgParselService } from './service.model.js';
-import { parseExplicitPublicFields } from '../shared/parser/field.parser.js';
 
 export function parseService(ast: ts.SourceFile, filePath: string): NgParselService {
   return {
     type: NgParselOutputType.SERVICE,
     className: parseClassName(ast),
+    extends: parseExtends(ast),
     filePath,
     fieldsPublicExplicit: parseExplicitPublicFields(ast),
     methods: parseMethods(ast),
diff --git a/src/parser/shared/parser/extends.parser.spec.ts b/src/parser/shared/parser/extends.parser.spec.ts
@@ -0,0 +1,56 @@
+import { tsquery } from '@phenomnomnominal/tsquery';
+import { parseExtends } from './extends.parser.js';
+
+describe('parseExtends', () => {
+  it('parses extends for a component-like class with extends', () => {
+    const ast = tsquery.ast(`
+      @Component({ selector: 'x' })
+      export class MyComponent extends BaseComponent {}
+    `);
+
+    expect(parseExtends(ast)).toEqual(['BaseComponent']);
+  });
+
+  it('returns an empty array for a component-like class without extends', () => {
+    const ast = tsquery.ast(`
+      @Component({ selector: 'x' })
+      export class MyComponent {}
+    `);
+
+    expect(parseExtends(ast)).toEqual([]);
+  });
+
+  it('parses extends for a directive-like class with extends', () => {
+    const ast = tsquery.ast(`
+      @Directive({ selector: '[x]' })
+      export class MyDirective extends SomeDirectiveBase {}
+    `);
+
+    expect(parseExtends(ast)).toEqual(['SomeDirectiveBase']);
+  });
+
+  it('returns an empty array for a directive-like class without extends', () => {
+    const ast = tsquery.ast(`
+      @Directive({ selector: '[x]' })
+      export class MyDirective {}
+    `);
+
+    expect(parseExtends(ast)).toEqual([]);
+  });
+
+  it('parses extends for a service-like class with generic extends', () => {
+    const ast = tsquery.ast(`
+      export class MyService extends BaseService<Thing> {}
+    `);
+
+    expect(parseExtends(ast)).toEqual(['BaseService']);
+  });
+
+  it('returns an empty array for a service-like class without extends', () => {
+    const ast = tsquery.ast(`
+      export class MyService {}
+    `);
+
+    expect(parseExtends(ast)).toEqual([]);
+  });
+});
diff --git a/src/parser/shared/parser/extends.parser.ts b/src/parser/shared/parser/extends.parser.ts
@@ -0,0 +1,21 @@
+import * as ts from 'typescript';
+import { tsquery } from '@phenomnomnominal/tsquery';
+
+export function parseExtends(ast: ts.SourceFile): string[] {
+  const identifiers = tsquery(ast, 'HeritageClause > ExpressionWithTypeArguments > Identifier') as any[];
+
+  const result: string[] = [];
+
+  for (const id of identifiers) {
+    // id.parent -> ExpressionWithTypeArguments
+    // id.parent.parent -> HeritageClause
+    const heritageClause = (id.parent as any)?.parent as ts.HeritageClause | undefined;
+    if (!heritageClause) continue;
+
+    if (heritageClause.token === ts.SyntaxKind.ExtendsKeyword) {
+      result.push(id.getText());
+    }
+  }
+
+  return result;
+}
