feat: fix issues in AST

BREAKING CHANGE: This changes the AST

Author: armano2

src/ast-node-types.ts

@@ -26,7 +26,6 @@ export const AST_NODE_TYPES: { [key: string]: string } = {
   ConditionalExpression: 'ConditionalExpression',
   ContinueStatement: 'ContinueStatement',
   DebuggerStatement: 'DebuggerStatement',
-  DeclareFunction: 'DeclareFunction',
   Decorator: 'Decorator',
   DoWhileStatement: 'DoWhileStatement',
   EmptyStatement: 'EmptyStatement',
@@ -100,6 +99,7 @@ export const AST_NODE_TYPES: { [key: string]: string } = {
   TSConstructorType: 'TSConstructorType',
   TSConstructSignature: 'TSConstructSignature',
   TSDeclareKeyword: 'TSDeclareKeyword',
+  TSDeclareFunction: 'TSDeclareFunction',
   TSEnumDeclaration: 'TSEnumDeclaration',
   TSEnumMember: 'TSEnumMember',
   TSExportAssignment: 'TSExportAssignment',

src/convert.ts

@@ -694,15 +694,8 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
 
     case SyntaxKind.FunctionDeclaration: {
       let functionDeclarationType = AST_NODE_TYPES.FunctionDeclaration;
-
-      if (node.modifiers && node.modifiers.length) {
-        const isDeclareFunction = nodeUtils.hasModifier(
-          SyntaxKind.DeclareKeyword,
-          node
-        );
-        if (isDeclareFunction) {
-          functionDeclarationType = AST_NODE_TYPES.DeclareFunction;
-        }
+      if (nodeUtils.hasModifier(SyntaxKind.DeclareKeyword, node)) {
+        functionDeclarationType = AST_NODE_TYPES.TSDeclareFunction;
       }
 
       Object.assign(result, {
@@ -712,14 +705,18 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
         expression: false,
         async: nodeUtils.hasModifier(SyntaxKind.AsyncKeyword, node),
         params: convertParameters(node.parameters),
-        body: convertChild(node.body)
+        body: convertChild(node.body) || undefined
       });
 
       // Process returnType
       if (node.type) {
         (result as any).returnType = convertTypeAnnotation(node.type);
       }
 
+      if (functionDeclarationType === AST_NODE_TYPES.TSDeclareFunction) {
+        result.declare = true;
+      }
+
       // Process typeParameters
       if (node.typeParameters && node.typeParameters.length) {
         result.typeParameters = convertTSTypeParametersToTypeParametersDeclaration(
@@ -758,6 +755,10 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
         kind: nodeUtils.getDeclarationKind(node.declarationList)
       });
 
+      if (nodeUtils.hasModifier(SyntaxKind.DeclareKeyword, node)) {
+        result.declare = true;
+      }
+
       // check for exports
       result = nodeUtils.fixExports(node, result as any, ast);
       break;
@@ -1618,6 +1619,10 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
         );
       }
 
+      if (nodeUtils.hasModifier(SyntaxKind.DeclareKeyword, node)) {
+        result.declare = true;
+      }
+
       if (node.decorators) {
         result.decorators = convertDecorators(node.decorators);
       }
@@ -2400,10 +2405,6 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
       }
 
       const hasImplementsClause = interfaceHeritageClauses.length > 0;
-      const hasAbstractKeyword = nodeUtils.hasModifier(
-        SyntaxKind.AbstractKeyword,
-        node
-      );
       const interfaceOpenBrace = nodeUtils.findNextToken(
         interfaceLastClassToken,
         ast,
@@ -2422,7 +2423,6 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
       };
 
       Object.assign(result, {
-        abstract: hasAbstractKeyword,
         type: AST_NODE_TYPES.TSInterfaceDeclaration,
         body: interfaceBody,
         id: convertChild(node.name),
@@ -2440,6 +2440,12 @@ export default function convert(config: ConvertConfig): ESTreeNode | null {
       if (node.decorators) {
         result.decorators = convertDecorators(node.decorators);
       }
+      if (nodeUtils.hasModifier(SyntaxKind.AbstractKeyword, node)) {
+        result.abstract = true;
+      }
+      if (nodeUtils.hasModifier(SyntaxKind.DeclareKeyword, node)) {
+        result.declare = true;
+      }
       // check for exports
       result = nodeUtils.fixExports(node, result as any, ast);
 

src/temp-types-based-on-js-source.ts

@@ -41,6 +41,7 @@ export interface ESTreeNode {
   static?: boolean;
   export?: boolean;
   parameter?: any;
+  abstract?: boolean;
 }
 
 export interface ESTreeComment extends ESTreeNode {}

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

@@ -395,11 +395,6 @@ let fixturePatternConfigsToTest = [
       'class-with-implements-generic',
       'class-with-implements',
       'class-with-extends-and-implements',
-      /**
-       * Babylon: TSDeclareFunction + declare: true
-       * tsep: DeclareFunction
-       */
-      'declare-function',
       /**
        * Babylon: TSTypeReference + identifier
        * tsep: TSUnknownKeyword
@@ -508,6 +503,18 @@ let fixturePatternConfigsToTest = [
     ]
   }),
 
+  createFixturePatternConfigFor('typescript/declare', {
+    fileType: 'ts',
+    ignore: [
+      /**
+       * AST difference
+       */
+      'type-alias',
+      'interface',
+      'abstract-class'
+    ]
+  }),
+
   createFixturePatternConfigFor('typescript/namespaces-and-modules', {
     fileType: 'ts',
     ignore: [

tests/fixtures/typescript/declare/abstract-class.src.ts

@@ -0,0 +1,3 @@
+declare abstract class Foo {
+
+}

tests/fixtures/typescript/declare/class.src.ts

@@ -0,0 +1,3 @@
+declare class Foo {
+
+}

tests/fixtures/typescript/declare/enum.src.ts

@@ -0,0 +1,4 @@
+declare enum Foo {
+    Bar,
+    Baz
+}

tests/fixtures/typescript/declare/function.src.ts

@@ -0,0 +1 @@
+declare function foo(): void

tests/fixtures/typescript/declare/interface.src.ts

@@ -0,0 +1,3 @@
+declare interface Foo {
+
+}

tests/fixtures/typescript/declare/module.src.ts

@@ -0,0 +1,3 @@
+declare module Foo {
+
+}

tests/fixtures/typescript/declare/namespace.src.ts

@@ -0,0 +1,3 @@
+declare namespace Foo {
+
+}

tests/fixtures/typescript/declare/type-alias.src.ts

@@ -0,0 +1 @@
+declare type Foo = string

tests/fixtures/typescript/declare/variable.src.ts

@@ -0,0 +1 @@
+declare var foo: any;

tests/lib/__snapshots__/javascript.ts.snap

@@ -105760,7 +105760,6 @@ Object {
   "body": Array [
     Object {
       "async": false,
-      "body": null,
       "expression": false,
       "generator": false,
       "id": Object {
@@ -106077,7 +106076,6 @@ Object {
   "body": Array [
     Object {
       "async": false,
-      "body": null,
       "expression": false,
       "generator": false,
       "id": Object {