From 4025afa6ce026aedf711af06a7287e1951f51879 Mon Sep 17 00:00:00 2001
From: jamesbt365 <jamesbt365@gmail.com>
Date: Mon, 20 May 2024 16:54:33 +0100
Subject: [PATCH] Add support for using manual cooldowns on single commands

---
 macros/Cargo.lock            | 2 +-
 macros/src/command/mod.rs    | 4 ++++
 macros/src/command/prefix.rs | 2 +-
 macros/src/command/slash.rs  | 4 ++--
 src/structs/command.rs       | 5 +++++
 5 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/macros/Cargo.lock b/macros/Cargo.lock
index de9713ee6919..c8eb1734ae78 100644
--- a/macros/Cargo.lock
+++ b/macros/Cargo.lock
@@ -51,7 +51,7 @@ checksum = "b9e0384b61958566e926dc50660321d12159025e767c18e043daf26b70104c39"
 
 [[package]]
 name = "poise_macros"
-version = "0.5.7"
+version = "0.6.1"
 dependencies = [
  "darling",
  "proc-macro2",
diff --git a/macros/src/command/mod.rs b/macros/src/command/mod.rs
index 8c7f8bbd6136..d94beb2b2d0f 100644
--- a/macros/src/command/mod.rs
+++ b/macros/src/command/mod.rs
@@ -49,6 +49,8 @@ pub struct CommandArgs {
     category: Option<String>,
     custom_data: Option<syn::Expr>,
 
+    manual_cooldowns: bool,
+
     // In seconds
     global_cooldown: Option<u64>,
     user_cooldown: Option<u64>,
@@ -280,6 +282,7 @@ fn generate_command(mut inv: Invocation) -> Result<proc_macro2::TokenStream, dar
     let description = wrap_option_to_string(inv.description.as_ref());
     let category = wrap_option_to_string(inv.args.category.as_ref());
 
+    let manual_cooldowns = inv.args.manual_cooldowns;
     let cooldown_config = generate_cooldown_config(&inv.args);
 
     let default_member_permissions = &inv.default_member_permissions;
@@ -354,6 +357,7 @@ fn generate_command(mut inv: Invocation) -> Result<proc_macro2::TokenStream, dar
                 description_localizations: #description_localizations,
                 help_text: #help_text,
                 hide_in_help: #hide_in_help,
+                manual_cooldowns: #manual_cooldowns,
                 cooldowns: std::sync::Mutex::new(::poise::Cooldowns::new()),
                 cooldown_config: #cooldown_config,
                 reuse_response: #reuse_response,
diff --git a/macros/src/command/prefix.rs b/macros/src/command/prefix.rs
index 681e86ebaa3a..a7ce403114c3 100644
--- a/macros/src/command/prefix.rs
+++ b/macros/src/command/prefix.rs
@@ -63,7 +63,7 @@ pub fn generate_prefix_action(inv: &Invocation) -> Result<proc_macro2::TokenStre
                 error,
             ))?;
 
-            if !ctx.framework.options.manual_cooldowns {
+            if !ctx.framework.options.manual_cooldowns && !ctx.command.manual_cooldowns {
                 ctx.command.cooldowns.lock().unwrap().start_cooldown(ctx.cooldown_context());
             }
 
diff --git a/macros/src/command/slash.rs b/macros/src/command/slash.rs
index ce27546b3a19..62487b60fef9 100644
--- a/macros/src/command/slash.rs
+++ b/macros/src/command/slash.rs
@@ -185,7 +185,7 @@ pub fn generate_slash_action(inv: &Invocation) -> Result<proc_macro2::TokenStrea
                 #( (#param_names: #param_types), )*
             ).await.map_err(|error| error.to_framework_error(ctx))?;
 
-            if !ctx.framework.options.manual_cooldowns {
+            if !ctx.framework.options.manual_cooldowns && !ctx.command.manual_cooldowns {
                 ctx.command.cooldowns.lock().unwrap().start_cooldown(ctx.cooldown_context());
             }
 
@@ -215,7 +215,7 @@ pub fn generate_context_menu_action(
     Ok(quote::quote! {
         <#param_type as ::poise::ContextMenuParameter<_, _>>::to_action(|ctx, value| {
             Box::pin(async move {
-                if !ctx.framework.options.manual_cooldowns {
+                if !ctx.framework.options.manual_cooldowns && !ctx.command.manual_cooldowns {
                     ctx.command.cooldowns.lock().unwrap().start_cooldown(ctx.cooldown_context());
                 }
 
diff --git a/src/structs/command.rs b/src/structs/command.rs
index 453a4840151f..540965f31d7a 100644
--- a/src/structs/command.rs
+++ b/src/structs/command.rs
@@ -58,6 +58,11 @@ pub struct Command<U, E> {
     /// Multiline description with detailed usage instructions. Displayed in the command specific
     /// help: `~help command_name`
     pub help_text: Option<String>,
+    /// if `true`, disables automatic cooldown handling before this commands invocation.
+    ///
+    /// Will override [`crate::FrameworkOptions::manual_cooldowns`] allowing manual cooldowns
+    /// on select commands.
+    pub manual_cooldowns: bool,
     /// Handles command cooldowns. Mainly for framework internal use
     pub cooldowns: std::sync::Mutex<crate::CooldownTracker>,
     /// Configuration for the [`crate::CooldownTracker`]