From 6183f6edf39abff1baf6f571065094a7a62a7479 Mon Sep 17 00:00:00 2001
From: Koitharu <nvasya95@gmail.com>
Date: Mon, 15 May 2023 10:21:32 +0300
Subject: [PATCH] [ComicK] Fix branching #163 #167

---
 .../kotatsu/parsers/site/ComickFunParser.kt   | 66 ++++++++++---------
 .../kotatsu/parsers/util/Collection.kt        | 11 +++-
 2 files changed, 45 insertions(+), 32 deletions(-)

diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ComickFunParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ComickFunParser.kt
index aae85a03f..d4aa5eacf 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ComickFunParser.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ComickFunParser.kt
@@ -2,7 +2,6 @@ package org.koitharu.kotatsu.parsers.site
 
 import androidx.collection.ArraySet
 import androidx.collection.SparseArrayCompat
-import org.json.JSONArray
 import org.json.JSONObject
 import org.koitharu.kotatsu.parsers.MangaLoaderContext
 import org.koitharu.kotatsu.parsers.MangaParser
@@ -126,7 +125,6 @@ internal class ComickFunParser(context: MangaLoaderContext) : MangaParser(contex
 		val jo = webClient.httpGet(
 			"https://api.${domain}/chapter/${chapter.url}?tachiyomi=true",
 		).parseJson().getJSONObject("chapter")
-		val referer = "https://${domain}/"
 		return jo.getJSONArray("images").mapJSON {
 			val url = it.getString("url")
 			MangaPage(
@@ -169,37 +167,43 @@ internal class ComickFunParser(context: MangaLoaderContext) : MangaParser(contex
 			url = "https://api.${domain}/comic/$hid/chapters?limit=$CHAPTERS_LIMIT",
 		).parseJson().getJSONArray("chapters")
 		val dateFormat = SimpleDateFormat("yyyy-MM-dd")
-		val counters = HashMap<Locale, Int>()
-		return ja.mapReversed { jo ->
-			val locale = Locale.forLanguageTag(jo.getString("lang"))
-			var number = counters[locale] ?: 0
-			number++
-			counters[locale] = number
-			MangaChapter(
-				id = generateUid(jo.getLong("id")),
-				name = buildString {
-					jo.getStringOrNull("vol")?.let { append("Vol ").append(it).append(' ') }
-					jo.getStringOrNull("chap")?.let { append("Chap ").append(it) }
-					jo.getStringOrNull("title")?.let { append(": ").append(it) }
-				},
-				number = number,
-				url = jo.getString("hid"),
-				scanlator = jo.optJSONArray("group_name")?.optString(0),
-				uploadDate = dateFormat.tryParse(jo.getString("created_at").substringBefore('T')),
-				branch = locale.getDisplayName(locale).toTitleCase(locale),
-				source = source,
-			)
+		val list = ja.toJSONList().reversed()
+		val dc = list.groupBy { jo -> jo.getStringOrNull("vol") to jo.getStringOrNull("chap") }
+		val branches = list.associateGrouping { jo ->
+			jo.getString("lang") to jo.optJSONArray("group_name")
+				?.joinToString { it.toString() }
+				?.takeUnless { it.isBlank() }
 		}
-	}
-
-	private inline fun <R> JSONArray.mapReversed(block: (JSONObject) -> R): List<R> {
-		val len = length()
-		val destination = ArrayList<R>(len)
-		for (i in (0 until len).reversed()) {
-			val jo = getJSONObject(i)
-			destination.add(block(jo))
+		val chaptersBuilder = ChaptersListBuilder(list.size)
+		var number = 0
+		for ((key, value) in dc) {
+			number++
+			val (vol, chap) = key
+			for (jo in value) {
+				val lang = jo.getString("lang")
+				val locale = Locale.forLanguageTag(lang)
+				val team = jo.optJSONArray("group_name")?.joinToString { it.toString() }?.takeUnless { it.isBlank() }
+				var branch = locale.getDisplayName(locale).toTitleCase(locale)
+				if (branches[lang].orEmpty().size > 1 && team != null) {
+					branch += " ($team)"
+				}
+				chaptersBuilder += MangaChapter(
+					id = generateUid(jo.getLong("id")),
+					name = buildString {
+						vol?.let { append("Vol ").append(it).append(' ') }
+						chap?.let { append("Chap ").append(it) }
+						jo.getStringOrNull("title")?.let { append(": ").append(it) }
+					},
+					number = number,
+					url = jo.getString("hid"),
+					scanlator = team,
+					uploadDate = dateFormat.tryParse(jo.getString("created_at").substringBefore('T')),
+					branch = branch,
+					source = source,
+				)
+			}
 		}
-		return destination
+		return chaptersBuilder.toList()
 	}
 
 	private fun JSONObject.selectGenres(name: String, tags: SparseArrayCompat<MangaTag>): Set<MangaTag> {
diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/util/Collection.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/util/Collection.kt
index fe4ef0e33..039ed1b42 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/util/Collection.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/util/Collection.kt
@@ -64,4 +64,13 @@ inline fun <T> List<T>.areItemsEquals(other: List<T>, equals: (T, T) -> Boolean)
 	return true
 }
 
-fun <T> Iterator<T>.nextOrNull(): T? = if (hasNext()) next() else null
\ No newline at end of file
+fun <T> Iterator<T>.nextOrNull(): T? = if (hasNext()) next() else null
+
+inline fun <T, K, V> Collection<T>.associateGrouping(transform: (T) -> Pair<K, V>): Map<K, List<V>> {
+	val result = LinkedHashMap<K, MutableList<V>>(size)
+	for (item in this) {
+		val (k, v) = transform(item)
+		result.getOrPut(k) { LinkedList() }.add(v)
+	}
+	return result
+}