feat: reintroduce PK support

Signed-off-by: seth <getchoo@tuta.io>

src/storage.rs

@@ -0,0 +1,40 @@
+use color_eyre::eyre::Result;
+use log::*;
+use poise::serenity_prelude::UserId;
+use redis::{AsyncCommands as _, Client};
+
+pub const USER_KEY: &str = "users-v1";
+
+#[derive(Clone, Debug)]
+pub struct Storage {
+    client: Client,
+}
+
+impl Storage {
+    pub fn new(redis_url: &str) -> Result<Self> {
+        let client = Client::open(redis_url)?;
+
+        Ok(Self { client })
+    }
+
+    pub async fn store_user_plurality(&self, sender: UserId) -> Result<()> {
+        let mut con = self.client.get_async_connection().await?;
+
+        info!("Marking {sender} as a PluralKit user");
+        let key = format!("{USER_KEY}:{sender}:pk");
+
+        // Just store some value. We only care about the presence of this key
+        con.set(&key, 0).await?;
+        con.expire(key, 7 * 24 * 60 * 60).await?;
+
+        Ok(())
+    }
+
+    pub async fn is_user_plural(&self, user_id: UserId) -> Result<bool> {
+        let key = format!("{USER_KEY}:{user_id}:pk");
+        let mut con = self.client.get_async_connection().await?;
+
+        let exists: bool = con.exists(key).await?;
+        Ok(exists)
+    }
+}