🍺 fixed the return statement issue

vivek12345 · Jan 31, 2019 · 723a62b · 723a62b
1 parent 5b95c25
commit 723a62b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/transforms/async-await-with-try-catch.js b/transforms/async-await-with-try-catch.js
@@ -31,7 +31,11 @@ export default function transformer(file, api) {
       (ExpressionTypes.indexOf(path.parent.node.type) >= 0 || path.parent.node.type === 'VariableDeclarator')
     )
       return;
-    replaceWithTryCatch(path, j.expressionStatement(path.node));
+    if (path.parent.node.type === 'ReturnStatement') {
+      replaceWithTryCatch(path.parent, path.parent.node);
+    } else {
+      replaceWithTryCatch(path, j.expressionStatement(path.node));
+    }
   });
   root.find(j.VariableDeclaration).forEach(path => {
     const variableDeclarators = path.node.declarations;