revanced-discord-bot/src/events/ready.rs

use chrono::Utc;

use super::*;
use crate::db::model::Muted;
use crate::utils::moderation::queue_unmute_member;

pub async fn load_muted_members(ctx: &serenity::Context, data: &Arc<RwLock<Data>>) {
    let data = &mut *data.write().await;
    let mute_role_id = data.configuration.mute.role;

    let mut cursor = data
        .database
        .find::<Muted>(
            "muted",
            Muted {
                ..Default::default()
            }
            .into(),
            None,
        )
        .await
        .unwrap();

    while cursor.advance().await.unwrap() {
        let current: Muted = cursor.deserialize_current().unwrap();
        let Some(expires) = current.expires else {
            continue;
        };
        let guild_id = current.guild_id.unwrap().parse::<u64>().unwrap();
        let user_id = current.user_id.unwrap().parse::<u64>().unwrap();

        let amount_left = std::cmp::max(expires as i64 - Utc::now().timestamp(), 0);

        data.pending_unmutes.insert(
            user_id,
            queue_unmute_member(
                ctx.clone(),
                data.database.clone(),
                guild_id.into(),
                user_id.into(),
                mute_role_id,
                amount_left as u64,
            ),
        );
    }
}