#!/usr/bin/env python3
"""HEOS panel: a phone-sized web remote plus the HTTP bridge it runs on.

    pip3 install -r requirements.txt
    python3 app.py                 # dev server, http://<pi-ip>:5443/

The service instead runs this under gunicorn -- see deploy/heos-panel.service
-- which imports `app` without ever calling main(), so the controller below
is built at import time rather than from main()'s argparse.

Everything the UI does goes through /api/*. The flatter, query-string
endpoints from the original heos_bridge.py (/volume/up?target=..., and
friends) are still here so existing Shortcuts and scripts keep working.
"""

import argparse
import os
from functools import wraps

from flask import Flask, jsonify, render_template, request, url_for
from werkzeug.middleware.proxy_fix import ProxyFix

import config
from controller import Controller, TargetError
from heos import HeosError
from spotify import SpotifyClient, SpotifyError
from zidoo import ZidooError

app = Flask(__name__)

# Served straight from port 5443 this changes nothing. Behind a reverse
# proxy that mounts us on a sub-path (Apache at /heos, say) it reads the
# X-Forwarded-Prefix that proxy sets, so every URL the app generates is
# /heos/... instead of /..., and the page works either way.
app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1, x_proto=1, x_host=1, x_prefix=1)

controller: Controller = None
spotify: dict = {}     # account key -> SpotifyClient, configured accounts only

# HEOS merges any In-Room Group into one player at the hardware level (see
# controller.py's module docstring), so its API has no field that says how
# many units heos_name actually names -- that has to come from config.json's
# "in_room_group" instead (config.IN_ROOM_GROUPS). The AVR/speaker split
# needs no such flag: this app's own design always makes HOST_KEY the AVR
# and ROOM_KEYS plain speakers, so it is known before any HEOS call is made.
#
# "kind" is passed straight through to the template as-is (one of
# config.IN_ROOM_GROUPS, or "avr") -- which icon that draws is the
# interface's business, not ours: see ICONS in static/app.js.
def _room_meta(key: str, kind: str) -> dict:
    return {"key": key, "kind": kind, **config.TARGETS[key]}


def _build_spotify() -> dict:
    if not (config.SPOTIFY_CLIENT_ID and config.SPOTIFY_CLIENT_SECRET):
        return {}
    return {
        key: SpotifyClient(config.SPOTIFY_CLIENT_ID, config.SPOTIFY_CLIENT_SECRET, account["refresh_token"])
        for key, account in config.SPOTIFY_ACCOUNTS.items()
        if account.get("refresh_token")
    }


if __name__ != "__main__":
    # Imported by a WSGI server rather than run as a script, so main()'s
    # argparse never executes -- build the one controller instance here
    # instead. HEOS_DEMO lets the fake-speakers mode work this way too.
    if os.environ.get("HEOS_DEMO", "").lower() in ("1", "true", "yes"):
        from demo import DemoController
        controller = DemoController(config)
    else:
        controller = Controller(config)
    spotify = _build_spotify()


def handle_errors(view):
    """One place to turn our three failure modes into sensible JSON."""
    @wraps(view)
    def wrapped(*args, **kwargs):
        try:
            return view(*args, **kwargs)
        except (TargetError, ValueError) as exc:
            return jsonify({"error": str(exc)}), 400
        except (HeosError, SpotifyError, ZidooError) as exc:
            return jsonify({"error": str(exc)}), 502
    return wrapped


@app.after_request
def no_store(response):
    """It is a LAN remote: never let a phone show a cached volume, and
    never let iOS pin an old copy of the UI to the home screen."""
    response.headers["Cache-Control"] = "no-store"
    return response


def _payload() -> dict:
    return request.get_json(silent=True) or request.form.to_dict() or request.args.to_dict()


def _target_from(data: dict, field: str = "target") -> str:
    key = data.get(field)
    if not key:
        raise ValueError(f"Missing '{field}'. Valid rooms: {', '.join(config.TARGETS)}")
    if key not in config.TARGETS:
        raise ValueError(f"Unknown room '{key}'. Valid rooms: {', '.join(config.TARGETS)}")
    return key


def _spotify_from(data: dict) -> SpotifyClient:
    if not spotify:
        raise ValueError("Spotify isn't configured -- see the README's Spotify section")
    key = data.get("account")
    if not key:
        raise ValueError(f"Missing 'account'. Spotify accounts: {', '.join(spotify)}")
    if key not in spotify:
        raise ValueError(f"Unknown Spotify account '{key}'. Spotify accounts: {', '.join(spotify)}")
    return spotify[key]


