在Node.js中导入钩子

时间:2014-07-07 02:00:46

标签: javascript node.js hook require

我遇到过这个问题,这个问题可以通过Python中的导入钩子来解决。在Node.js中是否有类似的机制,允许在挂钩安装后拦截对require()的任何调用?

我试过以下:

originalRequire = global.require;

global.require = function(module) {
    console.log('Importing ' + module + '...');

    return originalRequire(module);
};

然而,这没有任何作用。有什么想法吗?

1 个答案:

答案 0 :(得分:6)

我找到了方法,它只是内部的一点点。解决方案草案:

Module = require('module');
Module.prototype.requireOrig = Module.prototype.require;
Module.prototype.require = (path) ->
    console.log('Importing ' + path + '...');

    return this.require(path)

HereModule类的源代码。

更新:这就是我的结果。它做的很简单 - 当需要以':/'开头的模块时。 (例如':/ foo / bar'),查找它们与主模块相关的信息。

Module = require 'module'
assert = require 'assert'

do ->
    origRequire = Module::require
    _require = (context, path) -> origRequire.call context, path

    main = require.main

    Module::require = (path) ->
        assert typeof path == 'string', 'path must be a string'
        assert path, 'missing path'

        if path[...2] == ':/'
            return _require main, "./#{path[2...]}"

        return _require @, path

上面代码段的等效Javascript:

var Module = require('module');
var assert = require('assert');

(function() {
    var origRequire = Module.prototype.require;
    var _require = function(context, path) {
        return origRequire.call(context, path);
    };

    var main = require.main;

    Module.prototype.require = function(path) {
        assert(typeof path === 'string', 'path must be a string');
        assert(path, 'missing path');

        if (path.slice(0, 2) === ':/') {
            return _require(main, "./" + path.slice(2));
        }

        return _require(this, path);
    };
})();