ratpack 0.1.0

ratpack is a HTTP framework designed around simplicity and ease-of-use
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use ratpack::prelude::*;

#[derive(Clone)]
struct AuthedState {
    authed: Option<bool>,
}

impl TransientState for AuthedState {
    fn initial() -> Self {
        Self { authed: None }
    }
}

async fn validate_authtoken(
    req: Request<Body>,
    resp: Option<Response<Body>>,
    _params: Params,
    app: App<State, AuthedState>,
    mut authstate: AuthedState,
) -> HTTPResult<AuthedState> {
    let token = req.headers().get("X-AuthToken");
    if token.is_none() {
        return Err(Error::StatusCode(StatusCode::UNAUTHORIZED));
    }

    let token = token.unwrap();

    let state = app.state().await;
    if state.is_none() {
        return Err(Error::StatusCode(StatusCode::UNAUTHORIZED));
    }

    let state = state.unwrap();
    authstate.authed = Some(state.clone().lock().await.authtoken == token);

    return Ok((req, resp, authstate));
}

async fn hello(
    req: Request<Body>,
    _resp: Option<Response<Body>>,
    params: Params,
    _app: App<State, AuthedState>,
    authstate: AuthedState,
) -> HTTPResult<AuthedState> {
    if authstate.authed.is_some() && !authstate.authed.unwrap() {
        return Err(Error::StatusCode(StatusCode::UNAUTHORIZED));
    }

    let name = params.get("name").unwrap();
    let bytes = Body::from(format!("hello, {}!\n", name));

    return Ok((
        req,
        Some(Response::builder().status(200).body(bytes).unwrap()),
        authstate,
    ));
}

#[derive(Clone)]
struct State {
    authtoken: &'static str,
}

#[tokio::main]
async fn main() -> Result<(), ServerError> {
    let mut app = App::with_state(State {
        authtoken: "867-5309",
    });
    app.get("/auth/:name", compose_handler!(validate_authtoken, hello));
    app.get("/:name", compose_handler!(hello));

    app.serve("127.0.0.1:3000").await?;

    Ok(())
}