Merge pull request #1 from minneelyyyy/dev

add dox command
This commit is contained in:
mins
2024-10-10 14:37:37 -04:00
committed by GitHub

View File

@@ -1,11 +1,13 @@
use std::env; #![feature(async_closure)]
use poise::serenity_prelude as serenity; use std::env;
use poise::{serenity_prelude::{self as serenity, Colour}, CreateReply};
struct Data {} struct Data {}
type Error = Box<dyn std::error::Error + Send + Sync>; type Error = Box<dyn std::error::Error + Send + Sync>;
type Context<'a> = poise::Context<'a, Data, Error>; type Context<'a> = poise::Context<'a, Data, Error>;
/// Display the bot's latency to Discord's REST and Gateway APIs
#[poise::command(slash_command)] #[poise::command(slash_command)]
async fn ping(ctx: Context<'_>) -> Result<(), Error> { async fn ping(ctx: Context<'_>) -> Result<(), Error> {
use std::time::Instant; use std::time::Instant;
@@ -20,16 +22,83 @@ async fn ping(ctx: Context<'_>) -> Result<(), Error> {
Ok(()) Ok(())
} }
fn get_dox_output(ctx: &mut Context<'_>, user: &serenity::User, member: Option<&serenity::Member>, show_permissions: bool) -> String {
if user.bot {
return "This user is a bot.".into();
}
let mut output = String::new();
output.push_str(&format!("User ID: {}\n", user.id));
if let Some(locale) = &user.locale {
output.push_str(&format!("Locale: {locale}\n"));
}
if let Some(verified) = &user.verified {
output.push_str(&format!("Verified: {verified}\n"));
}
output.push_str(&format!("Account Created: {}\n", user.created_at()));
if let Some(Some(join_date)) = member.as_ref().map(|m| m.joined_at) {
output.push_str(&format!("Joined this Server at: {join_date}\n"));
}
if let Some(Some(premium_since)) = member.as_ref().map(|m| m.premium_since) {
output.push_str(&format!("Boosting this Server: Yes\nBoosting since: {premium_since}\n"));
}
if show_permissions {
if let Some(member) = member {
if let Ok(permissions) = member.permissions(ctx) {
output.push_str(&format!("Permissions: {}\n", permissions.get_permission_names().join(", ")))
}
}
}
output
}
/// Display information about a given user
#[poise::command(slash_command)]
async fn dox(
mut ctx: Context<'_>,
#[description = "The user to display information of"]
user: serenity::User,
#[rename = "permissions"]
#[description = "Rather or not to show the user's permissions"]
show_permissions: Option<bool>) -> Result<(), Error> {
let user = ctx.http().get_user(user.id).await?;
let member = if let Some(guild) = ctx.guild_id() {
guild.member(ctx.http(), user.id).await.ok()
} else {
None
};
let embed = serenity::CreateEmbed::default()
.title(format!("Information about {}", user.name))
.description(get_dox_output(&mut ctx, &user, member.as_ref(), show_permissions.unwrap_or(false)))
.colour(member.map(|m| m.colour(ctx.cache()))
.unwrap_or(None)
.unwrap_or(user.accent_colour.unwrap_or(Colour::from_rgb(255, 255, 255))))
.image(user.face());
ctx.send(CreateReply::default().embed(embed)).await?;
Ok(())
}
#[tokio::main] #[tokio::main]
async fn main() -> Result<(), Error> { async fn main() -> Result<(), Error> {
dotenv::dotenv().ok(); dotenv::dotenv().ok();
let token = env::var("DISCORD_BOT_TOKEN")?; let token = env::var("DISCORD_BOT_TOKEN")?;
let intents = serenity::GatewayIntents::non_privileged(); let intents = serenity::GatewayIntents::all();
let framework = poise::Framework::builder() let framework = poise::Framework::builder()
.options(poise::FrameworkOptions { .options(poise::FrameworkOptions {
commands: vec![ping()], commands: vec![ping(), dox()],
..Default::default() ..Default::default()
}) })
.setup(|ctx, _ready, framework| { .setup(|ctx, _ready, framework| {