218 lines
No EOL
6.5 KiB
JavaScript
Executable file
218 lines
No EOL
6.5 KiB
JavaScript
Executable file
'use strict';
|
|
|
|
var _slicedToArray = function () { function sliceIterator(arr, i) { var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"]) _i["return"](); } finally { if (_d) throw _e; } } return _arr; } return function (arr, i) { if (Array.isArray(arr)) { return arr; } else if (Symbol.iterator in Object(arr)) { return sliceIterator(arr, i); } else { throw new TypeError("Invalid attempt to destructure non-iterable instance"); } }; }();
|
|
|
|
const babylon = require('babylon');
|
|
const t = require('babel-types');
|
|
const traverse = require('babel-traverse').default;
|
|
const generate = require('babel-generator').default;
|
|
|
|
const EXPORTS_RE = /^\$([\d]+)\$exports$/;
|
|
const EXPORT_RE = /^\$([\d]+)\$export\$(.+)$/;
|
|
|
|
// TODO: minify
|
|
// TODO: source-map
|
|
|
|
module.exports = (code, exports, moduleMap, wildcards) => {
|
|
let ast = babylon.parse(code);
|
|
let addedExports = new Set();
|
|
let commentedBindings = new Set();
|
|
|
|
let resolveModule = (id, name) => {
|
|
let module = moduleMap.get(id);
|
|
return module.depAssets.get(module.dependencies.get(name)).id;
|
|
};
|
|
|
|
function replaceExportNode(id, name, path) {
|
|
path = getOuterStatement(path);
|
|
|
|
let node = tail(id, id => `$${id}$export$${name}`);
|
|
|
|
if (!node) {
|
|
// if there is no named export then lookup for a CommonJS export
|
|
node = tail(id, id => `$${id}$exports`) || t.identifier(`$${id}$exports`);
|
|
|
|
// if there is a CommonJS export return $id$exports.name
|
|
if (node) {
|
|
return t.memberExpression(node, t.identifier(name));
|
|
}
|
|
}
|
|
|
|
return node;
|
|
|
|
function tail(id, symbol) {
|
|
let computedSymbol = symbol(id);
|
|
|
|
// if the symbol is in the scope there is not need to remap it
|
|
if (path.scope.hasBinding(computedSymbol)) {
|
|
return t.identifier(computedSymbol);
|
|
}
|
|
|
|
if (exports.has(computedSymbol)) {
|
|
return t.identifier(exports.get(computedSymbol));
|
|
}
|
|
|
|
// if there is a wildcard for the module
|
|
// default exports are excluded from wildcard exports
|
|
if (wildcards.has(id) && name !== 'default') {
|
|
/* recursively lookup the symbol
|
|
* this is needed when there is deep export wildcards, like in the following:
|
|
* - a.js
|
|
* > export * from './b'
|
|
* - b.js
|
|
* > export * from './c'
|
|
* - c.js in es6
|
|
* > export * from 'lodash'
|
|
* - c.js in cjs
|
|
* > module.exports = require('lodash')
|
|
*/
|
|
let node = null;
|
|
|
|
wildcards.get(id).find(name => node = tail(resolveModule(id, name), symbol));
|
|
|
|
return node;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|
|
|
|
traverse(ast, {
|
|
CallExpression(path) {
|
|
var _path$node = path.node;
|
|
let args = _path$node.arguments,
|
|
callee = _path$node.callee;
|
|
|
|
// each require('module') call gets replaced with $parcel$require(id, 'module')
|
|
|
|
if (t.isIdentifier(callee, { name: '$parcel$require' })) {
|
|
var _args = _slicedToArray(args, 2);
|
|
|
|
let id = _args[0],
|
|
name = _args[1];
|
|
|
|
|
|
if (args.length !== 2 || !t.isNumericLiteral(id) || !t.isStringLiteral(name)) {
|
|
throw new Error('invariant: invalid signature, expected : $parcel$require(number, string)');
|
|
}
|
|
|
|
const mod = resolveModule(id.value, name.value);
|
|
|
|
if (typeof mod === 'undefined') {
|
|
throw new Error(`Cannot find module "${name.value}"`);
|
|
}
|
|
|
|
path.replaceWith(t.identifier(`$${mod}$exports`));
|
|
}
|
|
},
|
|
MemberExpression(path) {
|
|
if (!path.isReferenced()) {
|
|
return;
|
|
}
|
|
|
|
var _path$node2 = path.node;
|
|
let object = _path$node2.object,
|
|
property = _path$node2.property;
|
|
|
|
if (!t.isIdentifier(object) || !t.isIdentifier(property)) {
|
|
return;
|
|
}
|
|
|
|
let match = object.name.match(EXPORTS_RE);
|
|
if (match) {
|
|
let exportName = '$' + match[1] + '$export$' + property.name;
|
|
if (path.scope.hasBinding(exportName)) {
|
|
path.replaceWith(t.identifier(exportName));
|
|
}
|
|
}
|
|
},
|
|
Identifier(path) {
|
|
let name = path.node.name;
|
|
|
|
|
|
if (typeof name !== 'string') {
|
|
return;
|
|
}
|
|
|
|
let match = name.match(EXPORTS_RE);
|
|
|
|
if (match) {
|
|
if (!path.scope.hasBinding(name) && !addedExports.has(name)) {
|
|
let moduleExports = moduleMap.get(+match[1]).cacheData.exports;
|
|
|
|
addedExports.add(name);
|
|
|
|
getOuterStatement(path).insertBefore(t.variableDeclaration('var', [t.variableDeclarator(t.identifier(name), t.objectExpression(Object.keys(moduleExports).map(key => {
|
|
let binding = path.scope.getBinding(exports.get(key) || key);
|
|
if (!binding) {
|
|
return null;
|
|
}
|
|
|
|
let exportName = key.match(EXPORT_RE)[2];
|
|
let expr = replaceExportNode(+match[1], exportName, path);
|
|
|
|
if (expr === null) {
|
|
return null;
|
|
}
|
|
|
|
if (binding.constant) {
|
|
return t.objectProperty(t.identifier(exportName), expr);
|
|
} else {
|
|
if (!commentedBindings.has(binding)) {
|
|
commentedBindings.add(binding);
|
|
binding.constantViolations.forEach(path => path.getFunctionParent().addComment('leading', ` bailout: mutates ${generate(expr).code}`, '\n'));
|
|
}
|
|
|
|
return t.objectMethod('get', t.identifier(exportName), [], t.blockStatement([t.returnStatement(expr)]));
|
|
}
|
|
}).filter(property => property !== null)))]));
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
match = name.match(EXPORT_RE);
|
|
|
|
if (match && !path.scope.hasBinding(name) && !addedExports.has(name)) {
|
|
let id = Number(match[1]);
|
|
let exportName = match[2];
|
|
let node = replaceExportNode(id, exportName, path);
|
|
|
|
addedExports.add(name);
|
|
|
|
if (node) {
|
|
path.replaceWith(node);
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
return generate(ast, code).code;
|
|
};
|
|
|
|
function getOuterStatement(path) {
|
|
if (validate(path)) {
|
|
return path;
|
|
}
|
|
|
|
return path.findParent(validate);
|
|
|
|
function validate(path) {
|
|
if (!t.isStatement(path.node) || t.isBlockStatement(path.node)) {
|
|
return false;
|
|
}
|
|
|
|
// TODO: use scope?
|
|
let outerBlocks = 0;
|
|
|
|
path.findParent(parent => {
|
|
if (t.isBlockStatement(parent.node)) {
|
|
outerBlocks++;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
|
|
return outerBlocks === 1;
|
|
}
|
|
} |