use base64::{Engine as _, engine::general_purpose};
use ntex::web;

use crate::util::requester::Requester;

/// Extracts digits from a string and sums them.
fn ssut51(arg: &str) -> u32 {
    arg.chars()
        .filter(|c| c.is_ascii_digit())
        .map(|c| c.to_digit(10).unwrap())
        .sum()
}

/// Encodes a token: "<sum1>-<host>-<sum2>" using Base64 URL-safe variant.
fn boo(sum1: u32, sum2: u32) -> String {
    let raw = format!("{}-{}-{}", sum1, "sxyprn.com", sum2);
    let encoded = general_purpose::STANDARD.encode(raw);

    // Replace + → -, / → _, = → .
    encoded
        .replace('+', "-")
        .replace('/', "_")
        .replace('=', ".")
}

#[derive(Debug, Clone)]
pub struct SxyprnProxy {}

impl SxyprnProxy {
    pub fn new() -> Self {
        SxyprnProxy {}
    }

    pub async fn get_video_url(
        &self,
        url: String,
        requester: web::types::State<Requester>,
    ) -> String {
        let mut requester = requester.get_ref().clone();
        let url = "https://sxyprn.com/".to_string() + &url;
        // println!("Fetching URL: {}", url);
        let text = requester.get(&url, None).await.unwrap_or("".to_string());
        if text.is_empty() {
            return "".to_string();
        }
        let data_string = text.split("data-vnfo='").collect::<Vec<&str>>()[1]
            .split("\":\"")
            .collect::<Vec<&str>>()[1]
            .split("\"}")
            .collect::<Vec<&str>>()[0]
            .replace("\\", "");
        // println!("src: {}", data_string);
        let mut tmp = data_string
            .split("/")
            .map(|s| s.to_string())
            .collect::<Vec<String>>();
        // println!("tmp: {:?}", tmp);
        tmp[1] = format!(
            "{}8/{}",
            tmp[1],
            boo(ssut51(tmp[6].as_str()), ssut51(tmp[7].as_str()))
        );

        // println!("tmp[1]: {:?}", tmp[1]);
        //preda
        tmp[5] = format!(
            "{}",
            tmp[5].parse::<u32>().unwrap() - ssut51(tmp[6].as_str()) - ssut51(tmp[7].as_str())
        );
        // println!("tmp: {:?}", tmp);
        let sxyprn_video_url = format!("https://sxyprn.com{}", tmp.join("/"));
        // println!("sxyprn_video_url: {}", sxyprn_video_url);
        match crate::util::get_redirect_location(&sxyprn_video_url) {
            Ok(Some(loc)) => {return format!("https:{}", loc)},
            Ok(None)      => println!("No redirect found for {}", sxyprn_video_url),
            Err(e)        => eprintln!("Request failed: {}", e),
        }
        return "".to_string();
    }
}