mirror of
https://github.com/InsanusMokrassar/CaptchaPlaguBotPlugin.git
synced 2025-01-05 23:40:03 +00:00
fixes
This commit is contained in:
parent
7dead36cd9
commit
de6a8241c3
@ -1,33 +1,30 @@
|
|||||||
package dev.inmo.plagubot.plugins.captcha
|
package dev.inmo.plagubot.plugins.captcha
|
||||||
|
|
||||||
|
import dev.inmo.plagubot.plugins.captcha.provider.cancelData
|
||||||
import dev.inmo.tgbotapi.extensions.utils.SlotMachineReelImage
|
import dev.inmo.tgbotapi.extensions.utils.SlotMachineReelImage
|
||||||
import dev.inmo.tgbotapi.extensions.utils.types.buttons.*
|
import dev.inmo.tgbotapi.extensions.utils.types.buttons.*
|
||||||
|
import dev.inmo.tgbotapi.libraries.cache.admins.AdminsCacheAPI
|
||||||
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.CallbackDataInlineKeyboardButton
|
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.CallbackDataInlineKeyboardButton
|
||||||
|
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.InlineKeyboardButton
|
||||||
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardMarkup
|
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardMarkup
|
||||||
|
|
||||||
infix fun String.startingOf(target: String) = target.startsWith(this)
|
infix fun String.startingOf(target: String) = target.startsWith(this)
|
||||||
|
|
||||||
fun slotMachineReplyMarkup(
|
private val buttonsPreset: List<List<InlineKeyboardButton>> = SlotMachineReelImage.values().toList().chunked(2).map {
|
||||||
first: String? = null,
|
it.map {
|
||||||
second: String? = null,
|
CallbackDataInlineKeyboardButton(it.text, it.text)
|
||||||
third: String? = null,
|
}
|
||||||
): InlineKeyboardMarkup {
|
}
|
||||||
val texts = when {
|
|
||||||
first == null -> SlotMachineReelImage.values().map {
|
fun slotMachineReplyMarkup(
|
||||||
CallbackDataInlineKeyboardButton("${it.text}**", it.text)
|
adminCancelButton: Boolean = false
|
||||||
}
|
): InlineKeyboardMarkup {
|
||||||
second == null -> SlotMachineReelImage.values().map {
|
return inlineKeyboard {
|
||||||
CallbackDataInlineKeyboardButton("$first${it.text}*", it.text)
|
buttonsPreset.forEach(::add)
|
||||||
}
|
if (adminCancelButton) {
|
||||||
third == null -> SlotMachineReelImage.values().map {
|
row {
|
||||||
CallbackDataInlineKeyboardButton("$first$second${it.text}", it.text)
|
dataButton("Cancel (Admins only)", cancelData)
|
||||||
}
|
}
|
||||||
else -> listOf(CallbackDataInlineKeyboardButton("$first$second$third", "$first$second$third"))
|
}
|
||||||
}
|
|
||||||
return inlineKeyboard {
|
|
||||||
texts.chunked(2).forEach { add(it) }
|
|
||||||
// row {
|
|
||||||
// dataButton("Cancel (Admins only)", "cancel")
|
|
||||||
// }
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,12 +1,15 @@
|
|||||||
package dev.inmo.plagubot.plugins.captcha.provider
|
package dev.inmo.plagubot.plugins.captcha.provider
|
||||||
|
|
||||||
import com.benasher44.uuid.uuid4
|
import com.benasher44.uuid.uuid4
|
||||||
import com.soywiz.klock.DateTime
|
import com.soywiz.klock.*
|
||||||
import com.soywiz.klock.seconds
|
import dev.inmo.kslog.common.e
|
||||||
|
import dev.inmo.kslog.common.logger
|
||||||
import dev.inmo.micro_utils.coroutines.*
|
import dev.inmo.micro_utils.coroutines.*
|
||||||
import dev.inmo.plagubot.plugins.captcha.slotMachineReplyMarkup
|
import dev.inmo.plagubot.plugins.captcha.slotMachineReplyMarkup
|
||||||
|
import dev.inmo.tgbotapi.extensions.api.answers.answer
|
||||||
import dev.inmo.tgbotapi.extensions.api.answers.answerCallbackQuery
|
import dev.inmo.tgbotapi.extensions.api.answers.answerCallbackQuery
|
||||||
import dev.inmo.tgbotapi.extensions.api.chat.members.*
|
import dev.inmo.tgbotapi.extensions.api.chat.members.*
|
||||||
|
import dev.inmo.tgbotapi.extensions.api.delete
|
||||||
import dev.inmo.tgbotapi.extensions.api.deleteMessage
|
import dev.inmo.tgbotapi.extensions.api.deleteMessage
|
||||||
import dev.inmo.tgbotapi.extensions.api.edit.edit
|
import dev.inmo.tgbotapi.extensions.api.edit.edit
|
||||||
import dev.inmo.tgbotapi.extensions.api.edit.reply_markup.editMessageReplyMarkup
|
import dev.inmo.tgbotapi.extensions.api.edit.reply_markup.editMessageReplyMarkup
|
||||||
@ -15,7 +18,9 @@ import dev.inmo.tgbotapi.extensions.behaviour_builder.*
|
|||||||
import dev.inmo.tgbotapi.extensions.behaviour_builder.expectations.waitMessageDataCallbackQuery
|
import dev.inmo.tgbotapi.extensions.behaviour_builder.expectations.waitMessageDataCallbackQuery
|
||||||
import dev.inmo.tgbotapi.extensions.utils.asSlotMachineReelImage
|
import dev.inmo.tgbotapi.extensions.utils.asSlotMachineReelImage
|
||||||
import dev.inmo.tgbotapi.extensions.utils.calculateSlotMachineResult
|
import dev.inmo.tgbotapi.extensions.utils.calculateSlotMachineResult
|
||||||
|
import dev.inmo.tgbotapi.extensions.utils.extensions.sameMessage
|
||||||
import dev.inmo.tgbotapi.extensions.utils.shortcuts.executeUnsafe
|
import dev.inmo.tgbotapi.extensions.utils.shortcuts.executeUnsafe
|
||||||
|
import dev.inmo.tgbotapi.extensions.utils.shortcuts.sentMessages
|
||||||
import dev.inmo.tgbotapi.extensions.utils.types.buttons.*
|
import dev.inmo.tgbotapi.extensions.utils.types.buttons.*
|
||||||
import dev.inmo.tgbotapi.libraries.cache.admins.AdminsCacheAPI
|
import dev.inmo.tgbotapi.libraries.cache.admins.AdminsCacheAPI
|
||||||
import dev.inmo.tgbotapi.requests.DeleteMessage
|
import dev.inmo.tgbotapi.requests.DeleteMessage
|
||||||
@ -42,17 +47,100 @@ import kotlin.random.Random
|
|||||||
|
|
||||||
@Serializable
|
@Serializable
|
||||||
sealed class CaptchaProvider {
|
sealed class CaptchaProvider {
|
||||||
abstract suspend fun BehaviourContext.doAction(
|
abstract val checkTimeSpan: TimeSpan
|
||||||
|
|
||||||
|
interface CaptchaProviderWorker {
|
||||||
|
suspend fun BehaviourContext.doCaptcha(): Boolean
|
||||||
|
|
||||||
|
suspend fun BehaviourContext.onCloseCaptcha(passed: Boolean)
|
||||||
|
}
|
||||||
|
|
||||||
|
protected abstract suspend fun allocateWorker(
|
||||||
|
eventDateTime: DateTime,
|
||||||
|
chat: GroupChat,
|
||||||
|
user: User,
|
||||||
|
leftRestrictionsPermissions: ChatPermissions,
|
||||||
|
adminsApi: AdminsCacheAPI?,
|
||||||
|
kickOnUnsuccess: Boolean
|
||||||
|
): CaptchaProviderWorker
|
||||||
|
|
||||||
|
suspend fun BehaviourContext.doAction(
|
||||||
eventDateTime: DateTime,
|
eventDateTime: DateTime,
|
||||||
chat: GroupChat,
|
chat: GroupChat,
|
||||||
newUsers: List<User>,
|
newUsers: List<User>,
|
||||||
leftRestrictionsPermissions: ChatPermissions,
|
leftRestrictionsPermissions: ChatPermissions,
|
||||||
adminsApi: AdminsCacheAPI?,
|
adminsApi: AdminsCacheAPI?,
|
||||||
kickOnUnsuccess: Boolean
|
kickOnUnsuccess: Boolean
|
||||||
)
|
) {
|
||||||
|
val userBanDateTime = eventDateTime + checkTimeSpan
|
||||||
|
newUsers.map { user ->
|
||||||
|
launch {
|
||||||
|
createSubContextAndDoWithUpdatesFilter {
|
||||||
|
val worker = allocateWorker(
|
||||||
|
eventDateTime,
|
||||||
|
chat,
|
||||||
|
user,
|
||||||
|
leftRestrictionsPermissions,
|
||||||
|
adminsApi,
|
||||||
|
kickOnUnsuccess
|
||||||
|
)
|
||||||
|
val deferred = async {
|
||||||
|
runCatchingSafely {
|
||||||
|
with(worker) {
|
||||||
|
doCaptcha()
|
||||||
|
}
|
||||||
|
}.onFailure {
|
||||||
|
this@CaptchaProvider.logger.e("Unable to do captcha", it)
|
||||||
|
}.getOrElse { false }
|
||||||
|
}
|
||||||
|
|
||||||
|
val subscope = LinkedSupervisorScope()
|
||||||
|
subscope.launch {
|
||||||
|
delay((userBanDateTime - eventDateTime).millisecondsLong)
|
||||||
|
subscope.cancel()
|
||||||
|
}
|
||||||
|
subscope.launch {
|
||||||
|
deferred.await()
|
||||||
|
subscope.cancel()
|
||||||
|
}
|
||||||
|
|
||||||
|
subscope.coroutineContext.job.join()
|
||||||
|
|
||||||
|
val passed = runCatching {
|
||||||
|
deferred.getCompleted()
|
||||||
|
}.onFailure {
|
||||||
|
deferred.cancel()
|
||||||
|
}.getOrElse { false }
|
||||||
|
|
||||||
|
when {
|
||||||
|
passed -> {
|
||||||
|
safelyWithoutExceptions {
|
||||||
|
restrictChatMember(
|
||||||
|
chat,
|
||||||
|
user,
|
||||||
|
permissions = leftRestrictionsPermissions
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else -> {
|
||||||
|
send(chat, " ") {
|
||||||
|
+"User" + mention(user) + underline("didn't passed") + "captcha"
|
||||||
|
}
|
||||||
|
if (kickOnUnsuccess) {
|
||||||
|
banUser(chat, user, leftRestrictionsPermissions)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
with(worker) {
|
||||||
|
onCloseCaptcha(passed)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}.joinAll()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private const val cancelData = "cancel"
|
internal const val cancelData = "cancel"
|
||||||
|
|
||||||
private fun EntitiesBuilder.mention(user: User, defaultName: String = "User"): EntitiesBuilder {
|
private fun EntitiesBuilder.mention(user: User, defaultName: String = "User"): EntitiesBuilder {
|
||||||
return mention(
|
return mention(
|
||||||
@ -105,109 +193,94 @@ private suspend fun BehaviourContext.banUser(
|
|||||||
@Serializable
|
@Serializable
|
||||||
data class SlotMachineCaptchaProvider(
|
data class SlotMachineCaptchaProvider(
|
||||||
val checkTimeSeconds: Seconds = 300,
|
val checkTimeSeconds: Seconds = 300,
|
||||||
val captchaText: String = "solve this captcha: ",
|
val captchaText: String = "Solve this captcha: "
|
||||||
val kick: Boolean = true
|
|
||||||
) : CaptchaProvider() {
|
) : CaptchaProvider() {
|
||||||
@Transient
|
@Transient
|
||||||
private val checkTimeSpan = checkTimeSeconds.seconds
|
override val checkTimeSpan = checkTimeSeconds.seconds
|
||||||
|
|
||||||
override suspend fun BehaviourContext.doAction(
|
private inner class Worker(
|
||||||
|
private val chat: GroupChat,
|
||||||
|
private val user: User,
|
||||||
|
private val adminsApi: AdminsCacheAPI?
|
||||||
|
) : CaptchaProviderWorker {
|
||||||
|
private val messagesToDelete = mutableListOf<Message>()
|
||||||
|
|
||||||
|
override suspend fun BehaviourContext.doCaptcha(): Boolean {
|
||||||
|
val baseBuilder: EntitiesBuilderBody = {
|
||||||
|
mention(user)
|
||||||
|
regular(", $captchaText")
|
||||||
|
}
|
||||||
|
val sentMessage = send(
|
||||||
|
chat
|
||||||
|
) {
|
||||||
|
baseBuilder()
|
||||||
|
+": ✖✖✖"
|
||||||
|
}.also { messagesToDelete.add(it) }
|
||||||
|
val sentDice = sendDice(
|
||||||
|
sentMessage.chat,
|
||||||
|
SlotMachineDiceAnimationType,
|
||||||
|
replyToMessageId = sentMessage.messageId,
|
||||||
|
replyMarkup = slotMachineReplyMarkup(adminsApi != null)
|
||||||
|
).also { messagesToDelete.add(it) }
|
||||||
|
val reels = sentDice.content.dice.calculateSlotMachineResult()!!
|
||||||
|
val leftToClick = mutableListOf(
|
||||||
|
reels.left.asSlotMachineReelImage.text,
|
||||||
|
reels.center.asSlotMachineReelImage.text,
|
||||||
|
reels.right.asSlotMachineReelImage.text
|
||||||
|
)
|
||||||
|
val clicked = mutableListOf<String>()
|
||||||
|
fun buildTemplate() = "${clicked.joinToString("")}${leftToClick.joinToString("") { "✖" }}"
|
||||||
|
|
||||||
|
waitMessageDataCallbackQuery().filter {
|
||||||
|
when {
|
||||||
|
!it.message.sameMessage(sentDice) -> false
|
||||||
|
it.data == cancelData && adminsApi ?.isAdmin(chat.id, it.user.id) == true -> return@filter true
|
||||||
|
it.data == cancelData && adminsApi ?.isAdmin(chat.id, it.user.id) != true -> {
|
||||||
|
answer(
|
||||||
|
it,
|
||||||
|
"This button is only for admins"
|
||||||
|
)
|
||||||
|
false
|
||||||
|
}
|
||||||
|
it.user.id != user.id -> {
|
||||||
|
answer(it, "This button is not for you")
|
||||||
|
false
|
||||||
|
}
|
||||||
|
it.data != leftToClick.first() -> {
|
||||||
|
answer(it, "Nope")
|
||||||
|
false
|
||||||
|
}
|
||||||
|
else -> {
|
||||||
|
clicked.add(leftToClick.removeFirst())
|
||||||
|
answer(it, "Ok, next one")
|
||||||
|
edit(sentMessage) {
|
||||||
|
baseBuilder()
|
||||||
|
+": ${buildTemplate()}"
|
||||||
|
}
|
||||||
|
leftToClick.isEmpty()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}.first()
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
override suspend fun BehaviourContext.onCloseCaptcha(passed: Boolean) {
|
||||||
|
while (messagesToDelete.isNotEmpty()) {
|
||||||
|
runCatchingSafely { delete(messagesToDelete.removeFirst()) }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
override suspend fun allocateWorker(
|
||||||
eventDateTime: DateTime,
|
eventDateTime: DateTime,
|
||||||
chat: GroupChat,
|
chat: GroupChat,
|
||||||
newUsers: List<User>,
|
user: User,
|
||||||
leftRestrictionsPermissions: ChatPermissions,
|
leftRestrictionsPermissions: ChatPermissions,
|
||||||
adminsApi: AdminsCacheAPI?,
|
adminsApi: AdminsCacheAPI?,
|
||||||
kickOnUnsuccess: Boolean
|
kickOnUnsuccess: Boolean
|
||||||
) {
|
): CaptchaProviderWorker = Worker(chat, user, adminsApi)
|
||||||
val userBanDateTime = eventDateTime + checkTimeSpan
|
|
||||||
val authorized = Channel<User>(newUsers.size)
|
|
||||||
val messagesToDelete = Channel<Message>(Channel.UNLIMITED)
|
|
||||||
val subContexts = newUsers.map { user ->
|
|
||||||
createSubContextAndDoWithUpdatesFilter(stopOnCompletion = false) {
|
|
||||||
val sentMessage = send(
|
|
||||||
chat
|
|
||||||
) {
|
|
||||||
mention(user)
|
|
||||||
regular(", $captchaText")
|
|
||||||
}.also { messagesToDelete.send(it) }
|
|
||||||
val sentDice = sendDice(
|
|
||||||
sentMessage.chat,
|
|
||||||
SlotMachineDiceAnimationType,
|
|
||||||
replyToMessageId = sentMessage.messageId,
|
|
||||||
replyMarkup = slotMachineReplyMarkup()
|
|
||||||
).also { messagesToDelete.send(it) }
|
|
||||||
val reels = sentDice.content.dice.calculateSlotMachineResult()!!
|
|
||||||
val leftToClick = mutableListOf(
|
|
||||||
reels.left.asSlotMachineReelImage.text,
|
|
||||||
reels.center.asSlotMachineReelImage.text,
|
|
||||||
reels.right.asSlotMachineReelImage.text
|
|
||||||
)
|
|
||||||
|
|
||||||
launch {
|
|
||||||
val clicked = arrayOf<String?>(null, null, null)
|
|
||||||
while (leftToClick.isNotEmpty()) {
|
|
||||||
val userClicked =
|
|
||||||
waitMessageDataCallbackQuery().filter { it.user.id == user.id && it.message.messageId == sentDice.messageId }
|
|
||||||
.first()
|
|
||||||
|
|
||||||
when {
|
|
||||||
userClicked.data == leftToClick.first() -> {
|
|
||||||
clicked[3 - leftToClick.size] = leftToClick.removeAt(0)
|
|
||||||
if (clicked.contains(null)) {
|
|
||||||
safelyWithoutExceptions { answerCallbackQuery(userClicked, "Ok, next one") }
|
|
||||||
editMessageReplyMarkup(
|
|
||||||
sentDice,
|
|
||||||
slotMachineReplyMarkup(clicked[0], clicked[1], clicked[2])
|
|
||||||
)
|
|
||||||
} else {
|
|
||||||
safelyWithoutExceptions {
|
|
||||||
answerCallbackQuery(
|
|
||||||
userClicked,
|
|
||||||
"Thank you and welcome",
|
|
||||||
showAlert = true
|
|
||||||
)
|
|
||||||
}
|
|
||||||
safelyWithoutExceptions { deleteMessage(sentMessage) }
|
|
||||||
safelyWithoutExceptions { deleteMessage(sentDice) }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
else -> safelyWithoutExceptions { answerCallbackQuery(userClicked, "Nope") }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
authorized.send(user)
|
|
||||||
safelyWithoutExceptions {
|
|
||||||
restrictChatMember(
|
|
||||||
chat,
|
|
||||||
user,
|
|
||||||
permissions = leftRestrictionsPermissions
|
|
||||||
)
|
|
||||||
}
|
|
||||||
stop()
|
|
||||||
}
|
|
||||||
|
|
||||||
this to user
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
delay((userBanDateTime - eventDateTime).millisecondsLong)
|
|
||||||
|
|
||||||
authorized.close()
|
|
||||||
val authorizedUsers = authorized.toList()
|
|
||||||
|
|
||||||
subContexts.forEach { (context, user) ->
|
|
||||||
if (user !in authorizedUsers) {
|
|
||||||
context.stop()
|
|
||||||
if (kickOnUnsuccess) {
|
|
||||||
banUser(chat, user, leftRestrictionsPermissions)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
messagesToDelete.close()
|
|
||||||
for (message in messagesToDelete) {
|
|
||||||
executeUnsafe(DeleteMessage(message.chat.id, message.messageId), retries = 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Serializable
|
@Serializable
|
||||||
@ -217,86 +290,78 @@ data class SimpleCaptchaProvider(
|
|||||||
val buttonText: String = "Press me\uD83D\uDE0A"
|
val buttonText: String = "Press me\uD83D\uDE0A"
|
||||||
) : CaptchaProvider() {
|
) : CaptchaProvider() {
|
||||||
@Transient
|
@Transient
|
||||||
private val checkTimeSpan = checkTimeSeconds.seconds
|
override val checkTimeSpan = checkTimeSeconds.seconds
|
||||||
|
|
||||||
override suspend fun BehaviourContext.doAction(
|
private inner class Worker(
|
||||||
|
private val chat: GroupChat,
|
||||||
|
private val user: User,
|
||||||
|
private val adminsApi: AdminsCacheAPI?
|
||||||
|
) : CaptchaProviderWorker {
|
||||||
|
private var sentMessage: Message? = null
|
||||||
|
override suspend fun BehaviourContext.doCaptcha(): Boolean {
|
||||||
|
val callbackData = uuid4().toString()
|
||||||
|
val sentMessage = send(
|
||||||
|
chat,
|
||||||
|
replyMarkup = inlineKeyboard {
|
||||||
|
row {
|
||||||
|
dataButton(buttonText, callbackData)
|
||||||
|
}
|
||||||
|
if (adminsApi != null) {
|
||||||
|
row {
|
||||||
|
dataButton("Cancel (Admins only)", cancelData)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
) {
|
||||||
|
mention(user)
|
||||||
|
regular(", $captchaText")
|
||||||
|
}
|
||||||
|
this@Worker.sentMessage = sentMessage
|
||||||
|
|
||||||
|
val pushed = waitMessageDataCallbackQuery().filter {
|
||||||
|
when {
|
||||||
|
!it.message.sameMessage(sentMessage) -> false
|
||||||
|
it.data == callbackData && it.user.id == user.id -> true
|
||||||
|
it.data == cancelData && (adminsApi ?.isAdmin(chat.id, it.user.id) == true) -> true
|
||||||
|
it.data == callbackData -> {
|
||||||
|
answer(it, "This button is not for you")
|
||||||
|
false
|
||||||
|
}
|
||||||
|
it.data == cancelData -> {
|
||||||
|
answer(it, "This button is for admins only")
|
||||||
|
false
|
||||||
|
}
|
||||||
|
else -> false
|
||||||
|
}
|
||||||
|
}.first()
|
||||||
|
|
||||||
|
answer(
|
||||||
|
pushed,
|
||||||
|
when (pushed.data) {
|
||||||
|
cancelData -> "You have cancelled captcha"
|
||||||
|
else -> "Ok, thanks"
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
override suspend fun BehaviourContext.onCloseCaptcha(passed: Boolean) {
|
||||||
|
sentMessage ?.let {
|
||||||
|
delete(it)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
override suspend fun allocateWorker(
|
||||||
eventDateTime: DateTime,
|
eventDateTime: DateTime,
|
||||||
chat: GroupChat,
|
chat: GroupChat,
|
||||||
newUsers: List<User>,
|
user: User,
|
||||||
leftRestrictionsPermissions: ChatPermissions,
|
leftRestrictionsPermissions: ChatPermissions,
|
||||||
adminsApi: AdminsCacheAPI?,
|
adminsApi: AdminsCacheAPI?,
|
||||||
kickOnUnsuccess: Boolean
|
kickOnUnsuccess: Boolean
|
||||||
) {
|
): CaptchaProviderWorker = Worker(chat, user, adminsApi)
|
||||||
val userBanDateTime = eventDateTime + checkTimeSpan
|
|
||||||
newUsers.map { user ->
|
|
||||||
launchSafelyWithoutExceptions {
|
|
||||||
createSubContext(this).doInContext(stopOnCompletion = false) {
|
|
||||||
val callbackData = uuid4().toString()
|
|
||||||
val sentMessage = send(
|
|
||||||
chat,
|
|
||||||
replyMarkup = inlineKeyboard {
|
|
||||||
row {
|
|
||||||
dataButton(buttonText, callbackData)
|
|
||||||
}
|
|
||||||
if (adminsApi != null) {
|
|
||||||
row {
|
|
||||||
dataButton("Cancel (Admins only)", cancelData)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
) {
|
|
||||||
mention(user)
|
|
||||||
regular(", $captchaText")
|
|
||||||
}
|
|
||||||
|
|
||||||
suspend fun removeRedundantMessages() {
|
|
||||||
safelyWithoutExceptions {
|
|
||||||
deleteMessage(sentMessage)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
val job = launchSafely {
|
|
||||||
waitMessageDataCallbackQuery().filter { query ->
|
|
||||||
val baseCheck = query.message.messageId == sentMessage.messageId
|
|
||||||
val userAnswered = query.user.id == user.id && query.data == callbackData
|
|
||||||
val adminCanceled = (query.data == cancelData && (adminsApi?.isAdmin(
|
|
||||||
sentMessage.chat.id,
|
|
||||||
query.user.id
|
|
||||||
)) == true)
|
|
||||||
if (baseCheck && adminCanceled) {
|
|
||||||
sendAdminCanceledMessage(
|
|
||||||
sentMessage.chat,
|
|
||||||
user,
|
|
||||||
query.user
|
|
||||||
)
|
|
||||||
}
|
|
||||||
baseCheck && (adminCanceled || userAnswered)
|
|
||||||
}.first()
|
|
||||||
|
|
||||||
removeRedundantMessages()
|
|
||||||
safelyWithoutExceptions {
|
|
||||||
restrictChatMember(
|
|
||||||
chat,
|
|
||||||
user,
|
|
||||||
permissions = leftRestrictionsPermissions
|
|
||||||
)
|
|
||||||
}
|
|
||||||
stop()
|
|
||||||
}
|
|
||||||
|
|
||||||
delay((userBanDateTime - eventDateTime).millisecondsLong)
|
|
||||||
|
|
||||||
if (job.isActive) {
|
|
||||||
job.cancel()
|
|
||||||
if (kickOnUnsuccess) {
|
|
||||||
banUser(chat, user, leftRestrictionsPermissions)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
stop()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}.joinAll()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private object ExpressionBuilder {
|
private object ExpressionBuilder {
|
||||||
@ -355,130 +420,85 @@ data class ExpressionCaptchaProvider(
|
|||||||
val attempts: Int = 3
|
val attempts: Int = 3
|
||||||
) : CaptchaProvider() {
|
) : CaptchaProvider() {
|
||||||
@Transient
|
@Transient
|
||||||
private val checkTimeSpan = checkTimeSeconds.seconds
|
override val checkTimeSpan = checkTimeSeconds.seconds
|
||||||
|
|
||||||
override suspend fun BehaviourContext.doAction(
|
private inner class Worker(
|
||||||
|
private val chat: GroupChat,
|
||||||
|
private val user: User,
|
||||||
|
private val adminsApi: AdminsCacheAPI?
|
||||||
|
) : CaptchaProviderWorker {
|
||||||
|
private var sentMessage: Message? = null
|
||||||
|
override suspend fun BehaviourContext.doCaptcha(): Boolean {
|
||||||
|
val callbackData = ExpressionBuilder.createExpression(
|
||||||
|
maxPerNumber,
|
||||||
|
operations
|
||||||
|
)
|
||||||
|
val correctAnswer = callbackData.first.toString()
|
||||||
|
val answers = (0 until answers - 1).map {
|
||||||
|
ExpressionBuilder.generateResult(maxPerNumber, operations)
|
||||||
|
}.toMutableList().also { orderedAnswers ->
|
||||||
|
val correctAnswerPosition = Random.nextInt(orderedAnswers.size)
|
||||||
|
orderedAnswers.add(correctAnswerPosition, callbackData.first)
|
||||||
|
}.toList()
|
||||||
|
val sentMessage = send(
|
||||||
|
chat,
|
||||||
|
replyMarkup = inlineKeyboard {
|
||||||
|
answers.map {
|
||||||
|
CallbackDataInlineKeyboardButton(it.toString(), it.toString())
|
||||||
|
}.chunked(3).forEach(::add)
|
||||||
|
if (adminsApi != null) {
|
||||||
|
row {
|
||||||
|
dataButton("Cancel (Admins only)", cancelData)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
) {
|
||||||
|
mention(user)
|
||||||
|
regular(", $captchaText ")
|
||||||
|
bold(callbackData.second)
|
||||||
|
}
|
||||||
|
|
||||||
|
var leftAttempts = attempts
|
||||||
|
return waitMessageDataCallbackQuery().takeWhile { leftAttempts > 0 }.map { query ->
|
||||||
|
val baseCheck = query.message.messageId == sentMessage.messageId
|
||||||
|
val dataCorrect = (query.user.id == user.id && query.data == correctAnswer)
|
||||||
|
val adminCanceled = (query.data == cancelData && (adminsApi?.isAdmin(
|
||||||
|
sentMessage.chat.id,
|
||||||
|
query.user.id
|
||||||
|
)) == true)
|
||||||
|
baseCheck && if (dataCorrect || adminCanceled) {
|
||||||
|
if (adminCanceled) {
|
||||||
|
sendAdminCanceledMessage(
|
||||||
|
sentMessage.chat,
|
||||||
|
user,
|
||||||
|
query.user
|
||||||
|
)
|
||||||
|
}
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
leftAttempts--
|
||||||
|
if (leftAttempts > 0) {
|
||||||
|
answerCallbackQuery(query, leftRetriesText + leftAttempts)
|
||||||
|
}
|
||||||
|
false
|
||||||
|
}
|
||||||
|
}.firstOrNull() ?: false
|
||||||
|
}
|
||||||
|
|
||||||
|
override suspend fun BehaviourContext.onCloseCaptcha(passed: Boolean) {
|
||||||
|
sentMessage ?.let {
|
||||||
|
delete(it)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
override suspend fun allocateWorker(
|
||||||
eventDateTime: DateTime,
|
eventDateTime: DateTime,
|
||||||
chat: GroupChat,
|
chat: GroupChat,
|
||||||
newUsers: List<User>,
|
user: User,
|
||||||
leftRestrictionsPermissions: ChatPermissions,
|
leftRestrictionsPermissions: ChatPermissions,
|
||||||
adminsApi: AdminsCacheAPI?,
|
adminsApi: AdminsCacheAPI?,
|
||||||
kickOnUnsuccess: Boolean
|
kickOnUnsuccess: Boolean
|
||||||
) {
|
): CaptchaProviderWorker = Worker(chat, user, adminsApi)
|
||||||
val userBanDateTime = eventDateTime + checkTimeSpan
|
|
||||||
newUsers.map { user ->
|
|
||||||
launch {
|
|
||||||
createSubContextAndDoWithUpdatesFilter {
|
|
||||||
val callbackData = ExpressionBuilder.createExpression(
|
|
||||||
maxPerNumber,
|
|
||||||
operations
|
|
||||||
)
|
|
||||||
val correctAnswer = callbackData.first.toString()
|
|
||||||
val answers = (0 until answers - 1).map {
|
|
||||||
ExpressionBuilder.generateResult(maxPerNumber, operations)
|
|
||||||
}.toMutableList().also { orderedAnswers ->
|
|
||||||
val correctAnswerPosition = Random.nextInt(orderedAnswers.size)
|
|
||||||
orderedAnswers.add(correctAnswerPosition, callbackData.first)
|
|
||||||
}.toList()
|
|
||||||
val sentMessage = send(
|
|
||||||
chat,
|
|
||||||
replyMarkup = inlineKeyboard {
|
|
||||||
answers.map {
|
|
||||||
CallbackDataInlineKeyboardButton(it.toString(), it.toString())
|
|
||||||
}.chunked(3).forEach(::add)
|
|
||||||
if (adminsApi != null) {
|
|
||||||
row {
|
|
||||||
dataButton("Cancel (Admins only)", cancelData)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
) {
|
|
||||||
mention(user)
|
|
||||||
regular(", $captchaText ")
|
|
||||||
bold(callbackData.second)
|
|
||||||
}
|
|
||||||
|
|
||||||
suspend fun removeRedundantMessages(removeSentMessage: Boolean = true) {
|
|
||||||
safelyWithoutExceptions {
|
|
||||||
if (removeSentMessage) {
|
|
||||||
deleteMessage(sentMessage)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var passed: Boolean? = null
|
|
||||||
val passedMutex = Mutex()
|
|
||||||
val callback: suspend (Boolean) -> Unit = {
|
|
||||||
passedMutex.withLock {
|
|
||||||
if (passed == null) {
|
|
||||||
passed = it
|
|
||||||
runCatchingSafely<Unit> {
|
|
||||||
when {
|
|
||||||
it -> {
|
|
||||||
removeRedundantMessages()
|
|
||||||
safelyWithoutExceptions {
|
|
||||||
restrictChatMember(
|
|
||||||
chat,
|
|
||||||
user,
|
|
||||||
permissions = leftRestrictionsPermissions
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else -> {
|
|
||||||
removeRedundantMessages(removeSentMessage = false)
|
|
||||||
edit(sentMessage) {
|
|
||||||
+"User " + mention(user) + underline("didn't passed") + "captcha"
|
|
||||||
}
|
|
||||||
if (kickOnUnsuccess) {
|
|
||||||
banUser(chat, user, leftRestrictionsPermissions)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
val banJob = launch {
|
|
||||||
delay((userBanDateTime - eventDateTime).millisecondsLong)
|
|
||||||
|
|
||||||
if (passed == null) {
|
|
||||||
callback(false)
|
|
||||||
stop()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var leftAttempts = attempts
|
|
||||||
waitMessageDataCallbackQuery().takeWhile { leftAttempts > 0 }.filter { query ->
|
|
||||||
val baseCheck = query.message.messageId == sentMessage.messageId
|
|
||||||
val dataCorrect = (query.user.id == user.id && query.data == correctAnswer)
|
|
||||||
val adminCanceled = (query.data == cancelData && (adminsApi?.isAdmin(
|
|
||||||
sentMessage.chat.id,
|
|
||||||
query.user.id
|
|
||||||
)) == true)
|
|
||||||
baseCheck && if (dataCorrect || adminCanceled) {
|
|
||||||
banJob.cancel()
|
|
||||||
if (adminCanceled) {
|
|
||||||
sendAdminCanceledMessage(
|
|
||||||
sentMessage.chat,
|
|
||||||
user,
|
|
||||||
query.user
|
|
||||||
)
|
|
||||||
}
|
|
||||||
true
|
|
||||||
} else {
|
|
||||||
leftAttempts--
|
|
||||||
if (leftAttempts > 0) {
|
|
||||||
answerCallbackQuery(query, leftRetriesText + leftAttempts)
|
|
||||||
}
|
|
||||||
false
|
|
||||||
}
|
|
||||||
}.firstOrNull()
|
|
||||||
|
|
||||||
callback(leftAttempts > 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}.joinAll()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user