Jungfraujoch/frontend/src/components/useUpload.ts

import { useState } from "react";
import { useMutation, UseMutationOptions } from "@tanstack/react-query";

/** Pull a human-readable message out of a hey-api error (400 = string, 500 = error_message). */
export function errorMessage(error: unknown): string {
    if (typeof error === "string") return error;
    if (error && typeof error === "object" && "msg" in error)
        return String((error as { msg?: unknown }).msg ?? "Unknown error");
    return "Upload failed";
}

export type UploadSnackbar = {
    open: boolean;
    success: boolean;
    message: string;
    onClose: () => void;
};

/**
 * Wraps a generated hey-api mutation (e.g. putConfigFileWriterMutation()) and
 * surfaces the result as a snackbar. submit(body) sends the typed body; on
 * failure the server's error message is shown.
 */
export function useUpload<TError, TVars extends { body?: unknown }>(
    mutationOptions: UseMutationOptions<unknown, TError, TVars>,
) {
    const mutation = useMutation(mutationOptions);
    const [open, setOpen] = useState(false);

    const submit = (body: TVars["body"]) =>
        mutation.mutate({ body } as TVars, { onSettled: () => setOpen(true) });

    const snackbar: UploadSnackbar = {
        open,
        success: mutation.isSuccess,
        message: mutation.error ? errorMessage(mutation.error) : "",
        onClose: () => setOpen(false),
    };

    return { submit, pending: mutation.isPending, snackbar };
}