Fix Flow error

babel · Dec 6, 2019 · c25b1e7 · c25b1e7
1 parent fd1b6a5
commit c25b1e7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/packages/babel-parser/src/plugins/estree.js b/packages/babel-parser/src/plugins/estree.js
@@ -400,14 +400,17 @@ export default (superClass: Class<Parser>): Class<Parser> =>
     finishCallExpression<T: N.CallExpression | N.OptionalCallExpression>(
       node: T,
       optional: boolean,
-    ): T & { source?: N.Node } {
-      const finishedNode = super.finishCallExpression(node, optional);
-
+    ): T | N.EstreeImportExpression {
       if (node.callee.type === "Import") {
+        const finishedNode: N.EstreeImportExpression = (super.finishCallExpression(
+          node,
+          optional,
+        ): any);
         finishedNode.type = "ImportExpression";
         finishedNode.source = node.arguments[0];
+        return finishedNode;
       }
 
-      return finishedNode;
+      return super.finishCallExpression(node, optional);
     }
   };
diff --git a/packages/babel-parser/src/types.js b/packages/babel-parser/src/types.js
@@ -1012,7 +1012,7 @@ export type FlowInterfaceType = NodeBase & {
   body: FlowObjectTypeAnnotation,
 };
 
-// estree
+// ESTree
 
 export type EstreeProperty = NodeBase & {
   type: "Property",
@@ -1038,6 +1038,13 @@ export type EstreeMethodDefinition = NodeBase & {
   variance?: ?FlowVariance,
 };
 
+export type EstreeImportExpression = NodeBase & {
+  type: "ImportExpression",
+  callee: Import,
+  arguments: Array<Expression>,
+  source: Expression,
+};
+
 // === === === ===
 // TypeScript
 // === === === ===