Add shop module

Author: madhead

.deploy/lambda/lib/JProfByBotStack.ts

@@ -3,6 +3,8 @@ import {Construct} from 'constructs';
 import * as apigateway from 'aws-cdk-lib/aws-apigateway';
 import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
 import * as lambda from 'aws-cdk-lib/aws-lambda';
+import {Architecture} from 'aws-cdk-lib/aws-lambda';
+import * as secrets from 'aws-cdk-lib/aws-secretsmanager';
 import * as sfn from 'aws-cdk-lib/aws-stepfunctions';
 import * as tasks from 'aws-cdk-lib/aws-stepfunctions-tasks';
 import * as ses from 'aws-cdk-lib/aws-ses';
@@ -13,6 +15,8 @@ export class JProfByBotStack extends cdk.Stack {
     constructor(scope: Construct, id: string, props: JProfByBotStackProps) {
         super(scope, id, props);
 
+        const secretPaymentProviderTokens = new secrets.Secret(this, 'jprof-by-bot-secret-payment-provider-tokens');
+
         const votesTable = new dynamodb.Table(this, 'jprof-by-bot-table-votes', {
             tableName: 'jprof-by-bot-table-votes',
             partitionKey: {name: 'id', type: dynamodb.AttributeType.STRING},
@@ -123,13 +127,20 @@ export class JProfByBotStack extends cdk.Stack {
         });
 
         const layerLibGL = new lambda.LayerVersion(this, 'jprof-by-bot-lambda-layer-libGL', {
+            layerVersionName: 'libGL',
             code: lambda.Code.fromAsset('layers/libGL.zip'),
-            compatibleRuntimes: [lambda.Runtime.JAVA_11],
+            compatibleArchitectures: [Architecture.ARM_64],
         });
         const layerLibfontconfig = new lambda.LayerVersion(this, 'jprof-by-bot-lambda-layer-libfontconfig', {
+            layerVersionName: 'libfontconfig',
             code: lambda.Code.fromAsset('layers/libfontconfig.zip'),
-            compatibleRuntimes: [lambda.Runtime.JAVA_11],
+            compatibleArchitectures: [Architecture.ARM_64],
         });
+        const layerParametersAndSecretsLambdaExtension = lambda.LayerVersion.fromLayerVersionArn(
+            this,
+            'jprof-by-bot-lambda-layer-parametersAndSecretsLambdaExtension',
+            'arn:aws:lambda:us-east-1:177933569100:layer:AWS-Parameters-and-Secrets-Lambda-Extension:2'
+        )
 
         const lambdaWebhookTimeout = cdk.Duration.seconds(29);
         const lambdaWebhook = new lambda.Function(this, 'jprof-by-bot-lambda-webhook', {
@@ -138,6 +149,7 @@ export class JProfByBotStack extends cdk.Stack {
             layers: [
                 layerLibGL,
                 layerLibfontconfig,
+                layerParametersAndSecretsLambdaExtension,
             ],
             timeout: lambdaWebhookTimeout,
             maxEventAge: cdk.Duration.minutes(5),
@@ -198,6 +210,8 @@ export class JProfByBotStack extends cdk.Stack {
             ],
         });
 
+        secretPaymentProviderTokens.grantRead(lambdaWebhook);
+
         votesTable.grantReadWriteData(lambdaWebhook);
 
         youtubeChannelsWhitelistTable.grantReadData(lambdaWebhook);

README.adoc

@@ -12,7 +12,9 @@ Official Telegram bot of Java Professionals BY community.
 * Converts some currencies to EUR and USD
 * Posts scheduled messages from this repo's `posts` branch
 * Expands LeetCode links
+* Tells users' local times
 * Regulates our English Rooms & teaches us new English words
+* Sells pins and custom statuses 🤑
 
 So, it just brings some fun and interactivity in our chat.
 

gradle/libs.versions.toml

@@ -37,6 +37,7 @@ aws-lambda-java-events = { group = "com.amazonaws", name = "aws-lambda-java-even
 aws-lambda-java-core = { group = "com.amazonaws", name = "aws-lambda-java-core", version.ref = "aws-lambda-java-core" }
 aws-lambda-java-log4j2 = { group = "com.amazonaws", name = "aws-lambda-java-log4j2", version.ref = "aws-lambda-java-log4j2" }
 dynamodb = { group = "software.amazon.awssdk", name = "dynamodb", version.ref = "awssdk" }
+secretsmanager = { group = "software.amazon.awssdk", name = "secretsmanager", version.ref = "awssdk" }
 sfn = { group = "software.amazon.awssdk", name = "sfn", version.ref = "awssdk" }
 
 koin-core = { group = "io.insert-koin", name = "koin-core", version.ref = "koin" }

launchers/lambda/build.gradle.kts

@@ -5,6 +5,7 @@ plugins {
 
 dependencies {
     implementation(libs.bundles.aws.lambda)
+    implementation(libs.secretsmanager)
     implementation(libs.koin.core)
     implementation(libs.tgbotapi)
     implementation(libs.bundles.log4j)
@@ -28,4 +29,6 @@ dependencies {
     implementation(project.projects.english.urbanWordOfTheDay.dynamodb)
     implementation(project.projects.english.urbanDictionary)
     implementation(project.projects.english.dictionaryapiDev)
+    implementation(project.projects.shop.provider)
+    implementation(project.projects.shop)
 }

launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/JProf.kt

@@ -7,6 +7,7 @@ import by.jprof.telegram.bot.launchers.lambda.config.dictionaryApiDevModule
 import by.jprof.telegram.bot.launchers.lambda.config.envModule
 import by.jprof.telegram.bot.launchers.lambda.config.jsonModule
 import by.jprof.telegram.bot.launchers.lambda.config.pipelineModule
+import by.jprof.telegram.bot.launchers.lambda.config.secretsModule
 import by.jprof.telegram.bot.launchers.lambda.config.sfnModule
 import by.jprof.telegram.bot.launchers.lambda.config.telegramModule
 import by.jprof.telegram.bot.launchers.lambda.config.urbanDictionaryModule
@@ -45,6 +46,7 @@ class JProf : RequestHandler<APIGatewayV2HTTPEvent, APIGatewayV2HTTPResponse>, K
     init {
         startKoin {
             modules(
+                secretsModule,
                 envModule,
                 databaseModule,
                 jsonModule,

launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/env.kt

@@ -1,7 +1,13 @@
 package by.jprof.telegram.bot.launchers.lambda.config
 
+import by.jprof.telegram.bot.shop.provider.ChatProviderTokens
+import kotlinx.serialization.decodeFromString
+import kotlinx.serialization.json.Json
 import org.koin.core.qualifier.named
 import org.koin.dsl.module
+import software.amazon.awssdk.services.secretsmanager.SecretsManagerClient
+
+private const val SECRET_PAYMENT_PROVIDER_TOKENS = "jprof-by-bot-secret-payment-provider-tokens"
 
 const val TOKEN_TELEGRAM_BOT = "TOKEN_TELEGRAM_BOT"
 const val TOKEN_YOUTUBE_API = "TOKEN_YOUTUBE_API"
@@ -42,4 +48,14 @@ val envModule = module {
     single(named(TIMEOUT)) {
         System.getenv(TIMEOUT)!!.toLong()
     }
+
+    single {
+        val json: Json = get()
+        val secrets: SecretsManagerClient = get()
+        val secret = secrets.getSecretValue {
+            it.secretId(SECRET_PAYMENT_PROVIDER_TOKENS)
+        }
+
+        json.decodeFromString<ChatProviderTokens>(secret.secretString())
+    }
 }

launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/pipeline.kt

@@ -21,6 +21,12 @@ import by.jprof.telegram.bot.quizoji.QuizojiOptionUpdateProcessor
 import by.jprof.telegram.bot.quizoji.QuizojiQuestionUpdateProcessor
 import by.jprof.telegram.bot.quizoji.QuizojiStartCommandUpdateProcessor
 import by.jprof.telegram.bot.quizoji.QuizojiVoteUpdateProcessor
+import by.jprof.telegram.bot.shop.ForwardedPaymentStartCommandUpdateProcessor
+import by.jprof.telegram.bot.shop.PinsPreCheckoutQueryUpdateProcessor
+import by.jprof.telegram.bot.shop.RichCommandUpdateProcessor
+import by.jprof.telegram.bot.shop.RichPreCheckoutQueryUpdateProcessor
+import by.jprof.telegram.bot.shop.SupportCommandUpdateProcessor
+import by.jprof.telegram.bot.shop.SupportPreCheckoutQueryUpdateProcessor
 import by.jprof.telegram.bot.times.TimeCommandUpdateProcessor
 import by.jprof.telegram.bot.times.TimeZoneCommandUpdateProcessor
 import by.jprof.telegram.bot.youtube.YouTubeUpdateProcessor
@@ -116,6 +122,8 @@ val pipelineModule = module {
             pinDAO = get(),
             unpinScheduler = get(),
             bot = get(),
+            providerTokens = get(),
+            json = get(),
         )
     }
 
@@ -191,4 +199,48 @@ val pipelineModule = module {
             bot = get(),
         )
     }
+
+    single<UpdateProcessor>(named("RichCommandUpdateProcessor")) {
+        RichCommandUpdateProcessor(
+            bot = get(),
+            providerTokens = get(),
+            json = get(),
+        )
+    }
+
+    single<UpdateProcessor>(named("RichPreCheckoutQueryUpdateProcessor")) {
+        RichPreCheckoutQueryUpdateProcessor(
+            bot = get(),
+            json = get(),
+        )
+    }
+
+    single<UpdateProcessor>(named("SupportCommandUpdateProcessor")) {
+        SupportCommandUpdateProcessor(
+            bot = get(),
+            providerTokens = get(),
+            json = get(),
+        )
+    }
+
+    single<UpdateProcessor>(named("SupportPreCheckoutQueryUpdateProcessor")) {
+        SupportPreCheckoutQueryUpdateProcessor(
+            bot = get(),
+            json = get(),
+        )
+    }
+
+    single<UpdateProcessor>(named("ForwardedPaymentStartCommandUpdateProcessor")) {
+        ForwardedPaymentStartCommandUpdateProcessor(
+            bot = get(),
+        )
+    }
+
+    single<UpdateProcessor>(named("PinsPreCheckoutQueryUpdateProcessor")) {
+        PinsPreCheckoutQueryUpdateProcessor(
+            bot = get(),
+            json = get(),
+            moniesDAO = get(),
+        )
+    }
 }

launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/secrets.kt

@@ -0,0 +1,12 @@
+package by.jprof.telegram.bot.launchers.lambda.config
+
+import kotlinx.serialization.ExperimentalSerializationApi
+import org.koin.dsl.module
+import software.amazon.awssdk.services.secretsmanager.SecretsManagerClient
+
+@ExperimentalSerializationApi
+val secretsModule = module {
+    single {
+        SecretsManagerClient.create()
+    }
+}

pins/build.gradle.kts

@@ -6,6 +6,8 @@ dependencies {
     api(project.projects.core)
     api(libs.tgbotapi)
     api(project.projects.monies)
+    implementation(project.projects.shop.provider)
+    implementation(project.projects.shop.payload)
     implementation(project.projects.pins.dto)
     implementation(project.projects.pins.scheduler)
     implementation(libs.log4j.api)

pins/src/main/kotlin/by/jprof/telegram/bot/pins/PinCommandUpdateProcessor.kt

@@ -8,6 +8,7 @@ import by.jprof.telegram.bot.pins.dao.PinDAO
 import by.jprof.telegram.bot.pins.dto.Unpin
 import by.jprof.telegram.bot.pins.model.Pin
 import by.jprof.telegram.bot.pins.model.PinDuration
+import by.jprof.telegram.bot.pins.model.PinRequest
 import by.jprof.telegram.bot.pins.scheduler.UnpinScheduler
 import by.jprof.telegram.bot.pins.utils.PinRequestFinder
 import by.jprof.telegram.bot.pins.utils.beggar
@@ -16,13 +17,20 @@ import by.jprof.telegram.bot.pins.utils.negativeDuration
 import by.jprof.telegram.bot.pins.utils.tooManyPinnedMessages
 import by.jprof.telegram.bot.pins.utils.tooPositiveDuration
 import by.jprof.telegram.bot.pins.utils.unrecognizedDuration
+import by.jprof.telegram.bot.shop.payload.PinsPayload
+import by.jprof.telegram.bot.shop.provider.ChatProviderTokens
 import dev.inmo.tgbotapi.bot.RequestsExecutor
 import dev.inmo.tgbotapi.extensions.api.chat.modify.pinChatMessage
+import dev.inmo.tgbotapi.extensions.api.send.payments.sendInvoice
 import dev.inmo.tgbotapi.extensions.api.send.reply
 import dev.inmo.tgbotapi.types.message.MarkdownV2
+import dev.inmo.tgbotapi.types.payments.LabeledPrice
 import dev.inmo.tgbotapi.types.update.abstracts.Update
 import dev.inmo.tgbotapi.utils.PreviewFeature
 import java.time.Duration
+import kotlin.random.Random
+import kotlinx.serialization.encodeToString
+import kotlinx.serialization.json.Json
 import org.apache.logging.log4j.LogManager
 
 @PreviewFeature
@@ -31,6 +39,8 @@ class PinCommandUpdateProcessor(
     private val pinDAO: PinDAO,
     private val unpinScheduler: UnpinScheduler,
     private val bot: RequestsExecutor,
+    private val providerTokens: ChatProviderTokens,
+    private val json: Json,
     private val pinRequestFinder: PinRequestFinder = PinRequestFinder.DEFAULT
 ) : UpdateProcessor {
     companion object {
@@ -47,6 +57,7 @@ class PinCommandUpdateProcessor(
 
             if (pin.message == null) {
                 bot.reply(to = pin.request, text = help(pins), parseMode = MarkdownV2)
+                pinsShop(pin)
 
                 return
             }
@@ -71,6 +82,7 @@ class PinCommandUpdateProcessor(
 
             if (pins < pin.price) {
                 bot.reply(to = pin.request, text = beggar(pins, pin.price), parseMode = MarkdownV2)
+                pinsShop(pin)
 
                 return
             }
@@ -90,9 +102,36 @@ class PinCommandUpdateProcessor(
                     chatId = pin.chat.id.chatId
                     ttl = duration.duration.seconds
                 })
+                if (Random.nextInt(4) == 0) {
+                    pinsShop(pin)
+                }
             } catch (e: Exception) {
                 logger.error("Failed to pin a message", e)
             }
         }
     }
+
+    private suspend fun pinsShop(pin: PinRequest) {
+        val chatProviderToken = providerTokens[pin.request.chat.id.chatId]
+
+        if (chatProviderToken != null) {
+            bot.sendInvoice(
+                chatId = pin.request.chat.id,
+                title = "168 пинов",
+                description = "Неделя закрепа",
+                payload = json.encodeToString(PinsPayload(
+                    pins = 168,
+                    chat = pin.request.chat.id.chatId,
+                )),
+                providerToken = chatProviderToken,
+                currency = "USD",
+                prices = listOf(
+                    LabeledPrice("Пины × 168", 200)
+                ),
+                startParameter = "forwarded_payment",
+                replyToMessageId = pin.request.messageId,
+                allowSendingWithoutReply = true,
+            )
+        }
+    }
 }