Add invite creation, removal, fetch and usage.

src/guards/channel.rs

@@ -16,6 +16,7 @@ pub struct ChannelRef {
     #[serde(rename = "type")]
     pub channel_type: u8,
 
+    pub name: Option<String>,
     pub last_message: Option<LastMessage>,
 
     // information required for permission calculations
@@ -25,6 +26,31 @@ pub struct ChannelRef {
 }
 
 impl ChannelRef {
+    pub fn from(id: String) -> Option<ChannelRef> {
+        match database::get_collection("channels").find_one(
+            doc! { "_id": id },
+            FindOneOptions::builder()
+                .projection(doc! {
+                    "_id": 1,
+                    "type": 1,
+                    "name": 1,
+                    "last_message": 1,
+                    "recipients": 1,
+                    "guild": 1,
+                    "owner": 1,
+                })
+                .build(),
+        ) {
+            Ok(result) => match result {
+                Some(doc) => {
+                    Some(from_bson(bson::Bson::Document(doc)).expect("Failed to unwrap channel."))
+                }
+                None => None,
+            },
+            Err(_) => None,
+        }
+    }
+
     pub fn fetch_data(&self, projection: Document) -> Option<Document> {
         database::get_collection("channels")
             .find_one(
@@ -39,25 +65,8 @@ impl<'r> FromParam<'r> for ChannelRef {
     type Error = &'r RawStr;
 
     fn from_param(param: &'r RawStr) -> Result<Self, Self::Error> {
-        let id = param.to_string();
-        let result = database::get_collection("channels")
-            .find_one(
-                doc! { "_id": id },
-                FindOneOptions::builder()
-                    .projection(doc! {
-                        "_id": 1,
-                        "type": 1,
-                        "last_message": 1,
-                        "recipients": 1,
-                        "guild": 1,
-                        "owner": 1,
-                    })
-                    .build(),
-            )
-            .unwrap();
-
-        if let Some(channel) = result {
-            Ok(from_bson(bson::Bson::Document(channel)).expect("Failed to deserialize channel."))
+        if let Some(channel) = ChannelRef::from(param.to_string()) {
+            Ok(channel)
         } else {
             Err(param)
         }

src/guards/guild.rs

@@ -5,7 +5,7 @@ use rocket::request::FromParam;
 use serde::{Deserialize, Serialize};
 
 use crate::database;
-use crate::database::guild::{Ban, Member};
+use crate::database::guild::{Ban, Invite, Member};
 
 #[derive(Serialize, Deserialize, Debug, Clone)]
 pub struct GuildRef {
@@ -76,10 +76,10 @@ impl<'r> FromParam<'r> for GuildRef {
     }
 }
 
-pub fn get_member(guild: &GuildRef, member: &String) -> Option<Member> {
+pub fn get_member(guild_id: &String, member: &String) -> Option<Member> {
     if let Ok(result) = database::get_collection("members").find_one(
         doc! {
-            "_id.guild": &guild.id,
+            "_id.guild": &guild_id,
             "_id.user": &member,
         },
         None,
@@ -93,3 +93,43 @@ pub fn get_member(guild: &GuildRef, member: &String) -> Option<Member> {
         None
     }
 }
+
+pub fn get_invite(code: &String) -> Option<(String, String, Invite)> {
+    if let Ok(result) = database::get_collection("guilds").find_one(
+        doc! {
+            "invites": {
+                "$elemMatch": {
+                    "code": &code
+                }
+            }
+        },
+        FindOneOptions::builder()
+            .projection(doc! {
+                "_id": 1,
+                "name": 1,
+                "invites.$": 1,
+            })
+            .build(),
+    ) {
+        if let Some(doc) = result {
+            let invite = doc
+                .get_array("invites")
+                .unwrap()
+                .iter()
+                .next()
+                .unwrap()
+                .as_document()
+                .unwrap();
+
+            Some((
+                doc.get_str("_id").unwrap().to_string(),
+                doc.get_str("name").unwrap().to_string(),
+                from_bson(Bson::Document(invite.clone())).unwrap(),
+            ))
+        } else {
+            None
+        }
+    } else {
+        None
+    }
+}