65 lines
3.0 KiB
Kotlin
65 lines
3.0 KiB
Kotlin
@file:Suppress("unused")
|
|
|
|
package dev.inmo.tgbotapi.extensions.behaviour_builder.expectations
|
|
|
|
import dev.inmo.tgbotapi.extensions.behaviour_builder.BehaviourContext
|
|
import dev.inmo.tgbotapi.extensions.behaviour_builder.utils.SimpleFilter
|
|
import dev.inmo.tgbotapi.extensions.utils.callbackQueryUpdateOrNull
|
|
import dev.inmo.tgbotapi.requests.abstracts.Request
|
|
import dev.inmo.tgbotapi.types.queries.callback.*
|
|
import dev.inmo.tgbotapi.utils.RiskFeature
|
|
import dev.inmo.tgbotapi.utils.lowLevelRiskFeatureMessage
|
|
import kotlinx.coroutines.flow.Flow
|
|
import kotlinx.coroutines.flow.toList
|
|
|
|
typealias CallbackQueryMapper<T> = suspend T.() -> T?
|
|
|
|
@RiskFeature(lowLevelRiskFeatureMessage)
|
|
suspend inline fun <reified O> BehaviourContext.waitCallbackQueries(
|
|
initRequest: Request<*>? = null,
|
|
noinline errorFactory: NullableRequestBuilder<*> = { null }
|
|
): Flow<O> = expectFlow(
|
|
initRequest,
|
|
errorFactory
|
|
) {
|
|
(it.callbackQueryUpdateOrNull() ?.data as O).let(::listOfNotNull)
|
|
}
|
|
|
|
|
|
suspend fun BehaviourContext.waitDataCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<DataCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitGameShortNameCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<GameShortNameCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitInlineMessageIdCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<InlineMessageIdCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitInlineMessageIdDataCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<InlineMessageIdDataCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitInlineMessageIdGameShortNameCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<InlineMessageIdGameShortNameCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitMessageCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<MessageCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitMessageDataCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<MessageDataCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitMessageGameShortNameCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<MessageGameShortNameCallbackQuery>(initRequest, errorFactory)
|
|
suspend fun BehaviourContext.waitUnknownCallbackQuery(
|
|
initRequest: Request<*>? = null,
|
|
errorFactory: NullableRequestBuilder<*> = { null }
|
|
) = waitCallbackQueries<UnknownCallbackQueryType>(initRequest, errorFactory)
|