106 lines
4.1 KiB
JavaScript
106 lines
4.1 KiB
JavaScript
"use strict";
|
|
|
|
exports.__esModule = true;
|
|
exports.default = void 0;
|
|
var _babel = _interopRequireWildcard(require("@babel/core"));
|
|
function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function (e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, default: e }; if (null === e || "object" != typeof e && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (const t in e) "default" !== t && {}.hasOwnProperty.call(e, t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, t)) && (i.get || i.set) ? o(f, t, i) : f[t] = e[t]); return f; })(e, t); }
|
|
const {
|
|
types: t
|
|
} = _babel.default || _babel;
|
|
class ImportsCachedInjector {
|
|
constructor(resolver, getPreferredIndex) {
|
|
this._imports = new WeakMap();
|
|
this._anonymousImports = new WeakMap();
|
|
this._lastImports = new WeakMap();
|
|
this._resolver = resolver;
|
|
this._getPreferredIndex = getPreferredIndex;
|
|
}
|
|
storeAnonymous(programPath, url, moduleName, getVal) {
|
|
const key = this._normalizeKey(programPath, url);
|
|
const imports = this._ensure(this._anonymousImports, programPath, Set);
|
|
if (imports.has(key)) return;
|
|
const node = getVal(programPath.node.sourceType === "script", t.stringLiteral(this._resolver(url)));
|
|
imports.add(key);
|
|
this._injectImport(programPath, node, moduleName);
|
|
}
|
|
storeNamed(programPath, url, name, moduleName, getVal) {
|
|
const key = this._normalizeKey(programPath, url, name);
|
|
const imports = this._ensure(this._imports, programPath, Map);
|
|
if (!imports.has(key)) {
|
|
const {
|
|
node,
|
|
name: id
|
|
} = getVal(programPath.node.sourceType === "script", t.stringLiteral(this._resolver(url)), t.identifier(name));
|
|
imports.set(key, id);
|
|
this._injectImport(programPath, node, moduleName);
|
|
}
|
|
return t.identifier(imports.get(key));
|
|
}
|
|
_injectImport(programPath, node, moduleName) {
|
|
var _this$_lastImports$ge;
|
|
const newIndex = this._getPreferredIndex(moduleName);
|
|
const lastImports = (_this$_lastImports$ge = this._lastImports.get(programPath)) != null ? _this$_lastImports$ge : [];
|
|
const isPathStillValid = path => path.node &&
|
|
// Sometimes the AST is modified and the "last import"
|
|
// we have has been replaced
|
|
path.parent === programPath.node && path.container === programPath.node.body;
|
|
let last;
|
|
if (newIndex === Infinity) {
|
|
// Fast path: we can always just insert at the end if newIndex is `Infinity`
|
|
if (lastImports.length > 0) {
|
|
last = lastImports[lastImports.length - 1].path;
|
|
if (!isPathStillValid(last)) last = undefined;
|
|
}
|
|
} else {
|
|
for (const [i, data] of lastImports.entries()) {
|
|
const {
|
|
path,
|
|
index
|
|
} = data;
|
|
if (isPathStillValid(path)) {
|
|
if (newIndex < index) {
|
|
const [newPath] = path.insertBefore(node);
|
|
lastImports.splice(i, 0, {
|
|
path: newPath,
|
|
index: newIndex
|
|
});
|
|
return;
|
|
}
|
|
last = path;
|
|
}
|
|
}
|
|
}
|
|
if (last) {
|
|
const [newPath] = last.insertAfter(node);
|
|
lastImports.push({
|
|
path: newPath,
|
|
index: newIndex
|
|
});
|
|
} else {
|
|
const [newPath] = programPath.unshiftContainer("body", [node]);
|
|
this._lastImports.set(programPath, [{
|
|
path: newPath,
|
|
index: newIndex
|
|
}]);
|
|
}
|
|
}
|
|
_ensure(map, programPath, Collection) {
|
|
let collection = map.get(programPath);
|
|
if (!collection) {
|
|
collection = new Collection();
|
|
map.set(programPath, collection);
|
|
}
|
|
return collection;
|
|
}
|
|
_normalizeKey(programPath, url, name = "") {
|
|
const {
|
|
sourceType
|
|
} = programPath.node;
|
|
|
|
// If we rely on the imported binding (the "name" parameter), we also need to cache
|
|
// based on the sourceType. This is because the module transforms change the names
|
|
// of the import variables.
|
|
return `${name && sourceType}::${url}::${name}`;
|
|
}
|
|
}
|
|
exports.default = ImportsCachedInjector; |