discord-bot/src/bin/update_server-icon.rs

use chrono::{Datelike, Utc};
use rand::{rngs::SmallRng, seq::IndexedRandom, SeedableRng};
use serde::Deserialize;
use serenity::{
  all::GuildId,
  async_trait,
  builder::{CreateAttachment, EditGuild},
  client::{Context, EventHandler},
  model::gateway::{GatewayIntents, Ready},
  Client,
};
use skynet_discord_bot::{
  common::{
    database::{db_init, DataBase},
    renderer::{Args, Renderer},
  },
  get_config, get_now_iso, Config,
};
use sqlx::{Pool, Sqlite};
use std::{
  ffi::OsString,
  fs,
  path::PathBuf,
  process::{self, Command},
  sync::Arc,
};
use tokio::sync::RwLock;

#[tokio::main]
async fn main() {
  let config = get_config();
  let db = match db_init(&config).await {
    Ok(x) => x,
    Err(_) => return,
  };

  // Intents are a bitflag, bitwise operations can be used to dictate which intents to use
  let intents = GatewayIntents::GUILDS | GatewayIntents::GUILD_MESSAGES | GatewayIntents::MESSAGE_CONTENT | GatewayIntents::GUILD_MEMBERS;
  // Build our client.
  let mut client = Client::builder(&config.discord_token, intents)
    .event_handler(Handler {})
    .await
    .expect("Error creating client");

  {
    let mut data = client.data.write().await;

    data.insert::<Config>(Arc::new(RwLock::new(config)));
    data.insert::<DataBase>(Arc::new(RwLock::new(db)));
  }

  if let Err(why) = client.start().await {
    println!("Client error: {:?}", why);
  }
}

struct Handler;
#[async_trait]
impl EventHandler for Handler {
  async fn ready(&self, ctx: Context, ready: Ready) {
    let ctx = Arc::new(ctx);
    println!("{} is connected!", ready.user.name);

    // pull in the open governance repo
    // u[date committee server
    update_icon_main(Arc::clone(&ctx)).await;

    // finish up
    process::exit(0);
  }
}

async fn update_icon_main(ctx: Arc<Context>) {
  let db_lock = {
    let data_read = ctx.data.read().await;
    data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
  };
  let db = db_lock.read().await;

  let config_lock = {
    let data_read = ctx.data.read().await;
    data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
  };

  let config_global = config_lock.read().await;
  let config_toml = get_config_icons();
  let server = GuildId::new(689189992417067052);

  // clone repo into local folder
  clone_repo(&config_global, &config_toml);

  // 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<String>,
  exclusions: Vec<String>,
}

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
}

#[derive(Deserialize)]
struct ConfigToml {
  source: ConfigTomlSource,
  festivals: Vec<ConfigTomlFestivals>,
}
#[derive(Deserialize, Debug)]
struct ConfigTomlSource {
  repo: String,
  directory: String,
}

#[derive(Deserialize, Debug)]
struct ConfigTomlFestivals {
  name: String,
  all_year: bool,
  start: ConfigTomlFestivalsTime,
  end: ConfigTomlFestivalsTime,
}
#[derive(Deserialize, Debug)]
struct ConfigTomlFestivalsTime {
  day: u32,
  month: u32,
  year: i32,
}

fn get_config_icons() -> ConfigToml {
  let toml_raw = include_str!("../../.server-icons.toml");
  let config: ConfigToml = toml::from_str(toml_raw).unwrap();
  config
}
fn clone_repo(config: &Config, config_toml: &ConfigToml) {
  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");
}

#[derive(Debug)]
struct LogoData {
  name: OsString,
  path: PathBuf,
}
fn get_logos(config: &Config, config_toml: &ConfigToml) -> Vec<LogoData> {
  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<LogoData>) -> Vec<LogoData> {
  let mut filtered: Vec<LogoData> = 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
}

#[derive(Debug, Clone, sqlx::FromRow)]
pub struct ServerIcons {
  pub id: i64,
  pub name: String,
  pub date: String,
}

async fn logo_set(ctx: &Arc<Context>, db: &Pool<Sqlite>, 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<Sqlite>, 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);
    }
  }
}

// fn command(config_toml: &ConfigToml, logo_selected: &LogoData){
//   let web_url = format!("{}/src/branch/main/{}/{}",  &config_toml.source.repo, &config_toml.source.directory, &logo_selected.name.to_str().unwrap_or_default());
// }