feat: start of slash commands
This commit is contained in:
parent
92ebe3b931
commit
f405cbbb93
3 changed files with 94 additions and 1 deletions
58
src/commands/link_email.rs
Normal file
58
src/commands/link_email.rs
Normal 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
1
src/commands/mod.rs
Normal file
|
@ -0,0 +1 @@
|
||||||
|
pub mod link_email;
|
36
src/main.rs
36
src/main.rs
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue