Fix missing parens when function expressions is tag (#5193)

babel · Jan 23, 2017 · b69dc51 · b69dc51
1 parent c638985
commit b69dc51
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 10 deletions.
diff --git a/packages/babel-generator/src/node/parentheses.js b/packages/babel-generator/src/node/parentheses.js
@@ -170,16 +170,14 @@ export function FunctionExpression(node: Object, parent: Object, printStack: Arr
 }
 
 export function ArrowFunctionExpression(node: Object, parent: Object): boolean {
-  // export default (function () {});
-  if (t.isExportDeclaration(parent)) {
-    return true;
-  }
-
-  if (t.isBinaryExpression(parent) || t.isLogicalExpression(parent)) {
-    return true;
-  }
-
-  if (t.isUnaryExpression(parent)) {
+  if (
+    // export default (function () {});
+    t.isExportDeclaration(parent) ||
+    t.isBinaryExpression(parent) ||
+    t.isLogicalExpression(parent) ||
+    t.isUnaryExpression(parent) ||
+    t.isTaggedTemplateExpression(parent)
+  ) {
     return true;
   }
 
@@ -225,6 +223,10 @@ function isFirstInStatement(printStack: Array<Object>, {
       return true;
     }
 
+    if (t.isTaggedTemplateExpression(parent)) {
+      return true;
+    }
+
     if (considerDefaultExports && t.isExportDefaultDeclaration(parent, { declaration: node })) {
       return true;
     }

diff --git a/packages/babel-generator/test/fixtures/parentheses/tagged-template-expression/actual.js b/packages/babel-generator/test/fixtures/parentheses/tagged-template-expression/actual.js
@@ -0,0 +1,2 @@
+(() => {})``;
+(function(){}``);
diff --git a/packages/babel-generator/test/fixtures/parentheses/tagged-template-expression/expected.js b/packages/babel-generator/test/fixtures/parentheses/tagged-template-expression/expected.js
@@ -0,0 +1,2 @@
+(() => {})``;
+(function () {})``;