import json
import os
import time
from pathlib import Path
from urllib.parse import quote

import requests
from flask import Flask, Response, render_template_string, request, send_from_directory
from flask_cors import CORS

from config import BASE_DIR, CONFIG, VOICE_AVAILABLE

app = Flask(__name__, static_folder=str(BASE_DIR / "static"), static_url_path="/static")
CORS(app)

# Directories that must never be served by the catch-all asset route.
PRIVATE_DIRS = {"gateway", "reference", "static", ".git", ".venv", "__pycache__"}

# Everything the browser voice client needs; never include secrets here.
VOICE_PUBLIC_CONFIG = {
    "enabled": VOICE_AVAILABLE,
    "wsPath": "/ws/voice",
    "channelBanners": CONFIG["channel_banners"],
    "channelIcons": CONFIG["channel_icons"],
    "genericImages": CONFIG["generic_images"],
    "roleIcons": CONFIG["role_icons"],
}

ALLOWED_STATIC_EXTENSIONS = {
    ".css",
    ".png",
    ".jpg",
    ".jpeg",
    ".gif",
    ".webp",
    ".bmp",
    ".svg",
    ".ico",
}


def safe_url(url):
    raw = str(url or "").strip()
    if not raw:
        return ""
    return quote(raw, safe=":/?&=#%+~@!$,;*-._")



CLIENT_DETAIL_FIELDS = ("afk_text", "is_afk", "is_streaming", "stream_text")
# After a failed per-user clientinfo call, skip per-user calls for this long so a
# misbehaving server is not hammered on every refresh.
CLIENT_DETAILS_RETRY_SECONDS = 300
_client_details_paused_until = 0.0


class WebQueryError(Exception):
    """The WebQuery answered, but with an error (HTTP or TeamSpeak status)."""

    def __init__(self, message, code=None):
        super().__init__(message)
        self.code = code


# TeamSpeak "invalid clientID": the user left between clientlist and clientinfo.
TS_ERROR_INVALID_CLIENT_ID = "512"


def webquery_get(url, headers):
    res = requests.get(url, headers=headers, timeout=2)
    res.raise_for_status()
    payload = res.json()
    status = payload.get("status") if isinstance(payload, dict) else None
    if isinstance(status, dict) and str(status.get("code", 0)) != "0":
        code = str(status.get("code"))
        raise WebQueryError(f"{code}: {status.get('message', '')}", code=code)
    return payload


def fetch_client_details(base_url, headers, client_list):
    """Add AFK/streaming state per user via clientinfo.

    All or nothing: if any call fails, every user's details are dropped and the
    plain channel/user list is used instead. Returns True on success.
    """
    global _client_details_paused_until
    try:
        for client in client_list:
            clid = client.get("clid")
            if not clid or client.get("client_nickname") == "serveradmin":
                continue
            body = None
            try:
                body = webquery_get(f"{base_url}/clientinfo?clid={clid}", headers).get("body")
            except WebQueryError as e:
                if e.code != TS_ERROR_INVALID_CLIENT_ID:
                    raise
            client_info = body[0] if isinstance(body, list) and body else {}
            client["afk_text"] = client_info.get("client_away_message", "")
            client["is_afk"] = client_info.get("client_away") == "1"
            client["is_streaming"] = client_info.get("client_is_streaming") == "1"
            client["stream_text"] = client_info.get("client_stream_text", "")
        return True
    except (requests.RequestException, ValueError, WebQueryError) as e:
        print(
            f"clientinfo failed ({e}); showing users without AFK/streaming details "
            f"for the next {CLIENT_DETAILS_RETRY_SECONDS}s"
        )
        _client_details_paused_until = time.time() + CLIENT_DETAILS_RETRY_SECONDS
        for client in client_list:
            for field in CLIENT_DETAIL_FIELDS:
                client.pop(field, None)
        return False


