Merge pull request #171 from Rustmail/169-add-reminder-command

feat(commands): add add_reminder command

Author: Akinator31

.sqlx/query-1f2332afc3824ca0790ef097ff1d48074feaa48de9844e0e30d7e6fe96ad5e44.json

@@ -13,6 +13,7 @@ toml = "0.9.7"
 reqwest = "0.12.23"
 async-trait = "0.1.89"
 chrono = { version = "0.4.42", features = ["serde"] }
+regex = "1.11.3"
 
 [dependencies.uuid]
 version = "1.18.1"

.sqlx/query-8d256f5308e9c3af3f409d5097302f9794287b06f89da6c7f4a86efcd37b151a.json

@@ -0,0 +1,14 @@
+-- Create reminders table
+
+CREATE TABLE IF NOT EXISTS reminders (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    thread_id TEXT NOT NULL,
+    user_id BIGINT NOT NULL,
+    channel_id BIGINT NOT NULL,
+    guild_id BIGINT NOT NULL,
+    reminder_content TEXT NOT NULL,
+    trigger_time INTEGER NOT NULL,
+    created_at INTEGER NOT NULL DEFAULT (strftime('%s', 'now')),
+    completed BOOLEAN NOT NULL DEFAULT 0,
+    FOREIGN KEY (thread_id) REFERENCES threads(id) ON DELETE CASCADE ON UPDATE CASCADE
+);

.sqlx/query-b9420d5d850bf8fc05beb8594611533139e47e3df927153e8cb15fe7451fdc4b.json

@@ -0,0 +1,167 @@
+use crate::config::Config;
+use crate::db::reminders::{update_reminder_status, Reminder};
+use crate::utils::conversion::hex_string_to_int::hex_string_to_int;
+use crate::utils::message::message_builder::MessageBuilder;
+use chrono::Local;
+use serenity::all::{ChannelId, CommandInteraction, Context, Message, UserId};
+use sqlx::SqlitePool;
+use std::collections::HashMap;
+use std::time::Duration;
+use tokio::time::sleep;
+
+pub async fn send_register_confirmation_from_message(
+    reminder_id: i64,
+    reminder_content: &str,
+    ctx: &Context,
+    msg: &Message,
+    config: &Config,
+    trigger_timestamp: i64,
+) {
+    let mut params = HashMap::new();
+    params.insert("time".to_string(), format!("<t:{}:F>", trigger_timestamp));
+    params.insert(
+        "remaining_time".to_string(),
+        format!("<t:{}:R>", trigger_timestamp),
+    );
+
+    if !reminder_content.is_empty() {
+        params.insert("content".to_string(), reminder_content.to_string());
+    }
+
+    if !reminder_content.is_empty() {
+        let _ = MessageBuilder::system_message(&ctx, &config)
+            .translated_content(
+                "reminder.registered_with_content",
+                Some(&params),
+                None,
+                None,
+            )
+            .await
+            .to_channel(msg.channel_id)
+            .footer(format!("{}: {}", "ID", reminder_id))
+            .send()
+            .await;
+    } else {
+        let _ = MessageBuilder::system_message(&ctx, &config)
+            .translated_content(
+                "reminder.registered_without_content",
+                Some(&params),
+                None,
+                None,
+            )
+            .await
+            .to_channel(msg.channel_id)
+            .footer(format!("{}: {}", "ID", reminder_id))
+            .send()
+            .await;
+    }
+}
+
+pub async fn send_register_confirmation_from_command(
+    reminder_id: i64,
+    reminder_content: &str,
+    ctx: &Context,
+    command: &CommandInteraction,
+    config: &Config,
+    trigger_timestamp: i64,
+) {
+    let mut params = HashMap::new();
+    params.insert("time".to_string(), format!("<t:{}:F>", trigger_timestamp));
+    params.insert(
+        "remaining_time".to_string(),
+        format!("<t:{}:R>", trigger_timestamp),
+    );
+
+    if !reminder_content.is_empty() {
+        params.insert("content".to_string(), reminder_content.to_string());
+    }
+
+    if !reminder_content.is_empty() {
+        let response = MessageBuilder::system_message(&ctx, &config)
+            .translated_content(
+                "reminder.registered_with_content",
+                Some(&params),
+                None,
+                None,
+            )
+            .await
+            .to_channel(command.channel_id)
+            .footer(format!("{}: {}", "ID", reminder_id))
+            .build_interaction_message_followup()
+            .await;
+
+        let _ = command.create_followup(&ctx.http, response).await;
+    } else {
+        let response = MessageBuilder::system_message(&ctx, &config)
+            .translated_content(
+                "reminder.registered_without_content",
+                Some(&params),
+                None,
+                None,
+            )
+            .await
+            .to_channel(command.channel_id)
+            .footer(format!("{}: {}", "ID", reminder_id))
+            .build_interaction_message_followup()
+            .await;
+
+        let _ = command.create_followup(&ctx.http, response).await;
+    }
+}
+
+pub fn spawn_reminder(reminder: &Reminder, ctx: &Context, config: &Config, pool: &SqlitePool) {
+    let pool = pool.clone();
+    let config = config.clone();
+    let ctx = ctx.clone();
+    let reminder = reminder.clone();
+
+    tokio::spawn(async move {
+        let now = Local::now().timestamp();
+        let delay_duration = if reminder.trigger_time > now {
+            reminder.trigger_time - now
+        } else {
+            0
+        };
+        sleep(Duration::from_secs(delay_duration as u64)).await;
+
+        let mut params = HashMap::new();
+        params.insert(
+            "time".to_string(),
+            format!("<t:{}:F>", reminder.trigger_time),
+        );
+        params.insert(
+            "remaining_time".to_string(),
+            format!("<t:{}:R>", reminder.trigger_time),
+        );
+
+        params.insert("user".to_string(), reminder.user_id.to_string());
+        params.insert("content".to_string(), reminder.reminder_content.to_string());
+
+        let mut mentions = Vec::<UserId>::new();
+        mentions.push(UserId::new(reminder.user_id as u64));
+
+        if !reminder.reminder_content.is_empty() {
+            let _ = MessageBuilder::system_message(&ctx, &config)
+                .translated_content("reminder.show_with_content", Some(&params), None, None)
+                .await
+                .to_channel(ChannelId::new(reminder.channel_id as u64))
+                .color(hex_string_to_int(&config.reminders.embed_color) as u32)
+                .mention(mentions)
+                .send()
+                .await;
+        } else {
+            let _ = MessageBuilder::system_message(&ctx, &config)
+                .translated_content("reminder.show_without_content", Some(&params), None, None)
+                .await
+                .to_channel(ChannelId::new(reminder.channel_id as u64))
+                .color(hex_string_to_int(&config.reminders.embed_color) as u32)
+                .mention(mentions)
+                .send()
+                .await;
+        }
+
+        if let Err(e) = update_reminder_status(&reminder, true, &pool).await {
+            eprintln!("Failed to update reminder status: {}", e);
+        }
+    });
+}

Cargo.lock

@@ -0,0 +1,3 @@
+pub mod common;
+pub mod slash_command;
+pub mod text_command;