import { useState } from 'react';
import { Save, X, Bed, Stethoscope, User, Loader2, ClipboardList } from 'lucide-react';
import { Card, CardContent } from '@/components/ui/card';
import { Button } from '@/components/ui/button';
import { Input } from '@/components/ui/input';
import { Label } from '@/components/ui/label';
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from '@/components/ui/select';
import { Badge } from '@/components/ui/badge';
import { toast } from 'sonner';
import type { Paciente, Cama, Grupo, Internacion } from '@/types';
import { getNombreProfesional, calcularEdad, computeSector, sortCamas } from '@/lib/utils';
import { useHospitalStore } from '@/hooks/useHospitalStore';

interface EditIngresoProps {
  internacion: Internacion;
  paciente?: Paciente;
  cama?: Cama;
  pacientes: Paciente[];
  camas: Cama[];
  grupos: Grupo[];
  onActualizarInternacion: (id: string, datos: Partial<Internacion>) => Promise<void> | void;
  onAgregarCama?: (cama: Omit<Cama, 'id'>) => Promise<string>;
  onVolver: () => void;
  getGrupoName: (grupoId: string | undefined) => string;
}

export function EditIngreso({
  internacion,
  pacientes,
  camas,
  grupos,
  onActualizarInternacion,
  onAgregarCama,
  onVolver,
  getGrupoName,
}: EditIngresoProps) {
  const { currentUser } = useHospitalStore();
  const [pacienteSeleccionado] = useState(internacion.pacienteId);
  const [camaSeleccionada, setCamaSeleccionada] = useState(internacion.camaId);
  const [grupoSeleccionada] = useState(internacion.grupoId);
  const [camaInput, setCamaInput] = useState('');
  const [busqueda, setBusqueda] = useState('');
  const [modoCama, setModoCama] = useState<'seleccionar' | 'escribir'>('seleccionar');
  const effectiveMedico = getNombreProfesional(currentUser);
  const [fechaIngresoHospital, setFechaIngresoHospital] = useState(internacion.fechaIngresoHospital || '');
  const [fechaIngresoClinica, setFechaIngresoClinica] = useState(internacion.fechaIngresoClinica || '');
  const [motivoConsulta, setMotivoConsulta] = useState(internacion.motivoConsulta || '');
  const [diagnosticoIngreso, setDiagnosticoIngreso] = useState(internacion.diagnosticoIngreso || '');
  const [enfermedadActual, setEnfermedadActual] = useState(internacion.enfermedadActual || '');
  const [antecedentesEnfermedadActual, setAntecedentesEnfermedadActual] = useState(internacion.antecedentesEnfermedadActual || '');
  const [apache, setApache] = useState(internacion.apache || '');
  const [derivacion, setDerivacion] = useState(internacion.derivacion || '');
  const [isSubmitting, setIsSubmitting] = useState(false);

  const selectedPaciente = pacientes.find(p => p.id === pacienteSeleccionado);
  const normalizeStr = (str?: string) => (str || '').normalize('NFD').replace(/[\u0300-\u036f]/g, '').trim().toLowerCase();
  const grupoFueraDeGrupo = grupos.find(a => normalizeStr(a.nombre) === 'fuera de grupo');
  const grupoFueraDeGrupoId = grupoFueraDeGrupo?.id || grupos[0]?.id || 'fuera-de-grupo';

  const sortedCamas = sortCamas(camas);

  const handleSubmit = async () => {
    if (!pacienteSeleccionado) {
      toast.error('Debe seleccionar un paciente');
      return;
    }

    if (!diagnosticoIngreso.trim()) {
      toast.error('Debe ingresar el diagnóstico de ingreso');
      return;
    }

    if (!enfermedadActual.trim()) {
      toast.error('Debe ingresar la enfermedad actual');
      return;
    }

    if (!effectiveMedico.trim()) {
      toast.error('Debe ingresar el médico ingresante');
      return;
    }

    setIsSubmitting(true);

    try {
      let camaId = '';
      let newGrupoId = '';

      if (modoCama === 'escribir') {
        if (!camaInput.trim()) {
          toast.error('Ingrese el número de cama');
          setIsSubmitting(false);
          return;
        }
        const numeroCama = camaInput.trim();
        const camaExistente = camas.find(c => c.numero === numeroCama);
        if (camaExistente) {
          camaId = camaExistente.id;
          newGrupoId = camaExistente.grupoId || grupoFueraDeGrupoId;
        } else if (onAgregarCama) {
          camaId = await onAgregarCama({
            numero: numeroCama,
            grupoId: grupoFueraDeGrupoId,
            tipo: 'General',
            estado: 'Ocupada'
          });
          newGrupoId = grupoFueraDeGrupoId;
        } else {
          toast.error('No se puede crear la cama');
          setIsSubmitting(false);
          return;
        }
      } else {
        if (camaSeleccionada && camaSeleccionada !== internacion.camaId) {
          const camaElegida = sortedCamas.find(c => c.id === camaSeleccionada);
          if (!camaElegida) {
            toast.error('Cama no encontrada');
            setIsSubmitting(false);
            return;
          }
          newGrupoId = camaElegida.grupoId || grupoFueraDeGrupoId;
          camaId = camaSeleccionada;
        } else {
          newGrupoId = grupoSeleccionada || internacion.grupoId || grupoFueraDeGrupoId;
          camaId = internacion.camaId;
        }
      }

      await onActualizarInternacion(internacion.id, {
        pacienteId: pacienteSeleccionado,
        camaId: camaId,
        grupoId: newGrupoId,
        medicoIngresante: effectiveMedico.trim(),
        diagnosticoIngreso: diagnosticoIngreso.trim(),
        motivoConsulta: motivoConsulta.trim() || undefined,
        enfermedadActual: enfermedadActual.trim(),
        antecedentesEnfermedadActual: antecedentesEnfermedadActual.trim() || undefined,
        fechaIngresoHospital: fechaIngresoHospital || undefined,
        fechaIngresoClinica: fechaIngresoClinica || undefined,
        apache: apache.trim() || undefined,
        derivacion: derivacion.trim() || undefined,
      });

      toast.success('Ingreso actualizado correctamente');
      onVolver();
    } catch (err) {
      console.error('Error al actualizar ingreso:', err);
      toast.error('Error al actualizar el ingreso');
    } finally {
      setIsSubmitting(false);
    }
  };

  return (
    <div className="space-y-6 dark:text-white">
      <div className="flex flex-col sm:flex-row items-start sm:items-center justify-between gap-4">
        <div className="flex items-center gap-4">

          <div>
            <h1 className="text-2xl font-bold text-gray-900 dark:text-white flex items-center gap-2">
              <User className="h-6 w-6 text-blue-600" />
              Editar Ingreso
            </h1>
            <p className="text-gray-500 dark:text-gray-400">Modificar datos de internación</p>
          </div>
        </div>
        <div className="flex gap-2 ml-auto sm:ml-0">
          <Button variant="secondary" onClick={() => onVolver()} disabled={isSubmitting}>
            <X className="h-4 w-4 mr-2" />
            Cancelar
          </Button>
          <Button onClick={handleSubmit} disabled={isSubmitting}>
            {isSubmitting ? <Loader2 className="h-4 w-4 mr-2 animate-spin" /> : <Save className="h-4 w-4 mr-2" />}
            Guardar Cambios
          </Button>
        </div>
      </div>

      <div className="grid grid-cols-1 sm:grid-cols-2 gap-4">
        {/* Columna 1: Datos del Paciente, Cama y Datos Adicionales */}
        <div className="space-y-6">
          <Card className="min-h-[200px]">
            <CardContent className="pt-6 space-y-4">
              <h3 className="font-semibold text-gray-900 dark:text-white flex items-center gap-2">
                <User className="h-4 w-4" />
                Datos del Paciente
              </h3>

              {selectedPaciente ? (
                <div className="space-y-2">
                  <div className="flex flex-wrap gap-2">
                    <Badge variant="outline" className="bg-green-50 text-green-700 dark:bg-green-900 dark:text-green-300 dark:border-green-700">
                      {selectedPaciente?.apellido}, {selectedPaciente?.nombre}
                    </Badge>
                    <Badge variant="outline" className="bg-primary/10 text-primary border-primary/30">
                      {selectedPaciente?.fechaNacimiento ? calcularEdad(selectedPaciente.fechaNacimiento) + ' años' : ''}
                    </Badge>
                    <Badge variant="outline" className="bg-blue-50 text-blue-700 border-blue-200">
                      DNI: {selectedPaciente?.dni}
                    </Badge>
                  </div>
                  <div className="flex flex-wrap gap-2">
                    <Badge variant="secondary">
                      HC: {selectedPaciente?.historiaClinica || 'N/A'}
                    </Badge>
                    <Badge variant="secondary">
                      {selectedPaciente?.obraSocial || 'Sin obra social'}
                    </Badge>
                    <Badge variant="secondary">
                      {selectedPaciente?.nacionalidad || 'N/A'}
                    </Badge>
                  </div>
                </div>
              ) : (
                <>
                  <Label>Buscar Paciente *</Label>
                  <Input
                    placeholder="Apellido, nombre o DNI"
                    value={busqueda}
                    onChange={(e) => setBusqueda(e.target.value)}
                  />
                  <div className="max-h-48 overflow-auto border rounded-md">
                    {(() => {
                      const q = busqueda.trim().toLowerCase();
                      if (!q) {
                        return <div className="p-2 text-sm text-gray-500">Escriba para buscar</div>;
                      }
                      const matches = pacientes.filter(p =>
                        p.apellido.toLowerCase().includes(q) ||
                        p.nombre.toLowerCase().includes(q) ||
                        p.dni.includes(q)
                      );
                      if (matches.length === 0) {
                        return <div className="p-2 text-sm text-gray-500">Sin resultados</div>;
                      }
                      return matches.map(p => (
                        <button
                          key={p.id}
                          type="button"
                          onClick={() => setPacienteSeleccionado(p.id)}
                          className="w-full text-left p-2 hover:bg-gray-50 text-sm"
                        >
                          {p.apellido}, {p.nombre} — DNI: {p.dni}
                        </button>
                      ));
                    })()}
                  </div>
                </>
              )}
            </CardContent>
          </Card>
        </div>

        {/* Columna 2: Cama y Grupo */}
        <div className="space-y-6">
          <Card className="min-h-[200px]">
            <CardContent className="pt-6 space-y-4">
              <h3 className="font-semibold text-gray-900 dark:text-white flex items-center gap-2">
                <Bed className="h-4 w-4" />
                Asignación de Cama y Grupo
              </h3>
              
              <div className="flex gap-2 mb-4">
                <Button 
                  variant={modoCama === 'seleccionar' ? 'default' : 'outline'} 
                  onClick={() => setModoCama('seleccionar')}
                >
                  Seleccionar Cama
                </Button>
                <Button 
                  variant={modoCama === 'escribir' ? 'default' : 'outline'} 
                  onClick={() => setModoCama('escribir')}
                >
                  Escribir Cama
                </Button>
              </div>

              {modoCama === 'seleccionar' ? (
                <>
                  <div>
                    <Label>Cama</Label>
                    <Select value={camaSeleccionada} onValueChange={setCamaSeleccionada}>
                      <SelectTrigger>
                        <SelectValue placeholder="Seleccionar cama" />
                      </SelectTrigger>
                      <SelectContent>
                        {sortedCamas.map(c => (
                          <SelectItem key={c.id} value={c.id}>
                            {c.numero} - {getGrupoName(c.grupoId)}
                          </SelectItem>
                        ))}
                      </SelectContent>
                    </Select>
                  </div>
                </>
              ) : (
                <div>
                  <Label>Número de Cama (formato XXX-YY) *</Label>
                  <Input 
                    placeholder="201-1" 
                    value={camaInput}
                    onChange={(e) => setCamaInput(e.target.value)}
                  />
                  <p className="text-xs text-gray-500 mt-1">
                    {camaInput.trim() ? `Clasificación: ${computeSector(camaInput.trim())}` : 'Formato XXX-YY (Salas 3XX impares y 4XX son Fuera de Área)'}
                  </p>
                </div>
              )}
            </CardContent>
          </Card>
        </div>
      </div>


      <div className="space-y-6">
        <Card>
          <CardContent className="pt-6 space-y-4">
            <h3 className="font-semibold text-gray-900 dark:text-white flex items-center gap-2">
              <ClipboardList className="h-4 w-4" />
              Datos de Ingreso
            </h3>

            <div className="grid grid-cols-1 sm:grid-cols-2 gap-4">
              <div>
                <Label>Fecha Ingreso al Hospital</Label>
                <Input
                  type="date"
                  value={fechaIngresoHospital}
                  onChange={(e) => setFechaIngresoHospital(e.target.value)}
                />
              </div>
              <div>
                <Label>Fecha Ingreso a Clínica Médica</Label>
                <Input
                  type="date"
                  value={fechaIngresoClinica}
                  onChange={(e) => setFechaIngresoClinica(e.target.value)}
                />
              </div>

              <div>
                <Label>Derivado de</Label>
                <Input
                  placeholder="Hospital o clínica de derivación"
                  value={derivacion}
                  onChange={(e) => setDerivacion(e.target.value)}
                />
              </div>

              <div>
                <Label>Apache / Mortalidad</Label>
                <Input
                  placeholder="Valor Apache o riesgo"
                  value={apache}
                  onChange={(e) => setApache(e.target.value)}
                />
              </div>
            </div>
          </CardContent>
        </Card>

      </div>



      <div className="space-y-6">
        <Card>
          <CardContent className="pt-6 space-y-4">
            <h3 className="font-semibold text-gray-900 dark:text-white">
              Datos Clínicos
            </h3>

            <div>
              <Label>Motivo de Consulta *</Label>
              <textarea
                className="w-full p-2 border rounded-md text-sm min-h-[80px]"
                placeholder="Motivo por el cual consulta"
                value={motivoConsulta}
                onChange={(e) => setMotivoConsulta(e.target.value)}
              />
            </div>

            <div>
              <Label>Diagnóstico de Ingreso *</Label>
              <textarea
                className="w-full p-2 border rounded-md text-sm min-h-[80px]"
                placeholder="Diagnóstico principal"
                value={diagnosticoIngreso}
                onChange={(e) => setDiagnosticoIngreso(e.target.value)}
              />
            </div>

            <div>
              <Label>Enfermedad Actual *</Label>
              <textarea
                className="w-full p-2 border rounded-md text-sm min-h-[100px]"
                placeholder="Descripción de la enfermedad actual"
                value={enfermedadActual}
                onChange={(e) => setEnfermedadActual(e.target.value)}
              />
            </div>

            <div>
              <Label>Antecedentes de Enfermedad Actual</Label>
              <textarea
                className="w-full p-2 border rounded-md text-sm min-h-[80px]"
                placeholder="Antecedentes relevantes"
                value={antecedentesEnfermedadActual}
                onChange={(e) => setAntecedentesEnfermedadActual(e.target.value)}
              />
            </div>
          </CardContent>
        </Card>
      </div>


      {/* Columna 4: Médico Ingresante */}
      <div className="space-y-6">

        <Card>
          <CardContent className="pt-6 space-y-4">
            <h3 className="font-semibold text-gray-900 dark:text-white flex items-center gap-2">
              <Stethoscope className="h-4 w-4" />
              Médico Ingresante
            </h3>

            <div>
              <Label>Médico Ingresante *</Label>
              <Input
                placeholder="Nombre del médico"
                value={getNombreProfesional(currentUser)}
                disabled
              />
            </div>
          </CardContent>
        </Card>
      </div>

    </div>
  );
}