feat: refactor entire project

This commit is contained in:
Rekai Nyangadzayi Musuka 2021-02-11 21:47:06 -06:00
parent e325b1bbc0
commit ac0d7cc28a
5 changed files with 367 additions and 315 deletions

54
src/general.rs Normal file
View File

@ -0,0 +1,54 @@
use log::{info, warn};
use serenity::client::Context;
use serenity::framework::standard::CommandResult;
use serenity::model::channel::Message;
use serenity::utils::Colour;
use std::time::{SystemTime, UNIX_EPOCH};
const BOT_SOURCE: &str = "https://github.com/Paoda/arona";
const GACHA_SOURCE: &str = "https://github.com/Paoda/bluearch-recruitment";
const IMG_SOURCE: &str = "https://thearchive.gg";
pub const BLUE_ARCHIVE_BLUE: Colour = Colour::from_rgb(0, 215, 251);
pub async fn ping(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator);
info!("Ping requested from {}", author_name);
let now = SystemTime::now();
match now.duration_since(UNIX_EPOCH) {
Ok(now_timestamp) => {
let diff = now_timestamp.as_millis() - msg.timestamp.timestamp_millis() as u128;
info!("It took {}ms to receive {}'s ping", diff, author_name);
msg.reply(ctx, format!("Pong! (Response: {}ms)", diff))
.await?;
}
Err(_) => {
warn!("Failed to calculate UNIX Timestamp");
msg.reply(ctx, "Pong! (Response: ??ms)").await?;
}
}
Ok(())
}
pub async fn source(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator);
info!("{} requested bot / gacha / image sources", author_name);
let channel = msg.channel_id;
channel
.send_message(ctx, |m| {
m.embed(|embed| {
embed
.field("Bot Source", BOT_SOURCE, false)
.field("Gacha Source", GACHA_SOURCE, false)
.field("Image Source", IMG_SOURCE, false)
.colour(BLUE_ARCHIVE_BLUE)
})
})
.await?;
Ok(())
}

85
src/image.rs Normal file
View File

@ -0,0 +1,85 @@
use image::imageops::FilterType;
use image::io::Reader as ImageReader;
use image::{Rgba, RgbaImage};
use lazy_static::lazy_static;
use log::{error, info, warn};
use std::collections::HashMap;
use std::io::Cursor;
use std::sync::Mutex;
type Cache = Mutex<HashMap<String, RgbaImage>>;
lazy_static! {
static ref CACHE: Cache = Mutex::new(HashMap::new());
}
pub async fn get_image_from_url(url: &str, width: u32, height: u32) -> RgbaImage {
if let Some(img) = check_cache(url) {
info!("Cache Hit for {}", url);
return img;
}
info!("Downloading {}", url);
match reqwest::get(url).await {
Ok(resp) => match resp.bytes().await {
Ok(bytes) => match ImageReader::new(Cursor::new(bytes)).with_guessed_format() {
Ok(img_reader) => match img_reader.decode() {
Ok(dynamic_img) => {
info!("Successfully decoded image from {}", url);
let img = image::imageops::resize(
&dynamic_img,
width,
height,
FilterType::Nearest,
);
add_to_cache(url, &img);
img
}
Err(err) => {
warn!("Decoding Error: {}", err);
generate_default_img(width, height)
}
},
Err(err) => {
error!("Unexpected IO Error Occurred: {}", err);
// We can recover here, but maybe it's worth panicking here?
generate_default_img(width, height)
}
},
Err(err) => {
warn!("Response Parse failed: {:}?", err);
generate_default_img(width, height)
}
},
Err(err) => {
warn!("Download failed: {:?}", err);
generate_default_img(width, height)
}
}
}
fn check_cache(url: &str) -> Option<RgbaImage> {
if let Ok(lock) = CACHE.lock() {
lock.get(url).cloned()
} else {
None
}
}
fn add_to_cache(url: &str, img: &RgbaImage) {
if let Ok(ref mut lock) = CACHE.lock() {
lock.insert(url.to_string(), img.clone());
}
}
fn generate_default_img(width: u32, height: u32) -> RgbaImage {
let mut img = RgbaImage::new(width, height);
for x in 0..height {
for y in 0..width {
img.put_pixel(x, y, Rgba([160, 32, 240, 1]));
}
}
img
}

