| from typing import AsyncGenerator, Optional |
| from functools import lru_cache |
| from ..services.version_service import VersionService |
| from .supabase_repositories import ( |
| SupabaseVersionRepository, |
| SupabaseAgentRepository |
| ) |
| from services.supabase import DBConnection |
|
|
|
|
| _db_connection: Optional[DBConnection] = None |
|
|
| def set_db_connection(db: DBConnection): |
| global _db_connection |
| _db_connection = db |
|
|
| class DependencyContainer: |
| def __init__(self): |
| self._version_service = None |
| |
| async def get_db_client(self): |
| if not _db_connection: |
| raise RuntimeError("Database connection not initialized. Call set_db_connection first.") |
| return await _db_connection.client |
| |
| async def get_version_repository(self) -> SupabaseVersionRepository: |
| client = await self.get_db_client() |
| return SupabaseVersionRepository(client) |
| |
| async def get_agent_repository(self) -> SupabaseAgentRepository: |
| client = await self.get_db_client() |
| return SupabaseAgentRepository(client) |
| |
| async def get_version_service(self) -> VersionService: |
| if not self._version_service: |
| version_repo = await self.get_version_repository() |
| agent_repo = await self.get_agent_repository() |
| |
| self._version_service = VersionService( |
| version_repo=version_repo, |
| agent_repo=agent_repo |
| ) |
| |
| return self._version_service |
|
|
|
|
| @lru_cache() |
| def get_container() -> DependencyContainer: |
| return DependencyContainer() |
|
|
|
|
| async def get_version_service() -> VersionService: |
| container = get_container() |
| return await container.get_version_service() |
