From 29aa5e2463c68502ebd74143ab3aef5524ce0f6b Mon Sep 17 00:00:00 2001
From: toast <toast@toast-server.net>
Date: Tue, 5 Dec 2023 14:25:20 +1100
Subject: [PATCH] Refactor command registration logic

---
 src/main.rs | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index b8b68da..d158d36 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,19 +11,23 @@ async fn on_ready(
 ) -> Result<(), Error> {
   println!("Connected to API as {}", ready.user.name);
 
-  let builder = poise::builtins::create_application_commands(&framework.options().commands);
-  let commands = serenity::Command::set_global_application_commands(&ctx.http, |commands| {
-    *commands = builder.clone();
-    commands
-  }).await;
+  let register_commands = std::env::var("REGISTER_CMDS").unwrap_or_else(|_| String::from("true")).parse::<bool>().unwrap_or(true);
 
-  match commands {
-    Ok(cmdmap) => {
-      for command in cmdmap.iter() {
-        println!("Registered command globally: {}", command.name);
-      }
-    },
-    Err(why) => println!("Error registering commands: {:?}", why)
+  if register_commands {
+    let builder = poise::builtins::create_application_commands(&framework.options().commands);
+    let commands = serenity::Command::set_global_application_commands(&ctx.http, |commands| {
+      *commands = builder.clone();
+      commands
+    }).await;
+
+    match commands {
+      Ok(cmdmap) => {
+        for command in cmdmap.iter() {
+          println!("Registered command globally: {}", command.name);
+        }
+      },
+      Err(why) => println!("Error registering commands: {:?}", why)
+    }
   }
 
   Ok(())