nki/youmubot
1
0
Fork 0
mirror of https://github.com/natsukagami/youmubot.git synced 2025-04-19 08:48:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update youmubot

Browse source
This commit is contained in:
Natsu Kagami 2024-02-18 01:04:00 +01:00 • committed by Natsu Kagami
parent 0460e800e5
commit 4bdab3cefb
1 changed files with 30 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
youmubot/src/main.rs
View file

@ -1,6 +1,8 @@
use dotenv::var; use dotenv::var;
use serenity::{ use serenity::{
framework::standard::{macros::hook, CommandResult, DispatchError, StandardFramework}, framework::standard::{
macros::hook, BucketBuilder, CommandResult, Configuration, DispatchError, StandardFramework,
},
model::{ model::{
channel::{Channel, Message}, channel::{Channel, Message},
gateway, gateway,
@ -94,7 +96,7 @@ async fn main() {
let mut client = { let mut client = {
// Attempt to connect and set up a framework // Attempt to connect and set up a framework
let intents = GatewayIntents::GUILDS let intents = GatewayIntents::GUILDS
| GatewayIntents::GUILD_BANS | GatewayIntents::GUILD_MODERATION
| GatewayIntents::MESSAGE_CONTENT | GatewayIntents::MESSAGE_CONTENT
| GatewayIntents::GUILD_MESSAGES | GatewayIntents::GUILD_MESSAGES
| GatewayIntents::GUILD_MESSAGE_REACTIONS | GatewayIntents::GUILD_MESSAGE_REACTIONS
@ -167,40 +169,48 @@ async fn setup_framework(token: &str) -> StandardFramework {
.get_current_application_info() .get_current_application_info()
.await .await
.expect("Should be able to get app info") .expect("Should be able to get app info")
.owner; .owner
.unwrap();
let fw = StandardFramework::new() let fw = StandardFramework::new()
.configure(|c| {
c.with_whitespace(false)
.prefixes(
var("PREFIX")
.map(|v| v.split(',').map(|v| v.trim().to_owned()).collect())
.unwrap_or_else(|_| vec!["y!".to_owned(), "y2!".to_owned()]),
)
.delimiters(vec![" / ", "/ ", " /", "/"])
.owners([owner.id].iter().cloned().collect())
})
.help(&youmubot_core::HELP) .help(&youmubot_core::HELP)
.before(before_hook) .before(before_hook)
.after(after_hook) .after(after_hook)
.on_dispatch_error(on_dispatch_error) .on_dispatch_error(on_dispatch_error)
.bucket("voting", |c| { .bucket("voting", {
c.check(|ctx, msg| Box::pin(is_not_channel_mod(ctx, msg))) BucketBuilder::new_channel()
.check(|ctx, msg| Box::pin(is_not_channel_mod(ctx, msg)))
.delay(120 /* 2 minutes */) .delay(120 /* 2 minutes */)
.time_span(120) .time_span(120)
.limit(1) .limit(1)
}) })
.await .await
.bucket("images", |c| c.time_span(60).limit(2)) .bucket(
"images",
BucketBuilder::new_channel().time_span(60).limit(2),
)
.await .await
.bucket("community", |c| { .bucket(
c.check(|ctx, msg| Box::pin(is_not_channel_mod(ctx, msg))) "community",
BucketBuilder::new_guild()
.check(|ctx, msg| Box::pin(is_not_channel_mod(ctx, msg)))
.delay(30) .delay(30)
.time_span(30) .time_span(30)
.limit(1) .limit(1),
}) )
.await .await
.group(&prelude_commands::PRELUDE_GROUP); .group(&prelude_commands::PRELUDE_GROUP);
fw.configure(
Configuration::new()
.with_whitespace(false)
.prefixes(
var("PREFIX")
.map(|v| v.split(',').map(|v| v.trim().to_owned()).collect())
.unwrap_or_else(|_| vec!["y!".to_owned(), "y2!".to_owned()]),
)
.delimiters(vec![" / ", "/ ", " /", "/"])
.owners([owner.id].iter().cloned().collect()),
);
// groups here // groups here
#[cfg(feature = "core")] #[cfg(feature = "core")]
let fw = fw let fw = fw