import { zodResolver } from "@hookform/resolvers/zod";
import { useEffect, useMemo, useRef } from "react";
import { Controller, useForm, useWatch } from "react-hook-form";
import { z } from "zod";
import { Button } from "@/components/ui/button";
import { DatePicker } from "@/components/ui/date-picker";
import {
  ResponsiveDialog,
  ResponsiveDialogContent,
  ResponsiveDialogFooter,
  ResponsiveDialogHeader,
  ResponsiveDialogTitle,
} from "@/components/ui/responsive-dialog";
import {
  Select,
  SelectContent,
  SelectItem,
  SelectTrigger,
  SelectValue,
} from "@/components/ui/select";
import { useExpensesContext } from "../ExpensesProvider";
import { useQuotes, useWallets } from "@/lib/queries";

const newExpenseSchema = z.object({
  description: z
    .string()
    .min(1, "La descripción es obligatoria")
    .max(50, "Máximo 50 caracteres"),
  amount: z.number().positive("El monto debe ser mayor a 0"),
  dueDate: z.date({ message: "La fecha es obligatoria" }),
  walletId: z.number().positive("Seleccioná una billetera"),
  usdcConversionRate: z
    .number()
    .positive("La cotización debe ser mayor a 0")
    .optional(),
});

type NewExpenseFormValues = z.infer<typeof newExpenseSchema>;

interface NewExpenseDialogProps {
  open: boolean;
  onOpenChange: (open: boolean) => void;
}

