feat!: remove toGenerator helper

Author: sxzz

README.md

@@ -138,13 +138,13 @@ For more details on usage, refer to [quansync's docs](https://github.com/antfu-c
 ## How it works
 
 `unplugin-quansync` transforms your async functions into generator functions with `quansyncMacro`,
-and transforms `await` into `yield * toGenerator(...)`.
+and transforms `await` into `yield`.
 
 The example above is transformed into:
 
 ```ts
 import fs from 'node:fs'
-import { quansyncMacro, toGenerator } from 'quansync'
+import { quansyncMacro } from 'quansync'
 
 // No transformations needed for objects
 const readFile = quansyncMacro({
@@ -154,15 +154,17 @@ const readFile = quansyncMacro({
 
 // `async function` is transformed into a generator function
 const myFunction = quansyncMacro(function* (filename) {
-  // `await` is transformed into `yield * toGenerator(...)`
-  const code = yield* toGenerator(readFile(filename, 'utf8'))
+  // `await` is transformed into `yield ...`
+  const code = yield readFile(filename, 'utf8')
 
   return `// some custom prefix\n${code}`
 })
 ```
 
 ## Caveats
 
+### Arrow functions
+
 Both arrow functions and generators have been available since ES2015,
 but a "generator arrow function" syntax does not exist
 [yet](https://github.com/tc39/proposal-generator-arrow-functions).

package.json

@@ -70,7 +70,7 @@
     "prepublishOnly": "pnpm run build"
   },
   "peerDependencies": {
-    "quansync": ">=0.0.5"
+    "quansync": ">=0.1.0"
   },
   "dependencies": {
     "ast-kit": "^1.4.0",
@@ -88,7 +88,7 @@
     "eslint": "^9.21.0",
     "oxc-transform": "^0.52.0",
     "prettier": "^3.5.2",
-    "quansync": "^0.0.5",
+    "quansync": "^0.1.0",
     "tsdown": "^0.6.0",
     "tsx": "^4.19.3",
     "typescript": "^5.7.3",

pnpm-lock.yaml

@@ -39,23 +39,10 @@ export function transformQuansync(
   const s = new MagicString(code)
   const functionScopes: boolean[] = []
 
-  let inject = false
-  function injectHelper() {
-    if (inject) return
-    inject = true
-    s.prepend(`import { toGenerator as _QUANSYNC_TO_GEN } from 'quansync'\n`)
-  }
-
   walkAST<t.Node>(program, {
     enter(node, parent) {
       if (node.type === 'AwaitExpression' && functionScopes.at(-1)) {
-        injectHelper()
-        s.overwrite(
-          node.start!,
-          node.argument.start!,
-          'yield * _QUANSYNC_TO_GEN(',
-        )
-        s.appendLeft(node.end!, ')')
+        s.overwrite(node.start!, node.argument.start!, 'yield ')
         return
       }
 

src/core/index.ts

@@ -1,8 +1,7 @@
 // Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
 exports[`transform > ./fixtures/alias.js 1`] = `
-"import { toGenerator as _QUANSYNC_TO_GEN } from 'quansync'
-// @ts-check
+"// @ts-check
 import { quansync, quansyncMacro as x } from 'quansync'
 import { expect } from 'vitest'
 
@@ -13,7 +12,7 @@ export const echo = quansync({
 
 const echoNewLine = x(
   /** @param {string|Promise<string>} v */ function* (v) {
-    const contents = yield * _QUANSYNC_TO_GEN(echo(yield * _QUANSYNC_TO_GEN(v)))
+    const contents = yield echo(yield v)
     return contents + '\\n'
   },
 )
@@ -31,8 +30,7 @@ export default async () => {
 `;
 
 exports[`transform > ./fixtures/basic.js 1`] = `
-"import { toGenerator as _QUANSYNC_TO_GEN } from 'quansync'
-// @ts-check
+"// @ts-check
 import { quansyncMacro } from 'quansync'
 import { expect } from 'vitest'
 
@@ -42,43 +40,43 @@ export const getNumber = quansyncMacro({
 })
 
 const inc1 = quansyncMacro(function* (){
-  const value = yield * _QUANSYNC_TO_GEN(getNumber(1))
+  const value = yield getNumber(1)
   return value + 1
 })
 
 const inc2 = quansyncMacro(function* (id) {
-  const value = yield * _QUANSYNC_TO_GEN(getNumber(1))
+  const value = yield getNumber(1)
   return value + 1
 })
 
 const inc3 = quansyncMacro(function* named(){
-  const value = yield * _QUANSYNC_TO_GEN(getNumber(1))
+  const value = yield getNumber(1)
   return value + 1
 })
 
 const inc4 = quansyncMacro(function* named(id) {
-  const value = yield * _QUANSYNC_TO_GEN(getNumber(1))
+  const value = yield getNumber(1)
   return value + 1
 })
 
 const inc5 = quansyncMacro(() => {
 return function* () {
-  const value = yield * _QUANSYNC_TO_GEN(getNumber(1))
+  const value = yield getNumber(1)
   return value + 1
 }.call(this)
 })
 
 const inc6 = quansyncMacro((id) => {
 return function* () {
-  const value = yield * _QUANSYNC_TO_GEN(getNumber(1))
+  const value = yield getNumber(1)
   return value + 1
 }.call(this)
 })
 
 export const fn7 = quansyncMacro(() => {
 return function* () {
-  yield * _QUANSYNC_TO_GEN(1)
-  yield * _QUANSYNC_TO_GEN(2)
+  yield 1
+  yield 2
   const fn = async () => {
     await 3
   }
@@ -108,8 +106,7 @@ export default async () => {
 `;
 
 exports[`transform > ./fixtures/inlined-arrow.js 1`] = `
-"import { toGenerator as _QUANSYNC_TO_GEN } from 'quansync'
-// @ts-check
+"// @ts-check
 import { quansyncMacro } from 'quansync'
 import { expect } from 'vitest'
 
@@ -122,7 +119,7 @@ const echoNewLine = quansyncMacro(
   /** @param {string|Promise<string>} v */ (v) =>
     {
 return function* () {
-return (yield * _QUANSYNC_TO_GEN(echo(yield * _QUANSYNC_TO_GEN(v)))) + '\\n'
+return (yield echo(yield v)) + '\\n'
 }.call(this)
 },
 )