use std::vec;
use std::env;
use error_chain::error_chain;
use htmlentity::entity::{decode, ICodedDataTrait};
use reqwest::{Proxy};
use crate::providers::Provider;
use crate::util::cache::VideoCache;
use crate::util::flaresolverr::{FlareSolverrRequest, Flaresolverr};
use crate::videos::{self, VideoItem};
use crate::DbPool;
use crate::USER_AGENT; // Make sure Provider trait is imported

error_chain! {
    foreign_links {
        Io(std::io::Error);
        HttpRequest(reqwest::Error);
    }
}

pub struct SpankbangProvider {
    url: String,
}
impl SpankbangProvider {
    pub fn new() -> Self {
        SpankbangProvider {
            url: "https://spankbang.com/".to_string(),
        }
    }
    async fn get(&self, cache:VideoCache, page: u8, sort: String) -> Result<Vec<VideoItem>> {
        let mut url = format!("{}{}/{}/", self.url, sort, page);
        
        let old_items = match cache.get(&url) {
            Some((time, items)) => {
                if time.elapsed().unwrap_or_default().as_secs() < 60 * 60 {
                    println!("Cache hit for URL: {}", url);
                    return Ok(items.clone());
                }
                else{
                    items.clone()
                }
            }
            None => {
                vec![]
            }
        };

        
        let client = match env::var("BURP_URL").as_deref() {
            Ok(burp_url) => 
                reqwest::Client::builder()
                .user_agent(USER_AGENT)
                .proxy(Proxy::https(burp_url).unwrap())
                .danger_accept_invalid_certs(true)
                .build()?,
            Err(_) => reqwest::Client::builder()
                .user_agent(USER_AGENT)
                .danger_accept_invalid_certs(true)
                .build()?,
        };
        
        let response = client.get(url.clone()).send().await?;
        if response.status().is_success() {
            let text = response.text().await?;
            let video_items: Vec<VideoItem> = self.get_video_items_from_html(text.clone());
            if !video_items.is_empty() {
                cache.remove(&url);
                cache.insert(url.clone(), video_items.clone());
            } else{
                return Ok(old_items);
            }
            Ok(video_items)
        } else {
            let flare_url = env::var("FLARE_URL").expect("FLARE_URL not set");
            let flare = Flaresolverr::new(flare_url);
            let result = flare
                .solve(FlareSolverrRequest {
                    cmd: "request.get".to_string(),
                    url: url.clone(),
                    maxTimeout: 60000,
                })
                .await;
            let video_items = match result {
                Ok(res) => {
                    // println!("FlareSolverr response: {}", res);
                    self.get_video_items_from_html(res.solution.response)
                }
                Err(e) => {
                    println!("Error solving FlareSolverr: {}", e);
                    return Err("Failed to solve FlareSolverr".into());
                }
            };
            if !video_items.is_empty() {
                cache.remove(&url);
                cache.insert(url.clone(), video_items.clone());
            } else {
                return Ok(old_items);
            }
            Ok(video_items)
        }
    }
    
    async fn query(&self, cache: VideoCache, page: u8, query: &str,) -> Result<Vec<VideoItem>> {
        let url = format!("{}s/{}/{}/", self.url, query.replace(" ", "+"), page);
        
        let old_items = match cache.get(&url) {
            Some((time, items)) => {
                if time.elapsed().unwrap_or_default().as_secs() < 60 * 60 {
                    println!("Cache hit for URL: {}", url);
                    return Ok(items.clone());
                }
                else{
                    items.clone()
                }
            }
            None => {
                vec![]
            }
        };

        
        let client = match env::var("BURP_URL").as_deref() {
            Ok(burp_url) => 
                reqwest::Client::builder()
                .user_agent(USER_AGENT)
                .proxy(Proxy::https(burp_url).unwrap())
                .danger_accept_invalid_certs(true)
                .build()?,
            Err(_) => reqwest::Client::builder()
                .user_agent(USER_AGENT)
                .danger_accept_invalid_certs(true)
                .build()?,
        };
        
        let response = client.get(url.clone()).send().await?;
        if response.status().is_success() {
            let text = response.text().await?;
            let video_items: Vec<VideoItem> = self.get_video_items_from_html(text.clone());
            if !video_items.is_empty() {
                cache.remove(&url);
                cache.insert(url.clone(), video_items.clone());
            } else{
                return Ok(old_items);
            }
            Ok(video_items)
        } else {
            let flare_url = env::var("FLARE_URL").expect("FLARE_URL not set");
            let flare = Flaresolverr::new(flare_url);
            let result = flare
                .solve(FlareSolverrRequest {
                    cmd: "request.get".to_string(),
                    url: url.clone(),
                    maxTimeout: 60000,
                })
                .await;
            let video_items = match result {
                Ok(res) => {
                    // println!("FlareSolverr response: {}", res);
                    self.get_video_items_from_html(res.solution.response)
                }
                Err(e) => {
                    println!("Error solving FlareSolverr: {}", e);
                    return Err("Failed to solve FlareSolverr".into());
                }
            };
            if !video_items.is_empty() {
                cache.remove(&url);
                cache.insert(url.clone(), video_items.clone());
            } else {
                return Ok(old_items);
            }
            Ok(video_items)
        }
    }

