package com.streamflixreborn.streamflix.extractors

import com.tanasi.retrofit_jsoup.converter.JsoupConverterFactory
import com.streamflixreborn.streamflix.models.Video
import java.util.regex.Pattern
import okhttp3.OkHttpClient
import org.jsoup.nodes.Document
import retrofit2.Retrofit
import retrofit2.http.GET
import retrofit2.http.Header
import retrofit2.http.Url


class VideoSibNetExtractor : Extractor() {
    override val name = "VideoSibNet"
    override val mainUrl = "https://video.sibnet.ru/"

    override suspend fun extract(link: String): Video {
        val service = Service.build(mainUrl)

        val document = service.get(link, mainUrl)

        val relativeVideoUrl = extractRelativeVideoUrl(document)
            ?: throw Exception("Could not find video source in the webpage")

        val absoluteVideoUrl = if (relativeVideoUrl.startsWith("/")) {
            mainUrl.trimEnd('/') + relativeVideoUrl
        } else {
            relativeVideoUrl
        }

        return Video(
            source = absoluteVideoUrl,
            headers = mapOf(
                "Referer" to mainUrl,
                "User-Agent" to USER_AGENT
            )
        )
    }

    private fun extractRelativeVideoUrl(document: Document): String? {
        val scriptTags = document.select("script")
        for (script in scriptTags) {
            val scriptContent = script.html()
            if ("player.src" in scriptContent) {
                val pattern = Pattern.compile("src:\\s*\"([^\"]+)\"")
                val matcher = pattern.matcher(scriptContent)
                if (matcher.find()) {
                    return matcher.group(1)
                }
            }
        }
        return null
    }

    private interface Service {
        companion object {
            fun build(baseUrl: String): Service = Retrofit.Builder()
                .baseUrl(baseUrl)
                .client(OkHttpClient.Builder().build())
                .addConverterFactory(JsoupConverterFactory.create())
                .build()
                .create(Service::class.java)
        }

        @GET
        suspend fun get(
            @Url url: String,
            @Header("Referer") referer: String,
            @Header("Accept") accept: String = "text/html",
            @Header("User-Agent") userAgent: String = USER_AGENT,
        ): Document
    }

    companion object {
        private const val USER_AGENT =
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
    }
}