-
Notifications
You must be signed in to change notification settings - Fork 1
/
index.js
56 lines (49 loc) · 1.51 KB
/
index.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/**
* Smart Banner Webpack Plugin supports webpack 2
*
* Author: johvin
* Date: 2017/02/13
*/
'use strict';
const ConcatSource = require('webpack-sources').ConcatSource;
const ModuleFilenameHelpers = require('webpack/lib/ModuleFilenameHelpers');
function wrapComment(str) {
if(!str.includes('\n')) return `/*! ${str} */`;
return `/*!\n * ${str.split('\n').join('\n * ')}\n */`;
}
class BannerPlugin {
constructor(options) {
if(arguments.length > 1)
throw new Error('BannerPlugin only takes one argument (pass an options object)');
if(typeof options === 'string')
options = {
banner: options
};
this.options = options || {};
this.banner = this.options.raw ? options.banner : wrapComment(options.banner);
}
apply(compiler) {
let options = this.options;
let banner = this.banner;
compiler.plugin('compilation', (compilation) => {
compilation.plugin('optimize-chunk-assets', (chunks, callback) => {
chunks.forEach((chunk) => {
if (
options.entryOnly
&& 'isInitial' in chunk
&& !chunk.isInitial()
) return;
chunk.files
.filter(ModuleFilenameHelpers.matchObject.bind(undefined, options))
.forEach((file) =>
compilation.assets[file] = new ConcatSource(
banner.replace(/\[filename\]/g, file), '\n', compilation.assets[file]
)
);
});
callback();
});
});
}
}
module.exports = BannerPlugin;