from __future__ import annotations

import contextlib
import io
import json
import sys
from pathlib import Path
from typing import Any, Dict

REPO_ROOT = Path(__file__).resolve().parents[1]
if str(REPO_ROOT) not in sys.path:
    sys.path.insert(0, str(REPO_ROOT))


def main(argv: list[str] | None = None) -> int:
    args = list(sys.argv[1:] if argv is None else argv)
    if not args:
        payload: Dict[str, Any] = {
            "success": False,
            "stdout": "",
            "stderr": "",
            "error": "Missing url",
            "table": None,
        }
        print(json.dumps(payload, ensure_ascii=False))
        return 2

    url = str(args[0] or "").strip()
    captured_stdout = io.StringIO()
    captured_stderr = io.StringIO()
    with contextlib.redirect_stdout(captured_stdout), contextlib.redirect_stderr(captured_stderr):
        from MPV.pipeline_helper import _run_op

        payload = _run_op("ytdlp-formats", {"url": url})

    noisy_stdout = captured_stdout.getvalue().strip()
    noisy_stderr = captured_stderr.getvalue().strip()
    if noisy_stdout:
        payload["stdout"] = "\n".join(filter(None, [str(payload.get("stdout") or "").strip(), noisy_stdout]))
    if noisy_stderr:
        payload["stderr"] = "\n".join(filter(None, [str(payload.get("stderr") or "").strip(), noisy_stderr]))

    print(json.dumps(payload, ensure_ascii=False))
    return 0 if payload.get("success") else 1


if __name__ == "__main__":
    raise SystemExit(main())