Join sounds per-server

src/cmds/settings.rs

@@ -1,4 +1,7 @@
+use poise::serenity_prelude::{GuildId, User};
+
 use crate::{
+    cmds::autocomplete_sound,
     models::{guild_data::CtxGuildData, join_sound::JoinSoundCtx, sound::SoundCtx},
     Context, Error,
 };
@@ -31,18 +34,44 @@ pub async fn change_volume(
     Ok(())
 }
 
-/// Manage greet sounds on this server
+/// Manage greet sounds
 #[poise::command(slash_command, rename = "greet", guild_only = true)]
 pub async fn greet_sound(_ctx: Context<'_>) -> Result<(), Error> {
     Ok(())
 }
 
-/// Set a join sound
+/// Manage greet sounds in this server
+#[poise::command(slash_command, rename = "server")]
+pub async fn guild_greet_sound(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Set a user's guild-specific join sound
 #[poise::command(slash_command, rename = "set")]
-pub async fn set_greet_sound(
+pub async fn set_guild_greet_sound(
     ctx: Context<'_>,
-    #[description = "Name or ID of sound to set as your join sound"] name: String,
+    #[description = "Name or ID of sound to set as join sound"]
+    #[autocomplete = "autocomplete_sound"]
+    name: String,
+    #[description = "User to set join sound for"] user: User,
 ) -> Result<(), Error> {
+    if user.id != ctx.author().id {
+        let guild = ctx.guild().unwrap();
+        let permissions = guild
+            .member_permissions(&ctx.discord(), ctx.author().id)
+            .await;
+
+        if permissions.map_or(true, |p| !p.manage_guild()) {
+            ctx.send(|b| {
+                b.ephemeral(true)
+                    .content("Only admins can change other user's greet sounds.")
+            })
+            .await?;
+
+            return Ok(());
+        }
+    }
+
     let sound_vec = ctx
         .data()
         .search_for_sound(&name, ctx.guild_id().unwrap(), ctx.author().id, true)
@@ -51,7 +80,7 @@ pub async fn set_greet_sound(
     match sound_vec.first() {
         Some(sound) => {
             ctx.data()
-                .update_join_sound(ctx.author().id, Some(sound.id))
+                .update_join_sound(user.id, ctx.guild_id(), Some(sound.id))
                 .await;
 
             ctx.say(format!(
@@ -69,18 +98,104 @@ pub async fn set_greet_sound(
     Ok(())
 }
 
-/// Set a join sound
+/// Unset your global join sound
 #[poise::command(slash_command, rename = "unset", guild_only = true)]
-pub async fn unset_greet_sound(ctx: Context<'_>) -> Result<(), Error> {
-    ctx.data().update_join_sound(ctx.author().id, None).await;
+pub async fn unset_guild_greet_sound(
+    ctx: Context<'_>,
+    #[description = "User to set join sound for"] user: User,
+) -> Result<(), Error> {
+    if user.id != ctx.author().id {
+        let guild = ctx.guild().unwrap();
+        let permissions = guild
+            .member_permissions(&ctx.discord(), ctx.author().id)
+            .await;
+
+        if permissions.map_or(true, |p| !p.manage_guild()) {
+            ctx.send(|b| {
+                b.ephemeral(true)
+                    .content("Only admins can change other user's greet sounds.")
+            })
+            .await?;
+
+            return Ok(());
+        }
+    }
+
+    ctx.data()
+        .update_join_sound(user.id, ctx.guild_id(), None)
+        .await;
 
     ctx.say("Greet sound has been unset").await?;
 
     Ok(())
 }
 
+/// Manage your own greet sound
+#[poise::command(slash_command, rename = "user")]
+pub async fn user_greet_sound(_ctx: Context<'_>) -> Result<(), Error> {
+    Ok(())
+}
+
+/// Set your global join sound
+#[poise::command(slash_command, rename = "set")]
+pub async fn set_user_greet_sound(
+    ctx: Context<'_>,
+    #[description = "Name or ID of sound to set as your join sound"]
+    #[autocomplete = "autocomplete_sound"]
+    name: String,
+) -> Result<(), Error> {
+    let sound_vec = ctx
+        .data()
+        .search_for_sound(&name, ctx.guild_id().unwrap(), ctx.author().id, true)
+        .await?;
+
+    match sound_vec.first() {
+        Some(sound) => {
+            ctx.data()
+                .update_join_sound(ctx.author().id, None::<GuildId>, Some(sound.id))
+                .await;
+
+            ctx.send(|b| {
+                b.ephemeral(true).content(format!(
+                    "Greet sound has been set to {} (ID {})",
+                    sound.name, sound.id
+                ))
+            })
+            .await?;
+        }
+
+        None => {
+            ctx.send(|b| {
+                b.ephemeral(true)
+                    .content("Could not find a sound by that name.")
+            })
+            .await?;
+        }
+    }
+
+    Ok(())
+}
+
+/// Unset your global join sound
+#[poise::command(slash_command, rename = "unset", guild_only = true)]
+pub async fn unset_user_greet_sound(ctx: Context<'_>) -> Result<(), Error> {
+    ctx.data()
+        .update_join_sound(ctx.author().id, None::<GuildId>, None)
+        .await;
+
+    ctx.send(|b| b.ephemeral(true).content("Greet sound has been unset"))
+        .await?;
+
+    Ok(())
+}
+
 /// Disable greet sounds on this server
-#[poise::command(slash_command, rename = "disable", guild_only = true)]
+#[poise::command(
+    slash_command,
+    rename = "disable",
+    guild_only = true,
+    required_permissions = "MANAGE_GUILD"
+)]
 pub async fn disable_greet_sound(ctx: Context<'_>) -> Result<(), Error> {
     let guild_data_opt = ctx.guild_data(ctx.guild_id().unwrap()).await;
 
@@ -97,7 +212,12 @@ pub async fn disable_greet_sound(ctx: Context<'_>) -> Result<(), Error> {
 }
 
 /// Enable greet sounds on this server
-#[poise::command(slash_command, rename = "enable", guild_only = true)]
+#[poise::command(
+    slash_command,
+    rename = "enable",
+    guild_only = true,
+    required_permissions = "MANAGE_GUILD"
+)]
 pub async fn enable_greet_sound(ctx: Context<'_>) -> Result<(), Error> {
     let guild_data_opt = ctx.guild_data(ctx.guild_id().unwrap()).await;