chore(monorepo): delta, january, quark

crates/quark/src/impl/mongo/users/bot.rs

@@ -0,0 +1,68 @@
+use crate::models::bot::{Bot, FieldsBot, PartialBot};
+use crate::r#impl::mongo::IntoDocumentPath;
+use crate::{AbstractBot, Result};
+
+use super::super::MongoDb;
+
+static COL: &str = "bots";
+
+#[async_trait]
+impl AbstractBot for MongoDb {
+    async fn fetch_bot(&self, id: &str) -> Result<Bot> {
+        self.find_one_by_id(COL, id).await
+    }
+
+    async fn fetch_bot_by_token(&self, token: &str) -> Result<Bot> {
+        self.find_one(
+            COL,
+            doc! {
+                "token": token
+            },
+        )
+        .await
+    }
+
+    async fn insert_bot(&self, bot: &Bot) -> Result<()> {
+        self.insert_one(COL, &bot).await.map(|_| ())
+    }
+
+    async fn update_bot(&self, id: &str, bot: &PartialBot, remove: Vec<FieldsBot>) -> Result<()> {
+        self.update_one_by_id(
+            COL,
+            id,
+            bot,
+            remove.iter().map(|x| x as &dyn IntoDocumentPath).collect(),
+            None,
+        )
+        .await
+        .map(|_| ())
+    }
+
+    async fn delete_bot(&self, id: &str) -> Result<()> {
+        self.delete_one_by_id(COL, id).await.map(|_| ())
+    }
+
+    async fn fetch_bots_by_user(&self, user_id: &str) -> Result<Vec<Bot>> {
+        self.find(
+            COL,
+            doc! {
+                "owner": user_id
+            },
+        )
+        .await
+    }
+
+    async fn get_number_of_bots_by_user(&self, user_id: &str) -> Result<usize> {
+        // ! FIXME: move this to generic?
+        self.fetch_bots_by_user(user_id).await.map(|x| x.len())
+    }
+}
+
+impl IntoDocumentPath for FieldsBot {
+    fn as_path(&self) -> Option<&'static str> {
+        match self {
+            FieldsBot::InteractionsURL => Some("interactions_url"),
+            FieldsBot::Token => None,
+        }
+    }
+}