Add bot edit route for #8.

src/notifications/events.rs

@@ -69,6 +69,11 @@ pub enum RemoveMemberField {
     Avatar,
 }
 
+#[derive(Serialize, Deserialize, Debug, Clone)]
+pub enum RemoveBotField {
+    InteractionsURL,
+}
+
 #[derive(Serialize, Deserialize, Debug, Clone)]
 #[serde(tag = "type")]
 pub enum ClientboundNotification {

src/routes/bots/edit.rs

@@ -0,0 +1,119 @@
+use crate::notifications::events::ClientboundNotification;
+use crate::util::result::{Error, Result};
+use crate::{database::*, notifications::events::RemoveBotField};
+
+use mongodb::bson::doc;
+use regex::Regex;
+use rocket::serde::json::Json;
+use serde::{Deserialize, Serialize};
+use validator::Validate;
+
+// ! FIXME: should be global somewhere; maybe use config(?)
+// ! tip: CTRL + F, RE_USERNAME
+lazy_static! {
+    static ref RE_USERNAME: Regex = Regex::new(r"^[a-zA-Z0-9_.]+$").unwrap();
+}
+
+#[derive(Validate, Serialize, Deserialize)]
+pub struct Data {
+    #[validate(length(min = 2, max = 32), regex = "RE_USERNAME")]
+    #[serde(skip_serializing_if = "Option::is_none")]
+    name: Option<String>,
+    public: Option<bool>,
+    interactions_url: Option<String>,
+    remove: Option<RemoveBotField>,
+}
+
+#[patch("/<target>", data = "<data>")]
+pub async fn edit_bot(user: User, target: Ref, data: Json<Data>) -> Result<()> {
+    let data = data.into_inner();
+    data.validate()
+        .map_err(|error| Error::FailedValidation { error })?;
+
+    if data.name.is_none()
+        && data.public.is_none()
+        && data.interactions_url.is_none()
+        && data.remove.is_none()
+    {
+        return Ok(());
+    }
+
+    let bot = target.fetch_bot().await?;
+
+    if !bot.public {
+        if bot.owner != user.id {
+            return Err(Error::BotIsPrivate);
+        }
+    }
+
+    if let Some(name) = &data.name {
+        if User::is_username_taken(&name).await? {
+            return Err(Error::UsernameTaken);
+        }
+
+        get_collection("users")
+            .update_one(
+                doc! { "_id": &target.id },
+                doc! {
+                    "$set": {
+                        "username": name
+                    }
+                },
+                None
+            )
+            .await
+            .map_err(|_| Error::DatabaseError {
+                operation: "update_one",
+                with: "user",
+            })?;
+
+        ClientboundNotification::UserUpdate {
+            id: target.id.clone(),
+            data: json!({
+                "username": name
+            }),
+            clear: None,
+        }
+        .publish_as_user(target.id.clone());
+    }
+
+    let mut set = doc! {};
+    let mut unset = doc! {};
+
+    if let Some(remove) = &data.remove {
+        match remove {
+            RemoveBotField::InteractionsURL => {
+                unset.insert("interactions_url", 1);
+            }
+        }
+    }
+
+    if let Some(public) = &data.public {
+        set.insert("public", public);
+    }
+
+    if let Some(interactions_url) = &data.interactions_url {
+        set.insert("interactions_url", interactions_url);
+    }
+
+    let mut operations = doc! {};
+    if set.len() > 0 {
+        operations.insert("$set", &set);
+    }
+
+    if unset.len() > 0 {
+        operations.insert("$unset", unset);
+    }
+
+    if operations.len() > 0 {
+        get_collection("bots")
+            .update_one(doc! { "_id": &target.id }, operations, None)
+            .await
+            .map_err(|_| Error::DatabaseError {
+                operation: "update_one",
+                with: "bot",
+            })?;
+    }
+
+    Ok(())
+}

src/routes/bots/mod.rs

@@ -5,6 +5,7 @@ mod invite;
 mod fetch_public;
 mod fetch;
 mod fetch_owned;
+mod edit;
 
 pub fn routes() -> Vec<Route> {
     routes![
@@ -13,5 +14,6 @@ pub fn routes() -> Vec<Route> {
         fetch_public::fetch_public_bot,
         fetch::fetch_bot,
         fetch_owned::fetch_owned_bots,
+        edit::edit_bot,
     ]
 }