def _spotify_name(key: str) -> str:
    target = config.TARGETS[key]
    return target.get("spotify_name", target["heos_name"])


def _spotify_playing_on() -> dict:
    """Device name -> the key of the account playing on it. An account
    Spotify refuses (a revoked token, say) just plays on nothing, instead of
    failing whatever asked."""
    playing_on = {}
    for account, client in spotify.items():
        try:
            player = client.playback()
        except (SpotifyError, OSError):
            continue
        device = (player.get("device") or {}).get("name")
        # Should two accounts both claim a device, the one actually playing wins.
        if device and (device not in playing_on or player.get("is_playing")):
            playing_on[device] = account
    return playing_on


def _mark_spotify_accounts(rooms: list):
    """Tag each room HEOS says is on Spotify with the account playing it, so
    its card can pick out that account's button. Spotify is only asked when
    some room is on Spotify at all."""
    on_spotify = [room for room in rooms if room.get("spotify")]
    if not (spotify and on_spotify):
        return
    playing_on = _spotify_playing_on()
    for room in on_spotify:
        room["spotify_account"] = playing_on.get(_spotify_name(room["key"]))


def _link_zidoo_poster(avr: dict):
    """Point the AVR's now-playing cover at our own copy of the Zidoo's
    poster: the panel is served over https, and the Zidoo only speaks plain
    http, which a phone will not load into an https page."""
    track = avr.get("now_playing")
    if not track:
        return
    poster_id = track.pop("poster_id", None)
    if poster_id is not None:
        track["image"] = url_for("api_zidoo_poster", poster_id=poster_id)


# --- The UI -----------------------------------------------------------
@app.route("/")
def index():
    return render_template(
        "index.html",
        app_name=config.APP_NAME,
        host=_room_meta(config.HOST_KEY, "avr"),
        rooms=[
            _room_meta(key, config.TARGETS[key].get("in_room_group", "none"))
            for key in config.ROOM_KEYS
        ],
        step=config.VOLUME_STEP,
        spotify_accounts=[{"key": key, "label": config.SPOTIFY_ACCOUNTS[key]["label"]} for key in spotify],
    )


@app.get("/manifest.webmanifest")
def manifest():
    """Rendered rather than static, so APP_NAME is only written down once."""
    return app.response_class(
        render_template("manifest.webmanifest", app_name=config.APP_NAME),
        mimetype="application/manifest+json",
    )


# --- API the UI talks to ----------------------------------------------
@app.get("/api/state")
@handle_errors
def api_state():
    data = controller.state()
    # Demo rooms are not on anyone's real Spotify, so they bring their own account.
    if not data.get("demo"):
        _mark_spotify_accounts(data["rooms"])
    _link_zidoo_poster(data["avr"])
    return jsonify(data)


@app.get("/api/zidoo/poster/<int:poster_id>")
def api_zidoo_poster(poster_id):
    """A film's poster, fetched from the Zidoo on the phone's behalf -- see
    _link_zidoo_poster()."""
    zidoo = getattr(controller, "zidoo", None)
    image = zidoo.poster(poster_id) if zidoo else None
    if image is None:
        return "", 404
    body, mimetype = image
    return app.response_class(body, mimetype=mimetype)


@app.get("/api/targets")
@handle_errors
def api_targets():
    """Diagnostic: every player and group HEOS can see, with its exact
    name -- this is what you copy into config.py."""
    found = controller.scan()
    return jsonify({
        "players": [
            {"kind": "player", "name": p.get("name"), "pid": p.get("pid"), "model": p.get("model", "")}
            for p in found["players"]
        ],
        "groups": [
            {"kind": "group", "name": g.get("name"), "gid": g.get("gid"),
             "players": [{"name": m.get("name"), "pid": m.get("pid"), "role": m.get("role")}
                         for m in g.get("players", [])]}
            for g in found["groups"]
        ],
    })


@app.post("/api/volume")
@handle_errors
def api_volume():
    data = _payload()
    key = _target_from(data)
    if data.get("level") is not None:
        return jsonify({"target": key, "level": controller.set_volume(key, int(data["level"]))})
    if data.get("steps") is not None:
        # Taps, not points: each one lands on the next multiple of VOLUME_STEP.
        return jsonify({"target": key, "level": controller.step_volume(key, int(data["steps"]))})
    if data.get("delta") is not None:
        return jsonify({"target": key, "level": controller.nudge_volume(key, int(data["delta"]))})
    raise ValueError("Provide one of 'steps', 'delta' or 'level'")


