pub mod lsp_status; pub mod menu; pub mod pane; pub mod pane_group; pub mod settings; pub mod sidebar; mod status_bar; use anyhow::{anyhow, Result}; use client::{ proto, Authenticate, ChannelList, Client, PeerId, Subscription, TypedEnvelope, User, UserStore, }; use clock::ReplicaId; use collections::{HashMap, HashSet}; use gpui::{ action, color::Color, elements::*, geometry::{vector::vec2f, PathBuilder}, json::{self, to_string_pretty, ToJson}, keymap::Binding, platform::{CursorStyle, WindowOptions}, AnyModelHandle, AnyViewHandle, AppContext, AsyncAppContext, ClipboardItem, Entity, ImageData, ModelHandle, MutableAppContext, PathPromptOptions, PromptLevel, RenderContext, Task, View, ViewContext, ViewHandle, WeakViewHandle, }; use language::LanguageRegistry; use log::error; pub use pane::*; pub use pane_group::*; use postage::prelude::Stream; use project::{fs, Fs, Project, ProjectEntryId, ProjectPath, Worktree}; pub use settings::Settings; use sidebar::{Side, Sidebar, SidebarItemId, ToggleSidebarItem, ToggleSidebarItemFocus}; use status_bar::StatusBar; pub use status_bar::StatusItemView; use std::{ any::{Any, TypeId}, cell::RefCell, future::Future, path::{Path, PathBuf}, rc::Rc, sync::Arc, }; use theme::{Theme, ThemeRegistry}; type ProjectItemBuilders = HashMap< TypeId, fn(usize, ModelHandle, AnyModelHandle, &mut MutableAppContext) -> Box, >; type FollowedItemBuilder = fn( ViewHandle, ModelHandle, &mut Option, &mut MutableAppContext, ) -> Option>>>; type FollowedItemBuilders = HashMap< TypeId, ( FollowedItemBuilder, fn(AnyViewHandle) -> Box, ), >; action!(Open, Arc); action!(OpenNew, Arc); action!(OpenPaths, OpenParams); action!(ToggleShare); action!(JoinProject, JoinProjectParams); action!(Save); action!(DebugElements); pub fn init(client: &Arc, cx: &mut MutableAppContext) { pane::init(cx); menu::init(cx); cx.add_global_action(open); cx.add_global_action(move |action: &OpenPaths, cx: &mut MutableAppContext| { open_paths(&action.0.paths, &action.0.app_state, cx).detach(); }); cx.add_global_action(move |action: &OpenNew, cx: &mut MutableAppContext| { open_new(&action.0, cx) }); cx.add_global_action(move |action: &JoinProject, cx: &mut MutableAppContext| { join_project(action.0.project_id, &action.0.app_state, cx).detach(); }); cx.add_action(Workspace::toggle_share); cx.add_action( |workspace: &mut Workspace, _: &Save, cx: &mut ViewContext| { workspace.save_active_item(cx).detach_and_log_err(cx); }, ); cx.add_action(Workspace::debug_elements); cx.add_action(Workspace::toggle_sidebar_item); cx.add_action(Workspace::toggle_sidebar_item_focus); cx.add_bindings(vec![ Binding::new("cmd-s", Save, None), Binding::new("cmd-alt-i", DebugElements, None), Binding::new( "cmd-shift-!", ToggleSidebarItem(SidebarItemId { side: Side::Left, item_index: 0, }), None, ), Binding::new( "cmd-1", ToggleSidebarItemFocus(SidebarItemId { side: Side::Left, item_index: 0, }), None, ), ]); client.add_view_request_handler(Workspace::handle_follow); client.add_view_message_handler(Workspace::handle_unfollow); } pub fn register_project_item(cx: &mut MutableAppContext) { cx.update_default_global(|builders: &mut ProjectItemBuilders, _| { builders.insert(TypeId::of::(), |window_id, project, model, cx| { let item = model.downcast::().unwrap(); Box::new(cx.add_view(window_id, |cx| I::for_project_item(project, item, cx))) }); }); } pub fn register_followed_item(cx: &mut MutableAppContext) { cx.update_default_global(|builders: &mut FollowedItemBuilders, _| { builders.insert( TypeId::of::(), (I::for_state_message, |this| { Box::new(this.downcast::().unwrap()) }), ); }); } pub struct AppState { pub languages: Arc, pub themes: Arc, pub client: Arc, pub user_store: ModelHandle, pub fs: Arc, pub channel_list: ModelHandle, pub build_window_options: &'static dyn Fn() -> WindowOptions<'static>, pub build_workspace: &'static dyn Fn( ModelHandle, &Arc, &mut ViewContext, ) -> Workspace, } #[derive(Clone)] pub struct OpenParams { pub paths: Vec, pub app_state: Arc, } #[derive(Clone)] pub struct JoinProjectParams { pub project_id: u64, pub app_state: Arc, } pub trait Item: View { fn deactivated(&mut self, _: &mut ViewContext) {} fn navigate(&mut self, _: Box, _: &mut ViewContext) {} fn tab_content(&self, style: &theme::Tab, cx: &AppContext) -> ElementBox; fn project_path(&self, cx: &AppContext) -> Option; fn project_entry_id(&self, cx: &AppContext) -> Option; fn set_nav_history(&mut self, _: ItemNavHistory, _: &mut ViewContext); fn clone_on_split(&self, _: &mut ViewContext) -> Option where Self: Sized, { None } fn is_dirty(&self, _: &AppContext) -> bool { false } fn has_conflict(&self, _: &AppContext) -> bool { false } fn can_save(&self, cx: &AppContext) -> bool; fn save( &mut self, project: ModelHandle, cx: &mut ViewContext, ) -> Task>; fn can_save_as(&self, cx: &AppContext) -> bool; fn save_as( &mut self, project: ModelHandle, abs_path: PathBuf, cx: &mut ViewContext, ) -> Task>; fn should_activate_item_on_event(_: &Self::Event) -> bool { false } fn should_close_item_on_event(_: &Self::Event) -> bool { false } fn should_update_tab_on_event(_: &Self::Event) -> bool { false } fn act_as_type( &self, type_id: TypeId, self_handle: &ViewHandle, _: &AppContext, ) -> Option { if TypeId::of::() == type_id { Some(self_handle.into()) } else { None } } } pub trait ProjectItem: Item { type Item: project::Item; fn for_project_item( project: ModelHandle, item: ModelHandle, cx: &mut ViewContext, ) -> Self; } pub trait FollowedItem: Item { fn for_state_message( pane: ViewHandle, project: ModelHandle, state: &mut Option, cx: &mut MutableAppContext, ) -> Option>>> where Self: Sized; fn to_state_message(&self, cx: &AppContext) -> proto::view::Variant; fn to_update_message( &self, event: &Self::Event, cx: &AppContext, ) -> Option; } pub trait FollowedItemHandle { fn id(&self) -> usize; fn to_state_message(&self, cx: &AppContext) -> proto::view::Variant; fn to_update_message( &self, event: &dyn Any, cx: &AppContext, ) -> Option; } impl FollowedItemHandle for ViewHandle { fn id(&self) -> usize { self.id() } fn to_state_message(&self, cx: &AppContext) -> proto::view::Variant { self.read(cx).to_state_message(cx) } fn to_update_message( &self, event: &dyn Any, cx: &AppContext, ) -> Option { self.read(cx).to_update_message(event.downcast_ref()?, cx) } } pub trait ItemHandle: 'static { fn tab_content(&self, style: &theme::Tab, cx: &AppContext) -> ElementBox; fn project_path(&self, cx: &AppContext) -> Option; fn project_entry_id(&self, cx: &AppContext) -> Option; fn boxed_clone(&self) -> Box; fn set_nav_history(&self, nav_history: Rc>, cx: &mut MutableAppContext); fn clone_on_split(&self, cx: &mut MutableAppContext) -> Option>; fn added_to_pane( &self, workspace: &mut Workspace, pane: ViewHandle, cx: &mut ViewContext, ); fn deactivated(&self, cx: &mut MutableAppContext); fn navigate(&self, data: Box, cx: &mut MutableAppContext); fn id(&self) -> usize; fn to_any(&self) -> AnyViewHandle; fn is_dirty(&self, cx: &AppContext) -> bool; fn has_conflict(&self, cx: &AppContext) -> bool; fn can_save(&self, cx: &AppContext) -> bool; fn can_save_as(&self, cx: &AppContext) -> bool; fn save(&self, project: ModelHandle, cx: &mut MutableAppContext) -> Task>; fn save_as( &self, project: ModelHandle, abs_path: PathBuf, cx: &mut MutableAppContext, ) -> Task>; fn act_as_type(&self, type_id: TypeId, cx: &AppContext) -> Option; fn to_followed_item_handle(&self, cx: &AppContext) -> Option>; } pub trait WeakItemHandle { fn id(&self) -> usize; fn upgrade(&self, cx: &AppContext) -> Option>; } impl dyn ItemHandle { pub fn downcast(&self) -> Option> { self.to_any().downcast() } pub fn act_as(&self, cx: &AppContext) -> Option> { self.act_as_type(TypeId::of::(), cx) .and_then(|t| t.downcast()) } } impl ItemHandle for ViewHandle { fn tab_content(&self, style: &theme::Tab, cx: &AppContext) -> ElementBox { self.read(cx).tab_content(style, cx) } fn project_path(&self, cx: &AppContext) -> Option { self.read(cx).project_path(cx) } fn project_entry_id(&self, cx: &AppContext) -> Option { self.read(cx).project_entry_id(cx) } fn boxed_clone(&self) -> Box { Box::new(self.clone()) } fn clone_on_split(&self, cx: &mut MutableAppContext) -> Option> { self.update(cx, |item, cx| { cx.add_option_view(|cx| item.clone_on_split(cx)) }) .map(|handle| Box::new(handle) as Box) } fn set_nav_history(&self, nav_history: Rc>, cx: &mut MutableAppContext) { self.update(cx, |item, cx| { item.set_nav_history(ItemNavHistory::new(nav_history, &cx.handle()), cx); }) } fn added_to_pane( &self, workspace: &mut Workspace, pane: ViewHandle, cx: &mut ViewContext, ) { let pane = pane.downgrade(); cx.subscribe(self, move |workspace, item, event, cx| { let pane = if let Some(pane) = pane.upgrade(cx) { pane } else { log::error!("unexpected item event after pane was dropped"); return; }; if T::should_close_item_on_event(event) { pane.update(cx, |pane, cx| pane.close_item(item.id(), cx)); return; } if T::should_activate_item_on_event(event) { pane.update(cx, |pane, cx| { if let Some(ix) = pane.index_for_item(&item) { pane.activate_item(ix, cx); pane.activate(cx); } }); } if T::should_update_tab_on_event(event) { pane.update(cx, |_, cx| cx.notify()); } if let Some(message) = item .to_followed_item_handle(cx) .and_then(|i| i.to_update_message(event, cx)) {} }) .detach(); } fn deactivated(&self, cx: &mut MutableAppContext) { self.update(cx, |this, cx| this.deactivated(cx)); } fn navigate(&self, data: Box, cx: &mut MutableAppContext) { self.update(cx, |this, cx| this.navigate(data, cx)); } fn save(&self, project: ModelHandle, cx: &mut MutableAppContext) -> Task> { self.update(cx, |item, cx| item.save(project, cx)) } fn save_as( &self, project: ModelHandle, abs_path: PathBuf, cx: &mut MutableAppContext, ) -> Task> { self.update(cx, |item, cx| item.save_as(project, abs_path, cx)) } fn is_dirty(&self, cx: &AppContext) -> bool { self.read(cx).is_dirty(cx) } fn has_conflict(&self, cx: &AppContext) -> bool { self.read(cx).has_conflict(cx) } fn id(&self) -> usize { self.id() } fn to_any(&self) -> AnyViewHandle { self.into() } fn can_save(&self, cx: &AppContext) -> bool { self.read(cx).can_save(cx) } fn can_save_as(&self, cx: &AppContext) -> bool { self.read(cx).can_save_as(cx) } fn act_as_type(&self, type_id: TypeId, cx: &AppContext) -> Option { self.read(cx).act_as_type(type_id, self, cx) } fn to_followed_item_handle(&self, cx: &AppContext) -> Option> { if cx.has_global::() { let builders = cx.global::(); let item = self.to_any(); Some(builders.get(&item.view_type())?.1(item)) } else { None } } } impl Into for Box { fn into(self) -> AnyViewHandle { self.to_any() } } impl Clone for Box { fn clone(&self) -> Box { self.boxed_clone() } } impl WeakItemHandle for WeakViewHandle { fn id(&self) -> usize { self.id() } fn upgrade(&self, cx: &AppContext) -> Option> { self.upgrade(cx).map(|v| Box::new(v) as Box) } } #[derive(Clone)] pub struct WorkspaceParams { pub project: ModelHandle, pub client: Arc, pub fs: Arc, pub languages: Arc, pub user_store: ModelHandle, pub channel_list: ModelHandle, } impl WorkspaceParams { #[cfg(any(test, feature = "test-support"))] pub fn test(cx: &mut MutableAppContext) -> Self { let settings = Settings::test(cx); cx.set_global(settings); let fs = project::FakeFs::new(cx.background().clone()); let languages = Arc::new(LanguageRegistry::test()); let http_client = client::test::FakeHttpClient::new(|_| async move { Ok(client::http::ServerResponse::new(404)) }); let client = Client::new(http_client.clone()); let user_store = cx.add_model(|cx| UserStore::new(client.clone(), http_client, cx)); let project = Project::local( client.clone(), user_store.clone(), languages.clone(), fs.clone(), cx, ); Self { project, channel_list: cx .add_model(|cx| ChannelList::new(user_store.clone(), client.clone(), cx)), client, fs, languages, user_store, } } #[cfg(any(test, feature = "test-support"))] pub fn local(app_state: &Arc, cx: &mut MutableAppContext) -> Self { Self { project: Project::local( app_state.client.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), cx, ), client: app_state.client.clone(), fs: app_state.fs.clone(), languages: app_state.languages.clone(), user_store: app_state.user_store.clone(), channel_list: app_state.channel_list.clone(), } } } pub struct Workspace { weak_self: WeakViewHandle, client: Arc, user_store: ModelHandle, remote_entity_subscription: Option, fs: Arc, modal: Option, center: PaneGroup, left_sidebar: Sidebar, right_sidebar: Sidebar, panes: Vec>, active_pane: ViewHandle, status_bar: ViewHandle, project: ModelHandle, leader_state: LeaderState, follower_states_by_leader: HashMap, _observe_current_user: Task<()>, } #[derive(Default)] struct LeaderState { followers: HashSet, subscriptions: Vec, } struct FollowerState { current_view_id: Option, items_by_leader_view_id: HashMap>, } impl Workspace { pub fn new(params: &WorkspaceParams, cx: &mut ViewContext) -> Self { cx.observe(¶ms.project, |_, project, cx| { if project.read(cx).is_read_only() { cx.blur(); } cx.notify() }) .detach(); cx.subscribe(¶ms.project, move |this, project, event, cx| { if let project::Event::RemoteIdChanged(remote_id) = event { this.project_remote_id_changed(*remote_id, cx); } if project.read(cx).is_read_only() { cx.blur(); } cx.notify() }) .detach(); let pane = cx.add_view(|_| Pane::new()); let pane_id = pane.id(); cx.observe(&pane, move |me, _, cx| { let active_entry = me.active_project_path(cx); me.project .update(cx, |project, cx| project.set_active_path(active_entry, cx)); }) .detach(); cx.subscribe(&pane, move |me, _, event, cx| { me.handle_pane_event(pane_id, event, cx) }) .detach(); cx.focus(&pane); let status_bar = cx.add_view(|cx| StatusBar::new(&pane, cx)); let mut current_user = params.user_store.read(cx).watch_current_user().clone(); let mut connection_status = params.client.status().clone(); let _observe_current_user = cx.spawn_weak(|this, mut cx| async move { current_user.recv().await; connection_status.recv().await; let mut stream = Stream::map(current_user, drop).merge(Stream::map(connection_status, drop)); while stream.recv().await.is_some() { cx.update(|cx| { if let Some(this) = this.upgrade(cx) { this.update(cx, |_, cx| cx.notify()); } }) } }); let weak_self = cx.weak_handle(); cx.emit_global(WorkspaceCreated(weak_self.clone())); let mut this = Workspace { modal: None, weak_self, center: PaneGroup::new(pane.clone()), panes: vec![pane.clone()], active_pane: pane.clone(), status_bar, client: params.client.clone(), remote_entity_subscription: None, user_store: params.user_store.clone(), fs: params.fs.clone(), left_sidebar: Sidebar::new(Side::Left), right_sidebar: Sidebar::new(Side::Right), project: params.project.clone(), leader_state: Default::default(), follower_states_by_leader: Default::default(), _observe_current_user, }; this.project_remote_id_changed(this.project.read(cx).remote_id(), cx); this } pub fn weak_handle(&self) -> WeakViewHandle { self.weak_self.clone() } pub fn left_sidebar_mut(&mut self) -> &mut Sidebar { &mut self.left_sidebar } pub fn right_sidebar_mut(&mut self) -> &mut Sidebar { &mut self.right_sidebar } pub fn status_bar(&self) -> &ViewHandle { &self.status_bar } pub fn project(&self) -> &ModelHandle { &self.project } pub fn worktrees<'a>( &self, cx: &'a AppContext, ) -> impl 'a + Iterator> { self.project.read(cx).worktrees(cx) } pub fn contains_paths(&self, paths: &[PathBuf], cx: &AppContext) -> bool { paths.iter().all(|path| self.contains_path(&path, cx)) } pub fn contains_path(&self, path: &Path, cx: &AppContext) -> bool { for worktree in self.worktrees(cx) { let worktree = worktree.read(cx).as_local(); if worktree.map_or(false, |w| w.contains_abs_path(path)) { return true; } } false } pub fn worktree_scans_complete(&self, cx: &AppContext) -> impl Future + 'static { let futures = self .worktrees(cx) .filter_map(|worktree| worktree.read(cx).as_local()) .map(|worktree| worktree.scan_complete()) .collect::>(); async move { for future in futures { future.await; } } } pub fn open_paths( &mut self, abs_paths: &[PathBuf], cx: &mut ViewContext, ) -> Task, Arc>>>> { let entries = abs_paths .iter() .cloned() .map(|path| self.project_path_for_path(&path, cx)) .collect::>(); let fs = self.fs.clone(); let tasks = abs_paths .iter() .cloned() .zip(entries.into_iter()) .map(|(abs_path, project_path)| { cx.spawn(|this, mut cx| { let fs = fs.clone(); async move { let project_path = project_path.await.ok()?; if fs.is_file(&abs_path).await { Some( this.update(&mut cx, |this, cx| this.open_path(project_path, cx)) .await, ) } else { None } } }) }) .collect::>(); cx.foreground().spawn(async move { let mut items = Vec::new(); for task in tasks { items.push(task.await); } items }) } fn project_path_for_path( &self, abs_path: &Path, cx: &mut ViewContext, ) -> Task> { let entry = self.project().update(cx, |project, cx| { project.find_or_create_local_worktree(abs_path, true, cx) }); cx.spawn(|_, cx| async move { let (worktree, path) = entry.await?; Ok(ProjectPath { worktree_id: worktree.read_with(&cx, |t, _| t.id()), path: path.into(), }) }) } // Returns the model that was toggled closed if it was open pub fn toggle_modal( &mut self, cx: &mut ViewContext, add_view: F, ) -> Option> where V: 'static + View, F: FnOnce(&mut ViewContext, &mut Self) -> ViewHandle, { cx.notify(); // Whatever modal was visible is getting clobbered. If its the same type as V, then return // it. Otherwise, create a new modal and set it as active. let already_open_modal = self.modal.take().and_then(|modal| modal.downcast::()); if let Some(already_open_modal) = already_open_modal { cx.focus_self(); Some(already_open_modal) } else { let modal = add_view(cx, self); cx.focus(&modal); self.modal = Some(modal.into()); None } } pub fn modal(&self) -> Option<&AnyViewHandle> { self.modal.as_ref() } pub fn dismiss_modal(&mut self, cx: &mut ViewContext) { if self.modal.take().is_some() { cx.focus(&self.active_pane); cx.notify(); } } pub fn items<'a>( &'a self, cx: &'a AppContext, ) -> impl 'a + Iterator> { self.panes.iter().flat_map(|pane| pane.read(cx).items()) } pub fn item_of_type(&self, cx: &AppContext) -> Option> { self.items_of_type(cx).max_by_key(|item| item.id()) } pub fn items_of_type<'a, T: Item>( &'a self, cx: &'a AppContext, ) -> impl 'a + Iterator> { self.panes .iter() .flat_map(|pane| pane.read(cx).items_of_type()) } pub fn active_item(&self, cx: &AppContext) -> Option> { self.active_pane().read(cx).active_item() } fn active_project_path(&self, cx: &ViewContext) -> Option { self.active_item(cx).and_then(|item| item.project_path(cx)) } pub fn save_active_item(&mut self, cx: &mut ViewContext) -> Task> { let project = self.project.clone(); if let Some(item) = self.active_item(cx) { if item.can_save(cx) { if item.has_conflict(cx.as_ref()) { const CONFLICT_MESSAGE: &'static str = "This file has changed on disk since you started editing it. Do you want to overwrite it?"; let mut answer = cx.prompt( PromptLevel::Warning, CONFLICT_MESSAGE, &["Overwrite", "Cancel"], ); cx.spawn(|_, mut cx| async move { let answer = answer.recv().await; if answer == Some(0) { cx.update(|cx| item.save(project, cx)).await?; } Ok(()) }) } else { item.save(project, cx) } } else if item.can_save_as(cx) { let worktree = self.worktrees(cx).next(); let start_abs_path = worktree .and_then(|w| w.read(cx).as_local()) .map_or(Path::new(""), |w| w.abs_path()) .to_path_buf(); let mut abs_path = cx.prompt_for_new_path(&start_abs_path); cx.spawn(|_, mut cx| async move { if let Some(abs_path) = abs_path.recv().await.flatten() { cx.update(|cx| item.save_as(project, abs_path, cx)).await?; } Ok(()) }) } else { Task::ready(Ok(())) } } else { Task::ready(Ok(())) } } pub fn toggle_sidebar_item(&mut self, action: &ToggleSidebarItem, cx: &mut ViewContext) { let sidebar = match action.0.side { Side::Left => &mut self.left_sidebar, Side::Right => &mut self.right_sidebar, }; sidebar.toggle_item(action.0.item_index); if let Some(active_item) = sidebar.active_item() { cx.focus(active_item); } else { cx.focus_self(); } cx.notify(); } pub fn toggle_sidebar_item_focus( &mut self, action: &ToggleSidebarItemFocus, cx: &mut ViewContext, ) { let sidebar = match action.0.side { Side::Left => &mut self.left_sidebar, Side::Right => &mut self.right_sidebar, }; sidebar.activate_item(action.0.item_index); if let Some(active_item) = sidebar.active_item() { if active_item.is_focused(cx) { cx.focus_self(); } else { cx.focus(active_item); } } cx.notify(); } pub fn debug_elements(&mut self, _: &DebugElements, cx: &mut ViewContext) { match to_string_pretty(&cx.debug_elements()) { Ok(json) => { let kib = json.len() as f32 / 1024.; cx.as_mut().write_to_clipboard(ClipboardItem::new(json)); log::info!( "copied {:.1} KiB of element debug JSON to the clipboard", kib ); } Err(error) => { log::error!("error debugging elements: {}", error); } }; } fn add_pane(&mut self, cx: &mut ViewContext) -> ViewHandle { let pane = cx.add_view(|_| Pane::new()); let pane_id = pane.id(); cx.observe(&pane, move |me, _, cx| { let active_entry = me.active_project_path(cx); me.project .update(cx, |project, cx| project.set_active_path(active_entry, cx)); }) .detach(); cx.subscribe(&pane, move |me, _, event, cx| { me.handle_pane_event(pane_id, event, cx) }) .detach(); self.panes.push(pane.clone()); self.activate_pane(pane.clone(), cx); pane } pub fn add_item(&mut self, item: Box, cx: &mut ViewContext) { let pane = self.active_pane().clone(); Pane::add_item(self, pane, item, cx); } pub fn open_path( &mut self, path: impl Into, cx: &mut ViewContext, ) -> Task, Arc>> { let pane = self.active_pane().downgrade(); let task = self.load_path(path.into(), cx); cx.spawn(|this, mut cx| async move { let (project_entry_id, build_item) = task.await?; let pane = pane .upgrade(&cx) .ok_or_else(|| anyhow!("pane was closed"))?; this.update(&mut cx, |this, cx| { Ok(Pane::open_item( this, pane, project_entry_id, cx, build_item, )) }) }) } pub(crate) fn load_path( &mut self, path: ProjectPath, cx: &mut ViewContext, ) -> Task< Result<( ProjectEntryId, impl 'static + FnOnce(&mut MutableAppContext) -> Box, )>, > { let project = self.project().clone(); let project_item = project.update(cx, |project, cx| project.open_path(path, cx)); let window_id = cx.window_id(); cx.as_mut().spawn(|mut cx| async move { let (project_entry_id, project_item) = project_item.await?; let build_item = cx.update(|cx| { cx.default_global::() .get(&project_item.model_type()) .ok_or_else(|| anyhow!("no item builder for project item")) .cloned() })?; let build_item = move |cx: &mut MutableAppContext| build_item(window_id, project, project_item, cx); Ok((project_entry_id, build_item)) }) } pub fn open_project_item( &mut self, project_item: ModelHandle, cx: &mut ViewContext, ) -> ViewHandle where T: ProjectItem, { use project::Item as _; if let Some(item) = project_item .read(cx) .entry_id(cx) .and_then(|entry_id| self.active_pane().read(cx).item_for_entry(entry_id, cx)) .and_then(|item| item.downcast()) { self.activate_item(&item, cx); return item; } let item = cx.add_view(|cx| T::for_project_item(self.project().clone(), project_item, cx)); self.add_item(Box::new(item.clone()), cx); item } pub fn activate_item(&mut self, item: &dyn ItemHandle, cx: &mut ViewContext) -> bool { let result = self.panes.iter().find_map(|pane| { if let Some(ix) = pane.read(cx).index_for_item(item) { Some((pane.clone(), ix)) } else { None } }); if let Some((pane, ix)) = result { self.activate_pane(pane.clone(), cx); pane.update(cx, |pane, cx| pane.activate_item(ix, cx)); true } else { false } } pub fn activate_next_pane(&mut self, cx: &mut ViewContext) { let ix = self .panes .iter() .position(|pane| pane == &self.active_pane) .unwrap(); let next_ix = (ix + 1) % self.panes.len(); self.activate_pane(self.panes[next_ix].clone(), cx); } fn activate_pane(&mut self, pane: ViewHandle, cx: &mut ViewContext) { if self.active_pane != pane { self.active_pane = pane; self.status_bar.update(cx, |status_bar, cx| { status_bar.set_active_pane(&self.active_pane, cx); }); cx.focus(&self.active_pane); cx.notify(); } } fn handle_pane_event( &mut self, pane_id: usize, event: &pane::Event, cx: &mut ViewContext, ) { if let Some(pane) = self.pane(pane_id) { match event { pane::Event::Split(direction) => { self.split_pane(pane, *direction, cx); } pane::Event::Remove => { self.remove_pane(pane, cx); } pane::Event::Activate => { self.activate_pane(pane, cx); } } } else { error!("pane {} not found", pane_id); } } pub fn split_pane( &mut self, pane: ViewHandle, direction: SplitDirection, cx: &mut ViewContext, ) -> ViewHandle { let new_pane = self.add_pane(cx); self.activate_pane(new_pane.clone(), cx); if let Some(item) = pane.read(cx).active_item() { if let Some(clone) = item.clone_on_split(cx.as_mut()) { Pane::add_item(self, new_pane.clone(), clone, cx); } } self.center.split(&pane, &new_pane, direction).unwrap(); cx.notify(); new_pane } fn remove_pane(&mut self, pane: ViewHandle, cx: &mut ViewContext) { if self.center.remove(&pane).unwrap() { self.panes.retain(|p| p != &pane); self.activate_pane(self.panes.last().unwrap().clone(), cx); cx.notify(); } } pub fn panes(&self) -> &[ViewHandle] { &self.panes } fn pane(&self, pane_id: usize) -> Option> { self.panes.iter().find(|pane| pane.id() == pane_id).cloned() } pub fn active_pane(&self) -> &ViewHandle { &self.active_pane } fn toggle_share(&mut self, _: &ToggleShare, cx: &mut ViewContext) { self.project.update(cx, |project, cx| { if project.is_local() { if project.is_shared() { project.unshare(cx).detach(); } else { project.share(cx).detach(); } } }); } fn project_remote_id_changed(&mut self, remote_id: Option, cx: &mut ViewContext) { if let Some(remote_id) = remote_id { self.remote_entity_subscription = Some(self.client.add_view_for_remote_entity(remote_id, cx)); } else { self.remote_entity_subscription.take(); } } pub fn follow(&mut self, leader_id: PeerId, cx: &mut ViewContext) -> Task> { if let Some(project_id) = self.project.read(cx).remote_id() { let request = self.client.request(proto::Follow { project_id, leader_id: leader_id.0, }); cx.spawn_weak(|this, mut cx| async move { let mut response = request.await?; if let Some(this) = this.upgrade(&cx) { let mut item_tasks = Vec::new(); let (project, pane) = this.read_with(&cx, |this, _| { (this.project.clone(), this.active_pane().clone()) }); let item_builders = cx.update(|cx| { cx.default_global::() .values() .map(|b| b.0) .collect::>() .clone() }); for view in &mut response.views { let variant = view .variant .take() .ok_or_else(|| anyhow!("missing variant"))?; cx.update(|cx| { let mut variant = Some(variant); for build_item in &item_builders { if let Some(task) = build_item(pane.clone(), project.clone(), &mut variant, cx) { item_tasks.push(task); break; } else { assert!(variant.is_some()); } } }); } let items = futures::future::try_join_all(item_tasks).await?; let follower_state = FollowerState { current_view_id: response.current_view_id.map(|id| id as usize), items_by_leader_view_id: response .views .iter() .map(|v| v.id as usize) .zip(items) .collect(), }; let current_item = if let Some(current_view_id) = follower_state.current_view_id { Some( follower_state .items_by_leader_view_id .get(¤t_view_id) .ok_or_else(|| anyhow!("invalid current view id"))? .clone(), ) } else { None }; this.update(&mut cx, |this, cx| { if let Some(item) = current_item { Pane::add_item(this, pane, item, cx); } }); } Ok(()) }) } else { Task::ready(Err(anyhow!("project is not remote"))) } } fn render_connection_status(&self, cx: &mut RenderContext) -> Option { let theme = &cx.global::().theme; match &*self.client.status().borrow() { client::Status::ConnectionError | client::Status::ConnectionLost | client::Status::Reauthenticating | client::Status::Reconnecting { .. } | client::Status::ReconnectionError { .. } => Some( Container::new( Align::new( ConstrainedBox::new( Svg::new("icons/offline-14.svg") .with_color(theme.workspace.titlebar.offline_icon.color) .boxed(), ) .with_width(theme.workspace.titlebar.offline_icon.width) .boxed(), ) .boxed(), ) .with_style(theme.workspace.titlebar.offline_icon.container) .boxed(), ), client::Status::UpgradeRequired => Some( Label::new( "Please update Zed to collaborate".to_string(), theme.workspace.titlebar.outdated_warning.text.clone(), ) .contained() .with_style(theme.workspace.titlebar.outdated_warning.container) .aligned() .boxed(), ), _ => None, } } fn render_titlebar(&self, theme: &Theme, cx: &mut RenderContext) -> ElementBox { ConstrainedBox::new( Container::new( Stack::new() .with_child( Align::new( Label::new("zed".into(), theme.workspace.titlebar.title.clone()) .boxed(), ) .boxed(), ) .with_child( Align::new( Flex::row() .with_children(self.render_share_icon(theme, cx)) .with_children(self.render_collaborators(theme, cx)) .with_child(self.render_current_user( self.user_store.read(cx).current_user().as_ref(), self.project.read(cx).replica_id(), theme, cx, )) .with_children(self.render_connection_status(cx)) .boxed(), ) .right() .boxed(), ) .boxed(), ) .with_style(theme.workspace.titlebar.container) .boxed(), ) .with_height(theme.workspace.titlebar.height) .named("titlebar") } fn render_collaborators(&self, theme: &Theme, cx: &mut RenderContext) -> Vec { let mut collaborators = self .project .read(cx) .collaborators() .values() .cloned() .collect::>(); collaborators.sort_unstable_by_key(|collaborator| collaborator.replica_id); collaborators .into_iter() .filter_map(|collaborator| { Some(self.render_avatar( collaborator.user.avatar.clone()?, collaborator.replica_id, theme, )) }) .collect() } fn render_current_user( &self, user: Option<&Arc>, replica_id: ReplicaId, theme: &Theme, cx: &mut RenderContext, ) -> ElementBox { if let Some(avatar) = user.and_then(|user| user.avatar.clone()) { self.render_avatar(avatar, replica_id, theme) } else { MouseEventHandler::new::(0, cx, |state, _| { let style = if state.hovered { &theme.workspace.titlebar.hovered_sign_in_prompt } else { &theme.workspace.titlebar.sign_in_prompt }; Label::new("Sign in".to_string(), style.text.clone()) .contained() .with_style(style.container) .boxed() }) .on_click(|cx| cx.dispatch_action(Authenticate)) .with_cursor_style(CursorStyle::PointingHand) .aligned() .boxed() } } fn render_avatar( &self, avatar: Arc, replica_id: ReplicaId, theme: &Theme, ) -> ElementBox { ConstrainedBox::new( Stack::new() .with_child( ConstrainedBox::new( Image::new(avatar) .with_style(theme.workspace.titlebar.avatar) .boxed(), ) .with_width(theme.workspace.titlebar.avatar_width) .aligned() .boxed(), ) .with_child( AvatarRibbon::new(theme.editor.replica_selection_style(replica_id).cursor) .constrained() .with_width(theme.workspace.titlebar.avatar_ribbon.width) .with_height(theme.workspace.titlebar.avatar_ribbon.height) .aligned() .bottom() .boxed(), ) .boxed(), ) .with_width(theme.workspace.right_sidebar.width) .boxed() } fn render_share_icon(&self, theme: &Theme, cx: &mut RenderContext) -> Option { if self.project().read(cx).is_local() && self.client.user_id().is_some() { enum Share {} let color = if self.project().read(cx).is_shared() { theme.workspace.titlebar.share_icon_active_color } else { theme.workspace.titlebar.share_icon_color }; Some( MouseEventHandler::new::(0, cx, |_, _| { Align::new( ConstrainedBox::new( Svg::new("icons/broadcast-24.svg").with_color(color).boxed(), ) .with_width(24.) .boxed(), ) .boxed() }) .with_cursor_style(CursorStyle::PointingHand) .on_click(|cx| cx.dispatch_action(ToggleShare)) .boxed(), ) } else { None } } fn render_disconnected_overlay(&self, cx: &AppContext) -> Option { if self.project.read(cx).is_read_only() { let theme = &cx.global::().theme; Some( EventHandler::new( Label::new( "Your connection to the remote project has been lost.".to_string(), theme.workspace.disconnected_overlay.text.clone(), ) .aligned() .contained() .with_style(theme.workspace.disconnected_overlay.container) .boxed(), ) .capture(|_, _, _| true) .boxed(), ) } else { None } } // RPC handlers async fn handle_follow( this: ViewHandle, envelope: TypedEnvelope, _: Arc, mut cx: AsyncAppContext, ) -> Result { this.update(&mut cx, |this, cx| { this.leader_state .followers .insert(envelope.original_sender_id()?); let current_view_id = this .active_item(cx) .and_then(|i| i.to_followed_item_handle(cx)) .map(|i| i.id() as u64); Ok(proto::FollowResponse { current_view_id, views: this .items(cx) .filter_map(|item| { let id = item.id() as u64; let item = item.to_followed_item_handle(cx)?; let variant = item.to_state_message(cx); Some(proto::View { id, variant: Some(variant), }) }) .collect(), }) }) } async fn handle_unfollow( this: ViewHandle, envelope: TypedEnvelope, _: Arc, cx: AsyncAppContext, ) -> Result<()> { Ok(()) } } impl Entity for Workspace { type Event = (); } impl View for Workspace { fn ui_name() -> &'static str { "Workspace" } fn render(&mut self, cx: &mut RenderContext) -> ElementBox { let theme = cx.global::().theme.clone(); Stack::new() .with_child( Flex::column() .with_child(self.render_titlebar(&theme, cx)) .with_child( Stack::new() .with_child({ let mut content = Flex::row(); content.add_child(self.left_sidebar.render(&theme, cx)); if let Some(element) = self.left_sidebar.render_active_item(&theme, cx) { content.add_child(Flexible::new(0.8, false, element).boxed()); } content.add_child( Flex::column() .with_child( Flexible::new(1., true, self.center.render(&theme)) .boxed(), ) .with_child(ChildView::new(&self.status_bar).boxed()) .flexible(1., true) .boxed(), ); if let Some(element) = self.right_sidebar.render_active_item(&theme, cx) { content.add_child(Flexible::new(0.8, false, element).boxed()); } content.add_child(self.right_sidebar.render(&theme, cx)); content.boxed() }) .with_children(self.modal.as_ref().map(|m| ChildView::new(m).boxed())) .flexible(1.0, true) .boxed(), ) .contained() .with_background_color(theme.workspace.background) .boxed(), ) .with_children(self.render_disconnected_overlay(cx)) .named("workspace") } fn on_focus(&mut self, cx: &mut ViewContext) { cx.focus(&self.active_pane); } } pub trait WorkspaceHandle { fn file_project_paths(&self, cx: &AppContext) -> Vec; } impl WorkspaceHandle for ViewHandle { fn file_project_paths(&self, cx: &AppContext) -> Vec { self.read(cx) .worktrees(cx) .flat_map(|worktree| { let worktree_id = worktree.read(cx).id(); worktree.read(cx).files(true, 0).map(move |f| ProjectPath { worktree_id, path: f.path.clone(), }) }) .collect::>() } } pub struct AvatarRibbon { color: Color, } impl AvatarRibbon { pub fn new(color: Color) -> AvatarRibbon { AvatarRibbon { color } } } impl Element for AvatarRibbon { type LayoutState = (); type PaintState = (); fn layout( &mut self, constraint: gpui::SizeConstraint, _: &mut gpui::LayoutContext, ) -> (gpui::geometry::vector::Vector2F, Self::LayoutState) { (constraint.max, ()) } fn paint( &mut self, bounds: gpui::geometry::rect::RectF, _: gpui::geometry::rect::RectF, _: &mut Self::LayoutState, cx: &mut gpui::PaintContext, ) -> Self::PaintState { let mut path = PathBuilder::new(); path.reset(bounds.lower_left()); path.curve_to( bounds.origin() + vec2f(bounds.height(), 0.), bounds.origin(), ); path.line_to(bounds.upper_right() - vec2f(bounds.height(), 0.)); path.curve_to(bounds.lower_right(), bounds.upper_right()); path.line_to(bounds.lower_left()); cx.scene.push_path(path.build(self.color, None)); } fn dispatch_event( &mut self, _: &gpui::Event, _: gpui::geometry::rect::RectF, _: &mut Self::LayoutState, _: &mut Self::PaintState, _: &mut gpui::EventContext, ) -> bool { false } fn debug( &self, bounds: gpui::geometry::rect::RectF, _: &Self::LayoutState, _: &Self::PaintState, _: &gpui::DebugContext, ) -> gpui::json::Value { json::json!({ "type": "AvatarRibbon", "bounds": bounds.to_json(), "color": self.color.to_json(), }) } } impl std::fmt::Debug for OpenParams { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { f.debug_struct("OpenParams") .field("paths", &self.paths) .finish() } } fn open(action: &Open, cx: &mut MutableAppContext) { let app_state = action.0.clone(); let mut paths = cx.prompt_for_paths(PathPromptOptions { files: true, directories: true, multiple: true, }); cx.spawn(|mut cx| async move { if let Some(paths) = paths.recv().await.flatten() { cx.update(|cx| cx.dispatch_global_action(OpenPaths(OpenParams { paths, app_state }))); } }) .detach(); } pub struct WorkspaceCreated(WeakViewHandle); pub fn open_paths( abs_paths: &[PathBuf], app_state: &Arc, cx: &mut MutableAppContext, ) -> Task> { log::info!("open paths {:?}", abs_paths); // Open paths in existing workspace if possible let mut existing = None; for window_id in cx.window_ids().collect::>() { if let Some(workspace_handle) = cx.root_view::(window_id) { if workspace_handle.update(cx, |workspace, cx| { if workspace.contains_paths(abs_paths, cx.as_ref()) { cx.activate_window(window_id); existing = Some(workspace_handle.clone()); true } else { false } }) { break; } } } let workspace = existing.unwrap_or_else(|| { cx.add_window((app_state.build_window_options)(), |cx| { let project = Project::local( app_state.client.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), cx, ); (app_state.build_workspace)(project, &app_state, cx) }) .1 }); let task = workspace.update(cx, |workspace, cx| workspace.open_paths(abs_paths, cx)); cx.spawn(|_| async move { task.await; workspace }) } pub fn join_project( project_id: u64, app_state: &Arc, cx: &mut MutableAppContext, ) -> Task>> { for window_id in cx.window_ids().collect::>() { if let Some(workspace) = cx.root_view::(window_id) { if workspace.read(cx).project().read(cx).remote_id() == Some(project_id) { return Task::ready(Ok(workspace)); } } } let app_state = app_state.clone(); cx.spawn(|mut cx| async move { let project = Project::remote( project_id, app_state.client.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), &mut cx, ) .await?; Ok(cx.update(|cx| { cx.add_window((app_state.build_window_options)(), |cx| { (app_state.build_workspace)(project, &app_state, cx) }) .1 })) }) } fn open_new(app_state: &Arc, cx: &mut MutableAppContext) { let (window_id, workspace) = cx.add_window((app_state.build_window_options)(), |cx| { let project = Project::local( app_state.client.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), cx, ); (app_state.build_workspace)(project, &app_state, cx) }); cx.dispatch_action(window_id, vec![workspace.id()], &OpenNew(app_state.clone())); }