taskwarrior-tui/src/calendar.rs

// Based on https://gist.github.com/diwic/5c20a283ca3a03752e1a27b0f3ebfa30
// See https://old.reddit.com/r/rust/comments/4xneq5/the_calendar_example_challenge_ii_why_eddyb_all/

use std::fmt;

const COL_WIDTH: usize = 21;

use chrono::{Datelike, Local, Duration, NaiveDate, NaiveDateTime, TimeZone, Month};

use tui::{
    buffer::Buffer,
    layout::Rect,
    style::{Color, Modifier, Style},
    symbols,
    widgets::{Block, Widget},
};

use std::cmp::min;
use unicode_width::UnicodeWidthStr;

#[derive(Debug, Clone)]
pub struct Calendar<'a> {
    pub block: Option<Block<'a>>,
    pub year: i32,
    pub month: u32,
    pub style: Style,
    pub months_per_row: usize,
    pub date_style: Vec<(NaiveDate, Style)>
}

impl<'a> Default for Calendar<'a> {
    fn default() -> Calendar<'a> {
        let year = Local::today().year();
        let month = Local::today().month();
        Calendar {
            block: None,
            style: Default::default(),
            months_per_row: 0,
            year,
            month,
            date_style: vec![],
        }
    }
}

impl<'a> Calendar<'a> {
    pub fn block(mut self, block: Block<'a>) -> Self {
        self.block = Some(block);
        self
    }

    pub fn style(mut self, style: Style) -> Self {
        self.style = style;
        self
    }

    pub fn year(mut self, year: i32) -> Self {
        self.year = year;
        if self.year < 0 {
            self.year = 0;
        }
        self
    }

    pub fn month(mut self, month: u32) -> Self {
        self.month = month;
        self
    }

    pub fn date_style(mut self, date_style: Vec<(NaiveDate, Style)>) -> Self {
        self.date_style = date_style;
        self
    }
}

impl <'a> Widget for Calendar<'a> {
    fn render(mut self, area: Rect, buf: &mut Buffer) {
        let month_names = [
            Month::January.name(),
            Month::February.name(),
            Month::March.name(),
            Month::April.name(),
            Month::May.name(),
            Month::June.name(),
            Month::July.name(),
            Month::August.name(),
            Month::September.name(),
            Month::October.name(),
            Month::November.name(),
            Month::December.name(),
        ];
        buf.set_style(area, self.style);

        let area = match self.block.take() {
            Some(b) => {
                let inner_area = b.inner(area);
                b.render(area, buf);
                inner_area
            }
            None => area,
        };

        if area.height < 7 {
            return
        }

        let style = self.style;
        let today = Local::today();

        let year = self.year;
        let month = self.month;

        let months: Vec<_> = (0..12).collect();

        let mut days: Vec<_> = months.iter().map(|i| {
            let first = NaiveDate::from_ymd(year, i+1, 1);
            (first, first - Duration::days(first.weekday().num_days_from_sunday() as i64))
        }).collect();

        days.append(&mut months.iter().map(|i| {
                let first = NaiveDate::from_ymd(year + 1, i+1, 1);
                (first, first - Duration::days(first.weekday().num_days_from_sunday() as i64))
            }).collect::<Vec<_>>()
        );

        days.append(&mut months.iter().map(|i| {
                let first = NaiveDate::from_ymd(year + 2, i+1, 1);
                (first, first - Duration::days(first.weekday().num_days_from_sunday() as i64))
            }).collect::<Vec<_>>()
        );

        let mut startm = 0 as usize;
        if self.months_per_row > area.width as usize / 8 / 3 || self.months_per_row <= 0 {
            self.months_per_row = area.width as usize / 8 / 3;
        }
        let mut y = area.y;
        y += 1;

        let x = area.x;
        let s = format!("{year:^width$}", year = year, width = area.width as usize - 4);

        let mut year = 0;
        let mut style = Style::default().add_modifier(Modifier::UNDERLINED);
        if self.year + year as i32 == today.year() {
            style = style.add_modifier(Modifier::BOLD)
        }
        buf.set_string(x, y, &s, style);

        let startx = (area.width - 3 * 7 * self.months_per_row as u16 - self.months_per_row as u16) / 2;
        y += 2;
        loop {
            let endm = std::cmp::min(startm + self.months_per_row, 12);
            let mut x = area.x + startx;
            for c in startm..endm {
                if c > startm {
                    x += 1;
                }
                let d = &mut days[c];
                let m = d.0.month() as usize;
                let s = format!("{:^20}", month_names[m - 1]);
                let style = Style::default().bg(Color::Rgb(220, 220, 220));
                if m == today.month() as usize && self.year + year as i32 == today.year() {
                    buf.set_string(x, y, &s, style.add_modifier(Modifier::BOLD));
                } else {
                    buf.set_string(x, y, &s, style);
                }
                x += s.len() as u16 + 1;
            }
            y += 1;
            let mut x = area.x + startx;
            for c in startm..endm {
                let d = &mut days[c];
                let m = d.0.month() as usize;
                let style = Style::default().bg(Color::Rgb(220, 220, 220));
                if m == today.month() as usize && self.year + year as i32 == today.year() {
                    buf.set_string(x as u16, y, "Su Mo Tu We Th Fr Sa", style.add_modifier(Modifier::UNDERLINED));
                } else {
                    buf.set_string(x as u16, y, "Su Mo Tu We Th Fr Sa", style.add_modifier(Modifier::UNDERLINED));
                }
                x += 21 + 1;
            }
            y += 1;
            loop {
                let mut moredays = false;
                let mut x = area.x + startx;
                for c in startm..endm {
                    if c > startm {
                        x += 1;
                    }
                    let d = &mut days[c + year * 12];
                    for _ in 0..7 {
                        let s = if d.0.month() == d.1.month() {
                            format!("{:>2}", d.1.day())
                        } else {
                            "   ".to_string()
                        };
                        let mut style = Style::default();
                        let index = self.date_style.iter().position(|(date, style)| d.1 == *date);
                        match index {
                            Some(i) => style = self.date_style[i].1,
                            None => (),
                        }
                        if d.1 == Local::today().naive_local() {
                            buf.set_string(x, y, s, style.add_modifier(Modifier::BOLD));
                        } else {
                            buf.set_string(x, y, s, style);
                        }
                        x += 3;
                        d.1 = d.1 + Duration::days(1);
                    }
                    moredays |= d.0.month() == d.1.month() || d.1 < d.0;
                }
                y += 1;
                if !moredays {
                    break;
                }
            }
            startm += self.months_per_row;
            y += 1;
            if y + 8 > area.height {
                break
            } else if startm >= 12 {
                startm = 0;
                year += 1;
                let x = area.x;
                let s = format!("{year:^width$}", year = self.year as usize + year, width = area.width as usize - 4);
                let mut style = Style::default().add_modifier(Modifier::UNDERLINED);
                if self.year + year as i32 == today.year() {
                    style = style.add_modifier(Modifier::BOLD)
                }
                buf.set_string(x, y, &s, style);
                y += 1;
            }
            y += 1;
        }
    }
}