bnglws-server/src/routes.rs

use diesel::result::Error;
use rocket::http::Status;
use rocket::request::Request;
use rocket::response::status;
use rocket_contrib::json::Json;
use rocket_contrib::templates::Template;
use std::collections::HashMap;

use crate::connection::DbConn;
use crate::snippet;
use crate::snippet::{InsertableSnippet, Snippet};

fn error_response(error: Error) -> Status {
    match error {
        Error::NotFound => Status::NotFound,
        _ => Status::InternalServerError,
    }
}

#[get("/")]
pub fn index() -> Template {
    let context: HashMap<&str, String> = HashMap::new();
    Template::render("index", &context)
}

#[get("/snippets/<id>")]
pub fn show_snippet(id: i32, connection: DbConn) -> Template {
    let result = match snippet::get(&connection, id) {
        Ok(snippet) => Some(snippet),
        Err(_) => None,
    };

    let (id, title, body) = result.map_or(
        (
            String::from(""),
            String::from("404 - Snippet not found"),
            String::from(""),
        ),
        |snippet| {
            (
                format!("{}", snippet.id),
                snippet.title,
                snippet.formatted_body,
            )
        },
    );

    let mut context: HashMap<&str, String> = HashMap::new();
    context.insert("id", id);
    context.insert("title", title);
    context.insert("body", body);

    Template::render("snippets/show", &context)
}

#[get("/snippets/<id>/raw")]
pub fn show_raw_snippet(id: i32, connection: DbConn) -> String {
    let result = match snippet::get(&connection, id) {
        Ok(snippet) => Some(snippet),
        Err(_) => None,
    };

    result.map_or(String::from("Snippet not found"), |snippet| snippet.body)
}

#[post("/api/snippets", format = "application/json", data = "<snippet>")]
pub fn create_snippet(
    snippet: Json<InsertableSnippet>,
    connection: DbConn,
) -> Result<status::Created<Json<Snippet>>, Status> {
    snippet::insert(snippet.into_inner(), &connection)
        .map(|snippet| status::Created(String::from(""), Some(Json(snippet))))
        .map_err(error_response)
}

#[catch(400)]
pub fn bad_request(req: &Request) -> String {
    req.headers().get_one("content-length").map_or(
        String::from("{\"message\": \"Bad request\"}"),
        |length| {
            if length
                .parse::<i32>()
                .expect("Content length is non-numeric")
                > 1_000_000
            {
                String::from("{\"message\": \"Snippet must be under 1mb\"}")
            } else {
                String::from("{\"message\": \"Bad request\"}")
            }
        },
    )
}