import { SelectContent } from "@/components/ui/select";
import { SelectValue } from "@/components/ui/select";
import { SelectTrigger } from "@/components/ui/select";
import { DialogFooter } from "@/components/ui/dialog";
import { Reward, RewardType } from "@autumn/shared";
import { Select, SelectItem } from "@/components/ui/select";
import { DialogContent, DialogTitle } from "@/components/ui/dialog";
import { useState } from "react";
import { Button } from "@/components/ui/button";
import { useCustomerContext } from "../CustomerContext";
import { getBackendErr } from "@/utils/genUtils";
import { toast } from "sonner";
import { CusService } from "@/services/customers/CusService";
import { useAxiosInstance } from "@/services/useAxiosInstance";
import { getOriginalCouponId } from "@/utils/product/couponUtils";
import { WarningBox } from "@/components/general/modal-components/WarningBox";
import { useRewardsQuery } from "@/hooks/queries/useRewardsQuery";
import { useCusQuery } from "../hooks/useCusQuery";
import { useCusReferralQuery } from "../hooks/useCusReferralQuery";

const AddCouponDialogContent = ({
  setOpen,
}: {
  setOpen: (open: boolean) => void;
}) => {
  const { stripeCus } = useCusReferralQuery();
  const { customer, refetch } = useCusQuery();

  const [couponSelected, setCouponSelected] = useState<Reward | null>(null);
  const [loading, setLoading] = useState(false);
  const axiosInstance = useAxiosInstance();

  const { rewards } = useRewardsQuery();

  const handleAddClicked = async () => {
    try {
      setLoading(true);
      await CusService.addCouponToCustomer({
        axios: axiosInstance,
        customer_id: customer.id,
        coupon_id: couponSelected!.internal_id,
      });
      setOpen(false);
      await refetch();
      toast.success("Reward added to customer");
    } catch (error) {
      toast.error(getBackendErr(error, "Failed to create coupon"));
    } finally {
      setLoading(false);
    }
  };

  const existingDiscount = stripeCus?.discount;

  const getExistingCoupon = () => {
    if (existingDiscount) {
      return rewards.find(
        (c: Reward) => c.id === getOriginalCouponId(existingDiscount.coupon.id)
      );
    } else {
      return null;
    }
  };

  if (!rewards) return null;

  return (
    <DialogContent className="min-w-sm max-w-md">
      <DialogTitle>Add Reward</DialogTitle>
      {getExistingCoupon() && (
        <WarningBox>
          Reward {getExistingCoupon()?.name} already applied. Adding a new one
          will replace the existing one.
        </WarningBox>
      )}
      <div>
        <Select
          value={couponSelected?.internal_id}
          onValueChange={(value) => {
            const coupon = rewards.find((c: Reward) => c.internal_id === value);
            if (coupon) {
              setCouponSelected(coupon);
            }
          }}
        >
          <SelectTrigger>
            <SelectValue placeholder="Select Reward" />
          </SelectTrigger>
          <SelectContent>
            {/* If empty */}

            {rewards && rewards.length > 0 ? (
              rewards.map((coupon: Reward) => {
                if (coupon.type == RewardType.FreeProduct) return null;
                return (
                  <SelectItem
                    key={coupon.internal_id}
                    value={coupon.internal_id}
                  >
                    {coupon.name}
                  </SelectItem>
                );
              })
            ) : (
              <SelectItem value="none" disabled>
                No coupons found
              </SelectItem>
            )}
          </SelectContent>
        </Select>
      </div>
      <DialogFooter>
        <Button
          variant="gradientPrimary"
          onClick={() => handleAddClicked()}
          disabled={!couponSelected}
          isLoading={loading}
        >
          Add Reward
        </Button>
      </DialogFooter>
    </DialogContent>
  );
};

export default AddCouponDialogContent;