feat: start of slash commands

This commit is contained in:
silver 2023-09-16 22:47:26 +01:00
parent 92ebe3b931
commit f405cbbb93
3 changed files with 94 additions and 1 deletions

View file

@ -0,0 +1,58 @@
use serenity::builder::CreateApplicationCommand;
use serenity::client::Context;
use serenity::model::application::interaction::Interaction;
use serenity::model::application::interaction::application_command::ApplicationCommandInteraction;
use serenity::model::id::GuildId;
use serenity::model::prelude::command::CommandOptionType;
use serenity::model::prelude::interaction::application_command::{
CommandDataOption,
CommandDataOptionValue,
};
use serenity::model::user::User;
use sqlx::{Pool, Sqlite};
use skynet_discord_bot::DataBase;
pub async fn run(options: &ApplicationCommandInteraction, ctx: &Context) -> String {
let db_lock = {
let data_read = ctx.data.read().await;
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
};
let db = db_lock.read().await;
let option = options.data.options.get(0).expect("Expected email option").resolved.as_ref().expect("Expected email object");
if let CommandDataOptionValue::String(email) = option {
format!("Email is {}, user is {} {:?}", email, options.user.name, options.guild_id)
} else {
"Please provide a valid user".to_string()
}
}
pub fn register(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand {
command.name("link").description("Set Wolves Email").create_option(|option| {
option
.name("email")
.description("UL Wolves Email")
.kind(CommandOptionType::String)
.required(true)
})
}
async fn get_server_member(db: &Pool<Sqlite>, server: &GuildId) -> Vec<Accounts> {
sqlx::query_as::<_, Accounts>(
r#"
SELECT *
FROM wolves
WHERE server = ? AND discord IS NOT NULL AND expiry > ?
"#,
)
.bind(*server.as_u64() as i64)
.bind(get_now_iso(true))
.fetch_all(db)
.await
.unwrap_or_default()
}

1
src/commands/mod.rs Normal file
View file

@ -0,0 +1 @@
pub mod link_email;

View file

@ -1,13 +1,17 @@
mod commands;
use serenity::{ use serenity::{
async_trait, async_trait,
client::{Context, EventHandler}, client::{Context, EventHandler},
model::{ model::{
gateway::{GatewayIntents, Ready}, gateway::{GatewayIntents, Ready},
guild, guild,
application::{interaction::{Interaction, InteractionResponseType}, command::Command},
}, },
Client, Client,
}; };
use std::sync::Arc; use std::sync::Arc;
use serenity::model::prelude::interaction;
use skynet_discord_bot::{db_init, get_config, get_server_config, get_server_member, Config, DataBase}; use skynet_discord_bot::{db_init, get_config, get_server_config, get_server_member, Config, DataBase};
use tokio::sync::RwLock; use tokio::sync::RwLock;
@ -49,8 +53,38 @@ impl EventHandler for Handler {
} }
} }
async fn ready(&self, _ctx: Context, ready: Ready) { async fn ready(&self, ctx: Context, ready: Ready) {
println!("[Main] {} is connected!", ready.user.name); println!("[Main] {} is connected!", ready.user.name);
Command::set_global_application_commands(&ctx.http, |commands| {
commands
.create_application_command(|command| commands::link_email::register(command))
})
.await.ok();
}
async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
if let Interaction::ApplicationCommand(command) = interaction {
//println!("Received command interaction: {:#?}", command);
let content = match command.data.name.as_str() {
"link" => commands::link_email::run(&command, &ctx).await,
_ => "not implemented :(".to_string(),
};
if let Err(why) = command
.create_interaction_response(&ctx.http, |response| {
response
.kind(InteractionResponseType::ChannelMessageWithSource)
.interaction_response_data(|message| message.content(content).ephemeral(true))
})
.await
{
println!("Cannot respond to slash command: {}", why);
}
}
} }
} }