import subprocess
import threading
import uuid
import json
import os
from flask import Flask, request, jsonify, Response, stream_with_context
from flask_cors import CORS
import time

app = Flask(__name__, static_folder=".", static_url_path="")
CORS(app)

# Store active jobs: {job_id: {"lines": [], "done": bool, "error": bool}}
jobs = {}
jobs_lock = threading.Lock()

YTDLP_PATH = os.environ.get("YTDLP_PATH", "yt-dlp")
DOWNLOAD_DIR = os.environ.get("DOWNLOAD_DIR", os.getcwd())


def run_ytdlp(job_id, cmd):
    with jobs_lock:
        jobs[job_id] = {"lines": [], "done": False, "error": False}

    try:
        process = subprocess.Popen(
            cmd,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            text=True,
            bufsize=1,
            cwd=DOWNLOAD_DIR,
        )
        for line in process.stdout:
            line = line.rstrip()
            with jobs_lock:
                jobs[job_id]["lines"].append(line)
        process.wait()
        with jobs_lock:
            jobs[job_id]["done"] = True
            if process.returncode != 0:
                jobs[job_id]["error"] = True
    except Exception as e:
        with jobs_lock:
            jobs[job_id]["lines"].append(f"[ERROR] {str(e)}")
            jobs[job_id]["done"] = True
            jobs[job_id]["error"] = True


@app.route("/")
def index():
    return app.send_static_file("frontend.html")


@app.route("/api/info", methods=["POST"])
def get_info():
    data = request.json
    url = data.get("url", "").strip()
    cookies = data.get("cookies", "").strip()

    if not url:
        return jsonify({"error": "No URL provided"}), 400

    cmd = [YTDLP_PATH, "--dump-json", "--no-playlist"]
    if cookies:
        cmd += ["--cookies", cookies]
    cmd += ["--extractor-args", "generic:impersonate", url]

    try:
        result = subprocess.run(
            cmd, capture_output=True, text=True, timeout=30, cwd=DOWNLOAD_DIR
        )
        if result.returncode != 0:
            return jsonify({"error": result.stderr or "Failed to fetch info"}), 400
        info = json.loads(result.stdout)
        formats = []
        for f in info.get("formats", []):
            formats.append(
                {
                    "id": f.get("format_id"),
                    "ext": f.get("ext"),
                    "resolution": f.get("resolution") or f.get("format_note") or "",
                    "vcodec": f.get("vcodec", "none"),
                    "acodec": f.get("acodec", "none"),
                    "filesize": f.get("filesize") or f.get("filesize_approx"),
                    "tbr": f.get("tbr"),
                    "label": f.get("format"),
                }
            )
        return jsonify(
            {
                "title": info.get("title", "Unknown"),
                "thumbnail": info.get("thumbnail"),
                "duration": info.get("duration"),
                "uploader": info.get("uploader"),
                "formats": formats,
            }
        )
    except subprocess.TimeoutExpired:
        return jsonify({"error": "Timed out fetching video info"}), 408
    except Exception as e:
        return jsonify({"error": str(e)}), 500


@app.route("/api/download", methods=["POST"])
def start_download():
    data = request.json
    url = data.get("url", "").strip()
    format_id = data.get("format_id", "").strip()
    cookies = data.get("cookies", "").strip()
    extra_args = data.get("extra_args", "").strip()

    if not url:
        return jsonify({"error": "No URL provided"}), 400

    cmd = [YTDLP_PATH]
    if format_id:
        cmd += ["-f", format_id]
    cmd += ["--prefer-free-formats"]
    if cookies:
        cmd += ["--cookies", cookies]
    cmd += ["--extractor-args", "generic:impersonate"]
    if extra_args:
        cmd += extra_args.split()
    cmd.append(url)

    job_id = str(uuid.uuid4())
    thread = threading.Thread(target=run_ytdlp, args=(job_id, cmd), daemon=True)
    thread.start()

    return jsonify({"job_id": job_id})


@app.route("/api/status/<job_id>")
def job_status(job_id):
    def generate():
        sent = 0
        while True:
            with jobs_lock:
                job = jobs.get(job_id)
            if not job:
                yield f"data: {json.dumps({'error': 'Job not found'})}\n\n"
                break
            lines = job["lines"]
            while sent < len(lines):
                yield f"data: {json.dumps({'line': lines[sent]})}\n\n"
                sent += 1
            if job["done"]:
                yield f"data: {json.dumps({'done': True, 'error': job['error']})}\n\n"
                break
            time.sleep(0.2)

    return Response(
        stream_with_context(generate()),
        mimetype="text/event-stream",
        headers={"Cache-Control": "no-cache", "X-Accel-Buffering": "no"},
    )


if __name__ == "__main__":
    print(f"[yt-dlp UI] Serving on http://localhost:5000")
    print(f"[yt-dlp UI] Download directory: {DOWNLOAD_DIR}")
    app.run(debug=False, host="0.0.0.0", port=5000, threaded=True)