//! 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::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 driver::Extractor<(String,String)> for AudiophilItemPriceExtractor {
    fn extract(&mut self, uri: &Uri, doc: &extract::Document) -> Vec<(String,String)> {
        info!("Extracting info from {}", uri);
        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();
        info!("Extracted {:?}", zipped);
        zipped
    }
}

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, _: &Uri, _: &extract::Document) -> Vec<Uri> {
        vec![]
    }
}

pub struct DebuggingStorage { }

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