main.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from app.core.config import settings
from app.api.endpoints import pdf
from app.utils.ghostscript_inkcov import gs_available
import os
import subprocess
import time

enable_docs = settings.enable_docs or settings.environment in ("dev", "staging")

app = FastAPI(
    title=settings.api_title,
    version=settings.api_version,
    description=settings.api_description,
    docs_url="/docs" if enable_docs else None,
    redoc_url="/redoc" if enable_docs else None,
    openapi_url="/openapi.json" if enable_docs else None,
)
START_TIME = time.time()

def _git_commit_short() -> str:
    # Prefer env var if provided
    commit = os.getenv("GIT_COMMIT")
    if commit:
        return commit[:7]
    # Try git command
    try:
        out = subprocess.run([
            "git", "rev-parse", "--short", "HEAD"
        ], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        return out.stdout.decode().strip()
    except Exception:
        return "unknown"

# Add CORS middleware
app.add_middleware(
    CORSMiddleware,
    allow_origins=settings.cors_origins,
    allow_credentials=settings.cors_allow_credentials,
    allow_methods=settings.cors_allow_methods,
    allow_headers=settings.cors_allow_headers,
)

# Include routers
app.include_router(pdf.router)


@app.get("/")
async def root():
    return {
        "message": "PDF Dieline Processor API",
        "version": settings.api_version,
        "endpoints": {
            "analyze": "/api/pdf/analyze",
            "process": "/api/pdf/process",
            "process_with_json": "/api/pdf/process-with-json-file"
        }
    }


@app.get("/hello/{name}")
async def say_hello(name: str):
    return {"message": f"Hello {name}"}


def _pkg_version(modname: str) -> tuple[bool, str | None]:
    try:
        mod = __import__(modname)
        ver = getattr(mod, "__version__", None)
        if ver is None and modname == "fitz":
            # PyMuPDF historically exposes the version via VersionBind / docstring
            ver = getattr(mod, "VersionBind", None) or getattr(mod, "version", (None,))[0]
        return True, ver
    except Exception:
        return False, None


# Prime Ghostscript probe at import time (cached); never blocks if missing
_GS_PROBE = gs_available()


@app.get("/healthz")
async def healthz():
    pikepdf_ok, pikepdf_ver = _pkg_version("pikepdf")
    fitz_ok, fitz_ver = _pkg_version("fitz")
    pypdf_ok, pypdf_ver = _pkg_version("pypdf")
    gs_ok, gs_ver = _GS_PROBE
    return {
        "status": "ok",
        "uptime_seconds": round(time.time() - START_TIME, 2),
        "dependencies": {
            "pikepdf": {"available": pikepdf_ok, "version": pikepdf_ver},
            "pymupdf": {"available": fitz_ok, "version": fitz_ver},
            "pypdf":   {"available": pypdf_ok, "version": pypdf_ver},
            "ghostscript": {"available": gs_ok, "version": gs_ver},
        },
        "modes": {
            "color_analysis_heuristic": True,
            "color_analysis_measured": gs_ok,
        },
    }


@app.get("/version")
async def version():
    return {
        "name": settings.api_title,
        "version": settings.api_version,
        "git_commit": _git_commit_short(),
    }