xiaoxia-saas/packages/adapters/in_memory/asset_library_repository.py

"""AssetLibrary InMemory Repository 实现"""

from packages.domain import AssetLibrary, AssetLibraryKind


class InMemoryAssetLibraryRepository:
    """素材库 InMemory 仓储实现（用于同步用例和测试）"""

    def __init__(self):
        self._libraries: dict[str, AssetLibrary] = {}

    def create(self, library: AssetLibrary) -> AssetLibrary:
        self._libraries[library.id] = library
        return library

    def get(self, library_id: str) -> AssetLibrary | None:
        return self._libraries.get(library_id)

    def find_by_id(self, library_id: str) -> AssetLibrary | None:
        return self.get(library_id)

    def find_by_project(self, project_id: str, kind: AssetLibraryKind | None = None) -> list[AssetLibrary]:
        items = [library for library in self._libraries.values() if library.project_id == project_id]
        if kind is not None:
            items = [library for library in items if library.kind == kind]
        return items

    def update(self, library: AssetLibrary) -> AssetLibrary:
        self._libraries[library.id] = library
        return library

    def delete(self, library_id: str) -> bool:
        if library_id in self._libraries:
            del self._libraries[library_id]
            return True
        return False

    def increment_asset_count(self, library_id: str, count_delta: int = 1, size_delta: int = 0) -> None:
        library = self._libraries.get(library_id)
        if library:
            library.asset_count += count_delta
            library.total_size += size_delta

    def decrement_asset_count(self, library_id: str, count_delta: int = 1, size_delta: int = 0) -> None:
        library = self._libraries.get(library_id)
        if library:
            library.asset_count = max(0, library.asset_count - count_delta)
            library.total_size = max(0, library.total_size - size_delta)

    def recount_assets(self, library_id: str) -> int:
        """InMemory 实现无法真正重算（没有 asset 数据源），返回当前计数。"""
        library = self._libraries.get(library_id)
        return library.asset_count if library else 0

    def get_or_create_default_library(
        self,
        project_id: str,
        kind: AssetLibraryKind,
        *,
        name: str | None = None,
    ) -> AssetLibrary:
        """幂等获取/创建默认素材库（Issue #1775，内存实现，模拟唯一约束语义）。"""
        for lib in self._libraries.values():
            if lib.project_id == project_id and lib.kind == kind:
                return lib
        # 回退到 find_by_project
        for lib in self.find_by_project(project_id, kind):
            return lib
        library_name = name or f"{kind.value}素材库"
        library = AssetLibrary.create(project_id=project_id, name=library_name, kind=kind)
        return self.create(library)