use serde::Deserialize; use std::{ffi::OsString, fs, path::PathBuf}; pub mod get_config_icons { use super::*; use crate::Config; #[derive(Deserialize)] pub struct ConfigToml { pub source: ConfigTomlSource, pub festivals: Vec, } #[derive(Deserialize)] pub struct ConfigTomlLocal { pub source: ConfigTomlSource, } #[derive(Deserialize)] pub struct ConfigTomlRemote { pub festivals: Vec, } #[derive(Deserialize, Debug)] pub struct ConfigTomlSource { pub repo: String, pub directory: String, pub file: String, } #[derive(Deserialize, Debug)] pub struct ConfigTomlFestivals { pub name: String, pub all_year: bool, pub start: ConfigTomlFestivalsTime, pub end: ConfigTomlFestivalsTime, } #[derive(Deserialize, Debug)] pub struct ConfigTomlFestivalsTime { pub day: u32, pub month: u32, pub year: i32, } pub fn minimal() -> ConfigTomlLocal { let toml_raw_min = include_str!("../../.server-icons.toml"); let config_min: ConfigTomlLocal = toml::from_str(toml_raw_min).unwrap(); config_min } // since a copy of the festival file is in the repo we just need to get to it pub fn full(config: &Config) -> ConfigToml { let config_source = minimal(); let file_path = format!("{}/open-governance/{}/{}", &config.home, &config_source.source.directory, &config_source.source.file); let contents = fs::read_to_string(file_path).expect("Should have been able to read the file"); let config_festivals: ConfigTomlRemote = toml::from_str(&contents).unwrap(); ConfigToml { source: config_source.source, festivals: config_festivals.festivals, } } } #[derive(Debug)] pub struct LogoData { pub name: OsString, pub path: PathBuf, } #[derive(Debug, Clone, sqlx::FromRow)] pub struct ServerIcons { pub id: i64, pub name: String, pub date: String, } pub mod update_icon { use super::*; use crate::{ common::{ renderer::{Args, Renderer}, server_icon::get_config_icons::{self, ConfigToml, ConfigTomlLocal}, }, get_now_iso, Config, }; use chrono::{Datelike, Utc}; use rand::{rngs::SmallRng, seq::IndexedRandom, SeedableRng}; use serenity::{ all::GuildId, builder::{CreateAttachment, EditGuild}, client::Context, }; use sqlx::{Pool, Sqlite}; use std::process::Command; /// Update the server icon, pulling from open governance. pub async fn update_icon_main(ctx: &Context, db: &Pool, config_global: &Config, config_toml_local: &ConfigTomlLocal) { let server = GuildId::new(689189992417067052); // clone repo into local folder clone_repo(config_global, config_toml_local); // now the repo has been downloaded/updated we can now access the festivals let config_toml = get_config_icons::full(config_global); // see if there is a current festival let festival_data = get_festival(&config_toml); // get a list of all the graphics files let logos = get_logos(config_global, &config_toml); // filter them so only the current season (if any) are active let logos_filtered = logos_filter(&festival_data, logos); let mut rng = SmallRng::from_os_rng(); let logo_selected = logos_filtered.choose(&mut rng).unwrap(); logo_set(ctx, db, &server, logo_selected).await; } #[derive(Debug)] struct FestivalData { current: Vec, exclusions: Vec, } fn get_festival(config_toml: &ConfigToml) -> FestivalData { let today = Utc::now(); let day = today.day(); let month = today.month(); let year = today.year(); let mut result = FestivalData { current: vec![], exclusions: vec![], }; for festival in &config_toml.festivals { if (day >= festival.start.day && day <= festival.end.day) && (month >= festival.start.month && month <= festival.end.month) { if festival.start.year == 0 || festival.end.year == 0 || (year >= festival.start.year && year <= festival.end.year) { result.current.push(festival.name.to_owned()); } } else if !festival.all_year { result.exclusions.push(festival.name.to_owned()); } } result } fn clone_repo(config: &Config, config_toml: &ConfigTomlLocal) { let url = &config_toml.source.repo; let folder = format!("{}/open-governance", &config.home); Command::new("git").arg("clone").arg(url).arg(&folder).output().expect("failed to execute process"); Command::new("git") .arg("pull") .arg("origin") .arg("main") .current_dir(&folder) .output() .expect("failed to execute process"); } fn get_logos(config: &Config, config_toml: &ConfigToml) -> Vec { let folder = format!("{}/open-governance/{}", &config.home, &config_toml.source.directory); let folder_path = PathBuf::from(&folder); let mut folder_output = folder_path.clone(); folder_output.push("converted"); let paths = fs::read_dir(folder).unwrap(); let args = Args { input: folder_path.clone(), output: folder_output, colors: String::from(""), width: 1024, height: 1024, }; let mut r = Renderer::new(&args).unwrap(); let mut logos = vec![]; for tmp in paths.flatten() { let path_local = tmp.path().to_owned(); let path_local2 = tmp.path().to_owned(); let name = path_local2.file_name().unwrap().to_owned(); let mut path = tmp.path(); if path.is_dir() { continue; } match tmp.path().extension() { None => {} Some(ext) => { if ext == "svg" { let mut path_new = path_local.clone(); path_new.set_extension("png"); let filename_tmp = path_new.clone(); let filename = filename_tmp.file_name().unwrap_or_default(); path_new.pop(); path_new.push("converted"); path_new.push(filename); // check if exists if !path_new.exists() { // convert if it hasnt been converted already match r.render(&path_local, &args) { Ok(_) => {} Err(_e) => { dbg!("Failed to render {path_local:?}: {}"); } } } path = path_new; } } }; logos.push(LogoData { name, path, }); // println!("Name: {}", &tmp.path().display()); } logos } fn logos_filter(festival_data: &FestivalData, existing: Vec) -> Vec { let mut filtered: Vec = vec![]; 'outer: for logo in existing { let name_lowercase0 = logo.name.to_ascii_lowercase(); let name_lowercase = name_lowercase0.to_str().unwrap_or_default(); if !festival_data.current.is_empty() { // if its a current festival filter based on it for festival in &festival_data.current { if name_lowercase.contains(festival) { filtered.push(logo); continue 'outer; } } } else { // else filter using the excluded ones let mut excluded = false; for festival in &festival_data.exclusions { if name_lowercase.contains(festival) { excluded = true; } } if !excluded { filtered.push(logo); } } } filtered } async fn logo_set(ctx: &Context, db: &Pool, server: &GuildId, logo_selected: &LogoData) { // add to teh database logo_set_db(db, logo_selected).await; let icon = CreateAttachment::path(logo_selected.path.to_str().unwrap_or_default()).await.unwrap(); // assuming a `guild` has already been bound let builder = EditGuild::new().icon(Some(&icon)); server.edit(ctx, builder).await.unwrap(); } async fn logo_set_db(db: &Pool, logo_selected: &LogoData) { let name = logo_selected.name.to_str().unwrap_or_default(); match sqlx::query_as::<_, ServerIcons>( " INSERT OR REPLACE INTO server_icons (name, date) VALUES (?1, ?2) ", ) .bind(name) .bind(get_now_iso(false)) .fetch_optional(db) .await { Ok(_) => {} Err(e) => { dbg!(e); } } } }