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

"""封面模板 InMemory 仓储实现。"""

from packages.domain.cover_template import CoverTemplate


class InMemoryCoverTemplateRepository:
    """内存中的封面模板仓储，用于测试。"""

    def __init__(self) -> None:
        self._templates: dict[str, CoverTemplate] = {}

    def create(self, template: CoverTemplate) -> CoverTemplate:
        self._templates[template.id] = template
        return template

    def get(self, template_id: str) -> CoverTemplate | None:
        return self._templates.get(template_id)

    def list_for_user(self, user_id: str, skip: int = 0, limit: int = 100) -> list[CoverTemplate]:
        """列出系统模板 + 用户自己的模板。"""
        visible = [t for t in self._templates.values() if t.is_system or t.user_id == user_id]
        visible.sort(key=lambda t: (t.is_system, t.created_at), reverse=True)
        return visible[skip : skip + limit]

    def count_for_user(self, user_id: str) -> int:
        return sum(1 for t in self._templates.values() if t.is_system or t.user_id == user_id)

    def update(self, template: CoverTemplate) -> CoverTemplate:
        if template.id not in self._templates:
            raise ValueError(f"模板 {template.id} 不存在")
        self._templates[template.id] = template
        return template

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

    def list_system_templates(self) -> list[CoverTemplate]:
        return [t for t in self._templates.values() if t.is_system]