fix: handle existing parens

Author: sxzz

src/core/index.ts

@@ -24,7 +24,9 @@ export function transformQuansync(
   id: string,
 ): CodeTransform | undefined {
   const lang = getLang(id)
-  const program = babelParse(code, lang)
+  const program = babelParse(code, lang, {
+    createParenthesizedExpressions: true,
+  })
   const imports: Record<string, ImportBinding> = Object.create(null)
 
   for (const node of program.body) {

tests/__snapshots__/transform.test.ts.snap

@@ -154,7 +154,7 @@ export const echo = quansync({
 
 const echoNewLine = quansync(
   /** @param {string|Promise<string>} v */ function* (v) {
-return ((yield echo(yield v))) + '\\n'
+return (yield echo(yield v)) + '\\n'
 },
 )
 
@@ -169,3 +169,13 @@ export default async () => {
 }
 "
 `;
+
+exports[`transform > ./fixtures/parens.js 1`] = `
+"import { quansync } from 'quansync/macro'
+
+const parens = quansync(function* (obj) {
+  const config = yield (obj + 1)
+  return config
+})
+"
+`;

tests/fixtures/parens.js

@@ -0,0 +1,6 @@
+import { quansync } from 'quansync/macro'
+
+const parens = quansync(async (obj) => {
+  const config = await (obj + 1)
+  return config
+})

tests/transform.test.ts

@@ -21,7 +21,7 @@ describe('transform', async () => {
       const filePath = path.resolve(dirname, `temp/${path.basename(id)}`)
       await writeFile(filePath, result)
       const mod = await import(`${pathToFileURL(filePath).href}?${Date.now()}`)
-      await mod.default()
+      await mod.default?.()
 
       return result
     },