from auth import require_auth
from fastapi import APIRouter, Depends
from models.api import SourceHealthResponse
from watermark import watermarks_collection

router = APIRouter(dependencies=[Depends(require_auth)])

SOURCES = ["directory", "unified", "intune"]


@router.get("/source-health", response_model=list[SourceHealthResponse])
def source_health():
    """Return the last known fetch status for each ingestion source."""
    results = []
    for source in SOURCES:
        doc = watermarks_collection.find_one({"source": source})
        if doc and doc.get("last_fetch_time"):
            results.append({
                "source": source,
                "last_fetch_time": doc["last_fetch_time"],
                "status": "healthy",
            })
        else:
            results.append({
                "source": source,
                "last_fetch_time": None,
                "status": "unknown",
            })
    return results