3
src/lib.rs Normal file
View File

@ -0,0 +1,3 @@
pub mod general;
pub mod image;
pub mod recruitment;

View File

@ -1,14 +1,5 @@
use bluearch_recruitment::banner::{Banner, BannerBuilder};
use bluearch_recruitment::gacha::{GachaBuilder, Rarity, Recruitment as RecruitmentTrait};
use bluearch_recruitment::i18n::Language;
use bluearch_recruitment::student::Student;
use dotenv::dotenv; use dotenv::dotenv;
use image::codecs::jpeg::JpegEncoder; use log::{debug, error, info};
use image::imageops::FilterType;
use image::io::Reader as ImageReader;
use image::{ColorType, ImageEncoder, Rgba, RgbaImage};
use lazy_static::lazy_static;
use log::{debug, error, info, warn};
use serenity::async_trait; use serenity::async_trait;
use serenity::client::{Client, Context, EventHandler}; use serenity::client::{Client, Context, EventHandler};
use serenity::framework::standard::macros::{command, group, help}; use serenity::framework::standard::macros::{command, group, help};
@ -17,12 +8,9 @@ use serenity::framework::standard::{
}; };
use serenity::model::channel::Message; use serenity::model::channel::Message;
use serenity::model::id::UserId; use serenity::model::id::UserId;
use serenity::utils::Colour; use std::collections::HashSet;
use std::collections::{HashMap, HashSet};
use std::env; use std::env;
use std::io::Cursor;
use std::sync::Mutex;
use std::time::{Instant, SystemTime, UNIX_EPOCH};
#[group] #[group]
#[commands(ping, source)] #[commands(ping, source)]
struct General; struct General;
@ -36,22 +24,6 @@ struct Handler;
#[async_trait] #[async_trait]
impl EventHandler for Handler {} impl EventHandler for Handler {}
const STUDENTS_JSON: &str = include_str!("../data/students.json");
const CDN_URL: &str = "https://rerollcdn.com/BlueArchive";
const BOT_SOURCE: &str = "https://git.paoda.moe/paoda/arona";
const GACHA_SOURCE: &str = "https://github.com/Paoda/bluearch-recruitment";
const IMG_SOURCE: &str = "https://thearchive.gg";
const BANNER_IMG_URL: &str = "https://static.wikia.nocookie.net/blue-archive/images/e/e0/Gacha_Banner_01.png/revision/latest/";
const BLUE_ARCHIVE_BLUE: Colour = Colour::from_rgb(0, 215, 251);
type Cache = Mutex<HashMap<String, RgbaImage>>;
lazy_static! {
static ref STUDENTS: Vec<Student> = serde_json::from_str(STUDENTS_JSON).unwrap();
static ref BANNER: Banner = create_banner();
static ref CACHE: Cache = Mutex::new(HashMap::new());
}
#[tokio::main] #[tokio::main]
async fn main() { async fn main() {
dotenv().ok(); dotenv().ok();
@ -100,282 +72,30 @@ async fn main() {
#[command] #[command]
#[aliases(response)] #[aliases(response)]
async fn ping(ctx: &Context, msg: &Message) -> CommandResult { async fn ping(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator); arona::general::ping(ctx, msg).await
info!("Ping requested from {}", author_name);
let now = SystemTime::now();
match now.duration_since(UNIX_EPOCH) {
Ok(now_timestamp) => {
let diff = now_timestamp.as_millis() - msg.timestamp.timestamp_millis() as u128;
info!("It took {}ms to receive {}'s ping", diff, author_name);
msg.reply(ctx, format!("Pong! (Response: {}ms)", diff))
.await?;
}
Err(_) => {
warn!("Failed to calculate UNIX Timestamp");
msg.reply(ctx, "Pong! (Response: ??ms)").await?;
}
}
Ok(())
} }
#[command] #[command]
#[aliases(pull)] #[aliases(pull)]
async fn roll(ctx: &Context, msg: &Message) -> CommandResult { async fn roll(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator); arona::recruitment::roll(ctx, msg).await
info!("{} requested a single roll", author_name);
let channel = msg.channel_id;
let student = BANNER.roll();
let eng_name = student.name.get(Language::English).unwrap();
let url_name = if eng_name == "Junko" {
"Zunko"
} else {
&eng_name
};
let img_url = format!("{}/Characters/{}.png", CDN_URL, url_name);
let title_url = format!("https://www.thearchive.gg/characters/{}", url_name);
let icon_url = format!("{}/Icons/icon-brand.png", CDN_URL);
let rarity_colour = get_rarity_colour(student.rarity);
let rarity_str = match student.rarity {
Rarity::One => ":star:",
Rarity::Two => ":star::star:",
Rarity::Three => ":star::star::star:",
};
channel
.send_message(ctx, |m| {
m.embed(|embed| {
embed
.image(img_url)
.title(format!("{}", student.name))
.description(format!("{}\t{}", eng_name, rarity_str))
.url(title_url)
.footer(|footer| {
footer
.icon_url(icon_url)
.text("Image Source: https://thearchive.gg")
})
.colour(rarity_colour)
})
})
.await?;
Ok(())
} }
#[command] #[command]
#[aliases(tenroll)] #[aliases(tenroll)]
async fn roll10(ctx: &Context, msg: &Message) -> CommandResult { async fn roll10(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator); arona::recruitment::roll10(ctx, msg).await
info!("{} requested a ten roll", author_name);
let channel = msg.channel_id;
let typing = channel.start_typing(&ctx.http)?;
const RESIZE_WIDTH: u32 = 202; // OG: 404 (2020-02-11)
const RESIZE_HEIGHT: u32 = 228; // OG: 456 (2020-02-11)
const IMG_WIDTH: u32 = RESIZE_WIDTH * 5;
const IMG_HEIGHT: u32 = RESIZE_HEIGHT * 2;
let mut collage = RgbaImage::new(IMG_WIDTH, IMG_HEIGHT);
let mut images: Vec<RgbaImage> = Vec::with_capacity(10);
let students = BANNER.roll10();
let mut max_rarity = Rarity::One;
let start = Instant::now();
for student in students.iter() {
let eng_name = student.name.get(Language::English).unwrap();
let url_name = if eng_name == "Junko" {
"Zunko"
} else {
&eng_name
};
max_rarity = max_rarity.max(student.rarity);
let img_url = format!("{}/Characters/{}.png", CDN_URL, url_name);
let image = get_image_from_url(&img_url, RESIZE_WIDTH, RESIZE_HEIGHT).await;
images.push(image);
}
let elapsed_ms = (Instant::now() - start).as_millis();
info!("10-roll, DL, and resize took {}ms", elapsed_ms);
let start = Instant::now();
for x in (0..IMG_WIDTH).step_by(RESIZE_WIDTH as usize) {
let index: usize = (((IMG_WIDTH - x) / RESIZE_WIDTH) - 1) as usize;
// Top Image
image::imageops::overlay(&mut collage, &images[index], x, 0);
// Bottom Image
image::imageops::overlay(&mut collage, &images[index + 5], x, RESIZE_HEIGHT);
}
let elapsed_ms = (Instant::now() - start).as_millis();
info!("Collage Build took {}ms", elapsed_ms);
let mut jpeg = Vec::new();
let encoder = JpegEncoder::new(&mut jpeg);
let write_result = encoder.write_image(&collage, IMG_WIDTH, IMG_HEIGHT, ColorType::Rgba8);
if let Err(err) = write_result {
error!("Failed to Encode JPEG: {:?}", err);
msg.reply(
ctx,
"アロナ failed to perform your 10-roll. Please try again",
)
.await?;
return Ok(());
}
let files = vec![(jpeg.as_slice(), "result.jpeg")];
let _ = typing.stop();
channel
.send_files(ctx, files, |m| {
m.embed(|embed| {
embed
.title(format!("{} 10-roll", BANNER.name))
.description(BANNER.name.get(Language::English).unwrap())
.attachment("result.jpeg")
.colour(get_rarity_colour(max_rarity))
})
})
.await?;
Ok(())
}
async fn get_image_from_url(url: &str, width: u32, height: u32) -> RgbaImage {
if let Some(img) = check_cache(url) {
info!("Cache Hit for {}", url);
return img;
}
info!("Downloading {}", url);
match reqwest::get(url).await {
Ok(resp) => match resp.bytes().await {
Ok(bytes) => match ImageReader::new(Cursor::new(bytes)).with_guessed_format() {
Ok(img_reader) => match img_reader.decode() {
Ok(dynamic_img) => {
info!("Successfully decoded image from {}", url);
let img = image::imageops::resize(
&dynamic_img,
width,
height,
FilterType::Nearest,
);
add_to_cache(url, &img);
img
}
Err(err) => {
warn!("Decoding Error: {}", err);
generate_default_img(width, height)
}
},
Err(err) => {
error!("Unexpected IO Error Occurred: {}", err);
// We can recover here, but maybe it's worth panicking here?
generate_default_img(width, height)
}
},
Err(err) => {
warn!("Response Parse failed: {:}?", err);
generate_default_img(width, height)
}
},
Err(err) => {
warn!("Download failed: {:?}", err);
generate_default_img(width, height)
}
}
}
fn get_rarity_colour(rarity: Rarity) -> Colour {
match rarity {
Rarity::One => Colour::from_rgb(227, 234, 240),
Rarity::Two => Colour::from_rgb(255, 248, 124),
Rarity::Three => Colour::from_rgb(253, 198, 229),
}
}
fn generate_default_img(width: u32, height: u32) -> RgbaImage {
let mut img = RgbaImage::new(width, height);
for x in 0..height {
for y in 0..width {
img.put_pixel(x, y, Rgba([160, 32, 240, 1]));
}
}
img
}
fn check_cache(url: &str) -> Option<RgbaImage> {
if let Ok(lock) = CACHE.lock() {
lock.get(url).cloned()
} else {
None
}
}
fn add_to_cache(url: &str, img: &RgbaImage) {
if let Ok(ref mut lock) = CACHE.lock() {
lock.insert(url.to_string(), img.clone());
}
} }
#[command] #[command]
async fn banner(ctx: &Context, msg: &Message) -> CommandResult { async fn banner(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator); arona::recruitment::banner(ctx, msg).await
info!("{} requested banner information", author_name);
let channel = msg.channel_id;
let banner_eng = BANNER.name.get(Language::English).unwrap();
channel
.send_message(ctx, |m| {
m.embed(|embed| {
embed
.image(BANNER_IMG_URL)
.title(BANNER.name.clone())
.description(banner_eng)
.colour(BLUE_ARCHIVE_BLUE)
})
})
.await?;
Ok(())
} }
#[command] #[command]
#[aliases(github, code, dev)] #[aliases(github, code, dev)]
async fn source(ctx: &Context, msg: &Message) -> CommandResult { async fn source(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator); arona::general::source(ctx, msg).await
info!("{} requested bot / gacha / image sources", author_name);
let channel = msg.channel_id;
channel
.send_message(ctx, |m| {
m.embed(|embed| {
embed
.field("Bot Source", BOT_SOURCE, false)
.field("Gacha Source", GACHA_SOURCE, false)
.field("Image Source", IMG_SOURCE, false)
.colour(BLUE_ARCHIVE_BLUE)
})
})
.await?;
Ok(())
} }
#[help] #[help]
@ -392,30 +112,3 @@ async fn my_help(
let _ = help_commands::with_embeds(context, msg, args, help_options, groups, owners).await; let _ = help_commands::with_embeds(context, msg, args, help_options, groups, owners).await;
Ok(()) Ok(())
} }
pub fn create_banner() -> Banner {
let pool: Vec<Student> = STUDENTS
.iter()
.filter(|student| student.name != "ノゾミ")
.cloned()
.collect();
let sparkable: Vec<Student> = pool
.iter()
.filter(|student| student.name == "ホシノ" || student.name == "シロコ")
.cloned()
.collect();
let gacha = GachaBuilder::new(79.0, 18.5, 2.5)
.with_pool(pool)
.with_priority(&sparkable, 0.7)
.finish()
.unwrap();
BannerBuilder::new("ピックアップ募集")
.with_gacha(&gacha)
.with_name_translation(Language::English, "Rate-Up Recruitment")
.with_sparkable_students(&sparkable)
.finish()
.unwrap()
}

