feat: add presence persistence

Author: ryanccn

Cargo.lock

@@ -18,7 +18,8 @@ nanoid = "0.4.0"
 num = "0.4.1"
 once_cell = "1.18.0"
 owo-colors = { version = "3.5.0", features = ["supports-colors"] }
-poise = { git = "https://github.com/serenity-rs/poise.git", branch = "serenity-next", version = "0.5.5" }
+poise = { git = "https://github.com/serenity-rs/poise.git", branch = "serenity-next" }
+redis = { version = "0.23.3", features = ["tokio-comp", "tokio-rustls-comp"] }
 regex = "1.10.2"
 reqwest = { version = "0.11.22", default-features = false, features = ["rustls-tls", "json", "brotli"] }
 serde = { version = "1.0.189", features = ["derive"] }

Cargo.toml

@@ -24,3 +24,5 @@ pub fn vec() -> Vec<
         translate::translate(),
     ]
 }
+
+pub use presence::restore_presence;

src/commands/mod.rs

@@ -1,9 +1,12 @@
-use anyhow::Result;
+use owo_colors::OwoColorize;
 use poise::{serenity_prelude as serenity, CreateReply};
+use redis::AsyncCommands;
 
 use crate::Context;
+use anyhow::Result;
 
-#[derive(poise::ChoiceParameter)]
+#[derive(poise::ChoiceParameter, serde::Serialize, serde::Deserialize, Clone, Copy)]
+#[serde(rename_all = "lowercase")]
 pub enum PresenceChoice {
     Custom,
     Playing,
@@ -46,6 +49,13 @@ impl std::fmt::Display for PresenceChoice {
     }
 }
 
+#[derive(serde::Serialize, serde::Deserialize)]
+struct PresencePersistence {
+    content: String,
+    #[serde(rename = "type")]
+    type_: PresenceChoice,
+}
+
 /// Modify the Discord presence shown by the bot
 #[poise::command(
     slash_command,
@@ -73,11 +83,36 @@ pub async fn presence(
             serenity::CreateEmbed::new()
                 .title("Presence set!")
                 .field("Type", type_.to_string(), false)
-                .field("Content", content, false)
+                .field("Content", &content, false)
                 .color(0x4ade80),
         ),
     )
     .await?;
 
+    if let Some(redis) = &ctx.data().redis {
+        let mut conn = redis.get_async_connection().await?;
+        conn.set(
+            "presence-v1",
+            serde_json::to_string(&PresencePersistence { content, type_ })?,
+        )
+        .await?;
+    }
+
+    Ok(())
+}
+
+pub async fn restore_presence(ctx: &serenity::Context, redis_client: &redis::Client) -> Result<()> {
+    let mut conn = redis_client.get_async_connection().await?;
+    let data: Option<String> = conn.get("presence-v1").await?;
+
+    if let Some(data) = data {
+        let data: PresencePersistence = serde_json::from_str(&data)?;
+        ctx.set_presence(
+            Some(data.type_.make_activity(&data.content)),
+            serenity::OnlineStatus::Online,
+        );
+        println!("{} presence from Redis", "Restored".cyan());
+    }
+
     Ok(())
 }

src/commands/presence.rs

@@ -8,7 +8,9 @@ use poise::{
 
 use crate::utils::Pluralize;
 
-pub struct Data {}
+pub struct Data {
+    pub redis: Option<redis::Client>,
+}
 pub type Context<'a> = poise::Context<'a, Data, Error>;
 
 mod commands;
@@ -74,7 +76,19 @@ async fn main() -> Result<()> {
                         "command".pluralize(commands.len())
                     );
 
-                    Ok(Data {})
+                    if let Ok(redis_url) = std::env::var("REDIS_URL") {
+                        let client = redis::Client::open(redis_url)?;
+
+                        if let Err(err) = commands::restore_presence(&ctx, &client).await {
+                            eprintln!("{}", err);
+                        };
+
+                        return Ok(Data {
+                            redis: Some(client),
+                        });
+                    }
+
+                    return Ok(Data { redis: None });
                 })
             },
         ))