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

427 lines
19 KiB
Kotlin

package com.lagradost
import com.lagradost.cloudstream3.TvType
import com.lagradost.cloudstream3.MainAPI
import com.lagradost.cloudstream3.SearchResponse
import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.LoadResponse.Companion.addActors
import com.lagradost.cloudstream3.LoadResponse.Companion.addTrailer
import com.lagradost.cloudstream3.mvvm.logError
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.Qualities
import com.lagradost.cloudstream3.utils.SubtitleHelper
import com.lagradost.cloudstream3.utils.loadExtractor
class NginxProvider : MainAPI() {
override var name = "Nginx"
// override var lang = "en" //no specific language
override val hasMainPage = true
override val hasQuickSearch = false
override val supportedTypes = setOf(TvType.AnimeMovie, TvType.TvSeries, TvType.Movie)
// override val hasSearch = false
companion object {
var companionName = "Nginx"
var loginCredentials: String? = null
var overrideUrl: String? = null
var pathToLibrary: String? = null // the library displayed when getting overrideUrl /home/username/media/
const val ERROR_STRING = "No nginx url specified in the settings"
fun getDirectMediaLink(path: String): String? {
val storedPathToLibrary = pathToLibrary
val storedOverrideUrl = overrideUrl // may be null if change value later
if (storedPathToLibrary == null || storedOverrideUrl == null) {
return null
}
return path.replace(storedPathToLibrary, storedOverrideUrl)
// /home/username/media/Movies/The Northman (2022)/The.Northman.2022.MULTi.1080p.AMZN.WEBRip.DDP5.1.Atmos.x264-VROOMM.mkv
// becomes:
// https://website.hosting.com/Movies/The Northman (2022)/The.Northman.2022.MULTi.1080p.AMZN.WEBRip.DDP5.1.Atmos.x264-VROOMM.mkv
}
}
private fun getAuthHeader(): Map<String, String> {
val url = overrideUrl ?: throw ErrorLoadingException(ERROR_STRING)
mainUrl = url
println("OVERRIDING URL TO $overrideUrl")
if (mainUrl == "NONE" || mainUrl.isBlank()) {
throw ErrorLoadingException(ERROR_STRING)
}
val localCredentials = loginCredentials
if (localCredentials == null || localCredentials.trim() == ":") {
return mapOf("Authorization" to "Basic ") // no Authorization headers
}
val basicAuthToken =
base64Encode(localCredentials.toByteArray())
return mapOf("Authorization" to "Basic $basicAuthToken")
}
override suspend fun load(url: String): LoadResponse? {
val authHeader = getAuthHeader() // reload
val isValid = url.contains(".nfo")
val isSerie = url.contains("tvshow.nfo")
val metadataDocument = app.get(url, authHeader).document // get the metadata nfo file
val title = metadataDocument.selectFirst("title")?.text()?.replace("/", "") ?: url.substringBeforeLast("/").substringAfterLast("/")
// gets the content between slashes: https://g.com/nameOfShow/tvshow.nfo
val description = metadataDocument.selectFirst("plot")?.text()
val mediaRootUrl = url.substringBeforeLast("/") + "/"
val mediaRootDocument = app.get(mediaRootUrl, authHeader).document
if (!isValid) {
if (".mp4" in url || ".mkv" in url) {
val fixedUrl = fixUrl(url)
return newMovieLoadResponse(
title,
fixedUrl,
TvType.Movie,
fixedUrl,
)
}
return null
}
// There is metadata and library is correctly organised!
if (!isSerie) { // Movie
val poster = metadataDocument.selectFirst("thumb[aspect=poster]")?.text()
val fanart = metadataDocument.selectFirst("fanart > thumb")?.text()
val trailer = metadataDocument.selectFirst("trailer")?.text()?.replace(
"plugin://plugin.video.youtube/play/?video_id=",
"https://www.youtube.com/watch?v="
)
val date = metadataDocument.selectFirst("year")?.text()?.toIntOrNull()
val durationInMinutes = metadataDocument.selectFirst("duration")?.text()?.toIntOrNull()
val ratingAverage = metadataDocument.selectFirst("value")?.text()?.toIntOrNull()
val tagsList = metadataDocument.select("genre")
.mapNotNull { // all the tags like action, thriller ...
it?.text()
}
val actors = metadataDocument.select("actor").mapNotNull {
val name = it?.selectFirst("name")?.text() ?: return@mapNotNull null
val image = it.selectFirst("thumb")?.text() ?: return@mapNotNull null
Actor(name, image)
}
val mkvElementsResult = mediaRootDocument.getElementsByAttributeValueContaining( // list of all urls of the mkv url in the webpage
"href",
".mkv"
)
val mp4ElementsResult = mediaRootDocument.getElementsByAttributeValueContaining( // list of all urls of the webpage
"href",
".mp4"
)
val dataList = if (mkvElementsResult.isNotEmpty()) { // there is probably a better way to do it
mkvElementsResult[0].attr("href").toString()
} else {
if(mp4ElementsResult.isNotEmpty()) {
mp4ElementsResult[0].attr("href").toString()
} else {
null
}
}
if (dataList != null) {
return newMovieLoadResponse(
title,
mediaRootUrl,
TvType.Movie,
mediaRootUrl + dataList,
) {
this.year = date
this.plot = description
this.rating = ratingAverage
this.tags = tagsList
this.backgroundPosterUrl = fanart
this.duration = durationInMinutes
addPoster(poster, authHeader)
addActors(actors)
addTrailer(trailer)
}
}
} else {
// a tv show
val list = ArrayList<Pair<Int, String>>()
val posterUrl = mediaRootUrl + "poster.jpg"
val tagsList = metadataDocument.select("genre")
.mapNotNull { // all the tags like action, thriller ...; unused variable
it?.text()
}
val actorsList = metadataDocument.select("actor")
.mapNotNull { // all the tags like action, thriller ...; unused variable
if (it.selectFirst("name")?.text() != null && it.selectFirst("role")?.text() != null) {
it.selectFirst("name")?.text()
it.selectFirst("role")?.text()
} else null
}
val seasons =
mediaRootDocument.getElementsByAttributeValueContaining("href", "Season%20")
seasons.forEach { element ->
val season =
element.attr("href").replace("Season%20", "").replace("/", "").toIntOrNull()
val href = mediaRootUrl + element.attr("href")
if (season != null && season > 0 && href.isNotBlank()) {
list.add(Pair(season, href))
}
}
if (list.isEmpty()) throw ErrorLoadingException("No Seasons Found, make sure to use season folders")
val episodeList = ArrayList<Episode>()
list.apmap { (seasonInt, seasonString) ->
val seasonDocument = app.get(seasonString, authHeader).document
val episodes = seasonDocument.getElementsByAttributeValueContaining(
"href",
".nfo"
) // get metadata
episodes.forEach { episode ->
val nfoDocument = app.get(seasonString + episode.attr("href"), authHeader).document // get episode metadata file
val epNum = nfoDocument.selectFirst("episode")?.text()?.toIntOrNull()
val poster =
seasonString + episode.attr("href").replace(".nfo", "-thumb.jpg")
val name = nfoDocument.selectFirst("title")?.text()
// val seasonInt = nfoDocument.selectFirst("season").text().toIntOrNull()
val date = nfoDocument.selectFirst("aired")?.text()
val plot = nfoDocument.selectFirst("plot")?.text()
val dataList = seasonDocument.getElementsByAttributeValueContaining(
"href",
episode.attr("href").replace(".nfo", "")
)
val data = seasonString + dataList.firstNotNullOf { item -> item.takeIf { (!it.attr("href").contains(".nfo") && !it.attr("href").contains(".jpg"))} }.attr("href").toString() // exclude poster and nfo (metadata) file
episodeList.add(
newEpisode(data) {
this.name = name
this.season = seasonInt
this.episode = epNum
this.posterUrl = poster // will require headers too
this.description = plot
addDate(date)
}
)
}
}
return newTvSeriesLoadResponse(title, url, TvType.TvSeries, episodeList) {
this.name = title
this.url = url
this.episodes = episodeList
this.plot = description
this.tags = tagsList
addPoster(posterUrl, authHeader)
addActors(actorsList)
}
}
///
return null
}
override suspend fun loadLinks(
data: String,
isCasting: Boolean,
subtitleCallback: (SubtitleFile) -> Unit,
callback: (ExtractorLink) -> Unit
): Boolean {
// println("loadling link $data")
val authHeader = getAuthHeader() // refresh crendentials
try {
val mediaRootUrl = data.substringBeforeLast("/") + "/"
val mediaRootDocument = app.get(mediaRootUrl, authHeader).document
val subtitlesElements = mediaRootDocument.getElementsByAttributeValueContaining(
// list of all urls of the mkv url in the webpage
"href",
".srt", // all of subtitles files in the folder
)
subtitlesElements.forEach {
subtitleCallback.invoke(
SubtitleFile(
SubtitleHelper.fromTwoLettersToLanguage(it.attr("href").replace(".srt", "").substringAfterLast(".")) ?: "English",
mediaRootUrl + "/" + it.attr("href")
)
)
}
} catch (e: Exception) {
logError(e)
}
callback.invoke (
ExtractorLink(
name,
name,
data,
"", // referer not needed
Qualities.Unknown.value,
false,
authHeader,
)
)
return true
}
private fun cleanElement(elementUrl: String): String {
return if (elementUrl[0] == '/') { // if starts by "/", remove it
elementUrl.drop(1)
} else {
elementUrl
}
}
override suspend fun getMainPage(
page: Int,
request : MainPageRequest,
): HomePageResponse {
val authHeader = getAuthHeader() // reload
if (mainUrl == "NONE" || mainUrl == "" || mainUrl == "nginx_url_key"){ // mainurl: http://192.168.1.10/media/
throw ErrorLoadingException("No nginx url specified in the settings: Nginx Settigns > Nginx server url, try again in a few seconds") // getString(R.string.nginx_load_exception) or @strings/nginx_load_exception
}
val document = app.get(mainUrl, authHeader).document
val categories = document.select("a") // select all of the categories
val returnList = categories.mapNotNull { // for each category
val categoryPath = mainUrl + cleanElement(it.attr("href")) // ?: return@mapNotNull null // get the url of the category; like http://192.168.1.10/media/Movies/
val categoryTitle = it.text() // get the category title like Movies/ or Series/
if (categoryTitle != "../" && categoryTitle != "/") { // exclude parent dir
val categoryDocument = app.get(categoryPath, authHeader).document // queries the page http://192.168.1.10/media/Movies/
val contentLinks = categoryDocument.select("a") // select all links
val currentList = contentLinks.mapNotNull { head -> // for each of those elements
val linkToElement = head.attr("href")
if (linkToElement != "../" && linkToElement != "/") {
val isAFolderBool = linkToElement.endsWith("/")
try {
val mediaRootUrl =
categoryPath + cleanElement(linkToElement) // like http://192.168.1.10/Series/Chernobyl/ or http://192.168.1.10/Movies/zjfbfvz.mp4
if (isAFolderBool) { // content is organised in folders for each element
/*
├── Movies
├── Eternals (2021) // the media root folder will be 'http://192.168.1.10/media/Movies/Eternals (2021)/'
├── Eternals.2021.MULTi.WiTH.TRUEFRENCH.iMAX.1080p.DSNP.WEB-DL.DDP5.1.H264-FRATERNiTY.mkv
├── Eternals.2021.MULTi.WiTH.TRUEFRENCH.iMAX.1080p.DSNP.WEB-DL.DDP5.1.H264-FRATERNiTY.en.srt
├── Eternals.2021.MULTi.WiTH.TRUEFRENCH.iMAX.1080p.DSNP.WEB-DL.DDP5.1.H264-FRATERNiTY.nfo
├── fanart.jpg
└── poster.jpg
*/
val mediaDocument = app.get(mediaRootUrl, authHeader).document
val nfoFilename = try {
mediaDocument.getElementsByAttributeValueContaining(
"href",
".nfo"
)[0]?.attr("href")
} catch (e: Exception) {
logError(e)
null
}
val isSerieType =
nfoFilename.toString() == "tvshow.nfo" // will be a movie if no metadata
val nfoPath = if (nfoFilename != null) {
mediaRootUrl + nfoFilename // metadata must exist
} else {
return@mapNotNull newMovieSearchResponse(
linkToElement,
mediaRootUrl,
TvType.Movie,
)
}
val nfoContent = app.get(nfoPath, authHeader).document // get all the metadata
if (isSerieType) {
val serieName = nfoContent.select("title").text() ?: linkToElement // name of the tv show
val posterUrl = mediaRootUrl + "poster.jpg" // poster.jpg in the same folder
newTvSeriesSearchResponse(
serieName,
nfoPath,
TvType.TvSeries,
) {
addPoster(posterUrl, authHeader)
}
} else { // Movie
val movieName = nfoContent.select("title").text() ?: linkToElement
val posterUrl = nfoContent.selectFirst("thumb[aspect=poster]")?.text() // poster should be stored in the same folder
// val fanartUrl = mediaRootUrl + "fanart.jpg" // backdrop should be stored in the same folder
return@mapNotNull newMovieSearchResponse(
movieName,
nfoPath,
TvType.Movie,
) {
addPoster(posterUrl, authHeader)
}
}
} else { // return direct file
/*
├── Movies
├── Eternals.2021.MULTi.WiTH.TRUEFRENCH.iMAX.1080p.DSNP.WEB-DL.DDP5.1.H264-FRATERNiTY.mkv // the media root folder
├── Juste la fin du monde (2016) VOF 1080p mHD x264 AC3-SANTACRUZ.mkv
*/
return@mapNotNull newMovieSearchResponse(
linkToElement,
mediaRootUrl,
TvType.Movie,
)
}
} catch (e: Exception) { // can cause issues invisible errors
null
//logError(e) // not working because it changes the return type of currentList to Any
}
} else null
}
if (currentList.isNotEmpty() && categoryTitle != "../" && categoryTitle != "/") { // exclude upper dir
HomePageList(categoryTitle.replace("/", " "), currentList)
} else null
} else null // the path is ../ which is parent directory
}
// if (returnList.isEmpty()) return null // maybe doing nothing idk
return HomePageResponse(returnList)
}
}