Servers: Add invites. Create, view and join.

src/database/entities/channel.rs

@@ -3,7 +3,7 @@ use crate::notifications::events::ClientboundNotification;
 use crate::util::result::{Error, Result};
 use futures::StreamExt;
 use mongodb::{
-    bson::{doc, from_document, to_document, Document},
+    bson::{doc, to_document, Document},
     options::FindOptions,
 };
 use rocket_contrib::json::JsonValue;
@@ -76,22 +76,6 @@ impl Channel {
         }
     }
 
-    pub async fn get(id: &str) -> Result<Channel> {
-        let doc = get_collection("channels")
-            .find_one(doc! { "_id": id }, None)
-            .await
-            .map_err(|_| Error::DatabaseError {
-                operation: "find_one",
-                with: "channel",
-            })?
-            .ok_or_else(|| Error::UnknownChannel)?;
-
-        from_document::<Channel>(doc).map_err(|_| Error::DatabaseError {
-            operation: "from_document",
-            with: "channel",
-        })
-    }
-
     pub async fn publish(self) -> Result<()> {
         get_collection("channels")
             .insert_one(

src/database/entities/invites.rs

@@ -0,0 +1,88 @@
+use mongodb::bson::doc;
+use mongodb::bson::from_document;
+use mongodb::bson::to_document;
+use serde::{Deserialize, Serialize};
+
+use crate::database::get_collection;
+use crate::util::result::Error;
+use crate::util::result::Result;
+
+#[derive(Serialize, Deserialize, Debug, Clone)]
+#[serde(tag = "type")]
+pub enum Invite {
+    Server {
+        #[serde(rename = "_id")]
+        code: String,
+        creator: String,
+        channel: String,
+    },
+    Group {
+        #[serde(rename = "_id")]
+        code: String,
+        creator: String,
+        channel: String,
+    }, /* User {
+           code: String,
+           user: String
+       } */
+}
+
+impl Invite {
+    pub fn code(&self) -> &String {
+        match &self {
+            Invite::Server { code, .. } => code,
+            Invite::Group { code, .. } => code,
+        }
+    }
+
+    pub async fn get(code: &str) -> Result<Invite> {
+        let doc = get_collection("invites")
+            .find_one(doc! { "_id": code }, None)
+            .await
+            .map_err(|_| Error::DatabaseError {
+                operation: "find_one",
+                with: "invite",
+            })?
+            .ok_or_else(|| Error::UnknownServer)?;
+
+        from_document::<Invite>(doc).map_err(|_| Error::DatabaseError {
+            operation: "from_document",
+            with: "invite",
+        })
+    }
+
+    pub async fn save(self) -> Result<()> {
+        get_collection("invites")
+            .insert_one(
+                to_document(&self).map_err(|_| Error::DatabaseError {
+                    operation: "to_bson",
+                    with: "invite",
+                })?,
+                None,
+            )
+            .await
+            .map_err(|_| Error::DatabaseError {
+                operation: "insert_one",
+                with: "invite",
+            })?;
+
+        Ok(())
+    }
+
+    pub async fn delete(&self) -> Result<()> {
+        get_collection("invites")
+            .delete_one(
+                doc! {
+                    "_id": self.code()
+                },
+                None,
+            )
+            .await
+            .map_err(|_| Error::DatabaseError {
+                operation: "delete_one",
+                with: "invite",
+            })?;
+
+        Ok(())
+    }
+}

src/database/entities/mod.rs

@@ -1,4 +1,5 @@
 mod channel;
+mod invites;
 mod message;
 mod microservice;
 mod server;
@@ -9,6 +10,7 @@ use microservice::*;
 
 pub use autumn::*;
 pub use channel::*;
+pub use invites::*;
 pub use january::*;
 pub use message::*;
 pub use server::*;

src/database/entities/server.rs

@@ -17,14 +17,11 @@ pub struct MemberCompositeKey {
 pub struct Member {
     #[serde(rename = "_id")]
     pub id: MemberCompositeKey,
-    pub nickname: Option<String>,
-}
 
-#[derive(Serialize, Deserialize, Debug, Clone)]
-pub struct Invite {
-    pub code: String,
-    pub creator: String,
-    pub channel: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub nickname: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub avatar: Option<File>,
 }
 
 #[derive(Serialize, Deserialize, Debug, Clone)]
@@ -54,22 +51,6 @@ pub struct Server {
 }
 
 impl Server {
-    pub async fn get(id: &str) -> Result<Server> {
-        let doc = get_collection("servers")
-            .find_one(doc! { "_id": id }, None)
-            .await
-            .map_err(|_| Error::DatabaseError {
-                operation: "find_one",
-                with: "server",
-            })?
-            .ok_or_else(|| Error::UnknownServer)?;
-
-        from_document::<Server>(doc).map_err(|_| Error::DatabaseError {
-            operation: "from_document",
-            with: "server",
-        })
-    }
-
     pub async fn publish(self) -> Result<()> {
         get_collection("servers")
             .insert_one(
@@ -105,4 +86,30 @@ impl Server {
     pub async fn delete(&self) -> Result<()> {
         unimplemented!()
     }
+
+    pub async fn join_member(&self, id: &str) -> Result<()> {
+        get_collection("server_members")
+            .insert_one(
+                doc! {
+                    "_id": {
+                        "server": &self.id,
+                        "user": &id
+                    }
+                },
+                None,
+            )
+            .await
+            .map_err(|_| Error::DatabaseError {
+                operation: "insert_one",
+                with: "server_members",
+            })?;
+        
+        ClientboundNotification::ServerMemberJoin {
+            id: self.id.clone(),
+            user: id.to_string()
+        }
+        .publish(self.id.clone());
+
+        Ok(())
+    }
 }