import requests
import datetime
import time
import json
import os
from dateutil import parser
from zoneinfo import ZoneInfo

# --- CONFIGURAZIONE UTENTE ---
# 👇👇 INSERISCI QUI I TUOI DATI 👇👇
TELEGRAM_BOT_TOKEN = "8155587974:AAF9OekvBpixtk8ZH6KoIc0L8edbhdXt7A4"
TELEGRAM_CHAT_IDS = ["64463169", "24827341", "132455422", "5405962012"]

# --- SOGLIE DI ALLARME (Bologna) ---
SOGLIA_NEVE = 0.0        # cm (Basta neve per attivare)
SOGLIA_PIOGGIA_3H = 30.0 # mm in 3 ore (Pioggia molto forte)

# File di stato per la memoria
STATE_FILE = "/home/daniely/docker/telegram-bot/student_state.json"

# --- PUNTI DEL PERCORSO (Caselli A14) ---
POINTS = [
    {"name": "🎓 Bologna (V. Regnoli)", "lat": 44.4930, "lon": 11.3690, "type": "trigger"}, 
    {"name": "🛣️ Casello Imola",       "lat": 44.3798, "lon": 11.7397, "type": "route"},
    {"name": "🛣️ Casello Faenza",      "lat": 44.3223, "lon": 11.9040, "type": "route"},
    {"name": "🛣️ Casello Forlì",       "lat": 44.2502, "lon": 12.0910, "type": "route"},
    {"name": "🛣️ Casello Cesena",      "lat": 44.1675, "lon": 12.2835, "type": "route"},
    {"name": "🛣️ Casello Rimini",      "lat": 44.0362, "lon": 12.5659, "type": "route"},
    {"name": "🏠 San Marino",          "lat": 43.9356, "lon": 12.4296, "type": "end"}
]

# --- FUNZIONI DI UTILITÀ ---

def load_last_state():
    """Legge se c'era un allerta attiva"""
    if not os.path.exists(STATE_FILE): return False
    try:
        with open(STATE_FILE, 'r') as f:
            data = json.load(f)
            return data.get("alert_active", False)
    except: return False

def save_current_state(is_active):
    """Salva lo stato corrente"""
    try:
        with open(STATE_FILE, 'w') as f:
            json.dump({"alert_active": is_active, "updated": str(datetime.datetime.now())}, f)
    except Exception as e:
        print(f"Errore salvataggio stato: {e}")

def send_telegram_message(message):
    if not TELEGRAM_BOT_TOKEN or "INSERISCI" in TELEGRAM_BOT_TOKEN:
        print(f"[TEST OUT] {message}")
        return

    url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
    for chat_id in TELEGRAM_CHAT_IDS:
        try:
            requests.post(url, json={"chat_id": chat_id, "text": message, "parse_mode": "Markdown"}, timeout=10)
            time.sleep(0.2)
        except: pass

def get_forecast(lat, lon):
    url = "https://api.open-meteo.com/v1/forecast"
    params = {
        "latitude": lat, "longitude": lon,
        "hourly": "precipitation,snowfall",
        "models": "arome_france_hd",
        "timezone": "Europe/Rome",
        "forecast_days": 2
    }
    try:
        res = requests.get(url, params=params, timeout=5)
        res.raise_for_status()
        return res.json()
    except: return None

def get_stats(data):
    if not data: return None
    hourly = data.get("hourly", {})
    times = hourly.get("time", [])
    snow = hourly.get("snowfall", [])
    rain = hourly.get("precipitation", [])
    
    now = datetime.datetime.now(ZoneInfo("Europe/Rome"))
    
    start_idx = -1
    for i, t in enumerate(times):
        if parser.isoparse(t).replace(tzinfo=ZoneInfo("Europe/Rome")) >= now.replace(minute=0,second=0,microsecond=0):
            start_idx = i
            break
    if start_idx == -1: return None
    
    limit = min(start_idx + 24, len(times))
    
    def sum_slice(arr, hours):
        return sum(x for x in arr[start_idx:min(start_idx+hours, limit)] if x)

    return {
        "snow_3h": sum_slice(snow, 3),
        "snow_6h": sum_slice(snow, 6),
        "snow_12h": sum_slice(snow, 12),
        "snow_24h": sum_slice(snow, 24),
        "rain_3h": sum_slice(rain, 3),
        "rain_max": max(rain[start_idx:limit]) if rain else 0
    }

