109 lines
3.0 KiB
Rust
109 lines
3.0 KiB
Rust
use poise::{
|
|
serenity_prelude as serenity,
|
|
serenity_prelude::{
|
|
builder::CreateApplicationCommands,
|
|
http::CacheHttp,
|
|
interaction::MessageFlags,
|
|
model::id::{GuildId, UserId},
|
|
},
|
|
};
|
|
|
|
use crate::{
|
|
consts::{CNC_GUILD, SUBSCRIPTION_ROLES},
|
|
Data, Error,
|
|
};
|
|
|
|
pub async fn register_application_commands(
|
|
ctx: &serenity::Context,
|
|
framework: &poise::Framework<Data, Error>,
|
|
guild_id: Option<GuildId>,
|
|
) -> Result<(), serenity::Error> {
|
|
let mut commands_builder = CreateApplicationCommands::default();
|
|
let commands = &framework.options().commands;
|
|
for command in commands {
|
|
if let Some(slash_command) = command.create_as_slash_command() {
|
|
commands_builder.add_application_command(slash_command);
|
|
}
|
|
if let Some(context_menu_command) = command.create_as_context_menu_command() {
|
|
commands_builder.add_application_command(context_menu_command);
|
|
}
|
|
}
|
|
let commands_builder = poise::serenity_prelude::json::Value::Array(commands_builder.0);
|
|
|
|
if let Some(guild_id) = guild_id {
|
|
ctx.http.create_guild_application_commands(guild_id.0, &commands_builder).await?;
|
|
} else {
|
|
ctx.http.create_global_application_commands(&commands_builder).await?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn check_subscription(cache_http: impl CacheHttp, user_id: impl Into<UserId>) -> bool {
|
|
if let Some(subscription_guild) = *CNC_GUILD {
|
|
let guild_member = GuildId(subscription_guild).member(cache_http, user_id).await;
|
|
|
|
if let Ok(member) = guild_member {
|
|
for role in member.roles {
|
|
if SUBSCRIPTION_ROLES.contains(role.as_u64()) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
false
|
|
} else {
|
|
true
|
|
}
|
|
}
|
|
|
|
pub async fn check_guild_subscription(
|
|
cache_http: impl CacheHttp,
|
|
guild_id: impl Into<GuildId>,
|
|
) -> bool {
|
|
if let Some(guild) = cache_http.cache().unwrap().guild(guild_id) {
|
|
let owner = guild.owner_id;
|
|
|
|
check_subscription(&cache_http, owner).await
|
|
} else {
|
|
false
|
|
}
|
|
}
|
|
|
|
/// Sends the message, specified via [`crate::CreateReply`], to the interaction initial response
|
|
/// endpoint
|
|
pub fn send_as_initial_response(
|
|
data: poise::CreateReply<'_>,
|
|
f: &mut serenity::CreateInteractionResponseData,
|
|
) {
|
|
let poise::CreateReply {
|
|
content,
|
|
embeds,
|
|
attachments: _, // serenity doesn't support attachments in initial response yet
|
|
components,
|
|
ephemeral,
|
|
allowed_mentions,
|
|
reply: _,
|
|
} = data;
|
|
|
|
if let Some(content) = content {
|
|
f.content(content);
|
|
}
|
|
f.set_embeds(embeds);
|
|
if let Some(allowed_mentions) = allowed_mentions {
|
|
f.allowed_mentions(|f| {
|
|
*f = allowed_mentions.clone();
|
|
f
|
|
});
|
|
}
|
|
if let Some(components) = components {
|
|
f.components(|f| {
|
|
f.0 = components.0;
|
|
f
|
|
});
|
|
}
|
|
if ephemeral {
|
|
f.flags(MessageFlags::EPHEMERAL);
|
|
}
|
|
}
|