feat: update TSFunctionType and TSConstructorType nodes (#91)

BREAKING CHANGE: This changes the AST for TSFunctionType and TSConstructorType nodes

Author: armano2

src/convert.ts

@@ -122,7 +122,14 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
    */
   function convertTypeAnnotation(child: ts.TypeNode): ESTreeNode {
     const annotation = convertChildType(child);
-    const annotationStartCol = child.getFullStart() - 1;
+    // in FunctionType and ConstructorType typeAnnotation has 2 characters `=>` and in other places is just colon
+    const offset =
+      node.kind === SyntaxKind.FunctionType ||
+      node.kind === SyntaxKind.ConstructorType
+        ? 2
+        : 1;
+    const annotationStartCol = child.getFullStart() - offset;
+
     const loc = nodeUtils.getLocFor(annotationStartCol, child.end, ast);
     return {
       type: AST_NODE_TYPES.TSTypeAnnotation,
@@ -2439,14 +2446,28 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
 
       break;
     }
-
+    case SyntaxKind.ConstructorType:
+    case SyntaxKind.FunctionType:
     case SyntaxKind.ConstructSignature:
     case SyntaxKind.CallSignature: {
+      let type: AST_NODE_TYPES;
+      switch (node.kind) {
+        case SyntaxKind.ConstructSignature:
+          type = AST_NODE_TYPES.TSConstructSignatureDeclaration;
+          break;
+        case SyntaxKind.CallSignature:
+          type = AST_NODE_TYPES.TSCallSignatureDeclaration;
+          break;
+        case SyntaxKind.FunctionType:
+          type = AST_NODE_TYPES.TSFunctionType;
+          break;
+        case SyntaxKind.ConstructorType:
+        default:
+          type = AST_NODE_TYPES.TSConstructorType;
+          break;
+      }
       Object.assign(result, {
-        type:
-          node.kind === SyntaxKind.ConstructSignature
-            ? AST_NODE_TYPES.TSConstructSignatureDeclaration
-            : AST_NODE_TYPES.TSCallSignatureDeclaration,
+        type: type,
         params: convertParameters(node.parameters)
       });
 

tests/ast-alignment/fixtures-to-test.ts

@@ -384,7 +384,6 @@ tester.addFixturePatternConfig('typescript/basics', {
     'interface-with-optional-properties',
     'interface-without-type-annotation',
     'typed-this',
-    'export-type-function-declaration',
     'abstract-interface',
     /**
      * Babel bug for optional or abstract methods?
@@ -480,7 +479,14 @@ tester.addFixturePatternConfig('typescript/errorRecovery', {
 });
 
 tester.addFixturePatternConfig('typescript/types', {
-  fileType: 'ts'
+  fileType: 'ts',
+  ignore: [
+    /**
+     * AST difference
+     */
+    'function-with-rest',
+    'constructor-with-rest'
+  ]
 });
 
 tester.addFixturePatternConfig('typescript/declare', {

tests/ast-alignment/utils.ts

@@ -168,6 +168,32 @@ export function preprocessBabylonAST(ast: any): any {
           node.params = node.parameters;
           delete node.parameters;
         }
+      },
+      /**
+       * Awaiting feedback on Babel issue https://github.com/babel/babel/issues/9231
+       */
+      TSFunctionType(node: any) {
+        if (node.typeAnnotation) {
+          node.returnType = node.typeAnnotation;
+          delete node.typeAnnotation;
+        }
+        if (node.parameters) {
+          node.params = node.parameters;
+          delete node.parameters;
+        }
+      },
+      /**
+       * Awaiting feedback on Babel issue https://github.com/babel/babel/issues/9231
+       */
+      TSConstructorType(node: any) {
+        if (node.typeAnnotation) {
+          node.returnType = node.typeAnnotation;
+          delete node.typeAnnotation;
+        }
+        if (node.parameters) {
+          node.params = node.parameters;
+          delete node.parameters;
+        }
       }
     }
   );

tests/fixtures/typescript/types/constructor-generic.src.ts

@@ -0,0 +1 @@
+let f: new <T>(a: T) => T;

tests/fixtures/typescript/types/constructor-in-generic.src.ts

@@ -0,0 +1 @@
+let x: Array<new () => string>;

tests/fixtures/typescript/types/constructor-with-rest.src.ts

@@ -0,0 +1 @@
+let f: new (...a: number[]) => void;

tests/fixtures/typescript/types/constructor.src.ts

@@ -0,0 +1 @@
+let f: new (a: number, b?: number) => void;

tests/fixtures/typescript/types/function-generic.src.ts

@@ -0,0 +1 @@
+let f: <T>(a: T) => T;

tests/fixtures/typescript/types/function-in-generic.src.ts

@@ -0,0 +1 @@
+let x: Array<() => void>;

tests/fixtures/typescript/types/function-with-rest.src.ts

@@ -0,0 +1 @@
+let f: (...a: number[]) => void;