package com.suno.android.ui.bottom_sheets.comments

import com.suno.android.common_data.comments.Comment
import com.suno.android.common_data.comments.CommentReply
import com.suno.android.common_data.comments.CommentType
import com.suno.android.common_data.mentions.UserMention
import com.suno.android.common_ui.components.list_items.comments.CommentListItemState
import com.suno.android.common_ui.components.list_items.comments.CommentListItemType

fun Comment.toCommentListItemState(): CommentListItemState =
    CommentListItemState(
        id = id.value,
        entityId = mediaEntity.mediaId.value,
        userId = userId,
        userDisplayName = userDisplayName,
        userHandle = userHandle,
        userAvatarUrl = userAvatarUrl,
        userCommentsBlocked = userCommentsBlocked,
        content = content,
        createdAt = createdAt,
        numLikes = numLikes,
        reactionType = reactionType,
        numReplies = numReplies,
        replies = replies?.map(CommentReply::toCommentReplyState),
        userMentions = userMentions?.map(UserMention::toMentionState),
        trackTimestamp = trackTimestamp,
        entityType = when (type) {
            CommentType.Clip -> CommentListItemType.Clip
            CommentType.Hook -> CommentListItemType.Hook
        },
    )

fun CommentListItemType.toCommentType() =
    when (this) {
        CommentListItemType.Clip -> CommentType.Clip
        CommentListItemType.Hook -> CommentType.Hook
    }