feat(snippet): add base for snippet command

Author: Akinator31

migrations/20251125000000_create_snippets.sql

@@ -0,0 +1,11 @@
+-- Add migration script here
+CREATE TABLE IF NOT EXISTS "snippets" (
+    "id" INTEGER PRIMARY KEY AUTOINCREMENT,
+    "key" TEXT NOT NULL UNIQUE,
+    "content" TEXT NOT NULL,
+    "created_by" TEXT NOT NULL,
+    "created_at" DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE INDEX IF NOT EXISTS "idx_snippets_key" ON "snippets"("key");

rustmail/src/bot.rs

@@ -144,6 +144,7 @@ pub async fn run_bot(
     registry.register_command(TakeCommand);
     registry.register_command(ReleaseCommand);
     registry.register_command(PingCommand);
+    registry.register_command(SnippetCommand);
 
     let registry = Arc::new(registry);
 

rustmail/src/commands/anonreply/text_command/anonreply.rs

@@ -21,7 +21,24 @@ pub async fn anonreply(
         .as_ref()
         .ok_or_else(database_connection_failed)?;
 
-    let content = extract_reply_content(&msg.content, &config.command.prefix, &["anonreply", "ar"]);
+    let mut content = extract_reply_content(&msg.content, &config.command.prefix, &["anonreply", "ar"]);
+    
+    if let Some(text) = &content {
+        if let Some(stripped) = text.strip_prefix("{{").and_then(|s| s.strip_suffix("}}")) {
+            let snippet_key = stripped.trim();
+            match get_snippet_by_key(snippet_key, db_pool).await? {
+                Some(snippet) => {
+                    content = Some(snippet.content);
+                }
+                None => {
+                    msg.reply(&ctx.http, format!("❌ Snippet `{}` not found.", snippet_key))
+                        .await?;
+                    return Ok(());
+                }
+            }
+        }
+    }
+
     let intent = extract_intent(content, &msg.attachments).await;
 
     let Some(intent) = intent else {

rustmail/src/commands/mod.rs

@@ -28,6 +28,7 @@ pub mod release;
 pub mod remove_reminder;
 pub mod remove_staff;
 pub mod reply;
+pub mod snippet;
 pub mod take;
 
 pub use add_reminder::*;
@@ -49,6 +50,7 @@ pub use release::*;
 pub use remove_reminder::*;
 pub use remove_staff::*;
 pub use reply::*;
+pub use snippet::*;
 pub use take::*;
 
 pub type BoxFuture<'a, T> = Pin<Box<dyn Future<Output = T> + Send + 'a>>;

rustmail/src/commands/reply/slash_command/reply.rs

@@ -79,7 +79,15 @@ impl RegistrableCommand for ReplyCommand {
                             "message",
                             message_desc,
                         )
-                        .required(true),
+                        .required(false),
+                    )
+                    .add_option(
+                        CreateCommandOption::new(
+                            CommandOptionType::String,
+                            "snippet",
+                            "Use a snippet instead of typing a message",
+                        )
+                        .required(false),
                     )
                     .add_option(
                         CreateCommandOption::new(
@@ -119,6 +127,7 @@ impl RegistrableCommand for ReplyCommand {
             defer_response(&ctx, &command).await?;
 
             let mut content: Option<String> = None;
+            let mut snippet_key: Option<String> = None;
             let mut attachments: Vec<Attachment> = Vec::new();
             let mut anonymous: bool = false;
 
@@ -127,6 +136,9 @@ impl RegistrableCommand for ReplyCommand {
                     CommandDataOptionValue::String(val) if option.name == "message" => {
                         content = Some(val.clone());
                     }
+                    CommandDataOptionValue::String(val) if option.name == "snippet" => {
+                        snippet_key = Some(val.clone());
+                    }
                     CommandDataOptionValue::Attachment(att_id) if option.name == "attachment" => {
                         if let Some(att) = command.data.resolved.attachments.get(att_id) {
                             attachments.push(att.clone());
@@ -138,6 +150,19 @@ impl RegistrableCommand for ReplyCommand {
                     _ => {}
                 }
             }
+            
+            if let Some(key) = snippet_key {
+                match get_snippet_by_key(&key, db_pool).await? {
+                    Some(snippet) => {
+                        content = Some(snippet.content);
+                    }
+                    None => {
+                        return Err(ModmailError::Command(CommandError::CommandFailed(
+                            format!("Snippet '{}' not found", key),
+                        )));
+                    }
+                }
+            }
 
             let intent = extract_intent(content, &attachments).await;
 

rustmail/src/commands/reply/text_command/reply.rs

@@ -21,7 +21,24 @@ pub async fn reply(
         .as_ref()
         .ok_or_else(database_connection_failed)?;
 
-    let content = extract_reply_content(&msg.content, &config.command.prefix, &["reply", "r"]);
+    let mut content = extract_reply_content(&msg.content, &config.command.prefix, &["reply", "r"]);
+    
+    if let Some(text) = &content {
+        if let Some(stripped) = text.strip_prefix("{{").and_then(|s| s.strip_suffix("}}")) {
+            let snippet_key = stripped.trim();
+            match get_snippet_by_key(snippet_key, db_pool).await? {
+                Some(snippet) => {
+                    content = Some(snippet.content);
+                }
+                None => {
+                    msg.reply(&ctx.http, format!("❌ Snippet `{}` not found.", snippet_key))
+                        .await?;
+                    return Ok(());
+                }
+            }
+        }
+    }
+
     let intent = extract_intent(content, &msg.attachments).await;
 
     let Some(intent) = intent else {

rustmail/src/commands/snippet/mod.rs

@@ -0,0 +1,5 @@
+pub mod slash_command;
+pub mod text_command;
+
+pub use slash_command::*;
+pub use text_command::*;

rustmail/src/commands/snippet/slash_command/mod.rs

@@ -0,0 +1,3 @@
+pub mod snippet;
+
+pub use snippet::*;