"use client";

import { useState, useTransition } from "react";
import { toast } from "sonner";
import { Pencil, Plus } from "lucide-react";
import { createPlanSub, updatePlanSub, togglePlanSubActive } from "./actions";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Label } from "@/components/ui/label";
import { Switch } from "@/components/ui/switch";
import {
  Select,
  SelectContent,
  SelectItem,
  SelectTrigger,
  SelectValue,
} from "@/components/ui/select";
import {
  Dialog,
  DialogContent,
  DialogHeader,
  DialogTitle,
  DialogTrigger,
} from "@/components/ui/dialog";
import type { PlanMain, PlanSub } from "@/lib/types";

const NONE = "__none__";

export function PlanSubRowToggle({ id, isActive }: { id: string; isActive: boolean }) {
  const [pending, start] = useTransition();
  return (
    <Switch
      checked={isActive}
      disabled={pending}
      onCheckedChange={(next) =>
        start(async () => {
          await togglePlanSubActive(id, next);
          toast.success(next ? "Activated" : "Deactivated");
        })
      }
    />
  );
}

export function PlanSubDialog({
  plans,
  planSub,
}: {
  plans: PlanMain[];
  planSub?: PlanSub;
}) {
  const editing = !!planSub;
  const [open, setOpen] = useState(false);
  const [planId, setPlanId] = useState(planSub?.plan_id ?? NONE);
  const [pending, startTransition] = useTransition();

  const action = editing ? updatePlanSub : createPlanSub;

  function handleOpenChange(next: boolean) {
    if (next) {
      // Reset the controlled select to the source value each time the dialog opens.
      setPlanId(planSub?.plan_id ?? NONE);
    }
    setOpen(next);
  }

  function onSubmit(formData: FormData) {
    startTransition(async () => {
      const res = await action(undefined, formData);
      if (res?.ok) {
        toast.success(editing ? "Sub plan updated" : "Sub plan created");
        setOpen(false);
      } else if (res?.error) {
        toast.error(res.error);
      }
    });
  }

  return (
    <Dialog open={open} onOpenChange={handleOpenChange}>
      <DialogTrigger asChild>
        {editing ? (
          <Button size="icon" variant="ghost">
            <Pencil className="size-4" />
          </Button>
        ) : (
          <Button>
            <Plus className="size-4" /> Add Sub Plan
          </Button>
        )}
      </DialogTrigger>
      <DialogContent>
        <DialogHeader>
          <DialogTitle>{editing ? "Edit Sub Plan" : "Add Sub Plan"}</DialogTitle>
        </DialogHeader>
        <form action={onSubmit} className="grid grid-cols-2 gap-4">
          {editing && <input type="hidden" name="id" value={planSub.id} />}
          <input type="hidden" name="plan_id" value={planId} />

          <div className="col-span-2 space-y-2">
            <Label htmlFor="plan_id">Plan *</Label>
            <Select value={planId} onValueChange={setPlanId}>
              <SelectTrigger id="plan_id">
                <SelectValue placeholder="Select plan" />
              </SelectTrigger>
              <SelectContent>
                <SelectItem value={NONE}>Select plan</SelectItem>
                {plans.map((p) => (
                  <SelectItem key={p.id} value={p.id}>
                    {p.name}
                  </SelectItem>
                ))}
              </SelectContent>
            </Select>
          </div>

          <div className="space-y-2">
            <Label htmlFor="name">Sub plan name *</Label>
            <Input id="name" name="name" defaultValue={planSub?.name} required />
          </div>

          <div className="space-y-2">
            <Label htmlFor="amount">Amount</Label>
            <Input
              id="amount"
              name="amount"
              type="number"
              step="any"
              min="0"
              defaultValue={planSub?.amount ?? ""}
            />
          </div>

          <div className="col-span-2 flex justify-end">
            <Button type="submit" disabled={pending}>
              {pending ? "Saving…" : editing ? "Save changes" : "Save sub plan"}
            </Button>
          </div>
        </form>
      </DialogContent>
    </Dialog>
  );
}