    fn get_video_items_from_html(&self, html: String) -> Vec<VideoItem> {
        if html.is_empty() {
            println!("HTML is empty");
            return vec![];
        }
        let mut items: Vec<VideoItem> = Vec::new();
        let video_listing_content = html.split("class=\"video-list ").collect::<Vec<&str>>()[1];
        let raw_videos = video_listing_content
            .split("class=\"video-item\"")
            .collect::<Vec<&str>>()[1..]
            .to_vec();
        for video_segment in &raw_videos {
            let vid = video_segment.split("\n").collect::<Vec<&str>>();
            //for (index,line) in vid.iter().enumerate(){
            //    println!("Line {}: {}\n\n", index, line);
            //}
            let mut title = vid[4].split("title=\"").collect::<Vec<&str>>()[1].split("\"").collect::<Vec<&str>>()[0].to_string();
            title = decode(title.as_bytes()).to_string().unwrap_or(title);
            let thumb = vid[14].split("data-src=\"").collect::<Vec<&str>>()[1].split("\"").collect::<Vec<&str>>()[0].to_string();
            let mut preview = "".to_string();
            if vid[16].contains("data-preview=\""){
                preview = vid[16].split("data-preview=\"").collect::<Vec<&str>>()[1].split("\"").collect::<Vec<&str>>()[0].to_string();
            }
            let duration_str = vid.iter().find(|s| s.contains("<span class=\"video-badge l\">")).unwrap().split("<span class=\"video-badge l\">").collect::<Vec<&str>>()[1].split("m<").collect::<Vec<&str>>()[0];
            let duration: u32 = duration_str.parse::<u32>().unwrap_or(0) * 60;
            let view_and_rating_str: Vec<&str> = vid.iter().copied().filter(|s| s.contains("<span class=\"md:text-body-md\">")).collect();
            let views_str = view_and_rating_str[0].split(">").collect::<Vec<&str>>()[1].split("K<").collect::<Vec<&str>>()[0];
            let views = (views_str.parse::<f32>().unwrap_or(0.0) * 1000.0) as u32;
            let rate_str = view_and_rating_str[1].split(">").collect::<Vec<&str>>()[1].split("%<").collect::<Vec<&str>>()[0];
            let rating = rate_str.parse::<f32>().unwrap_or(0.0);
            let url_part = vid.iter().find(|s| s.contains("<a href=\"/")).unwrap().split("<a href=\"/").collect::<Vec<&str>>()[1].split("\"").collect::<Vec<&str>>()[0];
            let url = self.url.clone() + url_part.clone();
            let id = url_part.split("/").collect::<Vec<&str>>()[0].to_string();

            let quality_str = match vid[25].contains("<"){
                true => vid[25].split(">").collect::<Vec<&str>>()[1].split("<").collect::<Vec<&str>>()[0],
                false => "SD",
            };
            let quality = match quality_str{
                "HD" => "1080",
                "4k" => "2160",
                "SD" => "720",
                _ => "1080",
            };
            

            let mut format =
                videos::VideoFormat::new(url.clone(), quality.to_string(), "mp4".to_string());
            format.add_http_header("User-Agent".to_string(), USER_AGENT.to_string());

            let mut video_item = VideoItem::new(id, title, url.clone().to_string(), "spankbang".to_string(), thumb, duration)
            .views(views)
            .rating(rating)
            .formats(vec![format])
            .preview(preview);
            items.push(video_item);
        }
        return items;
    }

}

impl Provider for SpankbangProvider {
    async fn get_videos(
        &self,
        cache: VideoCache,
        pool: DbPool,
        _channel: String,
        mut sort: String,
        query: Option<String>,
        page: String,
        per_page: String,
        featured: String,
    ) -> Vec<VideoItem> {
        let _ = per_page;
        let _ = featured;
        let _ = pool;

        if sort == "date"{
            sort = "trending_videos".to_string();
        }
        let videos: std::result::Result<Vec<VideoItem>, Error> = match query {
            Some(q) => self.query(cache, page.parse::<u8>().unwrap_or(1), &q).await,
            None => self.get(cache, page.parse::<u8>().unwrap_or(1), sort).await,
        };
        match videos {
            Ok(v) => v,
            Err(e) => {
                println!("Error fetching videos: {}", e);
                vec![]
            }
        }
    }
}