import os
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import requests_unixsocket
import uvicorn

app = FastAPI(title="Podman MVP Control Plane", root_path="/api")
SESSION = requests_unixsocket.Session()
PODMAN_API_BASE = "http+unix://%2Frun%2Fuser%2F1000%2Fpodman%2Fpodman.sock/v5.4.2"
WORKLOADS_DIR = "/app/workloads"

class FileCreate(BaseModel):
    path: str
    content: str

@app.get("/workloads")
def list_workloads():
    yaml_files = []
    for root, dirs, files in os.walk(WORKLOADS_DIR):
        for file in files:
            if file.endswith((".yaml", ".kube")):
                rel_path = os.path.relpath(os.path.join(root, file), WORKLOADS_DIR)
                yaml_files.append(rel_path)
    return {"workloads": sorted(yaml_files)}

@app.get("/workloads/read/{filename:path}")
def read_workload(filename: str):
    path = os.path.join(WORKLOADS_DIR, filename)
    if not os.path.exists(path): raise HTTPException(status_code=404)
    with open(path, 'r') as f: return {"filename": filename, "content": f.read()}

@app.post("/workloads/save-file")
def save_file(file_data: FileCreate):
    target_path = os.path.join(WORKLOADS_DIR, file_data.path)
    os.makedirs(os.path.dirname(target_path), exist_ok=True)
    with open(target_path, 'w') as f: f.write(file_data.content)
    return {"status": "success"}

@app.post("/workloads/deploy/{filename:path}")
def deploy_workload(filename: str):
    path = os.path.join(WORKLOADS_DIR, filename)
    with open(path, 'r') as f: yaml_content = f.read()
    url = f"{PODMAN_API_BASE}/libpod/kube/play"
    return SESSION.post(url, data=yaml_content).json()

# --- PODS ---
@app.get("/pods")
def list_pods():
    # Cruciaal: ?all=true zorgt dat EXIT_STATE pods ook getoond worden
    url = f"{PODMAN_API_BASE}/libpod/pods/json?all=true"
    return SESSION.get(url).json()

@app.post("/pods/{action}/{name}")
def pod_action(action: str, name: str):
    # Podman API pad: /libpod/pods/{name}/{action}
    url = f"{PODMAN_API_BASE}/libpod/pods/{name}/{action}"
    res = SESSION.post(url)
    return {"status": "success", "code": res.status_code}

# --- CONTAINERS ---
@app.get("/containers")
def list_containers():
    # Ook hier ?all=true voor gestopte containers
    url = f"{PODMAN_API_BASE}/libpod/containers/json?all=true"
    return SESSION.get(url).json()

@app.post("/containers/{action}/{name}")
def container_action(action: str, name: str):
    # Podman API pad: /libpod/containers/{name}/{action}
    url = f"{PODMAN_API_BASE}/libpod/containers/{name}/{action}"
    res = SESSION.post(url)
    return {"status": "success", "code": res.status_code}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)