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.
2b-rs/src/commands/misc/pekofy.rs

117 lines
3.3 KiB
Rust

use rand::prelude::*;
use regex::Regex;
use serenity::framework::standard::{Args, CommandError, CommandResult};
use serenity::model::channel::Message;
use serenity::{framework::standard::macros::command, prelude::*};
// return a normal peko in most cases
static PEKOS: &[&str] = &[
"peko",
"p e k o",
"peeeeekooooo",
"pppeeekkkooo",
"🇵 🇪 🇰 🇴",
"p3k0",
];
#[command]
#[description("Pekofy messages")]
#[usage("(<content>)")]
#[example("Hello")]
#[aliases("peko")]
async fn pekofy(ctx: &Context, msg: &Message, args: Args) -> CommandResult {
let mut reference_message = msg.id;
let mut content = args.message().to_string();
if args.is_empty() {
if let Some(reference) = &msg.referenced_message {
reference_message = reference.id;
content = reference.content.clone();
} else {
let messages = msg
.channel_id
.messages(ctx, |ret| ret.before(&msg.id).limit(1))
.await?;
let reference = messages
.first()
.ok_or(CommandError::from("No message to pekofy"))?;
reference_message = reference.id;
content = reference.content.clone();
};
let _ = msg.delete(ctx).await;
}
if content.is_empty() {
return Err(CommandError::from("Can't pekofy empty message"));
}
log::debug!("Pekofying message '{}'", content);
let mut alpha_lowercase = content.to_lowercase();
alpha_lowercase.retain(|c| c.is_alphanumeric());
let pekofied: String = if alpha_lowercase == "pain" {
"https://tenor.com/view/pekora-usada-peko-hololive-died-gif-18114577".to_string()
} else if PEKOS.contains(&&*alpha_lowercase) {
random_peko()
} else {
content
.lines()
.into_iter()
.map(pekofy_line)
.collect::<Vec<String>>()
.join("\n")
};
let message = ctx
.http
.get_message(msg.channel_id.0, reference_message.0)
.await?;
log::debug!("Pekofied message is '{}'", pekofied);
message.reply(ctx, pekofied).await?;
Ok(())
}
/// Pekofies a single line
fn pekofy_line(mut line: &str) -> String {
lazy_static::lazy_static! { static ref FORMATTING_REGEX: Regex = Regex::new(r"^(.*?)((<:\w+:\d+>|\W)*)$").unwrap(); }
log::debug!("Pekofying line '{}'", line);
let original = line;
let mut md = "";
if let Some(captures) = FORMATTING_REGEX.captures(line) {
line = captures.get(1).unwrap().as_str();
md = captures.get(2).unwrap().as_str();
}
for peko in PEKOS {
if line.to_lowercase().ends_with(peko) {
log::debug!("Peko already found in message. Returning original");
return original.to_string();
}
}
let mut peko = random_peko();
if line
.chars()
.filter(|c| c.is_alphabetic())
.all(char::is_uppercase)
{
log::debug!("Message is all uppercase. Peko will also be uppercase");
peko = peko.to_uppercase();
}
format!("{} {}{}", line, peko, md)
}
/// Returns a random peko
fn random_peko() -> String {
let mut rng = rand::thread_rng();
if rng.gen_range(0..20) == 10 {
PEKOS.choose(&mut rng).unwrap().to_string()
} else {
"peko".to_string()
}
}