pangolin/src/components/EditSiteProvisioningKeyCredenza.tsx

"use client";

import {
    Credenza,
    CredenzaBody,
    CredenzaClose,
    CredenzaContent,
    CredenzaDescription,
    CredenzaFooter,
    CredenzaHeader,
    CredenzaTitle
} from "@app/components/Credenza";
import {
    Form,
    FormControl,
    FormDescription,
    FormField,
    FormItem,
    FormLabel,
    FormMessage
} from "@app/components/ui/form";
import { Button } from "@app/components/ui/button";
import { Checkbox } from "@app/components/ui/checkbox";
import { Input } from "@app/components/ui/input";
import { useEnvContext } from "@app/hooks/useEnvContext";
import { toast } from "@app/hooks/useToast";
import { createApiClient, formatAxiosError } from "@app/lib/api";
import { UpdateSiteProvisioningKeyResponse } from "@server/routers/siteProvisioning/types";
import { AxiosResponse } from "axios";
import { useTranslations } from "next-intl";
import { useRouter } from "next/navigation";
import { useEffect, useState } from "react";
import { useForm } from "react-hook-form";
import { z } from "zod";
import { zodResolver } from "@hookform/resolvers/zod";
import moment from "moment";

const FORM_ID = "edit-site-provisioning-key-form";

export type EditableSiteProvisioningKey = {
    id: string;
    name: string;
    maxBatchSize: number | null;
    validUntil: string | null;
};

type EditSiteProvisioningKeyCredenzaProps = {
    open: boolean;
    setOpen: (open: boolean) => void;
    orgId: string;
    provisioningKey: EditableSiteProvisioningKey | null;
};

