通过使用源代码字符串来更改函数非常简单。要针对特定实例执行此操作,请尝试:
eval(doSomething.toString().replace(/}\s*$/, ' return id; $&'));
现在doSomething
返回 ID。我通常不喜欢,但由于需要访问局部变量,因此此处不适用eval
正常的面向方面的编程技术。
如果doSomething
已经返回一个值,请尝试将主体包装在 a 中try ... finally
:
eval(doSomething.toString()
.replace(/^function *\w* *\([^)]*\) *{/, '$& try {')
.replace(/}\s*$/, '} finally { window.someID = id; } $&')
);
要将其转换为函数,我们需要使代码在全局范围内进行计算。最初,此答案with
用于更改 的范围eval
,但这目前在浏览器中不起作用。相反,.call
用于更改 to 的eval
范围window
。
(function () {
var begin = /^function\s*\w*\s*\([^)]*\)\s*{/,
end = /}\s*$/;
function alter(func, replacer) {
var newFunc = replacer(func.toString());
eval.call(window, newFunc);
}
function insertCode(func, replacer, pattern) {
alter(func, function (source) {
return source.replace(pattern, replacer);
});
};
/* Note: explicit `window` to mark these as globals */
window.before = function (func, code) {
return insertCode(func, '$& ' + code, begin);
};
window.after = function (func, code) {
return insertCode(func, code + ' $&', end);
};
window.around = function (func, pre, post) {
/* Can't simply call `before` and `after`, as a partial code insertion may produce a syntax error. */
alter(func, function(source) {
return source
.replace(begin, '$& ' + pre)
.replace(end, post + ' $&');
});
};
})();
...
after(doSomething, 'return id;');
/* or */
around(doSomething, 'try {', '} finally { window.someID = id; }');
如果要重写绑定到变量的方法和匿名函数,请更改alter
为:
...
function alter(func, replacer) {
var newFunc = replacer(eval('window.' + funcName).toString());
eval.call(window, newFunc);
}
...
function Foo() {}
Foo.prototype.bar = function () { var secret=0x09F91102; }
...
after('Foo.prototype.bar', 'return secret;');
请注意,函数的第一个参数现在是字符串。可以进一步改进以处理在全局范围内无法访问的方法。