Instrument a pre-call function hook in JS

instrumentPreCallHook.js

export function instrumentPreObjectFunctionCallHook(obj, methodName, hookFn) {
    const originalMethod = obj[methodName];
    obj[methodName] = (...args) => {
        const result = hookFn({ args, obj, methodName });
        return result instanceof Promise
            ? result.then(() => originalMethod.call(obj, args))
            : originalMethod.call(obj, args);
    };
    
    return obj[methodName].__deinstrument = () => {
        obj[methodName] = originalMethod;
    };
}
export function instrumentPostObjectFunctionCallHook(obj, methodName, hookFn) {
    const originalMethod = obj[methodName];
    obj[methodName] = (...args) => {
        const result = originalMethod.call(obj, args);
        return result instanceof Promise
            ? result.then((_result) => hookFn({ args, obj, methodName, result: _result }))
            : hookFn({ args, obj, methodName, result });
    };
    
    return obj[methodName].__deinstrument = () => {
        obj[methodName] = originalMethod;
    };
}
export function instrumentPreMethodCallHook(func, hookFn) {
    return (...args) => {
        const result = hookFn({ args, func });
        return result instanceof Promise
            ? result.then(() => func(...args))
            : func(...args);
    };
}
export function instrumentPostMethodCallHook(func, hookFn) {
    return (...args) => {
        const result = func(...args);
        return result instanceof Promise
            ? result.then((_result) => hookFn({ args, func, result: _result }))
            : hookFn({ args, func, result });
    };
}

typescript.ts

export function instrumentPreObjectFunctionCallHook(
    obj: any,
    methodName: string,
    hookFn: (props: { args: any[]; obj: any; methodName: string }) => any
) {
    const originalMethod = obj[methodName];
    obj[methodName] = (...args: any[]): any => {
        const result = hookFn({ args, obj, methodName });

        return result instanceof Promise
            ? result.then(() => originalMethod.call(obj, args))
            : originalMethod.call(obj, args);
    };

    return obj[methodName].__deinstrument = () => {
        obj[methodName] = originalMethod;
    };
}

export function instrumentPostObjectFunctionCallHook(
    obj: any,
    methodName: string,
    hookFn: (props: {
        args: any[];
        obj: any;
        methodName: string;
        result: any;
    }) => any
) {
    const originalMethod = obj[methodName];
    obj[methodName] = (...args: any[]): any => {
        const result = originalMethod.call(obj, args);
        return result instanceof Promise
            ? result.then((_result) =>
                  hookFn({ args, obj, methodName, result: _result })
              )
            : hookFn({ args, obj, methodName, result });
    };

    return obj[methodName].__deinstrument = () => {
        obj[methodName] = originalMethod;
    };
}

export function instrumentPreMethodCallHook(
    func: (...args: any[]) => any,
    hookFn: (props: { args: any[]; func: Function }) => any
) {
    return (...args: any[]): any => {
        const result = hookFn({ args, func });
        return result instanceof Promise
            ? result.then(() => func(...args))
            : func(...args);
    };
}

export function instrumentPostMethodCallHook(
    func: (...args: any[]) => any,
    hookFn: (props: { args: any[]; func: Function; result: any }) => any
) {
    return (...args: any[]): any => {
        const result = func(...args);
        return result instanceof Promise
            ? result.then((_result) => hookFn({ args, func, result: _result }))
            : hookFn({ args, func, result });
    };
}