export default function EditSiteProvisioningKeyCredenza({
    open,
    setOpen,
    orgId,
    provisioningKey
}: EditSiteProvisioningKeyCredenzaProps) {
    const t = useTranslations();
    const router = useRouter();
    const api = createApiClient(useEnvContext());
    const [loading, setLoading] = useState(false);

    const editFormSchema = z
        .object({
            name: z.string(),
            unlimitedBatchSize: z.boolean(),
            maxBatchSize: z
                .number()
                .int()
                .min(1, { message: t("provisioningKeysMaxBatchSizeInvalid") })
                .max(1_000_000, {
                    message: t("provisioningKeysMaxBatchSizeInvalid")
                }),
            validUntil: z.string().optional()
        })
        .superRefine((data, ctx) => {
            const v = data.validUntil;
            if (v == null || v.trim() === "") {
                return;
            }
            if (Number.isNaN(Date.parse(v))) {
                ctx.addIssue({
                    code: "custom",
                    message: t("provisioningKeysValidUntilInvalid"),
                    path: ["validUntil"]
                });
            }
        });

    type EditFormValues = z.infer<typeof editFormSchema>;

    const form = useForm<EditFormValues>({
        resolver: zodResolver(editFormSchema),
        defaultValues: {
            name: "",
            unlimitedBatchSize: false,
            maxBatchSize: 100,
            validUntil: ""
        }
    });

    useEffect(() => {
        if (!open || !provisioningKey) {
            return;
        }
        form.reset({
            name: provisioningKey.name,
            unlimitedBatchSize: provisioningKey.maxBatchSize == null,
            maxBatchSize: provisioningKey.maxBatchSize ?? 100,
            validUntil: provisioningKey.validUntil
                ? moment(provisioningKey.validUntil).format("YYYY-MM-DDTHH:mm")
                : ""
        });
    }, [open, provisioningKey, form]);

    async function onSubmit(data: EditFormValues) {
        if (!provisioningKey) {
            return;
        }
        setLoading(true);
        try {
            const res = await api
                .patch<
                    AxiosResponse<UpdateSiteProvisioningKeyResponse>
                >(
                    `/org/${orgId}/site-provisioning-key/${provisioningKey.id}`,
                    {
                        maxBatchSize: data.unlimitedBatchSize
                            ? null
                            : data.maxBatchSize,
                        validUntil:
                            data.validUntil == null ||
                            data.validUntil.trim() === ""
                                ? ""
                                : data.validUntil
                    }
                )
                .catch((e) => {
                    toast({
                        variant: "destructive",
                        title: t("provisioningKeysUpdateError"),
                        description: formatAxiosError(e)
                    });
                });

            if (res && res.status === 200) {
                toast({
                    title: t("provisioningKeysUpdated"),
                    description: t("provisioningKeysUpdatedDescription")
                });
                setOpen(false);
                router.refresh();
            }
        } finally {
            setLoading(false);
        }
    }

    const unlimitedBatchSize = form.watch("unlimitedBatchSize");

    if (!provisioningKey) {
        return null;
    }

    return (
        <Credenza open={open} onOpenChange={setOpen}>
            <CredenzaContent>
                <CredenzaHeader>
                    <CredenzaTitle>{t("provisioningKeysEdit")}</CredenzaTitle>
                    <CredenzaDescription>
                        {t("provisioningKeysEditDescription")}
                    </CredenzaDescription>
                </CredenzaHeader>
                <CredenzaBody>
                    <Form {...form}>
                        <form
                            id={FORM_ID}
                            onSubmit={form.handleSubmit(onSubmit)}
                            className="space-y-4"
                        >
                            <FormField
                                control={form.control}
                                name="name"
                                render={({ field }) => (
                                    <FormItem>
                                        <FormLabel>{t("name")}</FormLabel>
                                        <FormControl>
                                            <Input
                                                autoComplete="off"
                                                disabled
                                                {...field}
                                            />
                                        </FormControl>
                                    </FormItem>
                                )}
                            />
                            <FormField
                                control={form.control}
                                name="maxBatchSize"
                                render={({ field }) => (
                                    <FormItem>
                                        <FormLabel>
                                            {t("provisioningKeysMaxBatchSize")}
                                        </FormLabel>
                                        <FormControl>
                                            <Input
                                                type="number"
                                                min={1}
                                                max={1_000_000}
                                                autoComplete="off"
                                                disabled={unlimitedBatchSize}
                                                name={field.name}
                                                ref={field.ref}
                                                onBlur={field.onBlur}
                                                onChange={(e) => {
                                                    const v = e.target.value;
                                                    field.onChange(
                                                        v === ""
                                                            ? 100
                                                            : Number(v)
                                                    );
                                                }}
                                                value={field.value}
                                            />
                                        </FormControl>
                                        <FormMessage />
                                    </FormItem>
                                )}
                            />
                            <FormField
                                control={form.control}
                                name="unlimitedBatchSize"
                                render={({ field }) => (
                                    <FormItem className="flex flex-row items-center gap-3 space-y-0">
                                        <FormControl>
                                            <Checkbox
                                                id="provisioning-edit-unlimited-batch"
                                                checked={field.value}
                                                onCheckedChange={(c) =>
                                                    field.onChange(c === true)
                                                }
                                            />
                                        </FormControl>
                                        <FormLabel
                                            htmlFor="provisioning-edit-unlimited-batch"
                                            className="cursor-pointer font-normal !mt-0"
                                        >
                                            {t(
                                                "provisioningKeysUnlimitedBatchSize"
                                            )}
                                        </FormLabel>
                                    </FormItem>
                                )}
                            />
                            <FormField
                                control={form.control}
                                name="validUntil"
                                render={({ field }) => (
                                    <FormItem>
                                        <FormLabel>
                                            {t("provisioningKeysValidUntil")}
                                        </FormLabel>
                                        <FormControl>
                                            <Input
                                                type="datetime-local"
                                                {...field}
                                            />
                                        </FormControl>
                                        <FormDescription>
                                            {t("provisioningKeysValidUntilHint")}
                                        </FormDescription>
                                        <FormMessage />
                                    </FormItem>
                                )}
                            />
                        </form>
                    </Form>
                </CredenzaBody>
                <CredenzaFooter>
                    <CredenzaClose asChild>
                        <Button variant="outline">{t("close")}</Button>
                    </CredenzaClose>
                    <Button
                        type="submit"
                        form={FORM_ID}
                        loading={loading}
                        disabled={loading}
                    >
                        {t("save")}
                    </Button>
                </CredenzaFooter>
            </CredenzaContent>
        </Credenza>
    );
}