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 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="",
        )