blob: c8405a0ad3304cdc645047fe7b5ad0de3a7c33e0 (
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
|
use crate::Context;
use color_eyre::eyre::Result;
use log::debug;
use rand::Rng;
/// Generate some amount of uwurandom
#[poise::command(slash_command)]
pub async fn uwurandom(
ctx: Context<'_>,
#[description = "The amount of uwurandom to generate"]
#[min = 1]
#[max = 2000]
length: Option<u16>,
) -> Result<()> {
let gid = ctx.guild_id().unwrap_or_default();
let settings = ctx.data().storage.get_guild_settings(&gid).await?;
if !settings.optional_commands_enabled {
debug!("Not running uwurandom in {gid} since it's disabled");
ctx.say("I'm not allowed to do that here").await?;
return Ok(());
}
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(())
}
|
