cull-gmail 0.1.8

use std::fmt;

use clap::{Parser, ValueEnum};
use cull_gmail::{Error, MessageAge, Retention, Rules};

#[derive(Debug, Clone, Parser, ValueEnum)]
pub enum Period {
    Days,
    Weeks,
    Months,
    Years,
}

impl fmt::Display for Period {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Period::Days => write!(f, "days"),
            Period::Weeks => write!(f, "weeks"),
            Period::Months => write!(f, "months"),
            Period::Years => write!(f, "years"),
        }
    }
}

#[derive(Debug, Parser)]
pub struct AddRuleCli {
    /// Period for the rule
    #[arg(short, long)]
    period: Period,
    /// Count of the period
    #[arg(short, long, default_value = "1")]
    count: i64,
    /// Optional specific label; if not specified one will be generated
    #[arg(short, long)]
    label: Option<String>,
    /// Immediate delete instead of move to trash
    #[arg(long)]
    delete: bool,
}

impl AddRuleCli {
    pub fn run(&self, mut config: Rules) -> Result<(), Error> {
        let generate = self.label.is_none();
        let message_age = MessageAge::new(self.period.to_string().as_str(), self.count)?;
        let retention = Retention::new(message_age, generate);

        config.add_rule(retention, self.label.as_deref(), self.delete);
        config.save()
    }
}