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
|
const { sources } = require("webpack");
const terser = require("terser");
const getScriptCodeScoped = (scriptCode, loaded) => scriptCode
? (loaded
? `document.addEventListener(\"DOMContentLoaded\", () => {${scriptCode}});`
: `(() => {${scriptCode}})();`)
: "";
class InjectScriptPlugin {
apply(compiler) {
compiler.hooks.compilation.tap("InjectScriptPlugin", compilation => {
compilation.hooks.processAssets.tapPromise(
{
name: "InjectScriptPlugin",
stage: compiler.webpack.Compilation.PROCESS_ASSETS_STAGE_OPTIMIZE,
additionalAssets: true
},
async assets => {
for (const assetName in assets) {
if (assetName.endsWith("document_start.js")) {
const currentAssetContent = compilation.getAsset(assetName).source.source();
const scripts = {
before: "",
injected: "",
injected_and_loaded: ""
};
let scriptEnd = 0;
while (true) {
const start = currentAssetContent.indexOf("/*<--\t", scriptEnd + 1);
if (start < 0) {
break;
} else {
const end = currentAssetContent.indexOf("\t-->*/", start + 1);
if (end < 0) {
throw Error("Failed to parse scripts with document_start runtime");
} else {
scriptEnd = currentAssetContent.indexOf("\n/*<---->*/", end + 1);
if (scriptEnd < 0) {
throw Error("Failed to parse scripts with document_start runtime");
}
const scriptType = currentAssetContent.substring(start + 6, end);
scripts[scriptType.trim()] += currentAssetContent.substring(end + 6, scriptEnd);
scriptEnd += 11;
}
}
}
for (const scriptType in scripts) {
const currentScriptCode = (await terser.minify(scripts[scriptType].replaceAll("`", "\\`"))).code;
scripts[scriptType] = currentScriptCode
? (scriptType === "injected_and_loaded"
? `document.addEventListener(\"DOMContentLoaded\", () => {${currentScriptCode}});`
: `(() => {${currentScriptCode}})();`)
: "";
}
const injectedCode = (scripts.injected || scripts.injected_and_loaded)
? `const o = new MutationObserver(() => {
const s = document.querySelector("head > script:nth-child(54)");
if (s) {
s.innerHTML = \`${scripts.injected}${scripts.injected_and_loaded}\`;
o.disconnect();
}
});
o.observe(document, {childList: true, subtree: true});`
: "";
const resultantCode = scripts.before + injectedCode;
compilation.updateAsset(
assetName,
new sources.RawSource(resultantCode)
);
}
}
}
);
});
}
}
module.exports = InjectScriptPlugin;
|
