config/webpack/plugins/inject-script-webpack-plugin.js


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

const { sources } = require("webpack");
const path = require("path");

const { getFileNameHash } = webpackRequire("utils/filename-hash");
const { getInjectedCode } = webpackRequire("utils/injected-code");

class InjectScriptPlugin {
	constructor(insertedBefore = false) {
		this.insertedBefore = insertedBefore;
	}

	apply(compiler) {
		compiler.hooks.compilation.tap("InjectScriptPlugin", compilation => {
			compilation.hooks.processAssets.tapPromise(
				{
					name: "InjectScriptPlugin",
					stage: compiler.webpack.Compilation.PROCESS_ASSETS_STAGE_ADDITIONAL,
					additionalAssets: true
				},
				async assets => {
					for (const assetName in assets) {
						if (/\.js$/.test(assetName)) {
							const injectedCodeResponse = getInjectedCode(compilation.getAsset(assetName).source.source())
							const scriptInjectorModuleName = getFileNameHash("script-injector", "shared", true);
							compilation.updateAsset(
								assetName,
								new sources.RawSource(injectedCodeResponse[0])
							);
							assets["scripts/" + path.basename(assetName)] = new sources.RawSource(
								`$${scriptInjectorModuleName}.injectScriptHead("scripts/injected/${path.basename(assetName)}", ${injectedCodeResponse[1]});`
							);
						}
					}
				}
			);
		});
	}
}

module.exports = InjectScriptPlugin;