[Shinigami] Fix search with tags + Refactor

master
dragonx943 1 year ago
parent 6418422157
commit fecb1db3be

@ -16,249 +16,272 @@ import org.koitharu.kotatsu.parsers.Broken
@Broken("TODO: Add author search") @Broken("TODO: Add author search")
@MangaSourceParser("SHINIGAMI", "Shinigami", "id") @MangaSourceParser("SHINIGAMI", "Shinigami", "id")
internal class Shinigami(context: MangaLoaderContext) : internal class Shinigami(context: MangaLoaderContext) :
LegacyPagedMangaParser(context, MangaParserSource.SHINIGAMI, 24) { LegacyPagedMangaParser(context, MangaParserSource.SHINIGAMI, 24) {
override val configKeyDomain = ConfigKey.Domain("id.shinigami.asia") override val configKeyDomain = ConfigKey.Domain("id.shinigami.asia")
private val apiSuffix = "api.shngm.io/v1" private val apiSuffix = "api.shngm.io/v1"
private val cdnSuffix = "storage.shngm.id" private val cdnSuffix = "storage.shngm.id"
override fun onCreateConfig(keys: MutableCollection<ConfigKey<*>>) { override fun onCreateConfig(keys: MutableCollection<ConfigKey<*>>) {
super.onCreateConfig(keys) super.onCreateConfig(keys)
keys.add(userAgentKey) keys.add(userAgentKey)
} }
override fun getRequestHeaders(): Headers = Headers.Builder() override fun getRequestHeaders(): Headers = Headers.Builder()
.add("referer", "https://$domain/") .add("referer", "https://$domain/")
.add("sec-fetch-dest", "empty") .add("sec-fetch-dest", "empty")
.build() .build()
override val availableSortOrders: Set<SortOrder> = EnumSet.of( override val availableSortOrders: Set<SortOrder> = EnumSet.of(
SortOrder.POPULARITY, SortOrder.POPULARITY,
SortOrder.POPULARITY_ASC, SortOrder.POPULARITY_ASC,
SortOrder.NEWEST, SortOrder.NEWEST,
SortOrder.NEWEST_ASC, SortOrder.NEWEST_ASC,
SortOrder.RATING, SortOrder.RATING,
SortOrder.RATING_ASC, SortOrder.RATING_ASC,
) )
override suspend fun getFilterOptions(): MangaListFilterOptions { override suspend fun getFilterOptions(): MangaListFilterOptions {
return MangaListFilterOptions( return MangaListFilterOptions(
availableTags = fetchTags(), availableTags = fetchTags(),
availableStates = EnumSet.of(MangaState.ONGOING, MangaState.FINISHED, MangaState.PAUSED), availableStates = EnumSet.of(
availableContentTypes = EnumSet.of( MangaState.ONGOING,
ContentType.MANGA, MangaState.FINISHED,
ContentType.MANHWA, MangaState.PAUSED
ContentType.MANHUA ),
), availableContentTypes = EnumSet.of(
) ContentType.MANGA,
} ContentType.MANHWA,
ContentType.MANHUA
),
)
}
override val filterCapabilities: MangaListFilterCapabilities override val filterCapabilities: MangaListFilterCapabilities
get() = MangaListFilterCapabilities( get() = MangaListFilterCapabilities(
isMultipleTagsSupported = true, isMultipleTagsSupported = true,
isTagsExclusionSupported = true, isTagsExclusionSupported = true,
isSearchSupported = true, isSearchSupported = true,
) isSearchWithFiltersSupported = true,
)
override suspend fun getListPage(page: Int, order: SortOrder, filter: MangaListFilter): List<Manga> { override suspend fun getListPage(page: Int, order: SortOrder, filter: MangaListFilter): List<Manga> {
val url = buildString { val url = buildString {
append("https://") append("https://")
append(apiSuffix) append(apiSuffix)
append("/manga/list") append("/manga/list")
append("?page=") append("?page=")
append(page.toString()) append(page.toString())
append("&page_size=") append("&page_size=")
append(pageSize) append(pageSize)
append("&sort=") append("&sort=")
append( append(
when (order) { when (order) {
SortOrder.POPULARITY, SortOrder.POPULARITY_ASC -> "popularity" SortOrder.POPULARITY, SortOrder.POPULARITY_ASC -> "popularity"
SortOrder.NEWEST, SortOrder.NEWEST_ASC -> "latest" SortOrder.NEWEST, SortOrder.NEWEST_ASC -> "latest"
SortOrder.RATING, SortOrder.RATING_ASC -> "rating" SortOrder.RATING, SortOrder.RATING_ASC -> "rating"
else -> "latest" else -> "latest"
} }
) )
append("&sort_order=") append("&sort_order=")
append( append(
when (order) { when (order) {
SortOrder.POPULARITY_ASC, SortOrder.NEWEST_ASC, SortOrder.RATING_ASC -> "asc" SortOrder.POPULARITY_ASC, SortOrder.NEWEST_ASC, SortOrder.RATING_ASC -> "asc"
else -> "desc" else -> "desc"
} }
) )
filter.states.oneOrThrowIfMany()?.let { filter.states.oneOrThrowIfMany()?.let {
append("&status=") append("&status=")
append( append(
when (it) { when (it) {
MangaState.FINISHED -> "completed" MangaState.FINISHED -> "completed"
MangaState.ONGOING -> "ongoing" MangaState.ONGOING -> "ongoing"
MangaState.PAUSED -> "hiatus" MangaState.PAUSED -> "hiatus"
else -> "" else -> ""
} }
) )
} }
if (filter.types.isNotEmpty()) { if (filter.types.isNotEmpty()) {
filter.types.forEach { filter.types.forEach {
append("&format=") append("&format=")
append( append(
when (it) { when (it) {
ContentType.MANGA -> "manga" ContentType.MANGA -> "manga"
ContentType.MANHUA -> "manhua" ContentType.MANHUA -> "manhua"
ContentType.MANHWA -> "manhwa" ContentType.MANHWA -> "manhwa"
else -> "" else -> ""
} }
) )
} }
} }
if (filter.tags.isNotEmpty()) { if (filter.tags.isNotEmpty()) {
append("&genres_include=") append("&genre_include=")
filter.tags.joinTo(this, ",") { it.key } filter.tags.joinTo(this, separator = ",") { it.key }
append("&genres_include_mode=and") append("&genre_include_mode=and")
} }
if (filter.tagsExclude.isNotEmpty()) { if (filter.tagsExclude.isNotEmpty()) {
append("&genres_exclude=") append("&genre_exclude=")
filter.tagsExclude.joinTo(this, ",") { it.key } filter.tagsExclude.joinTo(this, separator = ",") { it.key }
append("&genres_exclude_mode=and") append("&genre_exclude_mode=and")
} }
if (!filter.query.isNullOrEmpty()) { if (!filter.query.isNullOrEmpty()) {
append("&q=") append("&q=")
val encodedQuery = filter.query.splitByWhitespace().joinToString(separator = "+") { part -> val encodedQuery = filter.query.splitByWhitespace()
part.urlEncoded() .joinToString(separator = "+") { part ->
} part.urlEncoded()
append(encodedQuery) }
} append(encodedQuery)
} }
}
val json = webClient.httpGet(url).parseJson() val json = webClient.httpGet(url).parseJson()
val data = json.getJSONArray("data") val data = json.getJSONArray("data")
return data.mapJSON { jo ->
val id = jo.getString("manga_id")
Manga(
id = generateUid(id),
url = "/manga/detail/$id",
publicUrl = "https://$domain/series/$id",
title = jo.getString("title"),
altTitles = setOf(jo.optString("alternative_title") ?: ""),
coverUrl = jo.getString("cover_image_url"),
largeCoverUrl = jo.optString("cover_portrait_url").takeIf { it.isNotEmpty() },
authors = jo.optJSONObject("taxonomy")?.optJSONArray("Author")?.mapJSONToSet { x ->
x.getString("name")
}.orEmpty(),
tags = jo.optJSONObject("taxonomy")?.optJSONArray("Genre")?.mapJSONToSet { x ->
MangaTag(
key = x.getString("slug"),
title = x.getString("name"),
source = source
)
}.orEmpty(),
state = when (jo.getInt("status")) {
1 -> MangaState.ONGOING
2 -> MangaState.FINISHED
3 -> MangaState.PAUSED
else -> null
},
description = jo.optString("description"),
contentRating = null,
source = source,
rating = RATING_UNKNOWN
)
}
}
override suspend fun getDetails(manga: Manga): Manga { return data.mapJSON { jo ->
val json = webClient.httpGet("https://$apiSuffix" + manga.url).parseJson() val id = jo.getString("manga_id")
val jo = json.getJSONObject("data") Manga(
val id = jo.getString("manga_id") id = generateUid(id),
return manga.copy( url = "/manga/detail/$id",
tags = jo.optJSONObject("taxonomy")?.optJSONArray("Genre")?.mapJSONToSet { x -> publicUrl = "https://$domain/series/$id",
MangaTag( title = jo.getString("title"),
key = x.getString("slug"), altTitles = setOf(jo.optString("alternative_title") ?: ""),
title = x.getString("name"), coverUrl = jo.getString("cover_image_url"),
source = source, largeCoverUrl = jo.optString("cover_portrait_url")
) .takeIf { it.isNotEmpty() },
}.orEmpty(), authors = jo.optJSONObject("taxonomy")
authors = jo.optJSONObject("taxonomy")?.optJSONArray("Author")?.mapJSONToSet { x -> ?.optJSONArray("Author")
x.getString("name") ?.mapJSONToSet { x ->
}.orEmpty(), x.getString("name")
state = when (jo.getInt("status")) { }.orEmpty(),
1 -> MangaState.ONGOING tags = jo.optJSONObject("taxonomy")
2 -> MangaState.FINISHED ?.optJSONArray("Genre")
3 -> MangaState.PAUSED ?.mapJSONToSet { x ->
else -> null MangaTag(
}, key = x.getString("slug"),
description = jo.optString("description"), title = x.getString("name"),
largeCoverUrl = jo.optString("cover_portrait_url").takeIf { it.isNotEmpty() }, source = source
chapters = getChapters(id) )
) }.orEmpty(),
} state = when (jo.getInt("status")) {
1 -> MangaState.ONGOING
2 -> MangaState.FINISHED
3 -> MangaState.PAUSED
else -> null
},
description = jo.optString("description"),
contentRating = null,
source = source,
rating = RATING_UNKNOWN
)
}
}
private suspend fun getChapters(mangaId: String): List<MangaChapter> { override suspend fun getDetails(manga: Manga): Manga {
val url = "https://$apiSuffix/chapter/$mangaId/list?page=1&page_size=9999&sort_by=chapter_number&sort_order=asc" val json = webClient.httpGet("https://$apiSuffix" + manga.url).parseJson()
val json = webClient.httpGet(url).parseJson() val jo = json.getJSONObject("data")
val data = json.getJSONArray("data") val id = jo.getString("manga_id")
return data.mapJSON { jo -> return manga.copy(
val chapterId = jo.getString("chapter_id") tags = jo.optJSONObject("taxonomy")
val number = jo.getInt("chapter_number") ?.optJSONArray("Genre")
val title = jo.optString("chapter_title").takeIf { it.isNotEmpty() } ?.mapJSONToSet { x ->
?: "Chapter $number" MangaTag(
key = x.getString("slug"),
title = x.getString("name"),
source = source,
)
}.orEmpty(),
authors = jo.optJSONObject("taxonomy")
?.optJSONArray("Author")
?.mapJSONToSet { x ->
x.getString("name")
}.orEmpty(),
state = when (jo.getInt("status")) {
1 -> MangaState.ONGOING
2 -> MangaState.FINISHED
3 -> MangaState.PAUSED
else -> null
},
description = jo.optString("description"),
largeCoverUrl = jo.optString("cover_portrait_url")
.takeIf { it.isNotEmpty() },
chapters = getChapters(id)
)
}
MangaChapter( private suspend fun getChapters(mangaId: String): List<MangaChapter> {
id = generateUid(chapterId), val url = "https://$apiSuffix/chapter/$mangaId/list?page=1&page_size=9999&sort_by=chapter_number&sort_order=asc"
title = title, val json = webClient.httpGet(url).parseJson()
number = number.toFloat(), val data = json.getJSONArray("data")
url = "chapter/detail/$chapterId",
scanlator = null,
uploadDate = jo.getString("release_date").parseDate(),
branch = null,
source = source,
volume = 0
)
}
}
override suspend fun getPages(chapter: MangaChapter): List<MangaPage> { return data.mapJSON { jo ->
val json = webClient.httpGet("https://$apiSuffix/${chapter.url}").parseJson() val chapterId = jo.getString("chapter_id")
val data = json.getJSONObject("data") val number = jo.getInt("chapter_number")
val chapterData = data.getJSONObject("chapter") val title = jo.optString("chapter_title")
val basePath = chapterData.getString("path") .takeIf { it.isNotEmpty() }
val datas = chapterData.getJSONArray("data") ?: "Chapter $number"
return List(datas.length()) { i -> MangaChapter(
val imgExt = datas.getString(i) id = generateUid(chapterId),
val imageUrl = "https://$cdnSuffix" + basePath + imgExt title = title,
MangaPage( number = number.toFloat(),
id = generateUid(imageUrl), url = "chapter/detail/$chapterId",
url = imageUrl, scanlator = null,
preview = null, uploadDate = jo.getString("release_date").parseDate(),
source = source, branch = null,
) source = source,
} volume = 0
} )
}
}
private suspend fun fetchTags(): Set<MangaTag> { override suspend fun getPages(chapter: MangaChapter): List<MangaPage> {
val json = webClient.httpGet("https://$apiSuffix/genre/list").parseJson() val json = webClient.httpGet("https://$apiSuffix/${chapter.url}").parseJson()
return json.getJSONArray("data").mapJSONToSet { x -> val data = json.getJSONObject("data")
MangaTag( val chapterData = data.getJSONObject("chapter")
key = x.getString("slug"), val basePath = chapterData.getString("path")
title = x.getString("name"), val datas = chapterData.getJSONArray("data")
source = source,
)
}
}
private fun String.parseDate(): Long { return List(datas.length()) { i ->
return try { val imgExt = datas.getString(i)
SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'", Locale.US).apply { val imageUrl = "https://$cdnSuffix" + basePath + imgExt
timeZone = TimeZone.getTimeZone("UTC")
}.parse(this)?.time ?: 0L MangaPage(
} catch (e: Exception) { id = generateUid(imageUrl),
0L url = imageUrl,
} preview = null,
} source = source,
)
}
}
private suspend fun fetchTags(): Set<MangaTag> {
val json = webClient.httpGet("https://$apiSuffix/genre/list").parseJson()
return json.getJSONArray("data").mapJSONToSet { x ->
MangaTag(
key = x.getString("slug"),
title = x.getString("name"),
source = source,
)
}
}
private fun String.parseDate(): Long {
return try {
SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'", Locale.US)
.apply {
timeZone = TimeZone.getTimeZone("UTC")
}
.parse(this)?.time ?: 0L
} catch (e: Exception) {
0L
}
}
} }
Loading…
Cancel
Save