added /eight_ball command
This commit is contained in:
+21
@@ -1,6 +1,7 @@
|
||||
use crate::{send_embed, send_msg, Context, EmbedOptions, Error};
|
||||
|
||||
use poise::serenity_prelude::{OnlineStatus, Timestamp};
|
||||
use rand::{seq::IteratorRandom, Rng};
|
||||
|
||||
|
||||
#[poise::command(slash_command, prefix_command)]
|
||||
@@ -65,5 +66,25 @@ pub async fn embed(
|
||||
}
|
||||
).await?;
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
|
||||
#[poise::command(slash_command, prefix_command)]
|
||||
pub async fn eight_ball(
|
||||
ctx: Context<'_>,
|
||||
#[description = "Question to ask."] question: String
|
||||
) -> Result<(), Error>
|
||||
{
|
||||
let is_quirky = rand::rng().random_bool(0.2);
|
||||
let list = &ctx.data().ball_prompts[if is_quirky { 1 } else { 0 }];
|
||||
let rand_item = list.iter().choose(&mut rand::rng());
|
||||
|
||||
send_msg(
|
||||
ctx,
|
||||
format!("Q: {}\nA: {}", question, rand_item.unwrap()),
|
||||
true
|
||||
).await?;
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
+5
-1
@@ -9,7 +9,7 @@ pub fn event_handler<'a>(
|
||||
ctx: &'a serenity::Context,
|
||||
event: &'a serenity::FullEvent,
|
||||
_framework: poise::FrameworkContext<'a, Data, Error>,
|
||||
_data: &'a Data,
|
||||
data: &'a Data,
|
||||
) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'a>> {
|
||||
Box::pin(async move {
|
||||
if let serenity::FullEvent::Ready { data_about_bot } = event {
|
||||
@@ -19,6 +19,10 @@ pub fn event_handler<'a>(
|
||||
data_about_bot.user.id
|
||||
);
|
||||
|
||||
if data.dev {
|
||||
println!("----- DEV MODE ENABLED -----");
|
||||
}
|
||||
|
||||
let file_text = std::fs::read_to_string("./data/status.txt").unwrap();
|
||||
let custom_activity = ActivityData::custom(file_text);
|
||||
// TODO: make custom rich presence
|
||||
|
||||
+13
-3
@@ -7,7 +7,8 @@ use std::env;
|
||||
use poise::serenity_prelude::{self as serenity, Color, CreateEmbed, Timestamp};
|
||||
|
||||
struct Data {
|
||||
dev: bool
|
||||
dev: bool,
|
||||
ball_prompts: [Vec<String>; 2]
|
||||
}
|
||||
type Error = Box<dyn std::error::Error + Send + Sync>;
|
||||
type Context<'a> = poise::Context<'a, Data, Error>;
|
||||
@@ -38,8 +39,16 @@ impl Default for EmbedOptions {
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
let args: Vec<String> = env::args().collect();
|
||||
|
||||
let ball_classic_str = std::fs::read_to_string("./data/8-ball_classic.txt").unwrap();
|
||||
let ball_quirk_str = std::fs::read_to_string("./data/8-ball_quirky.txt").unwrap();
|
||||
|
||||
let ball_classic: Vec<String> = ball_classic_str.lines().map(String::from).collect();
|
||||
let ball_quirk: Vec<String> = ball_quirk_str .lines().map(String::from).collect();
|
||||
|
||||
let data = Data {
|
||||
dev: args.contains(&"--dev".to_string())
|
||||
dev: args.contains(&"--dev".to_string()),
|
||||
ball_prompts: [ball_classic, ball_quirk]
|
||||
};
|
||||
|
||||
let token = std::env::var("ASSISTANT_TOKEN").expect("missing ASSISTANT_TOKEN env var");
|
||||
@@ -56,7 +65,8 @@ async fn main() {
|
||||
commands: vec![
|
||||
cmds::ping(),
|
||||
cmds::embed(),
|
||||
cmds::stop()
|
||||
cmds::stop(),
|
||||
cmds::eight_ball()
|
||||
],
|
||||
event_handler: events::event_handler,
|
||||
..Default::default()
|
||||
|
||||
Reference in New Issue
Block a user