from sqlalchemy.orm import Session

from tofu_api.common.database import SQLAlchemy


class Dependencies:
    """
    Container for dependency injection.
    """

    _dependency_cache: dict

    def __init__(self):
        self._dependency_cache = {}

    # Database dependencies

    def get_sqlalchemy(self) -> SQLAlchemy:
        if SQLAlchemy not in self._dependency_cache:
            self._dependency_cache[SQLAlchemy] = SQLAlchemy()
        return self._dependency_cache[SQLAlchemy]

    def get_db_session(self) -> Session:
        return self.get_sqlalchemy().session