cloudstream-extensions/TenshiProvider/src/main/kotlin/com/lagradost/TenshiProvider.kt

354 lines
14 KiB
Kotlin

package com.lagradost
import android.annotation.SuppressLint
import com.fasterxml.jackson.annotation.JsonProperty
import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.network.DdosGuardKiller
import com.lagradost.cloudstream3.network.getHeaders
import com.lagradost.cloudstream3.utils.AppUtils.parseJson
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.getQualityFromName
import org.jsoup.nodes.Document
import java.net.URI
import java.text.SimpleDateFormat
import java.util.*
class TenshiProvider : MainAPI() {
companion object {
//var token: String? = null
//var cookie: Map<String, String> = mapOf()
fun getType(t: String): TvType {
return if (t.contains("OVA") || t.contains("Special")) TvType.OVA
else if (t.contains("Movie")) TvType.AnimeMovie
else TvType.Anime
}
}
override var mainUrl = "https://tenshi.moe"
override var name = "Tenshi.moe"
override val hasQuickSearch = false
override val hasMainPage = true
override val supportedTypes = setOf(TvType.Anime, TvType.AnimeMovie, TvType.OVA)
private var ddosGuardKiller = DdosGuardKiller(true)
/*private fun loadToken(): Boolean {
return try {
val response = get(mainUrl)
cookie = response.cookies
val document = Jsoup.parse(response.text)
token = document.selectFirst("""meta[name="csrf-token"]""").attr("content")
token != null
} catch (e: Exception) {
false
}
}*/
override suspend fun getMainPage(page: Int, request : MainPageRequest): HomePageResponse {
val items = ArrayList<HomePageList>()
val soup = app.get(mainUrl, interceptor = ddosGuardKiller).document
for (section in soup.select("#content > section")) {
try {
if (section.attr("id") == "toplist-tabs") {
for (top in section.select(".tab-content > [role=\"tabpanel\"]")) {
val title = "Top - " + top.attr("id").split("-")[1].replaceFirstChar {
if (it.isLowerCase()) it.titlecase(
Locale.UK
) else it.toString()
}
val anime = top.select("li > a").map {
AnimeSearchResponse(
it.selectFirst(".thumb-title")!!.text(),
fixUrl(it.attr("href")),
this.name,
TvType.Anime,
it.selectFirst("img")!!.attr("src"),
null,
EnumSet.of(DubStatus.Subbed),
)
}
items.add(HomePageList(title, anime))
}
} else {
val title = section.selectFirst("h2")!!.text()
val anime = section.select("li > a").map {
AnimeSearchResponse(
it.selectFirst(".thumb-title")?.text() ?: "",
fixUrl(it.attr("href")),
this.name,
TvType.Anime,
it.selectFirst("img")!!.attr("src"),
null,
EnumSet.of(DubStatus.Subbed),
)
}
items.add(HomePageList(title, anime))
}
} catch (e: Exception) {
e.printStackTrace()
}
}
if (items.size <= 0) throw ErrorLoadingException()
return HomePageResponse(items)
}
private fun getIsMovie(type: String, id: Boolean = false): Boolean {
if (!id) return type == "Movie"
val movies = listOf("rrso24fa", "e4hqvtym", "bl5jdbqn", "u4vtznut", "37t6h2r4", "cq4azcrj")
val aniId = type.replace("$mainUrl/anime/", "")
return movies.contains(aniId)
}
private fun parseSearchPage(soup: Document): List<SearchResponse> {
val items = soup.select("ul.thumb > li > a")
return items.map {
val href = fixUrl(it.attr("href"))
val img = fixUrl(it.selectFirst("img")!!.attr("src"))
val title = it.attr("title")
if (getIsMovie(href, true)) {
MovieSearchResponse(
title, href, this.name, TvType.Movie, img, null
)
} else {
AnimeSearchResponse(
title,
href,
this.name,
TvType.Anime,
img,
null,
EnumSet.of(DubStatus.Subbed),
)
}
}
}
@SuppressLint("SimpleDateFormat")
private fun dateParser(dateString: String?): Date? {
if (dateString == null) return null
try {
val format = SimpleDateFormat("dd 'of' MMM',' yyyy")
val data = format.parse(
dateString.replace("th ", " ").replace("st ", " ").replace("nd ", " ")
.replace("rd ", " ")
) ?: return null
return data
} catch (e: Exception) {
return null
}
}
// data class TenshiSearchResponse(
// @JsonProperty("url") var url : String,
// @JsonProperty("title") var title : String,
// @JsonProperty("cover") var cover : String,
// @JsonProperty("genre") var genre : String,
// @JsonProperty("year") var year : Int,
// @JsonProperty("type") var type : String,
// @JsonProperty("eps") var eps : String,
// @JsonProperty("cen") var cen : String
// )
// override suspend fun quickSearch(query: String): ArrayList<SearchResponse>? {
// if (!autoLoadToken()) return quickSearch(query)
// val url = "$mainUrl/anime/search"
// val response = khttp.post(
// url,
// data=mapOf("q" to query),
// headers=mapOf("x-csrf-token" to token, "x-requested-with" to "XMLHttpRequest"),
// cookies = cookie
//
// )
//
// val items = mapper.readValue<List<TenshiSearchResponse>>(response.text)
//
// if (items.isEmpty()) return ArrayList()
//
// val returnValue = ArrayList<SearchResponse>()
// for (i in items) {
// val href = fixUrl(i.url)
// val title = i.title
// val img = fixUrl(i.cover)
// val year = i.year
//
// returnValue.add(
// if (getIsMovie(i.type)) {
// MovieSearchResponse(
// title, href, getSlug(href), this.name, TvType.Movie, img, year
// )
// } else {
// AnimeSearchResponse(
// title, href, getSlug(href), this.name,
// TvType.Anime, img, year, null,
// EnumSet.of(DubStatus.Subbed),
// null, null
// )
// }
// )
// }
// return returnValue
// }
override suspend fun search(query: String): List<SearchResponse> {
val url = "$mainUrl/anime"
var document = app.get(
url,
params = mapOf("q" to query),
cookies = mapOf("loop-view" to "thumb"),
interceptor = ddosGuardKiller
).document
val returnValue = parseSearchPage(document).toMutableList()
while (!document.select("""a.page-link[rel="next"]""").isEmpty()) {
val link = document.selectFirst("""a.page-link[rel="next"]""")?.attr("href")
if (!link.isNullOrBlank()) {
document = app.get(
link,
cookies = mapOf("loop-view" to "thumb"),
interceptor = ddosGuardKiller
).document
returnValue.addAll(parseSearchPage(document))
} else {
break
}
}
return returnValue
}
override suspend fun load(url: String): LoadResponse {
var document = app.get(
url,
cookies = mapOf("loop-view" to "thumb"),
interceptor = ddosGuardKiller
).document
val canonicalTitle = document.selectFirst("header.entry-header > h1.mb-3")!!.text().trim()
val episodeNodes = document.select("li[class*=\"episode\"] > a").toMutableList()
val totalEpisodePages = if (document.select(".pagination").size > 0)
document.select(".pagination .page-item a.page-link:not([rel])").last()!!.text()
.toIntOrNull()
else 1
if (totalEpisodePages != null && totalEpisodePages > 1) {
for (pageNum in 2..totalEpisodePages) {
document = app.get(
"$url?page=$pageNum",
cookies = mapOf("loop-view" to "thumb"),
interceptor = ddosGuardKiller
).document
episodeNodes.addAll(document.select("li[class*=\"episode\"] > a"))
}
}
val episodes = ArrayList(episodeNodes.map {
val title = it.selectFirst(".episode-title")?.text()?.trim()
newEpisode(it.attr("href")) {
this.name = if (title == "No Title") null else title
this.posterUrl = it.selectFirst("img")?.attr("src")
addDate(dateParser(it?.selectFirst(".episode-date")?.text()?.trim()))
this.description = it.attr("data-content").trim()
}
})
val similarAnime = document.select("ul.anime-loop > li > a").mapNotNull { element ->
val href = element.attr("href") ?: return@mapNotNull null
val title =
element.selectFirst("> .overlay > .thumb-title")?.text() ?: return@mapNotNull null
val img = element.selectFirst("> img")?.attr("src")
AnimeSearchResponse(title, href, this.name, TvType.Anime, img)
}
val type = document.selectFirst("a[href*=\"$mainUrl/type/\"]")?.text()?.trim()
return newAnimeLoadResponse(canonicalTitle, url, getType(type ?: "")) {
recommendations = similarAnime
posterUrl = document.selectFirst("img.cover-image")?.attr("src")
plot = document.selectFirst(".entry-description > .card-body")?.text()?.trim()
tags =
document.select("li.genre.meta-data > span.value")
.map { it?.text()?.trim().toString() }
synonyms =
document.select("li.synonym.meta-data > div.info-box > span.value")
.map { it?.text()?.trim().toString() }
engName =
document.selectFirst("span.value > span[title=\"English\"]")?.parent()?.text()
?.trim()
japName =
document.selectFirst("span.value > span[title=\"Japanese\"]")?.parent()?.text()
?.trim()
val pattern = Regex("(\\d{4})")
document.selectFirst("li.release-date .value")?.text()?.let {
year = pattern.find(it)?.groupValues?.get(1)?.toIntOrNull()
}
addEpisodes(DubStatus.Subbed, episodes)
showStatus = when (document.selectFirst("li.status > .value")?.text()?.trim()) {
"Ongoing" -> ShowStatus.Ongoing
"Completed" -> ShowStatus.Completed
else -> null
}
}
}
override suspend fun loadLinks(
data: String,
isCasting: Boolean,
subtitleCallback: (SubtitleFile) -> Unit,
callback: (ExtractorLink) -> Unit
): Boolean {
val soup = app.get(data, interceptor = ddosGuardKiller).document
data class Quality(
@JsonProperty("src") val src: String,
@JsonProperty("size") val size: Int
)
for (source in soup.select("""[aria-labelledby="mirror-dropdown"] > li > a.dropdown-item""")) {
val release = source.text().replace("/", "").trim()
val sourceHTML = app.get(
"https://tenshi.moe/embed?v=${source.attr("href").split("v=")[1].split("&")[0]}",
headers = mapOf("Referer" to data), interceptor = ddosGuardKiller
).text
val match = Regex("""sources: (\[(?:.|\s)+?type: ['"]video/.*?['"](?:.|\s)+?])""").find(
sourceHTML
)
if (match != null) {
val qualities = parseJson<List<Quality>>(
match.destructured.component1()
.replace("'", "\"")
.replace(Regex("""(\w+): """), "\"\$1\": ")
.replace(Regex("""\s+"""), "")
.replace(",}", "}")
.replace(",]", "]")
)
qualities.forEach {
callback.invoke(
ExtractorLink(
this.name,
"${this.name} $release",
fixUrl(it.src),
this.mainUrl,
getQualityFromName("${it.size}"),
headers = getHeaders(emptyMap(),
ddosGuardKiller.savedCookiesMap[URI(this.mainUrl).host]
?: emptyMap()
).toMap()
)
)
}
}
}
return true
}
}