#!/usr/bin/env python3
"""
Recounting dashboard — consumes live counter + recount APIs,
lists OUTPUT_DIR MP4 files, streams via go2rtc for preview.
"""
import json
import os
import subprocess
import requests
from datetime import datetime
from pathlib import Path

from flask import Flask, render_template, jsonify, request, Response
from werkzeug.serving import WSGIRequestHandler

from dotenv import load_dotenv
load_dotenv()

app = Flask(__name__, template_folder="templates")
app.config["SECRET_KEY"] = os.getenv("SECRET_KEY", "change-me-in-production")

OUTPUT_DIR = os.getenv("OUTPUT_DIR", os.getenv("OUTPUT_DIR", "/opt/bytetrack-counter"))
LIVE_API_URL = os.getenv("LIVE_API_URL", "http://localhost:5000")
RECOUNT_API_URL = os.getenv("RECOUNT_API_URL", "http://localhost:5001")
GO2RTC_API_URL = os.getenv("GO2RTC_API_URL", "http://localhost:1984")
GO2RTC_STREAM_NAME = os.getenv("GO2RTC_STREAM_NAME", "recount")

SITE_NAME = os.getenv("SITE_NAME", "RECOUNT")

DASHBOARD_PORT = int(os.getenv("RECOUNTING_DASHBOARD_PORT", "5002"))
DASHBOARD_HOST = os.getenv("DASHBOARD_HOST", "0.0.0.0")
FLASK_DEBUG = os.getenv("FLASK_DEBUG", "false").lower() == "true"

_http_session = requests.Session()
_http_session.timeout = 3


def _api_get(base_url, path, default=None):
    try:
        resp = _http_session.get(f"{base_url}{path}")
        if resp.status_code == 200:
            return resp.json()
    except Exception:
        pass
    return default


@app.route("/")
def index():
    return render_template("recounting.html", site_name=SITE_NAME, output_dir=OUTPUT_DIR)


@app.route("/api/live-progress")
def api_live_progress():
    data = _api_get(LIVE_API_URL, "/api/current-batch")
    if data and data.get("success"):
        return jsonify(data)
    return jsonify({"success": False, "count": 0, "batch_number": None, "error": "Live unreachable"}), 200


@app.route("/api/recount-progress")
def api_recount_progress():
    data = _api_get(RECOUNT_API_URL, "/api/current-batch")
    if data and data.get("success"):
        return jsonify(data)
    return jsonify({"success": False, "count": 0, "batch_number": None, "error": "Recount unreachable"}), 200


@app.route("/api/mp4-files")
def api_mp4_files():
    files = []
    output = Path(OUTPUT_DIR)
    if output.exists():
        for f in sorted(output.rglob("*.mp4"), key=lambda p: p.stat().st_mtime, reverse=True):
            st = f.stat()
            files.append({
                "name": f.name,
                "path": str(f),
                "size": st.st_size,
                "mtime": datetime.fromtimestamp(st.st_mtime).isoformat(),
            })
    return jsonify(files)


@app.route("/api/start-recount", methods=["POST"])
def start_recount():
    data = request.get_json(force=True) or {}
    mp4_path = data.get("path", "")
    if not mp4_path:
        return jsonify({"success": False, "error": "Missing 'path'"}), 400
    if not os.path.isfile(mp4_path):
        return jsonify({"success": False, "error": f"File not found: {mp4_path}"}), 404

    src = f"ffmpeg:{mp4_path}#video=h264#hardware"
    try:
        resp = requests.put(
            f"{GO2RTC_API_URL}/api/streams",
            params={"name": GO2RTC_STREAM_NAME, "src": src},
            timeout=5,
        )
        if resp.status_code not in (200, 201):
            return jsonify({"success": False, "error": f"go2rtc returned {resp.status_code}: {resp.text}"}), 502
    except Exception as e:
        return jsonify({"success": False, "error": f"go2rtc unreachable: {e}"}), 502

    stream_url = f"{GO2RTC_API_URL}/api/stream.mjpeg?src={GO2RTC_STREAM_NAME}"
    return jsonify({"success": True, "stream_url": stream_url, "file": Path(mp4_path).name})


@app.route("/api/stop-recount", methods=["POST"])
def stop_recount():
    try:
        requests.delete(f"{GO2RTC_API_URL}/api/streams", params={"name": GO2RTC_STREAM_NAME}, timeout=5)
    except Exception:
        pass
    return jsonify({"success": True})


if __name__ == "__main__":
    WSGIRequestHandler.protocol_version = "HTTP/1.1"
    print(f"Recounting dashboard at http://{DASHBOARD_HOST}:{DASHBOARD_PORT}")
    print(f"Live API: {LIVE_API_URL}")
    print(f"Recount API: {RECOUNT_API_URL}")
    print(f"go2rtc API: {GO2RTC_API_URL}")
    print(f"OUTPUT_DIR: {OUTPUT_DIR}")
    app.run(host=DASHBOARD_HOST, port=DASHBOARD_PORT, debug=FLASK_DEBUG)