123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276 |
- "use strict";
- const fs = require('fs');
- const _ = require('lodash');
- const acorn = require('acorn');
- const walk = require('acorn-walk');
- module.exports = {
- parseBundle
- };
- function parseBundle(bundlePath) {
- const content = fs.readFileSync(bundlePath, 'utf8');
- const ast = acorn.parse(content, {
- sourceType: 'script',
-
-
-
- ecmaVersion: 2050
- });
- const walkState = {
- locations: null,
- expressionStatementDepth: 0
- };
- walk.recursive(ast, walkState, {
- ExpressionStatement(node, state, c) {
- if (state.locations) return;
- state.expressionStatementDepth++;
- if (
- state.expressionStatementDepth === 1 && ast.body.includes(node) && isIIFE(node)) {
- const fn = getIIFECallExpression(node);
- if (
- fn.arguments.length === 0 &&
- fn.callee.params.length === 0) {
-
- const firstVariableDeclaration = fn.callee.body.body.find(node => node.type === 'VariableDeclaration');
- if (firstVariableDeclaration) {
- for (const declaration of firstVariableDeclaration.declarations) {
- if (declaration.init) {
- state.locations = getModulesLocations(declaration.init);
- if (state.locations) {
- break;
- }
- }
- }
- }
- }
- }
- if (!state.locations) {
- c(node.expression, state);
- }
- state.expressionStatementDepth--;
- },
- AssignmentExpression(node, state) {
- if (state.locations) return;
-
- const {
- left,
- right
- } = node;
- if (left && left.object && left.object.name === 'exports' && left.property && left.property.name === 'modules' && isModulesHash(right)) {
- state.locations = getModulesLocations(right);
- }
- },
- CallExpression(node, state, c) {
- if (state.locations) return;
- const args = node.arguments;
-
-
- if (node.callee.type === 'FunctionExpression' && !node.callee.id && args.length === 1 && isSimpleModulesList(args[0])) {
- state.locations = getModulesLocations(args[0]);
- return;
- }
-
-
- if (node.callee.type === 'Identifier' && mayBeAsyncChunkArguments(args) && isModulesList(args[1])) {
- state.locations = getModulesLocations(args[1]);
- return;
- }
-
-
- if (isAsyncChunkPushExpression(node)) {
- state.locations = getModulesLocations(args[0].elements[1]);
- return;
- }
-
-
- if (isAsyncWebWorkerChunkExpression(node)) {
- state.locations = getModulesLocations(args[1]);
- return;
- }
-
- args.forEach(arg => c(arg, state));
- }
- });
- let modules;
- if (walkState.locations) {
- modules = _.mapValues(walkState.locations, loc => content.slice(loc.start, loc.end));
- } else {
- modules = {};
- }
- return {
- modules,
- src: content,
- runtimeSrc: getBundleRuntime(content, walkState.locations)
- };
- }
- function getBundleRuntime(content, modulesLocations) {
- const sortedLocations = Object.values(modulesLocations || {}).sort((a, b) => a.start - b.start);
- let result = '';
- let lastIndex = 0;
- for (const {
- start,
- end
- } of sortedLocations) {
- result += content.slice(lastIndex, start);
- lastIndex = end;
- }
- return result + content.slice(lastIndex, content.length);
- }
- function isIIFE(node) {
- return node.type === 'ExpressionStatement' && (node.expression.type === 'CallExpression' || node.expression.type === 'UnaryExpression' && node.expression.argument.type === 'CallExpression');
- }
- function getIIFECallExpression(node) {
- if (node.expression.type === 'UnaryExpression') {
- return node.expression.argument;
- } else {
- return node.expression;
- }
- }
- function isModulesList(node) {
- return isSimpleModulesList(node) ||
- isOptimizedModulesArray(node);
- }
- function isSimpleModulesList(node) {
- return (
- isModulesHash(node) ||
- isModulesArray(node)
- );
- }
- function isModulesHash(node) {
- return node.type === 'ObjectExpression' && node.properties.map(node => node.value).every(isModuleWrapper);
- }
- function isModulesArray(node) {
- return node.type === 'ArrayExpression' && node.elements.every(elem =>
- !elem || isModuleWrapper(elem));
- }
- function isOptimizedModulesArray(node) {
-
-
-
- return node.type === 'CallExpression' && node.callee.type === 'MemberExpression' &&
- node.callee.object.type === 'CallExpression' && node.callee.object.callee.type === 'Identifier' && node.callee.object.callee.name === 'Array' && node.callee.object.arguments.length === 1 && isNumericId(node.callee.object.arguments[0]) &&
- node.callee.property.type === 'Identifier' && node.callee.property.name === 'concat' &&
- node.arguments.length === 1 && isModulesArray(node.arguments[0]);
- }
- function isModuleWrapper(node) {
- return (
- (node.type === 'FunctionExpression' || node.type === 'ArrowFunctionExpression') && !node.id ||
- isModuleId(node) ||
- node.type === 'ArrayExpression' && node.elements.length > 1 && isModuleId(node.elements[0])
- );
- }
- function isModuleId(node) {
- return node.type === 'Literal' && (isNumericId(node) || typeof node.value === 'string');
- }
- function isNumericId(node) {
- return node.type === 'Literal' && Number.isInteger(node.value) && node.value >= 0;
- }
- function isChunkIds(node) {
-
- return node.type === 'ArrayExpression' && node.elements.every(isModuleId);
- }
- function isAsyncChunkPushExpression(node) {
- const {
- callee,
- arguments: args
- } = node;
- return callee.type === 'MemberExpression' && callee.property.name === 'push' && callee.object.type === 'AssignmentExpression' && args.length === 1 && args[0].type === 'ArrayExpression' && mayBeAsyncChunkArguments(args[0].elements) && isModulesList(args[0].elements[1]);
- }
- function mayBeAsyncChunkArguments(args) {
- return args.length >= 2 && isChunkIds(args[0]);
- }
- function isAsyncWebWorkerChunkExpression(node) {
- const {
- callee,
- type,
- arguments: args
- } = node;
- return type === 'CallExpression' && callee.type === 'MemberExpression' && args.length === 2 && isChunkIds(args[0]) && isModulesList(args[1]);
- }
- function getModulesLocations(node) {
- if (node.type === 'ObjectExpression') {
-
- const modulesNodes = node.properties;
- return modulesNodes.reduce((result, moduleNode) => {
- const moduleId = moduleNode.key.name || moduleNode.key.value;
- result[moduleId] = getModuleLocation(moduleNode.value);
- return result;
- }, {});
- }
- const isOptimizedArray = node.type === 'CallExpression';
- if (node.type === 'ArrayExpression' || isOptimizedArray) {
-
- const minId = isOptimizedArray ?
- node.callee.object.arguments[0].value :
- 0;
- const modulesNodes = isOptimizedArray ?
- node.arguments[0].elements : node.elements;
- return modulesNodes.reduce((result, moduleNode, i) => {
- if (moduleNode) {
- result[i + minId] = getModuleLocation(moduleNode);
- }
- return result;
- }, {});
- }
- return {};
- }
- function getModuleLocation(node) {
- return {
- start: node.start,
- end: node.end
- };
- }
|