ironmount/apps/client/app/components/create-volume-form.tsx

import { arktypeResolver } from "@hookform/resolvers/arktype";
import { volumeConfigSchema } from "@ironmount/schemas";
import { useMutation } from "@tanstack/react-query";
import { type } from "arktype";
import { CheckCircle, Loader2, XCircle } from "lucide-react";
import { useState } from "react";
import { useForm } from "react-hook-form";
import { testConnectionMutation } from "~/api-client/@tanstack/react-query.gen";
import { slugify } from "~/lib/utils";
import { Button } from "./ui/button";
import { Form, FormControl, FormDescription, FormField, FormItem, FormLabel, FormMessage } from "./ui/form";
import { Input } from "./ui/input";
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "./ui/select";

export const formSchema = type({
	name: "2<=string<=32",
}).and(volumeConfigSchema);

type FormValues = typeof formSchema.inferIn;

type Props = {
	onSubmit: (values: FormValues) => void;
	mode?: "create" | "update";
	initialValues?: Partial<FormValues>;
	formId?: string;
};

export const CreateVolumeForm = ({ onSubmit, mode = "create", initialValues, formId }: Props) => {
	const form = useForm<FormValues>({
		resolver: arktypeResolver(formSchema),
		defaultValues: initialValues,
	});
	const { watch, getValues } = form;

	const watchedBackend = watch("backend");

	const [testStatus, setTestStatus] = useState<"idle" | "loading" | "success" | "error">("idle");
	const [testMessage, setTestMessage] = useState<string>("");

	const testBackendConnection = useMutation({
		...testConnectionMutation(),
		onMutate: () => {
			setTestStatus("loading");
		},
		onError: () => {
			setTestStatus("error");
			setTestMessage("Failed to test connection. Please try again.");
		},
		onSuccess: (data) => {
			if (data?.success) {
				setTestStatus("success");
				setTestMessage(data.message);
			} else {
				setTestStatus("error");
				setTestMessage(data?.message || "Connection test failed");
			}
		},
	});

	const handleTestConnection = async () => {
		const formValues = getValues();

		if (formValues.backend === "nfs") {
			testBackendConnection.mutate({
				body: { config: formValues },
			});
		}
	};

	return (
		<Form {...form}>
			<form id={formId} onSubmit={form.handleSubmit(onSubmit)} className="space-y-4">
				<FormField
					name="name"
					render={({ field }) => (
						<FormItem>
							<FormLabel>Name</FormLabel>
							<FormControl>
								<Input
									placeholder="Volume name"
									value={field.value ?? ""}
									onChange={(e) => field.onChange(slugify(e.target.value))}
									max={32}
									min={1}
									disabled={mode === "update"}
									className={mode === "update" ? "bg-gray-50" : ""}
								/>
							</FormControl>
							<FormDescription>Unique identifier for the volume.</FormDescription>
							<FormMessage />
						</FormItem>
					)}
				/>
				<FormField
					name="backend"
					render={({ field }) => (
						<FormItem>
							<FormLabel>Backend</FormLabel>
							<Select onValueChange={field.onChange} defaultValue={field.value} value={field.value}>
								<FormControl>
									<SelectTrigger>
										<SelectValue placeholder="Select a backend" />
									</SelectTrigger>
								</FormControl>
								<SelectContent>
									<SelectItem value="directory">Directory</SelectItem>
									<SelectItem value="nfs">NFS</SelectItem>
								</SelectContent>
							</Select>
							<FormDescription>Choose the storage backend for this volume.</FormDescription>
							<FormMessage />
						</FormItem>
					)}
				/>

				{watchedBackend === "nfs" && (
					<>
						<FormField
							name="server"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Server</FormLabel>
									<FormControl>
										<Input placeholder="192.168.1.100" value={field.value ?? ""} onChange={field.onChange} />
									</FormControl>
									<FormDescription>NFS server IP address or hostname.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							name="exportPath"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Export Path</FormLabel>
									<FormControl>
										<Input placeholder="/export/data" value={field.value ?? ""} onChange={field.onChange} />
									</FormControl>
									<FormDescription>Path to the NFS export on the server.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							name="port"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input
											type="number"
											placeholder="2049"
											value={field.value ?? ""}
											onChange={(e) => field.onChange(parseInt(e.target.value, 10) || undefined)}
										/>
									</FormControl>
									<FormDescription>NFS server port (default: 2049).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							name="version"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Version</FormLabel>
									<Select onValueChange={field.onChange} defaultValue={field.value}>
										<FormControl>
											<SelectTrigger>
												<SelectValue placeholder="Select NFS version" />
											</SelectTrigger>
										</FormControl>
										<SelectContent>
											<SelectItem value="3">NFS v3</SelectItem>
											<SelectItem value="4">NFS v4</SelectItem>
											<SelectItem value="4.1">NFS v4.1</SelectItem>
										</SelectContent>
									</Select>
									<FormDescription>NFS protocol version to use.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend === "nfs" && (
					<div className="space-y-3">
						<div className="flex items-center gap-2">
							<Button
								type="button"
								variant="outline"
								onClick={handleTestConnection}
								disabled={testStatus === "loading" || !form.watch("server") || !form.watch("exportPath")}
								className="flex-1"
							>
								{testStatus === "loading" && <Loader2 className="mr-2 h-4 w-4 animate-spin" />}
								{testStatus === "success" && <CheckCircle className="mr-2 h-4 w-4 text-green-500" />}
								{testStatus === "error" && <XCircle className="mr-2 h-4 w-4 text-red-500" />}
								{testStatus === "idle" && "Test Connection"}
								{testStatus === "loading" && "Testing..."}
								{testStatus === "success" && "Connection Successful"}
								{testStatus === "error" && "Test Failed"}
							</Button>
						</div>
						{testMessage && (
							<div
								className={`text-sm p-2 rounded-md ${
									testStatus === "success"
										? "bg-green-50 text-green-700 border border-green-200"
										: testStatus === "error"
											? "bg-red-50 text-red-700 border border-red-200"
											: "bg-gray-50 text-gray-700 border border-gray-200"
								}`}
							>
								{testMessage}
							</div>
						)}
					</div>
				)}
			</form>
		</Form>
	);
};