use atom_syndication::{ContentBuilder, Entry, EntryBuilder, LinkBuilder, Person, PersonBuilder}; use rocket::{ http::{RawStr, uri::{FromUriParam, Query}}, request::FromFormValue, response::NamedFile, }; use std::path::{Path, PathBuf}; use plume_models::{Connection, posts::Post}; const ITEMS_PER_PAGE: i32 = 12; #[derive(Copy, Clone, UriDisplayQuery)] pub struct Page(i32); impl<'v> FromFormValue<'v> for Page { type Error = &'v RawStr; fn from_form_value(form_value: &'v RawStr) -> Result { match form_value.parse::() { Ok(page) => Ok(Page(page)), _ => Err(form_value), } } } impl FromUriParam> for Page { type Target = Page; fn from_uri_param(val: Option) -> Page { val.unwrap_or_default() } } impl Page { /// Computes the total number of pages needed to display n_items pub fn total(n_items: i32) -> i32 { if n_items % ITEMS_PER_PAGE == 0 { n_items / ITEMS_PER_PAGE } else { (n_items / ITEMS_PER_PAGE) + 1 } } pub fn limits(&self) -> (i32, i32) { ((self.0 - 1) * ITEMS_PER_PAGE, self.0 * ITEMS_PER_PAGE) } } impl Default for Page { fn default() -> Self { Page(1) } } pub fn post_to_atom(post: Post, conn: &Connection) -> Entry { EntryBuilder::default() .title(format!("", post.title)) .content(ContentBuilder::default() .value(format!("", *post.content.get())) .src(post.ap_url.clone()) .content_type("html".to_string()) .build().expect("Atom feed: content error")) .authors(post.get_authors(&*conn).expect("Atom feed: author error") .into_iter() .map(|a| PersonBuilder::default() .name(a.display_name) .uri(a.ap_url) .build().expect("Atom feed: author error")) .collect::>()) .links(vec![LinkBuilder::default().href(post.ap_url).build().expect("Atom feed: link error")]) .build().expect("Atom feed: entry error") } pub mod blogs; pub mod comments; pub mod errors; pub mod instance; pub mod likes; pub mod medias; pub mod notifications; pub mod posts; pub mod reshares; pub mod session; pub mod tags; pub mod user; pub mod search; pub mod well_known; #[get("/static/", rank = 2)] pub fn static_files(file: PathBuf) -> Option { NamedFile::open(Path::new("static/").join(file)).ok() }