Blame view

node_modules/webpack/lib/BannerPlugin.js 1.93 KB
aaac7fed   liuqimichale   add
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
  /*
   MIT License http://www.opensource.org/licenses/mit-license.php
   Author Tobias Koppers @sokra
   */
  
  "use strict";
  
  const ConcatSource = require("webpack-sources").ConcatSource;
  const ModuleFilenameHelpers = require("./ModuleFilenameHelpers");
  
  const 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) {
  		const options = this.options;
  		const banner = this.banner;
  
  		compiler.plugin("compilation", (compilation) => {
  			compilation.plugin("optimize-chunk-assets", (chunks, callback) => {
  				chunks.forEach((chunk) => {
  					if(options.entryOnly && !chunk.isInitial()) return;
  					chunk.files
  						.filter(ModuleFilenameHelpers.matchObject.bind(undefined, options))
  						.forEach((file) => {
  							let basename;
  							let query = "";
  							let filename = file;
  							const hash = compilation.hash;
  							const querySplit = filename.indexOf("?");
  
  							if(querySplit >= 0) {
  								query = filename.substr(querySplit);
  								filename = filename.substr(0, querySplit);
  							}
  
  							const lastSlashIndex = filename.lastIndexOf("/");
  
  							if(lastSlashIndex === -1) {
  								basename = filename;
  							} else {
  								basename = filename.substr(lastSlashIndex + 1);
  							}
  
  							const comment = compilation.getPath(banner, {
  								hash,
  								chunk,
  								filename,
  								basename,
  								query,
  							});
  
  							return compilation.assets[file] = new ConcatSource(comment, "\n", compilation.assets[file]);
  						});
  				});
  				callback();
  			});
  		});
  	}
  }
  
  module.exports = BannerPlugin;