40 lines
1.3 KiB
Kotlin
40 lines
1.3 KiB
Kotlin
package dev.inmo.plaguposter.posts.models
|
|
|
|
import dev.inmo.tgbotapi.extensions.utils.possiblyMediaGroupMessageOrNull
|
|
import dev.inmo.tgbotapi.types.ChatId
|
|
import dev.inmo.tgbotapi.types.FullChatIdentifierSerializer
|
|
import dev.inmo.tgbotapi.types.IdChatIdentifier
|
|
import dev.inmo.tgbotapi.types.MessageIdentifier
|
|
import dev.inmo.tgbotapi.types.message.abstracts.ContentMessage
|
|
import dev.inmo.tgbotapi.types.message.content.MediaGroupContent
|
|
import kotlinx.serialization.Serializable
|
|
|
|
@Serializable
|
|
data class PostContentInfo(
|
|
@Serializable(FullChatIdentifierSerializer::class)
|
|
val chatId: IdChatIdentifier,
|
|
val messageId: MessageIdentifier,
|
|
val group: String?,
|
|
val order: Int
|
|
) {
|
|
companion object {
|
|
private fun fromMessage(message: ContentMessage<*>, order: Int) = PostContentInfo(
|
|
message.chat.id,
|
|
message.messageId,
|
|
message.possiblyMediaGroupMessageOrNull() ?.mediaGroupId,
|
|
order
|
|
)
|
|
fun fromMessage(message: ContentMessage<*>): List<PostContentInfo> {
|
|
val content = message.content
|
|
|
|
return if (content is MediaGroupContent<*>) {
|
|
content.group.mapIndexed { i, it ->
|
|
fromMessage(it.sourceMessage, i)
|
|
}
|
|
} else {
|
|
listOf(fromMessage(message, 0))
|
|
}
|
|
}
|
|
}
|
|
}
|