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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const { JAVASCRIPT_MODULE_TYPE_ESM } = require("../ModuleTypeConstants");
const DynamicExports = require("./DynamicExports");
const HarmonyCompatibilityDependency = require("./HarmonyCompatibilityDependency");
const HarmonyExports = require("./HarmonyExports");
 
/** @typedef {import("../javascript/JavascriptParser")} JavascriptParser */
/** @typedef {import("./HarmonyModulesPlugin").HarmonyModulesPluginOptions} HarmonyModulesPluginOptions */
 
module.exports = class HarmonyDetectionParserPlugin {
    /**
     * @param {HarmonyModulesPluginOptions} options options
     */
    constructor(options) {
        const { topLevelAwait = false } = options || {};
        this.topLevelAwait = topLevelAwait;
    }
 
    /**
     * @param {JavascriptParser} parser the parser
     * @returns {void}
     */
    apply(parser) {
        parser.hooks.program.tap("HarmonyDetectionParserPlugin", ast => {
            const isStrictHarmony =
                parser.state.module.type === JAVASCRIPT_MODULE_TYPE_ESM;
            const isHarmony =
                isStrictHarmony ||
                ast.body.some(
                    statement =>
                        statement.type === "ImportDeclaration" ||
                        statement.type === "ExportDefaultDeclaration" ||
                        statement.type === "ExportNamedDeclaration" ||
                        statement.type === "ExportAllDeclaration"
                );
            if (isHarmony) {
                const module = parser.state.module;
                const compatDep = new HarmonyCompatibilityDependency();
                compatDep.loc = {
                    start: {
                        line: -1,
                        column: 0
                    },
                    end: {
                        line: -1,
                        column: 0
                    },
                    index: -3
                };
                module.addPresentationalDependency(compatDep);
                DynamicExports.bailout(parser.state);
                HarmonyExports.enable(parser.state, isStrictHarmony);
                parser.scope.isStrict = true;
            }
        });
 
        parser.hooks.topLevelAwait.tap("HarmonyDetectionParserPlugin", () => {
            const module = parser.state.module;
            if (!this.topLevelAwait) {
                throw new Error(
                    "The top-level-await experiment is not enabled (set experiments.topLevelAwait: true to enabled it)"
                );
            }
            if (!HarmonyExports.isEnabled(parser.state)) {
                throw new Error(
                    "Top-level-await is only supported in EcmaScript Modules"
                );
            }
            module.buildMeta.async = true;
        });
 
        /**
         * @returns {boolean | undefined} true if in harmony
         */
        const skipInHarmony = () => {
            if (HarmonyExports.isEnabled(parser.state)) {
                return true;
            }
        };
 
        /**
         * @returns {null | undefined} null if in harmony
         */
        const nullInHarmony = () => {
            if (HarmonyExports.isEnabled(parser.state)) {
                return null;
            }
        };
 
        const nonHarmonyIdentifiers = ["define", "exports"];
        for (const identifier of nonHarmonyIdentifiers) {
            parser.hooks.evaluateTypeof
                .for(identifier)
                .tap("HarmonyDetectionParserPlugin", nullInHarmony);
            parser.hooks.typeof
                .for(identifier)
                .tap("HarmonyDetectionParserPlugin", skipInHarmony);
            parser.hooks.evaluate
                .for(identifier)
                .tap("HarmonyDetectionParserPlugin", nullInHarmony);
            parser.hooks.expression
                .for(identifier)
                .tap("HarmonyDetectionParserPlugin", skipInHarmony);
            parser.hooks.call
                .for(identifier)
                .tap("HarmonyDetectionParserPlugin", skipInHarmony);
        }
    }
};