extern crate restartables;
use reqwest;
use reqwest::Method;
use restartables::Restartable;
use std::default::Default;
use std::time::Duration;

#[derive(Debug)]
enum MyError {
    Reqwest(reqwest::Error),
    BadStatus(reqwest::StatusCode),
}

#[tokio::main]
async fn main() {
    example("https://google.com").await;
    example("https://google.asdfasdfasdf").await;
}

async fn example(url_to_hit: &'static str) {
    let url = reqwest::Url::parse(url_to_hit).unwrap();
    let req = reqwest::Request::new(Method::GET, url);
    let client: reqwest::Client = Default::default();
    let timeout = Duration::from_secs(2);

    let factory = || client.execute(req.try_clone().unwrap());
    let retrying = Restartable::new(factory, Some(timeout), |r| match r {
        // If the response is 200, resolve this future and return ()
        Ok(resp) if resp.status().is_success() => Ok(()),
        // Otherwise, return an error and restart the request.
        Ok(resp) => Err(MyError::BadStatus(resp.status())),
        Err(e) => Err(MyError::Reqwest(e)),
    });
    println!("Pinging {}", url_to_hit);
    let outcome = retrying.await;
    println!("{:?}", outcome);
}