initial rewrite in rust & moderation commands

Signed-off-by: seth <getchoo@tuta.io>
This commit is contained in:
seth 2023-12-03 04:11:57 -05:00
parent b17e357b75
commit 45403e9d9b
No known key found for this signature in database
GPG key ID: D31BD0D494BBEE86
53 changed files with 3297 additions and 2820 deletions

View file

@ -0,0 +1,80 @@
use crate::{consts::COLORS, Context};
use color_eyre::eyre::{eyre, Result};
use poise::serenity_prelude::{CreateEmbed, User};
fn create_moderation_embed(
title: String,
user: &User,
delete_messages_days: Option<u8>,
reason: String,
) -> impl FnOnce(&mut CreateEmbed) -> &mut CreateEmbed {
let fields = [
("User", format!("{} ({})", user.name, user.id), false),
("Reason", reason, false),
(
"Deleted messages",
format!("Last {} days", delete_messages_days.unwrap_or(0)),
false,
),
];
|e: &mut CreateEmbed| e.title(title).fields(fields).color(COLORS["red"])
}
// ban a user
#[poise::command(
slash_command,
prefix_command,
default_member_permissions = "BAN_MEMBERS"
)]
pub async fn ban_user(
ctx: Context<'_>,
user: User,
delete_messages_days: Option<u8>,
reason: Option<String>,
) -> Result<()> {
let days = delete_messages_days.unwrap_or(1);
let guild = ctx
.guild()
.ok_or_else(|| eyre!("Couldn't get guild from message; Unable to ban!"))?;
let reason = reason.unwrap_or("n/a".to_string());
if reason != "n/a" {
guild.ban_with_reason(ctx, &user, days, &reason).await?;
} else {
guild.ban(ctx, &user, days).await?;
}
let embed = create_moderation_embed("User banned!".to_string(), &user, Some(days), reason);
ctx.send(|m| m.embed(embed)).await?;
Ok(())
}
// kick a user
#[poise::command(
slash_command,
prefix_command,
default_member_permissions = "KICK_MEMBERS"
)]
pub async fn kick_user(ctx: Context<'_>, user: User, reason: Option<String>) -> Result<()> {
let guild = ctx
.guild()
.ok_or_else(|| eyre!("Couldn't get guild from message; Unable to ban!"))?;
let reason = reason.unwrap_or("n/a".to_string());
if reason != "n/a" {
guild.kick_with_reason(ctx, &user, &reason).await?;
} else {
guild.kick(ctx, &user).await?;
}
let embed = create_moderation_embed("User kicked!".to_string(), &user, None, reason);
ctx.send(|m| m.embed(embed)).await?;
Ok(())
}