From d1858e8c7b719251f3a987462dfcece8b0f0b44b Mon Sep 17 00:00:00 2001
From: Blatzar <46196380+Blatzar@users.noreply.github.com>
Date: Mon, 12 Sep 2022 17:23:24 +0200
Subject: [PATCH] Add premium seasons to Crunchy

---
 Crunchyroll/build.gradle.kts                  |  2 +-
 .../com/lagradost/CrunchyrollProvider.kt      | 44 +++++++++++++------
 2 files changed, 31 insertions(+), 15 deletions(-)

diff --git a/Crunchyroll/build.gradle.kts b/Crunchyroll/build.gradle.kts
index f650b1c..8c4e9a0 100644
--- a/Crunchyroll/build.gradle.kts
+++ b/Crunchyroll/build.gradle.kts
@@ -1,5 +1,5 @@
 // use an integer for version numbers
-version = 3
+version = 4
 
 
 cloudstream {
diff --git a/Crunchyroll/src/main/kotlin/com/lagradost/CrunchyrollProvider.kt b/Crunchyroll/src/main/kotlin/com/lagradost/CrunchyrollProvider.kt
index dc3a2af..45ac4ff 100644
--- a/Crunchyroll/src/main/kotlin/com/lagradost/CrunchyrollProvider.kt
+++ b/Crunchyroll/src/main/kotlin/com/lagradost/CrunchyrollProvider.kt
@@ -298,16 +298,19 @@ class KrunchyProvider : MainAPI() {
                     epDesc = "★ $epDesc ★"
                 }
 
+                val isPremiumDubbed =
+                    isPremium && seasonName != null && (seasonName.contains("Dub") || seasonName.contains(
+                        "Russian"
+                    ) || seasonName.contains("Spanish"))
+
                 val epi = Episode(
                     fixUrl(ep.attr("href")),
                     "$epTitle",
                     posterUrl = poster?.replace("widestar", "full")?.replace("wide", "full"),
-                    description = epDesc
+                    description = epDesc,
+                    season = if (isPremium) -1 else 1
                 )
-                if (isPremium && seasonName != null && (seasonName.contains("Dub") || seasonName.contains(
-                        "Russian"
-                    ) || seasonName.contains("Spanish"))
-                ) {
+                if (isPremiumDubbed) {
                     premiumDubEpisodes.add(epi)
                 } else if (isPremium) {
                     premiumSubEpisodes.add(epi)
@@ -341,20 +344,33 @@ class KrunchyProvider : MainAPI() {
             this.engName = title
             if (subEpisodes.isNotEmpty()) addEpisodes(DubStatus.Subbed, subEpisodes.reversed())
             if (dubEpisodes.isNotEmpty()) addEpisodes(DubStatus.Dubbed, dubEpisodes.reversed())
-            // TODO add arbitrary seasons
 
-            //if (premiumDubEpisodes.isNotEmpty()) addEpisodes(
-            //   DubStatus.PremiumDub,
-            //   premiumDubEpisodes.reversed()
-            //  )
-            // if (premiumSubEpisodes.isNotEmpty()) addEpisodes(
-            //    DubStatus.PremiumSub,
-            //   premiumSubEpisodes.reversed()
-            //  )
+            if (premiumDubEpisodes.isNotEmpty()) addEpisodes(
+                DubStatus.Dubbed,
+                premiumDubEpisodes.reversed()
+            )
+            if (premiumSubEpisodes.isNotEmpty()) addEpisodes(
+                DubStatus.Subbed,
+                premiumSubEpisodes.reversed()
+            )
+
             this.plot = description
             this.tags = genres
             this.year = year
+
             this.recommendations = recommendations
+            this.seasonNames = listOf(
+                SeasonData(
+                    1,
+                    "Free",
+                    null
+                ),
+                SeasonData(
+                    -1,
+                    "Premium",
+                    null
+                ),
+            )
         }
     }