From bd82ba5d510c30f060fdc92c661f68a809fe4d8d Mon Sep 17 00:00:00 2001
From: hexated <hexated@gmail.com>
Date: Sat, 31 Dec 2022 07:06:18 +0700
Subject: [PATCH] smol tweak

---
 SoraStream/src/main/kotlin/com/hexated/SoraExtractor.kt | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/SoraStream/src/main/kotlin/com/hexated/SoraExtractor.kt b/SoraStream/src/main/kotlin/com/hexated/SoraExtractor.kt
index 9095a025..0c81861b 100644
--- a/SoraStream/src/main/kotlin/com/hexated/SoraExtractor.kt
+++ b/SoraStream/src/main/kotlin/com/hexated/SoraExtractor.kt
@@ -1596,7 +1596,11 @@ object SoraExtractor : SoraStream() {
         val matchMedia = doc.select("article.mh-loop-item").map {
             it.select("a").attr("href") to it.select("a").text()
         }.find {
-            it.second.contains(Regex("(?i)($fixTitle)|($title)")) && it.first.contains("$year")
+            if(season == null) {
+                it.second.contains(Regex("(?i)($fixTitle)|($title)")) && it.first.contains("$year")
+            } else {
+                it.second.contains(Regex("(?i)($fixTitle)|($title)")) && it.second.contains(Regex("(?i)(Season\\s?$season)|(S0?$season)"))
+            }
         }
 
         val mediaLink = app.get(matchMedia?.first ?: return).document.selectFirst("a#jake1")?.attr("href")