2018-10-26 17:58:21 +00:00
|
|
|
// Add polyfill imports to the first file encountered.
|
2021-02-23 08:34:39 +00:00
|
|
|
const { addSideEffect } = require('@babel/helper-module-imports')
|
|
|
|
|
|
|
|
const modulePathMap = {
|
|
|
|
'regenerator-runtime': 'regenerator-runtime/runtime.js'
|
|
|
|
}
|
|
|
|
|
|
|
|
function getModulePath (mod) {
|
|
|
|
return modulePathMap[mod] || 'core-js/modules/' + mod + '.js'
|
|
|
|
}
|
|
|
|
|
|
|
|
function createImport (path, mod) {
|
|
|
|
return addSideEffect(path, getModulePath(mod))
|
|
|
|
}
|
|
|
|
|
2018-10-26 17:58:21 +00:00
|
|
|
module.exports = ({ types }) => {
|
|
|
|
let entryFile
|
|
|
|
return {
|
|
|
|
name: 'inject-polyfills',
|
|
|
|
visitor: {
|
2019-07-10 10:45:49 +00:00
|
|
|
Program (path, state) {
|
2018-10-26 17:58:21 +00:00
|
|
|
if (!entryFile) {
|
|
|
|
entryFile = state.filename
|
|
|
|
} else if (state.filename !== entryFile) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
const { polyfills } = state.opts
|
|
|
|
|
|
|
|
// Imports are injected in reverse order
|
|
|
|
polyfills.slice().reverse().forEach((p) => {
|
|
|
|
createImport(path, p)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|