use anyhow::Result; use cursive::Cursive; use cursive::Printer; use cursive::Rect; use cursive::View; use cursive::XY; use cursive::direction::Direction; use cursive::event::Callback; use cursive::event::Event; use cursive::event::EventResult; use cursive::view::Nameable; use cursive::view::Selector; use cursive::views::Dialog; use cursive::views::EditView; use cursive::views::NamedView; use cursive::views::ResizedView; use cursive::traits::Resizable; use cursive_multiplex::Mux; use crate::configuration::Configuration; use crate::maillist_view::MaillistView; use crate::maillist_view::MailListingData; pub const MAIN_VIEW_NAME: &'static str = "main_view"; pub const MAIN_MUX_NAME: &'static str = "main_mux"; pub const MAIN_MAIL_LIST_NAME: &'static str = "main_mail_list"; pub struct MainView { config: Configuration, muxroot: cursive_multiplex::Id, tabs: cursive_tabs::TabPanel, } impl View for MainView { fn draw(&self, printer: &Printer) { self.tabs.draw(printer) } fn layout(&mut self, xy: XY) { self.tabs.layout(xy) } fn needs_relayout(&self) -> bool { self.tabs.needs_relayout() } fn required_size(&mut self, constraint: XY) -> XY { self.tabs.required_size(constraint) } fn on_event(&mut self, e: Event) -> EventResult { match e { Event::Char('q') => { if self.tabs.tab_order().len() == 1 { EventResult::Ignored } else { if let Some(key) = self.tabs.active_tab().cloned() { debug!("Removing tab: {}", key); if let Err(e) = self.tabs.remove_tab(&key) { error!("{:?}", e); // TODO do more than just logging } debug!("Remove tab"); EventResult::Consumed(None) } else { debug!("No tab to remove found."); EventResult::Ignored } } }, Event::Key(cursive::event::Key::Enter) => { let muxroot = self.muxroot.clone(); EventResult::Consumed(Some(Callback::from_fn(move |siv| Self::add_mailview(siv, muxroot)))) }, Event::Char('o') => { EventResult::Consumed(Some(Callback::from_fn(MainView::add_notmuch_query_layer))) }, other => self.tabs.on_event(other), } } fn call_on_any<'a>(&mut self, s: &Selector, tpl: &'a mut (dyn FnMut(&mut (dyn View + 'static)) + 'a)) { self.tabs.call_on_any(s, tpl); } fn focus_view(&mut self, s: &Selector) -> Result<(), ()> { self.tabs.focus_view(s) } fn take_focus(&mut self, source: Direction) -> bool { self.tabs.take_focus(source) } fn important_area(&self, view_size: XY) -> Rect { self.tabs.important_area(view_size) } fn type_name(&self) -> &'static str { self.tabs.type_name() } } impl MainView { pub fn new(config: Configuration) -> Result> { use cursive::view::SizeConstraint; let mut tab = cursive_multiplex::Mux::new(); let muxroot = tab.root().build().unwrap(); { let dbpath = config.notmuch_database_path(); let dbquery = config.notmuch_default_query(); let mlname = MAIN_MAIL_LIST_NAME.to_string(); let view = MaillistView::create_for(dbpath.to_path_buf(), dbquery, mlname)?; let _ = tab.add_right_of(view, muxroot); } let tabs = cursive_tabs::TabPanel::default() .with_bar_alignment(cursive_tabs::Align::Start) .with_bar_placement(cursive_tabs::Placement::HorizontalTop) .with_tab(config.notmuch_default_query().clone(), tab.with_name(MAIN_MUX_NAME)); Ok(MainView { config, muxroot, tabs }.with_name(MAIN_VIEW_NAME)) } pub fn add_tab(&mut self, id: String, view: T) { self.tabs.add_tab(id, view) } pub fn config(&self) -> &Configuration { &self.config } fn add_notmuch_query_layer(siv: &mut Cursive) { use crate::util::dialog_for; use crate::util::error_dialog_for; let edit_view = EditView::new() .on_submit(move |siv: &mut Cursive, query: &str| { siv.call_on_name(MAIN_VIEW_NAME, move |main_view: &mut MainView| { main_view.config().notmuch_database_path().clone() }) .map(|dbpath| { let t = MaillistView::create_for(dbpath.to_path_buf(), query, query.to_string())? .full_screen() .with_name(format!("{}-view", query)); siv.call_on_name(MAIN_VIEW_NAME, move |main_view: &mut MainView| { main_view.add_tab(query.to_string(), t); }); siv.pop_layer(); Ok(()) }) .unwrap_or_else(|| { siv.pop_layer(); siv.add_layer(dialog_for("Failed to get database connection set up")); Ok(()) }) .unwrap_or_else(|e: anyhow::Error| { siv.pop_layer(); siv.add_layer(error_dialog_for(e)) }); }) .with_name("query"); siv.add_layer({ Dialog::around(edit_view) .title("Query") .min_width(80) }) } fn add_mailview(siv: &mut Cursive, muxroot: cursive_multiplex::Id) { debug!("Creating mailview"); siv.call_on_name(MAIN_MAIL_LIST_NAME, |maillist_view: &mut MaillistView| { if let Some(ml_data) = maillist_view.item().and_then(move |idx| maillist_view.borrow_item(idx)) { debug!("Found item"); Ok(ml_data.clone()) } else { unimplemented!() } }) .map(|maillist_item: Result| { if let Ok(mldata) = maillist_item { let _ = siv.call_on_name(MAIN_MUX_NAME, |mux: &mut Mux| { debug!("Adding mux pane"); let _ = mux.add_right_of(crate::mailview::MailView::create_for(mldata), muxroot); // TODO handle error }); // TODO handle error. } else { // do something } Ok(()) }) .unwrap_or_else(|| { siv.add_layer(crate::util::dialog_for("Failed to find View")); Ok(()) }) .unwrap_or_else(|e: anyhow::Error| { siv.add_layer(crate::util::error_dialog_for(e)) }); } }