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
97
98
99
100
101
102
103
104
105
106
107
108
|
var Module = require("module"),
fs = require("fs"),
__get__ = require("./__get__.js"),
__set__ = require("./__set__.js"),
getImportGlobalsSrc = require("./getImportGlobalsSrc.js"),
detectStrictMode = require("./detectStrictMode.js"),
moduleWrapper0 = Module.wrapper[0], // caching original wrapper
moduleWrapper1 = Module.wrapper[1], // caching original wrapper
rewiredModules = []; // cache for all rewired modules so it can be reset anytime
function restoreOriginalWrappers() {
Module.wrapper[0] = moduleWrapper0;
Module.wrapper[1] = moduleWrapper1;
}
/**
* Does actual rewiring the module. For further documentation @see index.js
*/
function rewire(parentModule, filename, cache) {
var testModule,
nodeRequire,
prepend,
append,
src;
/**
* Proxies the first require call in order to draw back all changes.
* Thus our changes don't influence other modules
*
* @param {!String} path
*/
function requireProxy(path) {
restoreOriginalWrappers(); // we need to restore the wrappers now so we don't influence other modules
testModule.require = nodeRequire; // restoring original nodeRequire
return nodeRequire.call(testModule, path); // node's require only works when "this" points to the module
}
// Checking params
if (typeof filename !== "string") {
throw new TypeError("Filename must be a string");
}
// Resolve full filename relative to the parent module
filename = Module._resolveFilename(filename, parentModule);
// Special support for older node versions that returned an array on Module._resolveFilename
// @see https://github.com/joyent/node/blob/865b077819a9271a29f982faaef99dc635b57fbc/lib/module.js#L319
if (Array.isArray(filename)) {
filename = filename[1];
}
// Create testModule as it would be created by require()
testModule = new Module(filename, parentModule);
// Patching requireProxy
nodeRequire = testModule.require;
testModule.require = requireProxy;
// We prepend a list of all globals declared with var so they can be overridden (without changing original globals)
prepend = getImportGlobalsSrc();
// We append our special setter and getter.
append = "module.exports.__set__ = " + __set__.toString() + "; ";
append += "module.exports.__get__ = " + __get__.toString() + "; ";
// Check if the module uses the strict mode.
// If so we must ensure that "use strict"; stays at the beginning of the module.
src = fs.readFileSync(filename, "utf8");
if (detectStrictMode(src) === true) {
prepend = ' "use strict"; ' + prepend;
}
// Apply prepend and appends
Module.wrapper[0] = moduleWrapper0 + prepend;
Module.wrapper[1] = append + moduleWrapper1;
//console.log(Module.wrapper);
// Let the show begin
testModule.load(testModule.id);
// Store the rewired module in the cache when enabled
if (cache) {
rewiredModules.push(filename); // save in private cache for .reset()
require.cache[filename] = testModule;
}
// This is only necessary if nothing has been required within the module
restoreOriginalWrappers();
return testModule.exports;
}
/**
* Deletes all rewired modules from the cache
*/
rewire.reset = function () {
var i;
for (i = 0; i < rewiredModules.length; i++) {
delete require.cache[rewiredModules[i]];
}
rewiredModules = [];
};
module.exports = rewire;
|