Fixed WtrLab, but only raw

Author: LagradOst

app/src/main/java/com/lagradost/quicknovel/providers/WtrLabProvider.kt

@@ -1,6 +1,7 @@
 package com.lagradost.quicknovel.providers
 
 import com.fasterxml.jackson.annotation.JsonProperty
+import com.lagradost.quicknovel.ChapterData
 import com.lagradost.quicknovel.ErrorLoadingException
 import com.lagradost.quicknovel.LoadResponse
 import com.lagradost.quicknovel.MainAPI
@@ -38,6 +39,29 @@ class WtrLabProvider : MainAPI() {
     }
 
 
+    private suspend fun getChapterRange(
+        url: String,
+        chaptersJson: ResultJsonResponse.Root,
+        start: Long,
+        end: Long
+    ): List<ChapterData> {
+        val chapterDataUrl =
+            "$mainUrl/api/chapters/${chaptersJson.props.pageProps.serie.serieData.rawId}?start=$start&end=$end"
+        println(chapterDataUrl)
+        val chaptersDataJson =
+            app.get(chapterDataUrl).text
+        val chaptersData = parseJson<ResultChaptersJsonResponse.Root>(chaptersDataJson)
+
+        return chaptersData.chapters.map { chapter ->
+            newChapterData(
+                "#${chapter.order} ${chapter.title}",
+                "${url.trimEnd('/')}/chapter-${chapter.order}"
+            ) {
+                dateOfRelease = chapter.updatedAt
+            }
+        }
+    }
+
     override suspend fun load(url: String): LoadResponse {
         val doc = app.get(url).document
         val titleWrap =
@@ -50,20 +74,34 @@ class WtrLabProvider : MainAPI() {
 
         val chaptersJson = parseJson<ResultJsonResponse.Root>(json)
 
-        val chapterDataUrl =
-            "$mainUrl/api/chapters/${chaptersJson.props.pageProps.serie.serieData.id}?start=1&end=${chaptersJson.props.pageProps.serie.serieData.rawChapterCount}"
-        val chaptersDataJson =
-            app.get(chapterDataUrl).text
-        val chaptersData = parseJson<ResultChaptersJsonResponse.Root>(chaptersDataJson)
 
-        val chapters = chaptersData.chapters.map { chapter ->
-            newChapterData(
-                "#${chapter.order} ${chapter.title}",
-                "${url.trimEnd('/')}/chapter-${chapter.order}"
-            ) {
-                dateOfRelease = chapter.updatedAt
-            }
+        val chapters = mutableListOf<ChapterData>()
+        chapters.addAll(
+            getChapterRange(
+                url,
+                chaptersJson,
+                1,
+                chaptersJson.props.pageProps.serie.serieData.rawChapterCount
+            )
+        )
+
+        /*
+        val chunks = chaptersJson.props.pageProps.serie.serieData.rawChapterCount / 250
+        val tail = chaptersJson.props.pageProps.serie.serieData.rawChapterCount % 250
+        for (chunk in 0 until chunks) {
+            chapters.addAll(getChapterRange(url, chaptersJson, chunk * 250 + 1, chunk * 250 + 250))
         }
+        if (tail > 0) {
+            chapters.addAll(
+                getChapterRange(
+                    url,
+                    chaptersJson,
+                    chunks * 250 + 1,
+                    chunks * 250 + tail
+                )
+            )
+        }*/
+
         /*doc.select(".toc-list > .chapter-item").map { select ->
             val href = select.attr("href") ?: throw ErrorLoadingException("No href on $select")
             val chapterTitle = select.selectFirst("span")?.text() ?: select.text()
@@ -91,13 +129,11 @@ class WtrLabProvider : MainAPI() {
 
         val root = app.post(
             "$mainUrl/api/reader/get", data = mapOf(
-                "chapter_id" to chapter.id.toString(),
                 "chapter_no" to chapter.slug,
-                "force_retry" to "false",
                 "language" to "en",
                 "raw_id" to chapter.rawId.toString(),
                 "retry" to "false",
-                "translate" to "ai",
+                "translate" to "web", // translate=ai just returns a job and I am too lazy to fix that
             )
         ).parsed<LoadJsonResponse2.Root>()
 
@@ -187,8 +223,10 @@ object ResultJsonResponse {
     )
 
     data class SerieData(
-        val id: Long,
-        /*val slug: String,
+        @JsonProperty("raw_id")
+        val rawId: Long,
+        /*
+        val id: Long,val slug: String,
         @JsonProperty("search_text")
         val searchText: String,
         val status: Long,
@@ -211,8 +249,6 @@ object ResultJsonResponse {
         val verified: Boolean,
         val from: String,
         val author: String,
-        @JsonProperty("raw_id")
-        val rawId: Long,
         @JsonProperty("ai_enabled")
         val aiEnabled: Boolean,
         @JsonProperty("released_by")
@@ -240,7 +276,7 @@ object ResultJsonResponse {
         val author: String,
         val description: String,
         @JsonProperty("from_user")
-        val fromUser: String,
+        val fromUser: String?,
         val raw: Raw,
         val image: String,
     )
@@ -251,7 +287,7 @@ object ResultJsonResponse {
         val description: String,
     )
 
-    data class Ranks(
+    /*data class Ranks(
         val week: Any?,
         val month: Any?,
         val all: String,
@@ -393,7 +429,7 @@ object ResultJsonResponse {
         val sid: String,
         @JsonProperty("serie_slug")
         val serieSlug: String,
-    )
+    )*/
 }
 
 object LoadJsonResponse2 {
@@ -426,7 +462,7 @@ object LoadJsonResponse2 {
     )
 
     data class Data2(
-        val body: List<String>,
+        val body: List<String> = emptyList(),
         /*val hans: String,
         val hash: String,
         val model: String,
@@ -444,7 +480,7 @@ object LoadJsonResponse2 {
 object LoadJsonResponse {
     data class Root(
         val props: Props,
-        val page: String,
+        /*val page: String,
         val query: Query,
         val buildId: String,
         val isFallback: Boolean,
@@ -453,18 +489,19 @@ object LoadJsonResponse {
         val locale: String,
         val locales: List<String>,
         val defaultLocale: String,
-        val scriptLoader: List<Any?>,
+        val scriptLoader: List<Any?>,*/
     )
 
     data class Props(
         val pageProps: PageProps,
-        @JsonProperty("__N_SSP")
+        /*@JsonProperty("__N_SSP")
         val nSsp: Boolean,
+        */
     )
 
     data class PageProps(
         val serie: Serie,
-        @JsonProperty("disabe_ads")
+        /*@JsonProperty("disabe_ads")
         val disabeAds: Boolean,
         @JsonProperty("server_time")
         val serverTime: String,
@@ -473,14 +510,14 @@ object LoadJsonResponse {
         @JsonProperty("_sentryTraceData")
         val sentryTraceData: String,
         @JsonProperty("_sentryBaggage")
-        val sentryBaggage: String,
+        val sentryBaggage: String,*/
     )
 
     data class Serie(
-        @JsonProperty("serie_data")
+        /*@JsonProperty("serie_data")
         val serieData: SerieData,
         @JsonProperty("default_service")
-        val defaultService: String,
+        val defaultService: String,*/
         val chapter: Chapter,
     )
 
@@ -520,10 +557,11 @@ object LoadJsonResponse {
 
     data class Chapter(
         val id: Long,
-        @JsonProperty("serie_id")
-        val serieId: Long,
+        val slug: String,
         @JsonProperty("raw_id")
         val rawId: Long,
+        /*@JsonProperty("serie_id")
+        val serieId: Long,
         val status: Long,
         val slug: String,
         val name: String,
@@ -535,7 +573,7 @@ object LoadJsonResponse {
         @JsonProperty("updated_at")
         val updatedAt: String,
         val title: String,
-        val code: String,
+        val code: String,*/
     )
 
     data class ActiveService(