subspedia-rs 0.1.4

//! This crate is a simple library for [subspedia](https://www.subspedia.tv/) based on api the
//! provided by site

extern crate hyper;
extern crate hyper_tls;
extern crate tokio;
extern crate futures;
#[macro_use]
extern crate serde_derive;
extern crate failure;
#[macro_use]
extern crate failure_derive;
extern crate serde;
extern crate serde_json;

use futures::{Future, Stream};
use std::sync::{Arc, Mutex};
use std::borrow::Cow;

/// An enumeration of possible error which can occur during http requests and the parsing of json
/// returned by the api
#[derive(Debug, Fail)]
pub enum FetchError {
    #[fail(display = "HTTP error: {}", _0)]
    Http(hyper::Error),
    #[fail(display = "JSON parsing error: {}", _0)]
    Json(serde_json::Error),
    #[fail(display = "{}", _0)]
    NotFound(String),
}

impl From<hyper::Error> for FetchError {
    fn from(err: hyper::Error) -> FetchError {
        FetchError::Http(err)
    }
}

impl From<serde_json::Error> for FetchError {
    fn from(err: serde_json::Error) -> FetchError {
        FetchError::Json(err)
    }
}

/// Trait that requests have to implement.
pub trait Request {
    type Response: serde::de::DeserializeOwned + std::fmt::Debug + std::marker::Send;

    /// Return api url based on type of response that are you looking for.
    fn url(&self) -> Cow<'static, str>;
}

/// Struct for store the television series in translation.
#[derive(Deserialize, Debug)]
pub struct SerieTraduzione {
    id_serie: usize,
    nome_serie: String,
    link_serie: String,
    id_thetvdb: usize,
    num_stagione: usize,
    num_episodio: usize,
    stato: String,
}

/// Struct to perform the request for the series that are in translation.
pub struct ReqSerieTraduzione;

impl Request for ReqSerieTraduzione {
    type Response = SerieTraduzione;

    fn url(&self) -> Cow<'static, str> {
        Cow::Borrowed("https://www.subspedia.tv/API/serie_traduzione")
    }
}

/// Struct for store the television series available on site.
#[derive(Deserialize, Debug, Clone)]
pub struct Serie {
    id_serie: usize,
    pub nome_serie: String,
    link_serie: String,
    id_thetvdb: usize,
    stato: String,
    anno: usize,
}

/// Struct to perform the request for the all series available on site.
pub struct ReqElencoSerie;

impl Request for ReqElencoSerie {
    type Response = Serie;

    fn url(&self) -> Cow<'static, str> {
        Cow::Borrowed("https://www.subspedia.tv/API/elenco_serie")
    }
}

/// Struct for store the subtitles.
#[derive(Deserialize, Debug)]
pub struct Sottotitolo {
    id_serie: usize,
    nome_serie: String,
    ep_titolo: String,
    num_stagione: usize,
    num_episodio: usize,
    immagine: String,
    link_sottotitoli: String,
    link_serie: String,
    link_file: String,
    descrizione: String,
    id_thetvdb: usize,
    data_uscita: String,
    grazie: usize,
}

/// Struct to perform the request for the last subtitles translated.
pub struct ReqUltimiSottotitoli;

impl Request for ReqUltimiSottotitoli {
    type Response = Sottotitolo;

    fn url(&self) -> Cow<'static, str> {
        Cow::Borrowed("https://www.subspedia.tv/API/ultimi_sottotitoli")
    }
}

/// Struct to perform the subtitle request for a series.
pub struct ReqSottotitoliSerie {
    id: usize,
}

impl ReqSottotitoliSerie {
    /// Create a new ReqSottotitoliSerie with the given series id.
    pub fn new(id: usize) -> ReqSottotitoliSerie {
        ReqSottotitoliSerie { id }
    }
}

impl Request for ReqSottotitoliSerie {
    type Response = Sottotitolo;

    fn url(&self) -> Cow<'static, str> {
        Cow::Owned(format!("https://www.subspedia.tv/API/sottotitoli_serie?serie={}", self.id))
    }
}

///Makes a request based on given type
///
/// # Errors
///
/// Returns error if something gone wrong during http requests and parsing json.
///
/// # Exemple
///
///```
///extern crate subspedia;
///
///use subspedia::ReqSerieTraduzione;
///
///fn main() {
///    println!("{:#?}", subspedia::get(ReqSerieTraduzione).unwrap());
///}
/// ```
pub fn get<R: 'static + Request>(req: &R) -> Result<Vec<R::Response>, FetchError>
{
    let url = req.url().parse().unwrap();
    let result = Arc::new(Mutex::new(Vec::new()));

    let tmp = Arc::clone(&result);

    tokio::run(futures::lazy(move || {
        fetch_json::<R::Response>(url)
            // use the parsed vector
            .map(move |mut serie| {
                tmp.lock().unwrap().append(&mut serie);
            })
            // if there was an error print it
            .map_err(|e| eprintln!("{}", e))
    }));

    Ok(Arc::try_unwrap(result).unwrap().into_inner().unwrap())
}

///Search serie based on a given name
///
/// # Errors
///
/// Returns error if something gone wrong during http requests, parsing json or if a series with that
/// name isn't found
///
/// # Exemple
///
///```
///extern crate subspedia;
///
///use subspedia::{FetchError, search_by_name};
///
///fn main() -> Result<(), FetchError> {
///    println!("{:#?}", search_by_name("serie name")?);
///    Ok(())
///}
/// ```
pub fn search_by_name(name: &str) -> Result<Vec<Serie>, FetchError> {
    let result = get(&ReqElencoSerie)?
        .iter()
        .filter(|s| s.nome_serie

            .to_lowercase()
            .as_str()
            .contains(name.to_lowercase().as_str())
        )
        .collect::<Vec<_>>()
        .iter()
        .map(|s| (**s).clone())
        .collect::<Vec<_>>();

    if !result.is_empty() {
        Ok(result)
    } else {
        Err(FetchError::NotFound(format!("Series with name {} not found", name)))
    }
}

///Search the series based on a given id
///
/// # Errors
///
/// Returns error if something gone wrong during http requests, parsing json or if a series with that
/// id isn't found
///
/// # Exemple
///
///```
/// extern crate subspedia;
///
/// use subspedia::{FetchError, search_by_id};
///
/// fn main() -> Result<(), FetchError> {
///    println!("{:#?}", search_by_id(500)?);
///    Ok(())
/// }
/// ```
pub fn search_by_id(id: usize) -> Result<Serie, FetchError> {
    match get(&ReqElencoSerie)?
        .iter()
        .filter(|s| s.id_serie == id)
        .collect::<Vec<_>>()
        .pop() {
        Some(s) => Ok(s.clone()),
        None => Err(FetchError::NotFound(format!("Series with id {} not found.", id)))
    }
}

fn fetch_json<T>(url: hyper::Uri) -> impl Future<Item=Vec<T>, Error=FetchError>
    where T: serde::de::DeserializeOwned + std::fmt::Debug
{
    let https = hyper_tls::HttpsConnector::new(4).unwrap();
    let client = hyper::Client::builder()
        .build::<_, hyper::Body>(https);

    client
        // Fetch the url...
        .get(url)
        // And then, if we get a response back...
        .and_then(|res| {
            // asynchronously concatenate chunks of the body
            res.into_body().concat2()
        })
        .from_err::<FetchError>()
        // use the body after concatenation
        .and_then(|body| {
            // try to parse as json with serde_json
            let serie = serde_json::from_slice(&body)?;
            Ok(serie)
        })
}