unleash.unleash/frontend/src/hooks/useChangeRequestToggle.ts

import { useCallback, useState } from 'react';
import useToast from 'hooks/useToast';
import { formatUnknownError } from 'utils/formatUnknownError';
import { useChangeRequestApi } from './api/actions/useChangeRequestApi/useChangeRequestApi';
import { usePendingChangeRequests } from './api/getters/usePendingChangeRequests/usePendingChangeRequests';

export const useChangeRequestToggle = (project: string) => {
    const { setToastData, setToastApiError } = useToast();
    const { addChange } = useChangeRequestApi();
    const { refetch: refetchChangeRequests } =
        usePendingChangeRequests(project);
    const [pending, setPending] = useState(false);

    const [changeRequestDialogDetails, setChangeRequestDialogDetails] =
        useState<{
            enabled?: boolean;
            shouldActivateDisabledStrategies?: boolean;
            featureName?: string;
            environment?: string;
            isOpen: boolean;
        }>({ isOpen: false });

    const onChangeRequestToggle = useCallback(
        (
            featureName: string,
            environment: string,
            enabled: boolean,
            shouldActivateDisabledStrategies: boolean,
        ) => {
            setChangeRequestDialogDetails({
                featureName,
                environment,
                enabled,
                shouldActivateDisabledStrategies,
                isOpen: true,
            });
        },
        [],
    );

    const onChangeRequestToggleClose = useCallback(() => {
        setChangeRequestDialogDetails((prev) => ({ ...prev, isOpen: false }));
    }, []);

    const onChangeRequestToggleConfirm = useCallback(async () => {
        try {
            setPending(true);
            await addChange(project, changeRequestDialogDetails.environment!, {
                feature: changeRequestDialogDetails.featureName!,
                action: 'updateEnabled',
                payload: {
                    enabled: Boolean(changeRequestDialogDetails.enabled),
                    shouldActivateDisabledStrategies: Boolean(
                        changeRequestDialogDetails.shouldActivateDisabledStrategies,
                    ),
                },
            });
            refetchChangeRequests();
            setChangeRequestDialogDetails((prev) => ({
                ...prev,
                isOpen: false,
            }));
            setToastData({
                type: 'success',
                title: 'Changes added to the draft!',
            });
        } catch (error) {
            setToastApiError(formatUnknownError(error));
            setChangeRequestDialogDetails((prev) => ({
                ...prev,
                isOpen: false,
            }));
        } finally {
            setPending(false);
        }
    }, [addChange]);

    return {
        pending,
        onChangeRequestToggle,
        onChangeRequestToggleClose,
        onChangeRequestToggleConfirm,
        changeRequestDialogDetails,
    };
};