use std::{collections::HashMap, env, sync::atomic::Ordering}; use log::{info, warn}; use poise::serenity::{client::Context, model::interactions::Interaction, utils::shard_id}; use crate::{Data, Error}; pub async fn listener(ctx: &Context, event: &poise::Event<'_>, data: &Data) -> Result<(), Error> { match event { poise::Event::CacheReady { .. } => { info!("Cache Ready!"); info!("Preparing to send reminders"); if !data.is_loop_running.load(Ordering::Relaxed) { let ctx1 = ctx.clone(); let ctx2 = ctx.clone(); let pool1 = data.database.clone(); let pool2 = data.database.clone(); let run_settings = env::var("DONTRUN").unwrap_or_else(|_| "".to_string()); if !run_settings.contains("postman") { tokio::spawn(async move { postman::initialize(ctx1, &pool1).await; }); } else { warn!("Not running postman") } if !run_settings.contains("web") { tokio::spawn(async move { reminder_web::initialize(ctx2, pool2).await.unwrap(); }); } else { warn!("Not running web") } data.is_loop_running.swap(true, Ordering::Relaxed); } } poise::Event::ChannelDelete { channel } => { sqlx::query!( " DELETE FROM channels WHERE channel = ? ", channel.id.as_u64() ) .execute(&data.database) .await .unwrap(); } poise::Event::GuildCreate { guild, is_new } => { if *is_new { let guild_id = guild.id.as_u64().to_owned(); sqlx::query!("INSERT INTO guilds (guild) VALUES (?)", guild_id) .execute(&data.database) .await .unwrap(); if let Ok(token) = env::var("DISCORDBOTS_TOKEN") { let shard_count = ctx.cache.shard_count(); let current_shard_id = shard_id(guild_id, shard_count); let guild_count = ctx .cache .guilds() .iter() .filter(|g| { shard_id(g.as_u64().to_owned(), shard_count) == current_shard_id }) .count() as u64; let mut hm = HashMap::new(); hm.insert("server_count", guild_count); hm.insert("shard_id", current_shard_id); hm.insert("shard_count", shard_count); let response = data .http .post( format!( "https://top.gg/api/bots/{}/stats", ctx.cache.current_user_id().as_u64() ) .as_str(), ) .header("Authorization", token) .json(&hm) .send() .await; if let Err(res) = response { println!("DiscordBots Response: {:?}", res); } } } } poise::Event::GuildDelete { incomplete, full } => { let _ = sqlx::query!("DELETE FROM guilds WHERE guild = ?", incomplete.id.0) .execute(&data.database) .await; } poise::Event::InteractionCreate { interaction } => match interaction { Interaction::MessageComponent(component) => { //let component_model = ComponentDataModel::from_custom_id(&component.data.custom_id); //component_model.act(&ctx, component).await; } _ => {} }, _ => {} } Ok(()) }