root app imports core

app.py

@@ -1,104 +1 @@
-import os, re, uuid, subprocess
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import FileResponse, StreamingResponse
-import yt_dlp
-
-app = FastAPI()
-os.makedirs("videos", exist_ok=True)
-
-def name_from_url(url: str, fallback: str) -> str:
-    # kick.com/streamer/videos/ID -> streamer_ID
-    # twitch.tv/videos/ID -> twitch_ID
-    # youtube.com/watch?v=ID -> youtube_ID
-    m = re.findall(r"/videos/([^/?]+)", url)
-    if m:
-        parts = url.split("/")
-        streamer = parts[-3] if len(parts) >= 3 and parts[-2] == "videos" else ("twitch" if "twitch" in url else "stream")
-        return f"{streamer}_{m[0][:12]}"
-    m = re.findall(r"[?&]v=([^&]+)", url)
-    if m:
-        return f"youtube_{m[0]}"
-    return fallback
-
-@app.get("/formats")
-def formats(url: str):
-    try:
-        with yt_dlp.YoutubeDL({"quiet": True, "no_warnings": True}) as y:
-            info = y.extract_info(url, download=False)
-    except Exception as e:
-        try:
-            r = subprocess.run(
-                ["streamlink", "--json", url],
-                capture_output=True, text=True, timeout=60
-            )
-            import json
-            data = json.loads(r.stdout)
-            return [{"id": k, "quality": k, "ext": "mp4", "size_mb": 0, "vcodec": "h264"}
-                    for k in data.get("streams", {}).keys()]
-        except Exception:
-            raise HTTPException(400, str(e))
-    out = []
-    for f in info.get("formats", []):
-        if f.get("vcodec") != "none" and f.get("height"):
-            out.append({
-                "id": f["format_id"],
-                "quality": f"{f['height']}p",
-                "ext": f.get("ext"),
-                "size_mb": round((f.get("filesize") or f.get("filesize_approx") or 0) / 1e6, 1),
-                "vcodec": f.get("vcodec"),
-            })
-    return out
-
-@app.post("/download")
-def download(url: str, format_id: str):
-    name = name_from_url(url, uuid.uuid4().hex[:8])
-    outtmpl = f"videos/{name}.%(ext)s"
-    try:
-        with yt_dlp.YoutubeDL({
-            "format": format_id,
-            "outtmpl": outtmpl,
-            "quiet": True,
-            "no_warnings": True,
-        }) as y:
-            info = y.extract_info(url, download=True)
-            ext = info.get("ext", "mp4")
-        return {"status": "ok", "file": f"{name}.{ext}"}
-    except Exception:
-        try:
-            out = f"videos/{name}.mp4"
-            subprocess.run(
-                ["streamlink", url, format_id, "-o", out],
-                capture_output=True, text=True, timeout=3600, check=True
-            )
-            return {"status": "ok", "file": f"{name}.mp4"}
-        except Exception as e2:
-            raise HTTPException(400, str(e2))
-
-@app.get("/stream")
-def stream(url: str, format_id: str):
-    name = name_from_url(url, uuid.uuid4().hex[:8])
-    sl = subprocess.Popen(
-        ["streamlink", url, format_id, "--stdout"],
-        stdout=subprocess.PIPE
-    )
-    ff = subprocess.Popen(
-        ["ffmpeg", "-i", "pipe:0", "-c", "copy",
-         "-bsf:a", "aac_adtstoasc",
-         "-movflags", "frag_keyframe+empty_moov",
-         "-f", "mp4", "pipe:1"],
-        stdin=sl.stdout,
-        stdout=subprocess.PIPE,
-        stderr=None
-    )
-    return StreamingResponse(
-        ff.stdout,
-        media_type="video/mp4",
-        headers={"Content-Disposition": f'attachment; filename="{name}.mp4"'}
-    )
-
-@app.get("/files/{name}")
-def get_file(name: str):
-    path = f"videos/{name}"
-    if not os.path.exists(path):
-        raise HTTPException(404, "not found")
-    return FileResponse(path)
+from core.app import app