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
|
/**
* requireHook - module wrapping a function that register require() listener and returns hook/unhook control
*
* @author Ciprian Popa (cyparu)
* @since 0.0.1
* @version 0.0.1
*/
"use strict";
var Module = require("module").Module,
_load = Module._load,
_hookedAt, _listener;
/**
* Module hooker function that will replace Module._load and invoke the _listener with module and timing information
*
* @function _hooker
*/
function _hooker(name, parent) {
var timeIn = Date.now(),
exports = _load.apply(Module, arguments),
timeOut = Date.now(),
mod = parent.children[parent.children.length - 1]; // should be the last loaded children
// call the listener
_listener({
name: name,
parent: parent,
module: mod,
filename: mod ? mod.filename : name,
exports: exports,
requiredOn: timeIn,
startedIn: timeOut - timeIn
});
return exports;
}
/**
* Hook Node's require() so the configured callback will be invocked with additional module and time loading information information
*
* @param {Function} [listener] - optional listener if
* @method hook
*/
function _hook(listener) {
if (typeof listener !== "undefined") {
if (typeof listener !== "function") {
throw new Error("The optional parameter for hook() should be a function but was " + (typeof listener));
}
// set the listener
_listener = listener;
}
// set the hoocker loader
Module._load = _hooker;
// mark hooked time
_hookedAt = new Date();
}
/**
* Unhook Node's require() to the original function
*
* @method unhook
*/
function _unhook() {
// set the original loader
Module._load = _load;
// reset hooking time
_hookedAt = undefined;
}
/**
* Export a function that set the callback and return hook/unhook control functionality
*
* @function
* @param {Function} listener - require() listener
* @param {Boolean} [autohook=true] - optional flag telling if the hooking will be started automatically
* @return hook/unhook control function
*/
module.exports = function(listener, autohook) {
if (typeof listener !== "function") {
throw new Error("The hooking function should be set");
}
// set the listener
_listener = listener;
// if autohook (by default),
if (autohook !== false) {
_hook();
}
return {
hookedAt: _hookedAt,
hook: _hook,
unhook: _unhook
};
};
|
