main.py

import os
from contextlib import asynccontextmanager

from fastapi import Depends, FastAPI
from sqlalchemy import inspect
from sqlalchemy.orm import Session

from shared.log_config import get_logger
from trustregistry import crud, db
from trustregistry.database import engine
from trustregistry.db import get_db
from trustregistry.registry import registry_actors, registry_schemas

logger = get_logger(__name__)

OPENAPI_NAME = os.getenv("OPENAPI_NAME", "Trust Registry")
PROJECT_VERSION = os.getenv("PROJECT_VERSION", "0.11.0")
ROOT_PATH = os.getenv("ROOT_PATH", "")


@asynccontextmanager
async def lifespan(_: FastAPI):
    db.Base.metadata.create_all(bind=engine)
    engine.dispose()

    logger.debug("TrustRegistry startup: Validate tables are created")
    with engine.connect() as connection:
        inspector = inspect(connection)
        table_names = inspector.get_table_names()
        logger.debug("TrustRegistry tables created: `{}`", table_names)
    # start-up logic is before the yield
    yield
    # shutdown logic after


def create_app():
    application = FastAPI(
        root_path=ROOT_PATH,
        title=OPENAPI_NAME,
        version=PROJECT_VERSION,
        description="Welcome to the OpenAPI interface to the Aries CloudAPI trust registry",
        lifespan=lifespan,
    )
    application.include_router(registry_actors.router)
    application.include_router(registry_schemas.router)
    return application


app = create_app()


@app.get("/")
async def root(db_session: Session = Depends(get_db)):
    logger.info("GET request received: Fetch actors and schemas from registry")
    db_schemas = crud.get_schemas(db_session)
    db_actors = crud.get_actors(db_session)
    schemas_repr = [schema.id for schema in db_schemas]
    logger.info("Successfully fetched actors and schemas from registry.")
    return {"actors": db_actors, "schemas": schemas_repr}


@app.get("/registry")
async def registry(db_session: Session = Depends(get_db)):
    return await root(db_session)