def fetch_teamspeak_data(cfg):
    cache_file = Path(cfg["cache_file"])
    if cache_file.exists() and (time.time() - cache_file.stat().st_mtime) < cfg["cache_time"]:
        try:
            return json.loads(cache_file.read_text(encoding="utf-8"))
        except (OSError, json.JSONDecodeError):
            pass

    base_url = f"https://{cfg['host']}:{cfg['port']}/{cfg['server_id']}"
    headers = {"x-api-key": cfg["api_key"]}

    try:
        channels_json = webquery_get(f"{base_url}/channellist", headers)
        clients_json = webquery_get(f"{base_url}/clientlist?-groups", headers)
    except (requests.RequestException, ValueError, WebQueryError) as e:
        print(f"Failed to fetch channel/client list: {e}")
        return {"channels": [], "clients": []}

    client_list = clients_json.get("body") or []

    afk_threshold = cfg.get("afk_status_max_clients")
    fetch_afk_status = (
        (afk_threshold is None or len(client_list) <= afk_threshold)
        and time.time() >= _client_details_paused_until
    )
    if fetch_afk_status:
        fetch_client_details(base_url, headers, client_list)

    data = {
        "channels": channels_json.get("body") or [],
        "clients": client_list or [],
    }

    try:
        cache_file.write_text(json.dumps(data), encoding="utf-8")
    except OSError:
        pass

    return data


def same_id(left, right):
    return str(left or 0) == str(right or 0)


def sort_ts_channels(channels, pid=0):
    ordered = []
    by_order = {}

    for channel in channels:
        if same_id(channel.get("pid", 0), pid):
            by_order[str(channel.get("channel_order", 0))] = channel

    current_order = "0"
    max_iterations = len(channels) + 1
    iterations = 0

    while current_order in by_order and iterations < max_iterations:
        channel = by_order[current_order]
        ordered.append(channel)
        ordered.extend(sort_ts_channels(channels, channel.get("cid")))
        current_order = str(channel.get("cid"))
        iterations += 1

    return ordered


def build_context():
    start_time = time.perf_counter()
    ts_data = fetch_teamspeak_data(CONFIG["server"])
    api_ping = round((time.perf_counter() - start_time) * 1000)

    raw_channels = ts_data.get("channels") or []
    afk_count = None
    for channel in raw_channels:
        original_name = channel.get("channel_name", "")
        channel["is_spacer"] = "[Cspacer]" in original_name and "AFK" not in original_name
        # Channels with the keyword "AFK" count all connected clients as AFK regardless of status
        if "AFK" in original_name:
            afk_count = channel.get("total_clients", None)
        channel["channel_name"] = original_name.replace("[Cspacer]", "")
    raw_clients = ts_data.get("clients") or []
    is_online = bool(raw_channels)
    if afk_count is not None:
        afk_count = int(afk_count) - 1
    else:
        afk_count = 0

    channels = sort_ts_channels(raw_channels, 0)

    clients_by_channel = {}
    total_real_clients = 0

    if is_online:
        for client in raw_clients:
            if int(client.get("client_type") or 0) == 1:
                continue
            if client.get("is_afk"):
                afk_count += 1
            clients_by_channel.setdefault(str(client.get("cid")), []).append(client)
            total_real_clients += 1

    return {
        "config": CONFIG,
        "api_ping": api_ping,
        "channels": channels,
        "clients_by_channel": clients_by_channel,
        "is_online": is_online,
        "safe_url": safe_url,
        "total_real_clients": total_real_clients,
        "afk_count": afk_count,
        "voice_available": VOICE_AVAILABLE,
        "voice_public": VOICE_PUBLIC_CONFIG,
    }


