import { useState } from 'react';
import { Bed, CheckCircle2, Clock, Wrench, User, Plus, Trash, Edit2 } from 'lucide-react';
import { Card, CardContent } from '@/components/ui/card';
import { Badge } from '@/components/ui/badge';
import { Button } from '@/components/ui/button';
import { Dialog, DialogContent, DialogHeader, DialogTitle, DialogTrigger } from '@/components/ui/dialog';
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from '@/components/ui/select';
import { Label } from '@/components/ui/label';
import { Input } from '@/components/ui/input';
import type { Cama, Paciente, Internacion, Area } from '@/types';

interface MapaCamasProps {
  camas: Cama[];
  areas: Area[];
  pacientes: Paciente[];
  internaciones: Internacion[];
  onActualizarCama: (id: string, datos: Partial<Cama>) => void;
  onAgregarCama: (cama: Omit<Cama, 'id'>) => string;
  onEliminarCama: (id: string) => void;
  onIniciarInternacion: (internacion: Omit<Internacion, 'id' | 'fechaIngreso' | 'activa'>) => void;
  onAgregarArea: (area: Omit<Area, 'id'>) => string;
  onActualizarArea: (id: string, datos: Partial<Area>) => void;
  onEliminarArea: (id: string) => void;
  getPacienteById: (id: string) => Paciente | undefined;
  getInternacionById: (id: string) => Internacion | undefined;
}

