summaryrefslogtreecommitdiff
path: root/src/commands/general/random.rs
blob: 053e4bb7c33e2d84f0ad04b44c76ee714f1484ce (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
use crate::{client::Context, http, utils};

use anyhow::Result;
use rand::Rng;

#[poise::command(slash_command, subcommands("teawie", "shiggy", "uwu"))]
#[allow(clippy::unused_async)]
pub async fn random(_: Context<'_>) -> Result<()> {
	Ok(())
}

/// Get a random teawie
#[poise::command(prefix_command, slash_command)]
pub async fn teawie(ctx: Context<'_>) -> Result<()> {
	let url = http::teawie::random(&ctx.data().http_client).await?;
	utils::send_url_as_embed(ctx, url).await?;

	Ok(())
}

/// Get a random shiggy
#[poise::command(prefix_command, slash_command)]
pub async fn shiggy(ctx: Context<'_>) -> Result<()> {
	let url = http::shiggy::random(&ctx.data().http_client).await?;
	utils::send_url_as_embed(ctx, url).await?;

	Ok(())
}

/// Some uwu
#[poise::command(prefix_command, slash_command)]
pub async fn uwu(
	ctx: Context<'_>,
	#[description = "The amount of uwurandom to generate"]
	#[min = 1]
	#[max = 100]
	length: Option<u16>,
) -> Result<()> {
	let length = length.unwrap_or(rand::thread_rng().gen_range(1..50));

	let mut result = String::with_capacity(length as usize);
	// ThreadRng is not Send(obviously), and rustc is slightly too paranoid about rng spilling to await point
	// So calm it by constraining it to a block
	{
		let mut rng = rand::thread_rng();
		let mut state_machine = uwurandom_rs::StateMachine::new(&mut rng);
		for _ in 0..length {
			let generated;
			(state_machine, generated) = state_machine.generate(&mut rng);
			result.push(generated);
		}
	}
	ctx.say(result).await?;

	Ok(())
}