Process LeetCode links

Author: madhead

leetcode/src/main/kotlin/by/jprof/telegram/bot/leetcode/LeetCodeUpdateProcessor.kt

@@ -1,27 +1,51 @@
 package by.jprof.telegram.bot.leetcode
 
 import by.jprof.telegram.bot.core.UpdateProcessor
+import dev.inmo.tgbotapi.bot.RequestsExecutor
+import dev.inmo.tgbotapi.extensions.api.send.reply
 import dev.inmo.tgbotapi.types.MessageEntity.textsources.TextLinkTextSource
 import dev.inmo.tgbotapi.types.MessageEntity.textsources.URLTextSource
+import dev.inmo.tgbotapi.types.ParseMode.MarkdownV2ParseMode
 import dev.inmo.tgbotapi.types.message.abstracts.ContentMessage
 import dev.inmo.tgbotapi.types.message.abstracts.Message
 import dev.inmo.tgbotapi.types.message.content.TextContent
 import dev.inmo.tgbotapi.types.update.MessageUpdate
 import dev.inmo.tgbotapi.types.update.abstracts.Update
+import dev.inmo.tgbotapi.utils.extensions.escapeMarkdownV2Common
+import kotlinx.coroutines.async
+import kotlinx.coroutines.awaitAll
+import kotlinx.coroutines.coroutineScope
 import org.apache.logging.log4j.LogManager
 
-class LeetCodeUpdateProcessor : UpdateProcessor {
+class LeetCodeUpdateProcessor(
+    private val bot: RequestsExecutor,
+) : UpdateProcessor {
     companion object {
         private val logger = LogManager.getLogger(LeetCodeUpdateProcessor::class.java)!!
         private val slugExtractor: SlugExtractor = ::NaiveRegexSlugExtractor
+        private val leetCodeClient: LeetCodeClient = GraphQLLeetCodeClient()
     }
 
     override suspend fun process(update: Update) {
         @Suppress("NAME_SHADOWING") val update = update as? MessageUpdate ?: return
 
-        val slugs = extractSlugs(update.data) ?: return
+        val slugs = extractSlugs(update.data)?.takeUnless { it.isEmpty() }?.distinct() ?: return
 
         logger.debug("LeetCode slugs: {}", slugs)
+
+        val questions = coroutineScope {
+            slugs.map { slug -> async { leetCodeClient.questionData(slug) } }.awaitAll()
+        }.filterNotNull()
+
+        logger.debug("Questions: {}", questions)
+
+        questions.forEach { question ->
+            bot.reply(
+                to = update.data,
+                text = questionInfo(question),
+                parseMode = MarkdownV2ParseMode,
+            )
+        }
     }
 
     private fun extractSlugs(message: Message): List<String>? =
@@ -37,4 +61,14 @@ class LeetCodeUpdateProcessor : UpdateProcessor {
                     }
             }
         }
+
+    private fun questionInfo(question: Question): String {
+        return """
+        |${question.level()} *${question.title.escapeMarkdownV2Common()}* \(${question.categoryTitle}\) ${question.paidIndicator()}
+        |
+        |${question.markdownContent()}
+        |
+        |_Likes: ${question.likes} / Dislikes: ${question.dislikes}_
+    """.trimMargin()
+    }
 }

leetcode/src/main/kotlin/by/jprof/telegram/bot/leetcode/Question.kt

@@ -1,5 +1,6 @@
 package by.jprof.telegram.bot.leetcode
 
+import dev.inmo.tgbotapi.utils.extensions.escapeMarkdownV2Common
 import kotlinx.serialization.Serializable
 
 @Serializable
@@ -12,4 +13,50 @@ data class Question(
     val likes: Int,
     val dislikes: Int,
     val categoryTitle: String,
-)
+) {
+    fun level(): String = when (this.difficulty) {
+        "Easy" -> "\uD83E\uDD64"
+        "Medium" -> "\uD83E\uDD14"
+        "Hard" -> "\uD83E\uDD18"
+        else -> ""
+    }
+
+    fun paidIndicator() = if (isPaidOnly) {
+        "\uD83E\uDD11"
+    } else {
+        ""
+    }
+
+    fun markdownContent(): String = this.content
+        .replace(" ", " ")
+        .replace("&lt;", "<")
+        .replace("&gt;", ">")
+        .replace(Regex("<sup>(?<sup>.+?)</sup>")) {
+            "^${it.groups["sup"]!!.value}"
+        }
+
+        .escapeMarkdownV2Common()
+
+        .replace("<code\\>", "`")
+        .replace("</code\\>", "`")
+
+        .replace("<pre\\>", "```")
+        .replace("</pre\\>", "```")
+
+        .replace("<strong\\>", "*")
+        .replace("</strong\\>", "*")
+
+        .replace("<em\\>", "_")
+        .replace("</em\\>", "_")
+
+        .replace("<li\\>", "• ")
+        .replace("</li\\>", "")
+
+        .replace("<p\\>", "")
+        .replace("</p\\>", "")
+        .replace("<ul\\>", "")
+        .replace("</ul\\>", "")
+
+        .replace("\\n", "\n")
+        .replace("\\t", "\t")
+}

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

@@ -134,6 +134,7 @@ val pipelineModule = module {
 
     single<UpdateProcessor>(named("LeetCodeUpdateProcessor")) {
         LeetCodeUpdateProcessor(
+            bot = get(),
         )
     }
 }