diff --git a/src/models/pretty.rs b/src/models/pretty.rs index a75b8fe..9750dea 100644 --- a/src/models/pretty.rs +++ b/src/models/pretty.rs @@ -1,10 +1,11 @@ use std::fs; +use std::path::PathBuf; use syntect::highlighting::ThemeSet; use syntect::html::highlighted_html_for_string; use syntect::parsing::SyntaxSet; -pub fn get_pretty_body(path: &String, ext: &String) -> String { +pub fn get_pretty_body(path: &PathBuf, ext: &String) -> String { let ss = SyntaxSet::load_defaults_newlines(); let mut theme_cursor = std::io::Cursor::new(include_bytes!("../../themes/ayu_dark.tmTheme")); diff --git a/src/routes/pretty_retrieve.rs b/src/routes/pretty_retrieve.rs index 11804a6..1efd53e 100644 --- a/src/routes/pretty_retrieve.rs +++ b/src/routes/pretty_retrieve.rs @@ -3,22 +3,22 @@ use rocket_dyn_templates::Template; use std::collections::HashMap; use std::path::Path; +use crate::get_upload_dir; use crate::models::paste_id::PasteId; use crate::models::pretty::get_pretty_body; #[get("/p/", rank = 2)] pub async fn pretty_retrieve(id: PasteId<'_>) -> Option