setup.sh

#!/bin/bash
# Setup iniziale di tango-crawler.
# Eseguire una sola volta dalla directory del progetto.
set -euo pipefail

# ── Argomenti ─────────────────────────────────────────────────────────────────
MODE="${1:-}"   # "build" per rebuild rapido, vuoto per setup completo

# ── Configurazione ────────────────────────────────────────────────────────────
NORMALIZE_AT="06:00"      # orario giornaliero di normalizzazione (HH:MM)
DATA_DIR="$HOME/.local/share/tango-crawler"
REPO_DIR="$(cd "$(dirname "$0")" && pwd)"
COMPOSE="docker compose -f $REPO_DIR/docker-compose.yml"

# ── Helpers ───────────────────────────────────────────────────────────────────
info()    { echo "[INFO]  $*"; }
success() { echo "[OK]    $*"; }
warning() { echo "[WARN]  $*"; }

# ── Modalità build rapido ──────────────────────────────────────────────────────
if [ "$MODE" = "build" ]; then
    info "Stop crawler..."
    $COMPOSE down
    info "Aggiorno repo..."
    git -C "$REPO_DIR" pull
    info "Rebuild + avvio..."
    $COMPOSE up -d --build
    success "Rebuild completato."
    exit 0
fi

# ── Modalità fresh: build + setup completo ────────────────────────────────────
if [ "$MODE" = "fresh" ]; then
    info "Stop crawler..."
    $COMPOSE down
    info "Aggiorno repo..."
    git -C "$REPO_DIR" pull
    info "Rilancio setup completo con rebuild..."
    exec "$0"
fi

# ── 1. Directory dati ─────────────────────────────────────────────────────────
info "Creo directory dati: $DATA_DIR"
mkdir -p "$DATA_DIR"
success "Directory pronta."

# ── 2. sqlite3 sull'host (per query dirette) ──────────────────────────────────
if command -v sqlite3 &>/dev/null; then
    success "sqlite3 già installato: $(sqlite3 --version | cut -d' ' -f1)"
else
    info "Installo sqlite3..."
    sudo apt-get install -y --quiet sqlite3
    success "sqlite3 installato."
fi

# ── 3. Build immagine Docker ──────────────────────────────────────────────────
info "Build immagine Docker..."
$COMPOSE build --quiet
success "Immagine pronta."

# ── 4. Permessi sulla directory dati (file creati da Docker sono di root) ─────
for dbfile in "$DATA_DIR"/*.db "$DATA_DIR"/*.log; do
    [ -f "$dbfile" ] || continue
    if [ "$(stat -c '%U' "$dbfile")" = "root" ]; then
        info "Correggo proprietario di $(basename "$dbfile")..."
        sudo chown "$USER" "$dbfile"
    fi
done

# ── 5. Migrazione/pulizia DB ──────────────────────────────────────────────────
if [ -f "$DATA_DIR/tracks.db" ]; then
    info "Eseguo manutenzione tracks.db (migrazione schema + pulizia cortine)..."
    python3 "$REPO_DIR/convert.py" --db "$DATA_DIR/tracks.db"
    success "DB pronto."
else
    info "Nessun tracks.db esistente, verrà creato al primo avvio."
fi

# ── 6. Cron job per normalizzazione giornaliera ───────────────────────────────
CRON_HOUR=$(echo "$NORMALIZE_AT" | cut -d: -f1)
CRON_MIN=$(echo  "$NORMALIZE_AT" | cut -d: -f2)
DOCKER_BIN="$(which docker)"
CRON_CMD="$CRON_MIN $CRON_HOUR * * * \"$DOCKER_BIN\" compose -f $REPO_DIR/docker-compose.yml --profile normalize run --rm normalizer >> $DATA_DIR/normalize.log 2>&1 # tango-normalize"

# Rimuove eventuali voci precedenti, aggiunge quella aggiornata
( crontab -l 2>/dev/null | { grep -v "# tango-normalize" || true; }; echo "$CRON_CMD" ) | crontab -
success "Cron job configurato: normalizzazione ogni giorno alle $NORMALIZE_AT."

# ── 7. Avvio crawler ──────────────────────────────────────────────────────────
info "Avvio crawler..."
$COMPOSE up -d
success "Crawler avviato."

# ── Riepilogo ─────────────────────────────────────────────────────────────────
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo " Setup completato."
echo ""
echo " Dati:        $DATA_DIR"
echo " Normalizza:  ogni giorno alle $NORMALIZE_AT"
echo " Log cron:    $DATA_DIR/normalize.log"
echo ""
echo " Comandi utili:"
echo "   ./bin/tango help                        # tutti i comandi"
echo "   ./bin/tango query --last 20             # ultimi brani"
echo "   ./bin/tango stats orchestras --limit 10 # top orchestre"
echo "   ./bin/tango logs                        # log crawler"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"