#!/bin/bash

# Matrix + Element Setup Script
set -e

echo "🚀 Configurando Matrix + Element + Synapse Admin..."

# Colores para output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

LIVEKIT_COMPOSE_FILE="docker-compose.livekit.yml"
LIVEKIT_CONFIG_FILE="configs/livekit/livekit.yaml"
LIVEKIT_CONFIG_TEMPLATE="configs/livekit/livekit.example.yaml"

# Verificar Docker
if ! command -v docker &> /dev/null; then
    echo -e "${RED}❌ Docker no está instalado${NC}"
    exit 1
fi

if ! command -v docker-compose &> /dev/null; then
    echo -e "${RED}❌ Docker Compose no está instalado${NC}"
    exit 1
fi

# Verificar archivo .env
if [ ! -f .env ]; then
    echo -e "${YELLOW}⚠️  Archivo .env no encontrado, copiando desde .env.example${NC}"
    cp .env.example .env
    echo -e "${RED}🔧 Por favor edita el archivo .env con tus configuraciones${NC}"
    echo -e "${BLUE}💡 Especialmente cambia las contraseñas por defecto${NC}"
    exit 1
fi

# Cargar variables de entorno
source .env

echo -e "${BLUE}📋 Verificando configuración...${NC}"

# Crear red Docker si no existe
echo -e "${BLUE}🌐 Creando red Docker...${NC}"
if ! docker network ls | grep -q matrix_net; then
    docker network create --driver=bridge --subnet=${MATRIX_NETWORK_SUBNET} --gateway=${MATRIX_NETWORK_GATEWAY} matrix_net
    echo -e "${GREEN}✅ Red matrix_net creada${NC}"
else
    echo -e "${YELLOW}ℹ️  Red matrix_net ya existe${NC}"
fi

# Crear directorios necesarios
echo -e "${BLUE}📁 Creando directorios...${NC}"
mkdir -p synapse_data/appservices
mkdir -p backups
mkdir -p configs/livekit

# Crear volúmenes externos requeridos por docker-compose si no existen
for volume in matrix_postgres_data; do
    if ! docker volume ls --format '{{.Name}}' | grep -qx "$volume"; then
        echo -e "${BLUE}🗄️  Creando volumen Docker $volume...${NC}"
        docker volume create "$volume" >/dev/null
        echo -e "${GREEN}✅ Volumen $volume creado${NC}"
    else
        echo -e "${YELLOW}ℹ️  Volumen $volume ya existe${NC}"
    fi
done

# Verificar configuración de LiveKit
if [ ! -f "${LIVEKIT_CONFIG_FILE}" ]; then
    if [ -f "${LIVEKIT_CONFIG_TEMPLATE}" ]; then
        cp "${LIVEKIT_CONFIG_TEMPLATE}" "${LIVEKIT_CONFIG_FILE}"
    fi
    echo -e "${RED}❌ No se encontró ${LIVEKIT_CONFIG_FILE}${NC}"
    echo -e "${YELLOW}   Se creó una plantilla base, edítala antes de continuar.${NC}"
    exit 1
fi

if [ ! -f "${LIVEKIT_COMPOSE_FILE}" ]; then
    echo -e "${RED}❌ Falta ${LIVEKIT_COMPOSE_FILE}. No se puede iniciar LiveKit.${NC}"
    exit 1
fi

# Generar configuración de Synapse si no existe
if [ ! -f synapse_data/homeserver.yaml ]; then
    echo -e "${BLUE}⚙️  Generando configuración de Synapse...${NC}"

    # Generar configuración inicial
    docker run --rm -v "$PWD/synapse_data:/data" \
        -e SYNAPSE_SERVER_NAME=${MATRIX_SERVER_NAME} \
        -e SYNAPSE_REPORT_STATS=${MATRIX_REPORT_STATS} \
        --user root \
        matrixdotorg/synapse:latest generate

    # Cambiar permisos
    docker run --rm -v "$PWD:/workdir" --user root alpine sh -c "chown -R $(id -u):$(id -g) /workdir/synapse_data"

    echo -e "${GREEN}✅ Configuración base de Synapse generada${NC}"
    echo -e "${YELLOW}🔧 Aplicando configuraciones personalizadas...${NC}"

    # E2EE por defecto en salas privadas/invite — idempotente.
    HS_YAML="synapse_data/homeserver.yaml"
    if [ -f "$HS_YAML" ] && ! grep -q "^encryption_enabled_by_default_for_room_type:" "$HS_YAML"; then
        echo "" >> "$HS_YAML"
        echo "# Cifrado E2EE por defecto en salas privadas + DMs (aplicado por setup.sh)" >> "$HS_YAML"
        echo "encryption_enabled_by_default_for_room_type: invite" >> "$HS_YAML"
        echo -e "${GREEN}🔒 Cifrado E2EE por defecto activado (invite rooms)${NC}"
    fi