FRAGMENT_TEMPLATE = """
{% if is_online %}
    <div class="server-stats">
        <div class="stat-item">
            <span class="stat-label">Users Connected</span>
            <span class="stat-value">{{ total_real_clients }}</span>
        </div>
        {% if afk_count is not none %}
        <div class="stat-item">
            <span class="stat-label">AFK Users</span>
            <span class="stat-value">{{ afk_count }}</span>
        </div>
        {% endif %}
    </div>
    <div class="channel-list">
        {% for channel in channels %}
            {% set c_name = channel.get('channel_name', '') %}
            {% set cid = channel.get('cid')|string %}
            {% set channel_clients = clients_by_channel.get(cid, []) %}
            {% set bg_img = config.channel_banners.get(c_name, config.generic_images.get('background', '')) %}
            {% set icon_img = config.channel_icons.get(c_name, config.generic_images.get('icon', '')) %}

            {% if bg_img %}
                {% set bg_style = "background-image: linear-gradient(90deg, rgba(9, 18, 32, 0.96) 0%, rgba(9, 18, 32, 0.78) 45%, rgba(9, 18, 32, 0.94) 100%), url('" ~ safe_url(bg_img) ~ "');" %}
            {% else %}
                {% set bg_style = "" %}
            {% endif %}

            {% if not channel.is_spacer %}
            <article class="channel" data-cid="{{ cid }}">
                <header class="channel-header" style="{{ bg_style }}">
                    <div class="channel-title-group">
                        {% if icon_img %}
                            <img src="{{ safe_url(icon_img) }}" class="channel-icon" alt="" aria-hidden="true">
                        {% endif %}
                        <h2>{{ c_name }}</h2>
                    </div>
                    {# Only show client count for non-spacer channels #}
                    <div class="channel-actions">
                        <span class="channel-meta">
                            {{ channel_clients|length }}
                        </span>
                        {% if voice_available %}
                            <button type="button" class="channel-join" data-cid="{{ cid }}" data-name="{{ c_name }}" aria-label="Join {{ c_name }} from the browser">join</button>
                        {% endif %}
                    </div>
                </header>

                {# Only show client list for non-spacer channels #}
                    <ul class="client-list">
                        {% for client in channel_clients %}
                            <li class="client" data-clid="{{ client.get('clid', '') }}">
                                {% set role_icon = namespace(path='') %}
                                {% for group_id in (client.get('client_servergroups', '')|string).split(',') %}
                                    {% if not role_icon.path and group_id in config.role_icons %}
                                        {% set role_icon.path = config.role_icons[group_id] %}
                                    {% endif %}
                                {% endfor %}

                                {% if role_icon.path %}
                                    <img src="{{ safe_url(role_icon.path) }}" class="role-icon" alt="Role">
                                {% else %}
                                    <span class="client-no-role-icon"></span>
                                {% endif %}
                                <span class="client-name">{{ client.get('client_nickname', '') }}</span>
                                {% if client.get('is_afk') %}
                                    <span class="client-afk">(AFK{% if client.get('afk_text') %}: {{ client.get('afk_text') }}{% endif %})</span>
                                {% endif %}
                                {% if client.get('is_streaming') %}
                                    <span class="client-stream">(Streaming{% if client.get('stream_text') %}: {{ client.get('stream_text') }}{% endif %})</span>
                                {% endif %}
                            </li>
                        {% endfor %}
                    </ul>
            </article>
            {% endif %}
        {% endfor %}
    </div>
{% else %}
    <div class="empty-server">Unable to reach the TeamSpeak server.</div>
{% endif %}
"""

