feat(lib): Convert StreamlareExtractor to shared lib (#1907)

2023-07-16 11:18:52 +00:00 · 2023-07-16 11:18:52 +00:00 · 451fd85cc2
commit 451fd85cc2
parent e8674107d3
2 changed files with 74 additions and 0 deletions
--- a/lib/streamlare-extractor/build.gradle.kts
+++ b/lib/streamlare-extractor/build.gradle.kts
@ -0,0 +1,17 @@
+plugins {
+    id("com.android.library")
+    kotlin("android")
+}
+
+android {
+    compileSdk = AndroidConfig.compileSdk
+    namespace = "eu.kanade.tachiyomi.lib.streamlareextractor"
+
+    defaultConfig {
+        minSdk = AndroidConfig.minSdk
+    }
+}
+
+dependencies {
+    compileOnly(libs.bundles.common)
+}
--- a/lib/streamlare-extractor/src/main/java/eu/kanade/tachiyomi/lib/streamlareextractor/StreamlareExtractor.kt
+++ b/lib/streamlare-extractor/src/main/java/eu/kanade/tachiyomi/lib/streamlareextractor/StreamlareExtractor.kt
@ -0,0 +1,57 @@
+package eu.kanade.tachiyomi.lib.streamlareextractor
+
+import eu.kanade.tachiyomi.animesource.model.Video
+import eu.kanade.tachiyomi.network.GET
+import eu.kanade.tachiyomi.network.POST
+import okhttp3.MediaType.Companion.toMediaType
+import okhttp3.OkHttpClient
+import okhttp3.RequestBody.Companion.toRequestBody
+
+class StreamlareExtractor(private val client: OkHttpClient) {
+    fun videosFromUrl(url: String, prefix: String = "", suffix: String = ""): List<Video> {
+        val id = url.split("/").last()
+        val playlist = client.newCall(
+            POST(
+                "https://slwatch.co/api/video/stream/get",
+                body = "{\"id\":\"$id\"}"
+                    .toRequestBody("application/json".toMediaType()),
+            ),
+        ).execute().body.string()
+
+        val type = playlist.substringAfter("\"type\":\"").substringBefore("\"")
+        return if (type == "hls") {
+            val masterPlaylistUrl = playlist.substringAfter("\"file\":\"").substringBefore("\"").replace("\\/", "/")
+            val masterPlaylist = client.newCall(GET(masterPlaylistUrl)).execute().body.string()
+
+            val separator = "#EXT-X-STREAM-INF"
+            masterPlaylist.substringAfter(separator).split(separator).map {
+                val quality = it.substringAfter("RESOLUTION=").substringAfter("x").substringBefore(",") + "p"
+                val videoUrl = it.substringAfter("\n").substringBefore("\n").let { urlPart ->
+                    when {
+                        !urlPart.startsWith("http") ->
+                            masterPlaylistUrl.substringBefore("master.m3u8") + urlPart
+                        else -> urlPart
+                    }
+                }
+                Video(videoUrl, buildQuality(quality, prefix, suffix), videoUrl)
+            }
+        } else {
+            val separator = "\"label\":\""
+            playlist.substringAfter(separator).split(separator).map {
+                val quality = it.substringAfter(separator).substringBefore("\",")
+                val apiUrl = it.substringAfter("\"file\":\"").substringBefore("\",")
+                    .replace("\\", "")
+                val response = client.newCall(POST(apiUrl)).execute()
+                val videoUrl = response.request.url.toString()
+                Video(videoUrl, buildQuality(quality, prefix, suffix), videoUrl)
+            }
+        }
+    }
+
+    private fun buildQuality(resolution: String, prefix: String = "", suffix: String = "") =
+        buildString {
+            if (prefix.isNotBlank()) append("$prefix ")
+            append("Streamlare:$resolution")
+            if (suffix.isNotBlank()) append(" $suffix")
+        }
+}