//! Implementations of common traits that are useful to plug together a Driver.

use std::iter;
use std::iter::FromIterator;

use crate::driver;
use crate::extract;
use crate::http;
use crate::err::HTTPError;

use hyper::Uri;
use log::{info,warn,error};
use rex_regex as rex;

pub struct AudiophilItemPriceExtractor {
}

fn substring(s: String, (start, len): (usize, usize)) -> String {
    String::from_iter(s.chars().skip(start).take(len))
}

impl extract::Extractor for AudiophilItemPriceExtractor {
    fn extract(&mut self, doc: &extract::Document) -> Option<Box<dyn extract::Extracted>> {
        let mut data = doc.get_contents(&[".bez.neu", ".preis strong"]).unwrap();
        let prices = data.pop().unwrap();
        let descs = data.pop().unwrap();

        let onlytext = rex::compile("^[a-zA-Z0-9\\.,+/ -]+").unwrap();

        let zipped: Vec<(String, String)> = descs
            .into_iter()
            .zip(prices)
            .map(|(desc, price)| (desc.trim().to_string(), price.trim().to_string()))
            .map(move |(desc, price)| {
                let desc2;
                let price2;
                let (ok, descmatch) = rex::match_re(&onlytext, &desc);
                if ok {
                    desc2 = substring(desc, descmatch[0]);
                } else {
                    desc2 = desc;
                }
                let (ok, pricematch) = rex::match_re(&onlytext, &price);
                if ok {
                    price2 = substring(price, pricematch[0]);
                } else {
                    price2 = price;
                }

                (desc2, price2)
            })
            .collect();
        println!("{:?}", zipped);
        None
    }
}

pub struct AudiophilExplorer {
    known: Vec<hyper::Uri>,
}

impl AudiophilExplorer {
    pub fn new() -> AudiophilExplorer {
        let want = vec![
            "https://audiophil-foto.de/de/shop/kameras/sony/",
            "https://audiophil-foto.de/de/shop/kameras/pentax-ricoh/",
            "https://audiophil-foto.de/de/shop/kameras/leica/",
            "https://audiophil-foto.de/de/shop/objektive/sony/",
            "https://audiophil-foto.de/de/shop/objektive/zeiss/",
            "https://audiophil-foto.de/de/shop/objektive/sigma/",
        ].into_iter().map(|s| s.parse::<Uri>().unwrap()).collect();
        AudiophilExplorer { known: want }
    }
}

impl driver::Explorer for AudiophilExplorer {
    fn idle(&mut self) -> Vec<Uri> {
        self.known.drain(..).collect()
    }
    fn next(&mut self, _: &extract::Document) -> Vec<Uri> {
        vec![]
    }
}

pub struct DebuggingStorage { }

#[async_trait::async_trait]
impl driver::Storage for DebuggingStorage {
    async fn store(&mut self, iter: Box<dyn iter::Iterator<Item=(String,String)>+Send>) -> Result<(), HTTPError> {
        let all = iter.collect::<Vec<(String,String)>>();
        info!("STORAGE: Received {:?}", all);
        Ok(())
    }
}