#![deny(warnings)]

use rweb::*;
use std::{convert::Infallible, str::FromStr, time::Duration};

#[tokio::main]
async fn main() {
    // Match `/:Seconds`...
    let routes = sleepy();

    rweb::serve(routes).run(([127, 0, 0, 1], 3030)).await;
}

#[get("/{seconds}")]
async fn sleepy(seconds: Seconds) -> Result<impl rweb::Reply, Infallible> {
    tokio::time::sleep(Duration::from_secs(seconds.0)).await;
    Ok(format!("I waited {} seconds!", seconds.0))
}

/// A newtype to enforce our maximum allowed seconds.
#[derive(Schema)]
struct Seconds(u64);

impl FromStr for Seconds {
    type Err = ();
    fn from_str(src: &str) -> Result<Self, Self::Err> {
        src.parse::<u64>().map_err(|_| ()).and_then(|num| {
            if num <= 5 {
                Ok(Seconds(num))
            } else {
                Err(())
            }
        })
    }
}
