website/components/ForgotPasswordDialog.tsx

"use client";

import { useState } from "react";
import { Button } from "@/components/ui/button";
import { useAppTheme } from "@/components/ThemeProvider";
import { Input } from "@/components/ui/input";
import {
  InputOTP,
  InputOTPGroup,
  InputOTPSlot,
} from "@/components/ui/input-otp";
import {
  Dialog,
  DialogContent,
  DialogDescription,
  DialogHeader,
  DialogTitle,
} from "@/components/ui/dialog";
import { Eye, EyeOff, Loader2, X, CheckCircle2 } from "lucide-react";
import { useAuth } from "@/hooks/useAuth";
import { 
  forgotPasswordSchema, 
  verifyPasswordResetOtpSchema, 
  resetPasswordSchema,
  type ForgotPasswordInput, 
  type VerifyPasswordResetOtpInput,
  type ResetPasswordInput 
} from "@/lib/schema/auth";
import { toast } from "sonner";

interface ForgotPasswordDialogProps {
  open: boolean;
  onOpenChange: (open: boolean) => void;
  onSuccess?: () => void;
}

type Step = "request" | "verify" | "reset";

export function ForgotPasswordDialog({ open, onOpenChange, onSuccess }: ForgotPasswordDialogProps) {
  const { theme } = useAppTheme();
  const isDark = theme === "dark";
  const { 
    forgotPasswordMutation, 
    verifyPasswordResetOtpMutation, 
    resetPasswordMutation,
    resendOtpMutation 
  } = useAuth();
  const [step, setStep] = useState<Step>("request");
  const [email, setEmail] = useState("");
  const [otpData, setOtpData] = useState<VerifyPasswordResetOtpInput>({
    email: "",
    otp: "",
  });
  const [resetData, setResetData] = useState<ResetPasswordInput>({
    email: "",
    otp: "",
    new_password: "",
    confirm_password: "",
  });
  const [showPassword, setShowPassword] = useState(false);
  const [showPassword2, setShowPassword2] = useState(false);

  const handleRequestOtp = async (e: React.FormEvent) => {
    e.preventDefault();
    
    const validation = forgotPasswordSchema.safeParse({ email });
    if (!validation.success) {
      const firstError = validation.error.issues[0];
      toast.error(firstError.message);
      return;
    }
    
    try {
      await forgotPasswordMutation.mutateAsync({ email });
      setOtpData({ email, otp: "" });
      setResetData({ email, otp: "", new_password: "", confirm_password: "" });
      setStep("verify");
      toast.success("Password reset OTP sent! Please check your email.");
    } catch (err) {
      const errorMessage = err instanceof Error ? err.message : "Failed to send OTP. Please try again.";
      toast.error(errorMessage);
    }
  };

  const handleVerifyOtp = async (e: React.FormEvent) => {
    e.preventDefault();
    
    const emailToVerify = email || otpData.email;
    if (!emailToVerify) {
      toast.error("Email is required");
      return;
    }
    
    const validation = verifyPasswordResetOtpSchema.safeParse({
      email: emailToVerify,
      otp: otpData.otp,
    });
    
    if (!validation.success) {
      const firstError = validation.error.issues[0];
      toast.error(firstError.message);
      return;
    }
    
    try {
      await verifyPasswordResetOtpMutation.mutateAsync({
        email: emailToVerify,
        otp: otpData.otp,
      });
      setResetData({ 
        email: emailToVerify, 
        otp: otpData.otp, 
        new_password: "", 
        confirm_password: "" 
      });
      setStep("reset");
      toast.success("OTP verified! Please set your new password.");
    } catch (err) {
      const errorMessage = err instanceof Error ? err.message : "OTP verification failed. Please try again.";
      toast.error(errorMessage);
    }
  };

  const handleResetPassword = async (e: React.FormEvent) => {
    e.preventDefault();
    
    const validation = resetPasswordSchema.safeParse(resetData);
    if (!validation.success) {
      const firstError = validation.error.issues[0];
      toast.error(firstError.message);
      return;
    }
    
    try {
      await resetPasswordMutation.mutateAsync(resetData);
      toast.success("Password reset successful! Please log in with your new password.");
      handleDialogChange(false);
      if (onSuccess) {
        onSuccess();
      }
    } catch (err) {
      const errorMessage = err instanceof Error ? err.message : "Password reset failed. Please try again.";
      toast.error(errorMessage);
    }
  };

  const handleResendOtp = async () => {
    const emailToResend = email || otpData.email;
    if (!emailToResend) {
      toast.error("Email is required");
      return;
    }
    
    try {
      await resendOtpMutation.mutateAsync({ 
        email: emailToResend, 
        context: "password_reset" 
      });
      toast.success("OTP resent successfully! Please check your email.");
    } catch (err) {
      const errorMessage = err instanceof Error ? err.message : "Failed to resend OTP";
      toast.error(errorMessage);
    }
  };

  const handleOtpChange = (field: keyof VerifyPasswordResetOtpInput, value: string) => {
    setOtpData((prev) => ({ ...prev, [field]: value }));
  };

  // Reset step when dialog closes
  const handleDialogChange = (isOpen: boolean) => {
    if (!isOpen) {
      setStep("request");
      setEmail("");
      setOtpData({ email: "", otp: "" });
      setResetData({ email: "", otp: "", new_password: "", confirm_password: "" });
    }
    onOpenChange(isOpen);
  };

  return (
    <Dialog open={open} onOpenChange={handleDialogChange}>
      <DialogContent 
        showCloseButton={false}
        className={`max-w-md max-h-[90vh] overflow-hidden flex flex-col p-0 ${isDark ? 'bg-gray-800 border-gray-700' : 'bg-white border-gray-200'}`}
      >
        {/* Header with Close Button - Fixed */}
        <div className="flex items-start justify-between p-6 pb-4 flex-shrink-0 border-b border-gray-200 dark:border-gray-700">
          <DialogHeader className="flex-1 pr-2">
            <DialogTitle className="text-2xl sm:text-3xl font-bold bg-gradient-to-r from-rose-600 via-pink-600 to-rose-600 bg-clip-text text-transparent">
              {step === "request" && "Reset Password"}
              {step === "verify" && "Verify OTP"}
              {step === "reset" && "Set New Password"}
            </DialogTitle>
            <DialogDescription className={`text-sm sm:text-base mt-1 ${isDark ? 'text-gray-400' : 'text-gray-600'}`}>
              {step === "request" && "Enter your email to receive a password reset code"}
              {step === "verify" && "Enter the verification code sent to your email"}
              {step === "reset" && "Enter your new password"}
            </DialogDescription>
          </DialogHeader>
          {/* Close Button */}
          <Button
            type="button"
            variant="ghost"
            size="icon"
            onClick={() => handleDialogChange(false)}
            className={`flex-shrink-0 w-8 h-8 rounded-full ${isDark ? 'text-gray-400 hover:text-gray-300 hover:bg-gray-700' : 'text-gray-500 hover:text-gray-700 hover:bg-gray-100'}`}
            aria-label="Close"
          >
            <X className="w-5 h-5" />
          </Button>
        </div>

        {/* Scrollable Content */}
        <div className="overflow-y-auto flex-1 px-6">
          {/* Request OTP Form */}
          {step === "request" && (
            <form className="space-y-4 sm:space-y-5 py-4 sm:py-6" onSubmit={handleRequestOtp}>
              <div className="space-y-1.5 sm:space-y-2">
                <label htmlFor="forgot-email" className={`text-sm font-medium ${isDark ? 'text-gray-300' : 'text-black'}`}>
                  Email address *
                </label>
                <Input
                  id="forgot-email"
                  type="email"
                  placeholder="Email address"
                  value={email}
                  onChange={(e) => setEmail(e.target.value)}
                  className={`h-11 sm:h-12 text-sm sm:text-base ${isDark ? 'bg-gray-700 border-gray-600 text-white placeholder:text-gray-400' : 'bg-white border-gray-300 text-gray-900'}`}
                  required
                />
              </div>

              <Button
                type="submit"
                disabled={forgotPasswordMutation.isPending}
                className="w-full h-11 sm:h-12 text-sm sm:text-base font-semibold bg-gradient-to-r from-rose-500 to-pink-600 hover:from-rose-600 hover:to-pink-700 text-white shadow-lg hover:shadow-xl transition-all disabled:opacity-50 disabled:cursor-not-allowed mt-4 sm:mt-6"
              >
                {forgotPasswordMutation.isPending ? (
                  <>
                    <Loader2 className="w-4 h-4 mr-2 animate-spin" />
                    Sending...
                  </>
                ) : (
                  "Send Reset Code"
                )}
              </Button>
            </form>
          )}

          {/* Verify OTP Form */}
          {step === "verify" && (
            <form className="space-y-4 sm:space-y-5 py-4 sm:py-6" onSubmit={handleVerifyOtp}>
              <div className={`p-3 sm:p-4 rounded-lg border ${isDark ? 'bg-blue-900/20 border-blue-800' : 'bg-blue-50 border-blue-200'}`}>
                <div className="flex items-start gap-3">
                  <CheckCircle2 className={`w-5 h-5 mt-0.5 flex-shrink-0 ${isDark ? 'text-blue-400' : 'text-blue-600'}`} />
                  <div>
                    <p className={`text-sm font-medium ${isDark ? 'text-blue-200' : 'text-blue-900'}`}>
                      Check your email
                    </p>
                    <p className={`text-xs sm:text-sm mt-1 ${isDark ? 'text-blue-300' : 'text-blue-700'}`}>
                      We've sent a 6-digit verification code to {email || otpData.email || "your email address"}.
                    </p>
                  </div>
                </div>
              </div>

              {/* Email Field (if not set) */}
              {!email && (
                <div className="space-y-1.5 sm:space-y-2">
                  <label htmlFor="verify-email" className={`text-sm font-medium ${isDark ? 'text-gray-300' : 'text-black'}`}>
                    Email address *
                  </label>
                  <Input
                    id="verify-email"
                    type="email"
                    placeholder="Email address"
                    value={otpData.email}
                    onChange={(e) => handleOtpChange("email", e.target.value)}
                    className={`h-11 sm:h-12 text-sm sm:text-base ${isDark ? 'bg-gray-700 border-gray-600 text-white placeholder:text-gray-400' : 'bg-white border-gray-300 text-gray-900'}`}
                    required
                  />
                </div>
              )}

              {/* OTP Field */}
              <div className="space-y-1.5 sm:space-y-2">
                <label className={`text-sm font-medium ${isDark ? 'text-gray-300' : 'text-black'}`}>
                  Verification Code *
                </label>
                <div className="flex justify-center">
                  <InputOTP
                    maxLength={6}
                    value={otpData.otp}
                    onChange={(value) => handleOtpChange("otp", value)}
                  >
                    <InputOTPGroup className="gap-2 sm:gap-3">
                      <InputOTPSlot index={0} className="h-12 w-12 sm:h-14 sm:w-14 text-lg sm:text-xl font-semibold" />
                      <InputOTPSlot index={1} className="h-12 w-12 sm:h-14 sm:w-14 text-lg sm:text-xl font-semibold" />
                      <InputOTPSlot index={2} className="h-12 w-12 sm:h-14 sm:w-14 text-lg sm:text-xl font-semibold" />
                      <InputOTPSlot index={3} className="h-12 w-12 sm:h-14 sm:w-14 text-lg sm:text-xl font-semibold" />
                      <InputOTPSlot index={4} className="h-12 w-12 sm:h-14 sm:w-14 text-lg sm:text-xl font-semibold" />
                      <InputOTPSlot index={5} className="h-12 w-12 sm:h-14 sm:w-14 text-lg sm:text-xl font-semibold" />
                    </InputOTPGroup>
                  </InputOTP>
                </div>
              </div>

              {/* Resend OTP */}
              <div className="text-center">
                <Button
                  type="button"
                  variant="link"
                  onClick={handleResendOtp}
                  disabled={resendOtpMutation?.isPending}
                  className={`h-auto p-0 text-xs sm:text-sm font-medium ${isDark ? 'text-blue-400 hover:text-blue-300' : 'text-blue-600 hover:text-blue-700'}`}
                >
                  {resendOtpMutation?.isPending ? "Sending..." : "Didn't receive the code? Resend"}
                </Button>
              </div>

              {/* Submit Button */}
              <Button
                type="submit"
                disabled={verifyPasswordResetOtpMutation.isPending}
                className="w-full h-11 sm:h-12 text-sm sm:text-base font-semibold bg-gradient-to-r from-rose-500 to-pink-600 hover:from-rose-600 hover:to-pink-700 text-white shadow-lg hover:shadow-xl transition-all disabled:opacity-50 disabled:cursor-not-allowed mt-4 sm:mt-6"
              >
                {verifyPasswordResetOtpMutation.isPending ? (
                  <>
                    <Loader2 className="w-4 h-4 mr-2 animate-spin" />
                    Verifying...
                  </>
                ) : (
                  "Verify Code"
                )}
              </Button>

              {/* Back to request */}
              <div className="text-center">
                <Button
                  type="button"
                  variant="link"
                  onClick={() => {
                    setStep("request");
                    setOtpData({ email: "", otp: "" });
                  }}
                  className={`h-auto p-0 text-xs sm:text-sm font-medium ${isDark ? 'text-gray-400 hover:text-gray-300' : 'text-gray-600 hover:text-gray-700'}`}
                >
                  ← Back
                </Button>
              </div>
            </form>
          )}

          {/* Reset Password Form */}
          {step === "reset" && (
            <form className="space-y-4 sm:space-y-5 py-4 sm:py-6" onSubmit={handleResetPassword}>
              {/* New Password Field */}
              <div className="space-y-1.5 sm:space-y-2">
                <label htmlFor="reset-password" className={`text-sm font-medium ${isDark ? 'text-gray-300' : 'text-black'}`}>
                  New Password *
                </label>
                <div className="relative">
                  <Input
                    id="reset-password"
                    type={showPassword ? "text" : "password"}
                    placeholder="New password (min 8 characters)"
                    value={resetData.new_password}
                    onChange={(e) => setResetData({ ...resetData, new_password: e.target.value })}
                    className={`h-11 sm:h-12 pr-12 text-sm sm:text-base ${isDark ? 'bg-gray-700 border-gray-600 text-white placeholder:text-gray-400' : 'bg-white border-gray-300 text-gray-900'}`}
                    required
                  />
                  <Button
                    type="button"
                    variant="ghost"
                    size="icon"
                    onClick={() => setShowPassword(!showPassword)}
                    className={`absolute right-3 sm:right-4 top-1/2 -translate-y-1/2 h-auto w-auto p-0 ${isDark ? 'text-gray-400 hover:text-gray-300' : 'text-gray-500 hover:text-gray-700'}`}
                    aria-label={showPassword ? "Hide password" : "Show password"}
                  >
                    {showPassword ? (
                      <EyeOff className="w-4 h-4 sm:w-5 sm:h-5" />
                    ) : (
                      <Eye className="w-4 h-4 sm:w-5 sm:h-5" />
                    )}
                  </Button>
                </div>
              </div>

              {/* Confirm Password Field */}
              <div className="space-y-1.5 sm:space-y-2">
                <label htmlFor="reset-password2" className={`text-sm font-medium ${isDark ? 'text-gray-300' : 'text-black'}`}>
                  Confirm New Password *
                </label>
                <div className="relative">
                  <Input
                    id="reset-password2"
                    type={showPassword2 ? "text" : "password"}
                    placeholder="Confirm new password"
                    value={resetData.confirm_password}
                    onChange={(e) => setResetData({ ...resetData, confirm_password: e.target.value })}
                    className={`h-11 sm:h-12 pr-12 text-sm sm:text-base ${isDark ? 'bg-gray-700 border-gray-600 text-white placeholder:text-gray-400' : 'bg-white border-gray-300 text-gray-900'}`}
                    required
                  />
                  <Button
                    type="button"
                    variant="ghost"
                    size="icon"
                    onClick={() => setShowPassword2(!showPassword2)}
                    className={`absolute right-3 sm:right-4 top-1/2 -translate-y-1/2 h-auto w-auto p-0 ${isDark ? 'text-gray-400 hover:text-gray-300' : 'text-gray-500 hover:text-gray-700'}`}
                    aria-label={showPassword2 ? "Hide password" : "Show password"}
                  >
                    {showPassword2 ? (
                      <EyeOff className="w-4 h-4 sm:w-5 sm:h-5" />
                    ) : (
                      <Eye className="w-4 h-4 sm:w-5 sm:h-5" />
                    )}
                  </Button>
                </div>
              </div>

              {/* Submit Button */}
              <Button
                type="submit"
                disabled={resetPasswordMutation.isPending}
                className="w-full h-11 sm:h-12 text-sm sm:text-base font-semibold bg-gradient-to-r from-rose-500 to-pink-600 hover:from-rose-600 hover:to-pink-700 text-white shadow-lg hover:shadow-xl transition-all disabled:opacity-50 disabled:cursor-not-allowed mt-4 sm:mt-6"
              >
                {resetPasswordMutation.isPending ? (
                  <>
                    <Loader2 className="w-4 h-4 mr-2 animate-spin" />
                    Resetting password...
                  </>
                ) : (
                  "Reset Password"
                )}
              </Button>

              {/* Back to verify */}
              <div className="text-center">
                <Button
                  type="button"
                  variant="link"
                  onClick={() => {
                    setStep("verify");
                    setResetData({ ...resetData, new_password: "", confirm_password: "" });
                  }}
                  className={`h-auto p-0 text-xs sm:text-sm font-medium ${isDark ? 'text-gray-400 hover:text-gray-300' : 'text-gray-600 hover:text-gray-700'}`}
                >
                  ← Back
                </Button>
              </div>
            </form>
          )}
        </div>
      </DialogContent>
    </Dialog>
  );
}