import { useState } from 'react';
import { motion, AnimatePresence } from 'framer-motion';
import type { Lang } from '@/lib/translations';
import { content } from '@/lib/translations';
import { ProgressBar } from '@/components/ProgressBar';
import { ApplicationForm, type FormData } from '@/components/ApplicationForm';
import { PaymentStep } from '@/components/PaymentStep';

const pageVariants = {
  enter: { opacity: 0, x: 10 },
  center: { opacity: 1, x: 0 },
  exit: { opacity: 0, x: -10 },
};

const Index = () => {
  const [lang, setLang] = useState<Lang>('en');
  const [step, setStep] = useState(1);
  const [_formData, setFormData] = useState<FormData | null>(null);
  const t = content[lang];

  const handleFormSubmit = (data: FormData) => {
    setFormData(data);
    setStep(2);
    window.scrollTo({ top: 0, behavior: 'smooth' });
  };

  const stepLabel = step === 1
    ? t.identity
    : t.pay.split('&')[0].trim();

  return (
    <div className="min-h-screen bg-background text-foreground selection:bg-primary/10">
      <nav className="flex justify-between items-center px-6 py-4 max-w-3xl mx-auto">
        <div className="flex items-center gap-2.5">
          <div className="w-8 h-8 bg-primary rounded-lg flex items-center justify-center text-primary-foreground font-bold text-sm">
            R
          </div>
          <span className="font-semibold tracking-tight text-foreground">ResidentLogic</span>
        </div>
        <button
          onClick={() => setLang(lang === 'en' ? 'es' : 'en')}
          className="text-xs font-bold uppercase tracking-widest text-muted-foreground hover:text-primary transition-colors px-3 py-1.5 rounded-md hover:bg-primary/5"
        >
          {lang === 'en' ? 'ESPAÑOL' : 'ENGLISH'}
        </button>
      </nav>

      <main className="max-w-xl mx-auto px-6 pb-24 pt-2">
        <ProgressBar step={step} totalSteps={2} lang={lang} stepLabel={stepLabel} />

        <AnimatePresence mode="wait">
          {step === 1 ? (
            <motion.div
              key="step1"
              variants={pageVariants}
              initial="enter"
              animate="center"
              exit="exit"
              transition={{ duration: 0.25, ease: [0.25, 0.1, 0.25, 1] }}
            >
              <ApplicationForm lang={lang} onSubmit={handleFormSubmit} />
            </motion.div>
          ) : (
            <motion.div
              key="step2"
              variants={pageVariants}
              initial="enter"
              animate="center"
              exit="exit"
              transition={{ duration: 0.25, ease: [0.25, 0.1, 0.25, 1] }}
            >
              <PaymentStep lang={lang} onBack={() => setStep(1)} />
            </motion.div>
          )}
        </AnimatePresence>
      </main>
    </div>
  );
};

export default Index;