fix(moderation): don't fail when taken_roles is empty

Axelen123 · Jan 4, 2023 · bf0c34e · bf0c34e
1 parent 503e095
commit bf0c34e
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/src/utils/moderation.rs b/src/utils/moderation.rs
@@ -93,18 +93,21 @@ pub fn queue_unmute_member(
             .await?
             .ok_or("User was not muted.")?;
 
-        let taken_roles = db_result
-            .taken_roles
-            .unwrap()
-            .into_iter()
-            .map(|r| RoleId::from(r.parse::<u64>().unwrap()))
-            .collect::<Vec<_>>();
-
-        let http = &ctx.http;
-
         // Update roles if they didn't leave the guild.
         if let Some(mut member) = ctx.cache.member(guild_id, user_id) {
-            member.add_roles(http, &taken_roles).await?;
+            let http = &ctx.http;
+
+            if let Some(taken_roles) = db_result.taken_roles {
+                member
+                    .add_roles(
+                        http,
+                        &taken_roles
+                            .into_iter()
+                            .map(|r| RoleId::from(r.parse::<u64>().unwrap()))
+                            .collect::<Vec<_>>(),
+                    )
+                    .await?;
+            }
             member.remove_role(http, mute_role_id).await?;
         }
         Ok(())