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.
124 lines
3.5 KiB
Rust
124 lines
3.5 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::*};
|
|
|
|
use crate::utils::context_data::get_database_from_context;
|
|
use crate::utils::error::{BotError, BotResult};
|
|
use crate::utils::get_previous_message_or_reply;
|
|
use bot_database::models::Media;
|
|
|
|
// return a normal peko in most cases
|
|
static PEKOS: &[&str] = &[
|
|
"peko",
|
|
"p e k o",
|
|
"peeeeekooooo",
|
|
"pppeeekkkooo",
|
|
"🇵 🇪 🇰 🇴",
|
|
"p3k0",
|
|
];
|
|
static MEDIA_CATEGORY: &str = "pain-peko";
|
|
|
|
#[command]
|
|
#[description("Pekofy messages")]
|
|
#[usage("[<content>...]")]
|
|
#[example("Hello")]
|
|
#[aliases("peko")]
|
|
#[bucket("general")]
|
|
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() {
|
|
let reference = get_previous_message_or_reply(ctx, msg)
|
|
.await?
|
|
.ok_or(CommandError::from("No message to pekofy"))?;
|
|
reference_message = reference.id;
|
|
content = reference.content;
|
|
|
|
let _ = msg.delete(ctx).await;
|
|
}
|
|
if content.is_empty() {
|
|
return Err(CommandError::from("Can't pekofy empty message"));
|
|
}
|
|
|
|
tracing::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" {
|
|
random_pain_media(ctx).await?.url
|
|
} 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?;
|
|
tracing::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(); }
|
|
tracing::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) {
|
|
tracing::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)
|
|
{
|
|
tracing::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()
|
|
}
|
|
}
|
|
|
|
/// Chooses a random pain peko gif
|
|
async fn random_pain_media(ctx: &Context) -> BotResult<Media> {
|
|
let database = get_database_from_context(ctx).await;
|
|
let gifs = database.get_media_by_category(MEDIA_CATEGORY).await?;
|
|
gifs.into_iter()
|
|
.choose(&mut rand::thread_rng())
|
|
.ok_or(BotError::from("No media found"))
|
|
}
|