RocketChat/Rocket.Chat

import type { UploadResult } from '@rocket.chat/ui-contexts';
import { useToastMessageDispatch, useUpload } from '@rocket.chat/ui-contexts';
import { useCallback } from 'react';

export const useEndpointUpload = (
    endpoint: Parameters<typeof useUpload>[0],
    successMessage: string,
): ((formData: FormData) => Promise<{ success: boolean }>) => {
    const sendData = useUpload(endpoint);
    const dispatchToastMessage = useToastMessageDispatch();

    return useCallback(
        async (formData: FormData) => {
            try {
                const data = sendData(formData);

                const promise = data instanceof Promise ? data : data.promise;

                const result = await (promise as unknown as Promise<UploadResult>);

                if (!result.success) {
                    throw new Error(String(result.status));
                }

                successMessage && dispatchToastMessage({ type: 'success', message: successMessage });

                return result as any;
            } catch (error) {
                dispatchToastMessage({ type: 'error', message: error });
                return { success: false };
            }
        },
        [dispatchToastMessage, sendData, successMessage],
    );
};