mise_server/

mise.rs

use crate::{
    routes::{RequestProcessor, RouteContext, RouteMethod},
    server::Server,
};
use serde_json::Value;
use std::{collections::HashMap, net::SocketAddr};

pub struct Request(pub http::Request<Value>);
pub struct Response(pub http::Response<Value>);

/// Final routes are always in JSON.
pub trait Route: FnMut(Request) -> Response + 'static {}
impl<T> Route for T where T: FnMut(Request) -> Response + 'static {}

/// In some cases it's possible to install a route that only returns text.
/// this is used for certain side behavior such as returning prometheus scrape
/// renders. These are meant to be static and have no parameter and respond only
/// on absolute paths: no request object is available and the response is always
/// a string.
pub trait TextRoute: FnMut() -> String + 'static {}
impl<T> TextRoute for T where T: FnMut() -> String + 'static {}

/// Server resource and entry point.
///
/// Example:
///
/// ```no_run
/// use mise_server::prelude::*;
/// use serde_json::json;
///
/// Mise::new()
///     .get("/found", |_| json!("hello world").into())
///     .get("/not", |_| StatusCode::NOT_FOUND.into())
///     .get("/param", |r| json!(r.query_param("a").unwrap()).into())
///     .get("/error", |_| panic!("error"))
///     .text("/text", || "result".to_string())
///     .post("/echo", |r| r.body().clone().into())
///     .get("/get_echo/*", |r| json!(r.name()).into())
///     .serve("127.0.0.1:8080".parse().unwrap());
/// ```
#[derive(Default)]
pub struct Mise {
    routes: HashMap<RouteMethod, HashMap<String, RouteContext>>,
    text_routes: HashMap<String, Box<dyn TextRoute>>,
}

impl Mise {
    /// Create a new default server.
    pub fn new() -> Self {
        Self::default()
    }

    /// Register a text result only.
    /// This is used for cases such as prometheus scrape renders.
    /// Text routes always take precedence even if the route is already defined
    /// for any other methods.
    pub fn text<F: TextRoute>(mut self, path: &str, f: F) -> Self {
        self.text_routes.insert(path.to_string(), Box::new(f));
        self
    }

    /// Register a get route.
    pub fn get<F: Route>(mut self, path: &str, f: F) -> Self {
        let routes = self.routes.entry(RouteMethod::Get).or_default();
        let d: Box<dyn Route> = Box::new(f);
        routes.insert(path.to_string(), (path, d).into());
        self
    }

    /// Registers a post route. Body is obtained from [Request::body] and it is
    /// always a json [Value].
    pub fn post<F: Route>(mut self, path: &str, f: F) -> Self {
        let routes = self.routes.entry(RouteMethod::Post).or_default();
        let d: Box<dyn Route> = Box::new(f);
        routes.insert(path.to_string(), (path, d).into());
        self
    }

    /// Starts the server. Blocks until the server quits.
    /// Can panic if cannot bind the server.
    pub fn serve(self, addr: SocketAddr) {
        Server::serve(
            RequestProcessor {
                routes: self.routes,
                text_routes: self.text_routes,
            },
            addr,
        )
        .run();
    }
}

impl From<Request> for Value {
    fn from(value: Request) -> Self {
        value.0.body().to_owned()
    }
}

impl From<Value> for Response {
    fn from(value: Value) -> Self {
        Response(http::Response::new(value))
    }
}

impl From<http::StatusCode> for Response {
    fn from(value: http::StatusCode) -> Self {
        Response(
            http::Response::builder()
                .status(value)
                .body(Value::Null)
                .expect("Statically built body should not fail"),
        )
    }
}

impl Request {
    /// Returns the uri path, without the query params
    pub fn path(&self) -> &str {
        self.0.uri().path()
    }

    pub fn query(&self) -> Option<&str> {
        self.0.uri().query()
    }

    /// Returns the query param by name.
    pub fn query_param(&self, name: &str) -> Option<&str> {
        let q = self.0.uri().query()?;
        let f = format!("{name}=");
        let idx = q.find(&f)?;
        let end = q[idx..].find('&').unwrap_or(q.len());
        Some(&q[idx + f.len()..end])
    }

    /// Returns the base of the path which is the path without the last item
    ///
    /// eg from /p1/p2/p3 returns '/p1/p2'
    ///
    /// empty string when is unavailable ("/a" = "")
    pub fn base(&self) -> &str {
        let p = self.0.uri().path();
        let n = self.name();
        p[..p.len() - n.len()].trim_end_matches("/")
    }

    /// Returns the last item of the path
    ///
    /// eg from /p1/p2/p3 returns 'p3'
    ///
    /// empty string when is unavailable
    pub fn name(&self) -> &str {
        if !self.0.uri().path().contains("/") {
            return "";
        }
        self.0.uri().path().split("/").last().unwrap_or("")
    }

    /// If there is a body in the request, then this will be the json of that
    ///
    /// If not available returns [Value::Null]
    pub fn body(&self) -> &Value {
        self.0.body()
    }

    pub(crate) fn base_star(&self) -> Option<String> {
        if self.name().is_empty() {
            // Cannot have a wildcard: no last item
            return None;
        }
        Some(format!("{}/*", self.base()))
    }
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn test_paths() {
        assert_eq!(requri("/").name(), "");
        assert_eq!(requri("/a").name(), "a");
        assert_eq!(requri("/a/b").name(), "b");

        assert_eq!(requri("/").base(), "");
        assert_eq!(requri("/a").base(), "");
        assert_eq!(requri("/a/b").base(), "/a");

        assert_eq!(requri("/").base_star(), None);
        assert_eq!(requri("/a").base_star(), Some("/*".to_string()));
        assert_eq!(requri("/a/b").base_star(), Some("/a/*".to_string()));

        assert_eq!(requri("/?b=a").query_param("b"), Some("a"));
    }

    fn requri(uri: &str) -> Request {
        Request(http::Request::builder().uri(uri).body(Value::Null).unwrap())
    }
}