You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.3 KiB
Rust
77 lines
2.3 KiB
Rust
|
|
use poise::serenity_prelude as serenity;
|
|
|
|
mod commands;
|
|
mod data_loader;
|
|
|
|
pub struct Data {} // User data, which is stored and accessible in all command invocations
|
|
type Error = Box<dyn std::error::Error + Send + Sync>;
|
|
type Context<'a> = poise::Context<'a, Data, Error>;
|
|
|
|
/// Displays your or another user's account creation date
|
|
#[poise::command(slash_command, prefix_command)]
|
|
async fn age(ctx: Context<'_>, #[description = "Selected user"] user: Option<serenity::User>) -> Result<(), Error>
|
|
{
|
|
let u = user.as_ref().unwrap_or_else(|| ctx.author());
|
|
let response = format!("{}'s account was created at {}", u.name, u.created_at());
|
|
ctx.say(response).await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// A command with two subcommands: `child1` and `child2`
|
|
///
|
|
/// Running this function directly, without any subcommand, is only supported in prefix commands.
|
|
/// Discord doesn't permit invoking the root command of a slash command if it has subcommands.
|
|
#[poise::command(prefix_command, slash_command, subcommands("child1", "child2"))]
|
|
pub async fn parent(ctx: Context<'_>) -> Result<(), Error>
|
|
{
|
|
ctx.say("Hello there!").await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// A subcommand of `parent`
|
|
#[poise::command(prefix_command, slash_command)]
|
|
pub async fn child1(ctx: Context<'_>) -> Result<(), Error>
|
|
{
|
|
ctx.say("You invoked the first child command!").await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// Another subcommand of `parent`
|
|
#[poise::command(prefix_command, slash_command)]
|
|
pub async fn child2(ctx: Context<'_>) -> Result<(), Error>
|
|
{
|
|
ctx.say("You invoked the second child command!").await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main()
|
|
{
|
|
let token = match data_loader::load_token("secrets/test.txt")
|
|
{
|
|
Ok(t) => t,
|
|
Err(why) => panic!("Could not load app token: {}", why),
|
|
};
|
|
|
|
println!("Connecting...");
|
|
|
|
let framework = poise::Framework::builder()
|
|
.options(poise::FrameworkOptions {
|
|
commands: vec![age(), parent()],
|
|
..Default::default()
|
|
})
|
|
.token(token)
|
|
.intents(serenity::GatewayIntents::non_privileged())
|
|
.setup(|ctx, _ready, framework|
|
|
{
|
|
Box::pin(async move
|
|
{
|
|
poise::builtins::register_globally(ctx, &framework.options().commands).await?;
|
|
Ok(Data {})
|
|
})
|
|
});
|
|
|
|
framework.run().await.unwrap();
|
|
}
|