pub mod normal { use crate::common::database::{DataBase, ServerMembersWolves, Servers, Wolves}; use crate::get_now_iso; use serenity::client::Context; use serenity::model::id::{GuildId, RoleId, UserId}; use sqlx::{Pool, Sqlite}; pub async fn update_server(ctx: &Context, server: &Servers, remove_roles: &[Option], members_changed: &[UserId]) { let db_lock = { let data_read = ctx.data.read().await; data_read.get::().expect("Expected Database in TypeMap.").clone() }; let db = db_lock.read().await; let Servers { server, role_past, role_current, .. } = server; let mut roles_set = [0, 0, 0]; let mut members = vec![]; for member in get_server_member_bulk(&db, server).await { if let Some(x) = member.discord { members.push(x); } } let mut members_all = members.len(); if let Ok(x) = server.members(ctx, None, None).await { for mut member in x { // members_changed acts as an override to only deal with teh users in it if !members_changed.is_empty() && !members_changed.contains(&member.user.id) { continue; } if members.contains(&member.user.id) { let mut roles = vec![]; if let Some(role) = &role_past { if !member.roles.contains(role) { roles_set[0] += 1; roles.push(role.to_owned()); } } if !member.roles.contains(role_current) { roles_set[1] += 1; roles.push(role_current.to_owned()); } if let Err(e) = member.add_roles(ctx, &roles).await { println!("{:?}", e); } } else { // old and never if let Some(role) = &role_past { if member.roles.contains(role) { members_all += 1; } } if member.roles.contains(role_current) { roles_set[2] += 1; // if theya re not a current member and have the role then remove it if let Err(e) = member.remove_role(ctx, role_current).await { println!("{:?}", e); } } } for role in remove_roles.iter().flatten() { if let Err(e) = member.remove_role(ctx, role).await { println!("{:?}", e); } } } } set_server_numbers(&db, server, members_all as i64, members.len() as i64).await; // small bit of logging to note changes over time println!("{:?} Changes: New: +{}, Current: +{}/-{}", server.as_u64(), roles_set[0], roles_set[1], roles_set[2]); } pub async fn get_server_member_bulk(db: &Pool, server: &GuildId) -> Vec { sqlx::query_as::<_, ServerMembersWolves>( r#" SELECT * FROM server_members JOIN wolves USING (id_wolves) WHERE ( server = ? AND discord IS NOT NULL AND expiry > ? ) "#, ) .bind(*server.as_u64() as i64) .bind(get_now_iso(true)) .fetch_all(db) .await .unwrap_or_default() } async fn set_server_numbers(db: &Pool, server: &GuildId, past: i64, current: i64) { match sqlx::query_as::<_, Wolves>( " UPDATE servers SET member_past = ?, member_current = ? WHERE server = ? ", ) .bind(past) .bind(current) .bind(*server.as_u64() as i64) .fetch_optional(db) .await { Ok(_) => {} Err(e) => { println!("Failure to insert into {}", server.as_u64()); println!("{:?}", e); } } } } // for updating committee members pub mod committee { use crate::common::database::{DataBase, Wolves}; use crate::common::wolves::committees::Committees; use crate::Config; use serenity::client::Context; use serenity::model::channel::ChannelType; use serenity::model::guild::Member; use serenity::model::id::ChannelId; use serenity::model::prelude::RoleId; use sqlx::{Pool, Sqlite}; use std::collections::HashMap; use std::sync::Arc; pub async fn check_committee(ctx: Arc) { let db_lock = { let data_read = ctx.data.read().await; data_read.get::().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::().expect("Expected Config in TypeMap.").clone() }; let config_global = config_lock.read().await; let server = config_global.committee_server; // because to use it to update a single user we need to pre-get the members of teh server let mut members = server.members(&ctx, None, None).await.unwrap_or_default(); update_committees(&db, &ctx, &config_global, &mut members).await; } /** This function can take a vec of members (or just one) and gives tehm the appropiate roles on teh committee server */ pub async fn update_committees(db: &Pool, ctx: &Context, config: &Config, members: &mut Vec) { let server = config.committee_server; let committee_member = config.committee_role; let committees = get_committees(db).await; let categories = vec![config.committee_category, ChannelId(1341457244973305927), ChannelId(1341457509717639279)]; // information about the server let roles = server.roles(&ctx).await.unwrap_or_default(); let channels = server.channels(&ctx).await.unwrap_or_default(); // make a hashmap of the nameof roles to quickly get them out again let mut roles_name = HashMap::new(); for role in roles.values() { roles_name.insert(role.name.to_owned(), role.to_owned()); } let mut channels_name = HashMap::new(); for channel in channels.values() { // we only care about teh channels in teh category if let Some(x) = channel.parent_id { for category in &categories { if x.eq(category) { channels_name.insert(channel.name.to_owned(), channel.to_owned()); } } } } // a map of users and the roles they are goign to be getting let mut users_roles = HashMap::new(); // a list of all the roles that can be removed from folks who should have them let mut committee_roles = vec![committee_member]; let mut category_index = 0; let mut i = 0; loop { if i >= committees.len() { break; } let committee = &committees[i]; // get the role for this committee/club/soc let role = match roles_name.get(&committee.name_full) { Some(x) => Some(x.to_owned()), None => { // create teh role if it does not exist match server.create_role(&ctx, |r| r.hoist(false).mentionable(true).name(&committee.name_full)).await { Ok(x) => Some(x), Err(_) => None, } } }; // create teh channel if it does nto exist if !channels_name.contains_key(&committee.name_profile) { match server .create_channel(&ctx, |c| c.name(&committee.name_profile).kind(ChannelType::Text).category(categories[category_index])) .await { Ok(x) => { // update teh channels name list channels_name.insert(x.name.to_owned(), x.to_owned()); println!("Created channel: {}", &committee.name_profile); } Err(x) => { let tmp = x.to_string(); if x.to_string().contains("Maximum number of channels in category reached (50)") { category_index += 1; continue; } dbg!("Unable to create channel: ", &tmp, &tmp.contains("Maximum number of channels in category reached (50)")); } } }; // so if the role exists if let Some(r) = role { committee_roles.push(r.id); for id_wolves in &committee.committee { // ID in this is the wolves ID, so we need to get a matching discord ID (if one exists) if let Some(x) = get_server_member_discord(db, id_wolves).await { if let Some(member_tmp) = x.discord { let values = users_roles.entry(member_tmp).or_insert(vec![committee_member]); values.push(r.id); } } } } i += 1; } // now we have a map of all users that should get roles time to go through all the folks on teh server for member in members { let roles_current = member.roles(ctx).unwrap_or_default(); let roles_required = match users_roles.get(&member.user.id) { None => { vec![] } Some(x) => { let mut tmp = x.to_owned(); if !tmp.is_empty() { tmp.push(RoleId(1226602779968274573)); } tmp } }; let mut roles_rem = vec![]; let mut roles_add = vec![]; // get a list of all the roles to remove from someone let mut roles_current_id = vec![]; for role in &roles_current { roles_current_id.push(role.id.to_owned()); if !roles_required.contains(&role.id) { roles_rem.push(role.id.to_owned()); } } for role in &roles_required { if !roles_current_id.contains(role) { roles_add.push(role.to_owned()); } } if !roles_rem.is_empty() { member.remove_roles(&ctx, &roles_rem).await.unwrap_or_default(); } if !roles_add.is_empty() { // these roles are flavor roles, only there to make folks mentionable member.add_roles(&ctx, &roles_add).await.unwrap_or_default(); } else { member.remove_roles(&ctx, &[RoleId(1226602779968274573)]).await.unwrap_or_default(); } } // finally re-order teh channels to make them visually apealing let mut channel_names = channels_name.clone().into_keys().collect::>(); channel_names.sort(); // get a list of all teh new positions let mut new_positions = vec![]; for (i, name) in channel_names.iter().enumerate() { if let Some(channel) = channels_name.get_mut(name) { let position_new = i as u64; if position_new != channel.position as u64 { new_positions.push((channel.id.to_owned(), position_new)); } } } if !new_positions.is_empty() { match server.reorder_channels(&ctx, new_positions).await { Ok(_) => { println!("Successfully re-orderd the committee category"); } Err(e) => { dbg!("Failed to re-order ", e); } } } } async fn get_committees(db: &Pool) -> Vec { sqlx::query_as::<_, Committees>( r#" SELECT * FROM committees "#, ) .fetch_all(db) .await.unwrap_or_else(|e| { dbg!(e); vec![] }) } async fn get_server_member_discord(db: &Pool, user: &i64) -> Option { sqlx::query_as::<_, Wolves>( r#" SELECT * FROM wolves WHERE id_wolves = ? "#, ) .bind(user) .fetch_one(db) .await .ok() } }