stoatchat/src/routes/bots/edit.rs

use crate::notifications::events::ClientboundNotification;
use crate::util::result::{Error, Result, EmptyResponse};
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<EmptyResponse> {
    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.owner != user.id {
        return Err(Error::MissingPermission);
    }

    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(EmptyResponse {})
}