/ud command update processor

madhead · madhead · commit f9c892cd6fc9 · 2023-01-08T22:51:26.000+01:00
diff --git a/.deploy/lambda/lib/JProfByBotStack.ts b/.deploy/lambda/lib/JProfByBotStack.ts
@@ -153,6 +153,7 @@ export class JProfByBotStack extends cdk.Stack {
                 'TABLE_PINS': pinsTable.tableName,
                 'TABLE_TIMEZONES': timezonesTable.tableName,
                 'TABLE_LANGUAGE_ROOMS': languageRoomsTable.tableName,
+                'TABLE_URBAN_WORDS_OF_THE_DAY': urbanWordsOfTheDayTable.tableName,
                 'STATE_MACHINE_UNPINS': stateMachineUnpin.stateMachineArn,
                 'TOKEN_TELEGRAM_BOT': props.telegramToken,
                 'TOKEN_YOUTUBE_API': props.youtubeToken,
@@ -209,6 +210,7 @@ export class JProfByBotStack extends cdk.Stack {
         languageRoomsTable.grantReadData(lambdaDailyUrbanDictionary);
 
         urbanWordsOfTheDayTable.grantWriteData(lambdaDailyUrbanDictionary);
+        urbanWordsOfTheDayTable.grantReadData(lambdaWebhook);
 
         stateMachineUnpin.grantStartExecution(lambdaWebhook);
         stateMachineUnpin.grantStartExecution(lambdaDailyUrbanDictionary);
diff --git a/english/README.adoc b/english/README.adoc
@@ -1 +1,5 @@
 = English
+
+This is an umbrella feature for several smaller sub-features:
+
+* The bot sends https://www.urbandictionary.com[Urban Word of the Day] into English rooms.
diff --git a/english/build.gradle.kts b/english/build.gradle.kts
@@ -5,6 +5,8 @@ plugins {
 dependencies {
     api(project.projects.core)
     implementation(projects.english.languageRooms)
+    implementation(projects.english.urbanWordOfTheDay)
+    implementation(projects.english.urbanWordOfTheDayFormatter)
     implementation(libs.log4j.api)
 
     testImplementation(libs.junit.jupiter.api)
diff --git a/english/src/main/kotlin/by/jprof/telegram/bot/english/UrbanWordOfTheDayUpdateProcessor.kt b/english/src/main/kotlin/by/jprof/telegram/bot/english/UrbanWordOfTheDayUpdateProcessor.kt
@@ -0,0 +1,69 @@
+package by.jprof.telegram.bot.english
+
+import by.jprof.telegram.bot.core.UpdateProcessor
+import by.jprof.telegram.bot.english.language_rooms.dao.LanguageRoomDAO
+import by.jprof.telegram.bot.english.language_rooms.model.Language
+import by.jprof.telegram.bot.english.urban_dictionary_definition_formatter.format
+import by.jprof.telegram.bot.english.urban_word_of_the_day.dao.UrbanWordOfTheDayDAO
+import by.jprof.telegram.bot.english.utils.downTo
+import dev.inmo.tgbotapi.bot.RequestsExecutor
+import dev.inmo.tgbotapi.extensions.api.send.reply
+import dev.inmo.tgbotapi.extensions.utils.asBaseMessageUpdate
+import dev.inmo.tgbotapi.extensions.utils.asBotCommandTextSource
+import dev.inmo.tgbotapi.extensions.utils.asContentMessage
+import dev.inmo.tgbotapi.extensions.utils.asTextContent
+import dev.inmo.tgbotapi.types.message.MarkdownV2
+import dev.inmo.tgbotapi.types.update.abstracts.Update
+import java.time.LocalDate
+import org.apache.logging.log4j.LogManager
+
+class UrbanWordOfTheDayUpdateProcessor(
+    private val languageRoomDAO: LanguageRoomDAO,
+    private val urbanWordOfTheDayDAO: UrbanWordOfTheDayDAO,
+    private val bot: RequestsExecutor,
+) : UpdateProcessor {
+    companion object {
+        private val logger = LogManager.getLogger(UrbanWordOfTheDayUpdateProcessor::class.java)!!
+    }
+
+    override suspend fun process(update: Update) {
+        val update = update.asBaseMessageUpdate() ?: return
+        val roomId = update.data.chat.id
+        val message = update.data.asContentMessage() ?: return
+        val content = message.content.asTextContent() ?: return
+
+        if (
+            content.textSources.none {
+                it.asBotCommandTextSource()?.command?.lowercase() in listOf("urban", "ud", "urbanword", "urbanwordoftheday")
+            }
+        ) {
+            return
+        }
+
+        if (
+            languageRoomDAO.get(roomId.chatId, roomId.threadId)?.takeIf { it.language == Language.ENGLISH } == null
+        ) {
+            return
+        }
+
+        val urbanWordOfTheDay = (LocalDate.now() downTo LocalDate.now().minusDays(5)).asSequence().firstNotNullOfOrNull {
+            urbanWordOfTheDayDAO.get(it)
+        }
+
+        if (urbanWordOfTheDay != null) {
+            bot.reply(
+                to = message,
+                text = urbanWordOfTheDay.format(),
+                parseMode = MarkdownV2,
+                disableWebPagePreview = true,
+            )
+        } else {
+            bot.reply(
+                to = message,
+                text = "No recent words of the day \uD83D\uDE22",
+                parseMode = MarkdownV2,
+                disableWebPagePreview = true,
+            )
+        }
+    }
+}
diff --git a/english/src/main/kotlin/by/jprof/telegram/bot/english/utils/LocalDateX.kt b/english/src/main/kotlin/by/jprof/telegram/bot/english/utils/LocalDateX.kt
@@ -0,0 +1,12 @@
+package by.jprof.telegram.bot.english.utils
+
+import java.time.LocalDate
+
+infix fun LocalDate.downTo(other: LocalDate): Iterator<LocalDate> = iterator {
+    var current = this@downTo
+
+    while (current >= other) {
+        yield(current)
+        current = current.minusDays(1)
+    }
+}
diff --git a/launchers/lambda/build.gradle.kts b/launchers/lambda/build.gradle.kts
@@ -25,4 +25,5 @@ dependencies {
     implementation(project.projects.times)
     implementation(project.projects.english)
     implementation(project.projects.english.languageRooms.dynamodb)
+    implementation(project.projects.english.urbanWordOfTheDay.dynamodb)
 }
diff --git a/launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/database.kt b/launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/database.kt
@@ -2,6 +2,7 @@ package by.jprof.telegram.bot.launchers.lambda.config
 
 import by.jprof.telegram.bot.dialogs.dao.DialogStateDAO
 import by.jprof.telegram.bot.english.language_rooms.dao.LanguageRoomDAO
+import by.jprof.telegram.bot.english.urban_word_of_the_day.dao.UrbanWordOfTheDayDAO
 import by.jprof.telegram.bot.kotlin.dao.KotlinMentionsDAO
 import by.jprof.telegram.bot.monies.dao.MoniesDAO
 import by.jprof.telegram.bot.pins.dao.PinDAO
@@ -15,6 +16,7 @@ import org.koin.dsl.module
 import software.amazon.awssdk.services.dynamodb.DynamoDbAsyncClient
 import by.jprof.telegram.bot.dialogs.dynamodb.dao.DialogStateDAO as DynamoDBDialogStateDAO
 import by.jprof.telegram.bot.english.language_rooms.dynamodb.dao.LanguageRoomDAO as DynamoDBLanguageRoomDAO
+import by.jprof.telegram.bot.english.urban_word_of_the_day.dynamodb.dao.UrbanWordOfTheDayDAO as DynamoDBUrbanWordOfTheDayDAO
 import by.jprof.telegram.bot.kotlin.dynamodb.dao.KotlinMentionsDAO as DynamoDBKotlinMentionsDAO
 import by.jprof.telegram.bot.monies.dynamodb.dao.MoniesDAO as DynamoDBMoniesDAO
 import by.jprof.telegram.bot.pins.dynamodb.dao.PinDAO as DynamoDBPinDAO
@@ -91,4 +93,11 @@ val databaseModule = module {
             get(named(TABLE_LANGUAGE_ROOMS))
         )
     }
+
+    single<UrbanWordOfTheDayDAO> {
+        DynamoDBUrbanWordOfTheDayDAO(
+            get(),
+            get(named(TABLE_URBAN_WORDS_OF_THE_DAY))
+        )
+    }
 }
diff --git a/launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/env.kt b/launchers/lambda/src/main/kotlin/by/jprof/telegram/bot/launchers/lambda/config/env.kt
@@ -14,6 +14,7 @@ const val TABLE_MONIES = "TABLE_MONIES"
 const val TABLE_PINS = "TABLE_PINS"
 const val TABLE_TIMEZONES = "TABLE_TIMEZONES"
 const val TABLE_LANGUAGE_ROOMS = "TABLE_LANGUAGE_ROOMS"
+const val TABLE_URBAN_WORDS_OF_THE_DAY = "TABLE_URBAN_WORDS_OF_THE_DAY"
 const val STATE_MACHINE_UNPINS = "STATE_MACHINE_UNPINS"
 
 val envModule = module {
@@ -29,6 +30,7 @@ val envModule = module {
         TABLE_PINS,
         TABLE_TIMEZONES,
         TABLE_LANGUAGE_ROOMS,
+        TABLE_URBAN_WORDS_OF_THE_DAY,
         STATE_MACHINE_UNPINS,
     ).forEach { variable ->
         single(named(variable)) {
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
@@ -4,6 +4,7 @@ import by.jprof.telegram.bot.core.UpdateProcessingPipeline
 import by.jprof.telegram.bot.core.UpdateProcessor
 import by.jprof.telegram.bot.currencies.CurrenciesUpdateProcessor
 import by.jprof.telegram.bot.english.EnglishCommandUpdateProcessor
+import by.jprof.telegram.bot.english.UrbanWordOfTheDayUpdateProcessor
 import by.jprof.telegram.bot.eval.EvalUpdateProcessor
 import by.jprof.telegram.bot.jep.JEPUpdateProcessor
 import by.jprof.telegram.bot.jep.JsoupJEPSummary
@@ -153,4 +154,12 @@ val pipelineModule = module {
             bot = get(),
         )
     }
+
+    single<UpdateProcessor>(named("UrbanWordOfTheDayUpdateProcessor")) {
+        UrbanWordOfTheDayUpdateProcessor(
+            languageRoomDAO = get(),
+            urbanWordOfTheDayDAO = get(),
+            bot = get(),
+        )
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -25,4 +25,5 @@ dependencies {`
`25`	`25`	`implementation(project.projects.times)`
`26`	`26`	`implementation(project.projects.english)`
`27`	`27`	`implementation(project.projects.english.languageRooms.dynamodb)`
	`28`	`+ implementation(project.projects.english.urbanWordOfTheDay.dynamodb)`
`28`	`29`	`}`