Fix LeetCode

Author: madhead

leetcode/build.gradle.kts

@@ -9,6 +9,7 @@ dependencies {
     implementation(libs.log4j.api)
     implementation(libs.graphql.kotlin.ktor.client)
     implementation(libs.kotlinx.serialization.core)
+    implementation(libs.jsoup)
 
     testImplementation(libs.junit.jupiter.api)
     testImplementation(libs.junit.jupiter.params)

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

@@ -63,12 +63,16 @@ class LeetCodeUpdateProcessor(
         }
 
     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()
+        return question.markdownContent()?.let {
+            """|${question.level()} *${question.title.escapeMarkdownV2Common()}* \(${question.categoryTitle}\) ${question.paidIndicator()}
+               |
+               |${it}
+               |
+               |_Likes: ${question.likes} / Dislikes: ${question.dislikes}_
+            """.trimMargin()
+        } ?: """|${question.level()} *${question.title.escapeMarkdownV2Common()}* \(${question.categoryTitle}\) ${question.paidIndicator()}
+                |
+                |_Likes: ${question.likes} / Dislikes: ${question.dislikes}_
+            """.trimMargin()
     }
 }

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

@@ -2,12 +2,13 @@ package by.jprof.telegram.bot.leetcode
 
 import dev.inmo.tgbotapi.utils.extensions.escapeMarkdownV2Common
 import kotlinx.serialization.Serializable
+import org.jsoup.Jsoup
 
 @Serializable
 data class Question(
     val title: String,
     val titleSlug: String,
-    val content: String,
+    val content: String?,
     val isPaidOnly: Boolean,
     val difficulty: String,
     val likes: Int,
@@ -27,40 +28,17 @@ data class Question(
         ""
     }
 
-    fun markdownContent(): String = this.content
-        .replace(" ", " ")
-        .replace("&lt;", "<")
-        .replace("&gt;", ">")
-        .replace("&quot;", "\"")
-        .replace("&#39;", "'")
-        .replace(Regex("<sup>(?<sup>.+?)</sup>")) {
-            "^${it.groups["sup"]!!.value}"
+    fun markdownContent(): String? {
+        return try {
+            val content = Jsoup.parse(this.content ?: return null)
+                .select("p")
+                .firstOrNull()
+                ?.text()
+                ?.takeUnless { it.isBlank() } ?: return null
+
+            ((content.takeIf { it.length < 3000 } ?: (content.take(3000) + "…"))).escapeMarkdownV2Common()
+        } catch (_: Exception) {
+            null
         }
-
-        .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("<ol\\>", "")
-        .replace("</ol\\>", "")
-
-        .replace("\\n", "\n")
-        .replace("\\t", "\t")
+    }
 }