mirror of
https://github.com/kdheepak/taskwarrior-tui.git
synced 2025-08-25 17:57:19 +02:00
Update Calendar widget
This commit is contained in:
Dheepak Krishnamurthy
parent
86e23671e8
commit
f77b7c71a6
1 changed files with 69 additions and 247 deletions
316
src/calendar.rs
316
src/calendar.rs
|
|
@ -5,10 +5,7 @@ use std::fmt;
|
|||
|
||||
const COL_WIDTH: usize = 21;
|
||||
|
||||
use Day::*;
|
||||
use Month::*;
|
||||
|
||||
use chrono::{Datelike, Local, Duration, NaiveDate, NaiveDateTime, TimeZone};
|
||||
use chrono::{Datelike, Local, Duration, NaiveDate, NaiveDateTime, TimeZone, Month};
|
||||
|
||||
use tui::{
|
||||
buffer::Buffer,
|
||||
|
|
@ -21,137 +18,13 @@ use tui::{
|
|||
use std::cmp::min;
|
||||
use unicode_width::UnicodeWidthStr;
|
||||
|
||||
#[derive(Clone, Copy, Debug)]
|
||||
#[repr(u8)]
|
||||
pub enum Day {
|
||||
Sun,
|
||||
Mon,
|
||||
Tue,
|
||||
Wed,
|
||||
Thu,
|
||||
Fri,
|
||||
Sat,
|
||||
}
|
||||
|
||||
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
|
||||
pub enum Month {
|
||||
Jan,
|
||||
Feb,
|
||||
Mar,
|
||||
Apr,
|
||||
May,
|
||||
Jun,
|
||||
Jul,
|
||||
Aug,
|
||||
Sep,
|
||||
Oct,
|
||||
Nov,
|
||||
Dec,
|
||||
}
|
||||
impl Month {
|
||||
fn len(self) -> u8 {
|
||||
match self {
|
||||
Jan => 31,
|
||||
Feb => 28,
|
||||
Mar => 31,
|
||||
Apr => 30,
|
||||
May => 31,
|
||||
Jun => 30,
|
||||
Jul => 31,
|
||||
Aug => 31,
|
||||
Sep => 30,
|
||||
Oct => 31,
|
||||
Nov => 30,
|
||||
Dec => 31,
|
||||
}
|
||||
}
|
||||
fn leap_len(self, leap_year: bool) -> u8 {
|
||||
match self {
|
||||
Feb => {
|
||||
if leap_year {
|
||||
29
|
||||
} else {
|
||||
28
|
||||
}
|
||||
}
|
||||
mon => mon.len(),
|
||||
}
|
||||
}
|
||||
fn first_day(self, year: i32) -> Day {
|
||||
let y = year - 1;
|
||||
let jan_first = (1 + (5 * (y % 4)) + (4 * (y % 100)) + (6 * (y % 400))) % 7;
|
||||
let mut len = 0;
|
||||
for m in Jan {
|
||||
if m == self {
|
||||
break;
|
||||
}
|
||||
len += m.leap_len(is_leap_year(year)) as i32;
|
||||
}
|
||||
match (len + jan_first) % 7 {
|
||||
0 => Sun,
|
||||
1 => Mon,
|
||||
2 => Tue,
|
||||
3 => Wed,
|
||||
4 => Thu,
|
||||
5 => Fri,
|
||||
_ => Sat,
|
||||
}
|
||||
}
|
||||
}
|
||||
impl Iterator for Month {
|
||||
type Item = Month;
|
||||
fn next(&mut self) -> Option<Month> {
|
||||
let ret = Some(*self);
|
||||
*self = match *self {
|
||||
Jan => Feb,
|
||||
Feb => Mar,
|
||||
Mar => Apr,
|
||||
Apr => May,
|
||||
May => Jun,
|
||||
Jun => Jul,
|
||||
Jul => Aug,
|
||||
Aug => Sep,
|
||||
Sep => Oct,
|
||||
Oct => Nov,
|
||||
Nov => Dec,
|
||||
Dec => Jan,
|
||||
};
|
||||
ret
|
||||
}
|
||||
}
|
||||
impl fmt::Display for Month {
|
||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||
let name = match *self {
|
||||
Jan => "January",
|
||||
Feb => "February",
|
||||
Mar => "March",
|
||||
Apr => "April",
|
||||
May => "May",
|
||||
Jun => "June",
|
||||
Jul => "July",
|
||||
Aug => "August",
|
||||
Sep => "September",
|
||||
Oct => "October",
|
||||
Nov => "November",
|
||||
Dec => "December",
|
||||
};
|
||||
let padding = COL_WIDTH - name.len();
|
||||
write!(f, "{:1$}", "", padding / 2)?;
|
||||
if padding % 2 != 0 {
|
||||
f.write_str(" ")?;
|
||||
}
|
||||
f.write_str(name)?;
|
||||
write!(f, "{:1$}", "", padding / 2)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Calendar<'a> {
|
||||
pub block: Option<Block<'a>>,
|
||||
pub year: i32,
|
||||
pub days: Vec<(NaiveDate, NaiveDate)>,
|
||||
pub style: Style,
|
||||
pub months_per_row: i8,
|
||||
pub months_per_row: usize,
|
||||
}
|
||||
|
||||
impl<'a> Default for Calendar<'a> {
|
||||
|
|
@ -193,12 +66,22 @@ impl<'a> Calendar<'a> {
|
|||
}
|
||||
}
|
||||
|
||||
fn is_leap_year(year: i32) -> bool {
|
||||
(year % 4 == 0 && year % 100 != 0) || year % 400 == 0
|
||||
}
|
||||
|
||||
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() {
|
||||
|
|
@ -216,135 +99,74 @@ impl <'a> Widget for Calendar<'a> {
|
|||
|
||||
let style = self.style;
|
||||
let today = Local::today();
|
||||
let cols = area.width;
|
||||
|
||||
let year = self.year;
|
||||
let leap_year = is_leap_year(year);
|
||||
let months = [Jan, Feb, Mar, Apr, May, Jun, Jul, Aug, Sep, Oct, Nov, Dec];
|
||||
let mut dates = [
|
||||
0..Jan.len(),
|
||||
0..Feb.leap_len(leap_year),
|
||||
0..Mar.len(),
|
||||
0..Apr.len(),
|
||||
0..May.len(),
|
||||
0..Jun.len(),
|
||||
0..Jul.len(),
|
||||
0..Aug.len(),
|
||||
0..Sep.len(),
|
||||
0..Oct.len(),
|
||||
0..Nov.len(),
|
||||
0..Dec.len(),
|
||||
];
|
||||
if self.months_per_row > cols as i8 / 8 / 3 || self.months_per_row <= 0 {
|
||||
self.months_per_row = cols as i8 / 8 / 3;
|
||||
let months: Vec<_> = (0..12).collect();
|
||||
let mut startm = 0;
|
||||
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 chunks = dates.chunks_mut(self.months_per_row as usize).zip(months.chunks(self.months_per_row as usize));
|
||||
let mut y = area.y;
|
||||
for (days_chunk, months) in chunks {
|
||||
let mut x = area.x;
|
||||
for month in months {
|
||||
buf.set_string(x as u16, y, month.to_string(), Style::default().add_modifier(Modifier::DIM));
|
||||
x = x + month.to_string().len() as u16 + 2;
|
||||
y += 1;
|
||||
let x = area.x;
|
||||
let s = format!("{year:^width$}", year = year, width = area.width as usize - 4);
|
||||
buf.set_string(x, y, &s, Style::default().add_modifier(Modifier::BOLD));
|
||||
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 self.days[c];
|
||||
let m = d.0.month() as usize;
|
||||
let s = format!("{:^21}", month_names[m - 1]);
|
||||
buf.set_string(x, y, &s, Style::default());
|
||||
x += s.len() as u16;
|
||||
}
|
||||
y += 1;
|
||||
let mut x = area.x;
|
||||
for month in months {
|
||||
let mut x = area.x + startx;
|
||||
for c in startm..endm {
|
||||
buf.set_string(x as u16, y, " Su Mo Tu We Th Fr Sa", Style::default());
|
||||
x += 21 + 1;
|
||||
}
|
||||
y += 1;
|
||||
let mut x = area.x;
|
||||
for (days, mon) in days_chunk.iter_mut().zip(months.iter()) {
|
||||
let first_day = mon.first_day(year) as u8;
|
||||
for _ in 0..(first_day) {
|
||||
buf.set_string(x as u16, y, "", Style::default());
|
||||
x += 3;
|
||||
}
|
||||
for _ in 0..(7 - first_day) {
|
||||
buf.set_string(x as u16, y, format!("{:>3}", days.next().unwrap() + 1), Style::default());
|
||||
x += 3;
|
||||
}
|
||||
buf.set_string(x as u16, y, " ", Style::default());
|
||||
x += 1;
|
||||
}
|
||||
y += 1;
|
||||
while !days_chunk.iter().all(|r| r.start == r.end) {
|
||||
let mut x = area.x;
|
||||
for days in days_chunk.iter_mut() {
|
||||
for _ in 0..7 {
|
||||
match days.next() {
|
||||
Some(s) => {
|
||||
buf.set_string(x as u16, y, format!("{:>3}", s + 1), Style::default());
|
||||
x += 3;
|
||||
},
|
||||
None => x += 3,
|
||||
}
|
||||
loop {
|
||||
let mut moredays = false;
|
||||
let mut x = area.x + startx;
|
||||
for c in startm..endm {
|
||||
if c > startm {
|
||||
x += 1;
|
||||
}
|
||||
x += 1;
|
||||
let d = &mut self.days[c];
|
||||
for _ in 0..7 {
|
||||
let s = if d.0.month() == d.1.month() {
|
||||
format!("{:>2}", d.1.day())
|
||||
} else {
|
||||
" ".to_string()
|
||||
};
|
||||
if d.1 == Local::today().naive_local() {
|
||||
buf.set_string(x, y, s, Style::default().add_modifier(Modifier::REVERSED));
|
||||
} else {
|
||||
buf.set_string(x, y, s, Style::default());
|
||||
}
|
||||
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;
|
||||
if startm >= 12 {
|
||||
return;
|
||||
}
|
||||
y += 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> fmt::Display for Calendar<'a> {
|
||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||
let today = Local::today();
|
||||
let cols = f.width().unwrap_or(3);
|
||||
let year = self.year;
|
||||
let leap_year = is_leap_year(year);
|
||||
let months = [Jan, Feb, Mar, Apr, May, Jun, Jul, Aug, Sep, Oct, Nov, Dec];
|
||||
let mut dates = [
|
||||
0..Jan.len(),
|
||||
0..Feb.leap_len(leap_year),
|
||||
0..Mar.len(),
|
||||
0..Apr.len(),
|
||||
0..May.len(),
|
||||
0..Jun.len(),
|
||||
0..Jul.len(),
|
||||
0..Aug.len(),
|
||||
0..Sep.len(),
|
||||
0..Oct.len(),
|
||||
0..Nov.len(),
|
||||
0..Dec.len(),
|
||||
];
|
||||
let chunks = dates.chunks_mut(cols).zip(months.chunks(cols));
|
||||
for (days_chunk, months) in chunks {
|
||||
for month in months {
|
||||
write!(f, "{:>1$} ", month, COL_WIDTH)?;
|
||||
}
|
||||
f.write_str("\n")?;
|
||||
for month in months {
|
||||
write!(f, "{:>1$} ", " S M T W T F S", COL_WIDTH)?;
|
||||
}
|
||||
f.write_str("\n")?;
|
||||
for (days, mon) in days_chunk.iter_mut().zip(months.iter()) {
|
||||
let first_day = mon.first_day(year) as u8;
|
||||
for _ in 0..(first_day) {
|
||||
f.write_str(" ")?;
|
||||
}
|
||||
for _ in 0..(7 - first_day) {
|
||||
write!(f, "{:>3}", days.next().unwrap() + 1)?;
|
||||
}
|
||||
f.write_str(" ")?;
|
||||
}
|
||||
f.write_str("\n")?;
|
||||
while !days_chunk.iter().all(|r| r.start == r.end) {
|
||||
for days in days_chunk.iter_mut() {
|
||||
for _ in 0..7 {
|
||||
match days.next() {
|
||||
Some(s) => write!(f, "{:>3}", s + 1)?,
|
||||
None => f.write_str(" ")?,
|
||||
}
|
||||
}
|
||||
f.write_str(" ")?;
|
||||
}
|
||||
f.write_str("\n")?;
|
||||
}
|
||||
f.write_str("\n")?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue