import {atom, useAtom} from 'jotai';
import {IDialogKey, IDialogType} from './components';

type injectedProps = 'visible' | 'hideDialog';
const dialogNameAtom = atom<IDialogKey | null>(null);
const payloadAtom = atom<
    | Omit<Parameters<IDialogType[keyof IDialogType]>[0], injectedProps>
    | undefined
>(undefined);

export default function useDialog() {
    const [dialogName, setDialogName] = useAtom(dialogNameAtom);
    const [payload, setPayload] = useAtom(payloadAtom);

    function showDialog<T extends keyof IDialogType>(
        name: T,
        payload?: Omit<Parameters<IDialogType[T]>[0], injectedProps>,
    ) {
        setDialogName(name);
        setPayload(payload);
    }

    function hideDialog() {
        setDialogName(null);
        setPayload(undefined);
    }

    return {dialogName, payload, showDialog, hideDialog};
}