From 83dec28ff07f2196dc1818bd1808556f524774b2 Mon Sep 17 00:00:00 2001
From: Zakhar Timoshenko <vp1984tanki@gmail.com>
Date: Sat, 21 May 2022 01:25:52 +0300
Subject: [PATCH] [New source] Nicovideo Seiga

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

diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/NicovideoSeigaParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/NicovideoSeigaParser.kt
new file mode 100644
index 00000000..3f324b27
--- /dev/null
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/NicovideoSeigaParser.kt
@@ -0,0 +1,132 @@
+package org.koitharu.kotatsu.parsers.site
+
+import org.koitharu.kotatsu.parsers.MangaLoaderContext
+import org.koitharu.kotatsu.parsers.MangaParser
+import org.koitharu.kotatsu.parsers.MangaParserAuthProvider
+import org.koitharu.kotatsu.parsers.MangaSourceParser
+import org.koitharu.kotatsu.parsers.config.ConfigKey
+import org.koitharu.kotatsu.parsers.exception.AuthRequiredException
+import org.koitharu.kotatsu.parsers.exception.ParseException
+import org.koitharu.kotatsu.parsers.model.*
+import org.koitharu.kotatsu.parsers.util.*
+import java.util.*
+
+@MangaSourceParser("NICOVIDEOSEIGA", "Nicovideo Seiga", "ja")
+class NicovideoSeigaParser(override val context: MangaLoaderContext) : MangaParser(MangaSource.NICOVIDEOSEIGA),
+	MangaParserAuthProvider {
+
+	override val authUrl: String
+		get() = "https://account.nicovideo.jp/login?site=seiga"
+
+	override val isAuthorized: Boolean
+		get() {
+			return context.cookieJar.getCookies(getDomain()).any {
+				it.name == "user_session"
+			}
+		}
+
+	override suspend fun getUsername(): String {
+		val body = context.httpGet("https://nicovideo.jp/my").parseHtml().body()
+		return "Nicovideo User" // TODO Figure out
+	}
+
+	override val sortOrders: Set<SortOrder> = EnumSet.of(
+		SortOrder.UPDATED,
+		SortOrder.POPULARITY,
+	)
+
+	override val configKeyDomain: ConfigKey.Domain = ConfigKey.Domain("seiga.nicovideo.jp", null)
+
+	override suspend fun getList(
+		offset: Int,
+		query: String?,
+		tags: Set<MangaTag>?,
+		sortOrder: SortOrder?,
+	): List<Manga> {
+		val page = (offset / 20f).toIntUp().inc()
+		val url = "/manga/list?page=$page&sort=${getSortKey(sortOrder)}".withDomain()
+		val doc = context.httpGet(url).parseHtml()
+		val comicList = doc.body().select("#comic_list > ul > li") ?: parseFailed("Container not found")
+		val items = comicList.select("div > .description > div > div")
+		return items.mapNotNull { item ->
+			val href = item.select(".comic_icon > div > a").attr("href") ?: return@mapNotNull null
+			val statusText = item.select(".mg_description_header > .mg_icon > .content_status > span").text()
+			Manga(
+				id = generateUid(href),
+				title = item.select(".mg_body > .title > a").text() ?: return@mapNotNull null,
+				coverUrl = item.select(".comic_icon > div > a > img").attr("src"),
+				altTitle = null,
+				author = item.select(".mg_description_header > .mg_author > a").text() ?: return@mapNotNull null,
+				rating = RATING_UNKNOWN,
+				url = href,
+				isNsfw = false,
+				tags = emptySet(),
+				state = when (statusText) {
+					"連載" -> MangaState.ONGOING
+					"完結" -> MangaState.FINISHED
+					else -> null
+				},
+				publicUrl = href.toAbsoluteUrl(item.host ?: getDomain()),
+				source = source,
+			)
+		}
+	}
+
+	override suspend fun getDetails(manga: Manga): Manga {
+		val doc = context.httpGet(manga.url.withDomain()).parseHtml()
+		val contents = doc.body().select("#contents") ?: parseFailed("Cannot find root")
+		val statusText = contents.select("div.mg_work_detail > div > div:nth-child(2) > div.tip.content_status.status_series > span").text()
+		return manga.copy(
+			description = contents.select("div.mg_work_detail > div > div.row > div.description_text").text(),
+			largeCoverUrl = contents.select("div.primaries > div.main_visual > a > img").attr("src"),
+			state = when (statusText) {
+				"連載" -> MangaState.ONGOING
+				"完結" -> MangaState.FINISHED
+				else -> null
+			},
+			chapters = contents.select("#episode_list > ul > li").mapIndexedNotNull { i, li ->
+				val href = li.select("div > div.description > div.title > a").attr("href").withDomain()
+				MangaChapter(
+					id = generateUid(href),
+					name = li.select("div > div.description > div.title > a").text(),
+					number = i + 1,
+					url = href,
+					scanlator = null,
+					branch = null,
+					uploadDate = 0,
+					source = source,
+				)
+			},
+		)
+	}
+
+	override suspend fun getPages(chapter: MangaChapter): List<MangaPage> {
+		val fullUrl = chapter.url.withDomain()
+		val doc = context.httpGet(fullUrl).parseHtml()
+		if (!doc.select("#login_manga").isEmpty())
+			throw AuthRequiredException(source)
+		val root = doc.body().select("#page_contents > li")
+			?: throw ParseException("Root not found")
+		return root.map { li ->
+			val url = li.select("div > img").attr("data-original")
+			MangaPage(
+				id = generateUid(url),
+				url = url,
+				preview = null,
+				referer = fullUrl,
+				source = source,
+			)
+		}
+	}
+
+	override suspend fun getTags(): Set<MangaTag> {
+		return emptySet()
+	}
+
+	private fun getSortKey(sortOrder: SortOrder?) =
+		when (sortOrder ?: sortOrders.minByOrNull { it.ordinal }) {
+			SortOrder.POPULARITY -> "manga_view"
+			SortOrder.UPDATED -> "manga_updated"
+			else -> "rate"
+		}
+}
\ No newline at end of file