export function NewExpenseDialog({
  open,
  onOpenChange,
}: NewExpenseDialogProps) {
  const { createNonPeriodicExpense } = useExpensesContext();
  const prevWalletRef = useRef(0);
  const { data: wallets } = useWallets();
  const { data: quotes } = useQuotes();
  const defaultWallet = useMemo(
    () => wallets?.find((w) => w.isDefault),
    [wallets],
  );

  const beloSell = useMemo(() => {
    if (!quotes) return null;
    const belo = quotes.find((q) => q.type === "BELO");
    return belo ? Number(belo.sell) : null;
  }, [quotes]);

  const {
    register,
    handleSubmit,
    control,
    formState: { errors, isSubmitting },
    reset,
    setValue,
    trigger,
  } = useForm<NewExpenseFormValues>({
    resolver: zodResolver(newExpenseSchema),
    defaultValues: {
      description: "",
      amount: 0,
      dueDate: new Date(),
      walletId: defaultWallet?.id ?? 0,
      usdcConversionRate: undefined,
    },
  });

  useEffect(() => {
    if (defaultWallet && !open) return;
    if (defaultWallet) {
      setValue("walletId", defaultWallet.id);
      prevWalletRef.current = defaultWallet.id;
    }
  }, [defaultWallet, setValue, open]);

  const usdcConversionRateField = register("usdcConversionRate", {
    setValueAs: (value) => (value === "" ? undefined : Number(value)),
  });

  const selectedWalletId = useWatch({ control, name: "walletId" });
  const selectedWallet = wallets?.find((w) => w.id === selectedWalletId);
  const isUSDC = selectedWallet?.currency === "USDC";

  useEffect(() => {
    if (
      isUSDC &&
      beloSell &&
      selectedWalletId &&
      selectedWalletId !== prevWalletRef.current
    ) {
      setValue("usdcConversionRate", beloSell);
      trigger("usdcConversionRate");
      prevWalletRef.current = selectedWalletId;
    }
  }, [isUSDC, beloSell, selectedWalletId, setValue, trigger]);

  function handleClose(open: boolean) {
    onOpenChange(open);
    if (!open) reset();
  }

  async function onSubmit(data: NewExpenseFormValues) {
    await createNonPeriodicExpense({
      description: data.description,
      amount: data.amount,
      dueDate: data.dueDate.toISOString(),
      walletId: data.walletId,
      usdcConversionRate: isUSDC ? data.usdcConversionRate : undefined,
    });
    handleClose(false);
  }

  return (
    <ResponsiveDialog open={open} onOpenChange={handleClose}>
      <ResponsiveDialogContent>
        <ResponsiveDialogHeader>
          <ResponsiveDialogTitle>Nuevo gasto</ResponsiveDialogTitle>
        </ResponsiveDialogHeader>

        <form
          id="new-expense-form"
          onSubmit={handleSubmit(onSubmit)}
          className="flex flex-col gap-4"
        >
          <div className="flex flex-col gap-1.5">
            <label htmlFor="description" className="text-sm font-medium">
              Descripción
            </label>
            <input
              id="description"
              {...register("description")}
              placeholder="Ej: Ropa"
              className="h-8 rounded-lg border border-input bg-background px-2.5 text-sm outline-none focus-visible:border-ring focus-visible:ring-3 focus-visible:ring-ring/50"
            />
            {errors.description && (
              <span className="text-xs text-destructive">
                {errors.description.message}
              </span>
            )}
          </div>

          <div className="flex flex-col gap-1.5">
            <label htmlFor="amount" className="text-sm font-medium">
              Monto
            </label>
            <input
              id="amount"
              type="number"
              step="0.01"
              min="0"
              {...register("amount", { valueAsNumber: true })}
              placeholder="2500"
              className="h-8 rounded-lg border border-input bg-background px-2.5 text-sm outline-none focus-visible:border-ring focus-visible:ring-3 focus-visible:ring-ring/50"
            />
            {errors.amount && (
              <span className="text-xs text-destructive">
                {errors.amount.message}
              </span>
            )}
          </div>

          <div className="flex flex-col gap-1.5">
            <label className="text-sm font-medium">Billetera</label>
            <Controller
              control={control}
              name="walletId"
              render={({ field }) => (
                <Select
                  value={field.value ? String(field.value) : ""}
                  onValueChange={(v) => field.onChange(Number(v))}
                >
                  <SelectTrigger className="h-8 w-full">
                    <SelectValue placeholder="Seleccionar billetera" />
                  </SelectTrigger>
                  <SelectContent>
                    {wallets?.map((w) => (
                      <SelectItem key={w.id} value={String(w.id)}>
                        {w.name} ({w.currency})
                      </SelectItem>
                    ))}
                  </SelectContent>
                </Select>
              )}
            />
            {errors.walletId && (
              <span className="text-xs text-destructive">
                {errors.walletId.message}
              </span>
            )}
          </div>

          {isUSDC && (
            <div className="flex flex-col gap-1.5">
              <label
                htmlFor="usdcConversionRate"
                className="text-sm font-medium"
              >
                Cotización USDC (ARS por USDC)
              </label>
              <input
                id="usdcConversionRate"
                type="text"
                inputMode="decimal"
                {...usdcConversionRateField}
                className="h-8 rounded-lg border border-input bg-background px-2.5 text-sm outline-none focus-visible:border-ring focus-visible:ring-3 focus-visible:ring-ring/50"
              />
              {errors.usdcConversionRate && (
                <span className="text-xs text-destructive">
                  {errors.usdcConversionRate.message}
                </span>
              )}
              <span className="text-xs text-muted-foreground">
                Pre-sugerido: {beloSell ?? "—"} ARS/USDC (BELO)
              </span>
            </div>
          )}

          <div className="flex flex-col gap-1.5">
            <span className="text-sm font-medium">Fecha del gasto</span>
            <Controller
              control={control}
              name="dueDate"
              render={({ field }) => (
                <DatePicker
                  value={field.value}
                  onChange={field.onChange}
                  placeholder="Seleccionar fecha"
                />
              )}
            />
            {errors.dueDate && (
              <span className="text-xs text-destructive">
                {errors.dueDate.message}
              </span>
            )}
          </div>
        </form>

        <ResponsiveDialogFooter>
          <Button
            variant="outline"
            onClick={() => handleClose(false)}
            disabled={isSubmitting}
          >
            Cancelar
          </Button>
          <Button type="submit" form="new-expense-form" disabled={isSubmitting}>
            Crear gasto
          </Button>
        </ResponsiveDialogFooter>
      </ResponsiveDialogContent>
    </ResponsiveDialog>
  );
}
