From 21635d7b2b1b53a45ecf4433b99e8de8dc8da108 Mon Sep 17 00:00:00 2001
From: Koitharu <nvasya95@gmail.com>
Date: Fri, 6 May 2022 14:44:16 +0300
Subject: [PATCH] [GroupLe] Select server for pages

---
 .../kotatsu/parsers/MangaLoaderContext.kt     | 10 ++++++
 .../kotatsu/parsers/site/GroupleParser.kt     | 35 ++++++++++++++-----
 2 files changed, 37 insertions(+), 8 deletions(-)

diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt
index 7511b29d..25fd49db 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt
@@ -28,6 +28,16 @@ abstract class MangaLoaderContext {
 		return httpClient.newCall(request.build()).await().ensureSuccess()
 	}
 
+	suspend fun httpHead(url: String, headers: Headers? = null): Response {
+		val request = Request.Builder()
+			.head()
+			.url(url)
+		if (headers != null) {
+			request.headers(headers)
+		}
+		return httpClient.newCall(request.build()).await().ensureSuccess()
+	}
+
 	suspend fun httpPost(
 		url: String,
 		form: Map<String, String>,
diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/GroupleParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/GroupleParser.kt
index 75e1595c..05ed85bc 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/GroupleParser.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/GroupleParser.kt
@@ -3,10 +3,12 @@ package org.koitharu.kotatsu.parsers.site
 import okhttp3.Headers
 import okhttp3.HttpUrl.Companion.toHttpUrl
 import okhttp3.Response
+import org.json.JSONArray
 import org.koitharu.kotatsu.parsers.MangaParser
 import org.koitharu.kotatsu.parsers.exception.ParseException
 import org.koitharu.kotatsu.parsers.model.*
 import org.koitharu.kotatsu.parsers.util.*
+import org.koitharu.kotatsu.parsers.util.json.mapJSON
 import java.text.SimpleDateFormat
 import java.util.*
 
@@ -168,16 +170,19 @@ internal abstract class GroupleParser(source: MangaSource) : MangaParser(source)
 			if (pos == -1) {
 				continue
 			}
-			val json = data.substring(pos).substringAfter('[').substringBeforeLast(']')
-			val matches = Regex("\\[.*?]").findAll(json).toList()
-			val regex = Regex("['\"].*?['\"]")
-			return matches.map { x ->
-				val parts = regex.findAll(x.value).toList()
-				val url = parts[0].value.removeSurrounding('"', '\'') +
-					parts[2].value.removeSurrounding('"', '\'')
+			val json = data.substring(pos).substringAfter('(').substringBefore('\n')
+				.substringBeforeLast(')')
+			val ja = JSONArray("[$json]")
+			val pages = ja.getJSONArray(1)
+			val servers = ja.getJSONArray(4).mapJSON { it.getString("path") }
+			val serversStr = servers.joinToString("|")
+			return (0 until pages.length()).map { i ->
+				val page = pages.getJSONArray(i)
+				val primaryServer = page.getString(0)
+				val url = page.getString(2)
 				MangaPage(
 					id = generateUid(url),
-					url = url,
+					url = "$primaryServer|$serversStr|$url",
 					preview = null,
 					referer = chapter.url,
 					source = source,
@@ -187,6 +192,20 @@ internal abstract class GroupleParser(source: MangaSource) : MangaParser(source)
 		throw ParseException("Pages list not found at ${chapter.url}")
 	}
 
+	override suspend fun getPageUrl(page: MangaPage): String {
+		val parts = page.url.split('|')
+		val path = parts.last()
+		val servers = parts.dropLast(1).toSet()
+		val headers = Headers.headersOf("Referer", page.referer)
+		for (server in servers) {
+			val url = server + path
+			if (context.httpHead(url, headers).isSuccessful) {
+				return url
+			}
+		}
+		throw IllegalArgumentException("Cannot find any page url")
+	}
+
 	override suspend fun getTags(): Set<MangaTag> {
 		val doc = context.httpGet("https://${getDomain()}/list/genres/sort_name", headers).parseHtml()
 		val root = doc.body().getElementById("mangaBox")?.selectFirst("div.leftContent")