mirror of
https://github.com/Noratrieb/widetom.git
synced 2026-01-14 08:55:02 +01:00
shutdown
This commit is contained in:
parent
85f9572c5b
commit
784b228e47
1 changed files with 16 additions and 8 deletions
24
src/main.rs
24
src/main.rs
|
|
@ -14,7 +14,7 @@ use serenity::model::id::UserId;
|
|||
use serenity::utils::{content_safe, ContentSafeOptions};
|
||||
use uwuifier::uwuify_str_sse;
|
||||
|
||||
use widertom::normal_message;
|
||||
use widertom::{normal_message, reply};
|
||||
|
||||
#[group]
|
||||
#[commands(say)]
|
||||
|
|
@ -26,6 +26,11 @@ struct General;
|
|||
#[description = "meme commands"]
|
||||
struct Meme;
|
||||
|
||||
#[group]
|
||||
#[commands(shutdown)]
|
||||
#[owners_only]
|
||||
#[description = "bot admin commands"]
|
||||
struct Admin;
|
||||
|
||||
struct Handler;
|
||||
|
||||
|
|
@ -39,7 +44,6 @@ impl EventHandler for Handler {
|
|||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
|
||||
let token = fs::read_to_string("bot_token")
|
||||
.expect("Expected bot token in file 'bot_token'");
|
||||
|
||||
|
|
@ -48,11 +52,8 @@ async fn main() {
|
|||
let (owners, bot_id) = match http.get_current_application_info().await {
|
||||
Ok(info) => {
|
||||
let mut owners = HashSet::new();
|
||||
if let Some(team) = info.team {
|
||||
owners.insert(team.owner_user_id);
|
||||
} else {
|
||||
owners.insert(info.owner.id);
|
||||
}
|
||||
owners.insert(UserId(414755070161453076)); //nils
|
||||
owners.insert(UserId(265849018662387712)); //yuki
|
||||
match http.get_current_user().await {
|
||||
Ok(bot_id) => (owners, bot_id.id),
|
||||
Err(why) => panic!("Could not access the bot id: {:?}", why),
|
||||
|
|
@ -72,7 +73,8 @@ async fn main() {
|
|||
.normal_message(normal_message)
|
||||
.help(&MY_HELP)
|
||||
.group(&GENERAL_GROUP)
|
||||
.group(&MEME_GROUP);
|
||||
.group(&MEME_GROUP)
|
||||
.group(&ADMIN_GROUP);
|
||||
|
||||
let mut client = Client::builder(&token)
|
||||
.event_handler(Handler)
|
||||
|
|
@ -109,6 +111,12 @@ async fn uwuify(ctx: &Context, msg: &Message, args: Args) -> CommandResult {
|
|||
Ok(())
|
||||
}
|
||||
|
||||
#[command]
|
||||
async fn shutdown(ctx: &Context, msg: &Message, _: Args) -> CommandResult {
|
||||
reply("<:tom:811324632082415626> bye <:tom:811324632082415626>", &msg, &ctx).await;
|
||||
std::process::exit(0);
|
||||
}
|
||||
|
||||
|
||||
#[help]
|
||||
#[individual_command_tip =
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue