database: add add_to_feed and remove_from_feed

Some database backends may have optimized ways of tracking feed
contents. Others might just use the "children" property directly.

2 files changed, 9 insertions, 2 deletions

diff --git a/kittybox-rs/src/database/mod.rs b/kittybox-rs/src/database/mod.rs
index aa675d3..cb4c0b9 100644
--- a/kittybox-rs/src/database/mod.rs
+++ b/kittybox-rs/src/database/mod.rs
@@ -293,6 +293,14 @@ pub trait Storage: std::fmt::Debug + Clone + Send + Sync {
     /// Note that the `post` object MUST have `post["properties"]["uid"][0]` defined.
     async fn put_post(&self, post: &'_ serde_json::Value, user: &'_ str) -> Result<()>;
 
+    /// Add post to feed. Some database implementations might have optimized ways to do this.
+    async fn add_to_feed(&self, feed: &'_ str, post: &'_ str) -> Result<()> {
+        self.update_post(feed, serde_json::json!({"add": {"children": [post]}})).await
+    }
+    async fn remove_from_feed(&self, feed: &'_ str, post: &'_ str) -> Result<()> {
+        self.update_post(feed, serde_json::json!({"delete": {"children": [post]}})).await
+    }
+
     /// Modify a post using an update object as defined in the Micropub spec.
     ///
     /// Note to implementors: the update operation MUST be atomic and
diff --git a/kittybox-rs/src/micropub/mod.rs b/kittybox-rs/src/micropub/mod.rs
index 5c297f9..da9c6a5 100644
--- a/kittybox-rs/src/micropub/mod.rs
+++ b/kittybox-rs/src/micropub/mod.rs
@@ -294,8 +294,7 @@ pub(crate) async fn _post<D: 'static + Storage>(
 
     for chan in &mut channels {
         if db.post_exists(chan).await? {
-            db.update_post(chan, json!({"add": {"children": [uid]}}))
-                .await?;
+            db.add_to_feed(chan, &uid).await?;
         } else if default_channels.iter().any(|i| chan == i) {
             util::create_feed(&db, &uid, chan, user).await?;
         } else {