today/src/files/commands.rs

use chrono::NaiveDate;

use super::primitives::{Span, Spanned, Time, Weekday};

#[derive(Debug, Clone, Copy)]
pub enum DeltaStep {
    /// `y`, move by a year, keeping the same month and day
    Year(i32),
    /// `m`, move by a month, keeping the same day `d`
    Month(i32),
    /// `M`, move by a month, keeping the same day `D`
    MonthReverse(i32),
    /// `d`
    Day(i32),
    /// `w`, move by 7 days
    Week(i32),
    /// `h`
    Hour(i32),
    /// `m`
    Minute(i32),
    /// `mon`, `tue`, `wed`, `thu`, `fri`, `sat`, `sun`
    ///
    /// Move to the next occurrence of the specified weekday
    Weekday(i32, Weekday),
}

impl DeltaStep {
    pub fn amount(&self) -> i32 {
        match self {
            DeltaStep::Year(i) => *i,
            DeltaStep::Month(i) => *i,
            DeltaStep::MonthReverse(i) => *i,
            DeltaStep::Day(i) => *i,
            DeltaStep::Week(i) => *i,
            DeltaStep::Hour(i) => *i,
            DeltaStep::Minute(i) => *i,
            DeltaStep::Weekday(i, _) => *i,
        }
    }

    pub fn name(&self) -> &'static str {
        match self {
            DeltaStep::Year(_) => "y",
            DeltaStep::Month(_) => "m",
            DeltaStep::MonthReverse(_) => "M",
            DeltaStep::Day(_) => "d",
            DeltaStep::Week(_) => "w",
            DeltaStep::Hour(_) => "h",
            DeltaStep::Minute(_) => "min",
            DeltaStep::Weekday(_, wd) => wd.name(),
        }
    }
}

#[derive(Debug, Default)]
pub struct Delta(pub Vec<Spanned<DeltaStep>>);

#[derive(Debug)]
pub struct Repeat {
    /// Start at the date when the latest `DONE` was created instead of the
    /// task's previous occurrence.
    pub start_at_done: bool,
    pub delta: Spanned<Delta>,
}

#[derive(Debug)]
pub struct DateSpec {
    pub start: NaiveDate,
    pub start_delta: Option<Delta>,
    pub start_time: Option<Time>,
    pub end: Option<Spanned<NaiveDate>>,
    pub end_delta: Option<Delta>,
    pub end_time: Option<Spanned<Time>>,
    pub repeat: Option<Repeat>,
}

#[derive(Debug)]
pub struct WeekdaySpec {
    pub start: Weekday,
    pub start_time: Option<Time>,
    pub end: Option<Spanned<Weekday>>,
    pub end_delta: Option<Delta>,
    pub end_time: Option<Spanned<Time>>,
}

#[derive(Debug, Clone, Copy)]
pub enum Var {
    /// `true`, always 1
    True,
    /// `false`, always 0
    False,
    /// `mon`, always 1
    Monday,
    /// `tue`, always 2
    Tuesday,
    /// `wed`, always 3
    Wednesday,
    /// `thu`, always 4
    Thursday,
    /// `fri`, always 5
    Friday,
    /// `sat`, always 6
    Saturday,
    /// `sun`, always 7
    Sunday,
    /// `j`, see <https://en.wikipedia.org/wiki/Julian_day>
    JulianDay,
    /// `y`
    Year,
    /// `yl`, length of the current year in days
    ///
    /// Equal to `isLeapYear ? 366 : 365`
    YearLength,
    /// `yd`, day of the year
    YearDay,
    /// `yD`, day of the year starting from the end
    ///
    /// Equal to `yl - yd + 1`
    YearDayReverse,
    /// `yw`, 1 during the first 7 days of the year, 2 during the next etc.
    ///
    /// Equal to `((yd - 1) / 7) + 1`
    YearWeek,
    /// `yW`, 1 during the last 7 days of the year, 2 during the previous etc.
    ///
    /// Equal to `((yD - 1) / 7) + 1`
    YearWeekReverse,
    /// `m`
    Month,
    /// `ml`, length of the current month in days
    MonthLength,
    /// `mw`, 1 during the first 7 days of the month, 2 during the next etc.
    ///
    /// Equal to `((md - 1) / 7) + 1`
    MonthWeek,
    /// `mW`, 1 during the last 7 days of the month, 2 during the previous etc.
    ///
    /// Equal to `((mD - 1) / 7) + 1`
    MonthWeekReverse,
    /// `d`, day of the month
    Day,
    /// `D`, day of the month starting from the end
    ///
    /// Equal to `ml - md + 1`
    DayReverse,
    /// `iy`, ISO 8601 year
    IsoYear,
    /// `iyl`, length of current ISO 8601 year **in weeks**
    IsoYearLength,
    /// `iw`, ISO 8601 week
    IsoWeek,
    /// `wd`, day of the week, starting at monday with 1
    Weekday,
    /// `e`, day of the year that easter falls on
    Easter,
    /// `isWeekday`, whether the current day is one of mon-fri
    IsWeekday,
    /// `isWeekend`, whether the current day is one of sat-sun
    IsWeekend,
    /// `isLeapYear`, whether the current year is a leap year
    IsLeapYear,
}