PAGE_TEMPLATE = """
<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <meta name="description" content="{{ config.ui.subtitle }}">
    <title>{{ config.ui.title }}</title>
    <link rel="icon" type="image/png" href="logo.png">
    <link rel="stylesheet" href="index.css">
    <style>
        :root {
            --bg-color: {{ config.ui.background_color }};
            --panel-bg: {{ config.ui.panel_color }};
            --accent: {{ config.ui.accent_color }};
            --text-main: {{ config.ui.text_color }};
        }

    </style>
</head>
<body>


<main class="app-container">
    {% if is_online %}<div class="progress-bar" id="refresh-bar"></div>{% endif %}

    {% if voice_available %}
    <section class="voice-topbar" id="voice-topbar" aria-label="Browser voice setup">
        <div class="voice-topbar-bar">
            <span class="voice-topbar-identity" id="voice-topbar-identity">[ not logged in ]</span>
            <div class="voice-topbar-actions">
                <button type="button" class="btn-voice btn-voice-rnnoise voice-topbar-btn" id="voice-wasm-open-btn" aria-pressed="true" title="Open RNNoise WASM settings">rnnoise: on</button>
                <button type="button" class="voice-login-btn" id="voice-login-btn">browser connect</button>
            </div>
        </div>
        <p class="voice-topbar-hint" id="voice-message" role="status" aria-live="polite" hidden></p>
    </section>
    {% endif %}

    {% if config.ui.banner_url %}
        <img src="{{ safe_url(config.ui.banner_url) }}" alt="Header banner" class="banner">
    {% endif %}

    <div class="content-wrapper">
        {% if voice_available %}<p class="voice-notice" id="voice-notice" role="status" aria-live="polite" hidden></p>{% endif %}

        <section id="ajax-container" aria-live="polite">
            {{ html_fragment|safe }}
        </section>

        {% if config.external_links %}
        <nav class="footer-links">
            {% for name, url in config.external_links.items() %}
                <a href="{{ url }}" target="_blank" rel="noopener">{{ name }}</a>
            {% endfor %}
        </nav>
        {% endif %}

        <header class="header">
            <h1>{{ config.ui.title }}</h1>
            <p>{{ config.ui.subtitle }}</p>
            {% if is_online %}
                <div class="status-badge status-online"><span class="pulse-dot"></span>SERVER ONLINE</div>
            {% else %}
                <div class="status-badge status-offline"><span class="pulse-dot"></span> SERVER OFFLINE</div>
            {% endif %}
        </header>

        <a href="ts3server://{{ config.server.ts_domain }}" class="btn-primary">{{ config.buttons.join }}</a>
        <p class="help-text">{{ config.buttons.download_text }} <a href="{{ config.buttons.download_url }}" target="_blank" rel="noopener">{{ config.buttons.download_link }}</a></p>

        <footer class="credits">
            {{ config.footer.text }} <a href="{{ config.footer.author_url }}" target="_blank" rel="noopener">{{ config.footer.author }}</a> — Refresh scan: <span id="timer-text">{{ config.server.cache_time }}</span>s
        </footer>
    </div>
</main>

{% if voice_available %}
<dialog class="voice-dialog" id="voice-dialog" aria-labelledby="voice-dialog-title">
    <form class="voice-dialog-form" id="voice-join-form" autocomplete="on">
        <header class="voice-dialog-header">
            <h2 id="voice-dialog-title">Browser connect</h2>
            <span class="voice-dialog-target" id="voice-dialog-target">default channel</span>
        </header>
        <div class="voice-dialog-body">
            <label class="voice-field">
                <span class="voice-field-tag">[ nickname ]</span>
                <input id="voice-nickname" name="nickname" minlength="3" maxlength="30" required autocomplete="nickname" spellcheck="false" placeholder="enter name...">
            </label>
            <label class="voice-field">
                <span class="voice-field-tag">[ server password ]</span>
                <input id="voice-password" name="password" type="password" required autocomplete="current-password" placeholder="enter password...">
            </label>
            <p class="voice-dialog-message" id="voice-dialog-message" role="status" aria-live="polite" hidden></p>
        </div>
        <footer class="voice-dialog-actions">
            <button type="button" class="btn-voice" id="voice-cancel">cancel</button>
            <button type="submit" class="btn-primary voice-submit" id="voice-submit">connect</button>
        </footer>
    </form>
</dialog>

<dialog class="voice-dialog" id="voice-wasm-dialog" aria-labelledby="voice-wasm-dialog-title">
    <div class="voice-dialog-form">
        <header class="voice-dialog-header">
            <h2 id="voice-wasm-dialog-title">RNNoise WASM</h2>
            <span class="voice-dialog-target voice-wasm-status" id="voice-wasm-status">not cached</span>
        </header>
        <div class="voice-dialog-body">
            <div class="voice-field">
                <span class="voice-field-tag">[ background noise suppression ]</span>
                <p class="voice-wasm-desc">Filters microphone input through RNNoise (WebAssembly) before audio is sent to the voice gateway.</p>
                <div class="voice-wasm-controls">
                    <button type="button" class="btn-voice btn-voice-rnnoise" id="voice-rnnoise-toggle" aria-pressed="true">rnnoise: on</button>
                </div>
                <p class="voice-wasm-desc">Processing: <span class="voice-wasm-status" id="voice-wasm-active" data-state="idle">not in a call</span></p>
            </div>
            <div class="voice-field">
                <span class="voice-field-tag">[ wasm model cache ]</span>
                <div class="voice-wasm-controls">
                    <button type="button" class="btn-voice" id="voice-wasm-redownload">re-download</button>
                    <button type="button" class="btn-voice btn-voice-danger" id="voice-wasm-delete">delete</button>
                </div>
            </div>
            <p class="voice-dialog-message" id="voice-wasm-message" role="status" aria-live="polite" hidden></p>
        </div>
        <footer class="voice-dialog-actions">
            <button type="button" class="btn-voice" id="voice-wasm-close">close</button>
        </footer>
    </div>
</dialog>

<section class="voice-bar" id="voice-bar" aria-label="Voice controls" hidden>
    <div class="voice-meter" aria-hidden="true">
        <div class="voice-meter-fill" id="voice-meter"></div>
        <div class="voice-meter-threshold" id="voice-threshold"></div>
    </div>
    <div class="voice-settings" id="voice-settings" hidden>
        <div class="voice-setting">
            <span class="stat-label">Input mode</span>
            <button type="button" class="btn-voice" id="voice-mode">voice activity</button>
        </div>
        <label class="voice-setting">
            <span class="stat-label">Mic sensitivity</span>
            <input type="range" id="voice-sensitivity" min="0" max="100" value="45">
        </label>
        <label class="voice-setting">
            <span class="stat-label">Volume</span>
            <input type="range" id="voice-volume" min="0" max="150" value="100">
        </label>
    </div>
    <p class="voice-bar-message" id="voice-bar-message" role="status" aria-live="polite" hidden></p>
    <div class="voice-bar-main">
        <div class="voice-bar-info">
            <span class="voice-live-badge"><span class="pulse-dot"></span>Live</span>
            <span class="voice-where"><span id="voice-channel">-</span><span class="voice-as">as</span><span id="voice-self">-</span></span>
        </div>
        <div class="voice-buttons">
            <button type="button" class="btn-voice btn-voice-talk" id="voice-ptt" hidden>talk (V)</button>
            <button type="button" class="btn-voice" id="voice-mute" aria-pressed="false">mute</button>
            <button type="button" class="btn-voice" id="voice-deafen" aria-pressed="false">deafen</button>
            <button type="button" class="btn-voice" id="voice-settings-toggle" aria-expanded="false" aria-controls="voice-settings">settings</button>
            <button type="button" class="btn-voice btn-voice-danger" id="voice-leave">leave</button>
        </div>
    </div>
</section>
<script id="voice-config" type="application/json">{{ voice_public|tojson }}</script>
{% endif %}

<script>
    // Reloads the server-rendered channel list. Skipped while the web voice
    // client renders it live; voice.js calls it again after leaving.
    window.refreshDashboard = function () {
        const container = document.getElementById('ajax-container');
        if (!container || document.body.classList.contains('voice-live')) return Promise.resolve();
        container.style.opacity = '0.5';
        return fetch(window.location.href.split('?')[0] + '?ajax=1', { cache: 'no-store', headers: {'X-Requested-With': 'XMLHttpRequest'} })
            .then(r => {
                if (!r.ok) throw new Error("HTTP error " + r.status);
                return r.text();
            })
            .then(html => {
                if (!document.body.classList.contains('voice-live')) container.innerHTML = html;
            })
            .catch(err => console.error("Refresh failed:", err))
            .finally(() => { container.style.opacity = '1'; });
    };

    {% if is_online %}
    document.addEventListener("DOMContentLoaded", () => {
        let timeLeft = {{ config.server.cache_time|int }};
        const totalTime = {{ config.server.cache_time|int }};

        const progressBar = document.getElementById('refresh-bar');
        const timerText = document.getElementById('timer-text');

        function resetProgressBar() {
            progressBar.style.transition = 'none';
            progressBar.style.width = '100%';
            void progressBar.offsetWidth;
            progressBar.style.transition = `width ${totalTime}s linear`;
            progressBar.style.width = '0%';
        }

        resetProgressBar();

        setInterval(() => {
            if (document.body.classList.contains('voice-live')) {
                timerText.innerText = 'live';
                timeLeft = totalTime;
                return;
            }
            timeLeft--;
            timerText.innerText = timeLeft;

            if (timeLeft <= 0) {
                timeLeft = totalTime;
                window.refreshDashboard().finally(() => {
                    timerText.innerText = timeLeft;
                    resetProgressBar();
                });
            }
        }, 1000);
    });
    {% endif %}
</script>
{% if voice_available %}<script src="/static/voice.js?v=rnnoise-2" defer></script>{% endif %}
</body>
</html>
"""


def render_fragment(context):
    return render_template_string(FRAGMENT_TEMPLATE, **context)


@app.route("/")
def index():
    context = build_context()
    html_fragment = render_fragment(context)

    if request.args.get("ajax") == "1":
        return Response(html_fragment, mimetype="text/html; charset=utf-8")

    return render_template_string(PAGE_TEMPLATE, html_fragment=html_fragment, **context)


@app.route("/<path:filename>")
def static_asset(filename):
    path = (BASE_DIR / filename).resolve()
    if (
        path.is_file()
        and BASE_DIR in path.parents
        and path.relative_to(BASE_DIR).parts[0] not in PRIVATE_DIRS
        and path.suffix.lower() in ALLOWED_STATIC_EXTENSIONS
    ):
        return send_from_directory(BASE_DIR, filename)
    return Response("Not found", status=404)


if __name__ == "__main__":
    # run.py sets DASHBOARD_HOST/DASHBOARD_LISTEN_PORT (internal port when the
    # voice gateway fronts the app). Run directly, the dashboard serves alone.
    app.run(
        host=os.environ.get("DASHBOARD_HOST", CONFIG["web"]["host"]),
        port=int(os.environ.get("DASHBOARD_LISTEN_PORT", CONFIG["web"]["port"])),
    )