reminder-bot/src/commands/pause.rs

use chrono::NaiveDateTime;
use serde::{Deserialize, Serialize};

use crate::{
    models::CtxData, time_parser::natural_parser, utils::Extract, ApplicationContext, Context,
    Error,
};

#[derive(Serialize, Deserialize, Extract)]
pub struct Options {
    until: Option<String>,
}

impl Extract for Options {
    fn extract(ctx: ApplicationContext) -> Self {
        Self { until: extract_arg!(ctx, "until", Option<String>) }
    }
}

pub async fn pause(ctx: Context<'_>, options: Options) -> Result<(), Error> {
    let timezone = ctx.timezone().await;

    let mut channel = ctx.channel_data().await.unwrap();

    match options.until {
        Some(until) => {
            let parsed = natural_parser(&until, &timezone.to_string()).await;

            if let Some(timestamp) = parsed {
                match NaiveDateTime::from_timestamp_opt(timestamp, 0) {
                    Some(dt) => {
                        channel.paused = true;
                        channel.paused_until = Some(dt);

                        channel.commit_changes(&ctx.data().database).await;

                        ctx.say(format!(
                            "Reminders in this channel have been silenced until **<t:{}:D>**",
                            timestamp
                        ))
                        .await?;
                    }

                    None => {
                        ctx.say(
                            "Time processed could not be interpreted as `DateTime`. Please write the time as clearly as possible",
                        )
                        .await?;
                    }
                }
            } else {
                ctx.say(
                    "Time could not be processed. Please write the time as clearly as possible",
                )
                .await?;
            }
        }
        _ => {
            channel.paused = !channel.paused;
            channel.paused_until = None;

            channel.commit_changes(&ctx.data().database).await;

            if channel.paused {
                ctx.say("Reminders in this channel have been silenced indefinitely").await?;
            } else {
                ctx.say("Reminders in this channel have been unsilenced").await?;
            }
        }
    }

    Ok(())
}

/// Pause all reminders on the current channel until a certain time or indefinitely
#[poise::command(
    slash_command,
    identifying_name = "pause",
    default_member_permissions = "MANAGE_GUILD"
)]
pub async fn command(
    ctx: Context<'_>,
    #[description = "When to pause until"] until: Option<String>,
) -> Result<(), Error> {
    pause(ctx, Options { until }).await
}