77 lines
3.0 KiB
JavaScript
77 lines
3.0 KiB
JavaScript
|
const fs = require('fs-extra');
|
|||
|
|
const { generateAsset } = require('./utils');
|
|||
|
|
const t = require('@babel/types');
|
|||
|
|
const babelGenerator = require('@babel/generator');
|
|||
|
|
const babelParser = require('@babel/parser');
|
|||
|
|
const { default: babelTraverse } = require('@babel/traverse');
|
|||
|
|
const getWxMpRuntime = require('../runtime/index');
|
|||
|
|
|
|||
|
|
// TODO 这个工作应该放在loader中做，后续优化
|
|||
|
|
class AddWxMpRuntimePlugin {
|
|||
|
|
apply (compiler) {
|
|||
|
|
compiler.hooks.emit.tapPromise('AddWxMpRuntimePlugin', compilation => {
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
try {
|
|||
|
|
// 收集独立分包路径下面的所有js文件
|
|||
|
|
// js文件都存储在 compilation.assets中 ， 因为需要注入 require(`${pkgRoot/common/index.js}`)
|
|||
|
|
|
|||
|
|
const thisCompilationAssets = compilation.assets;
|
|||
|
|
const independentPkgsInfo = Object.values(process.UNI_SUBPACKAGES).filter(info => info.independent) || [];
|
|||
|
|
const independentPkgRoots = independentPkgsInfo.map(info => `${info.root}`);
|
|||
|
|
if (!independentPkgRoots.length) {
|
|||
|
|
resolve();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
const mpRuntimePath = getWxMpRuntime();
|
|||
|
|
const wxMpRuntimeContent = fs.readFileSync(mpRuntimePath, 'utf8');
|
|||
|
|
|
|||
|
|
const ast = babelParser.parse(wxMpRuntimeContent, {
|
|||
|
|
sourceType: 'module',
|
|||
|
|
plugins: ['classProperties'],
|
|||
|
|
});
|
|||
|
|
babelTraverse(ast, {
|
|||
|
|
VariableDeclaration (path) {
|
|||
|
|
const v = path.node.declarations[0];
|
|||
|
|
const name = v.id.name;
|
|||
|
|
const value = v.init;
|
|||
|
|
if (name === 'independentRoots' && t.isArrayExpression(value)) {
|
|||
|
|
value.elements = independentPkgRoots.map(pkgRoot => t.StringLiteral(pkgRoot));
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
const { code } = babelGenerator.default(ast);
|
|||
|
|
const runtimeAssetsInfo = generateAsset(code);
|
|||
|
|
const mpRuntimeRelativePath = 'common/wxMpRuntime.js';
|
|||
|
|
|
|||
|
|
// 处理app.js
|
|||
|
|
const appJsName = 'app.js';
|
|||
|
|
thisCompilationAssets[mpRuntimeRelativePath] = runtimeAssetsInfo;
|
|||
|
|
const assetInfo = thisCompilationAssets[appJsName];
|
|||
|
|
if (!assetInfo) {
|
|||
|
|
return resolve();
|
|||
|
|
}
|
|||
|
|
const content = assetInfo.source();
|
|||
|
|
thisCompilationAssets[appJsName] = generateAsset(`require('./common/wxMpRuntime.js');${content}`);
|
|||
|
|
|
|||
|
|
// 处理独立分包
|
|||
|
|
independentPkgRoots.forEach(pkgRoot => {
|
|||
|
|
thisCompilationAssets[`${pkgRoot}/${mpRuntimeRelativePath}`] = runtimeAssetsInfo;
|
|||
|
|
const entryJsName = `${pkgRoot}/common/index.js`;
|
|||
|
|
const assetInfo = thisCompilationAssets[entryJsName];
|
|||
|
|
let content = assetInfo.source();
|
|||
|
|
content = generateAsset(`require('./wxMpRuntime.js');${content}`);
|
|||
|
|
thisCompilationAssets[entryJsName] = content;
|
|||
|
|
});
|
|||
|
|
resolve();
|
|||
|
|
} catch (e) {
|
|||
|
|
console.error('AddWxMpRuntimePlugin', e);
|
|||
|
|
reject(e);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = AddWxMpRuntimePlugin;
|