#[macro_use] extern crate rocket; mod paste_id; #[cfg(test)] mod tests; use std::io; use rocket::Data; use rocket::response::{content::Plain, Debug}; use rocket::tokio::fs::File; use crate::paste_id::PasteID; const HOST: &str = "http://localhost:8000"; const ID_LENGTH: usize = 3; #[post("/", data = "")] async fn upload(paste: Data) -> Result> { let id = PasteID::new(ID_LENGTH); let filename = format!("upload/{id}", id = id); let url = format!("{host}/{id}\n", host = HOST, id = id); paste.stream_to_file(filename).await?; Ok(url) } #[get("/")] async fn retrieve(id: PasteID<'_>) -> Option> { let filename = format!("upload/{id}", id = id); File::open(&filename).await.map(Plain).ok() } #[get("/")] fn index() -> &'static str { " USAGE POST / accepts raw data in the body of the request and responds with a URL of a page containing the body's content EXAMPLE: curl --data-binary @file.txt http://localhost:8000 GET / retrieves the content for the paste with id `` " } #[launch] fn rocket() -> rocket::Rocket { rocket::ignite().mount("/", routes![index, upload, retrieve]) }