Extract trait

src/commands/pause.rs

@@ -1,22 +1,28 @@
 use chrono::NaiveDateTime;
+use serde::{Deserialize, Serialize};
 
-use crate::{models::CtxData, time_parser::natural_parser, Context, Error};
+use crate::{
+    models::CtxData, time_parser::natural_parser, utils::Extract, ApplicationContext, Context,
+    Error,
+};
 
-/// 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 pause(
-    ctx: Context<'_>,
-    #[description = "When to pause until"] until: Option<String>,
-) -> Result<(), 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 until {
+    match options.until {
         Some(until) => {
             let parsed = natural_parser(&until, &timezone.to_string()).await;
 
@@ -65,3 +71,16 @@ pub async fn pause(
 
     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
+}