1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- 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 (title, body) = result.map_or(
- (
- String::from("404 - Snippet not found"),
- format!("<h3'>{}</h3>", "Snippet not found"),
- ),
- |snippet| (snippet.title, snippet.formatted_body),
- );
-
- let mut context: HashMap<&str, String> = HashMap::new();
- context.insert("title", title);
- context.insert("body", body);
-
- Template::render("snippets/show", &context)
- }
-
- #[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\"}")
- }
- },
- )
- }
|