66ea75de7f
This is to work with text/plain type files only Signed-off-by: Gunwant Jain <mail@wantguns.dev>
71 lines
1.6 KiB
Rust
71 lines
1.6 KiB
Rust
#![feature(proc_macro_hygiene, decl_macro)]
|
|
#[macro_use] extern crate rocket;
|
|
|
|
extern crate tree_magic;
|
|
|
|
use std::io;
|
|
use std::path::Path;
|
|
use std::fs::File;
|
|
|
|
use rocket::Data;
|
|
use rocket::http::RawStr;
|
|
use rocket_contrib::templates::Template;
|
|
|
|
mod paste_id;
|
|
|
|
use paste_id::PasteId;
|
|
|
|
#[get("/p/<id>")]
|
|
fn pretty_retrieve(id: PasteId) -> Option<File> {
|
|
let filename = format!("upload/{id}", id = id);
|
|
let filepath = Path::new(&filename);
|
|
|
|
match tree_magic::from_filepath(filepath).as_str() {
|
|
"text/plain" => File::open(&filename).ok(),
|
|
_ => Err("does not have the MIME type of a plaintext file").ok()
|
|
}
|
|
}
|
|
|
|
#[get("/<id>")]
|
|
fn retrieve(id: PasteId) -> Option<File> {
|
|
let filename = format!("upload/{id}", id = id);
|
|
|
|
File::open(&filename).ok()
|
|
}
|
|
|
|
#[post("/", data = "<paste>")]
|
|
fn upload(paste: Data) -> Result<String, std::io::Error> {
|
|
let id = PasteId::new(4);
|
|
let filename = format!("upload/{id}", id = id);
|
|
let url = format!("{host}/{id}\n", host = "http://localhost:8000", id = id);
|
|
|
|
paste.stream_to_file(Path::new(&filename))?;
|
|
Ok(url)
|
|
}
|
|
|
|
#[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
|
|
|
|
GET /<id>
|
|
|
|
retrieves the content for the paste with id `<id>`
|
|
|
|
GET /p/<id>
|
|
|
|
retrieves the content for the paste with id `<id>`, with syntax highlighting
|
|
"
|
|
}
|
|
|
|
fn main() {
|
|
rocket::ignite()
|
|
.mount("/", routes![index, upload, retrieve, pretty_retrieve])
|
|
.launch();
|
|
}
|