#!/bin/bash

# Script de instalación completo y robusto para el Agregador RSS.
# Instala y configura PostgreSQL, Python, y la aplicación como un servicio systemd.

set -e

# ========= CONFIGURACIÓN =========
APP_NAME="rss"
DB_NAME="rss"
DB_USER="rss"
DB_PASS="x" # ¡IMPORTANTE! Usar una contraseña más segura en un entorno real.
APP_USER="x" # El usuario no-root que ejecutará la aplicación.

# Directorios (se calculan automáticamente)
APP_DIR="/home/$APP_USER/$APP_NAME"
PYTHON_ENV="$APP_DIR/venv"
SERVICE_FILE="/etc/systemd/system/$APP_NAME.service"
FLASK_APP_ENTRY="app:app" # Formato para Gunicorn: "nombre_fichero:nombre_variable_app"

# ========= 1. INSTALAR DEPENDENCIAS DEL SISTEMA =========
echo "🟢 Instalando dependencias del sistema (PostgreSQL, Python, etc.)..."
sudo apt-get update
sudo apt-get install -y wget ca-certificates postgresql postgresql-contrib python3-venv python3-pip

# ========= 2. CONFIGURAR POSTGRESQL =========
echo "🛠️  Configurando PostgreSQL..."

# Ajustar autenticación para permitir login con contraseña
PG_HBA=$(find /etc/postgresql -name pg_hba.conf | head -1)
if [ -n "$PG_HBA" ] && grep -q "local\s\+all\s\+all\s\+peer" "$PG_HBA"; then
    echo "📝 Ajustando pg_hba.conf para autenticación md5..."
    sudo sed -i 's/local\s\+all\s\+all\s\+peer/local all all md5/' "$PG_HBA"
    sudo systemctl restart postgresql
else
    echo "✅ pg_hba.conf ya parece estar configurado para md5."
fi

# Crear usuario y base de datos de forma segura (solo si no existen)
echo "👤 Creando usuario y base de datos '$DB_NAME'..."
sudo -u postgres psql -c "DO \$\$ BEGIN IF NOT EXISTS (SELECT FROM pg_catalog.pg_user WHERE usename = '$DB_USER') THEN CREATE USER $DB_USER WITH PASSWORD '$DB_PASS'; END IF; END \$\$;"
if ! sudo -u postgres psql -lqt | cut -d \| -f 1 | grep -qw "$DB_NAME"; then
    sudo -u postgres psql -c "CREATE DATABASE $DB_NAME OWNER $DB_USER;"
    echo "Base de datos '$DB_NAME' creada."
else
    echo "✅ Base de datos '$DB_NAME' ya existe."
fi

# ========= 3. CREAR TABLAS DE LA APLICACIÓN =========
echo "📐 Creando tablas en la base de datos (si no existen)..."
export PGPASSWORD="$DB_PASS"
psql -U "$DB_USER" -h localhost -d "$DB_NAME" <<EOF
CREATE TABLE IF NOT EXISTS continentes (id SERIAL PRIMARY KEY, nombre VARCHAR(50) NOT NULL UNIQUE);
CREATE TABLE IF NOT EXISTS categorias (id SERIAL PRIMARY KEY, nombre VARCHAR(100) NOT NULL UNIQUE);
CREATE TABLE IF NOT EXISTS paises (id SERIAL PRIMARY KEY, nombre VARCHAR(100) NOT NULL UNIQUE, continente_id INTEGER REFERENCES continentes(id) ON DELETE SET NULL);
CREATE TABLE IF NOT EXISTS feeds (id SERIAL PRIMARY KEY, nombre VARCHAR(255), descripcion TEXT, url TEXT NOT NULL UNIQUE, categoria_id INTEGER REFERENCES categorias(id) ON DELETE SET NULL, pais_id INTEGER REFERENCES paises(id) ON DELETE SET NULL, idioma CHAR(2), activo BOOLEAN DEFAULT TRUE, fallos INTEGER DEFAULT 0);
CREATE TABLE IF NOT EXISTS noticias (id VARCHAR(32) PRIMARY KEY, titulo TEXT, resumen TEXT, url TEXT NOT NULL UNIQUE, fecha TIMESTAMP, imagen_url TEXT, categoria_id INTEGER REFERENCES categorias(id) ON DELETE SET NULL, pais_id INTEGER REFERENCES paises(id) ON DELETE SET NULL);
EOF
unset PGPASSWORD
echo "✅ Tablas creadas/verificadas."

# ========= 4. INSERTAR DATOS INICIALES (OPCIONAL) =========
for sql_file in continentes.sql paises.sql categorias.sql; do
    if [ -f "$APP_DIR/$sql_file" ]; then
        echo "🗂️  Insertando datos desde $sql_file..."
        export PGPASSWORD="$DB_PASS"
        psql -U "$DB_USER" -h localhost -d "$DB_NAME" -f "$APP_DIR/$sql_file"
        unset PGPASSWORD
    else
        echo "⚠️  Archivo de datos iniciales no encontrado, omitiendo: $sql_file"
    fi
done

# ========= 5. PREPARAR ENTORNO PYTHON =========
echo "🐍 Creando entorno virtual en $PYTHON_ENV"
python3 -m venv "$PYTHON_ENV"

echo "📦 Instalando dependencias desde requirements.txt..."
# Usamos el python del venv para ejecutar pip
"$PYTHON_ENV/bin/python" -m pip install --upgrade pip
"$PYTHON_ENV/bin/python" -m pip install -r "$APP_DIR/requirements.txt"

# ========= 6. CREAR SERVICIO SYSTEMD =========
echo "⚙️  Creando servicio systemd en: $SERVICE_FILE"
sudo tee "$SERVICE_FILE" > /dev/null <<EOF
[Unit]
Description=Servicio del Agregador RSS de Noticias (Simple)
After=network.target postgresql.service
Requires=postgresql.service

[Service]
# Mejora de seguridad: Ejecutar como usuario no-root.
User=$APP_USER
Group=$APP_USER

WorkingDirectory=$APP_DIR

# Comando final: Usa gunicorn, escucha en el puerto 8000, y tiene un timeout extendido.
ExecStart=$PYTHON_ENV/bin/gunicorn --bind 0.0.0.0:8000 --timeout 120 $FLASK_APP_ENTRY

Restart=always
Environment="PYTHONUNBUFFERED=1"

[Install]
WantedBy=multi-user.target
EOF

# ========= 7. ACTIVAR Y LANZAR EL SERVICIO =========
echo "🚀 Recargando systemd y activando el servicio '$APP_NAME'..."
sudo systemctl daemon-reload
sudo systemctl enable --now "$APP_NAME"

# --now hace start, pero un restart explícito asegura que usa la última conf.
sudo systemctl restart "$APP_NAME"

# ========= ESTADO FINAL =========
echo ""
echo "✅ ¡Instalación completada!"
echo "   El servicio '$APP_NAME' se ha instalado y está corriendo."
echo "   La aplicación es accesible en: http://<IP-DE-LA-VM>:8000"
echo ""
echo "👉 Comandos útiles:"
echo "   Revisar estado: sudo systemctl status $APP_NAME"
echo "   Ver logs en vivo: sudo journalctl -u $APP_NAME -f"
echo "   Reiniciar servicio: sudo systemctl restart $APP_NAME"
echo ""
echo "Recuerda configurar la redirección de puertos en VirtualBox y el firewall si es necesario."