add dox command
This commit is contained in:
77
src/main.rs
77
src/main.rs
@@ -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| {
|
||||||
|
|||||||
Reference in New Issue
Block a user