217
src/recruitment.rs Normal file
View File

@ -0,0 +1,217 @@
use crate::general::BLUE_ARCHIVE_BLUE;
use crate::image::get_image_from_url;
use bluearch_recruitment::banner::{Banner, BannerBuilder};
use bluearch_recruitment::gacha::Recruitment as RecruitmentTrait;
use bluearch_recruitment::gacha::{GachaBuilder, Rarity};
use bluearch_recruitment::i18n::Language;
use bluearch_recruitment::student::Student;
use image::jpeg::JpegEncoder;
use image::{ColorType, ImageEncoder, RgbaImage};
use lazy_static::lazy_static;
use log::{error, info};
use serenity::client::Context;
use serenity::framework::standard::CommandResult;
use serenity::model::channel::Message;
use serenity::utils::Colour;
use std::time::Instant;
const STUDENTS_JSON: &str = include_str!("../data/students.json");
const CDN_URL: &str = "https://rerollcdn.com/BlueArchive";
const BANNER_IMG_URL: &str = "https://static.wikia.nocookie.net/blue-archive/images/e/e0/Gacha_Banner_01.png/revision/latest/";
const THUMB_WIDTH: u32 = 202; // OG: 404 (2020-02-11) from https://thearchive.gg
const THUMB_HEIGHT: u32 = 228; // OG: 456 (2020-02-11) from https://thearchive.gg
lazy_static! {
static ref STUDENTS: Vec<Student> = serde_json::from_str(STUDENTS_JSON).unwrap();
static ref BANNER: Banner = create_banner();
}
pub async fn roll(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator);
info!("{} requested a single roll", author_name);
let channel = msg.channel_id;
let student = BANNER.roll();
let eng_name = student.name.get(Language::English).unwrap();
let url_name = if eng_name == "Junko" {
"Zunko"
} else {
&eng_name
};
let img_url = format!("{}/Characters/{}.png", CDN_URL, url_name);
let title_url = format!("https://www.thearchive.gg/characters/{}", url_name);
let icon_url = format!("{}/Icons/icon-brand.png", CDN_URL);
let rarity_colour = get_rarity_colour(student.rarity);
let rarity_str = match student.rarity {
Rarity::One => ":star:",
Rarity::Two => ":star::star:",
Rarity::Three => ":star::star::star:",
};
channel
.send_message(ctx, |m| {
m.embed(|embed| {
embed
.image(img_url)
.title(format!("{}", student.name))
.description(format!("{}\t{}", eng_name, rarity_str))
.url(title_url)
.footer(|footer| {
footer
.icon_url(icon_url)
.text("Image Source: https://thearchive.gg")
})
.colour(rarity_colour)
})
})
.await?;
Ok(())
}
pub async fn roll10(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator);
info!("{} requested a ten roll", author_name);
let channel = msg.channel_id;
let typing = channel.start_typing(&ctx.http)?;
const IMG_WIDTH: u32 = THUMB_WIDTH * 5;
const IMG_HEIGHT: u32 = THUMB_HEIGHT * 2;
let mut collage = RgbaImage::new(IMG_WIDTH, IMG_HEIGHT);
let mut images: Vec<RgbaImage> = Vec::with_capacity(10);
let students = BANNER.roll10();
let mut max_rarity = Rarity::One;
let start = Instant::now();
for student in students.iter() {
let eng_name = student.name.get(Language::English).unwrap();
let url_name = if eng_name == "Junko" {
"Zunko"
} else {
&eng_name
};
max_rarity = max_rarity.max(student.rarity);
let img_url = format!("{}/Characters/{}.png", CDN_URL, url_name);
let image = get_image_from_url(&img_url, THUMB_WIDTH, THUMB_HEIGHT).await;
images.push(image);
}
let elapsed_ms = (Instant::now() - start).as_millis();
info!("10-roll, DL, and resize took {}ms", elapsed_ms);
let start = Instant::now();
for x in (0..IMG_WIDTH).step_by(THUMB_WIDTH as usize) {
let index: usize = (((IMG_WIDTH - x) / THUMB_WIDTH) - 1) as usize;
// Top Image
image::imageops::overlay(&mut collage, &images[index], x, 0);
// Bottom Image
image::imageops::overlay(&mut collage, &images[index + 5], x, THUMB_HEIGHT);
}
let elapsed_ms = (Instant::now() - start).as_millis();
info!("Collage Build took {}ms", elapsed_ms);
let mut jpeg = Vec::new();
let encoder = JpegEncoder::new(&mut jpeg);
let write_result = encoder.write_image(&collage, IMG_WIDTH, IMG_HEIGHT, ColorType::Rgba8);
if let Err(err) = write_result {
error!("Failed to Encode JPEG: {:?}", err);
msg.reply(
ctx,
"アロナ failed to perform your 10-roll. Please try again",
)
.await?;
return Ok(());
}
let icon_url = format!("{}/Icons/icon-brand.png", CDN_URL);
let files = vec![(jpeg.as_slice(), "result.jpeg")];
let _ = typing.stop();
channel
.send_files(ctx, files, |m| {
m.embed(|embed| {
embed
.title(format!("{} 10-roll", BANNER.name))
.description(BANNER.name.get(Language::English).unwrap())
.attachment("result.jpeg")
.colour(get_rarity_colour(max_rarity))
.footer(|footer| {
footer
.icon_url(icon_url)
.text("Image Source: https://thearchive.gg")
})
})
})
.await?;
Ok(())
}
pub async fn banner(ctx: &Context, msg: &Message) -> CommandResult {
let author_name = format!("{}#{}", msg.author.name, msg.author.discriminator);
info!("{} requested banner information", author_name);
let channel = msg.channel_id;
let banner_eng = BANNER.name.get(Language::English).unwrap();
channel
.send_message(ctx, |m| {
m.embed(|embed| {
embed
.image(BANNER_IMG_URL)
.title(BANNER.name.clone())
.description(banner_eng)
.colour(BLUE_ARCHIVE_BLUE)
})
})
.await?;
Ok(())
}
pub fn create_banner() -> Banner {
let pool: Vec<Student> = STUDENTS
.iter()
.filter(|student| student.name != "ノゾミ")
.cloned()
.collect();
let sparkable: Vec<Student> = pool
.iter()
.filter(|student| student.name == "ホシノ" || student.name == "シロコ")
.cloned()
.collect();
let gacha = GachaBuilder::new(79.0, 18.5, 2.5)
.with_pool(pool)
.with_priority(&sparkable, 0.7)
.finish()
.unwrap();
BannerBuilder::new("ピックアップ募集")
.with_gacha(&gacha)
.with_name_translation(Language::English, "Rate-Up Recruitment")
.with_sparkable_students(&sparkable)
.finish()
.unwrap()
}
fn get_rarity_colour(rarity: Rarity) -> Colour {
match rarity {
Rarity::One => Colour::from_rgb(227, 234, 240),
Rarity::Two => Colour::from_rgb(255, 248, 124),
Rarity::Three => Colour::from_rgb(253, 198, 229),
}
}