feat(discord)!: add stats command

Author: Cyberdolfi

src/database/server.rs

@@ -255,4 +255,21 @@ pub async fn search_servers(filters: SearchFilters, limit: i64) -> Result<Option
     }).collect();
 
     Ok(Some(results))
+}
+
+pub async fn get_database_counts() -> Result<(i64, i64, i64), sqlx::Error> {
+    let pool = pool::get_pool();
+
+    let counts: (i64, i64, i64) = sqlx::query_as(
+        r#"
+        SELECT
+            (SELECT COUNT(*) FROM servers) as server_count,
+            (SELECT COUNT(*) FROM server_history) as history_count,
+            (SELECT COUNT(*) FROM player_history) as player_count
+        "#
+    )
+        .fetch_one(pool)
+        .await?;
+
+    Ok(counts)
 }

src/discord/mod.rs

@@ -54,7 +54,8 @@ pub async fn start_bot() {
                 commands: vec![
                     server_searcher::search_server(),
                     player_searcher::search_player(),
-                    cleanup_command::cleanup()
+                    cleanup_command::cleanup(),
+                    stats_command::stats()
                 ],
                 prefix_options: PrefixFrameworkOptions {
                     prefix: Some("!".into()),

src/discord/stats_command.rs

@@ -0,0 +1,45 @@
+use chrono::Utc;
+use poise::{command, CreateReply};
+use crate::database::server::get_database_counts;
+use crate::discord::{create_base_embed, create_error_embed, create_loading_embed, Context, Error};
+use crate::logger;
+
+/// Shows stats about the bot and database
+#[command(slash_command)]
+pub async fn stats(
+    ctx: Context<'_>
+) -> Result<(), Error> {
+    let start_time = Utc::now();
+
+    let reply = ctx.send(
+        CreateReply::default()
+            .embed(create_loading_embed("fetching stats"))
+    ).await?;
+
+    match get_database_counts().await {
+        Ok((server_count, history_count, player_count)) => {
+            let response = create_base_embed(Some(start_time))
+                .title("📶 Stats")
+                .description("Some stats about ServerRawler")
+                .field("💻 Indexed servers", format!("--> `{}`", server_count), true)
+                .field("📅 Server data entries", format!("--> `{}`", history_count), false)
+                .field("👥 Player join entries", format!("--> `{}`", player_count), false);
+
+            reply.edit(
+                ctx,
+                CreateReply::default()
+                    .embed(response)
+            ).await?;
+        }
+        Err(e) => {
+            reply.edit(
+                ctx,
+                CreateReply::default()
+                    .embed(create_error_embed("Database error", Some(start_time)))
+            ).await?;
+            logger::error(format!("Database stats error: {}", e)).prefix("Discord").send().await;
+        }
+    }
+
+    Ok(())
+}