import { useCallback, useState } from 'react';

const useSetState = <T extends object>(
    initialState: T = {} as T
): [T, (patch: Partial<T> | ((prevState: T) => Partial<T>)) => void] => {
    const [state, set] = useState<T>(initialState);
    const setState = useCallback(
        (patch: Partial<T> | ((prevState: T) => Partial<T>)) => {
            set((prevState) => ({
                ...prevState,
                ...(patch instanceof Function ? patch(prevState) : patch),
            }));
        },
        [set]
    );

    return [state, setState];
};

export default useSetState;