pySonnerie/frontend/app/routes.py

from __future__ import annotations

import re
from pathlib import Path

from flask import (
    Blueprint,
    Response,
    current_app,
    flash,
    jsonify,
    redirect,
    render_template,
    request,
    send_file,
    session,
    url_for,
)
from werkzeug.utils import secure_filename

from .backend_client import BackendApiError, BackendClient, read_default_backend_url


ui = Blueprint("ui", __name__, template_folder="templates")

GPIO_PATTERN = re.compile(r"^GPIO\d+$")
ALLOWED_AUDIO_EXTENSIONS = {".mp3", ".wav", ".ogg", ".flac", ".aac", ".m4a"}


def _backend_client() -> BackendClient | None:
    backend_url = session.get("backend_url")
    username = session.get("backend_username")
    password = session.get("backend_password")
    if not backend_url or not username or not password:
        return None
    return BackendClient(str(backend_url), str(username), str(password))


def _ensure_login() -> BackendClient | Response:
    client = _backend_client()
    if client is None:
        flash("Veuillez vous connecter d'abord.", "error")
        return redirect(url_for("ui.login"))
    return client


def _music_dir() -> Path:
    music_dir = current_app.config["MUSIC_DIR"]
    music_dir.mkdir(parents=True, exist_ok=True)
    return music_dir


def _list_audio_files() -> list[str]:
    files: list[str] = []
    for item in _music_dir().iterdir():
        if item.is_file() and item.suffix.lower() in ALLOWED_AUDIO_EXTENSIONS:
            files.append(item.name)
    return sorted(files, key=str.lower)


def _parse_optional_float(value: str) -> float | None:
    clean = value.strip()
    if clean == "":
        return None
    return float(clean)


def _next_numeric_trigger_id(triggers: dict[str, dict]) -> str:
    max_id = 0
    for key in triggers:
        if key.isdigit():
            max_id = max(max_id, int(key))
    return str(max_id + 1)


def _normalize_key(value: str) -> str:
    return value.strip().lower()


def _audio_redirect_target() -> str:
    if request.form.get("return_to", "").strip() == "audio_storage":
        return url_for("ui.audio_storage")
    return url_for("ui.dashboard")


def _trigger_sort_key(item: tuple[str, dict]) -> tuple[int, int | str]:
    trigger_id = item[0]
    if trigger_id.isdigit():
        return (0, int(trigger_id))
    return (1, trigger_id.lower())


@ui.get("/")
def home() -> Response:
    if _backend_client() is None:
        return redirect(url_for("ui.login"))
    return redirect(url_for("ui.dashboard"))


@ui.route("/login", methods=["GET", "POST"])
def login() -> str | Response:
    project_root: Path = current_app.config["PROJECT_ROOT"]
    default_url = read_default_backend_url(project_root)

    if request.method == "POST":
        backend_url = default_url
        username = request.form.get("username", "").strip()
        password = request.form.get("password", "")

        session["backend_url"] = backend_url
        session["backend_username"] = username
        session["backend_password"] = password

        client = _backend_client()
        assert client is not None
        try:
            client.health()
            client.list_triggers()
        except BackendApiError:
            return render_template(
                "login.html",
                login_error="Impossible de se connecter au service pour le moment. Vérifiez que le backend est demarré puis réessayez.",
                attempted_username=username,
            )

        flash("Connexion au backend reussie.", "success")
        return redirect(url_for("ui.dashboard"))

    return render_template("login.html", login_error=None, attempted_username="")


@ui.get("/logout")
def logout() -> Response:
    session.clear()
    flash("Session fermee.", "info")
    return redirect(url_for("ui.login"))


@ui.get("/dashboard")
def dashboard() -> str | Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        return client_or_redirect

    client = client_or_redirect
    triggers: dict[str, dict] = {}
    try:
        raw = client.list_triggers()
        valid_triggers = {k: v for k, v in raw.items() if isinstance(v, dict)}
        triggers = dict(sorted(valid_triggers.items(), key=_trigger_sort_key))
    except BackendApiError as exc:
        flash(f"Impossible de charger les triggers: {exc}", "error")

    audio_files = _list_audio_files()

    return render_template(
        "dashboard.html",
        triggers=triggers,
        audio_files=audio_files,
        backend_url=session.get("backend_url", ""),
        username=session.get("backend_username", ""),
    )