@app.post("/api/mute")
@handle_errors
def api_mute():
    key = _target_from(_payload())
    controller.toggle_mute(key)
    return jsonify({"target": key, "ok": True})


@app.post("/api/playback")
@handle_errors
def api_playback():
    """Start or stop a room. Send 'state' to be explicit, or leave it out to
    flip whatever the speakers are actually doing."""
    data = _payload()
    key = _target_from(data)
    state = data.get("state")
    if state is not None and state not in ("play", "pause", "stop"):
        raise ValueError("'state' must be play, pause or stop -- or left out to toggle")
    return jsonify({"target": key, "state": controller.toggle_play(key, state)})


@app.post("/api/skip")
@handle_errors
def api_skip():
    """Jump to the next (or previous) track in a room's queue."""
    data = _payload()
    key = _target_from(data)
    direction = data.get("direction", "next")
    if direction not in ("next", "previous"):
        raise ValueError("'direction' must be next or previous")
    controller.skip(key, direction)
    return jsonify({"target": key, "direction": direction})


@app.post("/api/seek")
@handle_errors
def api_seek():
    """Jump to a point in what a card is playing. HEOS itself cannot seek,
    so this goes around it: to the Zidoo for the AVR's card, and for a room,
    to Spotify, through whichever of your accounts is playing there."""
    data = _payload()
    key = _target_from(data)
    if data.get("position_ms") is None:
        raise ValueError("Provide 'position_ms', where to jump to")
    position = max(0, int(data["position_ms"]))
    if key == config.HOST_KEY:
        controller.zidoo_seek(position)
    else:
        account = _spotify_playing_on().get(_spotify_name(key))
        if account is None:
            raise ValueError("Only a Spotify stream one of your accounts is playing can seek -- HEOS itself cannot")
        spotify[account].seek(position)
    return jsonify({"target": key, "position_ms": position})


@app.post("/api/group")
@handle_errors
def api_group():
    """Join or leave the AVR's group. The AVR is always the host, so its
    content is what the joined rooms start playing."""
    data = _payload()
    key = _target_from(data)
    joined = data.get("joined")
    if isinstance(joined, str):
        joined = joined.lower() in ("1", "true", "yes", "on")
    if joined is None:
        raise ValueError("Provide 'joined': true to merge with the AVR, false to split off")
    return jsonify({"joined": controller.join(key) if joined else controller.leave(key)})


@app.post("/api/group/none")
@handle_errors
def api_group_none():
    """Every room back on its own."""
    return jsonify({"joined": controller.set_membership([])})


@app.get("/api/avr/inputs")
@handle_errors
def api_avr_inputs():
    return jsonify(controller.avr_inputs())


@app.post("/api/avr/input")
@handle_errors
def api_avr_set_input():
    code = _payload().get("code")
    if not code:
        raise ValueError("Provide 'code', e.g. inputs/aux_in_1 -- see GET /api/avr/inputs")
    return jsonify(controller.avr_select_input(code))


@app.get("/api/spotify/devices")
@handle_errors
def api_spotify_devices():
    """Diagnostic: every Spotify Connect receiver one account currently sees
    (?account=account1) -- use this to fill in a target's spotify_name if it
    differs from heos_name."""
    return jsonify(_spotify_from(request.args).devices())


@app.post("/api/spotify/resume")
@handle_errors
def api_spotify_resume():
    """Ask a room's own Spotify Connect receiver to resume one account's
    playback -- the reverse of connecting to it from the Spotify app."""
    data = _payload()
    client = _spotify_from(data)
    key = _target_from(data)
    device = client.resume(_spotify_name(key))
    return jsonify({"target": key, "account": data["account"], "device": device.get("name")})


# --- The original bridge's API, unchanged -----------------------------
@app.get("/targets")
@handle_errors
def legacy_targets():
    found = controller.scan()
    return jsonify(
        [{"kind": "player", "name": p.get("name"), "id": p.get("pid"), "model": p.get("model", "")}
         for p in found["players"]]
        + [{"kind": "group", "name": g.get("name"), "id": g.get("gid"),
            "members": [m.get("name") for m in g.get("players", [])]}
           for g in found["groups"]]
    )


@app.get("/volume")
@handle_errors
def legacy_get_volume():
    return jsonify({"level": controller.volume(_target_from(request.args))})


