use std::io; use std::path::{Path, PathBuf}; use rocket::Route; use rocket::response::NamedFile; use rocket_contrib::{Json, Value}; use auth::Headers; use CONFIG; pub fn routes() -> Vec { routes![index, files, attachments, alive] } // TODO: Might want to use in memory cache: https://github.com/hgzimmerman/rocket-file-cache #[get("/")] fn index() -> io::Result { NamedFile::open(Path::new(&CONFIG.web_vault_folder).join("index.html")) } #[get("/")] // Only match this if the other routes don't match fn files(p: PathBuf) -> io::Result { NamedFile::open(Path::new(&CONFIG.web_vault_folder).join(p)) } #[get("/attachments//")] fn attachments(uuid: String, file: PathBuf, headers: Headers) -> io::Result { if uuid != headers.user.uuid { return Err(io::Error::new(io::ErrorKind::PermissionDenied, "Permission denied")); } NamedFile::open(Path::new(&CONFIG.attachments_folder).join(file)) } #[get("/alive")] fn alive() -> Json { use util::format_date; use chrono::{NaiveDateTime, Utc}; Json(format_date(&Utc::now().naive_utc())) }