1
const HtmlWebpackPlugin = require('html-webpack-plugin')
2

3
// 启用 speed-measure-webpack-plugin 时此插件不能正常工作
4
class InjectVConsolePlugin {
5
    constructor(options = {}) {
6
        this.vconsoleUrl = options.vconsoleUrl || 'https://static.zhnysz.com/lib/vconsole.min.js'
7
        this.enabled = Boolean(process.env.VCONSOLE)
8
    }
9

10
    apply(compiler) {
11
        if (!this.enabled) {
12
            return
13
        }
14

15
        compiler.hooks.compilation.tap('InjectVConsolePlugin', (compilation) => {
16
            HtmlWebpackPlugin.getHooks(compilation).alterAssetTags.tap(
17
                'InjectVConsolePlugin',
18
                (data) => {
19
                    const scripts = data.assetTags?.scripts || data.scripts
20
                    if (scripts) {
21
                        scripts.unshift(
22
                            {
23
                                tagName: 'script',
24
                                voidTag: false,
25
                                attributes: {
26
                                    src: this.vconsoleUrl,
27
                                },
28
                            },
29
                            {
30
                                tagName: 'script',
31
                                voidTag: false,
32
                                innerHTML: 'new window.VConsole()',
33
                            }
34
                        )
35
                    }
36
                    return data
37
                }
38
            )
39
        })
40
    }
41
}
42

43
module.exports = InjectVConsolePlugin