#!/usr/bin/env python3
import subprocess
import json
from pathlib import Path

BASE_DIR = Path("/var/lib/ffplayout/tv-media")
SOURCES = {
    "communitymediavideos": "https://communitymedia.video/c/communitymediavideos/videos",
    "newellijay": "https://communitymedia.video/a/video@vod.newellijay.tv/video-channels"
}

for name, url in SOURCES.items():
    print(f"Fetching full list from {name}...")
    try:
        cmd = ["yt-dlp", "--flat-playlist", "--dump-json", url]
        output = subprocess.check_output(cmd, text=True, timeout=180)
        videos = [json.loads(line) for line in output.strip().split("\n") if line.strip()]
        output_file = BASE_DIR / f"peertube_{name}.json"
        output_file.write_text(json.dumps(videos, indent=2))
        print(f"  Saved {len(videos)} videos to {output_file.name}")
    except Exception as e:
        print(f"Error: {e}")

print("\nDone. Two JSON files created.")