InsanusMokrassar/CaptchaPlaguBotPlugin
1
0
Fork 0
mirror of https://github.com/InsanusMokrassar/CaptchaPlaguBotPlugin.git synced 2024-11-25 11:38:55 +00:00
Code Issues Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
InsanusMokrassar 2021-03-31 00:02:18 +06:00
parent 1d4baa8be9
commit 4a7339afd9
2 changed files with 43 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/kotlin/dev/inmo/plagubot/plugins/captcha/Plugin.kt
View File

@ -50,7 +50,7 @@ private const val enableSimpleCaptcha = "captcha_use_simple"
private const val enableExpressionCaptcha = "captcha_use_expression"
private val changeCaptchaMethodCommandRegex = Regex(
"captcha_use_(slot_machine)|(simple)|expression"
"captcha_use_((slot_machine)|(simple)|(expression))"
)
@Serializable
@ -108,22 +108,24 @@ class CaptchaBotPlugin : Plugin {
if (adminsAPI != null) {
onCommand(changeCaptchaMethodCommandRegex) {
val settings = it.chat.settings()
if (adminsAPI.sentByAdmin(it) != true) {
return@onCommand
}
val settings = it.chat.settings()
if (settings.autoRemoveCommands) {
safelyWithoutExceptions { deleteMessage(it) }
}
val commands = it.parseCommandsWithParams()
val changeCommand = commands.keys.first {
println(it)
changeCaptchaMethodCommandRegex.matches(it)
}
val captcha = when (changeCommand) {
enableSimpleCaptcha -> SimpleCaptchaProvider()
enableExpressionCaptcha -> ExpressionCaptchaProvider()
enableSlotMachineCaptcha -> SlotMachineCaptchaProvider()
println(changeCommand)
val captcha = when {
changeCommand.startsWith(enableSimpleCaptcha) -> SimpleCaptchaProvider()
changeCommand.startsWith(enableExpressionCaptcha) -> ExpressionCaptchaProvider()
changeCommand.startsWith(enableSlotMachineCaptcha) -> SlotMachineCaptchaProvider()
else -> return@onCommand
}
val newSettings = settings.copy(captchaProvider = captcha)

45
src/main/kotlin/dev/inmo/plagubot/plugins/captcha/provider/CaptchaProvider.kt
View File

@ -33,7 +33,6 @@ import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.channels.toList
import kotlinx.serialization.Serializable
import kotlinx.serialization.Transient
import kotlin.math.abs
import kotlin.random.Random
@Serializable
@ -230,7 +229,7 @@ private object ExpressionBuilder {
operations.forEach {
val rightOne = createNumber(max)
current = it.run { current.perform(rightOne) }
numbersString = " ${it.asString()} $numbersString"
numbersString += " ${it.asString()} $rightOne"
}
return current to numbersString
}
@ -240,9 +239,11 @@ private object ExpressionBuilder {
data class ExpressionCaptchaProvider(
val checkTimeSeconds: Seconds = 60,
val captchaText: String = "solve next captcha:",
val leftRetriesText: String = "Nope, left retries: ",
val maxPerNumber: Int = 10,
val operations: Int = 2,
val answers: Int = 6,
val attempts: Int = 3,
val kick: Boolean = true
) : CaptchaProvider() {
@Transient
@ -254,7 +255,7 @@ data class ExpressionCaptchaProvider(
newUsers: List<User>
) {
val userBanDateTime = eventDateTime + checkTimeSpan
newUsers.mapNotNull {
newUsers.mapNotNull { user ->
safelyWithoutExceptions {
launch {
doInSubContext {
@ -272,7 +273,7 @@ data class ExpressionCaptchaProvider(
val sentMessage = sendTextMessage(
chat,
buildEntities {
+it.mention(it.firstName)
+user.mention(user.firstName)
regular(", $captchaText ")
bold(callbackData.second)
},
@ -289,28 +290,44 @@ data class ExpressionCaptchaProvider(
}
}
var passed = true
val callback: suspend (Boolean) -> Unit = {
removeRedundantMessages()
passed = it
if (it) {
safelyWithoutExceptions { restrictChatMember(chat, user, permissions = LeftRestrictionsChatPermissions) }
} else {
if (kick) {
safelyWithoutExceptions { kickChatMember(chat, user) }
}
}
}
val job = parallel {
var leftAttempts = attempts
waitDataCallbackQuery {
if (it.id == user.id && this.data == correctAnswer) {
when {
this.user.id != user.id -> null
this.data != correctAnswer -> {
leftAttempts--
if (leftAttempts < 1) {
this
} else {
launch { answerCallbackQuery(this@waitDataCallbackQuery, leftRetriesText + leftAttempts) }
null
}
}
else -> this
}
}.first()
removeRedundantMessages()
safelyWithoutExceptions { restrictChatMember(chat, it, permissions = LeftRestrictionsChatPermissions) }
stop()
callback(leftAttempts > 0)
}
delay((userBanDateTime - eventDateTime).millisecondsLong)
if (job.isActive) {
job.cancel()
if (kick) {
safelyWithoutExceptions { kickChatMember(chat, it) }
}
}
if (job.isActive) job.cancel()
callback(passed)
}
}
}