INSANEMODE/aniyomi-extensions-backup
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(lib/okru-extractor): Add support to HLS/DASH playlists (#2140)

Browse Source
This commit is contained in:
Claudemirovsky
2023-09-03 18:10:53 -03:00
committed by GitHub
parent 52c5d3ba49
commit 0624746f5f
2 changed files with 38 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lib/okru-extractor/build.gradle.kts
View File

@ -14,4 +14,5 @@ android {
dependencies {
compileOnly(libs.bundles.common)
implementation(project(":lib-playlist-utils"))
}

51
lib/okru-extractor/src/main/java/eu/kanade/tachiyomi/lib/okruextractor/OkruExtractor.kt
View File

@ -3,9 +3,11 @@ package eu.kanade.tachiyomi.lib.okruextractor
import eu.kanade.tachiyomi.animesource.model.Video
import eu.kanade.tachiyomi.network.GET
import eu.kanade.tachiyomi.util.asJsoup
import eu.kanade.tachiyomi.lib.playlistutils.PlaylistUtils
import okhttp3.OkHttpClient
class OkruExtractor(private val client: OkHttpClient) {
private val playlistUtils by lazy { PlaylistUtils(client) }
private fun fixQuality(quality: String): String {
val qualities = listOf(
@ -23,15 +25,41 @@ class OkruExtractor(private val client: OkHttpClient) {
fun videosFromUrl(url: String, prefix: String = "", fixQualities: Boolean = true): List<Video> {
val document = client.newCall(GET(url)).execute().asJsoup()
val videosString = document.selectFirst("div[data-options]")
val videoString = document.selectFirst("div[data-options]")
?.attr("data-options")
?.substringAfter("\\\"videos\\\":[{\\\"name\\\":\\\"")
?.substringBefore("]")
?: return emptyList<Video>()
return videosString.split("{\\\"name\\\":\\\"").reversed().mapNotNull {
val videoUrl = it.substringAfter("url\\\":\\\"")
.substringBefore("\\\"")
.replace("\\\\u0026", "&")
return when {
"ondemandHls" in videoString -> {
val playlistUrl = videoString.extractLink("ondemandHls")
playlistUtils.extractFromHls(playlistUrl, videoNameGen = { "Okru:$it".addPrefix(prefix) })
}
"ondemandDash" in videoString -> {
val playlistUrl = videoString.extractLink("ondemandDash")
playlistUtils.extractFromDash(playlistUrl, videoNameGen = { it -> "Okru:$it".addPrefix(prefix) })
}
else -> videosFromJson(videoString, prefix, fixQualities)
}
}
private fun String.addPrefix(prefix: String) =
prefix.takeIf(String::isNotBlank)
?.let { "$prefix $this" }
?: this
private fun String.extractLink(attr: String) =
substringAfter("$attr\\\":\\\"")
.substringBefore("\\\"")
.replace("\\\\u0026", "&")
private fun videosFromJson(videoString: String, prefix: String = "", fixQualities: Boolean = true): List<Video> {
val arrayData = videoString.substringAfter("\\\"videos\\\":[{\\\"name\\\":\\\"")
.substringBefore("]")
return arrayData.split("{\\\"name\\\":\\\"").reversed().mapNotNull {
val videoUrl = it.extractLink("url")
val quality = it.substringBefore("\\\"").let {
if (fixQualities) {
fixQuality(it)
@ -39,13 +67,8 @@ class OkruExtractor(private val client: OkHttpClient) {
it
}
}
val videoQuality = ("Okru:" + quality).let {
if (prefix.isNotBlank()) {
"$prefix $it"
} else {
it
}
}
val videoQuality = "Okru:$quality".addPrefix(prefix)
if (videoUrl.startsWith("https://")) {
Video(videoUrl, videoQuality, videoUrl)
} else {