Allow exported generator functions (#290)

Fixes #289
alangpierce · Jul 2, 2018 · a1685b2 · a1685b2
1 parent d31351a
commit a1685b2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/transformers/CJSImportTransformer.ts b/src/transformers/CJSImportTransformer.ts
@@ -230,6 +230,7 @@ export default class CJSImportTransformer extends Transformer {
       return true;
     } else if (
       this.tokens.matches2(tt._export, tt._function) ||
+      // export async function
       this.tokens.matches3(tt._export, tt.name, tt._function)
     ) {
       this.processExportFunction();
@@ -357,6 +358,9 @@ export default class CJSImportTransformer extends Transformer {
       this.tokens.copyToken();
       this.tokens.copyToken();
     }
+    if (this.tokens.matches1(tt.star)) {
+      this.tokens.copyToken();
+    }
     if (!this.tokens.matches1(tt.name)) {
       throw new Error("Expected identifier for exported function name.");
     }

diff --git a/test/imports-test.ts b/test/imports-test.ts
@@ -988,4 +988,21 @@ module.exports = exports.default;
     `,
     );
   });
+
+  it("allows exported generator functions", () => {
+    assertResult(
+      `
+      export function* foo() {}
+      export async function* bar() {}
+      export default function* baz() {}
+      export default async function* qux() {}
+    `,
+      `"use strict";${ESMODULE_PREFIX}
+       function* foo() {} exports.foo = foo;
+       async function* bar() {} exports.bar = bar;
+      exports. default = function* baz() {}
+      exports. default = async function* qux() {}
+    `,
+    );
+  });
 });