export function MapaCamas({ 
  camas,
  areas,
  pacientes, 
  internaciones,
  onActualizarCama, 
  onAgregarCama,
  onEliminarCama,
  onIniciarInternacion,
  onAgregarArea,
  onActualizarArea,
  onEliminarArea,
  getPacienteById,
  getInternacionById
}: MapaCamasProps) {
  const [filtroSala, setFiltroSala] = useState<string>('todas');
  const [filtroTipo, setFiltroTipo] = useState<string>('todos');
  const [filtroEstado, setFiltroEstado] = useState<string>('todos');
  const [camaSeleccionada, setCamaSeleccionada] = useState<Cama | null>(null);
  const [pacienteSeleccionado, setPacienteSeleccionado] = useState<string>('');
  const [diagnostico, setDiagnostico] = useState('');
  const [enfermedadActual, setEnfermedadActual] = useState('');
  const [antecedentesEnfermedadActual, setAntecedentesEnfermedadActual] = useState('');
  const [medico, setMedico] = useState('');
  const [dialogoAbierto, setDialogoAbierto] = useState(false);
  // Area dialog state
  const [areaDialogOpen, setAreaDialogOpen] = useState(false);
  const [areaNombre, setAreaNombre] = useState('');
  const [editingAreaId, setEditingAreaId] = useState<string | null>(null);

  // Cama dialog state
  const [bedDialogOpen, setBedDialogOpen] = useState(false);
  const [editingBed, setEditingBed] = useState<Cama | null>(null);
  const [bedNumero, setBedNumero] = useState('');
  const [bedTipo, setBedTipo] = useState<Cama['tipo']>('General');
  const [bedAreaId, setBedAreaId] = useState<string | undefined>(undefined);

  const salas = Array.from(new Set(camas.map(c => c.areaId).filter(Boolean))) as string[];
  const areaById = Object.fromEntries(areas.map(a => [a.id, a.nombre]));
  const tipos = Array.from(new Set(camas.map(c => c.tipo)));

  const camasFiltradas = camas.filter(cama => {
    if (filtroSala !== 'todas' && cama.areaId !== filtroSala) return false;
    if (filtroTipo !== 'todos' && cama.tipo !== filtroTipo) return false;
    if (filtroEstado !== 'todos' && cama.estado !== filtroEstado) return false;
    return true;
  });

  const getEstadoColor = (cama: Cama) => {
    if (cama.estado === 'Disponible') {
      return 'bg-green-100 border-green-300 text-green-800';
    }
    if (cama.estado === 'Reservada') {
      return 'bg-orange-100 border-orange-300 text-orange-800';
    }
    if (cama.estado === 'Ocupada') {
      // Tipos de aislamiento
      const aislamientos = ['Aislamiento KPC', 'Aislamiento COVID', 'Aislamiento Clostridium', 'Aislamiento Neutropenico'];
      if (aislamientos.includes(cama.tipo)) {
        return 'bg-red-100 border-red-300 text-red-800';
      }
      return 'bg-blue-100 border-blue-300 text-blue-800';
    }
    return 'bg-amber-100 border-amber-300 text-amber-800';
  };

  const getEstadoIcono = (cama: Cama) => {
    switch (cama.estado) {
      case 'Disponible': return <CheckCircle2 className="h-5 w-5" />;
      case 'Ocupada': return <User className="h-5 w-5" />;
      case 'Reparacion': return <Wrench className="h-5 w-5" />;
      case 'Reservada': return <Clock className="h-5 w-5" />;
    }
  };

  const parseBedNumber = (numero: string) => {
    const match = numero.match(/^(\d{3})-(\d+)$/);
    if (!match) return { sala: 0, cama: 0 };
    return {
      sala: parseInt(match[1]),
      cama: parseInt(match[2])
    };
  };

  const sortedCamas = [...camasFiltradas].sort((a, b) => {
    const pa = parseBedNumber(a.numero);
    const pb = parseBedNumber(b.numero);

    if (pa.sala >= 200 && pa.sala < 300) {
      const salaImpar = [221, 217, 215, 213, 209, 205, 203, 201];
      const salaPar = [204, 214];

      if (salaImpar.includes(pa.sala)) {
        if (salaImpar.includes(pb.sala)) {
          if (pa.sala === pb.sala) {
            return pa.cama - pb.cama;
          }
          return salaImpar.indexOf(pa.sala) - salaImpar.indexOf(pb.sala);
        }
        return -1;
      }
      if (salaImpar.includes(pb.sala)) return 1;

      if (salaPar.includes(pa.sala)) {
        if (salaPar.includes(pb.sala)) {
          if (pa.sala === pb.sala) {
            return pa.cama - pb.cama;
          }
          return pa.sala - pb.sala;
        }
        return -1;
      }
      if (salaPar.includes(pb.sala)) return 1;

      return pa.cama - pb.cama;
    }

    if (pa.sala >= 300 && pa.sala < 400) {
      const salaPiso3 = [308, 310, 312];
      if (salaPiso3.includes(pa.sala)) {
        if (salaPiso3.includes(pb.sala)) {
          if (pa.sala === pb.sala) {
            return pa.cama - pb.cama;
          }
          return salaPiso3.indexOf(pa.sala) - salaPiso3.indexOf(pb.sala);
        }
        return -1;
      }
      if (salaPiso3.includes(pb.sala)) return 1;
      return pa.cama - pb.cama;
    }

    return a.numero.localeCompare(b.numero);
  });

  const handleOcuparCama = () => {
    if (camaSeleccionada && pacienteSeleccionado && diagnostico && enfermedadActual && medico) {
      onIniciarInternacion({
        pacienteId: pacienteSeleccionado,
        camaId: camaSeleccionada.id,
        diagnosticoIngreso: diagnostico,
        enfermedadActual,
        antecedentesEnfermedadActual: antecedentesEnfermedadActual || undefined,
        medicoIngresante: medico,
      });
      setDialogoAbierto(false);
      setCamaSeleccionada(null);
      setPacienteSeleccionado('');
      setDiagnostico('');
      setEnfermedadActual('');
      setAntecedentesEnfermedadActual('');
      setMedico('');
    }
  };

  const handleCambiarEstado = (cama: Cama, nuevoEstado: Cama['estado']) => {
    onActualizarCama(cama.id, { estado: nuevoEstado });
  };

  const pacientesSinInternar = pacientes.filter(p => {
    const internacionActiva = internaciones.find(i => i.pacienteId === p.id && i.activa);
    return !internacionActiva;
  });

  return (
    <div className="space-y-6">
      {/* Header */}
      <div className="flex flex-col sm:flex-row sm:items-center sm:justify-between gap-4">
        <div>
          <h1 className="text-2xl font-bold text-gray-900">Mapa de Camas</h1>
          <p className="text-gray-500">Gestión de camas del servicio</p>
        </div>
        <div className="flex items-center gap-2">
          <Badge variant="outline" className="bg-green-50 text-green-700">
            {camas.filter(c => c.estado === 'Disponible').length} Disponibles
          </Badge>
          <Badge variant="outline" className="bg-red-50 text-red-700">
            {camas.filter(c => c.estado === 'Ocupada').length} Ocupadas
          </Badge>
        <div className="ml-4 flex items-center gap-2">
          <Button onClick={() => {
            setEditingAreaId(null);
            setAreaNombre('');
            setAreaDialogOpen(true);
          }}>
            Administrar Áreas
          </Button>
          <Button onClick={() => {
            setEditingBed(null);
            setBedNumero('');
            setBedTipo('General');
            setBedAreaId(areas?.[0]?.id);
            setBedDialogOpen(true);
          }}>
            <Plus className="h-4 w-4 mr-1" />Agregar Cama
          </Button>
        </div>
        </div>
      </div>

      {/* Filtros */}
      <Card>
        <CardContent className="p-4">
          <div className="grid grid-cols-1 sm:grid-cols-3 gap-4">
            <div>
              <Label className="text-xs text-gray-500 mb-1 block">Área</Label>
              <Select value={filtroSala} onValueChange={setFiltroSala}>
                <SelectTrigger>
                  <SelectValue placeholder="Todas las áreas" />
                </SelectTrigger>
                <SelectContent>
                  <SelectItem value="todas">Todas las áreas</SelectItem>
                  {salas.map(sala => (
                    <SelectItem key={sala} value={sala}>{areaById[sala] ?? sala}</SelectItem>
                  ))}
                </SelectContent>
              </Select>
            </div>
            <div>
              <Label className="text-xs text-gray-500 mb-1 block">Tipo</Label>
              <Select value={filtroTipo} onValueChange={setFiltroTipo}>
                <SelectTrigger>
                  <SelectValue placeholder="Todos los tipos" />
                </SelectTrigger>
                <SelectContent>
                  <SelectItem value="todos">Todos los tipos</SelectItem>
                  {tipos.map(tipo => (
                    <SelectItem key={tipo} value={tipo}>{tipo}</SelectItem>
                  ))}
                </SelectContent>
              </Select>
            </div>
            <div>
              <Label className="text-xs text-gray-500 mb-1 block">Estado</Label>
              <Select value={filtroEstado} onValueChange={setFiltroEstado}>
                <SelectTrigger>
                  <SelectValue placeholder="Todos los estados" />
                </SelectTrigger>
                <SelectContent>
                  <SelectItem value="todos">Todos los estados</SelectItem>
                  <SelectItem value="Disponible">Disponible</SelectItem>
                  <SelectItem value="Ocupada">Ocupada</SelectItem>
                  <SelectItem value="Mantenimiento">Mantenimiento</SelectItem>
                  <SelectItem value="Reservada">Reservada</SelectItem>
                </SelectContent>
              </Select>
            </div>
          </div>
        </CardContent>
      </Card>

      {/* Mapa de Camas */}
      <div className="grid grid-cols-2 sm:grid-cols-3 md:grid-cols-4 lg:grid-cols-5 gap-2 sm:gap-4">
        {sortedCamas.map((cama) => {
          const paciente = cama.pacienteId ? getPacienteById(cama.pacienteId) : null;
          const internacion = cama.internacionId ? getInternacionById(cama.internacionId) : null;
          
          return (
            <Dialog key={cama.id}>
              <DialogTrigger asChild>
                <Card 
                  className={`cursor-pointer hover:shadow-md transition-shadow border-2 ${getEstadoColor(cama)}`}
                  onClick={() => setCamaSeleccionada(cama)}
                >
                  <CardContent className="p-4">
                    <div className="flex items-center justify-between mb-2">
                      <Bed className="h-6 w-6" />
                      {getEstadoIcono(cama)}
                    </div>
                    <p className="font-bold text-lg">{cama.numero}</p>
                    <p className="text-xs opacity-75">{cama.areaId ? areaById[cama.areaId] ?? 'Área desconocida' : 'Sin área'}</p>
                    <Badge variant="outline" className="mt-2 text-xs bg-white/50">
                      {cama.tipo}
                    </Badge>
                    {cama.estado === 'Ocupada' && paciente && (
                      <div className="mt-2 pt-2 border-t border-current/20">
                        <p className="text-xs font-medium truncate">
                          {paciente.apellido}, {paciente.nombre}
                        </p>
                      </div>
                    )}

                   
                  </CardContent>
                </Card>
              </DialogTrigger>
              <DialogContent className="sm:max-w-md">
                  <DialogHeader>
                  <DialogTitle className="flex items-center gap-2">
                    <Bed className="h-5 w-5" />
                    Cama {cama.numero} - {cama.areaId ? areaById[cama.areaId] ?? 'Área desconocida' : 'Sin área'}
                  </DialogTitle>
                </DialogHeader>
                <div className="space-y-4">
                  <div className="flex items-center gap-2">
                    <Badge className={getEstadoColor(cama)}>
                      {cama.estado}
                    </Badge>
                    <Badge variant="outline">{cama.tipo}</Badge>
                  </div>
                  
                  {cama.estado === 'Ocupada' && paciente && internacion && (
                    <div className="bg-gray-50 p-4 rounded-lg space-y-2">
                      <p className="font-medium">Paciente:</p>
                      <p className="text-lg">{paciente.apellido}, {paciente.nombre}</p>
                      <p className="text-sm text-gray-500">DNI: {paciente.dni}</p>
                      <p className="text-sm text-gray-500">Ingreso: {internacion.fechaIngreso}</p>
                      <p className="text-sm text-gray-500">Diagnóstico: {internacion.diagnosticoIngreso}</p>
                      <p className="text-sm text-gray-500">Médico: {internacion.medicoIngresante}</p>
                    </div>
                  )}

                  <div className="space-y-2">
                    <p className="font-medium">Acciones:</p>
                    <div className="grid grid-cols-2 gap-2">
                      {cama.estado === 'Disponible' && (
                        <Dialog open={dialogoAbierto} onOpenChange={setDialogoAbierto}>
                          <DialogTrigger asChild>
                            <Button 
                              className="w-full" 
                              onClick={() => {
                                setCamaSeleccionada(cama);
                                setDialogoAbierto(true);
                              }}
                            >
                              <Plus className="h-4 w-4 mr-1" />
                              Ocupar
                            </Button>
                          </DialogTrigger>
                          <DialogContent className="sm:max-w-lg">
                            <DialogHeader>
                              <DialogTitle>Nueva Internación - Cama {cama.numero}</DialogTitle>
                            </DialogHeader>
                            <div className="space-y-4">
                              <div>
                                <Label>Paciente *</Label>
                                <Select value={pacienteSeleccionado} onValueChange={setPacienteSeleccionado}>
                                  <SelectTrigger>
                                    <SelectValue placeholder="Seleccionar paciente" />
                                  </SelectTrigger>
                                  <SelectContent>
                                    {pacientesSinInternar.map(p => (
                                      <SelectItem key={p.id} value={p.id}>
                                        {p.apellido}, {p.nombre} - DNI: {p.dni}
                                      </SelectItem>
                                    ))}
                                  </SelectContent>
                                </Select>
                              </div>
                              <div>
                                <Label>Diagnóstico de Ingreso *</Label>
                                <textarea
                                  className="w-full p-2 border rounded-md text-sm"
                                  rows={3}
                                  value={diagnostico}
                                  onChange={(e) => setDiagnostico(e.target.value)}
                                  placeholder="Ingrese el diagnóstico..."
                                />
                              </div>
                              <div>
                                <Label>Enfermedad Actual *</Label>
                                <textarea
                                  className="w-full p-2 border rounded-md text-sm"
                                  rows={4}
                                  value={enfermedadActual}
                                  onChange={(e) => setEnfermedadActual(e.target.value)}
                                  placeholder="Describa la enfermedad actual..."
                                />
                              </div>
                              <div>
                                <Label>Antecedentes de Enfermedad Actual</Label>
                                <textarea
                                  className="w-full p-2 border rounded-md text-sm"
                                  rows={3}
                                  value={antecedentesEnfermedadActual}
                                  onChange={(e) => setAntecedentesEnfermedadActual(e.target.value)}
                                  placeholder="Antecedentes relevantes..."
                                />
                              </div>
                              <div>
                                <Label>Médico Tratante *</Label>
                                <input
                                  type="text"
                                  className="w-full p-2 border rounded-md text-sm"
                                  value={medico}
                                  onChange={(e) => setMedico(e.target.value)}
                                  placeholder="Nombre del médico..."
                                />
                              </div>
<Button 
                                 className="w-full" 
                                 onClick={handleOcuparCama}
                                 disabled={!pacienteSeleccionado || !diagnostico || !enfermedadActual || !medico}
                               >
                                <Plus className="h-4 w-4 mr-1" />
                                Iniciar Internación
                              </Button>
                            </div>
                          </DialogContent>
                        </Dialog>
                      )}
                      
                      {cama.estado !== 'Disponible' && (
                        <Button 
                          variant="outline" 
                          className="w-full"
                          onClick={() => handleCambiarEstado(cama, 'Disponible')}
                        >
                          <CheckCircle2 className="h-4 w-4 mr-1" />
                          Disponible
                        </Button>
                      )}
                      
                      {cama.estado !== 'Reparacion' && (
                        <Button 
                          variant="outline" 
                          className="w-full"
                          onClick={() => handleCambiarEstado(cama, 'Reparacion')}
                        >
                          <Wrench className="h-4 w-4 mr-1" />
                          Reparacion
                        </Button>
                      )}
               
                      {cama.estado !== 'Reservada' && (
                        <Button 
                          variant="outline" 
                          className="w-full"
                          onClick={() => handleCambiarEstado(cama, 'Reservada')}
                        >
                          <Clock className="h-4 w-4 mr-1" />
                          Reservar
                        </Button>
                      )}

                       
                      <Button variant="outline" onClick={() => {
                        setEditingBed(cama);
                        setBedNumero(cama.numero);
                        setBedTipo(cama.tipo);
                        setBedAreaId(cama.areaId);
                        setBedDialogOpen(true);
                      }}>
                        <Edit2 className="h-4 w-4 mr-1" />Editar
                      </Button>
                      <Button variant="destructive" onClick={() => {
                        if (confirm('Eliminar esta cama?')) {
                          onEliminarCama(cama.id);
                        }
                      }}>
                        <Trash className="h-4 w-4 mr-1" />Eliminar
                      </Button>
                    </div>
                    </div>
                  </div>
                
              </DialogContent>
            </Dialog>
          );
        })}
      </div>

      {/* Area dialog */}
      <Dialog open={areaDialogOpen} onOpenChange={setAreaDialogOpen}>
        <DialogContent className="sm:max-w-md">
          <DialogHeader>
            <DialogTitle>{editingAreaId ? 'Editar Área' : 'Nueva Área'}</DialogTitle>
          </DialogHeader>
          <div className="space-y-4">
            <div>
              <Label>Nombre</Label>
              <Input value={areaNombre} onChange={(e) => setAreaNombre(e.target.value)} />
            </div>
            <div className="flex gap-2 justify-end">
              {editingAreaId && (
                <Button variant="destructive" onClick={() => { if (confirm('Eliminar área?')) { onEliminarArea(editingAreaId); setAreaDialogOpen(false); } }}>
                  Eliminar
                </Button>
              )}
              <Button onClick={() => setAreaDialogOpen(false)}>Cancelar</Button>
              <Button onClick={() => {
                if (!areaNombre.trim()) return alert('Nombre requerido');
                if (editingAreaId) onActualizarArea(editingAreaId, { nombre: areaNombre });
                else onAgregarArea({ nombre: areaNombre });
                setAreaDialogOpen(false);
              }}>Guardar</Button>
            </div>
            <div>
              <p className="text-sm font-medium">Áreas existentes</p>
              <div className="mt-2 space-y-2">
                {areas.map(a => (
                  <div key={a.id} className="flex items-center justify-between">
                    <div>{a.nombre}</div>
                    <div className="flex gap-2">
                      <Button variant="ghost" onClick={() => { setEditingAreaId(a.id); setAreaNombre(a.nombre); }}>
                        Editar
                      </Button>
                      <Button variant="destructive" onClick={() => { if (confirm('Eliminar área?')) onEliminarArea(a.id); }}>
                        Eliminar
                      </Button>
                    </div>
                  </div>
                ))}
              </div>
            </div>
          </div>
        </DialogContent>
      </Dialog>

      {/* Bed dialog */}
      <Dialog open={bedDialogOpen} onOpenChange={setBedDialogOpen}>
        <DialogContent className="sm:max-w-md">
          <DialogHeader>
            <DialogTitle>{editingBed ? `Editar Cama ${editingBed.numero}` : 'Nueva Cama'}</DialogTitle>
          </DialogHeader>
          <div className="space-y-4">
            <div>
              <Label>Número</Label>
              <Input value={bedNumero} onChange={(e) => setBedNumero(e.target.value)} />
            </div>
            <div>
              <Label>Tipo</Label>
              <Select value={bedTipo} onValueChange={(v) => setBedTipo(v as Cama['tipo'])}>
                <SelectTrigger>
                  <SelectValue />
                </SelectTrigger>
                <SelectContent>
                  <SelectItem value="General">General</SelectItem>
                  <SelectItem value="Aislamiento KPC">Aislamiento KPC</SelectItem>
                  <SelectItem value="Aislamiento COVID">Aislamiento COVID</SelectItem>
                  <SelectItem value="Aislamiento Clostridium">Aislamiento Clostridium</SelectItem>
                  <SelectItem value="Aislamiento Neutropenico">Aislamiento Neutropenico</SelectItem>
                </SelectContent>
              </Select>
            </div>
            <div>
              <Label>Área</Label>
              <Select value={bedAreaId ?? '__none'} onValueChange={(v) => setBedAreaId(v === '__none' ? undefined : v)}>
                <SelectTrigger>
                  <SelectValue placeholder="Seleccione área" />
                </SelectTrigger>
                <SelectContent>
                  <SelectItem value="__none">Sin área</SelectItem>
                  {areas.map(a => (
                    <SelectItem key={a.id} value={a.id}>{a.nombre}</SelectItem>
                  ))}
                </SelectContent>
              </Select>
            </div>
            <div className="flex gap-2 justify-end">
              {editingBed && (
                <Button variant="destructive" onClick={() => { if (confirm('Eliminar cama?')) { onEliminarCama(editingBed.id); setBedDialogOpen(false); } }}>
                  Eliminar
                </Button>
              )}
              <Button onClick={() => setBedDialogOpen(false)}>Cancelar</Button>
              <Button onClick={() => {
                if (!bedNumero.trim()) return alert('Número requerido');
                if (editingBed) {
                  onActualizarCama(editingBed.id, { numero: bedNumero, tipo: bedTipo, areaId: bedAreaId });
                } else {
                  onAgregarCama({ numero: bedNumero, tipo: bedTipo, estado: 'Disponible', areaId: bedAreaId });
                }
                setBedDialogOpen(false);
              }}>Guardar</Button>
            </div>
          </div>
        </DialogContent>
      </Dialog>

      {camasFiltradas.length === 0 && (
        <div className="text-center py-12 text-gray-400">
          <Bed className="h-16 w-16 mx-auto mb-4 opacity-50" />
          <p className="text-lg">No se encontraron camas con los filtros seleccionados</p>
        </div>
      )}
    </div>
  );
}
