updates and fixes

posts/panel/src/jvmMain/kotlin/Plugin.kt

@@ -23,6 +23,7 @@ import dev.inmo.tgbotapi.extensions.utils.extensions.sameMessage
 import dev.inmo.tgbotapi.extensions.utils.types.buttons.dataButton
 import dev.inmo.tgbotapi.extensions.utils.types.buttons.flatInlineKeyboard
 import dev.inmo.tgbotapi.types.ChatId
+import dev.inmo.tgbotapi.types.IdChatIdentifier
 import dev.inmo.tgbotapi.types.MessageIdentifier
 import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.CallbackDataInlineKeyboardButton
 import dev.inmo.tgbotapi.types.buttons.InlineKeyboardMarkup
@@ -108,7 +109,7 @@ object Plugin : Plugin {
 
         suspend fun refreshPostMessage(
             postId: PostId,
-            chatId: ChatId,
+            chatId: IdChatIdentifier,
             messageId: MessageIdentifier
         ) {
             val post = postsRepo.getById(postId) ?: return

posts/panel/src/jvmMain/kotlin/repos/PostsMessages.kt

@@ -5,18 +5,20 @@ import dev.inmo.micro_utils.repos.exposed.keyvalue.ExposedKeyValueRepo
 import dev.inmo.micro_utils.repos.mappers.withMapper
 import dev.inmo.plaguposter.posts.models.PostId
 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 kotlinx.serialization.builtins.PairSerializer
 import kotlinx.serialization.builtins.serializer
 import kotlinx.serialization.json.Json
 import org.jetbrains.exposed.sql.Database
 
-private val ChatIdToMessageSerializer = PairSerializer(ChatId.serializer(), MessageIdentifier.serializer())
+private val ChatIdToMessageSerializer = PairSerializer(FullChatIdentifierSerializer, MessageIdentifier.serializer())
 
 fun PostsMessages(
     database: Database,
     json: Json
-): KeyValueRepo<PostId, Pair<ChatId, MessageIdentifier>> = ExposedKeyValueRepo<String, String>(
+): KeyValueRepo<PostId, Pair<IdChatIdentifier, MessageIdentifier>> = ExposedKeyValueRepo<String, String>(
     database,
     { text("postId") },
     { text("chatToMessage") },
@@ -25,5 +27,5 @@ fun PostsMessages(
     { string },
     { json.encodeToString(ChatIdToMessageSerializer, this) },
     { PostId(this) },
-    { json.decodeFromString(ChatIdToMessageSerializer, this) }
+    { json.decodeFromString(ChatIdToMessageSerializer, this) as Pair<IdChatIdentifier, MessageIdentifier> }
 )

posts/src/commonMain/kotlin/models/PostContentInfo.kt

@@ -1,25 +1,39 @@
 package dev.inmo.plaguposter.posts.models
 
-import dev.inmo.tgbotapi.extensions.utils.mediaGroupMessageOrNull
+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.MessageContent
+import dev.inmo.tgbotapi.types.message.content.MediaGroupContent
 import kotlinx.serialization.Serializable
 
 @Serializable
 data class PostContentInfo(
-    val chatId: ChatId,
+    @Serializable(FullChatIdentifierSerializer::class)
+    val chatId: IdChatIdentifier,
     val messageId: MessageIdentifier,
     val group: String?,
     val order: Int
 ) {
     companion object {
-        fun fromMessage(message: ContentMessage<*>, order: Int) = PostContentInfo(
+        private fun fromMessage(message: ContentMessage<*>, order: Int) = PostContentInfo(
             message.chat.id,
             message.messageId,
-            message.mediaGroupMessageOrNull() ?.mediaGroupId,
+            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))
+            }
+        }
     }
 }

posts/src/commonMain/kotlin/repo/ReadPostsRepo.kt

@@ -4,9 +4,10 @@ import com.soywiz.klock.DateTime
 import dev.inmo.micro_utils.repos.ReadCRUDRepo
 import dev.inmo.plaguposter.posts.models.*
 import dev.inmo.tgbotapi.types.ChatId
+import dev.inmo.tgbotapi.types.IdChatIdentifier
 import dev.inmo.tgbotapi.types.MessageIdentifier
 
 interface ReadPostsRepo : ReadCRUDRepo<RegisteredPost, PostId> {
-    suspend fun getIdByChatAndMessage(chatId: ChatId, messageId: MessageIdentifier): PostId?
+    suspend fun getIdByChatAndMessage(chatId: IdChatIdentifier, messageId: MessageIdentifier): PostId?
     suspend fun getPostCreationTime(postId: PostId): DateTime?
 }

posts/src/commonMain/kotlin/sending/PostPublisher.kt

@@ -12,6 +12,7 @@ import dev.inmo.tgbotapi.extensions.api.send.send
 import dev.inmo.tgbotapi.extensions.utils.*
 import dev.inmo.tgbotapi.types.*
 import dev.inmo.tgbotapi.types.message.content.MediaGroupContent
+import dev.inmo.tgbotapi.types.message.content.MediaGroupPartContent
 
 class PostPublisher(
     private val bot: TelegramBot,
@@ -37,14 +38,26 @@ class PostPublisher(
 
         sortedMessagesContents.forEach { (_, contents) ->
             contents.singleOrNull() ?.also {
-                bot.copyMessage(targetChatId, it.chatId, it.messageId)
+                runCatching {
+                    bot.copyMessage(targetChatId, it.chatId, it.messageId)
+                }.onFailure { _ ->
+                    runCatching {
+                        bot.forwardMessage(
+                            it.chatId,
+                            targetChatId,
+                            it.messageId
+                        )
+                    }.onSuccess {
+                        bot.copyMessage(targetChatId, it)
+                    }
+                }
                 return@forEach
             }
             val resultContents = contents.mapNotNull {
                 it.order to (bot.forwardMessage(toChatId = cachingChatId, fromChatId = it.chatId, messageId = it.messageId).contentMessageOrNull() ?: return@mapNotNull null)
-            }.sortedBy { it.first }.mapNotNull { (_, it) ->
-                it.withContentOrNull<MediaGroupContent>() ?: null.also { _ ->
-                    bot.copyMessage(targetChatId, it)
+            }.sortedBy { it.first }.mapNotNull { (_, forwardedMessage) ->
+                forwardedMessage.withContentOrNull<MediaGroupPartContent>() ?: null.also { _ ->
+                    bot.copyMessage(targetChatId, forwardedMessage)
                 }
             }
             resultContents.singleOrNull() ?.also {

posts/src/jvmMain/kotlin/exposed/ExposedContentInfoRepo.kt

@@ -5,6 +5,7 @@ import dev.inmo.micro_utils.repos.KeyValuesRepo
 import dev.inmo.micro_utils.repos.exposed.*
 import dev.inmo.plaguposter.posts.models.*
 import dev.inmo.tgbotapi.types.ChatId
+import dev.inmo.tgbotapi.types.IdChatIdentifier
 import org.jetbrains.exposed.sql.*
 
 internal class ExposedContentInfoRepo(
@@ -13,13 +14,14 @@ internal class ExposedContentInfoRepo(
 ) : ExposedRepo, Table(name = "posts_content") {
     val postIdColumn = text("post_id").references(postIdColumnReference, ReferenceOption.CASCADE, ReferenceOption.CASCADE)
     val chatIdColumn = long("chat_id")
+    val threadIdColumn = long("thread_id").nullable().default(null)
     val messageIdColumn = long("message_id")
     val groupColumn = text("group").nullable()
     val orderColumn = integer("order")
 
     val ResultRow.asObject
         get() = PostContentInfo(
-            ChatId(get(chatIdColumn)),
+            IdChatIdentifier(get(chatIdColumn), get(threadIdColumn)),
             get(messageIdColumn),
             get(groupColumn),
             get(orderColumn)

posts/src/jvmMain/kotlin/exposed/ExposedPostsRepo.kt

@@ -9,10 +9,13 @@ import dev.inmo.micro_utils.repos.exposed.initTable
 import dev.inmo.plaguposter.posts.models.*
 import dev.inmo.plaguposter.posts.repo.PostsRepo
 import dev.inmo.tgbotapi.types.ChatId
+import dev.inmo.tgbotapi.types.IdChatIdentifier
 import dev.inmo.tgbotapi.types.MessageIdentifier
 import kotlinx.coroutines.flow.*
 import kotlinx.serialization.json.Json
 import org.jetbrains.exposed.sql.*
+import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
+import org.jetbrains.exposed.sql.SqlExpressionBuilder.inList
 import org.jetbrains.exposed.sql.statements.*
 import org.jetbrains.exposed.sql.transactions.transaction
 
@@ -33,8 +36,8 @@ class ExposedPostsRepo(
 
     override val primaryKey: PrimaryKey = PrimaryKey(idColumn)
 
-    override val selectById: SqlExpressionBuilder.(PostId) -> Op<Boolean> = { idColumn.eq(it.string) }
-    override val selectByIds: SqlExpressionBuilder.(List<PostId>) -> Op<Boolean> = { idColumn.inList(it.map { it.string }) }
+    override val selectById: ISqlExpressionBuilder.(PostId) -> Op<Boolean> = { idColumn.eq(it.string) }
+    override val selectByIds: ISqlExpressionBuilder.(List<PostId>) -> Op<Boolean> = { idColumn.inList(it.map { it.string }) }
     override val ResultRow.asObject: RegisteredPost
         get() {
             val id = PostId(get(idColumn))
@@ -86,6 +89,7 @@ class ExposedPostsRepo(
                     insert {
                         it[postIdColumn] = post.id.string
                         it[chatIdColumn] = contentInfo.chatId.chatId
+                        it[threadIdColumn] = contentInfo.chatId.threadId
                         it[messageIdColumn] = contentInfo.messageId
                         it[groupColumn] = contentInfo.group
                         it[orderColumn] = contentInfo.order
@@ -102,14 +106,14 @@ class ExposedPostsRepo(
 
     override suspend fun onAfterCreate(values: List<Pair<NewPost, RegisteredPost>>): List<RegisteredPost> {
         values.forEach {
-            updateContent(it.second)
+            updateContent(it.second.copy(content = it.first.content))
         }
         return super.onAfterCreate(values)
     }
 
     override suspend fun onAfterUpdate(value: List<UpdatedValuePair<NewPost, RegisteredPost>>): List<RegisteredPost> {
         value.forEach {
-            updateContent(it.second)
+            updateContent(it.second.copy(content = it.first.content))
         }
         return super.onAfterUpdate(value)
     }
@@ -122,7 +126,7 @@ class ExposedPostsRepo(
         val existsIds = posts.keys.toList()
         transaction(db = database) {
             val deleted = deleteWhere(null, null) {
-                selectByIds(existsIds)
+                selectByIds(it, existsIds)
             }
             with(contentRepo) {
                 deleteWhere {
@@ -142,10 +146,10 @@ class ExposedPostsRepo(
         }
     }
 
-    override suspend fun getIdByChatAndMessage(chatId: ChatId, messageId: MessageIdentifier): PostId? {
+    override suspend fun getIdByChatAndMessage(chatId: IdChatIdentifier, messageId: MessageIdentifier): PostId? {
         return transaction(database) {
             with(contentRepo) {
-                select { chatIdColumn.eq(chatId.chatId).and(messageIdColumn.eq(messageId)) }.limit(1).firstOrNull() ?.get(postIdColumn)
+                select { chatIdColumn.eq(chatId.chatId).and(threadIdColumn.eq(chatId.threadId)).and(messageIdColumn.eq(messageId)) }.limit(1).firstOrNull() ?.get(postIdColumn)
             } ?.let(::PostId)
         }
     }