package com.stable

import com.lagradost.cloudstream3.app
import com.lagradost.cloudstream3.utils.ExtractorApi
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.Qualities

class UpstreamExtractor: ExtractorApi() {
    override val name: String = "Upstream.to"
    override val mainUrl: String = "https://upstream.to"
    override val requiresReferer = true

    override suspend fun getUrl(url: String, referer: String?): List<ExtractorLink> {
        // WIP: m3u8 link fetched but sometimes not playing
        //Log.i(this.name, "Result => (no extractor) ${url}")
        val sources: MutableList<ExtractorLink> = mutableListOf()
        val doc = app.get(url, referer = referer).text
        if (doc.isNotBlank()) {
            var reg = Regex("(?<=master)(.*)(?=hls)")
            val result = reg.find(doc)?.groupValues?.map {
                it.trim('|')
            }?.toList()
            reg = Regex("(?<=\\|file\\|)(.*)(?=\\|remove\\|)")
            val domainList = reg.find(doc)?.groupValues?.get(1)?.split("|")
            var domain = when (!domainList.isNullOrEmpty()) {
                true -> {
                    if (domainList.isNotEmpty()) {
                        var domName = ""
                        for (part in domainList) {
                            domName = "${part}.${domName}"
                        }
                        domName.trimEnd('.')
                    } else { "" }
                }
                false -> ""
            }
            //Log.i(this.name, "Result => (domain) ${domain}")
            if (domain.isEmpty()) {
                domain = "s96.upstreamcdn.co"
                //Log.i(this.name, "Result => (default domain) ${domain}")
            }

            result?.forEach {
                val linkUrl = "https://${domain}/hls/${it}/master.m3u8"
                sources.add(
                    ExtractorLink(
                    name = "Upstream m3u8",
                    source = this.name,
                    url = linkUrl,
                    quality = Qualities.Unknown.value,
                    referer = referer ?: linkUrl,
                    isM3u8 = true
                    )
                )
            }
        }
        return sources
    }
}