@ui.get("/audio-storage")
def audio_storage() -> str | Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        return client_or_redirect

    audio_files = _list_audio_files()
    return render_template("audio_storage.html", audio_files=audio_files)


@ui.post("/trigger/save")
def save_trigger() -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": "Non connecte"}), 401
        return client_or_redirect
    client = client_or_redirect

    original_id = request.form.get("original_id", "").strip()
    trigger_type = request.form.get("type", "").strip()
    name = request.form.get("name", "").strip()
    music_file = request.form.get("music_file", "").strip()
    start_raw = request.form.get("start_seconds", "0")
    end_raw = request.form.get("end_seconds", "")
    volume_raw = request.form.get("volume", "80")
    fade_in_raw = request.form.get("fade_in_seconds", "0")
    fade_out_raw = request.form.get("fade_out_seconds", "0")
    normalize_audio = request.form.get("normalize_audio") in {"on", "true", "1"}

    def _err(msg: str) -> Response:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": msg}), 400
        flash(msg, "error")
        return redirect(url_for("ui.dashboard"))

    if not GPIO_PATTERN.match(trigger_type):
        return _err("Le type doit respecter le format GPIO<number>.")

    if not name or not music_file:
        return _err("Le nom et le fichier audio sont obligatoires.")

    try:
        start_seconds = float(start_raw)
        end_seconds = _parse_optional_float(end_raw)
        volume = int(volume_raw)
        fade_in_seconds = float(fade_in_raw)
        fade_out_seconds = float(fade_out_raw)
    except ValueError:
        return _err("Les temps de debut/fin, fondus et le volume doivent être numériques.")

    if start_seconds < 0 or (end_seconds is not None and end_seconds <= start_seconds):
        return _err("Fenêtre temporelle invalide.")

    if not (0 <= volume <= 100):
        return _err("Le volume doit être compris entre 0 et 100.")

    if fade_in_seconds < 0 or fade_out_seconds < 0:
        return _err("Les durées de fondu doivent être supérieures ou égales à 0.")

    payload = {
        "name": name,
        "type": trigger_type,
        "music_file": music_file,
        "start_seconds": start_seconds,
        "end_seconds": end_seconds,
        "volume": volume,
        "fade_in_seconds": fade_in_seconds,
        "fade_out_seconds": fade_out_seconds,
        "normalize_audio": normalize_audio,
    }

    try:
        existing_raw = client.list_triggers()
        existing_triggers = {k: v for k, v in existing_raw.items() if isinstance(v, dict)}

        if original_id and original_id in existing_triggers:
            trigger_id = original_id
        else:
            trigger_id = _next_numeric_trigger_id(existing_triggers)

        for existing_id, existing_trigger in existing_triggers.items():
            if existing_id == trigger_id:
                continue
            if existing_trigger.get("type") == trigger_type:
                return _err("Ce type de trigger est déjà utilisé.")
            if _normalize_key(str(existing_trigger.get("name", ""))) == _normalize_key(name):
                return _err("Ce nom de trigger est déjà utilisé.")

        client.upsert_trigger(trigger_id, payload)
    except BackendApiError as exc:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": str(exc)}), 502
        flash(f"Echec d'enregistrement du trigger: {exc}", "error")
        return redirect(url_for("ui.dashboard"))

    if request.headers.get("X-Requested-With") == "fetch":
        return jsonify({
            "ok": True,
            "message": f"Trigger \"{name}\" enregistré.",
            "trigger_id": trigger_id,
            "original_id": original_id or trigger_id,
            "trigger": payload,
        })
    flash(f"Trigger \"{name}\" enregistré.", "success")
    return redirect(url_for("ui.dashboard"))


