马宇豪
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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
*/
 
"use strict";
 
const RuntimeGlobals = require("../RuntimeGlobals");
const Template = require("../Template");
const HelperRuntimeModule = require("./HelperRuntimeModule");
 
class GetTrustedTypesPolicyRuntimeModule extends HelperRuntimeModule {
    /**
     * @param {ReadonlySet<string>} runtimeRequirements runtime requirements
     */
    constructor(runtimeRequirements) {
        super("trusted types policy");
        this.runtimeRequirements = runtimeRequirements;
    }
 
    /**
     * @returns {string} runtime code
     */
    generate() {
        const { compilation } = this;
        const { runtimeTemplate, outputOptions } = compilation;
        const { trustedTypes } = outputOptions;
        const fn = RuntimeGlobals.getTrustedTypesPolicy;
        const wrapPolicyCreationInTryCatch = trustedTypes
            ? trustedTypes.onPolicyCreationFailure === "continue"
            : false;
 
        return Template.asString([
            "var policy;",
            `${fn} = ${runtimeTemplate.basicFunction("", [
                "// Create Trusted Type policy if Trusted Types are available and the policy doesn't exist yet.",
                "if (policy === undefined) {",
                Template.indent([
                    "policy = {",
                    Template.indent(
                        [
                            ...(this.runtimeRequirements.has(RuntimeGlobals.createScript)
                                ? [
                                        `createScript: ${runtimeTemplate.returningFunction(
                                            "script",
                                            "script"
                                        )}`
                                  ]
                                : []),
                            ...(this.runtimeRequirements.has(RuntimeGlobals.createScriptUrl)
                                ? [
                                        `createScriptURL: ${runtimeTemplate.returningFunction(
                                            "url",
                                            "url"
                                        )}`
                                  ]
                                : [])
                        ].join(",\n")
                    ),
                    "};",
                    ...(trustedTypes
                        ? [
                                'if (typeof trustedTypes !== "undefined" && trustedTypes.createPolicy) {',
                                Template.indent([
                                    ...(wrapPolicyCreationInTryCatch ? ["try {"] : []),
                                    ...[
                                        `policy = trustedTypes.createPolicy(${JSON.stringify(
                                            trustedTypes.policyName
                                        )}, policy);`
                                    ].map(line =>
                                        wrapPolicyCreationInTryCatch ? Template.indent(line) : line
                                    ),
                                    ...(wrapPolicyCreationInTryCatch
                                        ? [
                                                "} catch (e) {",
                                                Template.indent([
                                                    `console.warn('Could not create trusted-types policy ${JSON.stringify(
                                                        trustedTypes.policyName
                                                    )}');`
                                                ]),
                                                "}"
                                          ]
                                        : [])
                                ]),
                                "}"
                          ]
                        : [])
                ]),
                "}",
                "return policy;"
            ])};`
        ]);
    }
}
 
module.exports = GetTrustedTypesPolicyRuntimeModule;