Blame view

node_modules/webpack/lib/dependencies/RequireContextPlugin.js 3.26 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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
  /*
  	MIT License http://www.opensource.org/licenses/mit-license.php
  	Author Tobias Koppers @sokra
  */
  "use strict";
  
  const RequireContextDependency = require("./RequireContextDependency");
  const ContextElementDependency = require("./ContextElementDependency");
  
  const RequireContextDependencyParserPlugin = require("./RequireContextDependencyParserPlugin");
  
  class RequireContextPlugin {
  	constructor(modulesDirectories, extensions, mainFiles) {
  		if(!Array.isArray(modulesDirectories))
  			throw new Error("modulesDirectories must be an array");
  		if(!Array.isArray(extensions))
  			throw new Error("extensions must be an array");
  		this.modulesDirectories = modulesDirectories;
  		this.extensions = extensions;
  		this.mainFiles = mainFiles;
  	}
  
  	apply(compiler) {
  		compiler.plugin("compilation", (compilation, params) => {
  			const contextModuleFactory = params.contextModuleFactory;
  			const normalModuleFactory = params.normalModuleFactory;
  
  			compilation.dependencyFactories.set(RequireContextDependency, contextModuleFactory);
  			compilation.dependencyTemplates.set(RequireContextDependency, new RequireContextDependency.Template());
  
  			compilation.dependencyFactories.set(ContextElementDependency, normalModuleFactory);
  
  			params.normalModuleFactory.plugin("parser", (parser, parserOptions) => {
  
  				if(typeof parserOptions.requireContext !== "undefined" && !parserOptions.requireContext)
  					return;
  
  				parser.apply(new RequireContextDependencyParserPlugin());
  			});
  
  			params.contextModuleFactory.plugin("alternatives", (items, callback) => {
  				if(items.length === 0) return callback(null, items);
  
  				callback(null, items.map((obj) => {
  					return this.extensions.filter((ext) => {
  						const l = obj.request.length;
  						return l > ext.length && obj.request.substr(l - ext.length, l) === ext;
  					}).map((ext) => {
  						const l = obj.request.length;
  						return {
  							context: obj.context,
  							request: obj.request.substr(0, l - ext.length)
  						};
  					}).concat(obj);
  				}).reduce((a, b) => a.concat(b), []));
  			});
  
  			params.contextModuleFactory.plugin("alternatives", (items, callback) => {
  				if(items.length === 0) return callback(null, items);
  
  				callback(null, items.map((obj) => {
  					return this.mainFiles.filter((mainFile) => {
  						const l = obj.request.length;
  						return l > mainFile.length + 1 && obj.request.substr(l - mainFile.length - 1, l) === "/" + mainFile;
  					}).map((mainFile) => {
  						const l = obj.request.length;
  						return [{
  							context: obj.context,
  							request: obj.request.substr(0, l - mainFile.length)
  						}, {
  							context: obj.context,
  							request: obj.request.substr(0, l - mainFile.length - 1)
  						}];
  					}).reduce((a, b) => a.concat(b), []).concat(obj);
  				}).reduce((a, b) => a.concat(b), []));
  			});
  
  			params.contextModuleFactory.plugin("alternatives", (items, callback) => {
  				if(items.length === 0) return callback(null, items);
  
  				callback(null, items.map((obj) => {
  					for(let i = 0; i < this.modulesDirectories.length; i++) {
  						const dir = this.modulesDirectories[i];
  						const idx = obj.request.indexOf("./" + dir + "/");
  						if(idx === 0) {
  							obj.request = obj.request.slice(dir.length + 3);
  							break;
  						}
  					}
  					return obj;
  				}));
  			});
  		});
  	}
  }
  module.exports = RequireContextPlugin;