timr-tui/src/widgets/pomodoro.rs

use crate::{
    args::ClockStyle,
    constants::TICK_VALUE_MS,
    events::{Event, EventHandler},
    utils::center,
    widgets::clock::{Clock, ClockWidget, Countdown},
};
use ratatui::{
    buffer::Buffer,
    crossterm::event::KeyCode,
    layout::{Constraint, Layout, Rect},
    text::Line,
    widgets::{StatefulWidget, Widget},
};
use std::{cmp::max, time::Duration};

use strum::Display;

use super::clock::ClockArgs;

#[derive(Debug, Clone, Display, Hash, Eq, PartialEq)]
enum Mode {
    Work,
    Pause,
}

#[derive(Debug, Clone)]
pub struct ClockMap {
    work: Clock<Countdown>,
    pause: Clock<Countdown>,
}

impl ClockMap {
    fn get(&mut self, mode: &Mode) -> &mut Clock<Countdown> {
        match mode {
            Mode::Work => &mut self.work,
            Mode::Pause => &mut self.pause,
        }
    }
}

#[derive(Debug, Clone)]
pub struct Pomodoro {
    mode: Mode,
    clock_map: ClockMap,
}

pub struct PomodoroArgs {
    pub work: Duration,
    pub pause: Duration,
    pub style: ClockStyle,
    pub with_decis: bool,
}

impl Pomodoro {
    pub fn new(args: PomodoroArgs) -> Self {
        let PomodoroArgs {
            work,
            pause,
            style,
            with_decis,
        } = args;
        Self {
            mode: Mode::Work,
            clock_map: ClockMap {
                work: Clock::<Countdown>::new(ClockArgs {
                    initial_value: work,
                    tick_value: Duration::from_millis(TICK_VALUE_MS),
                    style,
                    with_decis,
                }),
                pause: Clock::<Countdown>::new(ClockArgs {
                    initial_value: pause,
                    tick_value: Duration::from_millis(TICK_VALUE_MS),
                    style,
                    with_decis,
                }),
            },
        }
    }

    fn get_clock(&mut self) -> &mut Clock<Countdown> {
        self.clock_map.get(&self.mode)
    }

    pub fn set_style(&mut self, style: crate::args::ClockStyle) {
        self.clock_map.work.style = style;
        self.clock_map.pause.style = style;
    }

    pub fn set_with_decis(&mut self, with_decis: bool) {
        self.clock_map.work.with_decis = with_decis;
        self.clock_map.pause.with_decis = with_decis;
    }

    pub fn next(&mut self) {
        self.mode = match self.mode {
            Mode::Pause => Mode::Work,
            Mode::Work => Mode::Pause,
        };
    }
}

impl EventHandler for Pomodoro {
    fn update(&mut self, event: Event) -> Option<Event> {
        let edit_mode = self.get_clock().is_edit_mode();
        match event {
            Event::Tick => {
                self.get_clock().tick();
            }
            Event::Key(key) => match key.code {
                KeyCode::Char('s') => {
                    self.get_clock().toggle_pause();
                }
                KeyCode::Char('e') => {
                    self.get_clock().toggle_edit();
                }
                KeyCode::Left if edit_mode => {
                    self.get_clock().edit_next();
                }
                KeyCode::Left => {
                    // `next` is acting as same as a `prev` function, we don't have
                    self.next();
                }
                KeyCode::Right if edit_mode => {
                    self.get_clock().edit_prev();
                }
                KeyCode::Right => {
                    self.next();
                }
                KeyCode::Up if edit_mode => {
                    self.get_clock().edit_up();
                }
                KeyCode::Down if edit_mode => {
                    self.get_clock().edit_down();
                }
                KeyCode::Char('r') => {
                    self.get_clock().reset();
                }
                _ => return Some(event),
            },
            _ => return Some(event),
        }
        None
    }
}

pub struct PomodoroWidget;

impl StatefulWidget for PomodoroWidget {
    type State = Pomodoro;
    fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
        let clock_widget = ClockWidget::new();
        let label = Line::raw(
            (format!(
                "Pomodoro {} {}",
                state.mode.clone(),
                state.get_clock().get_mode()
            ))
            .to_uppercase(),
        );

        let area = center(
            area,
            Constraint::Length(max(
                clock_widget.get_width(
                    &state.get_clock().get_format(),
                    state.get_clock().with_decis,
                ),
                label.width() as u16,
            )),
            Constraint::Length(clock_widget.get_height() + 1 /* height of mode_str */),
        );

        let [v1, v2] =
            Layout::vertical(Constraint::from_lengths([clock_widget.get_height(), 1])).areas(area);

        clock_widget.render(v1, buf, state.get_clock());
        label.centered().render(v2, buf);
    }
}