xiaoxia-saas/packages/domain/entities.py

from __future__ import annotations

from dataclasses import dataclass, field
from datetime import datetime, timezone
from enum import StrEnum
from typing import Any
from uuid import uuid4


class AssetLibraryKind(StrEnum):
    VIDEO = "video"
    VOICE = "voice"


class IngestJobStatus(StrEnum):
    PENDING = "pending"
    PROCESSING = "processing"
    COMPLETED = "completed"
    FAILED = "failed"


@dataclass(slots=True)
class User:
    id: str
    email: str
    display_name: str
    username: str = ""
    # 认证相关字段
    password_hash: str = ""
    email_verified: bool = False
    email_verification_token: str | None = None
    password_reset_token: str | None = None
    password_reset_expires_at: datetime | None = None
    last_login_at: datetime | None = None
    last_login_ip: str | None = None
    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))


@dataclass(slots=True)
class Workspace:
    id: str
    name: str
    owner_user_id: str
    # 订阅相关字段
    subscription_plan: str = "free"  # free, pro, enterprise
    subscription_status: str = "active"  # active, cancelled, expired
    subscription_expires_at: datetime | None = None
    # 配额限制
    max_projects: int = 3  # free: 3, pro: unlimited, enterprise: unlimited
    max_storage_gb: int = 10  # free: 10, pro: 100, enterprise: 1000
    used_storage_gb: float = 0.0
    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))


@dataclass(slots=True)
class Project:
    id: str
    workspace_id: str
    name: str
    description: str = ""
    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))

    @classmethod
    def create(cls, workspace_id: str, name: str, description: str = "") -> "Project":
        clean_name = name.strip()
        if not clean_name:
            raise ValueError("项目名称不能为空")
        return cls(
            id=uuid4().hex,
            workspace_id=workspace_id,
            name=clean_name,
            description=description.strip(),
        )


@dataclass(slots=True)
class AssetLibrary:
    id: str
    workspace_id: str
    project_id: str
    name: str
    kind: AssetLibraryKind
    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))

    @classmethod
    def create(
        cls,
        workspace_id: str,
        project_id: str,
        name: str,
        kind: AssetLibraryKind,
    ) -> "AssetLibrary":
        clean_name = name.strip()
        if not clean_name:
            raise ValueError("素材库名称不能为空")
        return cls(
            id=uuid4().hex,
            workspace_id=workspace_id,
            project_id=project_id,
            name=clean_name,
            kind=kind,
        )


@dataclass(slots=True)
class Asset:
    id: str
    workspace_id: str
    project_id: str
    library_id: str
    name: str
    storage_key: str
    mime_type: str
    metadata: dict[str, Any] = field(default_factory=dict)
    tags: list[str] = field(default_factory=list)
    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))

    @classmethod
    def create(
        cls,
        workspace_id: str,
        project_id: str,
        library_id: str,
        name: str,
        storage_key: str,
        mime_type: str,
        metadata: dict[str, Any] | None = None,
    ) -> "Asset":
        clean_name = name.strip()
        if not clean_name:
            raise ValueError("素材名称不能为空")
        if not storage_key.strip():
            raise ValueError("storage_key 不能为空")
        if not mime_type.strip():
            raise ValueError("mime_type 不能为空")
        return cls(
            id=uuid4().hex,
            workspace_id=workspace_id,
            project_id=project_id,
            library_id=library_id,
            name=clean_name,
            storage_key=storage_key.strip(),
            mime_type=mime_type.strip(),
            metadata=metadata or {},
            tags=[],
        )

    def add_tag(self, tag: str) -> None:
        """添加标签。空标签会被忽略，自动去重。"""
        clean_tag = tag.strip()
        if not clean_tag:
            raise ValueError("标签不能为空")
        if clean_tag not in self.tags:
            self.tags.append(clean_tag)

    def remove_tag(self, tag: str) -> None:
        """删除标签。如果标签不存在，不报错（幂等性）。"""
        clean_tag = tag.strip()
        if clean_tag in self.tags:
            self.tags.remove(clean_tag)


@dataclass(slots=True)
class IngestJob:
    id: str
    workspace_id: str
    project_id: str
    library_id: str
    storage_key: str
    status: IngestJobStatus = IngestJobStatus.PENDING
    error_message: str = ""
    result_asset_id: str = ""
    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
    updated_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))

    @classmethod
    def create(
        cls,
        workspace_id: str,
        project_id: str,
        library_id: str,
        storage_key: str,
    ) -> "IngestJob":
        if not workspace_id.strip():
            raise ValueError("workspace_id 不能为空")
        if not project_id.strip():
            raise ValueError("project_id 不能为空")
        if not library_id.strip():
            raise ValueError("library_id 不能为空")
        if not storage_key.strip():
            raise ValueError("storage_key 不能为空")
        return cls(
            id=uuid4().hex,
            workspace_id=workspace_id.strip(),
            project_id=project_id.strip(),
            library_id=library_id.strip(),
            storage_key=storage_key.strip(),
        )