#!/bin/bash

set -e

# ========= CONFIGURACIÓN =========
APP_NAME="rss"
APP_DIR="$(cd "$(dirname "$0")" && pwd)"
PYTHON_ENV="$APP_DIR/venv"
SERVICE_FILE="/etc/systemd/system/$APP_NAME.service"
FLASK_FILE="app:app" # Formato para Gunicorn

DB_NAME="rss"
DB_USER="rss"
DB_PASS="x" # Recuerda usar una contraseña más segura en un entorno real

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

# ========= CAMBIAR AUTENTICACIÓN DE peer A md5 =========
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 (contraseña)..."
    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 configurado para md5 o no se encontró."
fi

# ========= CONFIGURAR BASE DE DATOS Y USUARIO (VERSIÓN ROBUSTA) =========
echo "🛠️  Configurando PostgreSQL: usuario y base de datos..."

# Usamos un bloque DO para crear el usuario solo si no existe
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 \$\$;"

# Comprobamos si la base de datos existe antes de intentar crearla
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;"
    sudo -u postgres psql -c "GRANT ALL PRIVILEGES ON DATABASE $DB_NAME TO $DB_USER;"
    echo "Base de datos '$DB_NAME' creada."
else
    echo "Base de datos '$DB_NAME' ya existe."
fi

# ========= CREAR TABLAS =========
echo "📐 Creando tablas en PostgreSQL (si no existen)..."

# Exportamos la contraseña para que psql la use
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

echo "✅ Tablas creadas/verificadas."

# ========= DATOS INICIALES (si existen los archivos .sql) =========
for sql_file in continentes.sql paises.sql categorias.sql; do
    if [ -f "$APP_DIR/$sql_file" ]; then
        echo "🗂️  Insertando datos desde $sql_file..."
        psql -U "$DB_USER" -h localhost -d "$DB_NAME" -f "$APP_DIR/$sql_file"
    else
        echo "⚠️  No se encontró el archivo de datos iniciales: $sql_file"
    fi
done

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

echo "📦 Instalando dependencias desde requirements.txt"
source "$PYTHON_ENV/bin/activate"
pip install --upgrade pip
pip install -r "$APP_DIR/requirements.txt"
deactivate

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

[Service]
User=$USER
Group=$(id -gn $USER)
WorkingDirectory=$APP_DIR
ExecStart=$PYTHON_ENV/bin/gunicorn --workers 3 --bind unix:rss.sock -m 007 $FLASK_FILE
Restart=always
Environment="PYTHONUNBUFFERED=1"

[Install]
WantedBy=multi-user.target
EOF

# ========= ACTIVAR SERVICIO =========
echo "🚀 Recargando systemd y activando el servicio '$APP_NAME'"
sudo systemctl daemon-reload
sudo systemctl enable --now "$APP_NAME"
# sudo systemctl restart "$APP_NAME" # --now en enable ya lo inicia

# ========= ESTADO FINAL =========
echo "✅ ¡Instalación completada!"
echo "Para revisar el estado del servicio, ejecuta: systemctl status $APP_NAME"
echo "Para ver los logs en tiempo real, ejecuta: journalctl -u $APP_NAME -f"