"""PostgresConnectionManager dependency functions are defined here."""

from fastapi import FastAPI, Request

from {{project_slug}}.db.connection.manager import PostgresConnectionManager


def init_dispencer(app: FastAPI, connection_manager: PostgresConnectionManager) -> None:
    """Initialize PostgresConnectionManager dispencer at app's state."""
    if hasattr(app.state, "postgres_connection_manager_dep"):
        if not isinstance(app.state.postgres_connection_manager_dep, PostgresConnectionManager):
            raise ValueError(
                "postgres_connection_manager_dep attribute of app's state is already set"
                f"with other value ({app.state.postgres_connection_manager_dep})"
            )
        return

    app.state.postgres_connection_manager_dep = connection_manager


def obtain(app_or_request: FastAPI | Request) -> PostgresConnectionManager:
    """Get a PostgresConnectionManager from request's app state."""
    if isinstance(app_or_request, Request):
        app_or_request = app_or_request.app
    if not hasattr(app_or_request.state, "postgres_connection_manager_dep"):
        raise ValueError("PostgresConnectionManager dispencer was not initialized at app preparation")
    return app_or_request.state.postgres_connection_manager_dep