@ui.post("/trigger/delete")
def delete_trigger() -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": "Non connecte"}), 401
        return client_or_redirect
    client = client_or_redirect

    trigger_id = request.form.get("trigger_id", "").strip()
    if not trigger_id:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": "Identifiant du trigger manquant."}), 400
        flash("Identifiant du trigger manquant.", "error")
        return redirect(url_for("ui.dashboard"))

    try:
        client.delete_trigger(trigger_id)
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": True, "message": "Trigger supprimé.", "trigger_id": trigger_id})
        flash("Trigger supprimé.", "success")
    except BackendApiError as exc:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": f"Echec de suppression: {exc}"}), 500
        flash(f"Echec de suppression: {exc}", "error")
    return redirect(url_for("ui.dashboard"))


@ui.post("/trigger/play")
def play_trigger() -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": "Non connecte"}), 401
        return client_or_redirect
    client = client_or_redirect

    trigger_id = request.form.get("trigger_id", "").strip()
    if not trigger_id:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": "Identifiant du trigger manquant."}), 400
        flash("Identifiant du trigger manquant.", "error")
        return redirect(url_for("ui.dashboard"))

    try:
        client.play_trigger(trigger_id)
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": True, "message": "Trigger démarré."})
        flash("Trigger démarré.", "success")
    except BackendApiError as exc:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": str(exc)}), 502
        flash(f"Echec du lancement: {exc}", "error")
    return redirect(url_for("ui.dashboard"))


@ui.post("/audio/stop")
def stop_audio() -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": "Non connecte"}), 401
        return client_or_redirect
    client = client_or_redirect

    try:
        client.stop_audio()
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": True, "message": "Audio arrete."})
        flash("Audio arrete.", "info")
    except BackendApiError as exc:
        if request.headers.get("X-Requested-With") == "fetch":
            return jsonify({"ok": False, "error": str(exc)}), 502
        flash(f"Echec de l'arrêt audio: {exc}", "error")
    return redirect(url_for("ui.dashboard"))


@ui.post("/audio/upload")
def upload_audio() -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        return client_or_redirect

    audio = request.files.get("audio_file")
    if audio is None or audio.filename is None or audio.filename.strip() == "":
        flash("Sélectionnez d'abord un fichier.", "error")
        return redirect(_audio_redirect_target())

    filename = secure_filename(audio.filename)
    if filename == "":
        flash("Nom de fichier invalide.", "error")
        return redirect(_audio_redirect_target())

    ext = Path(filename).suffix.lower()
    if ext not in ALLOWED_AUDIO_EXTENSIONS:
        flash("Format audio non supporté.", "error")
        return redirect(_audio_redirect_target())

    existing_names = {item.name.lower() for item in _music_dir().iterdir() if item.is_file()}
    if filename.lower() in existing_names:
        flash("Un fichier avec ce nom existe déjà.", "error")
        return redirect(_audio_redirect_target())

    destination = _music_dir() / filename
    audio.save(destination)
    flash(f"Fichier {filename} téléversé.", "success")
    return redirect(_audio_redirect_target())


@ui.post("/audio/delete")
def delete_audio() -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        return client_or_redirect

    filename = request.form.get("filename", "").strip()
    if not filename:
        flash("Nom de fichier manquant.", "error")
        return redirect(_audio_redirect_target())

    target = _music_dir() / filename
    if not target.exists() or not target.is_file():
        flash("Fichier introuvable.", "error")
        return redirect(_audio_redirect_target())

    target.unlink()
    flash(f"Fichier {filename} supprimé.", "success")
    return redirect(_audio_redirect_target())


@ui.get("/audio/download/<path:filename>")
def download_audio(filename: str) -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        return client_or_redirect

    safe_name = Path(filename).name
    target = _music_dir() / safe_name
    if not target.exists() or not target.is_file():
        flash("Fichier introuvable.", "error")
        return redirect(url_for("ui.dashboard"))
    return send_file(target, as_attachment=True)


@ui.get("/audio/stream/<path:filename>")
def stream_audio(filename: str) -> Response:
    client_or_redirect = _ensure_login()
    if isinstance(client_or_redirect, Response):
        return client_or_redirect

    safe_name = Path(filename).name
    target = _music_dir() / safe_name
    if not target.exists() or not target.is_file():
        flash("Fichier introuvable.", "error")
        return redirect(url_for("ui.dashboard"))
    return send_file(target, as_attachment=False)