# wcs/kraska/KraskaClient.py
"""
KRA.sk API client pro Milionář addon.
Poskytuje login, resolve a správu předplatného.
"""

import requests
import xbmc
import xbmcgui


BASE = 'https://api.kra.sk'
UA = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
HEADERS = {'User-Agent': UA, 'Content-Type': 'application/json'}

_session = requests.Session()
_session.headers.update(HEADERS)

# Cache pro user info
_user_info_cache = None


def _log(msg, level=xbmc.LOGINFO):
    """Logování s prefixem."""
    xbmc.log(f"[plugin.video.milionar] KraskaClient: {msg}", level=level)


def _get_data(endpoint, data=None, token=None):
    """
    Volání KRA.sk API.
    
    Args:
        endpoint: API endpoint (např. '/api/user/login')
        data: dict s daty pro request
        token: session token (volitelné)
    
    Returns:
        dict s odpovědí nebo False při chybě
    """
    if data is None:
        data = {}
    if token:
        data['session_id'] = token
    
    try:
        _log(f"API call: {endpoint}")
        response = _session.post(BASE + endpoint, json=data, timeout=15)
        return response.json()
    except Exception as e:
        _log(f"API error: {e}", level=xbmc.LOGERROR)
        return False


def login(username, password, addon, popinfo, open_settings_on_fail=True):
    """
    Přihlášení ke KRA.sk.
    
    Args:
        username: uživatelské jméno
        password: heslo
        addon: xbmcaddon instance
        popinfo: callback pro notifikace
        open_settings_on_fail: otevřít nastavení při chybě
    
    Returns:
        session_id token nebo None
    """
    if not username or not password:
        popinfo(addon.getLocalizedString(33024), addon.getAddonInfo('name'), sound=True)
        if open_settings_on_fail:
            addon.openSettings()
        return None
    
    _log(f"Login attempt for user: {username}")
    
    data = {
        'data': {
            'username': username,
            'password': password
        }
    }
    
    response = _get_data('/api/user/login', data)
    
    if response and 'session_id' in response:
        token = response['session_id']
        addon.setSetting('kraska_token', token)
        addon.setSetting('kraska_user', username)
        addon.setSetting('kraska_pass', password)
        # Verify token was saved
        saved = addon.getSetting('kraska_token')
        _log("Login successful")
        return token
    else:
        error_msg = response.get('msg', 'Unknown error') if response else 'Connection failed'
        _log(f"Login failed: {error_msg}", level=xbmc.LOGWARNING)
        popinfo(addon.getLocalizedString(33024), addon.getAddonInfo('name'), icon='error', sound=True)
        if open_settings_on_fail:
            addon.openSettings()
        return None


def revalidate(addon, popinfo):
    """
    Ověření a obnova tokenu.
    
    Returns:
        Platný token nebo None
    """
    token = addon.getSetting('kraska_token')
    
    if not token:
        # Zkusit přihlásit se uloženými údaji
        username = addon.getSetting('kraska_user')
        password = addon.getSetting('kraska_pass')
        if username and password:
            return login(username, password, addon, popinfo, open_settings_on_fail=False)
        return None
    
    # Ověřit platnost tokenu
    info = get_user_info(addon, popinfo, token_override=token)
    if info:
        return token
    
    # Token neplatný, zkusit znovu přihlásit
    _log("Token invalid, attempting re-login")
    addon.setSetting('kraska_token', '')
    username = addon.getSetting('kraska_user')
    password = addon.getSetting('kraska_pass')
    if username and password:
        return login(username, password, addon, popinfo, open_settings_on_fail=False)
    
    return None


def get_user_info(addon, popinfo, token_override=None):
    """
    Získání informací o uživateli.
    
    Returns:
        dict s user info nebo None
    """
    global _user_info_cache
    
    token = token_override or addon.getSetting('kraska_token')
    if not token:
        return None
    
    response = _get_data('/api/user/info', {}, token=token)
    
    if response and 'data' in response:
        _user_info_cache = response['data']
        return _user_info_cache
    
    return None


def get_days_left(addon, popinfo):
    """
    Získání počtu dnů do expirace předplatného.
    
    Returns:
        int počet dnů nebo 0
    """
    info = get_user_info(addon, popinfo)
    if not info:
        return 0
    
    return info.get('days_left', 0)


