breakpilot-compliance/backend-compliance/main.py

"""
BreakPilot Compliance Backend

Extracted compliance-specific APIs from the monorepo backend.
Provides: Compliance Framework, Consent Admin, DSR, GDPR Export.

Runs on port 8002 with DB search_path=compliance,core,public.
"""

import logging
import os
from contextlib import asynccontextmanager

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

# Configure root logging so all modules' logger.info() etc. are visible
logging.basicConfig(
    level=logging.INFO,
    format="%(levelname)s:%(name)s: %(message)s",
)

logger = logging.getLogger(__name__)

# Compliance-specific API routers
from consent_api import router as consent_router
from consent_admin_api import router as consent_admin_router
from gdpr_api import router as gdpr_router, admin_router as gdpr_admin_router
# DSR proxy removed — now handled natively in compliance/api/dsr_routes.py

# Compliance framework sub-package
from compliance.api import router as compliance_framework_router

# Source Policy
from compliance.api.source_policy_router import router as source_policy_router

# Document Import & Screening
from compliance.api.import_routes import router as import_router
from compliance.api.screening_routes import router as screening_router

# Company Profile
from compliance.api.company_profile_routes import router as company_profile_router

# Agent (ZeroClaw compliance agent)
from compliance.api.agent_notification_routes import router as agent_notify_router
from compliance.api.agent_analyze_routes import router as agent_analyze_router
from compliance.api.agent_scan_routes import router as agent_scan_router
from compliance.api.agent_doc_check_routes import router as agent_doc_check_router

# Middleware
from middleware import (
    RequestIDMiddleware,
    SecurityHeadersMiddleware,
)

@asynccontextmanager
async def lifespan(app: FastAPI):
    """Run migrations on startup."""
    from migration_runner import run_migrations
    try:
        run_migrations()
        logger.info("Database migrations completed")
    except Exception as e:
        logger.error("Migration failed: %s — backend starting anyway", e)
    yield


app = FastAPI(
    title="BreakPilot Compliance Backend",
    description="GDPR/DSGVO Compliance, Consent Management, Data Subject Requests, and Regulatory Compliance Framework",
    version="1.0.0",
    lifespan=lifespan,
)

# --- CORS ---
ALLOWED_ORIGINS = os.getenv("CORS_ORIGINS", "*").split(",")
app.add_middleware(
    CORSMiddleware,
    allow_origins=ALLOWED_ORIGINS,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# --- Security & Tracing Middleware ---
app.add_middleware(RequestIDMiddleware)
app.add_middleware(SecurityHeadersMiddleware)


# --- Health Endpoint ---
@app.get("/health", tags=["system"])
async def health():
    """Health check endpoint for load balancers and orchestration."""
    return {
        "status": "healthy",
        "service": "backend-compliance",
        "version": "1.0.0",
    }


@app.get("/debug/routers", tags=["system"])
async def debug_routers():
    """Diagnostic: show which sub-routers loaded and which failed."""
    from compliance.api import _ROUTER_MODULES, _failed_routers, _loaded_count
    return {
        "total": len(_ROUTER_MODULES),
        "loaded": _loaded_count,
        "failed_count": len(_failed_routers),
        "failed": _failed_routers,
    }


# --- Compliance-specific Routers ---

# Consent (user-facing)
app.include_router(consent_router, prefix="/api")

# Consent Admin
app.include_router(consent_admin_router, prefix="/api")

# GDPR / Privacy (user-facing)
app.include_router(gdpr_router, prefix="/api")

# GDPR Admin
app.include_router(gdpr_admin_router, prefix="/api")

# DSR now handled natively via compliance_framework_router (dsr_routes.py)

# Compliance Framework (regulations, controls, evidence, risks, audits, ISMS)
app.include_router(compliance_framework_router, prefix="/api")

# Source Policy (allowed sources, PII rules, audit)
app.include_router(source_policy_router, prefix="/api")

# Document Import (PDF analysis, gap detection)
app.include_router(import_router, prefix="/api")

# System Screening (SBOM generation, vulnerability scan)
app.include_router(screening_router, prefix="/api")

# Company Profile (CRUD with audit logging)
app.include_router(company_profile_router, prefix="/api")

# Agent (ZeroClaw compliance agent → analyze + email via SMTP)
app.include_router(agent_notify_router, prefix="/api")
app.include_router(agent_analyze_router, prefix="/api")
app.include_router(agent_scan_router, prefix="/api")
app.include_router(agent_doc_check_router, prefix="/api")


if __name__ == "__main__":
    import uvicorn

    uvicorn.run(
        "main:app",
        host="0.0.0.0",
        port=int(os.getenv("PORT", "8002")),
        reload=os.getenv("ENVIRONMENT", "development") == "development",
    )