inject polyfills in visitor in usage-global

Author: zloirock

package-lock.json

@@ -62,6 +62,7 @@
   },
   "devDependencies": {
     "@babel/core": "^7.28.3",
+    "@babel/plugin-syntax-decorators": "^7.27.1",
     "@babel/plugin-transform-arrow-functions": "^7.27.1",
     "@babel/plugin-transform-block-scoped-functions": "^7.27.1",
     "@babel/plugin-transform-block-scoping": "^7.28.0",

package.json

@@ -37,11 +37,12 @@ function resolveHint(desc, meta) {
 }
 
 module.exports = defineProvider(({
+  babel,
   createMetaResolver,
   debug,
+  getUtils,
   method,
   targets,
-  babel: { types: t },
 }, {
   pkg,
   pkgs,
@@ -51,6 +52,9 @@ module.exports = defineProvider(({
   if (pkg === undefined) pkg = method === 'usage-pure' ? '@core-js/pure' : 'core-js';
   if (typeof pkg != 'string') throw new TypeError('Incorrect package name');
 
+  const t = babel.types;
+  const isWebpack = babel.caller(caller => caller?.name === 'babel-loader');
+
   const packages = pkgs ? [...defaultCoreJSPackages, ...pkgs] : [...defaultCoreJSPackages];
 
   const modulesListForTargetVersion = getModulesListForTargetVersion(version);
@@ -157,5 +161,47 @@ module.exports = defineProvider(({
       return true;
     },
     usagePure() { /* empty */ },
+    visitor: method === 'usage-global' && {
+      // import('foo')
+      CallExpression(path) {
+        if (path.get('callee').isImport()) {
+          const utils = getUtils(path);
+          if (isWebpack) {
+            // Webpack uses `Promise.all` to handle dynamic import.
+            injectCoreJSModulesForEntry(`${ mode }/promise/all`, utils);
+          } else {
+            injectCoreJSModulesForEntry(`${ mode }/promise/constructor`, utils);
+          }
+        }
+      },
+      // (async function () { }).finally(...)
+      Function(path) {
+        if (path.node.async) {
+          injectCoreJSModulesForEntry(`${ mode }/promise/constructor`, getUtils(path));
+        }
+      },
+      // for-of, [a, b] = c
+      'ForOfStatement|ArrayPattern'(path) {
+        injectCoreJSModulesForEntry(`${ mode }/get-iterator`, getUtils(path));
+      },
+      // [...spread]
+      SpreadElement(path) {
+        if (!path.parentPath.isObjectExpression()) {
+          injectCoreJSModulesForEntry(`${ mode }/get-iterator`, getUtils(path));
+        }
+      },
+      // yield *
+      YieldExpression(path) {
+        if (path.node.delegate) {
+          injectCoreJSModulesForEntry(`${ mode }/get-iterator`, getUtils(path));
+        }
+      },
+      // decorators metadata
+      Class(path) {
+        if (path.node.decorators?.length || path.node.body.body.some(el => el.decorators?.length)) {
+          injectCoreJSModulesForEntry(`${ mode }/symbol/metadata`, getUtils(path));
+        }
+      },
+    },
   };
 });

packages/core-js-babel-plugin/index.js

@@ -0,0 +1 @@
+async function f() { /* empty */ }

scripts/bundle-tests/package-lock.json

@@ -0,0 +1,14 @@
+{
+  "plugins": [
+    [
+      "@core-js",
+      {
+        "method": "usage-global",
+        "version": "4.0",
+        "targets": {
+          "ie": 11
+        }
+      }
+    ]
+  ]
+}

tests/babel-plugin/fixtures/usage-global/async-function/input.mjs

@@ -0,0 +1,5 @@
+import "core-js/modules/es.object.to-string";
+import "core-js/modules/es.promise.constructor";
+import "core-js/modules/es.promise.catch";
+import "core-js/modules/es.promise.finally";
+async function f() {/* empty */}

tests/babel-plugin/fixtures/usage-global/async-function/options.json

@@ -0,0 +1,3 @@
+class C {
+  @dec foo = 2;
+}

tests/babel-plugin/fixtures/usage-global/async-function/output.mjs

@@ -0,0 +1,15 @@
+{
+  "plugins": [
+    [
+      "@core-js",
+      {
+        "method": "usage-global",
+        "version": "4.0",
+        "targets": {
+          "ie": 11
+        }
+      }
+    ],
+    ["@babel/plugin-syntax-decorators", { "version": "2023-11" }]
+  ]
+}

tests/babel-plugin/fixtures/usage-global/decorator/input.mjs

@@ -0,0 +1,6 @@
+import "core-js/modules/esnext.function.metadata";
+import "core-js/modules/esnext.symbol.metadata";
+class C {
+  @dec
+  foo = 2;
+}