impl Var {
    pub fn name(&self) -> &'static str {
        match self {
            // Constants
            Var::True => "true",
            Var::False => "false",
            Var::Monday => "mon",
            Var::Tuesday => "tue",
            Var::Wednesday => "wed",
            Var::Thursday => "thu",
            Var::Friday => "fri",
            Var::Saturday => "sat",
            Var::Sunday => "sun",
            // Variables
            Var::JulianDay => "j",
            Var::Year => "y",
            Var::YearLength => "yl",
            Var::YearDay => "yd",
            Var::YearDayReverse => "yD",
            Var::YearWeek => "yw",
            Var::YearWeekReverse => "yW",
            Var::Month => "m",
            Var::MonthLength => "ml",
            Var::MonthWeek => "mw",
            Var::MonthWeekReverse => "mW",
            Var::Day => "d",
            Var::DayReverse => "D",
            Var::IsoYear => "iy",
            Var::IsoYearLength => "iyl",
            Var::IsoWeek => "iw",
            Var::Weekday => "wd",
            Var::Easter => "e",
            // Variables with "boolean" values
            Var::IsWeekday => "isWeekday",
            Var::IsWeekend => "isWeekend",
            Var::IsLeapYear => "isLeapYear",
        }
    }
}

#[derive(Debug, Clone)]
pub enum Expr {
    Lit(i64),
    Var(Var),
    Paren(Box<Spanned<Expr>>),
    // Integer-y operations
    Neg(Box<Spanned<Expr>>),
    Add(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Sub(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Mul(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Div(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Mod(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    // Comparisons
    Eq(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Neq(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Lt(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Lte(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Gt(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Gte(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    // Boolean-y operations
    Not(Box<Spanned<Expr>>),
    And(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Or(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
    Xor(Box<Spanned<Expr>>, Box<Spanned<Expr>>),
}

#[derive(Debug)]
pub struct FormulaSpec {
    pub start: Option<Spanned<Expr>>, // None: *
    pub start_delta: Option<Delta>,
    pub start_time: Option<Time>,
    pub end_delta: Option<Delta>,
    pub end_time: Option<Spanned<Time>>,
}

#[derive(Debug)]
pub enum Spec {
    Date(DateSpec),
    Weekday(WeekdaySpec),
    Formula(FormulaSpec),
}

#[derive(Debug)]
pub struct BirthdaySpec {
    pub date: NaiveDate,
    pub year_known: bool, // If year is unknown, use NaiveDate of year 0
}

#[derive(Debug)]
pub enum Statement {
    Date(Spec),
    BDate(BirthdaySpec),
    From(Option<NaiveDate>),
    Until(Option<NaiveDate>),
    Except(NaiveDate), // TODO Allow excluding ranges
    Move {
        span: Span,
        from: NaiveDate,
        to: NaiveDate,
    },
}

#[derive(Debug, Clone, Copy)]
pub enum DoneDate {
    Date {
        root: NaiveDate,
    },
    DateTime {
        root: NaiveDate,
        root_time: Time,
    },
    DateToDate {
        root: NaiveDate,
        other: NaiveDate,
    },
    DateTimeToTime {
        root: NaiveDate,
        root_time: Time,
        other_time: Time,
    },
    DateTimeToDateTime {
        root: NaiveDate,
        root_time: Time,
        other: NaiveDate,
        other_time: Time,
    },
}

impl DoneDate {
    pub fn root(self) -> NaiveDate {
        match self {
            DoneDate::Date { root } => root,
            DoneDate::DateTime { root, .. } => root,
            DoneDate::DateToDate { root, .. } => root,
            DoneDate::DateTimeToTime { root, .. } => root,
            DoneDate::DateTimeToDateTime { root, .. } => root,
        }
    }
}

#[derive(Debug)]
pub struct Done {
    pub date: Option<DoneDate>,
    pub done_at: NaiveDate,
}

#[derive(Debug)]
pub struct Task {
    pub title: String,
    pub statements: Vec<Statement>,
    pub done: Vec<Done>,
    pub desc: Vec<String>,
}

#[derive(Debug)]
pub struct Note {
    pub title: String,
    pub statements: Vec<Statement>,
    pub desc: Vec<String>,
}

#[derive(Debug)]
pub enum Command {
    Task(Task),
    Note(Note),
}

impl Command {
    pub fn title(&self) -> &str {
        match self {
            Self::Task(task) => &task.title,
            Self::Note(note) => &note.title,
        }
    }

    pub fn desc(&self) -> &[String] {
        match self {
            Self::Task(task) => &task.desc,
            Self::Note(note) => &note.desc,
        }
    }

    pub fn statements(&self) -> &[Statement] {
        match self {
            Self::Task(task) => &task.statements,
            Self::Note(note) => &note.statements,
        }
    }
}

#[derive(Debug)]
pub struct File {
    pub contents: String,
    pub includes: Vec<String>,
    pub timezone: Option<String>,
    pub commands: Vec<Command>,
}