'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
const t = require('@babel/types');
const babelTraverse = require('@babel/traverse').default;
const babelParser = require('@babel/parser');
const babelGenerator = require('@babel/generator');
const { generateAsset } = require('./utils');
const { collectIndependentJsAssets } = require('./optimize-components-position/util');
 
const visitor = {
  CallExpression (path) {
    const funNode = path.node;
    // https://developers.weixin.qq.com/miniprogram/dev/reference/api/getApp.html
    // FIX: 目前getApp仅支持一个参数(allowDefault)，如果后面增加更多的参数以下逻辑需要修改
    // 增减判断是否有该参数逻辑
 
    if (t.isIdentifier(path.node.callee)) {
      const logicGlobal = '(Function("return this")())'
      if (funNode.callee.name === 'getApp' && funNode.arguments.length === 0) {
        funNode.callee = t.MemberExpression(t.Identifier(logicGlobal), t.Identifier('getApp'));
      } else if (funNode.callee.name === 'App') {
        funNode.callee = t.MemberExpression(t.Identifier(logicGlobal), t.Identifier('App'));
      }
    }
  },
};
 
// 关键：需要在在整个emit阶段的最后（compilation.assets['/pages/chat-im/wxcomponents/...']
class AppInterceptorPlugin {
  apply (compiler) {
    compiler.hooks.emit.tapPromise('AppInterceptorPlugin', compilation => {
      return new Promise(resolve => {
        // 收集独立分包路径下面的所有js文件
        // js文件都存储在 compilation.assets中 ， 因为需要注入 require(`${pkgRoot/common/index.js}`)
        const thisCompilationAssets = compilation.assets;
        const independentJsAssets = collectIndependentJsAssets(thisCompilationAssets);
        independentJsAssets.forEach(({ jsAssets }) => {
          jsAssets.forEach(jsAssetName => {
              if (jsAssetName.endsWith('common/wxMpRuntime.js')) {
                  return;
              }
              const assetInfo = thisCompilationAssets[jsAssetName];
              let assetSource = assetInfo.source();
 
              // 有部分js文件在这里是Buffer类型
              if (assetSource instanceof Buffer) {
                assetSource = assetSource.toString();
              }
 
              const ast = babelParser.parse(assetSource, {
                sourceType: 'module',
                plugins: ['classProperties'],
              });
              babelTraverse(ast, visitor);
              const { code } = babelGenerator.default(ast);
              thisCompilationAssets[jsAssetName] = generateAsset(code);
            }
          );
        });
 
        resolve();
      });
    });
  }
}
 
module.exports = AppInterceptorPlugin;