# --- LOGICA PRINCIPALE ---

def main():
    print("--- Analisi Studente Bologna ---")
    
    now_str = datetime.datetime.now(ZoneInfo("Europe/Rome")).strftime('%H:%M')
    
    # 1. ANALISI BOLOGNA (Il Trigger)
    bo_point = POINTS[0]
    bo_data = get_forecast(bo_point["lat"], bo_point["lon"])
    bo_stats = get_stats(bo_data)
    
    if not bo_stats: return

    # Controlla se scatta l'allarme
    alarm_snow = bo_stats["snow_24h"] > SOGLIA_NEVE
    alarm_rain = bo_stats["rain_3h"] > SOGLIA_PIOGGIA_3H
    
    # Carica stato precedente
    WAS_ACTIVE = load_last_state()

    # --- SCENARIO A: C'È ALLERTA ---
    if alarm_snow or alarm_rain:
        icon_main = "❄️" if alarm_snow else "🌧️"
        
        msg = f"{icon_main} **ALLERTA METEO BOLOGNA**\n"
        msg += f"📅 _Aggiornamento ore {now_str}_\n\n"
        
        # Dettaglio Bologna
        msg += f"🎓 **A BOLOGNA:**\n"
        if alarm_snow:
            msg += f"• Neve 3h: **{bo_stats['snow_3h']:.1f}** cm\n"
            msg += f"• Neve 6h: **{bo_stats['snow_6h']:.1f}** cm\n"
            msg += f"• Neve 12h: **{bo_stats['snow_12h']:.1f}** cm\n"
            msg += f"• Neve 24h: **{bo_stats['snow_24h']:.1f}** cm\n"
        
        if alarm_rain:
            msg += f"• Pioggia 3h: **{bo_stats['rain_3h']:.1f}** mm (Intensa!)\n"
        
        msg += "\n🚗 **SITUAZIONE AI CASELLI (A14):**\n"
        
        # 2. ANALISI PERCORSO (Solo se c'è allerta)
        route_issues = False
        
        for p in POINTS[1:]:
            stats = get_stats(get_forecast(p["lat"], p["lon"]))
            if not stats: continue
            
            has_snow = stats["snow_24h"] > 0
            has_rain = stats["rain_3h"] > 5.0
            
            if has_snow or has_rain:
                route_issues = True
                line = f"**{p['name']}**: "
                if has_snow: line += f"❄️ {stats['snow_12h']:.1f}cm (12h) "
                if has_rain: line += f"🌧️ {stats['rain_3h']:.1f}mm "
                msg += f"{line}\n"
        
        if not route_issues:
            msg += "✅ I caselli autostradali sembrano puliti."
        
        send_telegram_message(msg)
        save_current_state(True)
        print("Allerta inviata.")

    # --- SCENARIO B: ALLARME RIENTRATO ---
    elif not (alarm_snow or alarm_rain) and WAS_ACTIVE:
        msg = (
            f"🟢 **ALLARME RIENTRATO (Bologna)**\n"
            f"📅 _Aggiornamento ore {now_str}_\n\n"
            f"Le previsioni non indicano più neve o piogge critiche per le prossime 24 ore.\n"
            f"Situazione tornata alla normalità."
        )
        send_telegram_message(msg)
        save_current_state(False)
        print("Allarme rientrato. Notifica inviata.")

    # --- SCENARIO C: TRANQUILLO ---
    else:
        save_current_state(False)
        print("Nessuna allerta.")

if __name__ == "__main__":
    main()