import os
import urllib.parse
import requests
from nicegui import ui, app

# NEU: Wir importieren unsere eigene Datenbank-Datei
import database

def setup_routes():
    client_id = os.getenv("DISCORD_CLIENT_ID")
    client_secret = os.getenv("DISCORD_CLIENT_SECRET")
    app_url = os.getenv("APP_URL")

    redirect_uri = f"{app_url}/login/discord"
    encoded_redirect_uri = urllib.parse.quote(redirect_uri, safe="")

    discord_auth_url = f"https://discord.com/api/oauth2/authorize?client_id={client_id}&redirect_uri={encoded_redirect_uri}&response_type=code&scope=identify"

    # --- HOME PAGE ---
    @ui.page('/')

    def home_page():
        ui.dark_mode(True)
        with ui.card().classes('w-full items-center mt-10'):
            ui.label('Warhammer League').classes('text-2xl font-bold')

            # --- LOGIN BEREICH ---
            if app.storage.user.get('authenticated', False):
                discord_name = app.storage.user.get('discord_name')
                db_id = app.storage.user.get('db_id')

                ui.label(f'Welcome back, {discord_name}!').classes('text-green-500 font-bold')

                ui.button('Enter Match', on_click=lambda: ui.navigate.to('/add-match')).classes('mt-4 bg-blue-500 text-white')

                def logout():
                    app.storage.user.clear()
                    ui.navigate.to('/')

                ui.button('Logout', on_click=logout).classes('mt-4 bg-red-500 text-white')
            else:
                ui.button('Login with Discord', on_click=lambda: ui.navigate.to(discord_auth_url))

        # --- LEADERBOARD BEREICH ---
        with ui.card().classes('w-full items-center mt-6'):
            ui.label('Leaderboard').classes('text-xl font-bold mb-4')

            # 1. Spalten (Columns) für die Tabelle definieren
            table_columns = [
                {'name': 'name', 'label': 'Player', 'field': 'name', 'align': 'left'},
                {'name': 'mmr', 'label': 'MMR', 'field': 'mmr', 'sortable': True},
                {'name': 'points', 'label': 'Points', 'field': 'points', 'sortable': True},
                {'name': 'games', 'label': 'Games Played', 'field': 'games', 'sortable': True},
            ]

            # 2. Daten (Rows) aus der Datenbank holen
            player_data = database.get_all_players()

            # 3. Tabelle zeichnen
            ui.table(columns=table_columns, rows=player_data, row_key='name').classes('w-full max-w-4xl')



    # --- DISCORD CALLBACK PAGE ---
    @ui.page('/login/discord')
    def discord_callback(code: str = None):
        if not code:
            ui.label('Fehler: Kein Code erhalten.').classes('text-red-500')
            return

        token_data = {
            'client_id': client_id,
            'client_secret': client_secret,
            'grant_type': 'authorization_code',
            'code': code,
            'redirect_uri': redirect_uri
        }

        token_response = requests.post('https://discord.com/api/oauth2/token', data=token_data)
        token_json = token_response.json()

        if 'access_token' in token_json:
            access_token = token_json['access_token']

            user_headers = {
                'Authorization': f"Bearer {access_token}"
            }
            user_response = requests.get('https://discord.com/api/users/@me', headers=user_headers)
            user_json = user_response.json()

            # --- NEU: DATENBANK INTEGRATION ---
            discord_id = user_json['id']
            discord_name = user_json['username']

            # 1. Spieler in der DB suchen oder neu anlegen
            player = database.get_or_create_player(discord_id, discord_name)

            # 2. Session-Daten aktualisieren
            app.storage.user['authenticated'] = True
            app.storage.user['discord_id'] = discord_id
            app.storage.user['discord_name'] = discord_name

            # player ist ein Tuple aus der DB, z.B.: (1, "123456", "Daniel", 1000, 0, 0)
            # player[0] ist die interne Datenbank-ID. Diese merken wir uns in der Session!
            app.storage.user['db_id'] = player[0]

            ui.navigate.to('/')
        else:
            ui.label('Fehler beim Login!').classes('text-red-500')
            ui.label(str(token_json))