add: ability to join/leave/be banned from communities

crates/core/src/database/memberships.rs

@@ -28,7 +28,7 @@ impl DataManager {
         }
     }
 
-    auto_method!(get_membership_by_id()@get_membership_from_row -> "SELECT * FROM memberships WHERE id = $1" --name="journal membership" --returns=CommunityMembership --cache-key-tmpl="atto.membership:{}");
+    auto_method!(get_membership_by_id()@get_membership_from_row -> "SELECT * FROM memberships WHERE id = $1" --name="community membership" --returns=CommunityMembership --cache-key-tmpl="atto.membership:{}");
 
     /// Replace a list of community memberships with the proper community.
     pub async fn fill_communities(&self, list: Vec<CommunityMembership>) -> Result<Vec<Community>> {
@@ -73,7 +73,7 @@ impl DataManager {
 
         let res = query_rows!(
             &conn,
-            "SELECT * FROM memberships WHERE owner = $1",
+            "SELECT * FROM memberships WHERE owner = $1 AND role IS NOT 33",
             &[&(owner as i64)],
             |x| { Self::get_membership_from_row(x) }
         );
@@ -90,6 +90,16 @@ impl DataManager {
     /// # Arguments
     /// * `data` - a mock [`CommunityMembership`] object to insert
     pub async fn create_membership(&self, data: CommunityMembership) -> Result<()> {
+        // make sure membership doesn't already exist
+        if self
+            .get_membership_by_owner_community(data.owner, data.community)
+            .await
+            .is_ok()
+        {
+            return Err(Error::MiscError("Already joined community".to_string()));
+        }
+
+        // ...
         let conn = match self.connect().await {
             Ok(c) => c,
             Err(e) => return Err(Error::DatabaseConnection(e.to_string())),

crates/core/src/database/posts.rs

@@ -167,6 +167,35 @@ impl DataManager {
         Ok(res.unwrap())
     }
 
+    /// Check if the given `uid` can post in the given `community`.
+    pub async fn check_can_post(&self, community: &Community, uid: usize) -> bool {
+        match community.write_access {
+            CommunityWriteAccess::Owner => {
+                if uid != community.owner {
+                    false
+                } else {
+                    true
+                }
+            }
+            CommunityWriteAccess::Joined => {
+                match self
+                    .get_membership_by_owner_community(uid, community.id)
+                    .await
+                {
+                    Ok(m) => {
+                        if !m.role.check_member() {
+                            false
+                        } else {
+                            true
+                        }
+                    }
+                    Err(_) => false,
+                }
+            }
+            _ => true,
+        }
+    }
+
     /// Create a new journal entry in the database.
     ///
     /// # Arguments
@@ -185,22 +214,9 @@ impl DataManager {
             Err(e) => return Err(e),
         };
 
-        match community.write_access {
-            CommunityWriteAccess::Owner => {
-                if data.owner != community.owner {
-                    return Err(Error::NotAllowed);
-                }
-            }
-            CommunityWriteAccess::Joined => {
-                if let Err(_) = self
-                    .get_membership_by_owner_community(data.owner, community.id)
-                    .await
-                {
-                    return Err(Error::NotAllowed);
-                }
-            }
-            _ => (),
-        };
+        if !self.check_can_post(&community, data.owner).await {
+            return Err(Error::NotAllowed);
+        }
 
         // check if we're blocked
         if let Some(replying_to) = data.replying_to {

crates/core/src/database/reactions.rs

@@ -57,7 +57,7 @@ impl DataManager {
         Ok(res.unwrap())
     }
 
-    /// Create a new journal membership in the database.
+    /// Create a new reaction in the database.
     ///
     /// # Arguments
     /// * `data` - a mock [`Reaction`] object to insert

crates/core/src/model/communities_permissions.rs

@@ -13,6 +13,7 @@ bitflags! {
         const MEMBER = 1 << 2;
         const MANAGE_POSTS = 1 << 3;
         const MANAGE_ROLES = 1 << 4;
+        const BANNED = 1 << 5;
 
         const _ = !0;
     }
@@ -89,6 +90,9 @@ impl CommunityPermission {
         if (self & CommunityPermission::ADMINISTRATOR) == CommunityPermission::ADMINISTRATOR {
             // has administrator permission, meaning everything else is automatically true
             return true;
+        } else if (self & CommunityPermission::BANNED) == CommunityPermission::BANNED {
+            // has banned permission, meaning everything else is automatically false
+            return false;
         }
 
         (self & permission) == permission
@@ -107,6 +111,6 @@ impl CommunityPermission {
 
 impl Default for CommunityPermission {
     fn default() -> Self {
-        Self::DEFAULT
+        Self::DEFAULT | Self::MEMBER
     }
 }