Draff/TachiyomiSY-Plus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve recommendations (#14)

Browse Source 
* Improve anilist recommendations by matching synonyms; Some bug fixes related to anilist

* Fix formatting

* Sort myanimelist results by result.title matching search.title

* Throw an exception if the result is not the manga we searched

Co-authored-by: she11sh0cked <she11sh0cked@users.noreply.github.com>
This commit is contained in:
she11sh0cked 2020-05-19 19:29:42 +02:00 committed by GitHub
parent 7e65e0de0e
commit 7ce15caded
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 57 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
app/src/main/java/eu/kanade/tachiyomi/ui/browse/source/browse/RecommendsPager.kt
View File

@ -7,6 +7,7 @@ import com.github.salomonbrys.kotson.jsonObject
import com.github.salomonbrys.kotson.nullString
import com.github.salomonbrys.kotson.obj
import com.github.salomonbrys.kotson.string
import com.google.gson.JsonArray
import com.google.gson.JsonParser
import eu.kanade.tachiyomi.data.database.models.Manga
import eu.kanade.tachiyomi.network.asObservableSuccess
@ -29,6 +30,13 @@ open class RecommendsPager(
) : Pager() {
private val client = OkHttpClient.Builder().build()
private fun countOccurance(array: JsonArray, search: String): Int {
return array.count {
val synonym = it.string
synonym.contains(search, true)
}
}
private fun myAnimeList(): Observable<List<SMangaImpl>>? {
fun getId(): Observable<String> {
val endpoint =
@ -54,8 +62,16 @@ open class RecommendsPager(
}
val response = JsonParser.parseString(responseBody).obj
val results = response["results"].array
val firstResult = results[0].obj
val id = firstResult["mal_id"].string
.sortedBy {
val title = it["title"].string
title.contains(manga.title, true)
}
val result = results.last()
val title = result["title"].string
if (!title.contains(manga.title, true)) {
throw Exception("Not found")
}
val id = result["mal_id"].string
if (id.isEmpty()) {
throw Exception("Not found")
}
@ -104,12 +120,14 @@ open class RecommendsPager(
val query =
"""
{
Media(search: "$manga.title", type: MANGA) {
title{
Page {
media(search: "${manga.title}", type: MANGA) {
title {
romaji
english
native
}
synonyms
recommendations {
edges {
node {
@ -117,6 +135,8 @@ open class RecommendsPager(
siteUrl
title {
romaji
english
native
}
coverImage {
large
@ -127,6 +147,7 @@ open class RecommendsPager(
}
}
}
}
""".trimIndent()
val variables = jsonObject()
val payload = jsonObject(
@ -148,8 +169,24 @@ open class RecommendsPager(
}
val response = JsonParser.parseString(responseBody).obj
val data = response["data"]!!.obj
val media = data["Media"].obj
val recommendations = media["recommendations"].obj
val page = data["Page"].obj
val media = page["media"].array
val results = media.sortedBy {
val synonyms = it["synonyms"].array
countOccurance(synonyms, manga.title)
}
val result = results.last()
val title = result["title"].obj
val synonyms = result["synonyms"].array
if (
title["romaji"].nullString?.contains("", true) != true &&
title["english"].nullString?.contains("", true) != true &&
title["native"].nullString?.contains("", true) != true &&
countOccurance(synonyms, manga.title) <= 0
) {
throw Exception("Not found")
}
val recommendations = result["recommendations"].obj
val edges = recommendations["edges"].array
edges.map {
val rec = it["node"]["mediaRecommendation"].obj