/* * meli - ui crate. * * Copyright 2017-2018 Manos Pitsidianakis * * This file is part of meli. * * meli is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * meli is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with meli. If not, see . */ use super::*; use crate::components::utilities::PageMovement; use std::ops::{Deref, DerefMut}; const MAX_COLS: usize = 500; struct EntryStrings { date: DateString, subject: SubjectString, flag: FlagString, from: FromString, } macro_rules! address_list { (($name:expr) as comma_sep_list) => {{ let mut ret: String = $name .into_iter() .fold(String::new(), |mut s: String, n: &Address| { s.extend(n.to_string().chars()); s.push_str(", "); s }); ret.pop(); ret.pop(); ret }}; } macro_rules! column_str { ( struct $name:ident(String)) => { pub struct $name(String); impl Deref for $name { type Target = String; fn deref(&self) -> &String { &self.0 } } impl DerefMut for $name { fn deref_mut(&mut self) -> &mut String { &mut self.0 } } }; } column_str!(struct DateString(String)); column_str!(struct FromString(String)); column_str!(struct SubjectString(String)); column_str!(struct FlagString(String)); /// A list of all mail (`Envelope`s) in a `Mailbox`. On `\n` it opens the `Envelope` content in a /// `ThreadView`. #[derive(Debug)] pub struct ConversationsListing { /// (x, y, z): x is accounts, y is folders, z is index inside a folder. cursor_pos: (usize, usize, usize), new_cursor_pos: (usize, usize, usize), length: usize, sort: (SortField, SortOrder), subsort: (SortField, SortOrder), order: FnvHashMap, /// Cache current view. content: CellBuffer, filter_term: String, filtered_selection: Vec, filtered_order: FnvHashMap, selection: FnvHashMap, /// If we must redraw on next redraw event dirty: bool, force_draw: bool, /// If `self.view` exists or not. unfocused: bool, view: ThreadView, row_updates: StackVec, movement: Option, id: ComponentId, } impl ListingTrait for ConversationsListing { fn coordinates(&self) -> (usize, usize, Option) { (self.cursor_pos.0, self.cursor_pos.1, None) } fn set_coordinates(&mut self, coordinates: (usize, usize, Option)) { self.new_cursor_pos = (coordinates.0, coordinates.1, 0); self.unfocused = false; } fn highlight_line(&mut self, grid: &mut CellBuffer, area: Area, idx: usize, context: &Context) { if self.length == 0 { return; } let i = self.get_thread_under_cursor(idx, context); let account = &context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; let thread_node = &threads.thread_nodes[&i]; let fg_color = if thread_node.has_unseen() { Color::Byte(0) } else { Color::Default }; let bg_color = if self.cursor_pos.2 == idx { Color::Byte(246) } else if self.selection[&i] { Color::Byte(210) } else if thread_node.has_unseen() { Color::Byte(251) } else { Color::Default }; copy_area( grid, &self.content, area, ((0, 3 * idx), pos_dec(self.content.size(), (1, 1))), ); let (upper_left, bottom_right) = area; let width = self.content.size().0; let (x, y) = upper_left; if self.cursor_pos.2 == idx || self.selection[&i] { for x in x..=get_x(bottom_right) { grid[(x, y)].set_fg(fg_color); grid[(x, y)].set_bg(bg_color); grid[(x, y + 1)].set_fg(fg_color); grid[(x, y + 1)].set_bg(bg_color); grid[(x, y + 2)].set_fg(Color::Byte(235)); grid[(x, y + 2)].set_bg(bg_color); } } else if width < width!(area) { /* fill any remaining columns, if our view is wider than self.content */ for x in (x + width)..=get_x(bottom_right) { grid[(x, y)].set_fg(fg_color); grid[(x, y)].set_bg(bg_color); grid[(x, y + 1)].set_fg(fg_color); grid[(x, y + 1)].set_bg(bg_color); grid[(x, y + 2)].set_fg(Color::Byte(235)); grid[(x, y + 2)].set_bg(bg_color); } } return; } /// Draw the list of `Envelope`s. fn draw_list(&mut self, grid: &mut CellBuffer, area: Area, context: &mut Context) { if self.cursor_pos.1 != self.new_cursor_pos.1 || self.cursor_pos.0 != self.new_cursor_pos.0 { self.refresh_mailbox(context); } let upper_left = upper_left!(area); let bottom_right = bottom_right!(area); if self.length == 0 { clear_area(grid, area); copy_area( grid, &self.content, area, ((0, 0), pos_dec(self.content.size(), (1, 1))), ); context.dirty_areas.push_back(area); return; } let rows = (get_y(bottom_right) - get_y(upper_left) + 1) / 3; if let Some(mvm) = self.movement.take() { match mvm { PageMovement::PageUp => { self.new_cursor_pos.2 = self.new_cursor_pos.2.saturating_sub(rows); } PageMovement::PageDown => { if self.new_cursor_pos.2 + rows + 1 < self.length { self.new_cursor_pos.2 += rows; } else if self.new_cursor_pos.2 + rows > self.length { self.new_cursor_pos.2 = self.length - 1; } else { self.new_cursor_pos.2 = (self.length / rows) * rows; } } PageMovement::Home => { self.new_cursor_pos.2 = 0; } PageMovement::End => { if self.new_cursor_pos.2 + rows > self.length { self.new_cursor_pos.2 = self.length - 1; } else { self.new_cursor_pos.2 = (self.length / rows) * rows; } } } } let prev_page_no = (self.cursor_pos.2).wrapping_div(rows); let page_no = (self.new_cursor_pos.2).wrapping_div(rows); let top_idx = page_no * rows; /* If cursor position has changed, remove the highlight from the previous position and * apply it in the new one. */ if self.cursor_pos.2 != self.new_cursor_pos.2 && prev_page_no == page_no { let old_cursor_pos = self.cursor_pos; self.cursor_pos = self.new_cursor_pos; for idx in &[old_cursor_pos.2, self.new_cursor_pos.2] { if *idx >= self.length { continue; //bounds check } let new_area = ( set_y(upper_left, get_y(upper_left) + 3 * (*idx % rows)), set_y(bottom_right, get_y(upper_left) + 3 * (*idx % rows) + 2), ); self.highlight_line(grid, new_area, *idx, context); context.dirty_areas.push_back(new_area); } return; } else if self.cursor_pos != self.new_cursor_pos { self.cursor_pos = self.new_cursor_pos; } if self.new_cursor_pos.2 >= self.length { self.new_cursor_pos.2 = self.length - 1; self.cursor_pos.2 = self.new_cursor_pos.2; } clear_area(grid, area); /* Page_no has changed, so draw new page */ copy_area( grid, &self.content, ( upper_left, set_x( bottom_right, std::cmp::min( get_x(bottom_right), get_x(upper_left) + self.content.size().0, ), ), ), ((0, 3 * top_idx), pos_dec(self.content.size(), (1, 1))), ); /* let mut flag_x = 0; let column_width = self.data_columns.columns[i].size().0; if i == 3 { flag_x = x; } copy_area( grid, &self.content, ( set_x(upper_left, x), set_x( bottom_right, std::cmp::min(get_x(bottom_right), x + (self.content)), ), ), ( (0, top_idx), (column_width.saturating_sub(1), self.length - 1), ), ); x += self.content + 2; // + SEPARATOR if x > get_x(bottom_right) { break; } for r in 0..cmp::min(self.length - top_idx, rows) { let (fg_color, bg_color) = { let c = &self.content[(0, r + top_idx)]; (c.fg(), c.bg()) }; change_colors( grid, ( pos_inc(upper_left, (0, r)), (flag_x.saturating_sub(1), get_y(upper_left) + r), ), fg_color, bg_color, ); for x in flag_x ..std::cmp::min( get_x(bottom_right), flag_x + 2 + self.content, ) { grid[(x, get_y(upper_left) + r)].set_bg(bg_color); } change_colors( grid, ( ( flag_x + 2 + self.content, get_y(upper_left) + r, ), (get_x(bottom_right), get_y(upper_left) + r), ), fg_color, bg_color, ); } */ /* TODO: highlight selected entries */ self.highlight_line( grid, ( pos_inc(upper_left, (0, 3 * (self.cursor_pos.2 % rows))), set_y( bottom_right, get_y(upper_left) + 3 * (self.cursor_pos.2 % rows) + 2, ), ), self.cursor_pos.2, context, ); /* calculate how many entries are visible in this page */ let rows = if top_idx + rows > self.length { clear_area( grid, ( pos_inc(upper_left, (0, 3 * (self.length - top_idx))), bottom_right, ), ); self.length - top_idx } else { rows }; /* fill any remaining columns, if our view is wider than self.content */ let width = self.content.size().0; if width < width!(area) { let y_offset = get_y(upper_left); for y in 0..rows { let bg_color = grid[(get_x(upper_left) + width - 1, y_offset + 3 * y)].bg(); for x in (get_x(upper_left) + width)..=get_x(bottom_right) { grid[(x, y_offset + 3 * y)].set_bg(bg_color); grid[(x, y_offset + 3 * y + 1)].set_ch('▁'); grid[(x, y_offset + 3 * y + 2)].set_fg(Color::Default); grid[(x, y_offset + 3 * y + 1)].set_bg(bg_color); grid[(x, y_offset + 3 * y + 2)].set_ch('▓'); grid[(x, y_offset + 3 * y + 2)].set_fg(Color::Byte(235)); grid[(x, y_offset + 3 * y + 2)].set_bg(bg_color); } } } context.dirty_areas.push_back(area); } fn filter(&mut self, filter_term: &str, context: &Context) { self.filtered_order.clear(); self.filtered_selection.clear(); for v in self.selection.values_mut() { *v = false; } self.filter_term.clear(); for (i, h) in self.order.keys().enumerate() { let account = &context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; let env_hash = { let thread_node = &threads.thread_nodes()[h]; if let Some(i) = thread_node.message() { i } else { let mut iter_ptr = thread_node.children()[0]; while threads.thread_nodes()[&iter_ptr].message().is_none() { iter_ptr = threads.thread_nodes()[&iter_ptr].children()[0]; } threads.thread_nodes()[&iter_ptr].message().unwrap() } }; let envelope = &account.collection[&env_hash]; if envelope.subject().contains(&filter_term) { self.filtered_selection.push(*h); self.filtered_order.insert(*h, i); self.selection.insert(*h, false); continue; } if envelope.field_from_to_string().contains(&filter_term) { self.filtered_selection.push(*h); self.filtered_order.insert(*h, i); self.selection.insert(*h, false); continue; } let op = account.operation(env_hash); let body = envelope.body(op); let decoded = decode_rec(&body, None); let body_text = String::from_utf8_lossy(&decoded); if body_text.contains(&filter_term) { self.filtered_selection.push(*h); self.filtered_order.insert(*h, i); self.selection.insert(*h, false); } } if !self.filtered_selection.is_empty() { self.filter_term = filter_term.to_string(); self.cursor_pos.2 = std::cmp::min(self.filtered_selection.len() - 1, self.cursor_pos.2); self.length = self.filtered_selection.len(); } else { self.length = 0; let message = format!("No results for `{}`.", filter_term); self.content = CellBuffer::new(message.len(), 1, Cell::with_char(' ')); write_string_to_grid( &message, &mut self.content, Color::Default, Color::Default, Attr::Default, ((0, 0), (message.len() - 1, 0)), false, ); } } } impl fmt::Display for ConversationsListing { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { write!(f, "mail") } } impl Default for ConversationsListing { fn default() -> Self { ConversationsListing::new() } } impl ConversationsListing { const DESCRIPTION: &'static str = "compact listing"; fn new() -> Self { ConversationsListing { cursor_pos: (0, 1, 0), new_cursor_pos: (0, 0, 0), length: 0, sort: (Default::default(), Default::default()), subsort: (SortField::Date, SortOrder::Desc), order: FnvHashMap::default(), filter_term: String::new(), filtered_selection: Vec::new(), filtered_order: FnvHashMap::default(), selection: FnvHashMap::default(), row_updates: StackVec::new(), content: Default::default(), dirty: true, force_draw: true, unfocused: false, view: ThreadView::default(), movement: None, id: ComponentId::new_v4(), } } fn make_entry_string(e: &Envelope, len: usize, is_snoozed: bool) -> EntryStrings { if len > 0 { EntryStrings { date: DateString(ConversationsListing::format_date(e)), subject: SubjectString(format!("{} ({})", e.subject(), len,)), flag: FlagString(format!( "{}{}", if e.has_attachments() { "📎" } else { "" }, if is_snoozed { "💤" } else { "" } )), from: FromString(address_list!((e.from()) as comma_sep_list)), } } else { EntryStrings { date: DateString(ConversationsListing::format_date(e)), subject: SubjectString(e.subject().to_string()), flag: FlagString(format!( "{}{}", if e.has_attachments() { "📎" } else { "" }, if is_snoozed { "💤" } else { "" } )), from: FromString(address_list!((e.from()) as comma_sep_list)), } } } /// Fill the `self.data_columns` `CellBuffers` with the contents of the account folder the user has /// chosen. fn refresh_mailbox(&mut self, context: &mut Context) { self.dirty = true; let old_cursor_pos = self.cursor_pos; if !(self.cursor_pos.0 == self.new_cursor_pos.0 && self.cursor_pos.1 == self.new_cursor_pos.1) { self.cursor_pos.2 = 0; self.new_cursor_pos.2 = 0; } self.cursor_pos.1 = self.new_cursor_pos.1; self.cursor_pos.0 = self.new_cursor_pos.0; let folder_hash = if let Some(h) = context.accounts[self.cursor_pos.0] .folders_order .get(self.cursor_pos.1) { *h } else { self.cursor_pos.1 = old_cursor_pos.1; self.dirty = false; return; }; context .replies .push_back(UIEvent::RefreshMailbox((self.cursor_pos.0, folder_hash))); // Get mailbox as a reference. // match context.accounts[self.cursor_pos.0].status(folder_hash) { Ok(()) => {} Err(_) => { let message: String = context.accounts[self.cursor_pos.0][folder_hash].to_string(); self.content = CellBuffer::new(message.len(), 1, Cell::with_char(' ')); self.length = 0; write_string_to_grid( message.as_str(), &mut self.content, Color::Default, Color::Default, Attr::Default, ((0, 0), (message.len() - 1, 0)), false, ); return; } } if old_cursor_pos == self.new_cursor_pos { self.view.update(context); } else if self.unfocused { self.view = ThreadView::new(self.new_cursor_pos, None, context); } let account = &context.accounts[self.cursor_pos.0]; let mailbox = &account[self.cursor_pos.1].unwrap(); let threads = &account.collection.threads[&mailbox.folder.hash()]; self.order.clear(); self.selection.clear(); self.length = 0; let mut rows = Vec::with_capacity(1024); let mut max_entry_columns = 0; threads.sort_by(self.sort, self.subsort, &account.collection); for (idx, root_idx) in threads.root_iter().enumerate() { self.length += 1; let thread_node = &threads.thread_nodes()[&root_idx]; let i = if let Some(i) = thread_node.message() { i } else { let mut iter_ptr = thread_node.children()[0]; while threads.thread_nodes()[&iter_ptr].message().is_none() { iter_ptr = threads.thread_nodes()[&iter_ptr].children()[0]; } threads.thread_nodes()[&iter_ptr].message().unwrap() }; if !context.accounts[self.cursor_pos.0].contains_key(i) { debug!("key = {}", i); debug!( "name = {} {}", mailbox.name(), context.accounts[self.cursor_pos.0].name() ); debug!("{:#?}", context.accounts); panic!(); } let root_envelope: &Envelope = &context.accounts[self.cursor_pos.0].get_env(&i); let strings = ConversationsListing::make_entry_string( root_envelope, thread_node.len(), threads.is_snoozed(root_idx), ); max_entry_columns = std::cmp::max( max_entry_columns, strings.flag.len() + 3 + strings.subject.len(), ); rows.push(strings); self.order.insert(root_idx, idx); self.selection.insert(root_idx, false); } let ConversationsListing { ref mut selection, ref order, .. } = self; selection.retain(|e, _| order.contains_key(e)); let width = std::cmp::min(MAX_COLS, max_entry_columns); self.content = CellBuffer::new(width, 4 * rows.len(), Cell::with_char(' ')); for ((idx, root_idx), strings) in threads.root_iter().enumerate().zip(rows) { let thread_node = &threads.thread_nodes()[&root_idx]; let i = if let Some(i) = thread_node.message() { i } else { let mut iter_ptr = thread_node.children()[0]; while threads.thread_nodes()[&iter_ptr].message().is_none() { iter_ptr = threads.thread_nodes()[&iter_ptr].children()[0]; } threads.thread_nodes()[&iter_ptr].message().unwrap() }; if !context.accounts[self.cursor_pos.0].contains_key(i) { panic!(); } let fg_color = if thread_node.has_unseen() { Color::Byte(0) } else { Color::Default }; let bg_color = if thread_node.has_unseen() { Color::Byte(251) } else { Color::Default }; /* draw flags */ let (x, _) = write_string_to_grid( &strings.flag, &mut self.content, fg_color, bg_color, Attr::Default, ((0, 3 * idx), (width - 1, 3 * idx)), false, ); for x in x..(x + 3) { self.content[(x, 3 * idx)].set_bg(bg_color); } /* draw subject */ let (x, _) = write_string_to_grid( &strings.subject, &mut self.content, fg_color, bg_color, Attr::Bold, ((x, 3 * idx), (width - 1, 3 * idx)), false, ); for x in x..width { self.content[(x, 3 * idx)].set_bg(bg_color); } /* Next line, draw date */ let (x, _) = write_string_to_grid( &strings.date, &mut self.content, fg_color, bg_color, Attr::Default, ((0, 3 * idx + 1), (width - 1, 3 * idx + 1)), false, ); for x in x..(x + 4) { self.content[(x, 3 * idx + 1)].set_ch('▁'); self.content[(x, 3 * idx + 1)].set_bg(bg_color); } /* draw from */ let (x, _) = write_string_to_grid( &strings.from, &mut self.content, fg_color, bg_color, Attr::Default, ((x + 4, 3 * idx + 1), (width - 1, 3 * idx + 1)), false, ); for x in x..width { self.content[(x, 3 * idx + 1)].set_ch('▁'); self.content[(x, 3 * idx + 1)].set_bg(bg_color); } for x in 0..width { self.content[(x, 3 * idx + 2)].set_ch('▓'); self.content[(x, 3 * idx + 2)].set_fg(Color::Byte(235)); self.content[(x, 3 * idx + 2)].set_bg(bg_color); } } if self.length == 0 { let message = format!("Folder `{}` is empty.", mailbox.folder.name()); self.content = CellBuffer::new(message.len(), 1, Cell::with_char(' ')); write_string_to_grid( &message, &mut self.content, Color::Default, Color::Default, Attr::Default, ((0, 0), (message.len() - 1, 0)), false, ); return; } } fn format_date(envelope: &Envelope) -> String { let d = std::time::UNIX_EPOCH + std::time::Duration::from_secs(envelope.date()); let now: std::time::Duration = std::time::SystemTime::now() .duration_since(d) .unwrap_or_else(|_| std::time::Duration::new(std::u64::MAX, 0)); match now.as_secs() { n if n < 10 * 60 * 60 => format!("{} hours ago", n / (60 * 60)), n if n < 24 * 60 * 60 => format!("{} hours ago", n / (60 * 60)), n if n < 4 * 24 * 60 * 60 => format!("{} days ago", n / (24 * 60 * 60),), _ => envelope.datetime().format("%Y-%m-%d %H:%M:%S").to_string(), } } fn draw_filtered_selection(&mut self, _context: &mut Context) { /* if self.filtered_selection.is_empty() { return; } let account = &context.accounts[self.cursor_pos.0]; let mailbox = if account[self.cursor_pos.1].is_available() { account[self.cursor_pos.1].unwrap() } else { return; }; let threads = &account.collection.threads[&mailbox.folder.hash()]; self.length = 0; let mut rows = Vec::with_capacity(1024); let mut min_width = (0, 0, 0, 0, 0); for (idx, envelope_hash) in self.filtered_selection.iter().enumerate() { self.length += 1; let envelope: &Envelope = &context.accounts[self.cursor_pos.0].get_env(&envelope_hash); let t_idx = envelope.thread(); let strings = ConversationsListing::make_entry_string( envelope, threads[&envelope.thread()].len(), idx, threads.is_snoozed(t_idx), ); min_width.0 = cmp::max(min_width.0, strings.0.grapheme_width()); /* index */ min_width.1 = cmp::max(min_width.1, strings.1.grapheme_width()); /* date */ min_width.2 = cmp::max(min_width.2, strings.2.grapheme_width()); /* from */ min_width.3 = cmp::max(min_width.3, strings.3.grapheme_width()); /* flags */ min_width.4 = cmp::max(min_width.4, strings.4.grapheme_width()); /* subject */ rows.push(strings); } /* index column */ self.data_columns.columns[0] = CellBuffer::new(min_width.0, rows.len(), Cell::with_char(' ')); /* date column */ self.data_columns.columns[1] = CellBuffer::new(min_width.1, rows.len(), Cell::with_char(' ')); /* from column */ self.data_columns.columns[2] = CellBuffer::new(min_width.2, rows.len(), Cell::with_char(' ')); /* flags column */ self.data_columns.columns[3] = CellBuffer::new(min_width.3, rows.len(), Cell::with_char(' ')); /* subject column */ self.data_columns.columns[4] = CellBuffer::new(min_width.4, rows.len(), Cell::with_char(' ')); for ((idx, envelope_hash), strings) in self.filtered_selection.iter().enumerate().zip(rows) { let envelope: &Envelope = &context.accounts[self.cursor_pos.0].get_env(&envelope_hash); let fg_color = if !envelope.is_seen() { Color::Byte(0) } else { Color::Default }; let bg_color = if self.selection[&envelope_hash] { Color::Byte(210) } else if !envelope.is_seen() { Color::Byte(251) } else { Color::Default }; let (x, _) = write_string_to_grid( &strings.0, &mut self.data_columns.columns[0], fg_color, bg_color, Attr::Default, ((0, idx), (min_width.0, idx)), false, ); for x in x..min_width.0 { self.data_columns.columns[0][(x, idx)].set_bg(bg_color); } let (x, _) = write_string_to_grid( &strings.1, &mut self.data_columns.columns[1], fg_color, bg_color, Attr::Default, ((0, idx), (min_width.1, idx)), false, ); for x in x..min_width.1 { self.data_columns.columns[1][(x, idx)].set_bg(bg_color); } let (x, _) = write_string_to_grid( &strings.2, &mut self.data_columns.columns[2], fg_color, bg_color, Attr::Default, ((0, idx), (min_width.2, idx)), false, ); for x in x..min_width.2 { self.data_columns.columns[2][(x, idx)].set_bg(bg_color); } let (x, _) = write_string_to_grid( &strings.3, &mut self.data_columns.columns[3], fg_color, bg_color, Attr::Default, ((0, idx), (min_width.3, idx)), false, ); for x in x..min_width.3 { self.data_columns.columns[3][(x, idx)].set_bg(bg_color); } let (x, _) = write_string_to_grid( &strings.4, &mut self.data_columns.columns[4], fg_color, bg_color, Attr::Default, ((0, idx), (min_width.4, idx)), false, ); for x in x..min_width.4 { self.data_columns.columns[4][(x, idx)].set_bg(bg_color); } match ( threads.is_snoozed(envelope.thread()), &context.accounts[self.cursor_pos.0] .get_env(&envelope_hash) .has_attachments(), ) { (true, true) => { self.data_columns.columns[3][(0, idx)].set_fg(Color::Byte(103)); self.data_columns.columns[3][(2, idx)].set_fg(Color::Red); } (true, false) => { self.data_columns.columns[3][(0, idx)].set_fg(Color::Red); } (false, true) => { self.data_columns.columns[3][(0, idx)].set_fg(Color::Byte(103)); } (false, false) => {} } } */ } fn get_thread_under_cursor(&self, cursor: usize, context: &Context) -> ThreadHash { let account = &context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; if self.filtered_selection.is_empty() { threads.root_set(cursor) } else { self.filtered_selection[cursor] } } } impl Component for ConversationsListing { fn draw(&mut self, grid: &mut CellBuffer, area: Area, context: &mut Context) { if !self.is_dirty() { return; } let (upper_left, bottom_right) = area; { let area = if self.unfocused { clear_area( grid, ( pos_inc(upper_left, (width!(area) / 3, 0)), set_x(bottom_right, get_x(upper_left) + width!(area) / 3 + 1), ), ); context.dirty_areas.push_back(( pos_inc(upper_left, (width!(area) / 3, 0)), set_x(bottom_right, get_x(upper_left) + width!(area) / 3 + 1), )); ( upper_left, set_x(bottom_right, get_x(upper_left) + width!(area) / 3 - 1), ) } else { area }; if !self.filtered_selection.is_empty() { self.draw_filtered_selection(context); let (x, y) = write_string_to_grid( &format!("Filter (Press ESC to exit): {}", self.filter_term), grid, Color::Default, Color::Default, Attr::Default, area, true, ); clear_area(grid, ((x, y), set_y(bottom_right, y))); context .dirty_areas .push_back((upper_left, set_y(bottom_right, y + 1))); self.draw_list(grid, (set_y(upper_left, y + 1), bottom_right), context); self.dirty = false; return; } if !self.row_updates.is_empty() { /* certain rows need to be updated (eg an unseen message was just set seen) * */ let (upper_left, bottom_right) = area; let width = self.content.size().0; while let Some(row) = self.row_updates.pop() { let row: usize = self.order[&row]; let i = self.get_thread_under_cursor(row, context); let account = &context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; let thread_node = &threads.thread_nodes[&i]; let fg_color = if thread_node.has_unseen() { Color::Byte(0) } else { Color::Default }; let bg_color = if thread_node.has_unseen() { Color::Byte(251) } else { Color::Default }; change_colors( &mut self.content, ((0, 3 * row), (width - 1, 3 * row + 1)), fg_color, bg_color, ); change_colors( &mut self.content, ((0, 3 * row + 2), (width - 1, 3 * row + 2)), Color::Byte(235), bg_color, ); let rows = (get_y(bottom_right) - get_y(upper_left) + 1) / 3; let page_no = (self.cursor_pos.2).wrapping_div(rows); let top_idx = page_no * rows; /* Update row only if it's currently visible */ if row >= top_idx && row <= top_idx + rows { let area = ( set_y(upper_left, get_y(upper_left) + (3 * (row % rows))), set_y(bottom_right, get_y(upper_left) + (3 * (row % rows) + 2)), ); self.highlight_line(grid, area, row, context); context.dirty_areas.push_back(area); } } if self.force_draw { /* Draw the entire list */ self.draw_list(grid, area, context); self.force_draw = false; } } else { /* Draw the entire list */ self.draw_list(grid, area, context); } } if self.unfocused { if self.length == 0 && self.dirty { clear_area(grid, area); context.dirty_areas.push_back(area); return; } let area = ( set_x(upper_left, get_x(upper_left) + width!(area) / 3 + 2), bottom_right, ); self.view.draw(grid, area, context); } self.dirty = false; } fn process_event(&mut self, event: &mut UIEvent, context: &mut Context) -> bool { if self.unfocused && self.view.process_event(event, context) { return true; } let shortcuts = &self.get_shortcuts(context)[ConversationsListing::DESCRIPTION]; match *event { UIEvent::Input(Key::Up) => { if self.cursor_pos.2 > 0 { self.new_cursor_pos.2 = self.new_cursor_pos.2.saturating_sub(1); self.dirty = true; } return true; } UIEvent::Input(Key::Down) => { if self.length > 0 && self.new_cursor_pos.2 < self.length - 1 { self.new_cursor_pos.2 += 1; self.dirty = true; } return true; } UIEvent::Input(ref k) if !self.unfocused && *k == shortcuts["open_thread"] => { if self.filtered_selection.is_empty() { self.view = ThreadView::new(self.cursor_pos, None, context); } else { let mut temp = self.cursor_pos; let thread_hash = self.get_thread_under_cursor(self.cursor_pos.2, context); let account = &context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; let root_thread_index = threads.root_iter().position(|t| t == thread_hash); if let Some(pos) = root_thread_index { temp.2 = pos; self.view = ThreadView::new(temp, Some(thread_hash), context); } else { return true; } } self.unfocused = true; self.dirty = true; return true; } UIEvent::Input(ref key) if *key == shortcuts["prev_page"] => { self.movement = Some(PageMovement::PageUp); self.set_dirty(); } UIEvent::Input(ref key) if *key == shortcuts["next_page"] => { self.movement = Some(PageMovement::PageDown); self.set_dirty(); } UIEvent::Input(ref key) if *key == Key::Home => { self.movement = Some(PageMovement::Home); self.set_dirty(); } UIEvent::Input(ref key) if *key == Key::End => { self.movement = Some(PageMovement::End); self.set_dirty(); } UIEvent::Input(ref k) if self.unfocused && *k == shortcuts["exit_thread"] => { self.unfocused = false; self.dirty = true; /* If self.row_updates is not empty and we exit a thread, the row_update events * will be performed but the list will not be drawn. So force a draw in any case. * */ self.force_draw = true; return true; } UIEvent::Input(ref key) if !self.unfocused && *key == shortcuts["select_entry"] => { let thread_hash = self.get_thread_under_cursor(self.cursor_pos.2, context); self.selection.entry(thread_hash).and_modify(|e| *e = !*e); } UIEvent::MailboxUpdate((ref idxa, ref idxf)) if (*idxa, *idxf) == ( self.new_cursor_pos.0, context.accounts[self.new_cursor_pos.0].folders_order [self.new_cursor_pos.1], ) => { self.refresh_mailbox(context); self.set_dirty(); } UIEvent::StartupCheck(ref f) if *f == context.accounts[self.cursor_pos.0].folders_order[self.new_cursor_pos.1] => { self.refresh_mailbox(context); self.set_dirty(); } UIEvent::EnvelopeRename(ref old_hash, ref new_hash) => { let account = &context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; if !account.collection.envelopes.contains_key(&new_hash) { return false; } let new_env_thread_hash = account.get_env(new_hash).thread(); if !threads.thread_nodes.contains_key(&new_env_thread_hash) { return false; } let thread_group = threads.thread_nodes[&new_env_thread_hash].thread_group(); let (&thread_hash, _): (&ThreadHash, &usize) = self .order .iter() .find(|(n, _)| threads.thread_nodes[&n].thread_group() == thread_group) .unwrap(); self.row_updates.push(thread_hash); self.dirty = true; self.view .process_event(&mut UIEvent::EnvelopeRename(*old_hash, *new_hash), context); } UIEvent::ChangeMode(UIMode::Normal) => { self.dirty = true; } UIEvent::Resize => { self.dirty = true; } UIEvent::Action(ref action) => match action { Action::ViewMailbox(idx) => { if context.accounts[self.cursor_pos.0] .folders_order .get(*idx) .is_none() { return true; } self.filtered_selection.clear(); self.new_cursor_pos.1 = *idx; self.refresh_mailbox(context); return true; } Action::SubSort(field, order) if !self.unfocused => { debug!("SubSort {:?} , {:?}", field, order); self.subsort = (*field, *order); self.refresh_mailbox(context); return true; } Action::Sort(field, order) if !self.unfocused => { debug!("Sort {:?} , {:?}", field, order); self.sort = (*field, *order); self.refresh_mailbox(context); return true; } Action::ToggleThreadSnooze if !self.unfocused => { let thread_hash = self.get_thread_under_cursor(self.cursor_pos.2, context); let account = &mut context.accounts[self.cursor_pos.0]; let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = account.collection.threads.entry(folder_hash).or_default(); let root_node = threads.thread_nodes.entry(thread_hash).or_default(); let is_snoozed = root_node.snoozed(); root_node.set_snoozed(!is_snoozed); self.row_updates.push(thread_hash); self.refresh_mailbox(context); return true; } Action::Listing(Filter(ref filter_term)) if !self.unfocused => { self.filter(filter_term, context); self.dirty = true; } Action::Listing(a @ ListingAction::SetSeen) | Action::Listing(a @ ListingAction::SetUnseen) | Action::Listing(a @ ListingAction::Delete) if !self.unfocused => { /* Iterate over selection if exists, else only over the envelope under the * cursor. Using two iterators allows chaining them which results into a Chain * type. We can't conditonally select either a slice iterator or a Map iterator * because of the type system */ let is_selection_empty = self.selection.values().cloned().any(std::convert::identity); let i = [self.get_thread_under_cursor(self.cursor_pos.2, context)]; let cursor_iter; let sel_iter = if is_selection_empty { cursor_iter = None; Some(self.selection.iter().filter(|(_, v)| **v).map(|(k, _)| k)) } else { cursor_iter = Some(i.iter()); None }; let iter = sel_iter .into_iter() .flatten() .chain(cursor_iter.into_iter().flatten()); for &i in iter { let account = &mut context.accounts[self.cursor_pos.0]; let mut envs_to_set: StackVec = StackVec::new(); { let folder_hash = account[self.cursor_pos.1].unwrap().folder.hash(); let threads = &account.collection.threads[&folder_hash]; let mut stack = StackVec::new(); stack.push(i); while let Some(thread_iter) = stack.pop() { if let Some(env_hash) = threads[&thread_iter].message() { if !account.contains_key(env_hash) { /* The envelope has been renamed or removed, so wait for the appropriate event to * arrive */ continue; } envs_to_set.push(env_hash); } for c in 0..threads[&thread_iter].children().len() { let c = threads[&thread_iter].children()[c]; stack.push(c); } } } for env_hash in envs_to_set { match a { ListingAction::SetSeen => { let hash = account.get_env(&env_hash).hash(); let op = account.operation(hash); let envelope: &mut Envelope = &mut account.get_env_mut(&env_hash); if let Err(e) = envelope.set_seen(op) { context.replies.push_back(UIEvent::StatusEvent( StatusEvent::DisplayMessage(e.to_string()), )); } self.row_updates.push(i); } ListingAction::SetUnseen => { let hash = account.get_env(&env_hash).hash(); let op = account.operation(hash); let envelope: &mut Envelope = &mut account.get_env_mut(&env_hash); if let Err(e) = envelope.set_unseen(op) { context.replies.push_back(UIEvent::StatusEvent( StatusEvent::DisplayMessage(e.to_string()), )); } self.row_updates.push(i); } ListingAction::Delete => { /* do nothing */ } _ => unreachable!(), } } } self.dirty = true; for v in self.selection.values_mut() { *v = false; } return true; } _ => {} }, UIEvent::Input(Key::Esc) if !self.unfocused && !self.filtered_selection.is_empty() => { self.filter_term.clear(); self.filtered_selection.clear(); for v in self.selection.values_mut() { *v = false; } self.filtered_order.clear(); self.refresh_mailbox(context); return true; } _ => {} } false } fn is_dirty(&self) -> bool { self.dirty || if self.unfocused { self.view.is_dirty() } else { false } } fn set_dirty(&mut self) { if self.unfocused { self.view.set_dirty(); } self.dirty = true; } fn get_shortcuts(&self, context: &Context) -> ShortcutMaps { let mut map = if self.unfocused { self.view.get_shortcuts(context) } else { ShortcutMaps::default() }; let config_map = context.settings.shortcuts.compact_listing.key_values(); map.insert( ConversationsListing::DESCRIPTION.to_string(), [ ( "open_thread", if let Some(key) = config_map.get("open_thread") { (*key).clone() } else { Key::Char('\n') }, ), ( "prev_page", if let Some(key) = config_map.get("prev_page") { (*key).clone() } else { Key::PageUp }, ), ( "next_page", if let Some(key) = config_map.get("next_page") { (*key).clone() } else { Key::PageDown }, ), ( "exit_thread", if let Some(key) = config_map.get("exit_thread") { (*key).clone() } else { Key::Char('i') }, ), ( "select_entry", if let Some(key) = config_map.get("select_entry") { (*key).clone() } else { Key::Char('v') }, ), ] .iter() .cloned() .collect(), ); map } fn id(&self) -> ComponentId { self.id } fn set_id(&mut self, id: ComponentId) { self.id = id; } }