def resolve(ident, addon, server=None):
    """
    Resolve ident na stream URL.
    
    Args:
        ident: identifikátor souboru
        addon: xbmcaddon instance
        server: volitelný custom server
    
    Returns:
        Stream URL nebo None
    """
    from wcs.ui import UI as ui
    
    token = revalidate(addon, ui.popinfo)
    if not token:
        _log("Resolve failed: no valid token")
        return None
    
    # Kontrola předplatného
    days_left = get_days_left(addon, ui.popinfo)
    if days_left <= 0:
        _log("Resolve failed: no active subscription")
        xbmcgui.Dialog().notification(
            addon.getAddonInfo('name'),
            addon.getLocalizedString(33020),  # Žádné aktivní předplatné
            xbmcgui.NOTIFICATION_WARNING,
            5000
        )
        return None
    
    # Varování na blížící se expiraci
    try:
        warn_days = int(addon.getSetting('kraska_warn_days') or '14')
    except Exception:
        warn_days = 14
    
    if days_left <= warn_days:
        xbmcgui.Dialog().notification(
            addon.getAddonInfo('name'),
            addon.getLocalizedString(33026).format(days=days_left),
            xbmcgui.NOTIFICATION_WARNING,
            5000
        )
    
    # Resolve souboru
    _log(f"Resolving ident: {ident}")
    
    data = {
        'data': {
            'ident': ident
        }
    }
    
    response = _get_data('/api/file/download', data, token=token)
    
    if response and 'data' in response:
        file_data = response['data']
        if 'link' in file_data:
            url = file_data['link']
            _log(f"Resolved URL: {url}")
            return url
    
    error_msg = response.get('msg', 'Unknown error') if response else 'Connection failed'
    _log(f"Resolve failed: {error_msg}", level=xbmc.LOGWARNING)
    return None


def refresh_subscription_info(addon, token_override=None, show_notification=False, notification_prefix=""):
    """
    Obnovení informací o předplatném v settings.
    
    Args:
        addon: xbmcaddon instance
        token_override: volitelný token
        show_notification: zobrazit notifikaci
        notification_prefix: prefix pro notifikaci (např. "Přihlášen: username, ")
    
    Returns:
        True pokud úspěšné
    """
    from wcs.ui import UI as ui
    
    info = get_user_info(addon, ui.popinfo, token_override=token_override)
    
    if not info:
        addon.setSetting('kraska_status', addon.getLocalizedString(33017))  # Nepřihlášen
        addon.setSetting('kraska_days_left', '')
        return False
    
    username = info.get('username', '')
    days_left = info.get('days_left', 0)
    subscribed_until = info.get('subscribed_until', '')
    
    # Nastavit status
    if username:
        addon.setSetting('kraska_status', addon.getLocalizedString(33018).format(name=username))
    else:
        addon.setSetting('kraska_status', addon.getLocalizedString(33017))
    
    # Nastavit dny do expirace
    vip_text = ""
    if days_left and days_left > 0:
        vip_text = addon.getLocalizedString(33019).format(days=days_left)
        addon.setSetting('kraska_days_left', vip_text)
    elif subscribed_until:
        vip_text = f"Do: {subscribed_until}"
        addon.setSetting('kraska_days_left', vip_text)
    else:
        vip_text = addon.getLocalizedString(33020)  # Žádné předplatné
        addon.setSetting('kraska_days_left', vip_text)
    
    if show_notification:
        # Sestavit notifikaci: prefix + VIP info (jako Webshare)
        if notification_prefix:
            notification_text = f"{notification_prefix}{vip_text}"
        else:
            notification_text = f"{username}: {vip_text}" if username else vip_text
        
        xbmcgui.Dialog().notification(
            addon.getAddonInfo('name'),
            notification_text,
            xbmcgui.NOTIFICATION_INFO,
            5000
        )
    
    _log(f"Subscription info updated: {days_left} days left")
    return True


def delete_account(addon):
    """
    Smazání uložených údajů KRA.sk.
    """
    addon.setSetting('kraska_user', '')
    addon.setSetting('kraska_pass', '')
    addon.setSetting('kraska_token', '')
    addon.setSetting('kraska_status', addon.getLocalizedString(33017))  # Nepřihlášen
    addon.setSetting('kraska_days_left', '')
    
    global _user_info_cache
    _user_info_cache = None
    
    xbmcgui.Dialog().notification(
        addon.getAddonInfo('name'),
        addon.getLocalizedString(33025),  # Účet smazán
        xbmcgui.NOTIFICATION_INFO,
        3000
    )
    _log("Account data deleted")
