implement blog tag filtering

3 files changed, 74 insertions, 35 deletions

diff --git a/src/main.rs b/src/main.rs
index c08c2e1..40f546d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,6 +5,7 @@ mod scrobbles;
 mod skweets;
 
 use std::borrow::Cow;
+use std::collections::HashSet;
 
 use rocket::fs::{relative, FileServer};
 use rocket::http::Status;
@@ -47,24 +48,6 @@ async fn home(clients: &State<Clients>) -> Template {
     )
 }
 
-#[get("/blog")]
-async fn blog() -> Template {
-    let mut blogposts = posts::get_blogposts().await.unwrap_or_default();
-    let tags = posts::get_tags(&blogposts);
-    for blogpost in &mut blogposts {
-        blogpost.render().await;
-    }
-    let reverse = "reverse".to_owned();
-    Template::render(
-        "blog",
-        context! {
-            reverse,
-            blogposts,
-            tags,
-        },
-    )
-}
-
 #[get("/blog/<blogpost>")]
 async fn blogpost(blogpost: &str) -> Result<Template> {
     let mut blogpost = posts::get_blogpost(blogpost).await?;
@@ -77,6 +60,33 @@ async fn blogpost(blogpost: &str) -> Result<Template> {
     ))
 }
 
+#[get("/blog?<filter>")]
+async fn blog(filter: Vec<String>) -> Result<Template> {
+    let mut blogposts = posts::get_blogposts().await?;
+    let tags: Vec<String> = posts::get_tags(&blogposts)
+        .into_iter()
+        .map(|tag| tag.to_owned())
+        .collect();
+    let mut filter_hashset: HashSet<String> = HashSet::new();
+    if !filter.is_empty() {
+        filter_hashset.extend(filter.into_iter());
+        blogposts = posts::filter_by_tags(blogposts, &filter_hashset);
+    }
+    for blogpost in &mut blogposts {
+        blogpost.render().await?;
+    }
+    let reverse = true;
+    Ok(Template::render(
+        "blog",
+        context! {
+            reverse,
+            tags,
+            filter_hashset,
+            blogposts,
+        },
+    ))
+}
+
 #[get("/contact")]
 async fn contact() -> Template {
     Template::render("contact", context! {})
diff --git a/src/posts/mod.rs b/src/posts/mod.rs
index be1e1a9..2dc5c4b 100644
--- a/src/posts/mod.rs
+++ b/src/posts/mod.rs
@@ -1,3 +1,6 @@
+mod article;
+mod note;
+
 use std::collections::HashSet;
 
 use async_trait::async_trait;
@@ -16,12 +19,6 @@ enum PostType {
     Note,
 }
 
-enum TextFormat {
-    Markdown,
-    Plaintext,
-    Html,
-}
-
 #[derive(Debug)]
 pub struct Post<D: Content> {
     // id: i64,
@@ -34,7 +31,7 @@ pub struct Post<D: Content> {
     data: D,
 }
 
-impl<D: Content + Serialize> Post<D> {
+impl<D: Content> Post<D> {
     pub async fn render(&mut self) -> Result<()> {
         self.render = Some(self.data.render().await?);
         Ok(())
@@ -74,6 +71,8 @@ pub async fn get_blogposts() -> Result<Vec<Post<Article>>> {
         let blogpost = Post::try_from(blogpost).await.unwrap();
         blogposts.push(blogpost);
     }
+    blogposts.sort_by_key(|post| post.created_at);
+    blogposts.reverse();
     Ok(blogposts)
 }
 
@@ -87,21 +86,45 @@ pub async fn get_blogpost(post_name: &str) -> Result<Post<Article>> {
                 path: file.path().to_str().unwrap_or_default().to_owned(),
                 name: name.to_owned(),
             };
-            let blogpost = Post::try_from(blogpost).await.unwrap();
+            let blogpost = Post::try_from(blogpost).await?;
             return Ok(blogpost);
         }
     }
     Err(BlossomError::NotFound(Status::new(404)))
 }
 
-pub fn get_tags<D: Content>(posts: &Vec<Post<D>>) -> HashSet<String> {
-    posts.into_iter().fold(HashSet::new(), |mut acc, post| {
-        let tags = &post.tags;
-        for tag in tags {
-            acc.insert(tag.to_owned());
-        }
-        acc
-    })
+pub fn get_tags<D: Content>(posts: &Vec<Post<D>>) -> Vec<&String> {
+    let mut tags = posts
+        .into_iter()
+        .fold(HashSet::new(), |mut acc, post| {
+            let tags = &post.tags;
+            for tag in tags {
+                acc.insert(tag);
+            }
+            acc
+        })
+        .into_iter()
+        .collect::<Vec<_>>();
+    tags.sort();
+    tags
+}
+
+pub fn filter_by_tags<'p, D: Content>(
+    posts: Vec<Post<D>>,
+    filter_tags: &HashSet<String>,
+) -> Vec<Post<D>> {
+    posts
+        .into_iter()
+        .filter(|post| {
+            for tag in &post.tags {
+                match filter_tags.contains(tag) {
+                    true => return true,
+                    false => continue,
+                }
+            }
+            false
+        })
+        .collect()
 }
 
 #[async_trait]
diff --git a/src/posts/note.rs b/src/posts/note.rs
index 95655d4..d7fa6c9 100644
--- a/src/posts/note.rs
+++ b/src/posts/note.rs
@@ -1 +1,7 @@
-struct note
+enum TextFormat {
+    Markdown,
+    Plaintext,
+    Html,
+}
+
+struct Note {}