76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
from contextlib import asynccontextmanager
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from sqlalchemy import select
|
|
|
|
from app.api.v1.router import api_router
|
|
from app.auth.api_key import hash_api_key
|
|
from app.config import get_settings
|
|
from app.database import SessionLocal
|
|
from app.models import ApiKey
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
|
|
|
|
def bootstrap_api_key() -> None:
|
|
settings = get_settings()
|
|
if not settings.sac_bootstrap_api_key:
|
|
return
|
|
db = SessionLocal()
|
|
try:
|
|
key_hash = hash_api_key(settings.sac_bootstrap_api_key)
|
|
exists = db.scalar(select(ApiKey).where(ApiKey.key_hash == key_hash))
|
|
if exists:
|
|
return
|
|
prefix = settings.sac_bootstrap_api_key[:12]
|
|
db.add(
|
|
ApiKey(
|
|
name="bootstrap",
|
|
key_prefix=prefix,
|
|
key_hash=key_hash,
|
|
is_active=True,
|
|
)
|
|
)
|
|
db.commit()
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_app: FastAPI):
|
|
bootstrap_api_key()
|
|
yield
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
settings = get_settings()
|
|
app = FastAPI(
|
|
title="Security Alert Center",
|
|
version="0.1.0",
|
|
lifespan=lifespan,
|
|
)
|
|
origins = [o.strip() for o in settings.cors_origins.split(",") if o.strip()]
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=origins if origins != ["*"] else ["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
from app.api.v1.health import router as health_router
|
|
|
|
app.include_router(api_router, prefix="/api/v1")
|
|
app.include_router(health_router)
|
|
|
|
frontend_dist = ROOT / "frontend" / "dist"
|
|
if frontend_dist.is_dir():
|
|
app.mount("/", StaticFiles(directory=str(frontend_dist), html=True), name="ui")
|
|
|
|
return app
|
|
|
|
|
|
app = create_app()
|