'a'

mh-two-thousand-and-two

2024-04-12 44d2c92345cd156a59fc327b3060292a282d2893

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
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;