fi

# Idempotente tambien para instancias ya generadas: asegurar la directiva.
HS_YAML="synapse_data/homeserver.yaml"
if [ -f "$HS_YAML" ] && ! grep -q "^encryption_enabled_by_default_for_room_type:" "$HS_YAML"; then
    echo "" >> "$HS_YAML"
    echo "# Cifrado E2EE por defecto en salas privadas + DMs (aplicado por setup.sh)" >> "$HS_YAML"
    echo "encryption_enabled_by_default_for_room_type: invite" >> "$HS_YAML"
    echo -e "${GREEN}🔒 Cifrado E2EE por defecto activado (invite rooms) en homeserver.yaml existente${NC}"
fi

echo -e "${BLUE}🐳 Iniciando contenedores...${NC}"
docker-compose up -d
echo -e "${BLUE}📡 Iniciando LiveKit + lk-jwt...${NC}"
docker-compose -f "${LIVEKIT_COMPOSE_FILE}" up -d

echo -e "${BLUE}⏳ Esperando que los servicios estén listos...${NC}"
sleep 20

# Verificar que todos los servicios estén funcionando
echo -e "${BLUE}🔍 Verificando servicios...${NC}"

services=(
    "postgres:5432:tcp"
    "synapse:8008:http"
    "element:8081:http"
    "synapse-admin:8082:http"
    "livekit:${LIVEKIT_HTTP_PORT:-7880}:http"
    "livekit-jwt:${LIVEKIT_JWT_PORT:-6080}:http"
)
for service in "${services[@]}"; do
    IFS=":" read -r name port proto <<<"$service"

    if [ "$proto" = "http" ]; then
        if curl -s --max-time 5 "http://localhost:$port" > /dev/null 2>&1; then
            echo -e "${GREEN}✅ $name (puerto $port) - OK${NC}"
        else
            echo -e "${RED}❌ $name (puerto $port) - ERROR${NC}"
        fi
    else
        if timeout 5 bash -c "cat < /dev/null > /dev/tcp/localhost/$port" > /dev/null 2>&1; then
            echo -e "${GREEN}✅ $name (puerto $port) - OK${NC}"
        else
            echo -e "${RED}❌ $name (puerto $port) - ERROR${NC}"
        fi
    fi
done

# Crear usuario administrador si no existe
echo -e "${BLUE}👤 Creando usuario administrador...${NC}"
if docker exec element_matrix_chat-synapse-1 register_new_matrix_user \
    -c /data/homeserver.yaml \
    -u ${ADMIN_USERNAME} \
    -p ${ADMIN_PASSWORD} \
    -a \
    http://localhost:8008 2>/dev/null; then
    echo -e "${GREEN}✅ Usuario administrador creado: ${ADMIN_USERNAME}${NC}"
else
    echo -e "${YELLOW}ℹ️  Usuario administrador ya existe o hubo un error${NC}"
fi

echo
echo -e "${GREEN}🎉 ¡Configuración completada!${NC}"
echo
echo -e "${BLUE}📍 Acceso a los servicios:${NC}"
echo -e "   • Element Web: ${GREEN}http://localhost:${ELEMENT_PORT}${NC}"
echo -e "   • Synapse Admin: ${GREEN}http://localhost:${SYNAPSE_ADMIN_PORT}${NC}"
echo -e "   • API Matrix: ${GREEN}http://localhost:${SYNAPSE_PORT}${NC}"
echo
echo -e "${BLUE}🔐 Credenciales de administrador:${NC}"
echo -e "   • Usuario: ${GREEN}${ADMIN_USERNAME}${NC}"
echo -e "   • Contraseña: ${GREEN}${ADMIN_PASSWORD}${NC}"
echo
echo -e "${YELLOW}💡 Para crear más usuarios usa: ./scripts/create-user.sh${NC}"