From 1cecd1fe94932c6a3d8d8af276b2ee61c05d047a Mon Sep 17 00:00:00 2001
From: scryptan <scryptan@gmail.com>
Date: Thu, 23 Jan 2025 13:31:07 +0500
Subject: [PATCH] add wamanga parser

---
 .../kotatsu/parsers/site/ru/WaMangaParser.kt  | 135 ++++++++++++++++++
 1 file changed, 135 insertions(+)
 create mode 100644 src/main/kotlin/org/koitharu/kotatsu/parsers/site/ru/WaMangaParser.kt

diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ru/WaMangaParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ru/WaMangaParser.kt
new file mode 100644
index 00000000..db6dd27a
--- /dev/null
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/ru/WaMangaParser.kt
@@ -0,0 +1,135 @@
+package org.koitharu.kotatsu.parsers.site.ru
+
+import org.json.JSONObject
+import org.koitharu.kotatsu.parsers.MangaLoaderContext
+import org.koitharu.kotatsu.parsers.MangaSourceParser
+import org.koitharu.kotatsu.parsers.SinglePageMangaParser
+import org.koitharu.kotatsu.parsers.config.ConfigKey
+import org.koitharu.kotatsu.parsers.model.*
+import org.koitharu.kotatsu.parsers.util.*
+import org.koitharu.kotatsu.parsers.util.json.*
+import java.text.SimpleDateFormat
+import java.util.*
+
+@MangaSourceParser("WAMANGA", "WaManga", "ru", type = ContentType.MANGA)
+internal class WaMangaParser(
+	context: MangaLoaderContext,
+) : SinglePageMangaParser(context, MangaParserSource.WAMANGA) {
+
+	override val configKeyDomain = ConfigKey.Domain("wamanga.me")
+
+	override val availableSortOrders: Set<SortOrder> = EnumSet.of(SortOrder.ALPHABETICAL)
+
+	override val filterCapabilities: MangaListFilterCapabilities
+		get() = MangaListFilterCapabilities(
+			isMultipleTagsSupported = true,
+			isSearchSupported = true,
+		)
+
+	override suspend fun getFilterOptions() = MangaListFilterOptions(availableTags = fetchAvailableTags())
+
+	override suspend fun getList(order: SortOrder, filter: MangaListFilter): List<Manga> {
+		return parseMangaList(webClient.httpGet("https://$domain/api/comics").parseJson())
+	}
+
+	private fun parseMangaList(docs: JSONObject): List<Manga> {
+		return docs.getJSONArray("comics").mapJSONNotNull { parseSmallMangaObject(it) }
+	}
+
+	private fun parseMangaTag(doc: JSONObject): MangaTag {
+		return MangaTag(
+			doc.getString("name")
+				.replaceFirstChar { if (it.isLowerCase()) it.titlecase(Locale.getDefault()) else it.toString() },
+			doc.getString("slug"),
+			MangaParserSource.WAMANGA,
+		)
+	}
+
+	private fun parseSmallMangaObject(doc: JSONObject): Manga {
+		val url = "https://$domain${doc.getString("url")}"
+		return Manga(
+			id = generateUid(doc.getString("url")),
+			url = doc.getString("url"),
+			title = doc.getString("title"),
+			altTitle = null,
+			publicUrl = url,
+			rating = doc.getFloatOrDefault("rating", 0f),
+			coverUrl = doc.getString("thumbnail_small"),
+			tags = doc.getJSONArray("genres").mapJSONToSet { tag -> parseMangaTag(tag) },
+			state = when (doc.getString("status").lowercase(Locale.getDefault())) {
+				"продолжается" -> MangaState.ONGOING
+				"окончен" -> MangaState.FINISHED
+				"закончен" -> MangaState.FINISHED
+				else -> MangaState.UPCOMING
+			},
+			author = doc.getStringOrNull("author"),
+			source = source,
+			contentRating = if (doc.getIntOrDefault("adult", 0) == 0) ContentRating.SAFE else ContentRating.ADULT,
+		)
+	}
+
+
+	override suspend fun getDetails(manga: Manga): Manga {
+		val url = "https://$domain/api${manga.url}"
+		val doc = webClient.httpGet(url).parseJson().getJSONObject("comic")
+
+		val dateFormat = SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'", sourceLocale)
+		val thumbnailUrl = doc.getString("thumbnail")
+		val mangaNamePath =
+			thumbnailUrl.slice(thumbnailUrl.indexOf(doc.getString("slug"))..<thumbnailUrl.length).split('/')[0]
+
+		val chapterBaseUrl = "https://$domain/public/storage/comics/$mangaNamePath"
+		val html = webClient.httpGet(chapterBaseUrl).parseHtml()
+		val chapters = html.getElementsByTag("a").map { it.ownText() }
+		return manga.copy(
+			id = generateUid(manga.url),
+			url = doc.getString("url"),
+			title = doc.getString("title"),
+			largeCoverUrl = thumbnailUrl,
+			description = doc.getString("description") ?: manga.description,
+			chapters = doc.getJSONArray("chapters").mapJSONNotNull {
+				val chapterPrefix = it.getString("slug_lang_vol_ch_sub")
+				val chapUrl = chapters.first { chap -> chap.startsWith(chapterPrefix) }
+				val fullChapUrl = "$chapterBaseUrl/$chapUrl"
+				MangaChapter(
+					id = generateUid(fullChapUrl),
+					url = fullChapUrl,
+					source = source,
+					number = it.getIntOrDefault("chapter", 0).toFloat(),
+					volume = it.getIntOrDefault("volume", 0),
+					name = it.getStringOrNull("full_title") ?: manga.title,
+					scanlator = it.getJSONArray("teams").optJSONObject(0, null)?.getStringOrNull("name"),
+					uploadDate = dateFormat.tryParse(
+						it.getStringOrNull("published_on"),
+					),
+					branch = null,
+				)
+			},
+		)
+	}
+
+	override suspend fun getPages(chapter: MangaChapter): List<MangaPage> {
+		val doc = webClient.httpGet(chapter.url).parseHtml()
+		val pages = doc.getElementsByTag("a").map { it.ownText() }
+		return pages.drop(1).map { img ->
+			val url = "${chapter.url}$img"
+			MangaPage(
+				id = generateUid(url),
+				url = url,
+				preview = null,
+				source = source,
+			)
+		}.toList()
+	}
+
+	private suspend fun fetchAvailableTags(): Set<MangaTag> {
+		val doc = webClient.httpGet("https://$domain/api/comics").parseJson()
+		return doc
+			.getJSONArray("comics")
+			.mapJSONNotNull { it.getJSONArray("genres").mapJSONToSet { tag -> parseMangaTag(tag) } }
+			.flatten()
+			.distinctBy { it.key }
+			.filter { it.key != "" }
+			.toSet()
+	}
+}