1
0
mirror of https://github.com/InsanusMokrassar/TelegramBotAPI.git synced 2024-06-17 07:15:26 +00:00
tgbotapi/tgbotapi.extensions.behaviour_builder/src/commonMain/kotlin/dev/inmo/tgbotapi/extensions/behaviour_builder/expectations/WaitContent.kt

197 lines
8.4 KiB
Kotlin
Raw Normal View History

@file:Suppress("unused")
package dev.inmo.tgbotapi.extensions.behaviour_builder.expectations
2021-01-06 17:06:48 +00:00
import dev.inmo.micro_utils.coroutines.safelyWithoutExceptions
import dev.inmo.tgbotapi.extensions.behaviour_builder.BehaviourContext
2021-01-07 18:22:19 +00:00
import dev.inmo.tgbotapi.extensions.utils.*
2021-01-06 17:06:48 +00:00
import dev.inmo.tgbotapi.requests.abstracts.Request
import dev.inmo.tgbotapi.types.message.abstracts.CommonMessage
2021-01-06 17:06:48 +00:00
import dev.inmo.tgbotapi.types.message.content.*
import dev.inmo.tgbotapi.types.message.content.abstracts.*
import dev.inmo.tgbotapi.types.message.content.media.*
import dev.inmo.tgbotapi.types.message.payments.InvoiceContent
import kotlinx.coroutines.flow.toList
typealias CommonMessageToContentMapper<T> = suspend CommonMessage<T>.() -> T?
2021-01-06 17:06:48 +00:00
private suspend fun <O> BehaviourContext.waitCommonMessage(
2021-01-06 17:06:48 +00:00
count: Int = 1,
initRequest: Request<*>? = null,
2021-01-07 18:22:19 +00:00
includeMediaGroups: Boolean = true,
2021-01-06 17:06:48 +00:00
errorFactory: NullableRequestBuilder<*> = { null },
mapper: suspend CommonMessage<MessageContent>.() -> O?
2021-01-06 17:06:48 +00:00
): List<O> = expectFlow(
initRequest,
count,
errorFactory
) {
2021-01-07 18:22:19 +00:00
if (includeMediaGroups) {
it.asSentMediaGroupUpdate() ?.data ?.mapNotNull {
(it as CommonMessage<MessageContent>).mapper()
2021-01-07 18:22:19 +00:00
} ?.let { return@expectFlow it }
}
it.asMessageUpdate() ?.data ?.asCommonMessage() ?.mapper().let(::listOfNotNull)
2021-01-06 17:06:48 +00:00
}.toList().toList()
2021-01-07 12:11:01 +00:00
private suspend inline fun <reified T : MessageContent> BehaviourContext.waitContent(
2021-01-06 17:06:48 +00:00
count: Int = 1,
initRequest: Request<*>? = null,
2021-01-07 18:22:19 +00:00
includeMediaGroups: Boolean = true,
2021-01-06 17:06:48 +00:00
noinline errorFactory: NullableRequestBuilder<*> = { null },
noinline filter: CommonMessageToContentMapper<T>? = null
) : List<T> = waitCommonMessage<T>(
2021-01-06 17:06:48 +00:00
count,
initRequest,
2021-01-07 18:22:19 +00:00
includeMediaGroups,
2021-01-06 17:06:48 +00:00
errorFactory
) {
if (content is T) {
@Suppress("UNCHECKED_CAST")
val message = (this as CommonMessage<T>)
2021-01-06 17:06:48 +00:00
if (filter == null) {
message.content
} else {
safelyWithoutExceptions { filter(message) }
2021-01-06 17:06:48 +00:00
}
} else {
null
}
}
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitContact(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<ContactContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitDice(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<DiceContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitGame(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<GameContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitLocation(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<LocationContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitPoll(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<PollContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitText(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<TextContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitVenue(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<VenueContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitAudioMediaGroup(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<AudioMediaGroupContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitDocumentMediaGroup(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<DocumentMediaGroupContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitMedia(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<MediaContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitMediaGroup(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<MediaGroupContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitVisualMediaGroup(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<VisualMediaGroupContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitAnimation(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<AnimationContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitAudio(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<AudioContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitDocument(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<DocumentContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitPhoto(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<PhotoContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitSticker(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<StickerContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitVideo(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
includeMediaGroups: Boolean = true,
filter: CommonMessageToContentMapper<VideoContent>? = null
) = waitContent(count, initRequest, includeMediaGroups, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitVideoNote(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<VideoNoteContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitVoice(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<VoiceContent>? = null
) = waitContent(count, initRequest, false, errorFactory, filter)
2021-01-07 12:11:01 +00:00
suspend fun BehaviourContext.waitInvoice(
2021-01-06 17:06:48 +00:00
initRequest: Request<*>? = null,
errorFactory: NullableRequestBuilder<*> = { null },
2021-01-08 04:29:29 +00:00
count: Int = 1,
filter: CommonMessageToContentMapper<InvoiceContent>? = null
2021-01-07 18:22:19 +00:00
) = waitContent(count, initRequest, false, errorFactory, filter)