import subprocess
from app_images import init_images_router
from app_files import init_files_router
from app_pods import init_pods_router
from app_containers import init_containers_router, start_stats_poller
from app_networks import init_networks_router
from app_system import init_system_router
from fastapi import FastAPI
import requests_unixsocket
from common import (
    _systemctl as _common_systemctl,
)
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"

@app.on_event("startup")
async def _startup_stats_poller():
    await start_stats_poller()

def _systemctl(cmd):
    return _common_systemctl(cmd, run)

# --- ROUTERS ---
# Images API lives in dedicated modules to keep this file from growing further.
app.include_router(init_system_router(SESSION, PODMAN_API_BASE, WORKLOADS_DIR))
app.include_router(init_images_router(SESSION, PODMAN_API_BASE))
app.include_router(init_files_router(SESSION, PODMAN_API_BASE, WORKLOADS_DIR))
app.include_router(init_networks_router(SESSION, PODMAN_API_BASE))
app.include_router(init_containers_router(
    SESSION,
    PODMAN_API_BASE,
    WORKLOADS_DIR,
    _systemctl,
))
app.include_router(init_pods_router(
    SESSION,
    PODMAN_API_BASE,
    WORKLOADS_DIR,
    _systemctl,
))


def run(cmd):
    try:
        result = subprocess.run(cmd, capture_output=True, text=True, check=False)
        output = (result.stdout or "") + (result.stderr or "")
        return result.returncode, output.strip()
    except Exception as e:
        return 1, str(e)

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