22 lines
841 B
Kotlin
22 lines
841 B
Kotlin
package dev.inmo.tgbotapi.extensions.utils.chat_events
|
|
|
|
import com.github.insanusmokrassar.TelegramBotAPI.types.message.*
|
|
import com.github.insanusmokrassar.TelegramBotAPI.types.message.abstracts.ChatEventMessage
|
|
import kotlinx.coroutines.flow.Flow
|
|
import kotlinx.coroutines.flow.mapNotNull
|
|
import kotlin.reflect.KClass
|
|
|
|
|
|
fun <T : ChatEventMessage> Flow<ChatEventMessage>.divideBySource(contentType: KClass<T>) = mapNotNull {
|
|
if (contentType.isInstance(it)) {
|
|
@Suppress("UNCHECKED_CAST")
|
|
it as T
|
|
} else {
|
|
null
|
|
}
|
|
}
|
|
|
|
fun Flow<ChatEventMessage>.onlyChannelEvents() = divideBySource(ChannelEventMessage::class)
|
|
fun Flow<ChatEventMessage>.onlyGroupEvents() = divideBySource(CommonGroupEventMessage::class)
|
|
fun Flow<ChatEventMessage>.onlySupergroupEvents() = divideBySource(CommonSupergroupEventMessage::class)
|