Dependency injection with mutation function

[SainteCroquette]

Hey !

using @tanstack/react-query i created i little wrapper to handle dependency injection on mutation functions while keeping type inference.

I am using dependency injection for my api modules called gateways in the case.

type WrappedMutation<R = unknown, P = unknown> = (param: P) => Promise<R>;

// the type of all my mutation callbacks which takes gateways in input, the APIs interface
type MutationCallback<R = unknown, P = unknown> = (param: P, gateways: Gateways) => Promise<R>;

// create an arrow function where you can inject variables from the scope
import gateways from '@core/gateways';

function useMutationFunctionWrapper<R, P>(callback: MutationCallback<R, P>): WrappedMutation<R, P> {
    return (arg: P): Promise<R> => {
        return callback(arg, gateways);
    };
}

// create the wrapped function
const updateCounter = useMutationFunctionWrapper(
    async (count: number, gateways: Gateways): Promise<number> => {
        await gateways.counterGateway.setCounter(1, count);
        return count;
    }
);

// no usage difference
const { isPending, isError, isSuccess, mutate } = useMutation({
    mutationFn: updateCounter,
    // others options....
});

Any feedback is welcome :)