use super::{ItemHandle, SplitDirection}; use crate::{Item, Settings, WeakItemHandle, Workspace}; use collections::{HashMap, VecDeque}; use gpui::{ action, elements::*, geometry::{rect::RectF, vector::vec2f}, keymap::Binding, platform::{CursorStyle, NavigationDirection}, AnyViewHandle, AppContext, Entity, MutableAppContext, Quad, RenderContext, Task, View, ViewContext, ViewHandle, WeakViewHandle, }; use project::{ProjectEntryId, ProjectPath}; use std::{ any::{Any, TypeId}, cell::RefCell, cmp, mem, rc::Rc, }; use util::ResultExt; action!(Split, SplitDirection); action!(ActivateItem, usize); action!(ActivatePrevItem); action!(ActivateNextItem); action!(CloseActiveItem); action!(CloseInactiveItems); action!(CloseItem, usize); action!(GoBack, Option>); action!(GoForward, Option>); const MAX_NAVIGATION_HISTORY_LEN: usize = 1024; pub fn init(cx: &mut MutableAppContext) { cx.add_action(|pane: &mut Pane, action: &ActivateItem, cx| { pane.activate_item(action.0, true, cx); }); cx.add_action(|pane: &mut Pane, _: &ActivatePrevItem, cx| { pane.activate_prev_item(cx); }); cx.add_action(|pane: &mut Pane, _: &ActivateNextItem, cx| { pane.activate_next_item(cx); }); cx.add_action(|pane: &mut Pane, _: &CloseActiveItem, cx| { pane.close_active_item(cx); }); cx.add_action(|pane: &mut Pane, _: &CloseInactiveItems, cx| { pane.close_inactive_items(cx); }); cx.add_action(|pane: &mut Pane, action: &CloseItem, cx| { pane.close_item(action.0, cx); }); cx.add_action(|pane: &mut Pane, action: &Split, cx| { pane.split(action.0, cx); }); cx.add_action(|workspace: &mut Workspace, action: &GoBack, cx| { Pane::go_back( workspace, action .0 .as_ref() .and_then(|weak_handle| weak_handle.upgrade(cx)), cx, ) .detach(); }); cx.add_action(|workspace: &mut Workspace, action: &GoForward, cx| { Pane::go_forward( workspace, action .0 .as_ref() .and_then(|weak_handle| weak_handle.upgrade(cx)), cx, ) .detach(); }); cx.add_bindings(vec![ Binding::new("shift-cmd-{", ActivatePrevItem, Some("Pane")), Binding::new("shift-cmd-}", ActivateNextItem, Some("Pane")), Binding::new("cmd-w", CloseActiveItem, Some("Pane")), Binding::new("alt-cmd-w", CloseInactiveItems, Some("Pane")), Binding::new("cmd-k up", Split(SplitDirection::Up), Some("Pane")), Binding::new("cmd-k down", Split(SplitDirection::Down), Some("Pane")), Binding::new("cmd-k left", Split(SplitDirection::Left), Some("Pane")), Binding::new("cmd-k right", Split(SplitDirection::Right), Some("Pane")), Binding::new("ctrl--", GoBack(None), Some("Pane")), Binding::new("shift-ctrl-_", GoForward(None), Some("Pane")), ]); } pub enum Event { Activate, ActivateItem { local: bool }, Remove, Split(SplitDirection), } pub struct Pane { items: Vec>, active_item_index: usize, nav_history: Rc>, toolbars: HashMap>, active_toolbar_type: Option, active_toolbar_visible: bool, } pub trait Toolbar: View { fn active_item_changed( &mut self, item: Option>, cx: &mut ViewContext, ) -> bool; fn on_dismiss(&mut self, cx: &mut ViewContext); } trait ToolbarHandle { fn active_item_changed( &self, item: Option>, cx: &mut MutableAppContext, ) -> bool; fn on_dismiss(&self, cx: &mut MutableAppContext); fn to_any(&self) -> AnyViewHandle; } pub struct ItemNavHistory { history: Rc>, item: Rc, } #[derive(Default)] pub struct NavHistory { mode: NavigationMode, backward_stack: VecDeque, forward_stack: VecDeque, paths_by_item: HashMap, } #[derive(Copy, Clone)] enum NavigationMode { Normal, GoingBack, GoingForward, Disabled, } impl Default for NavigationMode { fn default() -> Self { Self::Normal } } pub struct NavigationEntry { pub item: Rc, pub data: Option>, } impl Pane { pub fn new() -> Self { Self { items: Vec::new(), active_item_index: 0, nav_history: Default::default(), toolbars: Default::default(), active_toolbar_type: Default::default(), active_toolbar_visible: false, } } pub fn nav_history(&self) -> &Rc> { &self.nav_history } pub fn activate(&self, cx: &mut ViewContext) { cx.emit(Event::Activate); } pub fn go_back( workspace: &mut Workspace, pane: Option>, cx: &mut ViewContext, ) -> Task<()> { Self::navigate_history( workspace, pane.unwrap_or_else(|| workspace.active_pane().clone()), NavigationMode::GoingBack, cx, ) } pub fn go_forward( workspace: &mut Workspace, pane: Option>, cx: &mut ViewContext, ) -> Task<()> { Self::navigate_history( workspace, pane.unwrap_or_else(|| workspace.active_pane().clone()), NavigationMode::GoingForward, cx, ) } fn navigate_history( workspace: &mut Workspace, pane: ViewHandle, mode: NavigationMode, cx: &mut ViewContext, ) -> Task<()> { workspace.activate_pane(pane.clone(), cx); let to_load = pane.update(cx, |pane, cx| { loop { // Retrieve the weak item handle from the history. let entry = pane.nav_history.borrow_mut().pop(mode)?; // If the item is still present in this pane, then activate it. if let Some(index) = entry .item .upgrade(cx) .and_then(|v| pane.index_for_item(v.as_ref())) { if let Some(item) = pane.active_item() { pane.nav_history.borrow_mut().set_mode(mode); item.deactivated(cx); pane.nav_history .borrow_mut() .set_mode(NavigationMode::Normal); } let prev_active_index = mem::replace(&mut pane.active_item_index, index); pane.focus_active_item(cx); let mut navigated = prev_active_index != pane.active_item_index; if let Some(data) = entry.data { navigated |= pane.active_item()?.navigate(data, cx); } if navigated { cx.notify(); break None; } } // If the item is no longer present in this pane, then retrieve its // project path in order to reopen it. else { break pane .nav_history .borrow_mut() .paths_by_item .get(&entry.item.id()) .cloned() .map(|project_path| (project_path, entry)); } } }); if let Some((project_path, entry)) = to_load { // If the item was no longer present, then load it again from its previous path. let pane = pane.downgrade(); let task = workspace.load_path(project_path, cx); cx.spawn(|workspace, mut cx| async move { let task = task.await; if let Some(pane) = pane.upgrade(&cx) { if let Some((project_entry_id, build_item)) = task.log_err() { pane.update(&mut cx, |pane, _| { pane.nav_history.borrow_mut().set_mode(mode); }); let item = workspace.update(&mut cx, |workspace, cx| { Self::open_item( workspace, pane.clone(), project_entry_id, cx, build_item, ) }); pane.update(&mut cx, |pane, cx| { pane.nav_history .borrow_mut() .set_mode(NavigationMode::Normal); if let Some(data) = entry.data { item.navigate(data, cx); } }); } else { workspace .update(&mut cx, |workspace, cx| { Self::navigate_history(workspace, pane, mode, cx) }) .await; } } }) } else { Task::ready(()) } } pub(crate) fn open_item( workspace: &mut Workspace, pane: ViewHandle, project_entry_id: ProjectEntryId, cx: &mut ViewContext, build_item: impl FnOnce(&mut MutableAppContext) -> Box, ) -> Box { let existing_item = pane.update(cx, |pane, cx| { for (ix, item) in pane.items.iter().enumerate() { if item.project_entry_id(cx) == Some(project_entry_id) { let item = item.boxed_clone(); pane.activate_item(ix, true, cx); return Some(item); } } None }); if let Some(existing_item) = existing_item { existing_item } else { let item = build_item(cx); Self::add_item(workspace, pane, item.boxed_clone(), true, cx); item } } pub(crate) fn add_item( workspace: &mut Workspace, pane: ViewHandle, item: Box, local: bool, cx: &mut ViewContext, ) { // Prevent adding the same item to the pane more than once. if let Some(item_ix) = pane.read(cx).items.iter().position(|i| i.id() == item.id()) { pane.update(cx, |pane, cx| pane.activate_item(item_ix, local, cx)); return; } item.set_nav_history(pane.read(cx).nav_history.clone(), cx); item.added_to_pane(workspace, pane.clone(), cx); pane.update(cx, |pane, cx| { let item_idx = cmp::min(pane.active_item_index + 1, pane.items.len()); pane.items.insert(item_idx, item); pane.activate_item(item_idx, local, cx); cx.notify(); }); } pub fn items(&self) -> impl Iterator> { self.items.iter() } pub fn items_of_type<'a, T: View>(&'a self) -> impl 'a + Iterator> { self.items .iter() .filter_map(|item| item.to_any().downcast()) } pub fn active_item(&self) -> Option> { self.items.get(self.active_item_index).cloned() } pub fn project_entry_id_for_item( &self, item: &dyn ItemHandle, cx: &AppContext, ) -> Option { self.items.iter().find_map(|existing| { if existing.id() == item.id() { existing.project_entry_id(cx) } else { None } }) } pub fn item_for_entry( &self, entry_id: ProjectEntryId, cx: &AppContext, ) -> Option> { self.items.iter().find_map(|item| { if item.project_entry_id(cx) == Some(entry_id) { Some(item.boxed_clone()) } else { None } }) } pub fn index_for_item(&self, item: &dyn ItemHandle) -> Option { self.items.iter().position(|i| i.id() == item.id()) } pub fn activate_item(&mut self, index: usize, local: bool, cx: &mut ViewContext) { if index < self.items.len() { let prev_active_item_ix = mem::replace(&mut self.active_item_index, index); if prev_active_item_ix != self.active_item_index && prev_active_item_ix < self.items.len() { self.items[prev_active_item_ix].deactivated(cx); cx.emit(Event::ActivateItem { local }); } self.update_active_toolbar(cx); if local { self.focus_active_item(cx); self.activate(cx); } cx.notify(); } } pub fn activate_prev_item(&mut self, cx: &mut ViewContext) { let mut index = self.active_item_index; if index > 0 { index -= 1; } else if self.items.len() > 0 { index = self.items.len() - 1; } self.activate_item(index, true, cx); } pub fn activate_next_item(&mut self, cx: &mut ViewContext) { let mut index = self.active_item_index; if index + 1 < self.items.len() { index += 1; } else { index = 0; } self.activate_item(index, true, cx); } pub fn close_active_item(&mut self, cx: &mut ViewContext) { if !self.items.is_empty() { self.close_item(self.items[self.active_item_index].id(), cx) } } pub fn close_inactive_items(&mut self, cx: &mut ViewContext) { if !self.items.is_empty() { let active_item_id = self.items[self.active_item_index].id(); self.close_items(cx, |id| id != active_item_id); } } pub fn close_item(&mut self, view_id_to_close: usize, cx: &mut ViewContext) { self.close_items(cx, |view_id| view_id == view_id_to_close); } pub fn close_items( &mut self, cx: &mut ViewContext, should_close: impl Fn(usize) -> bool, ) { let mut item_ix = 0; let mut new_active_item_index = self.active_item_index; self.items.retain(|item| { if should_close(item.id()) { if item_ix == self.active_item_index { item.deactivated(cx); } if item_ix < self.active_item_index { new_active_item_index -= 1; } let mut nav_history = self.nav_history.borrow_mut(); if let Some(path) = item.project_path(cx) { nav_history.paths_by_item.insert(item.id(), path); } else { nav_history.paths_by_item.remove(&item.id()); } item_ix += 1; false } else { item_ix += 1; true } }); if self.items.is_empty() { cx.emit(Event::Remove); } else { self.active_item_index = cmp::min(new_active_item_index, self.items.len() - 1); self.focus_active_item(cx); self.activate(cx); } self.update_active_toolbar(cx); cx.notify(); } pub fn focus_active_item(&mut self, cx: &mut ViewContext) { if let Some(active_item) = self.active_item() { cx.focus(active_item); } } pub fn split(&mut self, direction: SplitDirection, cx: &mut ViewContext) { cx.emit(Event::Split(direction)); } pub fn show_toolbar(&mut self, cx: &mut ViewContext, build_toolbar: F) where F: FnOnce(&mut ViewContext) -> V, V: Toolbar, { let type_id = TypeId::of::(); if self.active_toolbar_type != Some(type_id) { self.dismiss_toolbar(cx); let active_item = self.active_item(); self.toolbars .entry(type_id) .or_insert_with(|| Box::new(cx.add_view(build_toolbar))); self.active_toolbar_type = Some(type_id); self.active_toolbar_visible = self.toolbars[&type_id].active_item_changed(active_item, cx); cx.notify(); } } pub fn dismiss_toolbar(&mut self, cx: &mut ViewContext) { if let Some(active_toolbar_type) = self.active_toolbar_type.take() { self.toolbars .get_mut(&active_toolbar_type) .unwrap() .on_dismiss(cx); self.active_toolbar_visible = false; self.focus_active_item(cx); cx.notify(); } } pub fn toolbar(&self) -> Option> { self.toolbars .get(&TypeId::of::()) .and_then(|toolbar| toolbar.to_any().downcast()) } pub fn active_toolbar(&self) -> Option { let type_id = self.active_toolbar_type?; let toolbar = self.toolbars.get(&type_id)?; if self.active_toolbar_visible { Some(toolbar.to_any()) } else { None } } fn update_active_toolbar(&mut self, cx: &mut ViewContext) { let active_item = self.items.get(self.active_item_index); for (toolbar_type_id, toolbar) in &self.toolbars { let visible = toolbar.active_item_changed(active_item.cloned(), cx); if Some(*toolbar_type_id) == self.active_toolbar_type { self.active_toolbar_visible = visible; } } } fn render_tabs(&self, cx: &mut RenderContext) -> ElementBox { let theme = cx.global::().theme.clone(); enum Tabs {} let tabs = MouseEventHandler::new::(0, cx, |mouse_state, cx| { let mut row = Flex::row(); for (ix, item) in self.items.iter().enumerate() { let is_active = ix == self.active_item_index; row.add_child({ let tab_style = if is_active { theme.workspace.active_tab.clone() } else { theme.workspace.tab.clone() }; let title = item.tab_content(&tab_style, cx); let mut style = if is_active { theme.workspace.active_tab.clone() } else { theme.workspace.tab.clone() }; if ix == 0 { style.container.border.left = false; } EventHandler::new( Container::new( Flex::row() .with_child( Align::new({ let diameter = 7.0; let icon_color = if item.has_conflict(cx) { Some(style.icon_conflict) } else if item.is_dirty(cx) { Some(style.icon_dirty) } else { None }; ConstrainedBox::new( Canvas::new(move |bounds, _, cx| { if let Some(color) = icon_color { let square = RectF::new( bounds.origin(), vec2f(diameter, diameter), ); cx.scene.push_quad(Quad { bounds: square, background: Some(color), border: Default::default(), corner_radius: diameter / 2., }); } }) .boxed(), ) .with_width(diameter) .with_height(diameter) .boxed() }) .boxed(), ) .with_child( Container::new(Align::new(title).boxed()) .with_style(ContainerStyle { margin: Margin { left: style.spacing, right: style.spacing, ..Default::default() }, ..Default::default() }) .boxed(), ) .with_child( Align::new( ConstrainedBox::new(if mouse_state.hovered { let item_id = item.id(); enum TabCloseButton {} let icon = Svg::new("icons/x.svg"); MouseEventHandler::new::( item_id, cx, |mouse_state, _| { if mouse_state.hovered { icon.with_color(style.icon_close_active) .boxed() } else { icon.with_color(style.icon_close).boxed() } }, ) .with_padding(Padding::uniform(4.)) .with_cursor_style(CursorStyle::PointingHand) .on_click(move |cx| { cx.dispatch_action(CloseItem(item_id)) }) .named("close-tab-icon") } else { Empty::new().boxed() }) .with_width(style.icon_width) .boxed(), ) .boxed(), ) .boxed(), ) .with_style(style.container) .boxed(), ) .on_mouse_down(move |cx| { cx.dispatch_action(ActivateItem(ix)); true }) .boxed() }) } row.add_child( Empty::new() .contained() .with_border(theme.workspace.tab.container.border) .flexible(0., true) .named("filler"), ); row.boxed() }); ConstrainedBox::new(tabs.boxed()) .with_height(theme.workspace.tab.height) .named("tabs") } } impl Entity for Pane { type Event = Event; } impl View for Pane { fn ui_name() -> &'static str { "Pane" } fn render(&mut self, cx: &mut RenderContext) -> ElementBox { let this = cx.handle(); EventHandler::new(if let Some(active_item) = self.active_item() { Flex::column() .with_child(self.render_tabs(cx)) .with_children( self.active_toolbar() .as_ref() .map(|view| ChildView::new(view).boxed()), ) .with_child(ChildView::new(active_item).flexible(1., true).boxed()) .boxed() } else { Empty::new().boxed() }) .on_navigate_mouse_down(move |direction, cx| { let this = this.clone(); match direction { NavigationDirection::Back => cx.dispatch_action(GoBack(Some(this))), NavigationDirection::Forward => cx.dispatch_action(GoForward(Some(this))), } true }) .named("pane") } fn on_focus(&mut self, cx: &mut ViewContext) { self.focus_active_item(cx); } } impl ToolbarHandle for ViewHandle { fn active_item_changed( &self, item: Option>, cx: &mut MutableAppContext, ) -> bool { self.update(cx, |this, cx| this.active_item_changed(item, cx)) } fn on_dismiss(&self, cx: &mut MutableAppContext) { self.update(cx, |this, cx| this.on_dismiss(cx)); } fn to_any(&self) -> AnyViewHandle { self.into() } } impl ItemNavHistory { pub fn new(history: Rc>, item: &ViewHandle) -> Self { Self { history, item: Rc::new(item.downgrade()), } } pub fn history(&self) -> Rc> { self.history.clone() } pub fn push(&self, data: Option) { self.history.borrow_mut().push(data, self.item.clone()); } } impl NavHistory { pub fn disable(&mut self) { self.mode = NavigationMode::Disabled; } pub fn enable(&mut self) { self.mode = NavigationMode::Normal; } pub fn pop_backward(&mut self) -> Option { self.backward_stack.pop_back() } pub fn pop_forward(&mut self) -> Option { self.forward_stack.pop_back() } fn pop(&mut self, mode: NavigationMode) -> Option { match mode { NavigationMode::Normal | NavigationMode::Disabled => None, NavigationMode::GoingBack => self.pop_backward(), NavigationMode::GoingForward => self.pop_forward(), } } fn set_mode(&mut self, mode: NavigationMode) { self.mode = mode; } pub fn push(&mut self, data: Option, item: Rc) { match self.mode { NavigationMode::Disabled => {} NavigationMode::Normal => { if self.backward_stack.len() >= MAX_NAVIGATION_HISTORY_LEN { self.backward_stack.pop_front(); } self.backward_stack.push_back(NavigationEntry { item, data: data.map(|data| Box::new(data) as Box), }); self.forward_stack.clear(); } NavigationMode::GoingBack => { if self.forward_stack.len() >= MAX_NAVIGATION_HISTORY_LEN { self.forward_stack.pop_front(); } self.forward_stack.push_back(NavigationEntry { item, data: data.map(|data| Box::new(data) as Box), }); } NavigationMode::GoingForward => { if self.backward_stack.len() >= MAX_NAVIGATION_HISTORY_LEN { self.backward_stack.pop_front(); } self.backward_stack.push_back(NavigationEntry { item, data: data.map(|data| Box::new(data) as Box), }); } } } }