mirror of
https://github.com/natsukagami/youmubot.git
synced 2025-04-21 09:48:54 +00:00
100 lines
2.8 KiB
Rust
100 lines
2.8 KiB
Rust
use serenity::{
|
|
framework::standard::{
|
|
macros::{command, group},
|
|
Args, CommandResult,
|
|
},
|
|
model::{
|
|
channel::{Channel, Message},
|
|
id::UserId,
|
|
},
|
|
};
|
|
use soft_ban::{SOFT_BAN_COMMAND, SOFT_BAN_INIT_COMMAND};
|
|
use std::{thread::sleep, time::Duration};
|
|
use youmubot_prelude::*;
|
|
|
|
mod soft_ban;
|
|
pub use soft_ban::watch_soft_bans;
|
|
|
|
#[group]
|
|
#[description = "Administrative commands for the server."]
|
|
#[commands(clean, ban, kick, soft_ban, soft_ban_init)]
|
|
struct Admin;
|
|
|
|
#[command]
|
|
#[aliases("cleanall")]
|
|
#[required_permissions(MANAGE_MESSAGES)]
|
|
#[description = "Clean at most X latest messages from the current channel (only clean Youmu's messages in DMs). Defaults to 10."]
|
|
#[usage = "clean 50"]
|
|
#[min_args(0)]
|
|
#[max_args(1)]
|
|
fn clean(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
|
let limit = args.single().unwrap_or(10);
|
|
let messages = msg
|
|
.channel_id
|
|
.messages(&ctx.http, |b| b.before(msg.id).limit(limit))?;
|
|
let channel = msg.channel_id.to_channel(&ctx)?;
|
|
match &channel {
|
|
Channel::Private(_) | Channel::Group(_) => {
|
|
let self_id = ctx.http.get_current_application_info()?.id;
|
|
messages
|
|
.into_iter()
|
|
.filter(|v| v.author.id == self_id)
|
|
.try_for_each(|m| m.delete(&ctx))?;
|
|
}
|
|
_ => {
|
|
msg.channel_id
|
|
.delete_messages(&ctx.http, messages.into_iter())?;
|
|
}
|
|
};
|
|
msg.react(&ctx, "🌋")?;
|
|
if let Channel::Guild(_) = &channel {
|
|
sleep(Duration::from_secs(2));
|
|
msg.delete(&ctx)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[command]
|
|
#[required_permissions(ADMINISTRATOR)]
|
|
#[description = "Ban an user with a certain reason."]
|
|
#[usage = "@user#1234/spam"]
|
|
#[min_args(1)]
|
|
#[max_args(2)]
|
|
#[only_in("guilds")]
|
|
fn ban(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
|
let user = args.single::<UserId>()?.to_user(&ctx)?;
|
|
let reason = args
|
|
.remains()
|
|
.map(|v| format!("`{}`", v))
|
|
.unwrap_or("no provided reason".to_owned());
|
|
|
|
msg.reply(
|
|
&ctx,
|
|
format!("🔨 Banning user {} for reason `{}`.", user.tag(), reason),
|
|
)?;
|
|
|
|
msg.guild_id
|
|
.ok_or("Can't get guild from message?")? // we had a contract
|
|
.ban(&ctx.http, user, &reason)?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[command]
|
|
#[required_permissions(ADMINISTRATOR)]
|
|
#[description = "Kick an user."]
|
|
#[usage = "@user#1234"]
|
|
#[num_args(1)]
|
|
#[only_in("guilds")]
|
|
fn kick(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
|
let user = args.single::<UserId>()?.to_user(&ctx)?;
|
|
|
|
msg.reply(&ctx, format!("🔫 Kicking user {}.", user.tag()))?;
|
|
|
|
msg.guild_id
|
|
.ok_or("Can't get guild from message?")? // we had a contract
|
|
.kick(&ctx.http, user)?;
|
|
|
|
Ok(())
|
|
}
|