init

马宇豪

2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18

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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const RequireEnsureDependenciesBlock = require("./RequireEnsureDependenciesBlock");
const RequireEnsureDependency = require("./RequireEnsureDependency");
const RequireEnsureItemDependency = require("./RequireEnsureItemDependency");
const getFunctionExpression = require("./getFunctionExpression");
 
module.exports = class RequireEnsureDependenciesBlockParserPlugin {
    apply(parser) {
        parser.hooks.call
            .for("require.ensure")
            .tap("RequireEnsureDependenciesBlockParserPlugin", expr => {
                let chunkName = null;
                let errorExpressionArg = null;
                let errorExpression = null;
                switch (expr.arguments.length) {
                    case 4: {
                        const chunkNameExpr = parser.evaluateExpression(expr.arguments[3]);
                        if (!chunkNameExpr.isString()) return;
                        chunkName = chunkNameExpr.string;
                    }
                    // falls through
                    case 3: {
                        errorExpressionArg = expr.arguments[2];
                        errorExpression = getFunctionExpression(errorExpressionArg);
 
                        if (!errorExpression && !chunkName) {
                            const chunkNameExpr = parser.evaluateExpression(
                                expr.arguments[2]
                            );
                            if (!chunkNameExpr.isString()) return;
                            chunkName = chunkNameExpr.string;
                        }
                    }
                    // falls through
                    case 2: {
                        const dependenciesExpr = parser.evaluateExpression(
                            expr.arguments[0]
                        );
                        const dependenciesItems = dependenciesExpr.isArray()
                            ? dependenciesExpr.items
                            : [dependenciesExpr];
                        const successExpressionArg = expr.arguments[1];
                        const successExpression =
                            getFunctionExpression(successExpressionArg);
 
                        if (successExpression) {
                            parser.walkExpressions(successExpression.expressions);
                        }
                        if (errorExpression) {
                            parser.walkExpressions(errorExpression.expressions);
                        }
 
                        const depBlock = new RequireEnsureDependenciesBlock(
                            chunkName,
                            expr.loc
                        );
                        const errorCallbackExists =
                            expr.arguments.length === 4 ||
                            (!chunkName && expr.arguments.length === 3);
                        const dep = new RequireEnsureDependency(
                            expr.range,
                            expr.arguments[1].range,
                            errorCallbackExists && expr.arguments[2].range
                        );
                        dep.loc = expr.loc;
                        depBlock.addDependency(dep);
                        const old = parser.state.current;
                        parser.state.current = depBlock;
                        try {
                            let failed = false;
                            parser.inScope([], () => {
                                for (const ee of dependenciesItems) {
                                    if (ee.isString()) {
                                        const ensureDependency = new RequireEnsureItemDependency(
                                            ee.string
                                        );
                                        ensureDependency.loc = ee.loc || expr.loc;
                                        depBlock.addDependency(ensureDependency);
                                    } else {
                                        failed = true;
                                    }
                                }
                            });
                            if (failed) {
                                return;
                            }
                            if (successExpression) {
                                if (successExpression.fn.body.type === "BlockStatement") {
                                    parser.walkStatement(successExpression.fn.body);
                                } else {
                                    parser.walkExpression(successExpression.fn.body);
                                }
                            }
                            old.addBlock(depBlock);
                        } finally {
                            parser.state.current = old;
                        }
                        if (!successExpression) {
                            parser.walkExpression(successExpressionArg);
                        }
                        if (errorExpression) {
                            if (errorExpression.fn.body.type === "BlockStatement") {
                                parser.walkStatement(errorExpression.fn.body);
                            } else {
                                parser.walkExpression(errorExpression.fn.body);
                            }
                        } else if (errorExpressionArg) {
                            parser.walkExpression(errorExpressionArg);
                        }
                        return true;
                    }
                }
            });
    }
};