WIP

JavaBy · Nov 19, 2022 · 79d756d · 79d756d
1 parent 7d63c59
commit 79d756d
Show file tree

Hide file tree

Showing 6 changed files with 62 additions and 6 deletions.
diff --git a/launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/pipeline.kt b/launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/pipeline.kt
@@ -186,6 +186,7 @@ val pipelineModule = module {
         PinsPreCheckoutQueryUpdateProcessor(
             bot = get(),
             json = get(),
+            moniesDAO = get(),
         )
     }
 

diff --git a/shop/build.gradle.kts b/shop/build.gradle.kts
@@ -4,11 +4,10 @@ plugins {
 
 dependencies {
     api(project.projects.core)
-    api(libs.tgbotapi.core)
+    api(libs.tgbotapi)
     implementation(project.projects.shop.provider)
     implementation(project.projects.shop.payload)
-    implementation(libs.tgbotapi.extensions.api)
-    implementation(libs.tgbotapi.extensions.utils)
+    implementation(project.projects.monies)
     implementation(libs.log4j.api)
 
     testImplementation(libs.junit.jupiter.api)

diff --git a/shop/src/main/kotlin/by/jprof/telegram/bot/shop/PinsPreCheckoutQueryUpdateProcessor.kt b/shop/src/main/kotlin/by/jprof/telegram/bot/shop/PinsPreCheckoutQueryUpdateProcessor.kt
@@ -1,9 +1,13 @@
 package by.jprof.telegram.bot.shop
 
 import by.jprof.telegram.bot.core.UpdateProcessor
+import by.jprof.telegram.bot.monies.dao.MoniesDAO
+import by.jprof.telegram.bot.monies.model.Monies
 import by.jprof.telegram.bot.shop.payload.PinsPayload
+import by.jprof.telegram.bot.shop.utils.tooManyPins
 import dev.inmo.tgbotapi.bot.RequestsExecutor
 import dev.inmo.tgbotapi.extensions.api.answers.payments.answerPreCheckoutQueryError
+import dev.inmo.tgbotapi.extensions.api.answers.payments.answerPreCheckoutQueryOk
 import dev.inmo.tgbotapi.extensions.utils.asPreCheckoutQueryUpdate
 import dev.inmo.tgbotapi.types.update.abstracts.Update
 import dev.inmo.tgbotapi.utils.PreviewFeature
@@ -15,6 +19,7 @@ import org.apache.logging.log4j.LogManager
 class PinsPreCheckoutQueryUpdateProcessor(
     private val bot: RequestsExecutor,
     private val json: Json,
+    private val moniesDAO: MoniesDAO,
 ) : UpdateProcessor {
     companion object {
         private val logger = LogManager.getLogger(PinsPreCheckoutQueryUpdateProcessor::class.java)!!
@@ -33,6 +38,13 @@ class PinsPreCheckoutQueryUpdateProcessor(
 
         logger.debug(payload)
 
-        bot.answerPreCheckoutQueryError(preCheckoutQuery, "Not implemented yet")
+        val monies = moniesDAO.get(preCheckoutQuery.user.id.chatId, payload.chat) ?: Monies(preCheckoutQuery.user.id.chatId, payload.chat)
+        val pins = monies.pins ?: 0
+
+        if (pins > 9999) {
+            bot.answerPreCheckoutQueryError(preCheckoutQuery, tooManyPins())
+        }
+
+        bot.answerPreCheckoutQueryOk(preCheckoutQuery)
     }
 }
diff --git a/shop/src/main/kotlin/by/jprof/telegram/bot/shop/SuccessfulPaymentUpdateProcessor.kt b/shop/src/main/kotlin/by/jprof/telegram/bot/shop/SuccessfulPaymentUpdateProcessor.kt
@@ -0,0 +1,35 @@
+package by.jprof.telegram.bot.shop
+
+import by.jprof.telegram.bot.core.UpdateProcessor
+import by.jprof.telegram.bot.shop.payload.SupportPayload
+import dev.inmo.tgbotapi.extensions.api.answers.payments.answerPreCheckoutQueryOk
+import dev.inmo.tgbotapi.extensions.utils.asBaseMessageUpdate
+import dev.inmo.tgbotapi.extensions.utils.asChatEventMessage
+import dev.inmo.tgbotapi.extensions.utils.asMessageUpdate
+import dev.inmo.tgbotapi.extensions.utils.asNewChatMembers
+import dev.inmo.tgbotapi.extensions.utils.asPreCheckoutQueryUpdate
+import dev.inmo.tgbotapi.extensions.utils.asVoiceChatParticipantsInvited
+import dev.inmo.tgbotapi.types.update.abstracts.Update
+import dev.inmo.tgbotapi.utils.PreviewFeature
+import kotlinx.serialization.decodeFromString
+import org.apache.logging.log4j.LogManager
+
+@OptIn(PreviewFeature::class)
+class SuccessfulPaymentUpdateProcessor() : UpdateProcessor {
+    override suspend fun process(update: Update) {
+        val a = update.asMessageUpdate().data.asChatEventMessage().chatEvent.asSuccessfulPaymentEvent
+        val preCheckoutQuery = update.asPreCheckoutQueryUpdate()?.data ?: return
+
+        logger.debug(preCheckoutQuery)
+
+        val payload = try {
+            json.decodeFromString<SupportPayload>(preCheckoutQuery.invoicePayload)
+        } catch (_: Exception) {
+            return
+        }
+
+        logger.debug(payload)
+
+        bot.answerPreCheckoutQueryOk(preCheckoutQuery)
+    }
+}
diff --git a/shop/src/main/kotlin/by/jprof/telegram/bot/shop/SupportPreCheckoutQueryUpdateProcessor.kt b/shop/src/main/kotlin/by/jprof/telegram/bot/shop/SupportPreCheckoutQueryUpdateProcessor.kt
@@ -3,7 +3,7 @@ package by.jprof.telegram.bot.shop
 import by.jprof.telegram.bot.core.UpdateProcessor
 import by.jprof.telegram.bot.shop.payload.SupportPayload
 import dev.inmo.tgbotapi.bot.RequestsExecutor
-import dev.inmo.tgbotapi.extensions.api.answers.payments.answerPreCheckoutQueryError
+import dev.inmo.tgbotapi.extensions.api.answers.payments.answerPreCheckoutQueryOk
 import dev.inmo.tgbotapi.extensions.utils.asPreCheckoutQueryUpdate
 import dev.inmo.tgbotapi.types.update.abstracts.Update
 import dev.inmo.tgbotapi.utils.PreviewFeature
@@ -33,6 +33,6 @@ class SupportPreCheckoutQueryUpdateProcessor(
 
         logger.debug(payload)
 
-        bot.answerPreCheckoutQueryError(preCheckoutQuery, "Not implemented yet")
+        bot.answerPreCheckoutQueryOk(preCheckoutQuery)
     }
 }
diff --git a/shop/src/main/kotlin/by/jprof/telegram/bot/shop/utils/messages.kt b/shop/src/main/kotlin/by/jprof/telegram/bot/shop/utils/messages.kt
@@ -15,3 +15,12 @@ private val forwardedPaymentsAreNotSupportedMessages = listOf(
 internal fun forwardedPaymentsAreNotSupported(): String {
     return forwardedPaymentsAreNotSupportedMessages.random()
 }
+
+private val tooManyPinsMessages = listOf(
+    "У тебя и так хватает пинов!",
+    "У тебя и так много пинов!",
+)
+
+internal fun tooManyPins(): String {
+    return tooManyPinsMessages.random()
+}