Adds a new reset command for vertification codes (#56)

Author: fatfingers23

trackscape-discord-bot/src/commands/mod.rs

@@ -4,6 +4,7 @@ pub mod info;
 pub(crate) mod manually_run_wom_sync_command;
 pub mod name_change_command;
 pub mod reset_broadcasts_thresholds;
+pub mod reset_verification_code;
 pub mod set_broadcast_channel;
 pub mod set_clan_chat_channel;
 pub mod set_diary_min_command;

trackscape-discord-bot/src/commands/reset_verification_code.rs

@@ -0,0 +1,28 @@
+use crate::database::BotMongoDb;
+use serenity::all::{CommandDataOption, CreateCommand};
+use serenity::client::Context;
+use serenity::model::prelude::Permissions;
+
+pub fn register() -> CreateCommand {
+    CreateCommand::new("reset_verification_code")
+        .description(
+            "Resets the verification code for the server. MUST UPDATE THE NEW CODE IN THE PLUGIN.",
+        )
+        .default_member_permissions(Permissions::MANAGE_GUILD)
+}
+
+pub async fn run(
+    _options: &[CommandDataOption],
+    _ctx: &Context,
+    db: &BotMongoDb,
+    guild_id: u64,
+) -> Option<String> {
+    let reset_code = db.guilds.reset_verification_code(guild_id).await;
+    match reset_code {
+        Ok(new_code) => Some(format!("The new verification code is: {}", new_code)),
+        Err(e) => {
+            eprintln!("Failed to reset verification code: {:?}", e);
+            Some("Failed to reset verification code.".to_string())
+        }
+    }
+}

trackscape-discord-bot/src/main.rs

@@ -307,6 +307,15 @@ impl EventHandler for Bot {
                     )
                     .await
                 }
+                "reset_verification_code" => {
+                    commands::reset_verification_code::run(
+                        &command.data.options,
+                        &ctx,
+                        &self.mongo_db,
+                        command.guild_id.unwrap().get(),
+                    )
+                    .await
+                }
                 _ => {
                     info!("not implemented :(");
                     None
@@ -349,6 +358,7 @@ fn get_commands() -> Vec<CreateCommand> {
     commands.push(commands::manually_run_wom_sync_command::register());
     commands.push(commands::set_leagues_broadcast_channel::register());
     commands.push(commands::stop_leagues_notifications::register());
+    commands.push(commands::reset_verification_code::register());
     commands
 }
 pub async fn create_commands_for_guild(guild_id: &GuildId, ctx: Context) {

trackscape-discord-shared/src/database/guilds_db.rs

@@ -134,6 +134,34 @@ impl GuildsDb {
         }
     }
 
+    pub async fn reset_verification_code(&self, guild_id: u64) -> Result<String, anyhow::Error> {
+        let saved_guild_query = self.get_by_guild_id(guild_id).await;
+        match saved_guild_query {
+            Ok(saved_guild) => match saved_guild {
+                Some(mut guild) => {
+                    let check_for_unique_code = self
+                        .recursive_check_for_unique_code(guild.verification_code.clone())
+                        .await;
+                    match check_for_unique_code {
+                        Ok(new_code) => {
+                            guild.verification_code = new_code.clone();
+                            guild.hashed_verification_code = hash_string(new_code.clone());
+                            self.update_guild(guild).await;
+                            Ok(new_code)
+                        }
+                        Err(_) => Err(anyhow::Error::msg("Failed to reset verification code.")),
+                    }
+                }
+                None => Err(anyhow::Error::msg(
+                    "Could not find a clan with that guild id.",
+                )),
+            },
+            Err(_) => Err(anyhow::Error::msg(
+                "Failed to get the saved guild to  reset verification code.",
+            )),
+        }
+    }
+
     pub async fn get_guild_by_code_and_clan_name(
         &self,
         code: String,