fix(okam-build): optimize the async await local polyfill for weixin #2

ecomfe · Nov 11, 2018 · f63bbcc · f63bbcc
1 parent 220d942
commit f63bbcc
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/packages/okam-build/lib/processor/js/polyfill/async.js b/packages/okam-build/lib/processor/js/polyfill/async.js
@@ -20,6 +20,32 @@ module.exports = function ({types: t}) {
              */
             AwaitExpression(path, state) {
                 importLocalPolyfill(false, path, state, t);
+            },
+
+            /**
+             * Process `async xxx` expression
+             *
+             * @param {Object} path the node path
+             * @param {Object} state the plugin state
+             */
+            FunctionExpression(path, state) {
+                let node = path.node;
+                if (node.async) {
+                    importLocalPolyfill(false, path, state, t);
+                }
+            },
+
+            /**
+             * Process `async xxx() {}` object member
+             *
+             * @param {Object} path the node path
+             * @param {Object} state the plugin state
+             */
+            ObjectMember(path, state) {
+                let node = path.node;
+                if (node.async) {
+                    importLocalPolyfill(false, path, state, t);
+                }
             }
         }
     };

diff --git a/packages/okam-build/lib/processor/js/polyfill/helper.js b/packages/okam-build/lib/processor/js/polyfill/helper.js
@@ -43,6 +43,7 @@ exports.importLocalPolyfill = function (name, path, state, t) {
     if (name === false) {
         name = importName;
     }
+
     if (!shouldPolyfill(importName, name, path)) {
         return;
     }