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"
    IMAGE = "image"


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))


class WorkspaceMemberRole(StrEnum):
    """工作空间成员角色"""

    OWNER = "owner"  # 所有者（创建者，唯一）
    ADMIN = "admin"  # 管理员（可管理成员和项目）
    MEMBER = "member"  # 成员（可创建和编辑项目）
    VIEWER = "viewer"  # 查看者（只读）


class InvitationStatus(StrEnum):
    """邀请状态"""

    PENDING = "pending"  # 待处理
    ACCEPTED = "accepted"  # 已接受
    DECLINED = "declined"  # 已拒绝
    EXPIRED = "expired"  # 已过期


@dataclass(slots=True)
class WorkspaceMember:
    """工作空间成员"""

    id: str
    workspace_id: str
    user_id: str
    role: str  # owner, admin, member, viewer
    invited_by: str | None = None  # 邀请人 user_id
    joined_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))


@dataclass(slots=True)
class WorkspaceInvitation:
    """工作空间邀请"""

    id: str
    workspace_id: str
    inviter_user_id: str  # 邀请人
    invitee_email: str  # 被邀请人邮箱
    role: str  # 邀请角色
    invitation_token: str  # 邀请令牌
    status: str = "pending"  # pending, accepted, declined, expired
    expires_at: datetime | None = None  # 过期时间
    accepted_at: datetime | None = None  # 接受时间
    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
    asset_count: int = 0
    total_size: int = 0
    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,
        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,
            asset_count=0,
            total_size=0,
        )


class AssetStatus(StrEnum):
    UPLOADING = "uploading"
    READY = "ready"
    PROCESSING = "processing"
    ERROR = "error"


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


@dataclass(slots=True)
class Asset:
    id: str
    workspace_id: str
    project_id: str
    library_id: str
    name: str
    storage_key: str
    mime_type: str
    file_size: int = 0
    thumbnail_url: str | None = None
    duration: float | None = None
    width: int | None = None
    height: int | None = None
    fps: float | None = None
    codec: str | None = None
    status: AssetStatus = AssetStatus.UPLOADING
    classification_status: ClassificationStatus = ClassificationStatus.PENDING
    quality_score: float | None = None
    uploaded_by_user_id: 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))
    updated_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,
        *,
        file_size: int = 0,
        thumbnail_url: str | None = None,
        duration: float | None = None,
        width: int | None = None,
        height: int | None = None,
        fps: float | None = None,
        codec: str | None = None,
        status: AssetStatus = AssetStatus.UPLOADING,
        classification_status: ClassificationStatus = ClassificationStatus.PENDING,
        quality_score: float | None = None,
        uploaded_by_user_id: str = "",
    ) -> "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(),
            file_size=file_size,
            thumbnail_url=thumbnail_url,
            duration=duration,
            width=width,
            height=height,
            fps=fps,
            codec=codec,
            status=status,
            classification_status=classification_status,
            quality_score=quality_score,
            uploaded_by_user_id=uploaded_by_user_id.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)
            self.updated_at = datetime.now(timezone.utc)

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


@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(),
        )