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

const SUPPORTS_CONNECTION_TEST = ["nfs", "smb", "webdav", "mariadb", "mysql", "postgres"];

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

export type FormValues = typeof formSchema.inferIn;

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

const defaultValuesForType = {
	directory: { backend: "directory" as const, path: "/" },
	nfs: { backend: "nfs" as const, port: 2049, version: "4.1" as const },
	smb: { backend: "smb" as const, port: 445, vers: "3.0" as const },
	webdav: { backend: "webdav" as const, port: 80, ssl: false },
	mariadb: { backend: "mariadb" as const, port: 3306 },
	mysql: { backend: "mysql" as const, port: 3306 },
	postgres: { backend: "postgres" as const, port: 5432, dumpFormat: "custom" as const },
};

export const CreateVolumeForm = ({ onSubmit, mode = "create", initialValues, formId, loading, className }: Props) => {
	const form = useForm<FormValues>({
		resolver: arktypeResolver(cleanSchema as unknown as typeof formSchema),
		defaultValues: initialValues,
		resetOptions: {
			keepDefaultValues: true,
			keepDirtyValues: false,
		},
	});

	const { watch, getValues } = form;

	const watchedBackend = watch("backend");

	useEffect(() => {
		if (mode === "create") {
			form.reset({
				name: form.getValues().name,
				...defaultValuesForType[watchedBackend as keyof typeof defaultValuesForType],
			});
		}
	}, [watchedBackend, form, mode]);

	const [testMessage, setTestMessage] = useState<{ success: boolean; message: string } | null>(null);

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

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

		if (SUPPORTS_CONNECTION_TEST.includes(formValues.backend)) {
			testBackendConnection.mutate({
				body: { config: formValues },
			});
		}
	};

	return (
		<Form {...form}>
			<form id={formId} onSubmit={form.handleSubmit(onSubmit)} className={cn("space-y-4", className)}>
				<FormField
					control={form.control}
					name="name"
					render={({ field }) => (
						<FormItem>
							<FormLabel>Name</FormLabel>
							<FormControl>
								<Input
									{...field}
									placeholder="Volume name"
									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
					control={form.control}
					name="backend"
					render={({ field }) => (
						<FormItem>
							<FormLabel>Backend</FormLabel>
							<Select onValueChange={field.onChange} defaultValue={field.value} value={field.value}>
								<FormControl>
									<SelectTrigger className="w-[280px]">
										<SelectValue placeholder="Select a backend" />
									</SelectTrigger>
								</FormControl>
								<SelectContent>
									<SelectGroup>
										<SelectItem value="directory">Directory</SelectItem>
									</SelectGroup>
									<SelectGroup>
										<SelectLabel>Network Storage</SelectLabel>
										<SelectItem value="nfs">NFS</SelectItem>
										<SelectItem value="smb">SMB</SelectItem>
										<SelectItem value="webdav">WebDAV</SelectItem>
									</SelectGroup>
									<SelectGroup>
										<SelectLabel>Databases</SelectLabel>
										<SelectItem value="mariadb">MariaDB</SelectItem>
										<SelectItem value="mysql">MySQL</SelectItem>
										<SelectItem value="postgres">PostgreSQL</SelectItem>
									</SelectGroup>
								</SelectContent>
							</Select>
							<FormDescription>Choose the storage backend for this volume.</FormDescription>
							<FormMessage />
						</FormItem>
					)}
				/>

				{watchedBackend === "directory" && (
					<FormField
						control={form.control}
						name="path"
						render={({ field }) => {
							return (
								<FormItem>
									<FormLabel>Directory Path</FormLabel>
									<FormControl>
										{field.value ? (
											<div className="flex items-center gap-2">
												<div className="flex-1 border rounded-md p-3 bg-muted/50">
													<div className="text-xs font-medium text-muted-foreground mb-1">Selected path:</div>
													<div className="text-sm font-mono break-all">{field.value}</div>
												</div>
												<Button type="button" variant="outline" size="sm" onClick={() => field.onChange("")}>
													Change
												</Button>
											</div>
										) : (
											<DirectoryBrowser onSelectPath={(path) => field.onChange(path)} selectedPath={field.value} />
										)}
									</FormControl>
									<FormDescription>Browse and select a directory on the host filesystem to track.</FormDescription>
									<FormMessage />
								</FormItem>
							);
						}}
					/>
				)}

				{watchedBackend === "nfs" && (
					<>
						<FormField
							control={form.control}
							name="server"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Server</FormLabel>
									<FormControl>
										<Input placeholder="192.168.1.100" {...field} />
									</FormControl>
									<FormDescription>NFS server IP address or hostname.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="exportPath"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Export Path</FormLabel>
									<FormControl>
										<Input placeholder="/export/data" {...field} />
									</FormControl>
									<FormDescription>Path to the NFS export on the server.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="port"
							defaultValue={2049}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input type="number" placeholder="2049" {...field} />
									</FormControl>
									<FormDescription>NFS server port (default: 2049).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="version"
							defaultValue="4.1"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Version</FormLabel>
									<Select onValueChange={field.onChange} defaultValue="4.1">
										<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>
							)}
						/>
						<FormField
							control={form.control}
							name="readOnly"
							defaultValue={false}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Read-only Mode</FormLabel>
									<FormControl>
										<div className="flex items-center space-x-2">
											<input
												type="checkbox"
												checked={field.value ?? false}
												onChange={(e) => field.onChange(e.target.checked)}
												className="rounded border-gray-300"
											/>
											<span className="text-sm">Mount volume as read-only</span>
										</div>
									</FormControl>
									<FormDescription>
										Prevent any modifications to the volume. Recommended for backup sources and sensitive data.
									</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend === "webdav" && (
					<>
						<FormField
							control={form.control}
							name="server"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Server</FormLabel>
									<FormControl>
										<Input placeholder="example.com" {...field} />
									</FormControl>
									<FormDescription>WebDAV server hostname or IP address.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="path"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Path</FormLabel>
									<FormControl>
										<Input placeholder="/webdav" {...field} />
									</FormControl>
									<FormDescription>Path to the WebDAV directory on the server.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="username"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Username (Optional)</FormLabel>
									<FormControl>
										<Input placeholder="admin" {...field} />
									</FormControl>
									<FormDescription>Username for WebDAV authentication (optional).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="password"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Password (Optional)</FormLabel>
									<FormControl>
										<Input type="password" placeholder="••••••••" {...field} />
									</FormControl>
									<FormDescription>Password for WebDAV authentication (optional).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="port"
							defaultValue={80}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input type="number" placeholder="80" {...field} />
									</FormControl>
									<FormDescription>WebDAV server port (default: 80 for HTTP, 443 for HTTPS).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="ssl"
							defaultValue={false}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Use SSL/HTTPS</FormLabel>
									<FormControl>
										<div className="flex items-center space-x-2">
											<input
												type="checkbox"
												checked={field.value ?? false}
												onChange={(e) => field.onChange(e.target.checked)}
												className="rounded border-gray-300"
											/>
											<span className="text-sm">Enable HTTPS for secure connections</span>
										</div>
									</FormControl>
									<FormDescription>Use HTTPS instead of HTTP for secure connections.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="readOnly"
							defaultValue={false}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Read-only Mode</FormLabel>
									<FormControl>
										<div className="flex items-center space-x-2">
											<input
												type="checkbox"
												checked={field.value ?? false}
												onChange={(e) => field.onChange(e.target.checked)}
												className="rounded border-gray-300"
											/>
											<span className="text-sm">Mount volume as read-only</span>
										</div>
									</FormControl>
									<FormDescription>
										Prevent any modifications to the volume. Recommended for backup sources and sensitive data.
									</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend === "smb" && (
					<>
						<FormField
							control={form.control}
							name="server"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Server</FormLabel>
									<FormControl>
										<Input placeholder="192.168.1.100" value={field.value} onChange={field.onChange} />
									</FormControl>
									<FormDescription>SMB server IP address or hostname.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="share"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Share</FormLabel>
									<FormControl>
										<Input placeholder="myshare" value={field.value} onChange={field.onChange} />
									</FormControl>
									<FormDescription>SMB share name on the server.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="username"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Username</FormLabel>
									<FormControl>
										<Input placeholder="admin" value={field.value} onChange={field.onChange} />
									</FormControl>
									<FormDescription>Username for SMB authentication.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="password"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Password</FormLabel>
									<FormControl>
										<Input type="password" placeholder="••••••••" value={field.value} onChange={field.onChange} />
									</FormControl>
									<FormDescription>Password for SMB authentication.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="vers"
							defaultValue="3.0"
							render={({ field }) => (
								<FormItem>
									<FormLabel>SMB Version</FormLabel>
									<Select onValueChange={field.onChange} defaultValue={field.value || "3.0"}>
										<FormControl>
											<SelectTrigger>
												<SelectValue placeholder="Select SMB version" />
											</SelectTrigger>
										</FormControl>
										<SelectContent>
											<SelectItem value="1.0">SMB v1.0</SelectItem>
											<SelectItem value="2.0">SMB v2.0</SelectItem>
											<SelectItem value="2.1">SMB v2.1</SelectItem>
											<SelectItem value="3.0">SMB v3.0</SelectItem>
										</SelectContent>
									</Select>
									<FormDescription>SMB protocol version to use (default: 3.0).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="domain"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Domain (Optional)</FormLabel>
									<FormControl>
										<Input placeholder="WORKGROUP" value={field.value} onChange={field.onChange} />
									</FormControl>
									<FormDescription>Domain or workgroup for authentication (optional).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="port"
							defaultValue={445}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input
											type="number"
											placeholder="445"
											value={field.value}
											defaultValue={445}
											onChange={(e) => field.onChange(parseInt(e.target.value, 10) || undefined)}
										/>
									</FormControl>
									<FormDescription>SMB server port (default: 445).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="readOnly"
							defaultValue={false}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Read-only Mode</FormLabel>
									<FormControl>
										<div className="flex items-center space-x-2">
											<input
												type="checkbox"
												checked={field.value ?? false}
												onChange={(e) => field.onChange(e.target.checked)}
												className="rounded border-gray-300"
											/>
											<span className="text-sm">Mount volume as read-only</span>
										</div>
									</FormControl>
									<FormDescription>
										Prevent any modifications to the volume. Recommended for backup sources and sensitive data.
									</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend === "mariadb" && (
					<>
						<FormField
							control={form.control}
							name="host"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Host</FormLabel>
									<FormControl>
										<Input placeholder="localhost" {...field} />
									</FormControl>
									<FormDescription>MariaDB server hostname or IP address.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="port"
							defaultValue={3306}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input type="number" placeholder="3306" {...field} />
									</FormControl>
									<FormDescription>MariaDB server port (default: 3306).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="username"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Username</FormLabel>
									<FormControl>
										<Input placeholder="root" {...field} />
									</FormControl>
									<FormDescription>Database user with backup privileges.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="password"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Password</FormLabel>
									<FormControl>
										<Input type="password" placeholder="••••••••" {...field} />
									</FormControl>
									<FormDescription>Password for database authentication.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="database"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Database</FormLabel>
									<FormControl>
										<Input placeholder="myapp_production" {...field} />
									</FormControl>
									<FormDescription>Name of the database to backup.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend === "mysql" && (
					<>
						<FormField
							control={form.control}
							name="host"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Host</FormLabel>
									<FormControl>
										<Input placeholder="localhost" {...field} />
									</FormControl>
									<FormDescription>MySQL server hostname or IP address.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="port"
							defaultValue={3306}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input type="number" placeholder="3306" {...field} />
									</FormControl>
									<FormDescription>MySQL server port (default: 3306).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="username"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Username</FormLabel>
									<FormControl>
										<Input placeholder="root" {...field} />
									</FormControl>
									<FormDescription>Database user with backup privileges.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="password"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Password</FormLabel>
									<FormControl>
										<Input type="password" placeholder="••••••••" {...field} />
									</FormControl>
									<FormDescription>Password for database authentication.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="database"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Database</FormLabel>
									<FormControl>
										<Input placeholder="myapp_production" {...field} />
									</FormControl>
									<FormDescription>Name of the database to backup.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend === "postgres" && (
					<>
						<FormField
							control={form.control}
							name="host"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Host</FormLabel>
									<FormControl>
										<Input placeholder="localhost" {...field} />
									</FormControl>
									<FormDescription>PostgreSQL server hostname or IP address.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="port"
							defaultValue={5432}
							render={({ field }) => (
								<FormItem>
									<FormLabel>Port</FormLabel>
									<FormControl>
										<Input type="number" placeholder="5432" {...field} />
									</FormControl>
									<FormDescription>PostgreSQL server port (default: 5432).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="username"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Username</FormLabel>
									<FormControl>
										<Input placeholder="postgres" {...field} />
									</FormControl>
									<FormDescription>Database user with backup privileges.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="password"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Password</FormLabel>
									<FormControl>
										<Input type="password" placeholder="••••••••" {...field} />
									</FormControl>
									<FormDescription>Password for database authentication.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="database"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Database</FormLabel>
									<FormControl>
										<Input placeholder="myapp_production" {...field} />
									</FormControl>
									<FormDescription>Name of the database to backup.</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
						<FormField
							control={form.control}
							name="dumpFormat"
							defaultValue="custom"
							render={({ field }) => (
								<FormItem>
									<FormLabel>Dump Format</FormLabel>
									<Select onValueChange={field.onChange} defaultValue={field.value || "custom"}>
										<FormControl>
											<SelectTrigger>
												<SelectValue placeholder="Select dump format" />
											</SelectTrigger>
										</FormControl>
										<SelectContent>
											<SelectItem value="custom">Custom (Compressed)</SelectItem>
											<SelectItem value="plain">Plain SQL</SelectItem>
											<SelectItem value="directory">Directory</SelectItem>
										</SelectContent>
									</Select>
									<FormDescription>Format for database dumps (custom recommended).</FormDescription>
									<FormMessage />
								</FormItem>
							)}
						/>
					</>
				)}

				{watchedBackend !== "directory" && (
					<div className="space-y-3">
						<div className="flex items-center gap-2">
							<Button
								type="button"
								variant="outline"
								onClick={handleTestConnection}
								disabled={testBackendConnection.isPending}
								className="flex-1"
							>
								{testBackendConnection.isPending && <Loader2 className="mr-2 h-4 w-4 animate-spin" />}
								{!testBackendConnection.isPending && testMessage?.success && (
									<CheckCircle className="mr-2 h-4 w-4 text-green-500" />
								)}
								{!testBackendConnection.isPending && testMessage && !testMessage.success && (
									<XCircle className="mr-2 h-4 w-4 text-red-500" />
								)}
								{testBackendConnection.isPending
									? "Testing..."
									: testMessage
										? testMessage.success
											? "Connection Successful"
											: "Test Failed"
										: "Test Connection"}
							</Button>
						</div>
						{testMessage && (
							<div
								className={cn("text-xs p-2 rounded-md text-wrap wrap-anywhere", {
									"bg-green-50 text-green-700 border border-green-200": testMessage.success,
									"bg-red-50 text-red-700 border border-red-200": !testMessage.success,
								})}
							>
								{testMessage.message}
							</div>
						)}
					</div>
				)}
				{mode === "update" && (
					<Button type="submit" className="w-full" loading={loading}>
						Save Changes
					</Button>
				)}
			</form>
		</Form>
	);
};