"use client";

import { useEffect } from "react";
import { useForm } from "react-hook-form";
import { zodResolver } from "@hookform/resolvers/zod";
import { z } from "zod";
import {
  ShipmentTrackingDTOSchema,
  UpdateShipmentSchema,
} from "@/lib/db/schemas/api";
import { updateShipment } from "@/actions/sales";
import { Button } from "@/components/ui/button";
import { OrderReferenceField } from "@/components/dashboard/OrderReferenceField";
import {
  Dialog,
  DialogContent,
  DialogDescription,
  DialogHeader,
  DialogTitle,
} from "@/components/ui/dialog";
import {
  Form,
  FormControl,
  FormField,
  FormItem,
  FormLabel,
  FormMessage,
} from "@/components/ui/form";
import { Input } from "@/components/ui/input";
import {
  Select,
  SelectContent,
  SelectItem,
  SelectTrigger,
  SelectValue,
} from "@/components/ui/select";
import { useAction } from "@/hooks/useAction";
import { toast } from "sonner";

type ShipmentEditFormData = z.infer<typeof UpdateShipmentSchema>;

interface ShipmentEditFormProps {
  shipment: z.infer<typeof ShipmentTrackingDTOSchema> | null | undefined;
  open: boolean;
  onOpenChange: (open: boolean) => void;
}

export function ShipmentEditForm({
  shipment,
  open,
  onOpenChange,
}: ShipmentEditFormProps) {
  const { execute, isLoading, validationErrors } = useAction(updateShipment);

  const form = useForm<ShipmentEditFormData>({
    resolver: zodResolver(UpdateShipmentSchema),
    defaultValues: {
      orderId: shipment?.orderId || "",
      trackingNumber: shipment?.trackingNumber || "",
      carrier: shipment?.carrier || "",
      shipmentType: shipment?.shipmentType || undefined,
      status: shipment?.status || "",
    },
  });

  // Sync server-side validation errors
  useEffect(() => {
    if (validationErrors) {
      Object.keys(validationErrors).forEach((key) => {
        form.setError(key as any, {
          type: "server",
          message: validationErrors[key][0],
        });
      });
    }
  }, [validationErrors, form]);

  const onSubmit = async (data: ShipmentEditFormData) => {
    if (!shipment) return;
    const result = await execute({ id: shipment.id, data });
    if (result?.success) {
      toast.success("Sendung erfolgreich aktualisiert");
      onOpenChange(false);
    }
  };

  return (
    <Dialog open={open} onOpenChange={onOpenChange}>
      <DialogContent className="sm:max-w-[425px]">
        <DialogHeader>
          <DialogTitle>Sendung bearbeiten</DialogTitle>
          <DialogDescription>
            Aktualisieren Sie die Informationen der Sendung.
          </DialogDescription>
        </DialogHeader>
        <Form {...form}>
          <form onSubmit={form.handleSubmit(onSubmit)} className="space-y-4">
            <FormField
              control={form.control}
              name="orderId"
              render={({ field }) => (
                <FormItem>
                  <FormLabel>Bestellungs-ID (optional)</FormLabel>
                  <FormControl>
                    <OrderReferenceField
                      value={field.value || ""}
                      onChange={field.onChange}
                      placeholder="Bestellung suchen..."
                      disabled={isLoading}
                    />
                  </FormControl>
                  <FormMessage />
                </FormItem>
              )}
            />
            <FormField
              control={form.control}
              name="trackingNumber"
              render={({ field }) => (
                <FormItem>
                  <FormLabel>Sendungsnummer</FormLabel>
                  <FormControl>
                    <Input
                      placeholder="Tracking Nummer"
                      {...field}
                      value={field.value || ""}
                      disabled={isLoading}
                    />
                  </FormControl>
                  <FormMessage />
                </FormItem>
              )}
            />
            <FormField
              control={form.control}
              name="carrier"
              render={({ field }) => (
                <FormItem>
                  <FormLabel>Versanddienstleister</FormLabel>
                  <FormControl>
                    <Input
                      placeholder="z.B. DHL, UPS"
                      {...field}
                      value={field.value || ""}
                      disabled={isLoading}
                    />
                  </FormControl>
                  <FormMessage />
                </FormItem>
              )}
            />
            <FormField
              control={form.control}
              name="shipmentType"
              render={({ field }) => (
                <FormItem>
                  <FormLabel>Typ</FormLabel>
                  <Select
                    onValueChange={field.onChange}
                    value={field.value || undefined}
                    disabled={isLoading}
                  >
                    <FormControl>
                      <SelectTrigger>
                        <SelectValue placeholder="Typ auswählen" />
                      </SelectTrigger>
                    </FormControl>
                    <SelectContent>
                      <SelectItem value="Original">Original</SelectItem>
                      <SelectItem value="Return">Rücksendung</SelectItem>
                      <SelectItem value="Replacement">Ersatz</SelectItem>
                    </SelectContent>
                  </Select>
                  <FormMessage />
                </FormItem>
              )}
            />
            <FormField
              control={form.control}
              name="status"
              render={({ field }) => (
                <FormItem>
                  <FormLabel>Status</FormLabel>
                  <FormControl>
                    <Input
                      placeholder="z.B. Versendet, Zugestellt"
                      {...field}
                      value={field.value || ""}
                      disabled={isLoading}
                    />
                  </FormControl>
                  <FormMessage />
                </FormItem>
              )}
            />
            <div className="flex justify-end gap-2">
              <Button
                type="button"
                variant="outline"
                onClick={() => onOpenChange(false)}
                disabled={isLoading}
              >
                Abbrechen
              </Button>
              <Button type="submit" disabled={isLoading}>
                {isLoading ? "Speichern..." : "Speichern"}
              </Button>
            </div>
          </form>
        </Form>
      </DialogContent>
    </Dialog>
  );
}