use finishToken instead of finishNode

babel · Apr 24, 2019 · aa3982b · aa3982b
1 parent e64cbdf
commit aa3982b
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/packages/babel-parser/src/plugins/flow.js b/packages/babel-parser/src/plugins/flow.js
@@ -95,18 +95,13 @@ export default (superClass: Class<Parser>): Class<Parser> =>
       return this.getPluginOption("flow", "all") || this.flowPragma === "flow";
     }
 
-    finishNodeAt<T: N.Node>(
-      node: T,
-      type: string,
-      pos: number,
-      loc: Position,
-    ): T {
-      if (type !== "StringLiteral") {
+    finishToken(type: TokenType, val: any): void {
+      if (!(type === tt.string || type === tt.semi)) {
         if (this.flowPragma === undefined) {
           this.flowPragma = null;
         }
       }
-      return super.finishNodeAt(node, type, pos, loc);
+      return super.finishToken(type, val);
     }
 
     addComment(comment: N.Comment): void {