#!/usr/bin/env bash
###############################################################################
# Sistema di Pulizia - Warren Scanner
# Elimina dati temporanei e prepara sistema per produzione
###############################################################################

SCRIPT_DIR="/mnt/ssd/data/python-lab/Trading"
cd "$SCRIPT_DIR" || exit 1

echo "========================================================================"
echo "🧹 PULIZIA SISTEMA WARREN SCANNER"
echo "========================================================================"
echo ""

# 1. Pulizia HTML vecchi (mantieni solo ultimo)
echo "📁 Step 1: Pulizia Report HTML"
echo "----------------------------------------------------------------"
HTML_COUNT=$(ls -1 warren_scan_*.html 2>/dev/null | wc -l)
if [ "$HTML_COUNT" -gt 1 ]; then
    LATEST_HTML=$(ls -1t warren_scan_*.html | head -1)
    echo "  Mantenendo: $LATEST_HTML"
    echo "  Eliminando: $((HTML_COUNT - 1)) file vecchi..."
    ls -1t warren_scan_*.html | tail -n +2 | xargs rm -v
    echo "  ✅ Completato"
else
    echo "  ✅ Solo 1 HTML presente, nessuna pulizia necessaria"
fi
echo ""

# 2. Pulizia Log (richiede sudo)
echo "📊 Step 2: Pulizia Log Files"
echo "----------------------------------------------------------------"
if [ -f "logs/warren_error.log" ]; then
    LOG_SIZE=$(du -h logs/warren_error.log | cut -f1)
    echo "  warren_error.log: $LOG_SIZE"

    if [ -w "logs/warren_error.log" ]; then
        > logs/warren_error.log
        echo "  ✅ warren_error.log svuotato"
    else
        echo "  ⚠️  Richiede sudo: sudo truncate -s 0 logs/warren_error.log"
    fi
fi

if [ -f "logs/warren_scan_cron.log" ]; then
    LOG_SIZE=$(du -h logs/warren_scan_cron.log | cut -f1)
    echo "  warren_scan_cron.log: $LOG_SIZE (mantenuto)"
fi
echo ""

# 3. Pulizia Database Opportunities (opzionale)
echo "🗄️  Step 3: Database Cleanup (Opzionale)"
echo "----------------------------------------------------------------"
echo "  Vuoi pulire le Opportunities di test? (y/N)"
read -r -p "  > " CLEAN_DB

if [[ "$CLEAN_DB" =~ ^[Yy]$ ]]; then
    echo "  Eliminando Opportunities di test..."
    source venv/bin/activate
    python - <<EOF
from src.database.db_manager import DatabaseManager
from src.database.models import Opportunity

db = DatabaseManager('data/trading_system.db')

with db.get_session() as session:
    count = session.query(Opportunity).count()
    print(f'    Prima: {count} Opportunities')

    session.query(Opportunity).delete()
    session.commit()

    print(f'    Dopo: 0 Opportunities')
    print('    ✅ Opportunities eliminate')
EOF
else
    echo "  ⏭️  Saltato (Opportunities mantenute)"
fi
echo ""

# 4. Verifica Sistema
echo "✅ Step 4: Verifica Sistema Pulito"
echo "----------------------------------------------------------------"

# Count files
HTML_FINAL=$(ls -1 warren_scan_*.html 2>/dev/null | wc -l)
TEST_FILES=$(ls -1 test_*.py test_*.sh 2>/dev/null | wc -l)

echo "  📁 Report HTML: $HTML_FINAL"
echo "  🧪 Script test: $TEST_FILES"

# Database stats
source venv/bin/activate
python - <<EOF
from src.database.db_manager import DatabaseManager
from src.database.models import Stock, PriceData, FundamentalData, Opportunity

db = DatabaseManager('data/trading_system.db')

with db.get_session() as session:
    stocks = session.query(Stock).count()
    prices = session.query(PriceData).count()
    funds = session.query(FundamentalData).count()
    opps = session.query(Opportunity).count()

    print(f'  📊 Database:')
    print(f'     • Stocks: {stocks}')
    print(f'     • Prices: {prices:,}')
    print(f'     • Fundamentals: {funds}')
    print(f'     • Opportunities: {opps}')
EOF

echo ""
echo "========================================================================"
echo "✅ PULIZIA COMPLETATA"
echo "========================================================================"
echo ""
echo "📌 Prossimi Passi:"
echo "   1. Sistema pronto per scan domenicale"
echo "   2. Cron attivo: Domenica 19:00"
echo "   3. Email: maurof.gagliardi@gmail.com"
echo ""
echo "💡 Per pulizia log root:"
echo "   sudo truncate -s 0 logs/warren_error.log"
echo ""