data_models

@dataclass
class ActivityData:
    """Simple dataclass to cache representation data."""

    representation_id: str = ""
    hash: int = field(init=False)
    activity_list: list[
        CommentModel | VersionPublishModel | StatusChangeModel
    ] = field(default_factory=list)

    def __post_init__(self):
        """Compute the hash of the activity list."""
        self.hash = hash(tuple(self.activity_list))

@dataclass
class CommentCategory:
    """Model for powerpack `activity_categories` settings"""

    name: str
    color: str
    access: dict[str, Any]

@dataclass(unsafe_hash=True)
class CommentModel:
    activity_id: str = ""
    user_full_name: str = ""
    user_name: str = ""
    user_src: str = ""
    comment: str = ""  # type: ignore
    category: str = ""
    category_color: str = ""
    comment_date: str = ""
    short_date: str = field(init=False, hash=False)
    type: ActivityCategory = field(
        init=False, default=ActivityCategory.COMMENT, hash=False
    )
    files: list[FileModel] = field(default_factory=list, hash=False)
    annotations: list[AnnotationModel] = field(
        default_factory=list, hash=False
    )
    _comment: str = field(init=False, repr=False)

    def __post_init__(self):
        """Set the date if not set and compute the short date."""
        if not self.comment_date:
            self.comment_date = datetime.now(timezone.utc).isoformat()

        self.short_date = short_date(self.comment_date)

    @property
    def comment(self):  # noqa: F811
        return self._comment

    @comment.setter
    def comment(self, value: str):
        """Check if the comment contains a checklist."""
        if re.search(RE_HAS_CHECKLIST, str(value)):
            self.type = ActivityCategory.COMMENT | ActivityCategory.CHECKLIST
        self._comment = value

@dataclass
class EntityMention:
    """Entity (version or task) that can be mentioned in a comment.

    A mention is stored in the comment markdown as
    ``[label](entity_type:id)``.
    """

    entity_type: str  # "version" or "task"
    id: str
    label: str
    context: str = ""  # e.g. product name of a version, folder of a task
    # Material symbol and color of the product type or task type
    icon: str = ""
    color: str = ""
    suffix: str = ""  # e.g. how long ago a version was created

@dataclass
class FileModel:
    """Model for activity attached files.

    Attachments could be images, there could be also preview available for
    them.
    """

    id: str
    mime: str
    local_path: str = ""
    thumb_local_path: str = ""
    frame: int = 0
    start_frame: int = 0  # for sequences, to compute the actual frame number

@dataclass
class ProjectData:
    """Model to pass project data - anatomy, users, teams"""

    project_name: str
    users: List[User]
    teams: List[Team]
    anatomy: dict[str, Any]
    comment_category: List[CommentCategory]
    current_user: Optional[User]

    @staticmethod
    def not_set():
        ns = "PROJECT NOT SET"
        return ProjectData(
            project_name=ns,
            users=[],
            teams=[],
            anatomy={},
            comment_category=[],
            current_user=None,
        )

@dataclass
class User:
    """Data model for user"""

    name: str
    short_name: str
    full_name: str
    email: str
    avatar_url: str = ""
    avatar_local_path: str = ""

@dataclass
class VersionData:
    """Model to pass enhanced version data"""

    id: str
    name: str
    author: str
    tags: list[str]
    status: str
    product_name: str
    task_name: str
    priority: str
    folder_path: str
    assignees: list[str]
    attrib: dict[str, str]
    thumbnail_id: str = ""
    thumbnail_local_path: str = ""

    @staticmethod
    def not_set():
        ns = "VERSION NOT SET"
        return VersionData(
            id=ns,
            name=ns,
            author=ns,
            tags=[],
            status=ns,
            product_name=ns,
            task_name=ns,
            priority=ns,
            folder_path=ns,
            assignees=[],
            attrib={},
            thumbnail_id="",
            thumbnail_local_path="",
        )

@dataclass(unsafe_hash=True)
class VersionPublishModel:
    """Publish of a version.

    Attributes:
        status: Current status of the published version, not the status
            it had when it was published. Not shown when empty.
        thumbnail_key: Key of the version thumbnail in the
            :class:`~ayon_core.ui.image_cache.ImageCache`. The thumbnail
            is shown if the key is cached, otherwise the widget asks its
            thumbnail loader for it. Not shown when empty.
    """

    activity_id: str = ""
    user_full_name: str = ""
    user_name: str = ""
    user_src: str = ""
    version: str = ""
    product: str = ""
    date: str = ""
    status: str = field(default="", hash=False)
    thumbnail_key: str = field(default="", hash=False)
    short_date: str = field(init=False, hash=False)
    type: ActivityCategory = field(
        init=False, default=ActivityCategory.VERSION_PUBLISH, hash=False
    )

    def __post_init__(self):
        self.short_date = short_date(self.date)