rat_widget/calendar/
single_selection.rs

1use crate::calendar::event::CalOutcome;
2use crate::calendar::{CalendarSelection, CalendarState, MonthState};
3use chrono::NaiveDate;
4use rat_event::util::item_at;
5use rat_event::{ct_event, flow, ConsumedEvent, HandleEvent, MouseOnly, Regular};
6use rat_focus::HasFocus;
7
8/// Can select a single date.
9///
10/// Movement with the arrow keys and PageUp/PageDown.
11/// Ctrl+Home moves to today.
12///
13#[derive(Debug, Default, Clone)]
14pub struct SingleSelection {
15    selected: Option<NaiveDate>,
16}
17
18impl CalendarSelection for SingleSelection {
19    fn count(&self) -> usize {
20        if self.selected.is_some() {
21            1
22        } else {
23            0
24        }
25    }
26
27    fn is_selected(&self, date: NaiveDate) -> bool {
28        self.selected == Some(date)
29    }
30
31    fn lead_selection(&self) -> Option<NaiveDate> {
32        self.selected
33    }
34}
35
36impl SingleSelection {
37    pub fn clear(&mut self) {
38        self.selected = None;
39    }
40
41    pub fn select(&mut self, date: NaiveDate) -> bool {
42        let old = self.selected;
43        self.selected = Some(date);
44        old != self.selected
45    }
46
47    pub fn selected(&self) -> Option<NaiveDate> {
48        self.selected
49    }
50}
51
52impl HandleEvent<crossterm::event::Event, Regular, CalOutcome> for MonthState<SingleSelection> {
53    fn handle(&mut self, event: &crossterm::event::Event, _qualifier: Regular) -> CalOutcome {
54        if self.is_focused() {
55            flow!(match event {
56                ct_event!(keycode press Up) => self.prev_day(7),
57                ct_event!(keycode press Down) => self.next_day(7),
58                ct_event!(keycode press Left) => self.prev_day(1),
59                ct_event!(keycode press Right) => self.next_day(1),
60                _ => CalOutcome::Continue,
61            })
62        }
63
64        self.handle(event, MouseOnly)
65    }
66}
67
68impl HandleEvent<crossterm::event::Event, MouseOnly, CalOutcome> for MonthState<SingleSelection> {
69    fn handle(&mut self, event: &crossterm::event::Event, _qualifier: MouseOnly) -> CalOutcome {
70        match event {
71            ct_event!(mouse drag Left for x, y) | ct_event!(mouse down Left for x, y) => {
72                if let Some(sel) = item_at(&self.area_days, *x, *y) {
73                    self.select_day(sel)
74                } else {
75                    CalOutcome::Continue
76                }
77            }
78
79            _ => CalOutcome::Continue,
80        }
81    }
82}
83
84impl<const N: usize> HandleEvent<crossterm::event::Event, Regular, CalOutcome>
85    for CalendarState<N, SingleSelection>
86{
87    fn handle(&mut self, event: &crossterm::event::Event, _qualifier: Regular) -> CalOutcome {
88        let mut r = 'f: {
89            for month in &mut self.months {
90                let r = month.handle(event, Regular);
91                if r.is_consumed() {
92                    //todo: change to on selected
93                    self.focus_lead();
94                    break 'f r;
95                }
96            }
97            CalOutcome::Continue
98        };
99
100        r = r.or_else(|| {
101            if self.is_focused() {
102                match event {
103                    ct_event!(keycode press CONTROL-Home) => self.move_to_today(),
104                    ct_event!(keycode press PageUp) => self.prev_month(1),
105                    ct_event!(keycode press PageDown) => self.next_month(1),
106                    ct_event!(keycode press Up) => self.prev_day(7),
107                    ct_event!(keycode press Down) => self.next_day(7),
108                    ct_event!(keycode press Left) => self.prev_day(1),
109                    ct_event!(keycode press Right) => self.next_day(1),
110
111                    _ => CalOutcome::Continue,
112                }
113            } else {
114                CalOutcome::Continue
115            }
116        });
117
118        r.or_else(|| self.handle(event, MouseOnly))
119    }
120}
121
122impl<const N: usize> HandleEvent<crossterm::event::Event, MouseOnly, CalOutcome>
123    for CalendarState<N, SingleSelection>
124{
125    fn handle(&mut self, event: &crossterm::event::Event, _qualifier: MouseOnly) -> CalOutcome {
126        for i in 0..self.months.len() {
127            if self.months[i].gained_focus() {
128                self.set_primary_idx(i);
129                break;
130            }
131        }
132
133        let all_areas = self
134            .months
135            .iter()
136            .map(|v| v.area)
137            .reduce(|v, w| v.union(w))
138            .unwrap_or_default();
139        match event {
140            ct_event!(scroll up for x,y) if all_areas.contains((*x, *y).into()) => {
141                self.scroll_back(self.step())
142            }
143            ct_event!(scroll down for x,y) if all_areas.contains((*x, *y).into()) => {
144                self.scroll_forward(self.step())
145            }
146            _ => CalOutcome::Continue,
147        }
148    }
149}
rat_widget/calendar/single_selection.rs

rat_widget/calendar/
single_selection.rs
