feat(commands): add base for ping command

Author: Akinator31

rustmail/src/commands/mod.rs

@@ -22,6 +22,7 @@ pub mod id;
 pub mod logs;
 pub mod move_thread;
 pub mod new_thread;
+pub mod ping;
 pub mod recover;
 pub mod release;
 pub mod remove_reminder;
@@ -42,6 +43,7 @@ pub use id::*;
 pub use logs::*;
 pub use move_thread::*;
 pub use new_thread::*;
+pub use ping::*;
 pub use recover::*;
 pub use release::*;
 pub use remove_reminder::*;

rustmail/src/commands/ping/mod.rs

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

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

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

rustmail/src/commands/ping/slash_command/ping.rs

@@ -0,0 +1,77 @@
+use crate::bot::ShardManagerKey;
+use crate::commands::{BoxFuture, CommunityRegistrable, RegistrableCommand};
+use crate::config::Config;
+use crate::errors::ModmailResult;
+use crate::handlers::InteractionHandler;
+use crate::i18n::get_translated_message;
+use serenity::FutureExt;
+use serenity::all::{CommandInteraction, Context, CreateCommand, ResolvedOption};
+use std::sync::Arc;
+
+pub struct PingCommand;
+
+#[async_trait::async_trait]
+impl RegistrableCommand for PingCommand {
+    fn as_community(&self) -> Option<&dyn CommunityRegistrable> {
+        None
+    }
+
+    fn name(&self) -> &'static str {
+        "ping"
+    }
+
+    fn doc<'a>(&self, config: &'a Config) -> BoxFuture<'a, String> {
+        async move { get_translated_message(config, "help.ping", None, None, None, None).await }
+            .boxed()
+    }
+
+    fn register(&self, config: &Config) -> BoxFuture<'_, Vec<CreateCommand>> {
+        let config = config.clone();
+
+        Box::pin(async move {
+            let cmd_desc = get_translated_message(
+                &config,
+                "slash_command.ping_command_desc",
+                None,
+                None,
+                None,
+                None,
+            )
+            .await;
+
+            vec![CreateCommand::new(self.name()).description(cmd_desc)]
+        })
+    }
+
+    fn run(
+        &self,
+        ctx: &Context,
+        command: &CommandInteraction,
+        _options: &[ResolvedOption<'_>],
+        config: &Config,
+        _handler: Arc<InteractionHandler>,
+    ) -> BoxFuture<'_, ModmailResult<()>> {
+        let ctx = ctx.clone();
+        let command = command.clone();
+        let config = config.clone();
+
+        Box::pin(async move {
+            let shard_manager = ctx
+                .data
+                .read()
+                .await
+                .get::<ShardManagerKey>()
+                .unwrap()
+                .clone();
+
+            let latency = {
+                let runners = shard_manager.runners.lock().await;
+                runners.get(&ctx.shard_id).and_then(|runner| runner.latency)
+            };
+
+            println!("Latency for shard {}: {:?}", ctx.shard_id, latency);
+
+            Ok(())
+        })
+    }
+}

rustmail/src/commands/ping/text_command/mod.rs

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

rustmail/src/commands/ping/text_command/ping.rs

@@ -0,0 +1,30 @@
+use crate::bot::ShardManagerKey;
+use crate::prelude::config::*;
+use crate::prelude::errors::*;
+use crate::prelude::handlers::*;
+use serenity::all::{Context, Message};
+use std::sync::Arc;
+
+pub async fn ping(
+    ctx: Context,
+    msg: Message,
+    config: &Config,
+    _handler: Arc<GuildMessagesHandler>,
+) -> ModmailResult<()> {
+    let shard_manager = ctx
+        .data
+        .read()
+        .await
+        .get::<ShardManagerKey>()
+        .unwrap()
+        .clone();
+
+    let latency = {
+        let runners = shard_manager.runners.lock().await;
+        runners.get(&ctx.shard_id).and_then(|runner| runner.latency)
+    };
+
+    println!("Latency for shard {}: {:?}", ctx.shard_id, latency);
+
+    Ok(())
+}

rustmail/src/handlers/guild_messages_handler.rs

@@ -77,6 +77,7 @@ impl GuildMessagesHandler {
         wrap_command!(lock, "logs", logs);
         wrap_command!(lock, "take", take);
         wrap_command!(lock, "release", release);
+        wrap_command!(lock, "ping", ping);
 
         drop(lock);
         h

rustmail/src/i18n/language/en.rs

@@ -883,6 +883,10 @@ pub fn load_english_messages(dict: &mut ErrorDictionary) {
         "help.release".to_string(),
         DictionaryMessage::new("Releases ownership of a ticket previously taken with the `!take` command. To release a ticket, use `!release` in the ticket."),
     );
+    dict.messages.insert(
+        "help.ping".to_string(),
+        DictionaryMessage::new("Show the actual latency of the Discord API."),
+    );
     dict.messages.insert(
         "add_reminder.helper".to_string(),
         DictionaryMessage::new(
@@ -928,4 +932,8 @@ pub fn load_english_messages(dict: &mut ErrorDictionary) {
         "slash_command.help_command_argument_desc".to_string(),
         DictionaryMessage::new("The command to get help with"),
     );
+    dict.messages.insert(
+        "slash_command.ping_command_desc".to_string(),
+        DictionaryMessage::new("Check the discord API latency"),
+    );
 }

rustmail/src/i18n/language/fr.rs

@@ -904,6 +904,10 @@ pub fn load_french_messages(dict: &mut ErrorDictionary) {
         "help.release".to_string(),
         DictionaryMessage::new("Permet de ne plus prendre en charge un ticket pris en charge via la commande `take`. Pour libérer un ticket, faites `!release` dans le ticket."),
     );
+    dict.messages.insert(
+        "help.ping".to_string(),
+        DictionaryMessage::new("Permet d'afficher la latence actuelle de l'API Discord."),
+    );
     dict.messages.insert(
         "add_reminder.helper".to_string(),
         DictionaryMessage::new("Format incorrect. Utilisation : `{prefix}remind ou {prefix}rem <HH:MM> [contenu du rappel]`"),
@@ -940,4 +944,8 @@ pub fn load_french_messages(dict: &mut ErrorDictionary) {
         "slash_command.help_command_argument_desc".to_string(),
         DictionaryMessage::new("Le nom de la commande pour laquelle vous souhaitez de l'aide"),
     );
+    dict.messages.insert(
+        "slash_command.ping_command_desc".to_string(),
+        DictionaryMessage::new("Afficher la latence actuelle de l'API Discord"),
+    );
 }