@app.post("/volume/set")
@handle_errors
def legacy_set_volume():
    level = request.args.get("level", type=int)
    if level is None or not 0 <= level <= 100:
        raise ValueError("provide integer 'level' between 0 and 100")
    return jsonify({"level": controller.set_volume(_target_from(request.args), level)})


@app.post("/volume/up")
@handle_errors
def legacy_volume_up():
    key = _target_from(request.args)
    step = request.args.get("step", type=int)
    # No ?step= means one tap of the panel's own button, snapping to the
    # next multiple; an explicit ?step= stays a raw number of points.
    level = controller.step_volume(key, 1) if step is None else controller.nudge_volume(key, step)
    return jsonify({"level": level})


@app.post("/volume/down")
@handle_errors
def legacy_volume_down():
    key = _target_from(request.args)
    step = request.args.get("step", type=int)
    level = controller.step_volume(key, -1) if step is None else controller.nudge_volume(key, -step)
    return jsonify({"level": level})


@app.post("/volume/mute")
@handle_errors
def legacy_mute():
    controller.toggle_mute(_target_from(request.args))
    return jsonify({"ok": True})


@app.post("/playback/<state>")
@handle_errors
def legacy_playback(state):
    key = _target_from(request.args)
    if state in ("play", "pause", "stop"):
        controller.play_state(key, state)
        return jsonify({"state": state})
    if state in ("next", "previous"):
        controller.skip(key, state)
        return jsonify({"ok": True})
    raise ValueError(f"Unknown playback action '{state}'")


@app.post("/group/create")
@handle_errors
def legacy_group_create():
    host = _target_from(request.args, "host")
    members = [m.strip() for m in request.args.get("members", "").split(",") if m.strip()]
    if not members:
        raise ValueError("provide '?host=<room>&members=<comma-separated rooms>'")
    for key in members:
        _target_from({"target": key})
    controller.group_targets(host, members)
    return jsonify({"host": host, "members": members})


@app.post("/group/remove")
@handle_errors
def legacy_group_remove():
    controller.ungroup(_target_from(request.args))
    return jsonify({"ok": True})


@app.get("/inputs")
@handle_errors
def legacy_inputs():
    return jsonify(controller.heos_inputs(_target_from(request.args)))


@app.post("/input/set")
@handle_errors
def legacy_input_set():
    input_id = request.args.get("input")
    if not input_id:
        raise ValueError("provide '?input=' -- see GET /inputs for valid values")
    controller.play_heos_input(_target_from(request.args), input_id)
    return jsonify({"input": input_id})


@app.post("/input/relay")
@handle_errors
def legacy_input_relay():
    input_id = request.args.get("input")
    source = request.args.get("from")
    if not input_id or not source:
        raise ValueError("provide '?from=<source room>&input=<input id>'")
    controller.play_heos_input(_target_from(request.args), input_id, _target_from(request.args, "from"))
    return jsonify({"input": input_id, "from": source})


@app.get("/raw/<path:subpath>")
@handle_errors
def legacy_raw(subpath):
    """Diagnostic: forward any heos:// command as-is.
    e.g. GET /raw/browse/browse?sid=1027"""
    return jsonify(controller.heos.command(subpath, **request.args.to_dict()))


@app.get("/avr/inputs")
@handle_errors
def legacy_avr_inputs():
    return jsonify(controller.avr_inputs())


@app.route("/avr/input", methods=["GET", "POST"])
@handle_errors
def legacy_avr_input():
    if request.method == "GET":
        return jsonify(controller.avr_current_input() or {"error": "AVR not reachable"})
    code = request.args.get("input")
    if not code:
        raise ValueError("provide '?input=<code>', e.g. inputs/aux_in_1 -- see GET /avr/inputs")
    return jsonify(controller.avr_select_input(code))


def main():
    global controller, spotify
    parser = argparse.ArgumentParser(description="HEOS panel")
    parser.add_argument("--port", type=int, default=config.WEB_PORT)
    parser.add_argument("--host", default="0.0.0.0",
                        help="0.0.0.0 so your phone can reach it directly; "
                             "127.0.0.1 to allow only a local reverse proxy")
    parser.add_argument("--demo", action="store_true",
                        help="run with fake speakers, for working on the UI away from the kit")
    args = parser.parse_args()

    if args.demo:
        from demo import DemoController
        controller = DemoController(config)
    else:
        controller = Controller(config)
    spotify = _build_spotify()

    app.run(host=args.host, port=args.port, threaded=True)


if __name__ == "__main__":
    main()