import { useState, useEffect } from "react"
import { Link, useNavigate } from "react-router"
import { FormInput } from "@/components/shared/form-input"
import { FormButton } from "@/components/shared/form-button"
import { useAuth } from "@/app/context/auth-context"
import { register } from "@/lib/api"

export default function RegisterPage() {
    const navigate = useNavigate()
    const { user, loading: authLoading } = useAuth()

    useEffect(() => {
        if (!authLoading && user) {
            navigate("/forms")
        }
    }, [user, authLoading, navigate])

    const [form, setForm] = useState({
        firstName: "",
        lastName: "",
        email: "",
        password: "",
        confirmPassword: "",
    })

    const [errors, setErrors] = useState<Partial<typeof form>>({})

    function handleChange(e: React.ChangeEvent<HTMLInputElement>) {
        setForm((prev) => ({ ...prev, [e.target.name]: e.target.value }))
        setErrors((prev) => ({ ...prev, [e.target.name]: "" }))
    }

    function validate() {
        const newErrors: Partial<typeof form> = {}

        if (!form.firstName.trim()) newErrors.firstName = "First name is required"
        if (!form.lastName.trim()) newErrors.lastName = "Last name is required"
        if (!form.email.trim()) newErrors.email = "Email is required"
        if (form.password.length < 8)
            newErrors.password = "Must be at least 8 characters"
        if (form.confirmPassword !== form.password)
            newErrors.confirmPassword = "Passwords do not match"

        return newErrors
    }

    async function handleSubmit(e: React.FormEvent) {
        e.preventDefault()
        const newErrors = validate()
        if (Object.keys(newErrors).length > 0) {
            setErrors(newErrors)
            return
        }

        try {
            await register(form.email, form.password)
            navigate("/login")
        } catch (err) {
            setErrors({ email: err instanceof Error ? err.message : "Something went wrong. Please try again." })
        }
    }

    return (
        <div className="flex min-h-screen flex-col bg-background">
            <div className="flex flex-1 items-center justify-center px-4 py-12">
                <div className="w-full max-w-md">
                    <div className="mb-8 flex flex-col items-center gap-3">
                        <h1 className="text-2xl font-bold text-foreground">
                            Create your account
                        </h1>
                        <p className="text-sm text-muted-foreground">
                            Start building beautiful forms in minutes
                        </p>
                    </div>

                    <div className="rounded-xl border border-border bg-card p-6 shadow-sm">
                        <form onSubmit={handleSubmit} className="flex flex-col gap-4">
                            <div className="flex gap-3">
                                <FormInput
                                    label="First name"
                                    type="text"
                                    name="firstName"
                                    placeholder="bagas"
                                    required
                                    autoComplete="given-name"
                                    className="w-full"
                                    value={form.firstName}
                                    onChange={handleChange}
                                    error={errors.firstName}
                                />
                                <FormInput
                                    label="Last name"
                                    type="text"
                                    name="lastName"
                                    placeholder="pacil"
                                    required
                                    autoComplete="family-name"
                                    className="w-full"
                                    value={form.lastName}
                                    onChange={handleChange}
                                    error={errors.lastName}
                                />
                            </div>
                            <FormInput
                                label="Email"
                                type="email"
                                name="email"
                                placeholder="bagas@example.com"
                                required
                                autoComplete="email"
                                value={form.email}
                                onChange={handleChange}
                                error={errors.email}
                            />
                            <FormInput
                                label="Password"
                                type="password"
                                name="password"
                                placeholder="Create a password"
                                required
                                hint="Must be at least 8 characters"
                                autoComplete="new-password"
                                value={form.password}
                                onChange={handleChange}
                                error={errors.password}
                            />
                            <FormInput
                                label="Confirm password"
                                type="password"
                                name="confirmPassword"
                                placeholder="Confirm your password"
                                required
                                autoComplete="new-password"
                                value={form.confirmPassword}
                                onChange={handleChange}
                                error={errors.confirmPassword}
                            />

                            <FormButton type="submit" size="lg" className="mt-2 w-full">
                                Create account
                            </FormButton>
                        </form>
                    </div>

                    <p className="mt-6 text-center text-sm text-muted-foreground">
                        Already have an account?{" "}
                        <Link
                            to="/login"
                            className="font-medium text-primary hover:underline"
                        >
                            Sign in
                        </Link>
                    </p>
                </div>
            </div>
        </div>
    )
}