from fastapi import FastAPI, UploadFile, File, Form, HTTPException
from fastapi.responses import FileResponse
from fastapi.middleware.cors import CORSMiddleware
import yaml
import os
import subprocess
from pathlib import Path
from utils import parse_yaml_config, get_presets_dir

app = FastAPI()

# Allow CORS (frontend access)
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.post("/upload-config/")
async def upload_config(file: UploadFile = File(...)):
    config_data = yaml.safe_load(await file.read())
    return {"parsed": config_data}

@app.post("/transcode/")
async def transcode(yaml_config: dict):
    try:
        ffmpeg_cmd = parse_yaml_config(yaml_config)
        subprocess.run(ffmpeg_cmd, check=True)
        return {"status": "success", "cmd": ffmpeg_cmd}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@app.get("/browse/")
def browse(path: str = "/data"):
    if not os.path.exists(path):
        raise HTTPException(status_code=404, detail="Path not found")
    entries = [{"name": f, "is_dir": os.path.isdir(os.path.join(path, f))} for f in os.listdir(path)]
    return {"path": path, "entries": entries}

@app.post("/save-preset/")
def save_preset(name: str = Form(...), file: UploadFile = File(...)):
    path = get_presets_dir() / f"{name}.yml"
    with open(path, "wb") as f:
        f.write(file.file.read())
    return {"message": "Preset saved"}

@app.get("/list-presets/")
def list_presets():
    return {"presets": [p.stem for p in get_presets_dir().glob("*.yml")]}

@app.get("/preset/{name}")
def get_preset(name: str):
    path = get_presets_dir() / f"{name}.yml"
    if not path.exists():
        raise HTTPException(status_code=404, detail="Preset not found")
    return FileResponse(path)