You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
44 lines
1.2 KiB
function monkeyPatch(name, parentObj, patches) {
|
|
const injId = Math.random().toString(36).slice(2).toUpperCase()
|
|
const before = (patches["before"] === undefined) ? (...args)=>args : patches["before"];
|
|
const instead = (patches["instead"] === undefined) ? (...args)=>args : patches["instead"];
|
|
const after = (patches["after"] === undefined) ? (args, res)=>res : patches["after"];
|
|
const handler = {
|
|
apply: (target, thisArg, args) => {
|
|
before(args);
|
|
res = (patches["instead"] === undefined) ? target(...args) : instead(...args);
|
|
return after(...args, res);
|
|
}
|
|
};
|
|
const prox = new Proxy(parentObj[name], handler);
|
|
const orig = parentObj[name];
|
|
parentObj[name] = prox;
|
|
const unpatch = () => {
|
|
parentObj[name] = orig;
|
|
}
|
|
parentObj[injId] = {
|
|
name: name,
|
|
orig: orig,
|
|
unpatch: unpatch
|
|
};
|
|
return unpatch;
|
|
}
|
|
function before(name, parentObj, func) {
|
|
return monkeyPatch(name, parentObj, {before: func})
|
|
}
|
|
|
|
function instead(name, parentObj, func) {
|
|
return monkeyPatch(name, parentObj, {instead: func})
|
|
}
|
|
|
|
function after(name, parentObj, func) {
|
|
return monkeyPatch(name, parentObj, {after: func})
|
|
}
|
|
|
|
module.exports = {
|
|
monkeyPatch,
|
|
before,
|
|
instead,
|
|
after
|
|
}
|