1pub mod dock;
2pub mod item;
3pub mod notifications;
4pub mod pane;
5pub mod pane_group;
6mod persistence;
7pub mod searchable;
8pub mod shared_screen;
9mod status_bar;
10mod toolbar;
11mod workspace_settings;
12
13use anyhow::{anyhow, Context, Result};
14use call::ActiveCall;
15use client::{
16 proto::{self, PeerId},
17 Client, Status, TypedEnvelope, UserStore,
18};
19use collections::{hash_map, HashMap, HashSet};
20use drag_and_drop::DragAndDrop;
21use futures::{
22 channel::{mpsc, oneshot},
23 future::try_join_all,
24 FutureExt, StreamExt,
25};
26use gpui::{
27 actions,
28 elements::*,
29 geometry::{
30 rect::RectF,
31 vector::{vec2f, Vector2F},
32 },
33 impl_actions,
34 platform::{
35 CursorStyle, ModifiersChangedEvent, MouseButton, PathPromptOptions, Platform, PromptLevel,
36 WindowBounds, WindowOptions,
37 },
38 AnyModelHandle, AnyViewHandle, AnyWeakViewHandle, AnyWindowHandle, AppContext, AsyncAppContext,
39 Entity, ModelContext, ModelHandle, SizeConstraint, Subscription, Task, View, ViewContext,
40 ViewHandle, WeakViewHandle, WindowContext, WindowHandle,
41};
42use item::{FollowableItem, FollowableItemHandle, Item, ItemHandle, ProjectItem};
43use itertools::Itertools;
44use language::{LanguageRegistry, Rope};
45use node_runtime::NodeRuntime;
46use std::{
47 any::TypeId,
48 borrow::Cow,
49 cmp, env,
50 future::Future,
51 path::{Path, PathBuf},
52 rc::Rc,
53 str,
54 sync::{atomic::AtomicUsize, Arc},
55 time::Duration,
56};
57
58use crate::{
59 notifications::{simple_message_notification::MessageNotification, NotificationTracker},
60 persistence::model::{
61 DockData, DockStructure, SerializedPane, SerializedPaneGroup, SerializedWorkspace,
62 },
63};
64use dock::{Dock, DockPosition, Panel, PanelButtons, PanelHandle};
65use lazy_static::lazy_static;
66use notifications::{NotificationHandle, NotifyResultExt};
67pub use pane::*;
68pub use pane_group::*;
69use persistence::{model::SerializedItem, DB};
70pub use persistence::{
71 model::{ItemId, WorkspaceLocation},
72 WorkspaceDb, DB as WORKSPACE_DB,
73};
74use postage::prelude::Stream;
75use project::{Project, ProjectEntryId, ProjectPath, Worktree, WorktreeId};
76use serde::Deserialize;
77use shared_screen::SharedScreen;
78use status_bar::StatusBar;
79pub use status_bar::StatusItemView;
80use theme::{Theme, ThemeSettings};
81pub use toolbar::{ToolbarItemLocation, ToolbarItemView};
82use util::ResultExt;
83pub use workspace_settings::{AutosaveSetting, GitGutterSetting, WorkspaceSettings};
84
85lazy_static! {
86 static ref ZED_WINDOW_SIZE: Option<Vector2F> = env::var("ZED_WINDOW_SIZE")
87 .ok()
88 .as_deref()
89 .and_then(parse_pixel_position_env_var);
90 static ref ZED_WINDOW_POSITION: Option<Vector2F> = env::var("ZED_WINDOW_POSITION")
91 .ok()
92 .as_deref()
93 .and_then(parse_pixel_position_env_var);
94}
95
96pub trait Modal: View {
97 fn has_focus(&self) -> bool;
98 fn dismiss_on_event(event: &Self::Event) -> bool;
99}
100
101trait ModalHandle {
102 fn as_any(&self) -> &AnyViewHandle;
103 fn has_focus(&self, cx: &WindowContext) -> bool;
104}
105
106impl<T: Modal> ModalHandle for ViewHandle<T> {
107 fn as_any(&self) -> &AnyViewHandle {
108 self
109 }
110
111 fn has_focus(&self, cx: &WindowContext) -> bool {
112 self.read(cx).has_focus()
113 }
114}
115
116#[derive(Clone, PartialEq)]
117pub struct RemoveWorktreeFromProject(pub WorktreeId);
118
119actions!(
120 workspace,
121 [
122 Open,
123 NewFile,
124 NewWindow,
125 CloseWindow,
126 CloseInactiveTabsAndPanes,
127 AddFolderToProject,
128 Unfollow,
129 SaveAs,
130 ReloadActiveItem,
131 ActivatePreviousPane,
132 ActivateNextPane,
133 FollowNextCollaborator,
134 NewTerminal,
135 NewCenterTerminal,
136 ToggleTerminalFocus,
137 NewSearch,
138 Feedback,
139 Restart,
140 Welcome,
141 ToggleZoom,
142 ToggleLeftDock,
143 ToggleRightDock,
144 ToggleBottomDock,
145 CloseAllDocks,
146 ]
147);
148
149#[derive(Clone, PartialEq)]
150pub struct OpenPaths {
151 pub paths: Vec<PathBuf>,
152}
153
154#[derive(Clone, Deserialize, PartialEq)]
155pub struct ActivatePane(pub usize);
156
157#[derive(Clone, Deserialize, PartialEq)]
158pub struct ActivatePaneInDirection(pub SplitDirection);
159
160#[derive(Clone, Deserialize, PartialEq)]
161pub struct SwapPaneInDirection(pub SplitDirection);
162
163#[derive(Clone, Deserialize, PartialEq)]
164pub struct NewFileInDirection(pub SplitDirection);
165
166#[derive(Clone, PartialEq, Debug, Deserialize)]
167#[serde(rename_all = "camelCase")]
168pub struct SaveAll {
169 pub save_intent: Option<SaveIntent>,
170}
171
172#[derive(Clone, PartialEq, Debug, Deserialize)]
173#[serde(rename_all = "camelCase")]
174pub struct Save {
175 pub save_intent: Option<SaveIntent>,
176}
177
178#[derive(Clone, PartialEq, Debug, Deserialize, Default)]
179#[serde(rename_all = "camelCase")]
180pub struct CloseAllItemsAndPanes {
181 pub save_intent: Option<SaveIntent>,
182}
183
184#[derive(Deserialize)]
185pub struct Toast {
186 id: usize,
187 msg: Cow<'static, str>,
188 #[serde(skip)]
189 on_click: Option<(Cow<'static, str>, Arc<dyn Fn(&mut WindowContext)>)>,
190}
191
192impl Toast {
193 pub fn new<I: Into<Cow<'static, str>>>(id: usize, msg: I) -> Self {
194 Toast {
195 id,
196 msg: msg.into(),
197 on_click: None,
198 }
199 }
200
201 pub fn on_click<F, M>(mut self, message: M, on_click: F) -> Self
202 where
203 M: Into<Cow<'static, str>>,
204 F: Fn(&mut WindowContext) + 'static,
205 {
206 self.on_click = Some((message.into(), Arc::new(on_click)));
207 self
208 }
209}
210
211impl PartialEq for Toast {
212 fn eq(&self, other: &Self) -> bool {
213 self.id == other.id
214 && self.msg == other.msg
215 && self.on_click.is_some() == other.on_click.is_some()
216 }
217}
218
219impl Clone for Toast {
220 fn clone(&self) -> Self {
221 Toast {
222 id: self.id,
223 msg: self.msg.to_owned(),
224 on_click: self.on_click.clone(),
225 }
226 }
227}
228
229#[derive(Clone, Deserialize, PartialEq)]
230pub struct OpenTerminal {
231 pub working_directory: PathBuf,
232}
233
234impl_actions!(
235 workspace,
236 [
237 ActivatePane,
238 ActivatePaneInDirection,
239 SwapPaneInDirection,
240 NewFileInDirection,
241 Toast,
242 OpenTerminal,
243 SaveAll,
244 Save,
245 CloseAllItemsAndPanes,
246 ]
247);
248
249pub type WorkspaceId = i64;
250
251pub fn init_settings(cx: &mut AppContext) {
252 settings::register::<WorkspaceSettings>(cx);
253 settings::register::<item::ItemSettings>(cx);
254}
255
256pub fn init(app_state: Arc<AppState>, cx: &mut AppContext) {
257 init_settings(cx);
258 pane::init(cx);
259 notifications::init(cx);
260
261 cx.add_global_action({
262 let app_state = Arc::downgrade(&app_state);
263 move |_: &Open, cx: &mut AppContext| {
264 let mut paths = cx.prompt_for_paths(PathPromptOptions {
265 files: true,
266 directories: true,
267 multiple: true,
268 });
269
270 if let Some(app_state) = app_state.upgrade() {
271 cx.spawn(move |mut cx| async move {
272 if let Some(paths) = paths.recv().await.flatten() {
273 cx.update(|cx| {
274 open_paths(&paths, &app_state, None, cx).detach_and_log_err(cx)
275 });
276 }
277 })
278 .detach();
279 }
280 }
281 });
282 cx.add_async_action(Workspace::open);
283
284 cx.add_async_action(Workspace::follow_next_collaborator);
285 cx.add_async_action(Workspace::close);
286 cx.add_async_action(Workspace::close_inactive_items_and_panes);
287 cx.add_async_action(Workspace::close_all_items_and_panes);
288 cx.add_global_action(Workspace::close_global);
289 cx.add_global_action(restart);
290 cx.add_async_action(Workspace::save_all);
291 cx.add_action(Workspace::add_folder_to_project);
292
293 cx.add_action(
294 |workspace: &mut Workspace, _: &Unfollow, cx: &mut ViewContext<Workspace>| {
295 let pane = workspace.active_pane().clone();
296 workspace.unfollow(&pane, cx);
297 },
298 );
299 cx.add_action(
300 |workspace: &mut Workspace, action: &Save, cx: &mut ViewContext<Workspace>| {
301 workspace
302 .save_active_item(action.save_intent.unwrap_or(SaveIntent::Save), cx)
303 .detach_and_log_err(cx);
304 },
305 );
306 cx.add_action(
307 |workspace: &mut Workspace, _: &SaveAs, cx: &mut ViewContext<Workspace>| {
308 workspace
309 .save_active_item(SaveIntent::SaveAs, cx)
310 .detach_and_log_err(cx);
311 },
312 );
313 cx.add_action(|workspace: &mut Workspace, _: &ActivatePreviousPane, cx| {
314 workspace.activate_previous_pane(cx)
315 });
316 cx.add_action(|workspace: &mut Workspace, _: &ActivateNextPane, cx| {
317 workspace.activate_next_pane(cx)
318 });
319
320 cx.add_action(
321 |workspace: &mut Workspace, action: &ActivatePaneInDirection, cx| {
322 workspace.activate_pane_in_direction(action.0, cx)
323 },
324 );
325
326 cx.add_action(
327 |workspace: &mut Workspace, action: &SwapPaneInDirection, cx| {
328 workspace.swap_pane_in_direction(action.0, cx)
329 },
330 );
331
332 cx.add_action(|workspace: &mut Workspace, _: &ToggleLeftDock, cx| {
333 workspace.toggle_dock(DockPosition::Left, cx);
334 });
335 cx.add_action(|workspace: &mut Workspace, _: &ToggleRightDock, cx| {
336 workspace.toggle_dock(DockPosition::Right, cx);
337 });
338 cx.add_action(|workspace: &mut Workspace, _: &ToggleBottomDock, cx| {
339 workspace.toggle_dock(DockPosition::Bottom, cx);
340 });
341 cx.add_action(|workspace: &mut Workspace, _: &CloseAllDocks, cx| {
342 workspace.close_all_docks(cx);
343 });
344 cx.add_action(Workspace::activate_pane_at_index);
345 cx.add_action(|workspace: &mut Workspace, _: &ReopenClosedItem, cx| {
346 workspace.reopen_closed_item(cx).detach();
347 });
348 cx.add_action(|workspace: &mut Workspace, _: &GoBack, cx| {
349 workspace
350 .go_back(workspace.active_pane().downgrade(), cx)
351 .detach();
352 });
353 cx.add_action(|workspace: &mut Workspace, _: &GoForward, cx| {
354 workspace
355 .go_forward(workspace.active_pane().downgrade(), cx)
356 .detach();
357 });
358
359 cx.add_action(|_: &mut Workspace, _: &install_cli::Install, cx| {
360 cx.spawn(|workspace, mut cx| async move {
361 let err = install_cli::install_cli(&cx)
362 .await
363 .context("Failed to create CLI symlink");
364
365 workspace.update(&mut cx, |workspace, cx| {
366 if matches!(err, Err(_)) {
367 err.notify_err(workspace, cx);
368 } else {
369 workspace.show_notification(1, cx, |cx| {
370 cx.add_view(|_| {
371 MessageNotification::new("Successfully installed the `zed` binary")
372 })
373 });
374 }
375 })
376 })
377 .detach();
378 });
379}
380
381type ProjectItemBuilders = HashMap<
382 TypeId,
383 fn(ModelHandle<Project>, AnyModelHandle, &mut ViewContext<Pane>) -> Box<dyn ItemHandle>,
384>;
385pub fn register_project_item<I: ProjectItem>(cx: &mut AppContext) {
386 cx.update_default_global(|builders: &mut ProjectItemBuilders, _| {
387 builders.insert(TypeId::of::<I::Item>(), |project, model, cx| {
388 let item = model.downcast::<I::Item>().unwrap();
389 Box::new(cx.add_view(|cx| I::for_project_item(project, item, cx)))
390 });
391 });
392}
393
394type FollowableItemBuilder = fn(
395 ViewHandle<Pane>,
396 ViewHandle<Workspace>,
397 ViewId,
398 &mut Option<proto::view::Variant>,
399 &mut AppContext,
400) -> Option<Task<Result<Box<dyn FollowableItemHandle>>>>;
401type FollowableItemBuilders = HashMap<
402 TypeId,
403 (
404 FollowableItemBuilder,
405 fn(&AnyViewHandle) -> Box<dyn FollowableItemHandle>,
406 ),
407>;
408pub fn register_followable_item<I: FollowableItem>(cx: &mut AppContext) {
409 cx.update_default_global(|builders: &mut FollowableItemBuilders, _| {
410 builders.insert(
411 TypeId::of::<I>(),
412 (
413 |pane, workspace, id, state, cx| {
414 I::from_state_proto(pane, workspace, id, state, cx).map(|task| {
415 cx.foreground()
416 .spawn(async move { Ok(Box::new(task.await?) as Box<_>) })
417 })
418 },
419 |this| Box::new(this.clone().downcast::<I>().unwrap()),
420 ),
421 );
422 });
423}
424
425type ItemDeserializers = HashMap<
426 Arc<str>,
427 fn(
428 ModelHandle<Project>,
429 WeakViewHandle<Workspace>,
430 WorkspaceId,
431 ItemId,
432 &mut ViewContext<Pane>,
433 ) -> Task<Result<Box<dyn ItemHandle>>>,
434>;
435pub fn register_deserializable_item<I: Item>(cx: &mut AppContext) {
436 cx.update_default_global(|deserializers: &mut ItemDeserializers, _cx| {
437 if let Some(serialized_item_kind) = I::serialized_item_kind() {
438 deserializers.insert(
439 Arc::from(serialized_item_kind),
440 |project, workspace, workspace_id, item_id, cx| {
441 let task = I::deserialize(project, workspace, workspace_id, item_id, cx);
442 cx.foreground()
443 .spawn(async { Ok(Box::new(task.await?) as Box<_>) })
444 },
445 );
446 }
447 });
448}
449
450pub struct AppState {
451 pub languages: Arc<LanguageRegistry>,
452 pub client: Arc<Client>,
453 pub user_store: ModelHandle<UserStore>,
454 pub workspace_store: ModelHandle<WorkspaceStore>,
455 pub fs: Arc<dyn fs::Fs>,
456 pub build_window_options:
457 fn(Option<WindowBounds>, Option<uuid::Uuid>, &dyn Platform) -> WindowOptions<'static>,
458 pub initialize_workspace:
459 fn(WeakViewHandle<Workspace>, bool, Arc<AppState>, AsyncAppContext) -> Task<Result<()>>,
460 pub background_actions: BackgroundActions,
461 pub node_runtime: Arc<dyn NodeRuntime>,
462}
463
464pub struct WorkspaceStore {
465 workspaces: HashSet<WeakViewHandle<Workspace>>,
466 followers: Vec<Follower>,
467 client: Arc<Client>,
468 _subscriptions: Vec<client::Subscription>,
469}
470
471#[derive(PartialEq, Eq, PartialOrd, Ord, Debug)]
472struct Follower {
473 project_id: Option<u64>,
474 peer_id: PeerId,
475}
476
477impl AppState {
478 #[cfg(any(test, feature = "test-support"))]
479 pub fn test(cx: &mut AppContext) -> Arc<Self> {
480 use node_runtime::FakeNodeRuntime;
481 use settings::SettingsStore;
482
483 if !cx.has_global::<SettingsStore>() {
484 cx.set_global(SettingsStore::test(cx));
485 }
486
487 let fs = fs::FakeFs::new(cx.background().clone());
488 let languages = Arc::new(LanguageRegistry::test());
489 let http_client = util::http::FakeHttpClient::with_404_response();
490 let client = Client::new(http_client.clone(), cx);
491 let user_store = cx.add_model(|cx| UserStore::new(client.clone(), http_client, cx));
492 let workspace_store = cx.add_model(|cx| WorkspaceStore::new(client.clone(), cx));
493
494 theme::init((), cx);
495 client::init(&client, cx);
496 crate::init_settings(cx);
497
498 Arc::new(Self {
499 client,
500 fs,
501 languages,
502 user_store,
503 // channel_store,
504 workspace_store,
505 node_runtime: FakeNodeRuntime::new(),
506 initialize_workspace: |_, _, _, _| Task::ready(Ok(())),
507 build_window_options: |_, _, _| Default::default(),
508 background_actions: || &[],
509 })
510 }
511}
512
513struct DelayedDebouncedEditAction {
514 task: Option<Task<()>>,
515 cancel_channel: Option<oneshot::Sender<()>>,
516}
517
518impl DelayedDebouncedEditAction {
519 fn new() -> DelayedDebouncedEditAction {
520 DelayedDebouncedEditAction {
521 task: None,
522 cancel_channel: None,
523 }
524 }
525
526 fn fire_new<F>(&mut self, delay: Duration, cx: &mut ViewContext<Workspace>, func: F)
527 where
528 F: 'static + FnOnce(&mut Workspace, &mut ViewContext<Workspace>) -> Task<Result<()>>,
529 {
530 if let Some(channel) = self.cancel_channel.take() {
531 _ = channel.send(());
532 }
533
534 let (sender, mut receiver) = oneshot::channel::<()>();
535 self.cancel_channel = Some(sender);
536
537 let previous_task = self.task.take();
538 self.task = Some(cx.spawn(|workspace, mut cx| async move {
539 let mut timer = cx.background().timer(delay).fuse();
540 if let Some(previous_task) = previous_task {
541 previous_task.await;
542 }
543
544 futures::select_biased! {
545 _ = receiver => return,
546 _ = timer => {}
547 }
548
549 if let Some(result) = workspace
550 .update(&mut cx, |workspace, cx| (func)(workspace, cx))
551 .log_err()
552 {
553 result.await.log_err();
554 }
555 }));
556 }
557}
558
559pub enum Event {
560 PaneAdded(ViewHandle<Pane>),
561 ContactRequestedJoin(u64),
562}
563
564pub struct Workspace {
565 weak_self: WeakViewHandle<Self>,
566 modal: Option<ActiveModal>,
567 zoomed: Option<AnyWeakViewHandle>,
568 zoomed_position: Option<DockPosition>,
569 center: PaneGroup,
570 left_dock: ViewHandle<Dock>,
571 bottom_dock: ViewHandle<Dock>,
572 right_dock: ViewHandle<Dock>,
573 panes: Vec<ViewHandle<Pane>>,
574 panes_by_item: HashMap<usize, WeakViewHandle<Pane>>,
575 active_pane: ViewHandle<Pane>,
576 last_active_center_pane: Option<WeakViewHandle<Pane>>,
577 last_active_view_id: Option<proto::ViewId>,
578 status_bar: ViewHandle<StatusBar>,
579 titlebar_item: Option<AnyViewHandle>,
580 notifications: Vec<(TypeId, usize, Box<dyn NotificationHandle>)>,
581 project: ModelHandle<Project>,
582 follower_states: HashMap<ViewHandle<Pane>, FollowerState>,
583 last_leaders_by_pane: HashMap<WeakViewHandle<Pane>, PeerId>,
584 window_edited: bool,
585 active_call: Option<(ModelHandle<ActiveCall>, Vec<Subscription>)>,
586 leader_updates_tx: mpsc::UnboundedSender<(PeerId, proto::UpdateFollowers)>,
587 database_id: WorkspaceId,
588 app_state: Arc<AppState>,
589 subscriptions: Vec<Subscription>,
590 _apply_leader_updates: Task<Result<()>>,
591 _observe_current_user: Task<Result<()>>,
592 _schedule_serialize: Option<Task<()>>,
593 pane_history_timestamp: Arc<AtomicUsize>,
594}
595
596struct ActiveModal {
597 view: Box<dyn ModalHandle>,
598 previously_focused_view_id: Option<usize>,
599}
600
601#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
602pub struct ViewId {
603 pub creator: PeerId,
604 pub id: u64,
605}
606
607#[derive(Default)]
608struct FollowerState {
609 leader_id: PeerId,
610 active_view_id: Option<ViewId>,
611 items_by_leader_view_id: HashMap<ViewId, Box<dyn FollowableItemHandle>>,
612}
613
614enum WorkspaceBounds {}
615
616impl Workspace {
617 pub fn new(
618 workspace_id: WorkspaceId,
619 project: ModelHandle<Project>,
620 app_state: Arc<AppState>,
621 cx: &mut ViewContext<Self>,
622 ) -> Self {
623 cx.observe(&project, |_, _, cx| cx.notify()).detach();
624 cx.subscribe(&project, move |this, _, event, cx| {
625 match event {
626 project::Event::RemoteIdChanged(_) => {
627 this.update_window_title(cx);
628 }
629
630 project::Event::CollaboratorLeft(peer_id) => {
631 this.collaborator_left(*peer_id, cx);
632 }
633
634 project::Event::WorktreeRemoved(_) | project::Event::WorktreeAdded => {
635 this.update_window_title(cx);
636 this.serialize_workspace(cx);
637 }
638
639 project::Event::DisconnectedFromHost => {
640 this.update_window_edited(cx);
641 cx.blur();
642 }
643
644 project::Event::Closed => {
645 cx.remove_window();
646 }
647
648 project::Event::DeletedEntry(entry_id) => {
649 for pane in this.panes.iter() {
650 pane.update(cx, |pane, cx| {
651 pane.handle_deleted_project_item(*entry_id, cx)
652 });
653 }
654 }
655
656 project::Event::Notification(message) => this.show_notification(0, cx, |cx| {
657 cx.add_view(|_| MessageNotification::new(message.clone()))
658 }),
659
660 _ => {}
661 }
662 cx.notify()
663 })
664 .detach();
665
666 let weak_handle = cx.weak_handle();
667 let pane_history_timestamp = Arc::new(AtomicUsize::new(0));
668
669 let center_pane = cx.add_view(|cx| {
670 Pane::new(
671 weak_handle.clone(),
672 project.clone(),
673 app_state.background_actions,
674 pane_history_timestamp.clone(),
675 cx,
676 )
677 });
678 cx.subscribe(¢er_pane, Self::handle_pane_event).detach();
679 cx.focus(¢er_pane);
680 cx.emit(Event::PaneAdded(center_pane.clone()));
681
682 app_state.workspace_store.update(cx, |store, _| {
683 store.workspaces.insert(weak_handle.clone());
684 });
685
686 let mut current_user = app_state.user_store.read(cx).watch_current_user();
687 let mut connection_status = app_state.client.status();
688 let _observe_current_user = cx.spawn(|this, mut cx| async move {
689 current_user.recv().await;
690 connection_status.recv().await;
691 let mut stream =
692 Stream::map(current_user, drop).merge(Stream::map(connection_status, drop));
693
694 while stream.recv().await.is_some() {
695 this.update(&mut cx, |_, cx| cx.notify())?;
696 }
697 anyhow::Ok(())
698 });
699
700 // All leader updates are enqueued and then processed in a single task, so
701 // that each asynchronous operation can be run in order.
702 let (leader_updates_tx, mut leader_updates_rx) =
703 mpsc::unbounded::<(PeerId, proto::UpdateFollowers)>();
704 let _apply_leader_updates = cx.spawn(|this, mut cx| async move {
705 while let Some((leader_id, update)) = leader_updates_rx.next().await {
706 Self::process_leader_update(&this, leader_id, update, &mut cx)
707 .await
708 .log_err();
709 }
710
711 Ok(())
712 });
713
714 cx.emit_global(WorkspaceCreated(weak_handle.clone()));
715
716 let left_dock = cx.add_view(|_| Dock::new(DockPosition::Left));
717 let bottom_dock = cx.add_view(|_| Dock::new(DockPosition::Bottom));
718 let right_dock = cx.add_view(|_| Dock::new(DockPosition::Right));
719 let left_dock_buttons =
720 cx.add_view(|cx| PanelButtons::new(left_dock.clone(), weak_handle.clone(), cx));
721 let bottom_dock_buttons =
722 cx.add_view(|cx| PanelButtons::new(bottom_dock.clone(), weak_handle.clone(), cx));
723 let right_dock_buttons =
724 cx.add_view(|cx| PanelButtons::new(right_dock.clone(), weak_handle.clone(), cx));
725 let status_bar = cx.add_view(|cx| {
726 let mut status_bar = StatusBar::new(¢er_pane.clone(), cx);
727 status_bar.add_left_item(left_dock_buttons, cx);
728 status_bar.add_right_item(right_dock_buttons, cx);
729 status_bar.add_right_item(bottom_dock_buttons, cx);
730 status_bar
731 });
732
733 cx.update_default_global::<DragAndDrop<Workspace>, _, _>(|drag_and_drop, _| {
734 drag_and_drop.register_container(weak_handle.clone());
735 });
736
737 let mut active_call = None;
738 if cx.has_global::<ModelHandle<ActiveCall>>() {
739 let call = cx.global::<ModelHandle<ActiveCall>>().clone();
740 let mut subscriptions = Vec::new();
741 subscriptions.push(cx.subscribe(&call, Self::on_active_call_event));
742 active_call = Some((call, subscriptions));
743 }
744
745 let subscriptions = vec![
746 cx.observe_fullscreen(|_, _, cx| cx.notify()),
747 cx.observe_window_activation(Self::on_window_activation_changed),
748 cx.observe_window_bounds(move |_, mut bounds, display, cx| {
749 // Transform fixed bounds to be stored in terms of the containing display
750 if let WindowBounds::Fixed(mut window_bounds) = bounds {
751 if let Some(screen) = cx.platform().screen_by_id(display) {
752 let screen_bounds = screen.bounds();
753 window_bounds
754 .set_origin_x(window_bounds.origin_x() - screen_bounds.origin_x());
755 window_bounds
756 .set_origin_y(window_bounds.origin_y() - screen_bounds.origin_y());
757 bounds = WindowBounds::Fixed(window_bounds);
758 }
759 }
760
761 cx.background()
762 .spawn(DB.set_window_bounds(workspace_id, bounds, display))
763 .detach_and_log_err(cx);
764 }),
765 cx.observe(&left_dock, |this, _, cx| {
766 this.serialize_workspace(cx);
767 cx.notify();
768 }),
769 cx.observe(&bottom_dock, |this, _, cx| {
770 this.serialize_workspace(cx);
771 cx.notify();
772 }),
773 cx.observe(&right_dock, |this, _, cx| {
774 this.serialize_workspace(cx);
775 cx.notify();
776 }),
777 ];
778
779 cx.defer(|this, cx| this.update_window_title(cx));
780 Workspace {
781 weak_self: weak_handle.clone(),
782 modal: None,
783 zoomed: None,
784 zoomed_position: None,
785 center: PaneGroup::new(center_pane.clone()),
786 panes: vec![center_pane.clone()],
787 panes_by_item: Default::default(),
788 active_pane: center_pane.clone(),
789 last_active_center_pane: Some(center_pane.downgrade()),
790 last_active_view_id: None,
791 status_bar,
792 titlebar_item: None,
793 notifications: Default::default(),
794 left_dock,
795 bottom_dock,
796 right_dock,
797 project: project.clone(),
798 follower_states: Default::default(),
799 last_leaders_by_pane: Default::default(),
800 window_edited: false,
801 active_call,
802 database_id: workspace_id,
803 app_state,
804 _observe_current_user,
805 _apply_leader_updates,
806 _schedule_serialize: None,
807 leader_updates_tx,
808 subscriptions,
809 pane_history_timestamp,
810 }
811 }
812
813 fn new_local(
814 abs_paths: Vec<PathBuf>,
815 app_state: Arc<AppState>,
816 requesting_window: Option<WindowHandle<Workspace>>,
817 cx: &mut AppContext,
818 ) -> Task<(
819 WeakViewHandle<Workspace>,
820 Vec<Option<Result<Box<dyn ItemHandle>, anyhow::Error>>>,
821 )> {
822 let project_handle = Project::local(
823 app_state.client.clone(),
824 app_state.node_runtime.clone(),
825 app_state.user_store.clone(),
826 app_state.languages.clone(),
827 app_state.fs.clone(),
828 cx,
829 );
830
831 cx.spawn(|mut cx| async move {
832 let serialized_workspace = persistence::DB.workspace_for_roots(&abs_paths.as_slice());
833
834 let paths_to_open = Arc::new(abs_paths);
835
836 // Get project paths for all of the abs_paths
837 let mut worktree_roots: HashSet<Arc<Path>> = Default::default();
838 let mut project_paths: Vec<(PathBuf, Option<ProjectPath>)> =
839 Vec::with_capacity(paths_to_open.len());
840 for path in paths_to_open.iter().cloned() {
841 if let Some((worktree, project_entry)) = cx
842 .update(|cx| {
843 Workspace::project_path_for_path(project_handle.clone(), &path, true, cx)
844 })
845 .await
846 .log_err()
847 {
848 worktree_roots.insert(worktree.read_with(&mut cx, |tree, _| tree.abs_path()));
849 project_paths.push((path, Some(project_entry)));
850 } else {
851 project_paths.push((path, None));
852 }
853 }
854
855 let workspace_id = if let Some(serialized_workspace) = serialized_workspace.as_ref() {
856 serialized_workspace.id
857 } else {
858 DB.next_id().await.unwrap_or(0)
859 };
860
861 let window = if let Some(window) = requesting_window {
862 window.replace_root(&mut cx, |cx| {
863 Workspace::new(workspace_id, project_handle.clone(), app_state.clone(), cx)
864 });
865 window
866 } else {
867 {
868 let window_bounds_override = window_bounds_env_override(&cx);
869 let (bounds, display) = if let Some(bounds) = window_bounds_override {
870 (Some(bounds), None)
871 } else {
872 serialized_workspace
873 .as_ref()
874 .and_then(|serialized_workspace| {
875 let display = serialized_workspace.display?;
876 let mut bounds = serialized_workspace.bounds?;
877
878 // Stored bounds are relative to the containing display.
879 // So convert back to global coordinates if that screen still exists
880 if let WindowBounds::Fixed(mut window_bounds) = bounds {
881 if let Some(screen) = cx.platform().screen_by_id(display) {
882 let screen_bounds = screen.bounds();
883 window_bounds.set_origin_x(
884 window_bounds.origin_x() + screen_bounds.origin_x(),
885 );
886 window_bounds.set_origin_y(
887 window_bounds.origin_y() + screen_bounds.origin_y(),
888 );
889 bounds = WindowBounds::Fixed(window_bounds);
890 } else {
891 // Screen no longer exists. Return none here.
892 return None;
893 }
894 }
895
896 Some((bounds, display))
897 })
898 .unzip()
899 };
900
901 // Use the serialized workspace to construct the new window
902 cx.add_window(
903 (app_state.build_window_options)(bounds, display, cx.platform().as_ref()),
904 |cx| {
905 Workspace::new(
906 workspace_id,
907 project_handle.clone(),
908 app_state.clone(),
909 cx,
910 )
911 },
912 )
913 }
914 };
915
916 // We haven't yielded the main thread since obtaining the window handle,
917 // so the window exists.
918 let workspace = window.root(&cx).unwrap();
919
920 (app_state.initialize_workspace)(
921 workspace.downgrade(),
922 serialized_workspace.is_some(),
923 app_state.clone(),
924 cx.clone(),
925 )
926 .await
927 .log_err();
928
929 window.update(&mut cx, |cx| cx.activate_window());
930
931 let workspace = workspace.downgrade();
932 notify_if_database_failed(&workspace, &mut cx);
933 let opened_items = open_items(
934 serialized_workspace,
935 &workspace,
936 project_paths,
937 app_state,
938 cx,
939 )
940 .await
941 .unwrap_or_default();
942
943 (workspace, opened_items)
944 })
945 }
946
947 pub fn weak_handle(&self) -> WeakViewHandle<Self> {
948 self.weak_self.clone()
949 }
950
951 pub fn left_dock(&self) -> &ViewHandle<Dock> {
952 &self.left_dock
953 }
954
955 pub fn bottom_dock(&self) -> &ViewHandle<Dock> {
956 &self.bottom_dock
957 }
958
959 pub fn right_dock(&self) -> &ViewHandle<Dock> {
960 &self.right_dock
961 }
962
963 pub fn add_panel<T: Panel>(&mut self, panel: ViewHandle<T>, cx: &mut ViewContext<Self>)
964 where
965 T::Event: std::fmt::Debug,
966 {
967 self.add_panel_with_extra_event_handler(panel, cx, |_, _, _, _| {})
968 }
969
970 pub fn add_panel_with_extra_event_handler<T: Panel, F>(
971 &mut self,
972 panel: ViewHandle<T>,
973 cx: &mut ViewContext<Self>,
974 handler: F,
975 ) where
976 T::Event: std::fmt::Debug,
977 F: Fn(&mut Self, &ViewHandle<T>, &T::Event, &mut ViewContext<Self>) + 'static,
978 {
979 let dock = match panel.position(cx) {
980 DockPosition::Left => &self.left_dock,
981 DockPosition::Bottom => &self.bottom_dock,
982 DockPosition::Right => &self.right_dock,
983 };
984
985 self.subscriptions.push(cx.subscribe(&panel, {
986 let mut dock = dock.clone();
987 let mut prev_position = panel.position(cx);
988 move |this, panel, event, cx| {
989 if T::should_change_position_on_event(event) {
990 let new_position = panel.read(cx).position(cx);
991 let mut was_visible = false;
992 dock.update(cx, |dock, cx| {
993 prev_position = new_position;
994
995 was_visible = dock.is_open()
996 && dock
997 .visible_panel()
998 .map_or(false, |active_panel| active_panel.id() == panel.id());
999 dock.remove_panel(&panel, cx);
1000 });
1001
1002 if panel.is_zoomed(cx) {
1003 this.zoomed_position = Some(new_position);
1004 }
1005
1006 dock = match panel.read(cx).position(cx) {
1007 DockPosition::Left => &this.left_dock,
1008 DockPosition::Bottom => &this.bottom_dock,
1009 DockPosition::Right => &this.right_dock,
1010 }
1011 .clone();
1012 dock.update(cx, |dock, cx| {
1013 dock.add_panel(panel.clone(), cx);
1014 if was_visible {
1015 dock.set_open(true, cx);
1016 dock.activate_panel(dock.panels_len() - 1, cx);
1017 }
1018 });
1019 } else if T::should_zoom_in_on_event(event) {
1020 dock.update(cx, |dock, cx| dock.set_panel_zoomed(&panel, true, cx));
1021 if !panel.has_focus(cx) {
1022 cx.focus(&panel);
1023 }
1024 this.zoomed = Some(panel.downgrade().into_any());
1025 this.zoomed_position = Some(panel.read(cx).position(cx));
1026 } else if T::should_zoom_out_on_event(event) {
1027 dock.update(cx, |dock, cx| dock.set_panel_zoomed(&panel, false, cx));
1028 if this.zoomed_position == Some(prev_position) {
1029 this.zoomed = None;
1030 this.zoomed_position = None;
1031 }
1032 cx.notify();
1033 } else if T::is_focus_event(event) {
1034 let position = panel.read(cx).position(cx);
1035 this.dismiss_zoomed_items_to_reveal(Some(position), cx);
1036 if panel.is_zoomed(cx) {
1037 this.zoomed = Some(panel.downgrade().into_any());
1038 this.zoomed_position = Some(position);
1039 } else {
1040 this.zoomed = None;
1041 this.zoomed_position = None;
1042 }
1043 this.update_active_view_for_followers(cx);
1044 cx.notify();
1045 } else {
1046 handler(this, &panel, event, cx)
1047 }
1048 }
1049 }));
1050
1051 dock.update(cx, |dock, cx| dock.add_panel(panel, cx));
1052 }
1053
1054 pub fn status_bar(&self) -> &ViewHandle<StatusBar> {
1055 &self.status_bar
1056 }
1057
1058 pub fn app_state(&self) -> &Arc<AppState> {
1059 &self.app_state
1060 }
1061
1062 pub fn user_store(&self) -> &ModelHandle<UserStore> {
1063 &self.app_state.user_store
1064 }
1065
1066 pub fn project(&self) -> &ModelHandle<Project> {
1067 &self.project
1068 }
1069
1070 pub fn recent_navigation_history(
1071 &self,
1072 limit: Option<usize>,
1073 cx: &AppContext,
1074 ) -> Vec<(ProjectPath, Option<PathBuf>)> {
1075 let mut abs_paths_opened: HashMap<PathBuf, HashSet<ProjectPath>> = HashMap::default();
1076 let mut history: HashMap<ProjectPath, (Option<PathBuf>, usize)> = HashMap::default();
1077 for pane in &self.panes {
1078 let pane = pane.read(cx);
1079 pane.nav_history()
1080 .for_each_entry(cx, |entry, (project_path, fs_path)| {
1081 if let Some(fs_path) = &fs_path {
1082 abs_paths_opened
1083 .entry(fs_path.clone())
1084 .or_default()
1085 .insert(project_path.clone());
1086 }
1087 let timestamp = entry.timestamp;
1088 match history.entry(project_path) {
1089 hash_map::Entry::Occupied(mut entry) => {
1090 let (_, old_timestamp) = entry.get();
1091 if ×tamp > old_timestamp {
1092 entry.insert((fs_path, timestamp));
1093 }
1094 }
1095 hash_map::Entry::Vacant(entry) => {
1096 entry.insert((fs_path, timestamp));
1097 }
1098 }
1099 });
1100 }
1101
1102 history
1103 .into_iter()
1104 .sorted_by_key(|(_, (_, timestamp))| *timestamp)
1105 .map(|(project_path, (fs_path, _))| (project_path, fs_path))
1106 .rev()
1107 .filter(|(history_path, abs_path)| {
1108 let latest_project_path_opened = abs_path
1109 .as_ref()
1110 .and_then(|abs_path| abs_paths_opened.get(abs_path))
1111 .and_then(|project_paths| {
1112 project_paths
1113 .iter()
1114 .max_by(|b1, b2| b1.worktree_id.cmp(&b2.worktree_id))
1115 });
1116
1117 match latest_project_path_opened {
1118 Some(latest_project_path_opened) => latest_project_path_opened == history_path,
1119 None => true,
1120 }
1121 })
1122 .take(limit.unwrap_or(usize::MAX))
1123 .collect()
1124 }
1125
1126 fn navigate_history(
1127 &mut self,
1128 pane: WeakViewHandle<Pane>,
1129 mode: NavigationMode,
1130 cx: &mut ViewContext<Workspace>,
1131 ) -> Task<Result<()>> {
1132 let to_load = if let Some(pane) = pane.upgrade(cx) {
1133 cx.focus(&pane);
1134
1135 pane.update(cx, |pane, cx| {
1136 loop {
1137 // Retrieve the weak item handle from the history.
1138 let entry = pane.nav_history_mut().pop(mode, cx)?;
1139
1140 // If the item is still present in this pane, then activate it.
1141 if let Some(index) = entry
1142 .item
1143 .upgrade(cx)
1144 .and_then(|v| pane.index_for_item(v.as_ref()))
1145 {
1146 let prev_active_item_index = pane.active_item_index();
1147 pane.nav_history_mut().set_mode(mode);
1148 pane.activate_item(index, true, true, cx);
1149 pane.nav_history_mut().set_mode(NavigationMode::Normal);
1150
1151 let mut navigated = prev_active_item_index != pane.active_item_index();
1152 if let Some(data) = entry.data {
1153 navigated |= pane.active_item()?.navigate(data, cx);
1154 }
1155
1156 if navigated {
1157 break None;
1158 }
1159 }
1160 // If the item is no longer present in this pane, then retrieve its
1161 // project path in order to reopen it.
1162 else {
1163 break pane
1164 .nav_history()
1165 .path_for_item(entry.item.id())
1166 .map(|(project_path, _)| (project_path, entry));
1167 }
1168 }
1169 })
1170 } else {
1171 None
1172 };
1173
1174 if let Some((project_path, entry)) = to_load {
1175 // If the item was no longer present, then load it again from its previous path.
1176 let task = self.load_path(project_path, cx);
1177 cx.spawn(|workspace, mut cx| async move {
1178 let task = task.await;
1179 let mut navigated = false;
1180 if let Some((project_entry_id, build_item)) = task.log_err() {
1181 let prev_active_item_id = pane.update(&mut cx, |pane, _| {
1182 pane.nav_history_mut().set_mode(mode);
1183 pane.active_item().map(|p| p.id())
1184 })?;
1185
1186 pane.update(&mut cx, |pane, cx| {
1187 let item = pane.open_item(project_entry_id, true, cx, build_item);
1188 navigated |= Some(item.id()) != prev_active_item_id;
1189 pane.nav_history_mut().set_mode(NavigationMode::Normal);
1190 if let Some(data) = entry.data {
1191 navigated |= item.navigate(data, cx);
1192 }
1193 })?;
1194 }
1195
1196 if !navigated {
1197 workspace
1198 .update(&mut cx, |workspace, cx| {
1199 Self::navigate_history(workspace, pane, mode, cx)
1200 })?
1201 .await?;
1202 }
1203
1204 Ok(())
1205 })
1206 } else {
1207 Task::ready(Ok(()))
1208 }
1209 }
1210
1211 pub fn go_back(
1212 &mut self,
1213 pane: WeakViewHandle<Pane>,
1214 cx: &mut ViewContext<Workspace>,
1215 ) -> Task<Result<()>> {
1216 self.navigate_history(pane, NavigationMode::GoingBack, cx)
1217 }
1218
1219 pub fn go_forward(
1220 &mut self,
1221 pane: WeakViewHandle<Pane>,
1222 cx: &mut ViewContext<Workspace>,
1223 ) -> Task<Result<()>> {
1224 self.navigate_history(pane, NavigationMode::GoingForward, cx)
1225 }
1226
1227 pub fn reopen_closed_item(&mut self, cx: &mut ViewContext<Workspace>) -> Task<Result<()>> {
1228 self.navigate_history(
1229 self.active_pane().downgrade(),
1230 NavigationMode::ReopeningClosedItem,
1231 cx,
1232 )
1233 }
1234
1235 pub fn client(&self) -> &Client {
1236 &self.app_state.client
1237 }
1238
1239 pub fn set_titlebar_item(&mut self, item: AnyViewHandle, cx: &mut ViewContext<Self>) {
1240 self.titlebar_item = Some(item);
1241 cx.notify();
1242 }
1243
1244 pub fn titlebar_item(&self) -> Option<AnyViewHandle> {
1245 self.titlebar_item.clone()
1246 }
1247
1248 /// Call the given callback with a workspace whose project is local.
1249 ///
1250 /// If the given workspace has a local project, then it will be passed
1251 /// to the callback. Otherwise, a new empty window will be created.
1252 pub fn with_local_workspace<T, F>(
1253 &mut self,
1254 cx: &mut ViewContext<Self>,
1255 callback: F,
1256 ) -> Task<Result<T>>
1257 where
1258 T: 'static,
1259 F: 'static + FnOnce(&mut Workspace, &mut ViewContext<Workspace>) -> T,
1260 {
1261 if self.project.read(cx).is_local() {
1262 Task::Ready(Some(Ok(callback(self, cx))))
1263 } else {
1264 let task = Self::new_local(Vec::new(), self.app_state.clone(), None, cx);
1265 cx.spawn(|_vh, mut cx| async move {
1266 let (workspace, _) = task.await;
1267 workspace.update(&mut cx, callback)
1268 })
1269 }
1270 }
1271
1272 pub fn worktrees<'a>(
1273 &self,
1274 cx: &'a AppContext,
1275 ) -> impl 'a + Iterator<Item = ModelHandle<Worktree>> {
1276 self.project.read(cx).worktrees(cx)
1277 }
1278
1279 pub fn visible_worktrees<'a>(
1280 &self,
1281 cx: &'a AppContext,
1282 ) -> impl 'a + Iterator<Item = ModelHandle<Worktree>> {
1283 self.project.read(cx).visible_worktrees(cx)
1284 }
1285
1286 pub fn worktree_scans_complete(&self, cx: &AppContext) -> impl Future<Output = ()> + 'static {
1287 let futures = self
1288 .worktrees(cx)
1289 .filter_map(|worktree| worktree.read(cx).as_local())
1290 .map(|worktree| worktree.scan_complete())
1291 .collect::<Vec<_>>();
1292 async move {
1293 for future in futures {
1294 future.await;
1295 }
1296 }
1297 }
1298
1299 pub fn close_global(_: &CloseWindow, cx: &mut AppContext) {
1300 cx.spawn(|mut cx| async move {
1301 let window = cx
1302 .windows()
1303 .into_iter()
1304 .find(|window| window.is_active(&cx).unwrap_or(false));
1305 if let Some(window) = window {
1306 //This can only get called when the window's project connection has been lost
1307 //so we don't need to prompt the user for anything and instead just close the window
1308 window.remove(&mut cx);
1309 }
1310 })
1311 .detach();
1312 }
1313
1314 pub fn close(
1315 &mut self,
1316 _: &CloseWindow,
1317 cx: &mut ViewContext<Self>,
1318 ) -> Option<Task<Result<()>>> {
1319 let window = cx.window();
1320 let prepare = self.prepare_to_close(false, cx);
1321 Some(cx.spawn(|_, mut cx| async move {
1322 if prepare.await? {
1323 window.remove(&mut cx);
1324 }
1325 Ok(())
1326 }))
1327 }
1328
1329 pub fn prepare_to_close(
1330 &mut self,
1331 quitting: bool,
1332 cx: &mut ViewContext<Self>,
1333 ) -> Task<Result<bool>> {
1334 let active_call = self.active_call().cloned();
1335 let window = cx.window();
1336
1337 cx.spawn(|this, mut cx| async move {
1338 let workspace_count = cx
1339 .windows()
1340 .into_iter()
1341 .filter(|window| window.root_is::<Workspace>())
1342 .count();
1343
1344 if let Some(active_call) = active_call {
1345 if !quitting
1346 && workspace_count == 1
1347 && active_call.read_with(&cx, |call, _| call.room().is_some())
1348 {
1349 let answer = window.prompt(
1350 PromptLevel::Warning,
1351 "Do you want to leave the current call?",
1352 &["Close window and hang up", "Cancel"],
1353 &mut cx,
1354 );
1355
1356 if let Some(mut answer) = answer {
1357 if answer.next().await == Some(1) {
1358 return anyhow::Ok(false);
1359 } else {
1360 active_call
1361 .update(&mut cx, |call, cx| call.hang_up(cx))
1362 .await
1363 .log_err();
1364 }
1365 }
1366 }
1367 }
1368
1369 Ok(this
1370 .update(&mut cx, |this, cx| {
1371 this.save_all_internal(SaveIntent::Close, cx)
1372 })?
1373 .await?)
1374 })
1375 }
1376
1377 fn save_all(
1378 &mut self,
1379 action: &SaveAll,
1380 cx: &mut ViewContext<Self>,
1381 ) -> Option<Task<Result<()>>> {
1382 let save_all =
1383 self.save_all_internal(action.save_intent.unwrap_or(SaveIntent::SaveAll), cx);
1384 Some(cx.foreground().spawn(async move {
1385 save_all.await?;
1386 Ok(())
1387 }))
1388 }
1389
1390 fn save_all_internal(
1391 &mut self,
1392 mut save_intent: SaveIntent,
1393 cx: &mut ViewContext<Self>,
1394 ) -> Task<Result<bool>> {
1395 if self.project.read(cx).is_read_only() {
1396 return Task::ready(Ok(true));
1397 }
1398 let dirty_items = self
1399 .panes
1400 .iter()
1401 .flat_map(|pane| {
1402 pane.read(cx).items().filter_map(|item| {
1403 if item.is_dirty(cx) {
1404 Some((pane.downgrade(), item.boxed_clone()))
1405 } else {
1406 None
1407 }
1408 })
1409 })
1410 .collect::<Vec<_>>();
1411
1412 let project = self.project.clone();
1413 cx.spawn(|workspace, mut cx| async move {
1414 // Override save mode and display "Save all files" prompt
1415 if save_intent == SaveIntent::Close && dirty_items.len() > 1 {
1416 let mut answer = workspace.update(&mut cx, |_, cx| {
1417 let prompt = Pane::file_names_for_prompt(
1418 &mut dirty_items.iter().map(|(_, handle)| handle),
1419 dirty_items.len(),
1420 cx,
1421 );
1422 cx.prompt(
1423 PromptLevel::Warning,
1424 &prompt,
1425 &["Save all", "Discard all", "Cancel"],
1426 )
1427 })?;
1428 match answer.next().await {
1429 Some(0) => save_intent = SaveIntent::SaveAll,
1430 Some(1) => save_intent = SaveIntent::Skip,
1431 _ => {}
1432 }
1433 }
1434 for (pane, item) in dirty_items {
1435 let (singleton, project_entry_ids) =
1436 cx.read(|cx| (item.is_singleton(cx), item.project_entry_ids(cx)));
1437 if singleton || !project_entry_ids.is_empty() {
1438 if let Some(ix) =
1439 pane.read_with(&cx, |pane, _| pane.index_for_item(item.as_ref()))?
1440 {
1441 if !Pane::save_item(
1442 project.clone(),
1443 &pane,
1444 ix,
1445 &*item,
1446 save_intent,
1447 &mut cx,
1448 )
1449 .await?
1450 {
1451 return Ok(false);
1452 }
1453 }
1454 }
1455 }
1456 Ok(true)
1457 })
1458 }
1459
1460 pub fn open(&mut self, _: &Open, cx: &mut ViewContext<Self>) -> Option<Task<Result<()>>> {
1461 let mut paths = cx.prompt_for_paths(PathPromptOptions {
1462 files: true,
1463 directories: true,
1464 multiple: true,
1465 });
1466
1467 Some(cx.spawn(|this, mut cx| async move {
1468 if let Some(paths) = paths.recv().await.flatten() {
1469 if let Some(task) = this
1470 .update(&mut cx, |this, cx| this.open_workspace_for_paths(paths, cx))
1471 .log_err()
1472 {
1473 task.await?
1474 }
1475 }
1476 Ok(())
1477 }))
1478 }
1479
1480 pub fn open_workspace_for_paths(
1481 &mut self,
1482 paths: Vec<PathBuf>,
1483 cx: &mut ViewContext<Self>,
1484 ) -> Task<Result<()>> {
1485 let window = cx.window().downcast::<Self>();
1486 let is_remote = self.project.read(cx).is_remote();
1487 let has_worktree = self.project.read(cx).worktrees(cx).next().is_some();
1488 let has_dirty_items = self.items(cx).any(|item| item.is_dirty(cx));
1489 let close_task = if is_remote || has_worktree || has_dirty_items {
1490 None
1491 } else {
1492 Some(self.prepare_to_close(false, cx))
1493 };
1494 let app_state = self.app_state.clone();
1495
1496 cx.spawn(|_, mut cx| async move {
1497 let window_to_replace = if let Some(close_task) = close_task {
1498 if !close_task.await? {
1499 return Ok(());
1500 }
1501 window
1502 } else {
1503 None
1504 };
1505 cx.update(|cx| open_paths(&paths, &app_state, window_to_replace, cx))
1506 .await?;
1507 Ok(())
1508 })
1509 }
1510
1511 #[allow(clippy::type_complexity)]
1512 pub fn open_paths(
1513 &mut self,
1514 mut abs_paths: Vec<PathBuf>,
1515 visible: bool,
1516 cx: &mut ViewContext<Self>,
1517 ) -> Task<Vec<Option<Result<Box<dyn ItemHandle>, anyhow::Error>>>> {
1518 log::info!("open paths {:?}", abs_paths);
1519
1520 let fs = self.app_state.fs.clone();
1521
1522 // Sort the paths to ensure we add worktrees for parents before their children.
1523 abs_paths.sort_unstable();
1524 cx.spawn(|this, mut cx| async move {
1525 let mut tasks = Vec::with_capacity(abs_paths.len());
1526 for abs_path in &abs_paths {
1527 let project_path = match this
1528 .update(&mut cx, |this, cx| {
1529 Workspace::project_path_for_path(
1530 this.project.clone(),
1531 abs_path,
1532 visible,
1533 cx,
1534 )
1535 })
1536 .log_err()
1537 {
1538 Some(project_path) => project_path.await.log_err(),
1539 None => None,
1540 };
1541
1542 let this = this.clone();
1543 let task = cx.spawn(|mut cx| {
1544 let fs = fs.clone();
1545 let abs_path = abs_path.clone();
1546 async move {
1547 let (worktree, project_path) = project_path?;
1548 if fs.is_file(&abs_path).await {
1549 Some(
1550 this.update(&mut cx, |this, cx| {
1551 this.open_path(project_path, None, true, cx)
1552 })
1553 .log_err()?
1554 .await,
1555 )
1556 } else {
1557 this.update(&mut cx, |workspace, cx| {
1558 let worktree = worktree.read(cx);
1559 let worktree_abs_path = worktree.abs_path();
1560 let entry_id = if abs_path == worktree_abs_path.as_ref() {
1561 worktree.root_entry()
1562 } else {
1563 abs_path
1564 .strip_prefix(worktree_abs_path.as_ref())
1565 .ok()
1566 .and_then(|relative_path| {
1567 worktree.entry_for_path(relative_path)
1568 })
1569 }
1570 .map(|entry| entry.id);
1571 if let Some(entry_id) = entry_id {
1572 workspace.project().update(cx, |_, cx| {
1573 cx.emit(project::Event::ActiveEntryChanged(Some(entry_id)));
1574 })
1575 }
1576 })
1577 .log_err()?;
1578 None
1579 }
1580 }
1581 });
1582 tasks.push(task);
1583 }
1584
1585 futures::future::join_all(tasks).await
1586 })
1587 }
1588
1589 fn add_folder_to_project(&mut self, _: &AddFolderToProject, cx: &mut ViewContext<Self>) {
1590 let mut paths = cx.prompt_for_paths(PathPromptOptions {
1591 files: false,
1592 directories: true,
1593 multiple: true,
1594 });
1595 cx.spawn(|this, mut cx| async move {
1596 if let Some(paths) = paths.recv().await.flatten() {
1597 let results = this
1598 .update(&mut cx, |this, cx| this.open_paths(paths, true, cx))?
1599 .await;
1600 for result in results.into_iter().flatten() {
1601 result.log_err();
1602 }
1603 }
1604 anyhow::Ok(())
1605 })
1606 .detach_and_log_err(cx);
1607 }
1608
1609 fn project_path_for_path(
1610 project: ModelHandle<Project>,
1611 abs_path: &Path,
1612 visible: bool,
1613 cx: &mut AppContext,
1614 ) -> Task<Result<(ModelHandle<Worktree>, ProjectPath)>> {
1615 let entry = project.update(cx, |project, cx| {
1616 project.find_or_create_local_worktree(abs_path, visible, cx)
1617 });
1618 cx.spawn(|cx| async move {
1619 let (worktree, path) = entry.await?;
1620 let worktree_id = worktree.read_with(&cx, |t, _| t.id());
1621 Ok((
1622 worktree,
1623 ProjectPath {
1624 worktree_id,
1625 path: path.into(),
1626 },
1627 ))
1628 })
1629 }
1630
1631 /// Returns the modal that was toggled closed if it was open.
1632 pub fn toggle_modal<V, F>(
1633 &mut self,
1634 cx: &mut ViewContext<Self>,
1635 add_view: F,
1636 ) -> Option<ViewHandle<V>>
1637 where
1638 V: 'static + Modal,
1639 F: FnOnce(&mut Self, &mut ViewContext<Self>) -> ViewHandle<V>,
1640 {
1641 cx.notify();
1642 // Whatever modal was visible is getting clobbered. If its the same type as V, then return
1643 // it. Otherwise, create a new modal and set it as active.
1644 if let Some(already_open_modal) = self
1645 .dismiss_modal(cx)
1646 .and_then(|modal| modal.downcast::<V>())
1647 {
1648 cx.focus_self();
1649 Some(already_open_modal)
1650 } else {
1651 let modal = add_view(self, cx);
1652 cx.subscribe(&modal, |this, _, event, cx| {
1653 if V::dismiss_on_event(event) {
1654 this.dismiss_modal(cx);
1655 }
1656 })
1657 .detach();
1658 let previously_focused_view_id = cx.focused_view_id();
1659 cx.focus(&modal);
1660 self.modal = Some(ActiveModal {
1661 view: Box::new(modal),
1662 previously_focused_view_id,
1663 });
1664 None
1665 }
1666 }
1667
1668 pub fn modal<V: 'static + View>(&self) -> Option<ViewHandle<V>> {
1669 self.modal
1670 .as_ref()
1671 .and_then(|modal| modal.view.as_any().clone().downcast::<V>())
1672 }
1673
1674 pub fn dismiss_modal(&mut self, cx: &mut ViewContext<Self>) -> Option<AnyViewHandle> {
1675 if let Some(modal) = self.modal.take() {
1676 if let Some(previously_focused_view_id) = modal.previously_focused_view_id {
1677 if modal.view.has_focus(cx) {
1678 cx.window_context().focus(Some(previously_focused_view_id));
1679 }
1680 }
1681 cx.notify();
1682 Some(modal.view.as_any().clone())
1683 } else {
1684 None
1685 }
1686 }
1687
1688 pub fn items<'a>(
1689 &'a self,
1690 cx: &'a AppContext,
1691 ) -> impl 'a + Iterator<Item = &Box<dyn ItemHandle>> {
1692 self.panes.iter().flat_map(|pane| pane.read(cx).items())
1693 }
1694
1695 pub fn item_of_type<T: Item>(&self, cx: &AppContext) -> Option<ViewHandle<T>> {
1696 self.items_of_type(cx).max_by_key(|item| item.id())
1697 }
1698
1699 pub fn items_of_type<'a, T: Item>(
1700 &'a self,
1701 cx: &'a AppContext,
1702 ) -> impl 'a + Iterator<Item = ViewHandle<T>> {
1703 self.panes
1704 .iter()
1705 .flat_map(|pane| pane.read(cx).items_of_type())
1706 }
1707
1708 pub fn active_item(&self, cx: &AppContext) -> Option<Box<dyn ItemHandle>> {
1709 self.active_pane().read(cx).active_item()
1710 }
1711
1712 fn active_project_path(&self, cx: &ViewContext<Self>) -> Option<ProjectPath> {
1713 self.active_item(cx).and_then(|item| item.project_path(cx))
1714 }
1715
1716 pub fn save_active_item(
1717 &mut self,
1718 save_intent: SaveIntent,
1719 cx: &mut ViewContext<Self>,
1720 ) -> Task<Result<()>> {
1721 let project = self.project.clone();
1722 let pane = self.active_pane();
1723 let item_ix = pane.read(cx).active_item_index();
1724 let item = pane.read(cx).active_item();
1725 let pane = pane.downgrade();
1726
1727 cx.spawn(|_, mut cx| async move {
1728 if let Some(item) = item {
1729 Pane::save_item(project, &pane, item_ix, item.as_ref(), save_intent, &mut cx)
1730 .await
1731 .map(|_| ())
1732 } else {
1733 Ok(())
1734 }
1735 })
1736 }
1737
1738 pub fn close_inactive_items_and_panes(
1739 &mut self,
1740 _: &CloseInactiveTabsAndPanes,
1741 cx: &mut ViewContext<Self>,
1742 ) -> Option<Task<Result<()>>> {
1743 self.close_all_internal(true, SaveIntent::Close, cx)
1744 }
1745
1746 pub fn close_all_items_and_panes(
1747 &mut self,
1748 action: &CloseAllItemsAndPanes,
1749 cx: &mut ViewContext<Self>,
1750 ) -> Option<Task<Result<()>>> {
1751 self.close_all_internal(false, action.save_intent.unwrap_or(SaveIntent::Close), cx)
1752 }
1753
1754 fn close_all_internal(
1755 &mut self,
1756 retain_active_pane: bool,
1757 save_intent: SaveIntent,
1758 cx: &mut ViewContext<Self>,
1759 ) -> Option<Task<Result<()>>> {
1760 let current_pane = self.active_pane();
1761
1762 let mut tasks = Vec::new();
1763
1764 if retain_active_pane {
1765 if let Some(current_pane_close) = current_pane.update(cx, |pane, cx| {
1766 pane.close_inactive_items(&CloseInactiveItems, cx)
1767 }) {
1768 tasks.push(current_pane_close);
1769 };
1770 }
1771
1772 for pane in self.panes() {
1773 if retain_active_pane && pane.id() == current_pane.id() {
1774 continue;
1775 }
1776
1777 if let Some(close_pane_items) = pane.update(cx, |pane: &mut Pane, cx| {
1778 pane.close_all_items(
1779 &CloseAllItems {
1780 save_intent: Some(save_intent),
1781 },
1782 cx,
1783 )
1784 }) {
1785 tasks.push(close_pane_items)
1786 }
1787 }
1788
1789 if tasks.is_empty() {
1790 None
1791 } else {
1792 Some(cx.spawn(|_, _| async move {
1793 for task in tasks {
1794 task.await?
1795 }
1796 Ok(())
1797 }))
1798 }
1799 }
1800
1801 pub fn toggle_dock(&mut self, dock_side: DockPosition, cx: &mut ViewContext<Self>) {
1802 let dock = match dock_side {
1803 DockPosition::Left => &self.left_dock,
1804 DockPosition::Bottom => &self.bottom_dock,
1805 DockPosition::Right => &self.right_dock,
1806 };
1807 let mut focus_center = false;
1808 let mut reveal_dock = false;
1809 dock.update(cx, |dock, cx| {
1810 let other_is_zoomed = self.zoomed.is_some() && self.zoomed_position != Some(dock_side);
1811 let was_visible = dock.is_open() && !other_is_zoomed;
1812 dock.set_open(!was_visible, cx);
1813
1814 if let Some(active_panel) = dock.active_panel() {
1815 if was_visible {
1816 if active_panel.has_focus(cx) {
1817 focus_center = true;
1818 }
1819 } else {
1820 cx.focus(active_panel.as_any());
1821 reveal_dock = true;
1822 }
1823 }
1824 });
1825
1826 if reveal_dock {
1827 self.dismiss_zoomed_items_to_reveal(Some(dock_side), cx);
1828 }
1829
1830 if focus_center {
1831 cx.focus_self();
1832 }
1833
1834 cx.notify();
1835 self.serialize_workspace(cx);
1836 }
1837
1838 pub fn close_all_docks(&mut self, cx: &mut ViewContext<Self>) {
1839 let docks = [&self.left_dock, &self.bottom_dock, &self.right_dock];
1840
1841 for dock in docks {
1842 dock.update(cx, |dock, cx| {
1843 dock.set_open(false, cx);
1844 });
1845 }
1846
1847 cx.focus_self();
1848 cx.notify();
1849 self.serialize_workspace(cx);
1850 }
1851
1852 /// Transfer focus to the panel of the given type.
1853 pub fn focus_panel<T: Panel>(&mut self, cx: &mut ViewContext<Self>) -> Option<ViewHandle<T>> {
1854 self.focus_or_unfocus_panel::<T>(cx, |_, _| true)?
1855 .as_any()
1856 .clone()
1857 .downcast()
1858 }
1859
1860 /// Focus the panel of the given type if it isn't already focused. If it is
1861 /// already focused, then transfer focus back to the workspace center.
1862 pub fn toggle_panel_focus<T: Panel>(&mut self, cx: &mut ViewContext<Self>) {
1863 self.focus_or_unfocus_panel::<T>(cx, |panel, cx| !panel.has_focus(cx));
1864 }
1865
1866 /// Focus or unfocus the given panel type, depending on the given callback.
1867 fn focus_or_unfocus_panel<T: Panel>(
1868 &mut self,
1869 cx: &mut ViewContext<Self>,
1870 should_focus: impl Fn(&dyn PanelHandle, &mut ViewContext<Dock>) -> bool,
1871 ) -> Option<Rc<dyn PanelHandle>> {
1872 for dock in [&self.left_dock, &self.bottom_dock, &self.right_dock] {
1873 if let Some(panel_index) = dock.read(cx).panel_index_for_type::<T>() {
1874 let mut focus_center = false;
1875 let mut reveal_dock = false;
1876 let panel = dock.update(cx, |dock, cx| {
1877 dock.activate_panel(panel_index, cx);
1878
1879 let panel = dock.active_panel().cloned();
1880 if let Some(panel) = panel.as_ref() {
1881 if should_focus(&**panel, cx) {
1882 dock.set_open(true, cx);
1883 cx.focus(panel.as_any());
1884 reveal_dock = true;
1885 } else {
1886 // if panel.is_zoomed(cx) {
1887 // dock.set_open(false, cx);
1888 // }
1889 focus_center = true;
1890 }
1891 }
1892 panel
1893 });
1894
1895 if focus_center {
1896 cx.focus_self();
1897 }
1898
1899 self.serialize_workspace(cx);
1900 cx.notify();
1901 return panel;
1902 }
1903 }
1904 None
1905 }
1906
1907 pub fn panel<T: Panel>(&self, cx: &WindowContext) -> Option<ViewHandle<T>> {
1908 for dock in [&self.left_dock, &self.bottom_dock, &self.right_dock] {
1909 let dock = dock.read(cx);
1910 if let Some(panel) = dock.panel::<T>() {
1911 return Some(panel);
1912 }
1913 }
1914 None
1915 }
1916
1917 fn zoom_out(&mut self, cx: &mut ViewContext<Self>) {
1918 for pane in &self.panes {
1919 pane.update(cx, |pane, cx| pane.set_zoomed(false, cx));
1920 }
1921
1922 self.left_dock.update(cx, |dock, cx| dock.zoom_out(cx));
1923 self.bottom_dock.update(cx, |dock, cx| dock.zoom_out(cx));
1924 self.right_dock.update(cx, |dock, cx| dock.zoom_out(cx));
1925 self.zoomed = None;
1926 self.zoomed_position = None;
1927
1928 cx.notify();
1929 }
1930
1931 #[cfg(any(test, feature = "test-support"))]
1932 pub fn zoomed_view(&self, cx: &AppContext) -> Option<AnyViewHandle> {
1933 self.zoomed.and_then(|view| view.upgrade(cx))
1934 }
1935
1936 fn dismiss_zoomed_items_to_reveal(
1937 &mut self,
1938 dock_to_reveal: Option<DockPosition>,
1939 cx: &mut ViewContext<Self>,
1940 ) {
1941 // If a center pane is zoomed, unzoom it.
1942 for pane in &self.panes {
1943 if pane != &self.active_pane || dock_to_reveal.is_some() {
1944 pane.update(cx, |pane, cx| pane.set_zoomed(false, cx));
1945 }
1946 }
1947
1948 // If another dock is zoomed, hide it.
1949 let mut focus_center = false;
1950 for dock in [&self.left_dock, &self.right_dock, &self.bottom_dock] {
1951 dock.update(cx, |dock, cx| {
1952 if Some(dock.position()) != dock_to_reveal {
1953 if let Some(panel) = dock.active_panel() {
1954 if panel.is_zoomed(cx) {
1955 focus_center |= panel.has_focus(cx);
1956 dock.set_open(false, cx);
1957 }
1958 }
1959 }
1960 });
1961 }
1962
1963 if focus_center {
1964 cx.focus_self();
1965 }
1966
1967 if self.zoomed_position != dock_to_reveal {
1968 self.zoomed = None;
1969 self.zoomed_position = None;
1970 }
1971
1972 cx.notify();
1973 }
1974
1975 fn add_pane(&mut self, cx: &mut ViewContext<Self>) -> ViewHandle<Pane> {
1976 let pane = cx.add_view(|cx| {
1977 Pane::new(
1978 self.weak_handle(),
1979 self.project.clone(),
1980 self.app_state.background_actions,
1981 self.pane_history_timestamp.clone(),
1982 cx,
1983 )
1984 });
1985 cx.subscribe(&pane, Self::handle_pane_event).detach();
1986 self.panes.push(pane.clone());
1987 cx.focus(&pane);
1988 cx.emit(Event::PaneAdded(pane.clone()));
1989 pane
1990 }
1991
1992 pub fn add_item_to_center(
1993 &mut self,
1994 item: Box<dyn ItemHandle>,
1995 cx: &mut ViewContext<Self>,
1996 ) -> bool {
1997 if let Some(center_pane) = self.last_active_center_pane.clone() {
1998 if let Some(center_pane) = center_pane.upgrade(cx) {
1999 center_pane.update(cx, |pane, cx| pane.add_item(item, true, true, None, cx));
2000 true
2001 } else {
2002 false
2003 }
2004 } else {
2005 false
2006 }
2007 }
2008
2009 pub fn add_item(&mut self, item: Box<dyn ItemHandle>, cx: &mut ViewContext<Self>) {
2010 self.active_pane
2011 .update(cx, |pane, cx| pane.add_item(item, true, true, None, cx));
2012 }
2013
2014 pub fn split_item(
2015 &mut self,
2016 split_direction: SplitDirection,
2017 item: Box<dyn ItemHandle>,
2018 cx: &mut ViewContext<Self>,
2019 ) {
2020 let new_pane = self.split_pane(self.active_pane.clone(), split_direction, cx);
2021 new_pane.update(cx, move |new_pane, cx| {
2022 new_pane.add_item(item, true, true, None, cx)
2023 })
2024 }
2025
2026 pub fn open_abs_path(
2027 &mut self,
2028 abs_path: PathBuf,
2029 visible: bool,
2030 cx: &mut ViewContext<Self>,
2031 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
2032 cx.spawn(|workspace, mut cx| async move {
2033 let open_paths_task_result = workspace
2034 .update(&mut cx, |workspace, cx| {
2035 workspace.open_paths(vec![abs_path.clone()], visible, cx)
2036 })
2037 .with_context(|| format!("open abs path {abs_path:?} task spawn"))?
2038 .await;
2039 anyhow::ensure!(
2040 open_paths_task_result.len() == 1,
2041 "open abs path {abs_path:?} task returned incorrect number of results"
2042 );
2043 match open_paths_task_result
2044 .into_iter()
2045 .next()
2046 .expect("ensured single task result")
2047 {
2048 Some(open_result) => {
2049 open_result.with_context(|| format!("open abs path {abs_path:?} task join"))
2050 }
2051 None => anyhow::bail!("open abs path {abs_path:?} task returned None"),
2052 }
2053 })
2054 }
2055
2056 pub fn split_abs_path(
2057 &mut self,
2058 abs_path: PathBuf,
2059 visible: bool,
2060 cx: &mut ViewContext<Self>,
2061 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
2062 let project_path_task =
2063 Workspace::project_path_for_path(self.project.clone(), &abs_path, visible, cx);
2064 cx.spawn(|this, mut cx| async move {
2065 let (_, path) = project_path_task.await?;
2066 this.update(&mut cx, |this, cx| this.split_path(path, cx))?
2067 .await
2068 })
2069 }
2070
2071 pub fn open_path(
2072 &mut self,
2073 path: impl Into<ProjectPath>,
2074 pane: Option<WeakViewHandle<Pane>>,
2075 focus_item: bool,
2076 cx: &mut ViewContext<Self>,
2077 ) -> Task<Result<Box<dyn ItemHandle>, anyhow::Error>> {
2078 let pane = pane.unwrap_or_else(|| {
2079 self.last_active_center_pane.clone().unwrap_or_else(|| {
2080 self.panes
2081 .first()
2082 .expect("There must be an active pane")
2083 .downgrade()
2084 })
2085 });
2086
2087 let task = self.load_path(path.into(), cx);
2088 cx.spawn(|_, mut cx| async move {
2089 let (project_entry_id, build_item) = task.await?;
2090 pane.update(&mut cx, |pane, cx| {
2091 pane.open_item(project_entry_id, focus_item, cx, build_item)
2092 })
2093 })
2094 }
2095
2096 pub fn split_path(
2097 &mut self,
2098 path: impl Into<ProjectPath>,
2099 cx: &mut ViewContext<Self>,
2100 ) -> Task<Result<Box<dyn ItemHandle>, anyhow::Error>> {
2101 let pane = self.last_active_center_pane.clone().unwrap_or_else(|| {
2102 self.panes
2103 .first()
2104 .expect("There must be an active pane")
2105 .downgrade()
2106 });
2107
2108 if let Member::Pane(center_pane) = &self.center.root {
2109 if center_pane.read(cx).items_len() == 0 {
2110 return self.open_path(path, Some(pane), true, cx);
2111 }
2112 }
2113
2114 let task = self.load_path(path.into(), cx);
2115 cx.spawn(|this, mut cx| async move {
2116 let (project_entry_id, build_item) = task.await?;
2117 this.update(&mut cx, move |this, cx| -> Option<_> {
2118 let pane = pane.upgrade(cx)?;
2119 let new_pane = this.split_pane(pane, SplitDirection::Right, cx);
2120 new_pane.update(cx, |new_pane, cx| {
2121 Some(new_pane.open_item(project_entry_id, true, cx, build_item))
2122 })
2123 })
2124 .map(|option| option.ok_or_else(|| anyhow!("pane was dropped")))?
2125 })
2126 }
2127
2128 pub(crate) fn load_path(
2129 &mut self,
2130 path: ProjectPath,
2131 cx: &mut ViewContext<Self>,
2132 ) -> Task<
2133 Result<(
2134 ProjectEntryId,
2135 impl 'static + FnOnce(&mut ViewContext<Pane>) -> Box<dyn ItemHandle>,
2136 )>,
2137 > {
2138 let project = self.project().clone();
2139 let project_item = project.update(cx, |project, cx| project.open_path(path, cx));
2140 cx.spawn(|_, mut cx| async move {
2141 let (project_entry_id, project_item) = project_item.await?;
2142 let build_item = cx.update(|cx| {
2143 cx.default_global::<ProjectItemBuilders>()
2144 .get(&project_item.model_type())
2145 .ok_or_else(|| anyhow!("no item builder for project item"))
2146 .cloned()
2147 })?;
2148 let build_item =
2149 move |cx: &mut ViewContext<Pane>| build_item(project, project_item, cx);
2150 Ok((project_entry_id, build_item))
2151 })
2152 }
2153
2154 pub fn open_project_item<T>(
2155 &mut self,
2156 project_item: ModelHandle<T::Item>,
2157 cx: &mut ViewContext<Self>,
2158 ) -> ViewHandle<T>
2159 where
2160 T: ProjectItem,
2161 {
2162 use project::Item as _;
2163
2164 let entry_id = project_item.read(cx).entry_id(cx);
2165 if let Some(item) = entry_id
2166 .and_then(|entry_id| self.active_pane().read(cx).item_for_entry(entry_id, cx))
2167 .and_then(|item| item.downcast())
2168 {
2169 self.activate_item(&item, cx);
2170 return item;
2171 }
2172
2173 let item = cx.add_view(|cx| T::for_project_item(self.project().clone(), project_item, cx));
2174 self.add_item(Box::new(item.clone()), cx);
2175 item
2176 }
2177
2178 pub fn split_project_item<T>(
2179 &mut self,
2180 project_item: ModelHandle<T::Item>,
2181 cx: &mut ViewContext<Self>,
2182 ) -> ViewHandle<T>
2183 where
2184 T: ProjectItem,
2185 {
2186 use project::Item as _;
2187
2188 let entry_id = project_item.read(cx).entry_id(cx);
2189 if let Some(item) = entry_id
2190 .and_then(|entry_id| self.active_pane().read(cx).item_for_entry(entry_id, cx))
2191 .and_then(|item| item.downcast())
2192 {
2193 self.activate_item(&item, cx);
2194 return item;
2195 }
2196
2197 let item = cx.add_view(|cx| T::for_project_item(self.project().clone(), project_item, cx));
2198 self.split_item(SplitDirection::Right, Box::new(item.clone()), cx);
2199 item
2200 }
2201
2202 pub fn open_shared_screen(&mut self, peer_id: PeerId, cx: &mut ViewContext<Self>) {
2203 if let Some(shared_screen) = self.shared_screen_for_peer(peer_id, &self.active_pane, cx) {
2204 self.active_pane.update(cx, |pane, cx| {
2205 pane.add_item(Box::new(shared_screen), false, true, None, cx)
2206 });
2207 }
2208 }
2209
2210 pub fn activate_item(&mut self, item: &dyn ItemHandle, cx: &mut ViewContext<Self>) -> bool {
2211 let result = self.panes.iter().find_map(|pane| {
2212 pane.read(cx)
2213 .index_for_item(item)
2214 .map(|ix| (pane.clone(), ix))
2215 });
2216 if let Some((pane, ix)) = result {
2217 pane.update(cx, |pane, cx| pane.activate_item(ix, true, true, cx));
2218 true
2219 } else {
2220 false
2221 }
2222 }
2223
2224 fn activate_pane_at_index(&mut self, action: &ActivatePane, cx: &mut ViewContext<Self>) {
2225 let panes = self.center.panes();
2226 if let Some(pane) = panes.get(action.0).map(|p| (*p).clone()) {
2227 cx.focus(&pane);
2228 } else {
2229 self.split_and_clone(self.active_pane.clone(), SplitDirection::Right, cx);
2230 }
2231 }
2232
2233 pub fn activate_next_pane(&mut self, cx: &mut ViewContext<Self>) {
2234 let panes = self.center.panes();
2235 if let Some(ix) = panes.iter().position(|pane| **pane == self.active_pane) {
2236 let next_ix = (ix + 1) % panes.len();
2237 let next_pane = panes[next_ix].clone();
2238 cx.focus(&next_pane);
2239 }
2240 }
2241
2242 pub fn activate_previous_pane(&mut self, cx: &mut ViewContext<Self>) {
2243 let panes = self.center.panes();
2244 if let Some(ix) = panes.iter().position(|pane| **pane == self.active_pane) {
2245 let prev_ix = cmp::min(ix.wrapping_sub(1), panes.len() - 1);
2246 let prev_pane = panes[prev_ix].clone();
2247 cx.focus(&prev_pane);
2248 }
2249 }
2250
2251 pub fn activate_pane_in_direction(
2252 &mut self,
2253 direction: SplitDirection,
2254 cx: &mut ViewContext<Self>,
2255 ) {
2256 if let Some(pane) = self.find_pane_in_direction(direction, cx) {
2257 cx.focus(pane);
2258 }
2259 }
2260
2261 pub fn swap_pane_in_direction(
2262 &mut self,
2263 direction: SplitDirection,
2264 cx: &mut ViewContext<Self>,
2265 ) {
2266 if let Some(to) = self
2267 .find_pane_in_direction(direction, cx)
2268 .map(|pane| pane.clone())
2269 {
2270 self.center.swap(&self.active_pane.clone(), &to);
2271 cx.notify();
2272 }
2273 }
2274
2275 fn find_pane_in_direction(
2276 &mut self,
2277 direction: SplitDirection,
2278 cx: &mut ViewContext<Self>,
2279 ) -> Option<&ViewHandle<Pane>> {
2280 let Some(bounding_box) = self.center.bounding_box_for_pane(&self.active_pane) else {
2281 return None;
2282 };
2283 let cursor = self.active_pane.read(cx).pixel_position_of_cursor(cx);
2284 let center = match cursor {
2285 Some(cursor) if bounding_box.contains_point(cursor) => cursor,
2286 _ => bounding_box.center(),
2287 };
2288
2289 let distance_to_next = theme::current(cx).workspace.pane_divider.width + 1.;
2290
2291 let target = match direction {
2292 SplitDirection::Left => vec2f(bounding_box.origin_x() - distance_to_next, center.y()),
2293 SplitDirection::Right => vec2f(bounding_box.max_x() + distance_to_next, center.y()),
2294 SplitDirection::Up => vec2f(center.x(), bounding_box.origin_y() - distance_to_next),
2295 SplitDirection::Down => vec2f(center.x(), bounding_box.max_y() + distance_to_next),
2296 };
2297 self.center.pane_at_pixel_position(target)
2298 }
2299
2300 fn handle_pane_focused(&mut self, pane: ViewHandle<Pane>, cx: &mut ViewContext<Self>) {
2301 if self.active_pane != pane {
2302 self.active_pane = pane.clone();
2303 self.status_bar.update(cx, |status_bar, cx| {
2304 status_bar.set_active_pane(&self.active_pane, cx);
2305 });
2306 self.active_item_path_changed(cx);
2307 self.last_active_center_pane = Some(pane.downgrade());
2308 }
2309
2310 self.dismiss_zoomed_items_to_reveal(None, cx);
2311 if pane.read(cx).is_zoomed() {
2312 self.zoomed = Some(pane.downgrade().into_any());
2313 } else {
2314 self.zoomed = None;
2315 }
2316 self.zoomed_position = None;
2317 self.update_active_view_for_followers(cx);
2318
2319 cx.notify();
2320 }
2321
2322 fn handle_pane_event(
2323 &mut self,
2324 pane: ViewHandle<Pane>,
2325 event: &pane::Event,
2326 cx: &mut ViewContext<Self>,
2327 ) {
2328 match event {
2329 pane::Event::AddItem { item } => item.added_to_pane(self, pane, cx),
2330 pane::Event::Split(direction) => {
2331 self.split_and_clone(pane, *direction, cx);
2332 }
2333 pane::Event::Remove => self.remove_pane(pane, cx),
2334 pane::Event::ActivateItem { local } => {
2335 if *local {
2336 self.unfollow(&pane, cx);
2337 }
2338 if &pane == self.active_pane() {
2339 self.active_item_path_changed(cx);
2340 }
2341 }
2342 pane::Event::ChangeItemTitle => {
2343 if pane == self.active_pane {
2344 self.active_item_path_changed(cx);
2345 }
2346 self.update_window_edited(cx);
2347 }
2348 pane::Event::RemoveItem { item_id } => {
2349 self.update_window_edited(cx);
2350 if let hash_map::Entry::Occupied(entry) = self.panes_by_item.entry(*item_id) {
2351 if entry.get().id() == pane.id() {
2352 entry.remove();
2353 }
2354 }
2355 }
2356 pane::Event::Focus => {
2357 self.handle_pane_focused(pane.clone(), cx);
2358 }
2359 pane::Event::ZoomIn => {
2360 if pane == self.active_pane {
2361 pane.update(cx, |pane, cx| pane.set_zoomed(true, cx));
2362 if pane.read(cx).has_focus() {
2363 self.zoomed = Some(pane.downgrade().into_any());
2364 self.zoomed_position = None;
2365 }
2366 cx.notify();
2367 }
2368 }
2369 pane::Event::ZoomOut => {
2370 pane.update(cx, |pane, cx| pane.set_zoomed(false, cx));
2371 if self.zoomed_position.is_none() {
2372 self.zoomed = None;
2373 }
2374 cx.notify();
2375 }
2376 }
2377
2378 self.serialize_workspace(cx);
2379 }
2380
2381 pub fn split_pane(
2382 &mut self,
2383 pane_to_split: ViewHandle<Pane>,
2384 split_direction: SplitDirection,
2385 cx: &mut ViewContext<Self>,
2386 ) -> ViewHandle<Pane> {
2387 let new_pane = self.add_pane(cx);
2388 self.center
2389 .split(&pane_to_split, &new_pane, split_direction)
2390 .unwrap();
2391 cx.notify();
2392 new_pane
2393 }
2394
2395 pub fn split_and_clone(
2396 &mut self,
2397 pane: ViewHandle<Pane>,
2398 direction: SplitDirection,
2399 cx: &mut ViewContext<Self>,
2400 ) -> Option<ViewHandle<Pane>> {
2401 let item = pane.read(cx).active_item()?;
2402 let maybe_pane_handle = if let Some(clone) = item.clone_on_split(self.database_id(), cx) {
2403 let new_pane = self.add_pane(cx);
2404 new_pane.update(cx, |pane, cx| pane.add_item(clone, true, true, None, cx));
2405 self.center.split(&pane, &new_pane, direction).unwrap();
2406 Some(new_pane)
2407 } else {
2408 None
2409 };
2410 cx.notify();
2411 maybe_pane_handle
2412 }
2413
2414 pub fn split_pane_with_item(
2415 &mut self,
2416 pane_to_split: WeakViewHandle<Pane>,
2417 split_direction: SplitDirection,
2418 from: WeakViewHandle<Pane>,
2419 item_id_to_move: usize,
2420 cx: &mut ViewContext<Self>,
2421 ) {
2422 let Some(pane_to_split) = pane_to_split.upgrade(cx) else {
2423 return;
2424 };
2425 let Some(from) = from.upgrade(cx) else {
2426 return;
2427 };
2428
2429 let new_pane = self.add_pane(cx);
2430 self.move_item(from.clone(), new_pane.clone(), item_id_to_move, 0, cx);
2431 self.center
2432 .split(&pane_to_split, &new_pane, split_direction)
2433 .unwrap();
2434 cx.notify();
2435 }
2436
2437 pub fn split_pane_with_project_entry(
2438 &mut self,
2439 pane_to_split: WeakViewHandle<Pane>,
2440 split_direction: SplitDirection,
2441 project_entry: ProjectEntryId,
2442 cx: &mut ViewContext<Self>,
2443 ) -> Option<Task<Result<()>>> {
2444 let pane_to_split = pane_to_split.upgrade(cx)?;
2445 let new_pane = self.add_pane(cx);
2446 self.center
2447 .split(&pane_to_split, &new_pane, split_direction)
2448 .unwrap();
2449
2450 let path = self.project.read(cx).path_for_entry(project_entry, cx)?;
2451 let task = self.open_path(path, Some(new_pane.downgrade()), true, cx);
2452 Some(cx.foreground().spawn(async move {
2453 task.await?;
2454 Ok(())
2455 }))
2456 }
2457
2458 pub fn move_item(
2459 &mut self,
2460 source: ViewHandle<Pane>,
2461 destination: ViewHandle<Pane>,
2462 item_id_to_move: usize,
2463 destination_index: usize,
2464 cx: &mut ViewContext<Self>,
2465 ) {
2466 let item_to_move = source
2467 .read(cx)
2468 .items()
2469 .enumerate()
2470 .find(|(_, item_handle)| item_handle.id() == item_id_to_move);
2471
2472 if item_to_move.is_none() {
2473 log::warn!("Tried to move item handle which was not in `from` pane. Maybe tab was closed during drop");
2474 return;
2475 }
2476 let (item_ix, item_handle) = item_to_move.unwrap();
2477 let item_handle = item_handle.clone();
2478
2479 if source != destination {
2480 // Close item from previous pane
2481 source.update(cx, |source, cx| {
2482 source.remove_item(item_ix, false, cx);
2483 });
2484 }
2485
2486 // This automatically removes duplicate items in the pane
2487 destination.update(cx, |destination, cx| {
2488 destination.add_item(item_handle, true, true, Some(destination_index), cx);
2489 cx.focus_self();
2490 });
2491 }
2492
2493 fn remove_pane(&mut self, pane: ViewHandle<Pane>, cx: &mut ViewContext<Self>) {
2494 if self.center.remove(&pane).unwrap() {
2495 self.force_remove_pane(&pane, cx);
2496 self.unfollow(&pane, cx);
2497 self.last_leaders_by_pane.remove(&pane.downgrade());
2498 for removed_item in pane.read(cx).items() {
2499 self.panes_by_item.remove(&removed_item.id());
2500 }
2501
2502 cx.notify();
2503 } else {
2504 self.active_item_path_changed(cx);
2505 }
2506 }
2507
2508 pub fn panes(&self) -> &[ViewHandle<Pane>] {
2509 &self.panes
2510 }
2511
2512 pub fn active_pane(&self) -> &ViewHandle<Pane> {
2513 &self.active_pane
2514 }
2515
2516 fn collaborator_left(&mut self, peer_id: PeerId, cx: &mut ViewContext<Self>) {
2517 self.follower_states.retain(|_, state| {
2518 if state.leader_id == peer_id {
2519 for item in state.items_by_leader_view_id.values() {
2520 item.set_leader_peer_id(None, cx);
2521 }
2522 false
2523 } else {
2524 true
2525 }
2526 });
2527 cx.notify();
2528 }
2529
2530 fn start_following(
2531 &mut self,
2532 leader_id: PeerId,
2533 cx: &mut ViewContext<Self>,
2534 ) -> Option<Task<Result<()>>> {
2535 let pane = self.active_pane().clone();
2536
2537 self.last_leaders_by_pane
2538 .insert(pane.downgrade(), leader_id);
2539 self.unfollow(&pane, cx);
2540 self.follower_states.insert(
2541 pane.clone(),
2542 FollowerState {
2543 leader_id,
2544 active_view_id: None,
2545 items_by_leader_view_id: Default::default(),
2546 },
2547 );
2548 cx.notify();
2549
2550 let room_id = self.active_call()?.read(cx).room()?.read(cx).id();
2551 let project_id = self.project.read(cx).remote_id();
2552 let request = self.app_state.client.request(proto::Follow {
2553 room_id,
2554 project_id,
2555 leader_id: Some(leader_id),
2556 });
2557
2558 Some(cx.spawn(|this, mut cx| async move {
2559 let response = request.await?;
2560 this.update(&mut cx, |this, _| {
2561 let state = this
2562 .follower_states
2563 .get_mut(&pane)
2564 .ok_or_else(|| anyhow!("following interrupted"))?;
2565 state.active_view_id = if let Some(active_view_id) = response.active_view_id {
2566 Some(ViewId::from_proto(active_view_id)?)
2567 } else {
2568 None
2569 };
2570 Ok::<_, anyhow::Error>(())
2571 })??;
2572 Self::add_views_from_leader(
2573 this.clone(),
2574 leader_id,
2575 vec![pane],
2576 response.views,
2577 &mut cx,
2578 )
2579 .await?;
2580 this.update(&mut cx, |this, cx| this.leader_updated(leader_id, cx))?;
2581 Ok(())
2582 }))
2583 }
2584
2585 pub fn follow_next_collaborator(
2586 &mut self,
2587 _: &FollowNextCollaborator,
2588 cx: &mut ViewContext<Self>,
2589 ) -> Option<Task<Result<()>>> {
2590 let collaborators = self.project.read(cx).collaborators();
2591 let next_leader_id = if let Some(leader_id) = self.leader_for_pane(&self.active_pane) {
2592 let mut collaborators = collaborators.keys().copied();
2593 for peer_id in collaborators.by_ref() {
2594 if peer_id == leader_id {
2595 break;
2596 }
2597 }
2598 collaborators.next()
2599 } else if let Some(last_leader_id) =
2600 self.last_leaders_by_pane.get(&self.active_pane.downgrade())
2601 {
2602 if collaborators.contains_key(last_leader_id) {
2603 Some(*last_leader_id)
2604 } else {
2605 None
2606 }
2607 } else {
2608 None
2609 };
2610
2611 let pane = self.active_pane.clone();
2612 let Some(leader_id) = next_leader_id.or_else(|| collaborators.keys().copied().next())
2613 else {
2614 return None;
2615 };
2616 if Some(leader_id) == self.unfollow(&pane, cx) {
2617 return None;
2618 }
2619 self.follow(leader_id, cx)
2620 }
2621
2622 pub fn follow(
2623 &mut self,
2624 leader_id: PeerId,
2625 cx: &mut ViewContext<Self>,
2626 ) -> Option<Task<Result<()>>> {
2627 let room = ActiveCall::global(cx).read(cx).room()?.read(cx);
2628 let project = self.project.read(cx);
2629
2630 let Some(remote_participant) = room.remote_participant_for_peer_id(leader_id) else {
2631 return None;
2632 };
2633
2634 let other_project_id = match remote_participant.location {
2635 call::ParticipantLocation::External => None,
2636 call::ParticipantLocation::UnsharedProject => None,
2637 call::ParticipantLocation::SharedProject { project_id } => {
2638 if Some(project_id) == project.remote_id() {
2639 None
2640 } else {
2641 Some(project_id)
2642 }
2643 }
2644 };
2645
2646 // if they are active in another project, follow there.
2647 if let Some(project_id) = other_project_id {
2648 let app_state = self.app_state.clone();
2649 return Some(crate::join_remote_project(
2650 project_id,
2651 remote_participant.user.id,
2652 app_state,
2653 cx,
2654 ));
2655 }
2656
2657 // if you're already following, find the right pane and focus it.
2658 for (pane, state) in &self.follower_states {
2659 if leader_id == state.leader_id {
2660 cx.focus(pane);
2661 return None;
2662 }
2663 }
2664
2665 // Otherwise, follow.
2666 self.start_following(leader_id, cx)
2667 }
2668
2669 pub fn unfollow(
2670 &mut self,
2671 pane: &ViewHandle<Pane>,
2672 cx: &mut ViewContext<Self>,
2673 ) -> Option<PeerId> {
2674 let state = self.follower_states.remove(pane)?;
2675 let leader_id = state.leader_id;
2676 for (_, item) in state.items_by_leader_view_id {
2677 item.set_leader_peer_id(None, cx);
2678 }
2679
2680 if self
2681 .follower_states
2682 .values()
2683 .all(|state| state.leader_id != state.leader_id)
2684 {
2685 let project_id = self.project.read(cx).remote_id();
2686 let room_id = self.active_call()?.read(cx).room()?.read(cx).id();
2687 self.app_state
2688 .client
2689 .send(proto::Unfollow {
2690 room_id,
2691 project_id,
2692 leader_id: Some(leader_id),
2693 })
2694 .log_err();
2695 }
2696
2697 cx.notify();
2698 Some(leader_id)
2699 }
2700
2701 pub fn is_being_followed(&self, peer_id: PeerId) -> bool {
2702 self.follower_states
2703 .values()
2704 .any(|state| state.leader_id == peer_id)
2705 }
2706
2707 fn render_titlebar(&self, theme: &Theme, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
2708 // TODO: There should be a better system in place for this
2709 // (https://github.com/zed-industries/zed/issues/1290)
2710 let is_fullscreen = cx.window_is_fullscreen();
2711 let container_theme = if is_fullscreen {
2712 let mut container_theme = theme.titlebar.container;
2713 container_theme.padding.left = container_theme.padding.right;
2714 container_theme
2715 } else {
2716 theme.titlebar.container
2717 };
2718
2719 enum TitleBar {}
2720 MouseEventHandler::new::<TitleBar, _>(0, cx, |_, cx| {
2721 Stack::new()
2722 .with_children(
2723 self.titlebar_item
2724 .as_ref()
2725 .map(|item| ChildView::new(item, cx)),
2726 )
2727 .contained()
2728 .with_style(container_theme)
2729 })
2730 .on_click(MouseButton::Left, |event, _, cx| {
2731 if event.click_count == 2 {
2732 cx.zoom_window();
2733 }
2734 })
2735 .constrained()
2736 .with_height(theme.titlebar.height)
2737 .into_any_named("titlebar")
2738 }
2739
2740 fn active_item_path_changed(&mut self, cx: &mut ViewContext<Self>) {
2741 let active_entry = self.active_project_path(cx);
2742 self.project
2743 .update(cx, |project, cx| project.set_active_path(active_entry, cx));
2744 self.update_window_title(cx);
2745 }
2746
2747 fn update_window_title(&mut self, cx: &mut ViewContext<Self>) {
2748 let project = self.project().read(cx);
2749 let mut title = String::new();
2750
2751 if let Some(path) = self.active_item(cx).and_then(|item| item.project_path(cx)) {
2752 let filename = path
2753 .path
2754 .file_name()
2755 .map(|s| s.to_string_lossy())
2756 .or_else(|| {
2757 Some(Cow::Borrowed(
2758 project
2759 .worktree_for_id(path.worktree_id, cx)?
2760 .read(cx)
2761 .root_name(),
2762 ))
2763 });
2764
2765 if let Some(filename) = filename {
2766 title.push_str(filename.as_ref());
2767 title.push_str(" — ");
2768 }
2769 }
2770
2771 for (i, name) in project.worktree_root_names(cx).enumerate() {
2772 if i > 0 {
2773 title.push_str(", ");
2774 }
2775 title.push_str(name);
2776 }
2777
2778 if title.is_empty() {
2779 title = "empty project".to_string();
2780 }
2781
2782 if project.is_remote() {
2783 title.push_str(" ↙");
2784 } else if project.is_shared() {
2785 title.push_str(" ↗");
2786 }
2787
2788 cx.set_window_title(&title);
2789 }
2790
2791 fn update_window_edited(&mut self, cx: &mut ViewContext<Self>) {
2792 let is_edited = !self.project.read(cx).is_read_only()
2793 && self
2794 .items(cx)
2795 .any(|item| item.has_conflict(cx) || item.is_dirty(cx));
2796 if is_edited != self.window_edited {
2797 self.window_edited = is_edited;
2798 cx.set_window_edited(self.window_edited)
2799 }
2800 }
2801
2802 fn render_disconnected_overlay(
2803 &self,
2804 cx: &mut ViewContext<Workspace>,
2805 ) -> Option<AnyElement<Workspace>> {
2806 if self.project.read(cx).is_read_only() {
2807 enum DisconnectedOverlay {}
2808 Some(
2809 MouseEventHandler::new::<DisconnectedOverlay, _>(0, cx, |_, cx| {
2810 let theme = &theme::current(cx);
2811 Label::new(
2812 "Your connection to the remote project has been lost.",
2813 theme.workspace.disconnected_overlay.text.clone(),
2814 )
2815 .aligned()
2816 .contained()
2817 .with_style(theme.workspace.disconnected_overlay.container)
2818 })
2819 .with_cursor_style(CursorStyle::Arrow)
2820 .capture_all()
2821 .into_any_named("disconnected overlay"),
2822 )
2823 } else {
2824 None
2825 }
2826 }
2827
2828 fn render_notifications(
2829 &self,
2830 theme: &theme::Workspace,
2831 cx: &AppContext,
2832 ) -> Option<AnyElement<Workspace>> {
2833 if self.notifications.is_empty() {
2834 None
2835 } else {
2836 Some(
2837 Flex::column()
2838 .with_children(self.notifications.iter().map(|(_, _, notification)| {
2839 ChildView::new(notification.as_any(), cx)
2840 .contained()
2841 .with_style(theme.notification)
2842 }))
2843 .constrained()
2844 .with_width(theme.notifications.width)
2845 .contained()
2846 .with_style(theme.notifications.container)
2847 .aligned()
2848 .bottom()
2849 .right()
2850 .into_any(),
2851 )
2852 }
2853 }
2854
2855 // RPC handlers
2856
2857 fn handle_follow(
2858 &mut self,
2859 follower_project_id: Option<u64>,
2860 cx: &mut ViewContext<Self>,
2861 ) -> proto::FollowResponse {
2862 let client = &self.app_state.client;
2863 let project_id = self.project.read(cx).remote_id();
2864
2865 let active_view_id = self.active_item(cx).and_then(|i| {
2866 Some(
2867 i.to_followable_item_handle(cx)?
2868 .remote_id(client, cx)?
2869 .to_proto(),
2870 )
2871 });
2872
2873 cx.notify();
2874
2875 self.last_active_view_id = active_view_id.clone();
2876 proto::FollowResponse {
2877 active_view_id,
2878 views: self
2879 .panes()
2880 .iter()
2881 .flat_map(|pane| {
2882 let leader_id = self.leader_for_pane(pane);
2883 pane.read(cx).items().filter_map({
2884 let cx = &cx;
2885 move |item| {
2886 let item = item.to_followable_item_handle(cx)?;
2887 if (project_id.is_none() || project_id != follower_project_id)
2888 && item.is_project_item(cx)
2889 {
2890 return None;
2891 }
2892 let id = item.remote_id(client, cx)?.to_proto();
2893 let variant = item.to_state_proto(cx)?;
2894 Some(proto::View {
2895 id: Some(id),
2896 leader_id,
2897 variant: Some(variant),
2898 })
2899 }
2900 })
2901 })
2902 .collect(),
2903 }
2904 }
2905
2906 fn handle_update_followers(
2907 &mut self,
2908 leader_id: PeerId,
2909 message: proto::UpdateFollowers,
2910 _cx: &mut ViewContext<Self>,
2911 ) {
2912 self.leader_updates_tx
2913 .unbounded_send((leader_id, message))
2914 .ok();
2915 }
2916
2917 async fn process_leader_update(
2918 this: &WeakViewHandle<Self>,
2919 leader_id: PeerId,
2920 update: proto::UpdateFollowers,
2921 cx: &mut AsyncAppContext,
2922 ) -> Result<()> {
2923 match update.variant.ok_or_else(|| anyhow!("invalid update"))? {
2924 proto::update_followers::Variant::UpdateActiveView(update_active_view) => {
2925 this.update(cx, |this, _| {
2926 for (_, state) in &mut this.follower_states {
2927 if state.leader_id == leader_id {
2928 state.active_view_id =
2929 if let Some(active_view_id) = update_active_view.id.clone() {
2930 Some(ViewId::from_proto(active_view_id)?)
2931 } else {
2932 None
2933 };
2934 }
2935 }
2936 anyhow::Ok(())
2937 })??;
2938 }
2939 proto::update_followers::Variant::UpdateView(update_view) => {
2940 let variant = update_view
2941 .variant
2942 .ok_or_else(|| anyhow!("missing update view variant"))?;
2943 let id = update_view
2944 .id
2945 .ok_or_else(|| anyhow!("missing update view id"))?;
2946 let mut tasks = Vec::new();
2947 this.update(cx, |this, cx| {
2948 let project = this.project.clone();
2949 for (_, state) in &mut this.follower_states {
2950 if state.leader_id == leader_id {
2951 let view_id = ViewId::from_proto(id.clone())?;
2952 if let Some(item) = state.items_by_leader_view_id.get(&view_id) {
2953 tasks.push(item.apply_update_proto(&project, variant.clone(), cx));
2954 }
2955 }
2956 }
2957 anyhow::Ok(())
2958 })??;
2959 try_join_all(tasks).await.log_err();
2960 }
2961 proto::update_followers::Variant::CreateView(view) => {
2962 let panes = this.read_with(cx, |this, _| {
2963 this.follower_states
2964 .iter()
2965 .filter_map(|(pane, state)| (state.leader_id == leader_id).then_some(pane))
2966 .cloned()
2967 .collect()
2968 })?;
2969 Self::add_views_from_leader(this.clone(), leader_id, panes, vec![view], cx).await?;
2970 }
2971 }
2972 this.update(cx, |this, cx| this.leader_updated(leader_id, cx))?;
2973 Ok(())
2974 }
2975
2976 async fn add_views_from_leader(
2977 this: WeakViewHandle<Self>,
2978 leader_id: PeerId,
2979 panes: Vec<ViewHandle<Pane>>,
2980 views: Vec<proto::View>,
2981 cx: &mut AsyncAppContext,
2982 ) -> Result<()> {
2983 let this = this
2984 .upgrade(cx)
2985 .ok_or_else(|| anyhow!("workspace dropped"))?;
2986
2987 let item_builders = cx.update(|cx| {
2988 cx.default_global::<FollowableItemBuilders>()
2989 .values()
2990 .map(|b| b.0)
2991 .collect::<Vec<_>>()
2992 });
2993
2994 let mut item_tasks_by_pane = HashMap::default();
2995 for pane in panes {
2996 let mut item_tasks = Vec::new();
2997 let mut leader_view_ids = Vec::new();
2998 for view in &views {
2999 let Some(id) = &view.id else { continue };
3000 let id = ViewId::from_proto(id.clone())?;
3001 let mut variant = view.variant.clone();
3002 if variant.is_none() {
3003 Err(anyhow!("missing view variant"))?;
3004 }
3005 for build_item in &item_builders {
3006 let task = cx
3007 .update(|cx| build_item(pane.clone(), this.clone(), id, &mut variant, cx));
3008 if let Some(task) = task {
3009 item_tasks.push(task);
3010 leader_view_ids.push(id);
3011 break;
3012 } else {
3013 assert!(variant.is_some());
3014 }
3015 }
3016 }
3017
3018 item_tasks_by_pane.insert(pane, (item_tasks, leader_view_ids));
3019 }
3020
3021 for (pane, (item_tasks, leader_view_ids)) in item_tasks_by_pane {
3022 let items = futures::future::try_join_all(item_tasks).await?;
3023 this.update(cx, |this, cx| {
3024 let state = this.follower_states.get_mut(&pane)?;
3025 for (id, item) in leader_view_ids.into_iter().zip(items) {
3026 item.set_leader_peer_id(Some(leader_id), cx);
3027 state.items_by_leader_view_id.insert(id, item);
3028 }
3029
3030 Some(())
3031 });
3032 }
3033 Ok(())
3034 }
3035
3036 fn update_active_view_for_followers(&mut self, cx: &AppContext) {
3037 let mut is_project_item = true;
3038 let mut update = proto::UpdateActiveView::default();
3039 if self.active_pane.read(cx).has_focus() {
3040 let item = self
3041 .active_item(cx)
3042 .and_then(|item| item.to_followable_item_handle(cx));
3043 if let Some(item) = item {
3044 is_project_item = item.is_project_item(cx);
3045 update = proto::UpdateActiveView {
3046 id: item
3047 .remote_id(&self.app_state.client, cx)
3048 .map(|id| id.to_proto()),
3049 leader_id: self.leader_for_pane(&self.active_pane),
3050 };
3051 }
3052 }
3053
3054 if update.id != self.last_active_view_id {
3055 self.last_active_view_id = update.id.clone();
3056 self.update_followers(
3057 is_project_item,
3058 proto::update_followers::Variant::UpdateActiveView(update),
3059 cx,
3060 );
3061 }
3062 }
3063
3064 fn update_followers(
3065 &self,
3066 project_only: bool,
3067 update: proto::update_followers::Variant,
3068 cx: &AppContext,
3069 ) -> Option<()> {
3070 let project_id = if project_only {
3071 self.project.read(cx).remote_id()
3072 } else {
3073 None
3074 };
3075 self.app_state().workspace_store.read_with(cx, |store, cx| {
3076 store.update_followers(project_id, update, cx)
3077 })
3078 }
3079
3080 pub fn leader_for_pane(&self, pane: &ViewHandle<Pane>) -> Option<PeerId> {
3081 self.follower_states.get(pane).map(|state| state.leader_id)
3082 }
3083
3084 fn leader_updated(&mut self, leader_id: PeerId, cx: &mut ViewContext<Self>) -> Option<()> {
3085 cx.notify();
3086
3087 let call = self.active_call()?;
3088 let room = call.read(cx).room()?.read(cx);
3089 let participant = room.remote_participant_for_peer_id(leader_id)?;
3090 let mut items_to_activate = Vec::new();
3091
3092 let leader_in_this_app;
3093 let leader_in_this_project;
3094 match participant.location {
3095 call::ParticipantLocation::SharedProject { project_id } => {
3096 leader_in_this_app = true;
3097 leader_in_this_project = Some(project_id) == self.project.read(cx).remote_id();
3098 }
3099 call::ParticipantLocation::UnsharedProject => {
3100 leader_in_this_app = true;
3101 leader_in_this_project = false;
3102 }
3103 call::ParticipantLocation::External => {
3104 leader_in_this_app = false;
3105 leader_in_this_project = false;
3106 }
3107 };
3108
3109 for (pane, state) in &self.follower_states {
3110 if state.leader_id != leader_id {
3111 continue;
3112 }
3113 if let (Some(active_view_id), true) = (state.active_view_id, leader_in_this_app) {
3114 if let Some(item) = state.items_by_leader_view_id.get(&active_view_id) {
3115 if leader_in_this_project || !item.is_project_item(cx) {
3116 items_to_activate.push((pane.clone(), item.boxed_clone()));
3117 }
3118 } else {
3119 log::warn!(
3120 "unknown view id {:?} for leader {:?}",
3121 active_view_id,
3122 leader_id
3123 );
3124 }
3125 continue;
3126 }
3127 if let Some(shared_screen) = self.shared_screen_for_peer(leader_id, pane, cx) {
3128 items_to_activate.push((pane.clone(), Box::new(shared_screen)));
3129 }
3130 }
3131
3132 for (pane, item) in items_to_activate {
3133 let pane_was_focused = pane.read(cx).has_focus();
3134 if let Some(index) = pane.update(cx, |pane, _| pane.index_for_item(item.as_ref())) {
3135 pane.update(cx, |pane, cx| pane.activate_item(index, false, false, cx));
3136 } else {
3137 pane.update(cx, |pane, cx| {
3138 pane.add_item(item.boxed_clone(), false, false, None, cx)
3139 });
3140 }
3141
3142 if pane_was_focused {
3143 pane.update(cx, |pane, cx| pane.focus_active_item(cx));
3144 }
3145 }
3146
3147 None
3148 }
3149
3150 fn shared_screen_for_peer(
3151 &self,
3152 peer_id: PeerId,
3153 pane: &ViewHandle<Pane>,
3154 cx: &mut ViewContext<Self>,
3155 ) -> Option<ViewHandle<SharedScreen>> {
3156 let call = self.active_call()?;
3157 let room = call.read(cx).room()?.read(cx);
3158 let participant = room.remote_participant_for_peer_id(peer_id)?;
3159 let track = participant.video_tracks.values().next()?.clone();
3160 let user = participant.user.clone();
3161
3162 for item in pane.read(cx).items_of_type::<SharedScreen>() {
3163 if item.read(cx).peer_id == peer_id {
3164 return Some(item);
3165 }
3166 }
3167
3168 Some(cx.add_view(|cx| SharedScreen::new(&track, peer_id, user.clone(), cx)))
3169 }
3170
3171 pub fn on_window_activation_changed(&mut self, active: bool, cx: &mut ViewContext<Self>) {
3172 if active {
3173 self.update_active_view_for_followers(cx);
3174 cx.background()
3175 .spawn(persistence::DB.update_timestamp(self.database_id()))
3176 .detach();
3177 } else {
3178 for pane in &self.panes {
3179 pane.update(cx, |pane, cx| {
3180 if let Some(item) = pane.active_item() {
3181 item.workspace_deactivated(cx);
3182 }
3183 if matches!(
3184 settings::get::<WorkspaceSettings>(cx).autosave,
3185 AutosaveSetting::OnWindowChange | AutosaveSetting::OnFocusChange
3186 ) {
3187 for item in pane.items() {
3188 Pane::autosave_item(item.as_ref(), self.project.clone(), cx)
3189 .detach_and_log_err(cx);
3190 }
3191 }
3192 });
3193 }
3194 }
3195 }
3196
3197 fn active_call(&self) -> Option<&ModelHandle<ActiveCall>> {
3198 self.active_call.as_ref().map(|(call, _)| call)
3199 }
3200
3201 fn on_active_call_event(
3202 &mut self,
3203 _: ModelHandle<ActiveCall>,
3204 event: &call::room::Event,
3205 cx: &mut ViewContext<Self>,
3206 ) {
3207 match event {
3208 call::room::Event::ParticipantLocationChanged { participant_id }
3209 | call::room::Event::RemoteVideoTracksChanged { participant_id } => {
3210 self.leader_updated(*participant_id, cx);
3211 }
3212 _ => {}
3213 }
3214 }
3215
3216 pub fn database_id(&self) -> WorkspaceId {
3217 self.database_id
3218 }
3219
3220 fn location(&self, cx: &AppContext) -> Option<WorkspaceLocation> {
3221 let project = self.project().read(cx);
3222
3223 if project.is_local() {
3224 Some(
3225 project
3226 .visible_worktrees(cx)
3227 .map(|worktree| worktree.read(cx).abs_path())
3228 .collect::<Vec<_>>()
3229 .into(),
3230 )
3231 } else {
3232 None
3233 }
3234 }
3235
3236 fn remove_panes(&mut self, member: Member, cx: &mut ViewContext<Workspace>) {
3237 match member {
3238 Member::Axis(PaneAxis { members, .. }) => {
3239 for child in members.iter() {
3240 self.remove_panes(child.clone(), cx)
3241 }
3242 }
3243 Member::Pane(pane) => {
3244 self.force_remove_pane(&pane, cx);
3245 }
3246 }
3247 }
3248
3249 fn force_remove_pane(&mut self, pane: &ViewHandle<Pane>, cx: &mut ViewContext<Workspace>) {
3250 self.panes.retain(|p| p != pane);
3251 cx.focus(self.panes.last().unwrap());
3252 if self.last_active_center_pane == Some(pane.downgrade()) {
3253 self.last_active_center_pane = None;
3254 }
3255 cx.notify();
3256 }
3257
3258 fn schedule_serialize(&mut self, cx: &mut ViewContext<Self>) {
3259 self._schedule_serialize = Some(cx.spawn(|this, cx| async move {
3260 cx.background().timer(Duration::from_millis(100)).await;
3261 this.read_with(&cx, |this, cx| this.serialize_workspace(cx))
3262 .ok();
3263 }));
3264 }
3265
3266 fn serialize_workspace(&self, cx: &ViewContext<Self>) {
3267 fn serialize_pane_handle(
3268 pane_handle: &ViewHandle<Pane>,
3269 cx: &AppContext,
3270 ) -> SerializedPane {
3271 let (items, active) = {
3272 let pane = pane_handle.read(cx);
3273 let active_item_id = pane.active_item().map(|item| item.id());
3274 (
3275 pane.items()
3276 .filter_map(|item_handle| {
3277 Some(SerializedItem {
3278 kind: Arc::from(item_handle.serialized_item_kind()?),
3279 item_id: item_handle.id(),
3280 active: Some(item_handle.id()) == active_item_id,
3281 })
3282 })
3283 .collect::<Vec<_>>(),
3284 pane.has_focus(),
3285 )
3286 };
3287
3288 SerializedPane::new(items, active)
3289 }
3290
3291 fn build_serialized_pane_group(
3292 pane_group: &Member,
3293 cx: &AppContext,
3294 ) -> SerializedPaneGroup {
3295 match pane_group {
3296 Member::Axis(PaneAxis {
3297 axis,
3298 members,
3299 flexes,
3300 bounding_boxes: _,
3301 }) => SerializedPaneGroup::Group {
3302 axis: *axis,
3303 children: members
3304 .iter()
3305 .map(|member| build_serialized_pane_group(member, cx))
3306 .collect::<Vec<_>>(),
3307 flexes: Some(flexes.borrow().clone()),
3308 },
3309 Member::Pane(pane_handle) => {
3310 SerializedPaneGroup::Pane(serialize_pane_handle(&pane_handle, cx))
3311 }
3312 }
3313 }
3314
3315 fn build_serialized_docks(this: &Workspace, cx: &ViewContext<Workspace>) -> DockStructure {
3316 let left_dock = this.left_dock.read(cx);
3317 let left_visible = left_dock.is_open();
3318 let left_active_panel = left_dock.visible_panel().and_then(|panel| {
3319 Some(
3320 cx.view_ui_name(panel.as_any().window(), panel.id())?
3321 .to_string(),
3322 )
3323 });
3324 let left_dock_zoom = left_dock
3325 .visible_panel()
3326 .map(|panel| panel.is_zoomed(cx))
3327 .unwrap_or(false);
3328
3329 let right_dock = this.right_dock.read(cx);
3330 let right_visible = right_dock.is_open();
3331 let right_active_panel = right_dock.visible_panel().and_then(|panel| {
3332 Some(
3333 cx.view_ui_name(panel.as_any().window(), panel.id())?
3334 .to_string(),
3335 )
3336 });
3337 let right_dock_zoom = right_dock
3338 .visible_panel()
3339 .map(|panel| panel.is_zoomed(cx))
3340 .unwrap_or(false);
3341
3342 let bottom_dock = this.bottom_dock.read(cx);
3343 let bottom_visible = bottom_dock.is_open();
3344 let bottom_active_panel = bottom_dock.visible_panel().and_then(|panel| {
3345 Some(
3346 cx.view_ui_name(panel.as_any().window(), panel.id())?
3347 .to_string(),
3348 )
3349 });
3350 let bottom_dock_zoom = bottom_dock
3351 .visible_panel()
3352 .map(|panel| panel.is_zoomed(cx))
3353 .unwrap_or(false);
3354
3355 DockStructure {
3356 left: DockData {
3357 visible: left_visible,
3358 active_panel: left_active_panel,
3359 zoom: left_dock_zoom,
3360 },
3361 right: DockData {
3362 visible: right_visible,
3363 active_panel: right_active_panel,
3364 zoom: right_dock_zoom,
3365 },
3366 bottom: DockData {
3367 visible: bottom_visible,
3368 active_panel: bottom_active_panel,
3369 zoom: bottom_dock_zoom,
3370 },
3371 }
3372 }
3373
3374 if let Some(location) = self.location(cx) {
3375 // Load bearing special case:
3376 // - with_local_workspace() relies on this to not have other stuff open
3377 // when you open your log
3378 if !location.paths().is_empty() {
3379 let center_group = build_serialized_pane_group(&self.center.root, cx);
3380 let docks = build_serialized_docks(self, cx);
3381
3382 let serialized_workspace = SerializedWorkspace {
3383 id: self.database_id,
3384 location,
3385 center_group,
3386 bounds: Default::default(),
3387 display: Default::default(),
3388 docks,
3389 };
3390
3391 cx.background()
3392 .spawn(persistence::DB.save_workspace(serialized_workspace))
3393 .detach();
3394 }
3395 }
3396 }
3397
3398 pub(crate) fn load_workspace(
3399 workspace: WeakViewHandle<Workspace>,
3400 serialized_workspace: SerializedWorkspace,
3401 paths_to_open: Vec<Option<ProjectPath>>,
3402 cx: &mut AppContext,
3403 ) -> Task<Result<Vec<Option<Box<dyn ItemHandle>>>>> {
3404 cx.spawn(|mut cx| async move {
3405 let (project, old_center_pane) = workspace.read_with(&cx, |workspace, _| {
3406 (
3407 workspace.project().clone(),
3408 workspace.last_active_center_pane.clone(),
3409 )
3410 })?;
3411
3412 let mut center_group = None;
3413 let mut center_items = None;
3414 // Traverse the splits tree and add to things
3415 if let Some((group, active_pane, items)) = serialized_workspace
3416 .center_group
3417 .deserialize(&project, serialized_workspace.id, &workspace, &mut cx)
3418 .await
3419 {
3420 center_items = Some(items);
3421 center_group = Some((group, active_pane))
3422 }
3423
3424 let mut items_by_project_path = cx.read(|cx| {
3425 center_items
3426 .unwrap_or_default()
3427 .into_iter()
3428 .filter_map(|item| {
3429 let item = item?;
3430 let project_path = item.project_path(cx)?;
3431 Some((project_path, item))
3432 })
3433 .collect::<HashMap<_, _>>()
3434 });
3435
3436 let opened_items = paths_to_open
3437 .into_iter()
3438 .map(|path_to_open| {
3439 path_to_open
3440 .and_then(|path_to_open| items_by_project_path.remove(&path_to_open))
3441 })
3442 .collect::<Vec<_>>();
3443
3444 // Remove old panes from workspace panes list
3445 workspace.update(&mut cx, |workspace, cx| {
3446 if let Some((center_group, active_pane)) = center_group {
3447 workspace.remove_panes(workspace.center.root.clone(), cx);
3448
3449 // Swap workspace center group
3450 workspace.center = PaneGroup::with_root(center_group);
3451
3452 // Change the focus to the workspace first so that we retrigger focus in on the pane.
3453 cx.focus_self();
3454
3455 if let Some(active_pane) = active_pane {
3456 cx.focus(&active_pane);
3457 } else {
3458 cx.focus(workspace.panes.last().unwrap());
3459 }
3460 } else {
3461 let old_center_handle = old_center_pane.and_then(|weak| weak.upgrade(cx));
3462 if let Some(old_center_handle) = old_center_handle {
3463 cx.focus(&old_center_handle)
3464 } else {
3465 cx.focus_self()
3466 }
3467 }
3468
3469 let docks = serialized_workspace.docks;
3470 workspace.left_dock.update(cx, |dock, cx| {
3471 dock.set_open(docks.left.visible, cx);
3472 if let Some(active_panel) = docks.left.active_panel {
3473 if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
3474 dock.activate_panel(ix, cx);
3475 }
3476 }
3477 dock.active_panel()
3478 .map(|panel| panel.set_zoomed(docks.left.zoom, cx));
3479 if docks.left.visible && docks.left.zoom {
3480 cx.focus_self()
3481 }
3482 });
3483 // TODO: I think the bug is that setting zoom or active undoes the bottom zoom or something
3484 workspace.right_dock.update(cx, |dock, cx| {
3485 dock.set_open(docks.right.visible, cx);
3486 if let Some(active_panel) = docks.right.active_panel {
3487 if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
3488 dock.activate_panel(ix, cx);
3489 }
3490 }
3491 dock.active_panel()
3492 .map(|panel| panel.set_zoomed(docks.right.zoom, cx));
3493
3494 if docks.right.visible && docks.right.zoom {
3495 cx.focus_self()
3496 }
3497 });
3498 workspace.bottom_dock.update(cx, |dock, cx| {
3499 dock.set_open(docks.bottom.visible, cx);
3500 if let Some(active_panel) = docks.bottom.active_panel {
3501 if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
3502 dock.activate_panel(ix, cx);
3503 }
3504 }
3505
3506 dock.active_panel()
3507 .map(|panel| panel.set_zoomed(docks.bottom.zoom, cx));
3508
3509 if docks.bottom.visible && docks.bottom.zoom {
3510 cx.focus_self()
3511 }
3512 });
3513
3514 cx.notify();
3515 })?;
3516
3517 // Serialize ourself to make sure our timestamps and any pane / item changes are replicated
3518 workspace.read_with(&cx, |workspace, cx| workspace.serialize_workspace(cx))?;
3519
3520 Ok(opened_items)
3521 })
3522 }
3523
3524 #[cfg(any(test, feature = "test-support"))]
3525 pub fn test_new(project: ModelHandle<Project>, cx: &mut ViewContext<Self>) -> Self {
3526 use node_runtime::FakeNodeRuntime;
3527
3528 let client = project.read(cx).client();
3529 let user_store = project.read(cx).user_store();
3530
3531 let workspace_store = cx.add_model(|cx| WorkspaceStore::new(client.clone(), cx));
3532 let app_state = Arc::new(AppState {
3533 languages: project.read(cx).languages().clone(),
3534 workspace_store,
3535 client,
3536 user_store,
3537 fs: project.read(cx).fs().clone(),
3538 build_window_options: |_, _, _| Default::default(),
3539 initialize_workspace: |_, _, _, _| Task::ready(Ok(())),
3540 background_actions: || &[],
3541 node_runtime: FakeNodeRuntime::new(),
3542 });
3543 Self::new(0, project, app_state, cx)
3544 }
3545
3546 fn render_dock(&self, position: DockPosition, cx: &WindowContext) -> Option<AnyElement<Self>> {
3547 let dock = match position {
3548 DockPosition::Left => &self.left_dock,
3549 DockPosition::Right => &self.right_dock,
3550 DockPosition::Bottom => &self.bottom_dock,
3551 };
3552 let active_panel = dock.read(cx).visible_panel()?;
3553 let element = if Some(active_panel.id()) == self.zoomed.as_ref().map(|zoomed| zoomed.id()) {
3554 dock.read(cx).render_placeholder(cx)
3555 } else {
3556 ChildView::new(dock, cx).into_any()
3557 };
3558
3559 Some(
3560 element
3561 .constrained()
3562 .dynamically(move |constraint, _, cx| match position {
3563 DockPosition::Left | DockPosition::Right => SizeConstraint::new(
3564 Vector2F::new(20., constraint.min.y()),
3565 Vector2F::new(cx.window_size().x() * 0.8, constraint.max.y()),
3566 ),
3567 DockPosition::Bottom => SizeConstraint::new(
3568 Vector2F::new(constraint.min.x(), 20.),
3569 Vector2F::new(constraint.max.x(), cx.window_size().y() * 0.8),
3570 ),
3571 })
3572 .into_any(),
3573 )
3574 }
3575}
3576
3577fn window_bounds_env_override(cx: &AsyncAppContext) -> Option<WindowBounds> {
3578 ZED_WINDOW_POSITION
3579 .zip(*ZED_WINDOW_SIZE)
3580 .map(|(position, size)| {
3581 WindowBounds::Fixed(RectF::new(
3582 cx.platform().screens()[0].bounds().origin() + position,
3583 size,
3584 ))
3585 })
3586}
3587
3588async fn open_items(
3589 serialized_workspace: Option<SerializedWorkspace>,
3590 workspace: &WeakViewHandle<Workspace>,
3591 mut project_paths_to_open: Vec<(PathBuf, Option<ProjectPath>)>,
3592 app_state: Arc<AppState>,
3593 mut cx: AsyncAppContext,
3594) -> Result<Vec<Option<Result<Box<dyn ItemHandle>>>>> {
3595 let mut opened_items = Vec::with_capacity(project_paths_to_open.len());
3596
3597 if let Some(serialized_workspace) = serialized_workspace {
3598 let workspace = workspace.clone();
3599 let restored_items = cx
3600 .update(|cx| {
3601 Workspace::load_workspace(
3602 workspace,
3603 serialized_workspace,
3604 project_paths_to_open
3605 .iter()
3606 .map(|(_, project_path)| project_path)
3607 .cloned()
3608 .collect(),
3609 cx,
3610 )
3611 })
3612 .await?;
3613
3614 let restored_project_paths = cx.read(|cx| {
3615 restored_items
3616 .iter()
3617 .filter_map(|item| item.as_ref()?.project_path(cx))
3618 .collect::<HashSet<_>>()
3619 });
3620
3621 for restored_item in restored_items {
3622 opened_items.push(restored_item.map(Ok));
3623 }
3624
3625 project_paths_to_open
3626 .iter_mut()
3627 .for_each(|(_, project_path)| {
3628 if let Some(project_path_to_open) = project_path {
3629 if restored_project_paths.contains(project_path_to_open) {
3630 *project_path = None;
3631 }
3632 }
3633 });
3634 } else {
3635 for _ in 0..project_paths_to_open.len() {
3636 opened_items.push(None);
3637 }
3638 }
3639 assert!(opened_items.len() == project_paths_to_open.len());
3640
3641 let tasks =
3642 project_paths_to_open
3643 .into_iter()
3644 .enumerate()
3645 .map(|(i, (abs_path, project_path))| {
3646 let workspace = workspace.clone();
3647 cx.spawn(|mut cx| {
3648 let fs = app_state.fs.clone();
3649 async move {
3650 let file_project_path = project_path?;
3651 if fs.is_file(&abs_path).await {
3652 Some((
3653 i,
3654 workspace
3655 .update(&mut cx, |workspace, cx| {
3656 workspace.open_path(file_project_path, None, true, cx)
3657 })
3658 .log_err()?
3659 .await,
3660 ))
3661 } else {
3662 None
3663 }
3664 }
3665 })
3666 });
3667
3668 for maybe_opened_path in futures::future::join_all(tasks.into_iter())
3669 .await
3670 .into_iter()
3671 {
3672 if let Some((i, path_open_result)) = maybe_opened_path {
3673 opened_items[i] = Some(path_open_result);
3674 }
3675 }
3676
3677 Ok(opened_items)
3678}
3679
3680fn notify_of_new_dock(workspace: &WeakViewHandle<Workspace>, cx: &mut AsyncAppContext) {
3681 const NEW_PANEL_BLOG_POST: &str = "https://zed.dev/blog/new-panel-system";
3682 const NEW_DOCK_HINT_KEY: &str = "show_new_dock_key";
3683 const MESSAGE_ID: usize = 2;
3684
3685 if workspace
3686 .read_with(cx, |workspace, cx| {
3687 workspace.has_shown_notification_once::<MessageNotification>(MESSAGE_ID, cx)
3688 })
3689 .unwrap_or(false)
3690 {
3691 return;
3692 }
3693
3694 if db::kvp::KEY_VALUE_STORE
3695 .read_kvp(NEW_DOCK_HINT_KEY)
3696 .ok()
3697 .flatten()
3698 .is_some()
3699 {
3700 if !workspace
3701 .read_with(cx, |workspace, cx| {
3702 workspace.has_shown_notification_once::<MessageNotification>(MESSAGE_ID, cx)
3703 })
3704 .unwrap_or(false)
3705 {
3706 cx.update(|cx| {
3707 cx.update_global::<NotificationTracker, _, _>(|tracker, _| {
3708 let entry = tracker
3709 .entry(TypeId::of::<MessageNotification>())
3710 .or_default();
3711 if !entry.contains(&MESSAGE_ID) {
3712 entry.push(MESSAGE_ID);
3713 }
3714 });
3715 });
3716 }
3717
3718 return;
3719 }
3720
3721 cx.spawn(|_| async move {
3722 db::kvp::KEY_VALUE_STORE
3723 .write_kvp(NEW_DOCK_HINT_KEY.to_string(), "seen".to_string())
3724 .await
3725 .ok();
3726 })
3727 .detach();
3728
3729 workspace
3730 .update(cx, |workspace, cx| {
3731 workspace.show_notification_once(2, cx, |cx| {
3732 cx.add_view(|_| {
3733 MessageNotification::new_element(|text, _| {
3734 Text::new(
3735 "Looking for the dock? Try ctrl-`!\nshift-escape now zooms your pane.",
3736 text,
3737 )
3738 .with_custom_runs(vec![26..32, 34..46], |_, bounds, cx| {
3739 let code_span_background_color = settings::get::<ThemeSettings>(cx)
3740 .theme
3741 .editor
3742 .document_highlight_read_background;
3743
3744 cx.scene().push_quad(gpui::Quad {
3745 bounds,
3746 background: Some(code_span_background_color),
3747 border: Default::default(),
3748 corner_radii: (2.0).into(),
3749 })
3750 })
3751 .into_any()
3752 })
3753 .with_click_message("Read more about the new panel system")
3754 .on_click(|cx| cx.platform().open_url(NEW_PANEL_BLOG_POST))
3755 })
3756 })
3757 })
3758 .ok();
3759}
3760
3761fn notify_if_database_failed(workspace: &WeakViewHandle<Workspace>, cx: &mut AsyncAppContext) {
3762 const REPORT_ISSUE_URL: &str ="https://github.com/zed-industries/community/issues/new?assignees=&labels=defect%2Ctriage&template=2_bug_report.yml";
3763
3764 workspace
3765 .update(cx, |workspace, cx| {
3766 if (*db::ALL_FILE_DB_FAILED).load(std::sync::atomic::Ordering::Acquire) {
3767 workspace.show_notification_once(0, cx, |cx| {
3768 cx.add_view(|_| {
3769 MessageNotification::new("Failed to load the database file.")
3770 .with_click_message("Click to let us know about this error")
3771 .on_click(|cx| cx.platform().open_url(REPORT_ISSUE_URL))
3772 })
3773 });
3774 }
3775 })
3776 .log_err();
3777}
3778
3779impl Entity for Workspace {
3780 type Event = Event;
3781
3782 fn release(&mut self, cx: &mut AppContext) {
3783 self.app_state.workspace_store.update(cx, |store, _| {
3784 store.workspaces.remove(&self.weak_self);
3785 })
3786 }
3787}
3788
3789impl View for Workspace {
3790 fn ui_name() -> &'static str {
3791 "Workspace"
3792 }
3793
3794 fn render(&mut self, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
3795 let theme = theme::current(cx).clone();
3796 Stack::new()
3797 .with_child(
3798 Flex::column()
3799 .with_child(self.render_titlebar(&theme, cx))
3800 .with_child(
3801 Stack::new()
3802 .with_child({
3803 let project = self.project.clone();
3804 Flex::row()
3805 .with_children(self.render_dock(DockPosition::Left, cx))
3806 .with_child(
3807 Flex::column()
3808 .with_child(
3809 FlexItem::new(
3810 self.center.render(
3811 &project,
3812 &theme,
3813 &self.follower_states,
3814 self.active_call(),
3815 self.active_pane(),
3816 self.zoomed
3817 .as_ref()
3818 .and_then(|zoomed| zoomed.upgrade(cx))
3819 .as_ref(),
3820 &self.app_state,
3821 cx,
3822 ),
3823 )
3824 .flex(1., true),
3825 )
3826 .with_children(
3827 self.render_dock(DockPosition::Bottom, cx),
3828 )
3829 .flex(1., true),
3830 )
3831 .with_children(self.render_dock(DockPosition::Right, cx))
3832 })
3833 .with_child(Overlay::new(
3834 Stack::new()
3835 .with_children(self.zoomed.as_ref().and_then(|zoomed| {
3836 enum ZoomBackground {}
3837 let zoomed = zoomed.upgrade(cx)?;
3838
3839 let mut foreground_style =
3840 theme.workspace.zoomed_pane_foreground;
3841 if let Some(zoomed_dock_position) = self.zoomed_position {
3842 foreground_style =
3843 theme.workspace.zoomed_panel_foreground;
3844 let margin = foreground_style.margin.top;
3845 let border = foreground_style.border.top;
3846
3847 // Only include a margin and border on the opposite side.
3848 foreground_style.margin.top = 0.;
3849 foreground_style.margin.left = 0.;
3850 foreground_style.margin.bottom = 0.;
3851 foreground_style.margin.right = 0.;
3852 foreground_style.border.top = false;
3853 foreground_style.border.left = false;
3854 foreground_style.border.bottom = false;
3855 foreground_style.border.right = false;
3856 match zoomed_dock_position {
3857 DockPosition::Left => {
3858 foreground_style.margin.right = margin;
3859 foreground_style.border.right = border;
3860 }
3861 DockPosition::Right => {
3862 foreground_style.margin.left = margin;
3863 foreground_style.border.left = border;
3864 }
3865 DockPosition::Bottom => {
3866 foreground_style.margin.top = margin;
3867 foreground_style.border.top = border;
3868 }
3869 }
3870 }
3871
3872 Some(
3873 ChildView::new(&zoomed, cx)
3874 .contained()
3875 .with_style(foreground_style)
3876 .aligned()
3877 .contained()
3878 .with_style(theme.workspace.zoomed_background)
3879 .mouse::<ZoomBackground>(0)
3880 .capture_all()
3881 .on_down(
3882 MouseButton::Left,
3883 |_, this: &mut Self, cx| {
3884 this.zoom_out(cx);
3885 },
3886 ),
3887 )
3888 }))
3889 .with_children(self.modal.as_ref().map(|modal| {
3890 // Prevent clicks within the modal from falling
3891 // through to the rest of the workspace.
3892 enum ModalBackground {}
3893 MouseEventHandler::new::<ModalBackground, _>(
3894 0,
3895 cx,
3896 |_, cx| ChildView::new(modal.view.as_any(), cx),
3897 )
3898 .on_click(MouseButton::Left, |_, _, _| {})
3899 .contained()
3900 .with_style(theme.workspace.modal)
3901 .aligned()
3902 .top()
3903 }))
3904 .with_children(self.render_notifications(&theme.workspace, cx)),
3905 ))
3906 .provide_resize_bounds::<WorkspaceBounds>()
3907 .flex(1.0, true),
3908 )
3909 .with_child(ChildView::new(&self.status_bar, cx))
3910 .contained()
3911 .with_background_color(theme.workspace.background),
3912 )
3913 .with_children(DragAndDrop::render(cx))
3914 .with_children(self.render_disconnected_overlay(cx))
3915 .into_any_named("workspace")
3916 }
3917
3918 fn focus_in(&mut self, _: AnyViewHandle, cx: &mut ViewContext<Self>) {
3919 if cx.is_self_focused() {
3920 cx.focus(&self.active_pane);
3921 }
3922 }
3923
3924 fn modifiers_changed(&mut self, e: &ModifiersChangedEvent, cx: &mut ViewContext<Self>) -> bool {
3925 DragAndDrop::<Workspace>::update_modifiers(e.modifiers, cx)
3926 }
3927}
3928
3929impl WorkspaceStore {
3930 pub fn new(client: Arc<Client>, cx: &mut ModelContext<Self>) -> Self {
3931 Self {
3932 workspaces: Default::default(),
3933 followers: Default::default(),
3934 _subscriptions: vec![
3935 client.add_request_handler(cx.handle(), Self::handle_follow),
3936 client.add_message_handler(cx.handle(), Self::handle_unfollow),
3937 client.add_message_handler(cx.handle(), Self::handle_update_followers),
3938 ],
3939 client,
3940 }
3941 }
3942
3943 pub fn update_followers(
3944 &self,
3945 project_id: Option<u64>,
3946 update: proto::update_followers::Variant,
3947 cx: &AppContext,
3948 ) -> Option<()> {
3949 if !cx.has_global::<ModelHandle<ActiveCall>>() {
3950 return None;
3951 }
3952
3953 let room_id = ActiveCall::global(cx).read(cx).room()?.read(cx).id();
3954 let follower_ids: Vec<_> = self
3955 .followers
3956 .iter()
3957 .filter_map(|follower| {
3958 if follower.project_id == project_id || project_id.is_none() {
3959 Some(follower.peer_id.into())
3960 } else {
3961 None
3962 }
3963 })
3964 .collect();
3965 if follower_ids.is_empty() {
3966 return None;
3967 }
3968 self.client
3969 .send(proto::UpdateFollowers {
3970 room_id,
3971 project_id,
3972 follower_ids,
3973 variant: Some(update),
3974 })
3975 .log_err()
3976 }
3977
3978 async fn handle_follow(
3979 this: ModelHandle<Self>,
3980 envelope: TypedEnvelope<proto::Follow>,
3981 _: Arc<Client>,
3982 mut cx: AsyncAppContext,
3983 ) -> Result<proto::FollowResponse> {
3984 this.update(&mut cx, |this, cx| {
3985 let follower = Follower {
3986 project_id: envelope.payload.project_id,
3987 peer_id: envelope.original_sender_id()?,
3988 };
3989 let active_project = ActiveCall::global(cx)
3990 .read(cx)
3991 .location()
3992 .map(|project| project.id());
3993
3994 let mut response = proto::FollowResponse::default();
3995 for workspace in &this.workspaces {
3996 let Some(workspace) = workspace.upgrade(cx) else {
3997 continue;
3998 };
3999
4000 workspace.update(cx.as_mut(), |workspace, cx| {
4001 let handler_response = workspace.handle_follow(follower.project_id, cx);
4002 if response.views.is_empty() {
4003 response.views = handler_response.views;
4004 } else {
4005 response.views.extend_from_slice(&handler_response.views);
4006 }
4007
4008 if let Some(active_view_id) = handler_response.active_view_id.clone() {
4009 if response.active_view_id.is_none()
4010 || Some(workspace.project.id()) == active_project
4011 {
4012 response.active_view_id = Some(active_view_id);
4013 }
4014 }
4015 });
4016 }
4017
4018 if let Err(ix) = this.followers.binary_search(&follower) {
4019 this.followers.insert(ix, follower);
4020 }
4021
4022 Ok(response)
4023 })
4024 }
4025
4026 async fn handle_unfollow(
4027 this: ModelHandle<Self>,
4028 envelope: TypedEnvelope<proto::Unfollow>,
4029 _: Arc<Client>,
4030 mut cx: AsyncAppContext,
4031 ) -> Result<()> {
4032 this.update(&mut cx, |this, _| {
4033 let follower = Follower {
4034 project_id: envelope.payload.project_id,
4035 peer_id: envelope.original_sender_id()?,
4036 };
4037 if let Ok(ix) = this.followers.binary_search(&follower) {
4038 this.followers.remove(ix);
4039 }
4040 Ok(())
4041 })
4042 }
4043
4044 async fn handle_update_followers(
4045 this: ModelHandle<Self>,
4046 envelope: TypedEnvelope<proto::UpdateFollowers>,
4047 _: Arc<Client>,
4048 mut cx: AsyncAppContext,
4049 ) -> Result<()> {
4050 let leader_id = envelope.original_sender_id()?;
4051 let update = envelope.payload;
4052 this.update(&mut cx, |this, cx| {
4053 for workspace in &this.workspaces {
4054 let Some(workspace) = workspace.upgrade(cx) else {
4055 continue;
4056 };
4057 workspace.update(cx.as_mut(), |workspace, cx| {
4058 let project_id = workspace.project.read(cx).remote_id();
4059 if update.project_id != project_id && update.project_id.is_some() {
4060 return;
4061 }
4062 workspace.handle_update_followers(leader_id, update.clone(), cx);
4063 });
4064 }
4065 Ok(())
4066 })
4067 }
4068}
4069
4070impl Entity for WorkspaceStore {
4071 type Event = ();
4072}
4073
4074impl ViewId {
4075 pub(crate) fn from_proto(message: proto::ViewId) -> Result<Self> {
4076 Ok(Self {
4077 creator: message
4078 .creator
4079 .ok_or_else(|| anyhow!("creator is missing"))?,
4080 id: message.id,
4081 })
4082 }
4083
4084 pub(crate) fn to_proto(&self) -> proto::ViewId {
4085 proto::ViewId {
4086 creator: Some(self.creator),
4087 id: self.id,
4088 }
4089 }
4090}
4091
4092pub trait WorkspaceHandle {
4093 fn file_project_paths(&self, cx: &AppContext) -> Vec<ProjectPath>;
4094}
4095
4096impl WorkspaceHandle for ViewHandle<Workspace> {
4097 fn file_project_paths(&self, cx: &AppContext) -> Vec<ProjectPath> {
4098 self.read(cx)
4099 .worktrees(cx)
4100 .flat_map(|worktree| {
4101 let worktree_id = worktree.read(cx).id();
4102 worktree.read(cx).files(true, 0).map(move |f| ProjectPath {
4103 worktree_id,
4104 path: f.path.clone(),
4105 })
4106 })
4107 .collect::<Vec<_>>()
4108 }
4109}
4110
4111impl std::fmt::Debug for OpenPaths {
4112 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
4113 f.debug_struct("OpenPaths")
4114 .field("paths", &self.paths)
4115 .finish()
4116 }
4117}
4118
4119pub struct WorkspaceCreated(pub WeakViewHandle<Workspace>);
4120
4121pub fn activate_workspace_for_project(
4122 cx: &mut AsyncAppContext,
4123 predicate: impl Fn(&mut Project, &mut ModelContext<Project>) -> bool,
4124) -> Option<WeakViewHandle<Workspace>> {
4125 for window in cx.windows() {
4126 let handle = window
4127 .update(cx, |cx| {
4128 if let Some(workspace_handle) = cx.root_view().clone().downcast::<Workspace>() {
4129 let project = workspace_handle.read(cx).project.clone();
4130 if project.update(cx, &predicate) {
4131 cx.activate_window();
4132 return Some(workspace_handle.clone());
4133 }
4134 }
4135 None
4136 })
4137 .flatten();
4138
4139 if let Some(handle) = handle {
4140 return Some(handle.downgrade());
4141 }
4142 }
4143 None
4144}
4145
4146pub async fn last_opened_workspace_paths() -> Option<WorkspaceLocation> {
4147 DB.last_workspace().await.log_err().flatten()
4148}
4149
4150async fn join_channel_internal(
4151 channel_id: u64,
4152 app_state: &Arc<AppState>,
4153 requesting_window: Option<WindowHandle<Workspace>>,
4154 active_call: &ModelHandle<ActiveCall>,
4155 cx: &mut AsyncAppContext,
4156) -> Result<bool> {
4157 let (should_prompt, open_room) = active_call.read_with(cx, |active_call, cx| {
4158 let Some(room) = active_call.room().map(|room| room.read(cx)) else {
4159 return (false, None);
4160 };
4161
4162 let already_in_channel = room.channel_id() == Some(channel_id);
4163 let should_prompt = room.is_sharing_project()
4164 && room.remote_participants().len() > 0
4165 && !already_in_channel;
4166 let open_room = if already_in_channel {
4167 active_call.room().cloned()
4168 } else {
4169 None
4170 };
4171 (should_prompt, open_room)
4172 });
4173
4174 if let Some(room) = open_room {
4175 let task = room.update(cx, |room, cx| {
4176 if let Some((project, host)) = room.most_active_project(cx) {
4177 return Some(join_remote_project(project, host, app_state.clone(), cx));
4178 }
4179
4180 None
4181 });
4182 if let Some(task) = task {
4183 task.await?;
4184 }
4185 return anyhow::Ok(true);
4186 }
4187
4188 if should_prompt {
4189 if let Some(workspace) = requesting_window {
4190 if let Some(window) = workspace.update(cx, |cx| cx.window()) {
4191 let answer = window.prompt(
4192 PromptLevel::Warning,
4193 "Leaving this call will unshare your current project.\nDo you want to switch channels?",
4194 &["Yes, Join Channel", "Cancel"],
4195 cx,
4196 );
4197
4198 if let Some(mut answer) = answer {
4199 if answer.next().await == Some(1) {
4200 return Ok(false);
4201 }
4202 }
4203 } else {
4204 return Ok(false); // unreachable!() hopefully
4205 }
4206 } else {
4207 return Ok(false); // unreachable!() hopefully
4208 }
4209 }
4210
4211 let client = cx.read(|cx| active_call.read(cx).client());
4212
4213 let mut client_status = client.status();
4214
4215 // this loop will terminate within client::CONNECTION_TIMEOUT seconds.
4216 'outer: loop {
4217 let Some(status) = client_status.recv().await else {
4218 return Err(anyhow!("error connecting"));
4219 };
4220
4221 match status {
4222 Status::Connecting
4223 | Status::Authenticating
4224 | Status::Reconnecting
4225 | Status::Reauthenticating => continue,
4226 Status::Connected { .. } => break 'outer,
4227 Status::SignedOut => return Err(anyhow!("not signed in")),
4228 Status::UpgradeRequired => return Err(anyhow!("zed is out of date")),
4229 Status::ConnectionError | Status::ConnectionLost | Status::ReconnectionError { .. } => {
4230 return Err(anyhow!("zed is offline"))
4231 }
4232 }
4233 }
4234
4235 let room = active_call
4236 .update(cx, |active_call, cx| {
4237 active_call.join_channel(channel_id, cx)
4238 })
4239 .await?;
4240
4241 room.update(cx, |room, _| room.room_update_completed())
4242 .await;
4243
4244 let task = room.update(cx, |room, cx| {
4245 if let Some((project, host)) = room.most_active_project(cx) {
4246 return Some(join_remote_project(project, host, app_state.clone(), cx));
4247 }
4248
4249 None
4250 });
4251 if let Some(task) = task {
4252 task.await?;
4253 return anyhow::Ok(true);
4254 }
4255 anyhow::Ok(false)
4256}
4257
4258pub fn join_channel(
4259 channel_id: u64,
4260 app_state: Arc<AppState>,
4261 requesting_window: Option<WindowHandle<Workspace>>,
4262 cx: &mut AppContext,
4263) -> Task<Result<()>> {
4264 let active_call = ActiveCall::global(cx);
4265 cx.spawn(|mut cx| async move {
4266 let result = join_channel_internal(
4267 channel_id,
4268 &app_state,
4269 requesting_window,
4270 &active_call,
4271 &mut cx,
4272 )
4273 .await;
4274
4275 // join channel succeeded, and opened a window
4276 if matches!(result, Ok(true)) {
4277 return anyhow::Ok(());
4278 }
4279
4280 if requesting_window.is_some() {
4281 return anyhow::Ok(());
4282 }
4283
4284 // find an existing workspace to focus and show call controls
4285 let mut active_window = activate_any_workspace_window(&mut cx);
4286 if active_window.is_none() {
4287 // no open workspaces, make one to show the error in (blergh)
4288 cx.update(|cx| Workspace::new_local(vec![], app_state.clone(), requesting_window, cx))
4289 .await;
4290 }
4291
4292 active_window = activate_any_workspace_window(&mut cx);
4293 if active_window.is_none() {
4294 return result.map(|_| ()); // unreachable!() assuming new_local always opens a window
4295 }
4296
4297 if let Err(err) = result {
4298 let prompt = active_window.unwrap().prompt(
4299 PromptLevel::Critical,
4300 &format!("Failed to join channel: {}", err),
4301 &["Ok"],
4302 &mut cx,
4303 );
4304 if let Some(mut prompt) = prompt {
4305 prompt.next().await;
4306 } else {
4307 return Err(err);
4308 }
4309 }
4310
4311 // return ok, we showed the error to the user.
4312 return anyhow::Ok(());
4313 })
4314}
4315
4316pub fn activate_any_workspace_window(cx: &mut AsyncAppContext) -> Option<AnyWindowHandle> {
4317 for window in cx.windows() {
4318 let found = window.update(cx, |cx| {
4319 let is_workspace = cx.root_view().clone().downcast::<Workspace>().is_some();
4320 if is_workspace {
4321 cx.activate_window();
4322 }
4323 is_workspace
4324 });
4325 if found == Some(true) {
4326 return Some(window);
4327 }
4328 }
4329 None
4330}
4331
4332#[allow(clippy::type_complexity)]
4333pub fn open_paths(
4334 abs_paths: &[PathBuf],
4335 app_state: &Arc<AppState>,
4336 requesting_window: Option<WindowHandle<Workspace>>,
4337 cx: &mut AppContext,
4338) -> Task<
4339 Result<(
4340 WeakViewHandle<Workspace>,
4341 Vec<Option<Result<Box<dyn ItemHandle>, anyhow::Error>>>,
4342 )>,
4343> {
4344 let app_state = app_state.clone();
4345 let abs_paths = abs_paths.to_vec();
4346 cx.spawn(|mut cx| async move {
4347 // Open paths in existing workspace if possible
4348 let existing = activate_workspace_for_project(&mut cx, |project, cx| {
4349 project.contains_paths(&abs_paths, cx)
4350 });
4351
4352 if let Some(existing) = existing {
4353 Ok((
4354 existing.clone(),
4355 existing
4356 .update(&mut cx, |workspace, cx| {
4357 workspace.open_paths(abs_paths, true, cx)
4358 })?
4359 .await,
4360 ))
4361 } else {
4362 Ok(cx
4363 .update(|cx| {
4364 Workspace::new_local(abs_paths, app_state.clone(), requesting_window, cx)
4365 })
4366 .await)
4367 }
4368 })
4369}
4370
4371pub fn open_new(
4372 app_state: &Arc<AppState>,
4373 cx: &mut AppContext,
4374 init: impl FnOnce(&mut Workspace, &mut ViewContext<Workspace>) + 'static,
4375) -> Task<()> {
4376 let task = Workspace::new_local(Vec::new(), app_state.clone(), None, cx);
4377 cx.spawn(|mut cx| async move {
4378 let (workspace, opened_paths) = task.await;
4379
4380 workspace
4381 .update(&mut cx, |workspace, cx| {
4382 if opened_paths.is_empty() {
4383 init(workspace, cx)
4384 }
4385 })
4386 .log_err();
4387 })
4388}
4389
4390pub fn create_and_open_local_file(
4391 path: &'static Path,
4392 cx: &mut ViewContext<Workspace>,
4393 default_content: impl 'static + Send + FnOnce() -> Rope,
4394) -> Task<Result<Box<dyn ItemHandle>>> {
4395 cx.spawn(|workspace, mut cx| async move {
4396 let fs = workspace.read_with(&cx, |workspace, _| workspace.app_state().fs.clone())?;
4397 if !fs.is_file(path).await {
4398 fs.create_file(path, Default::default()).await?;
4399 fs.save(path, &default_content(), Default::default())
4400 .await?;
4401 }
4402
4403 let mut items = workspace
4404 .update(&mut cx, |workspace, cx| {
4405 workspace.with_local_workspace(cx, |workspace, cx| {
4406 workspace.open_paths(vec![path.to_path_buf()], false, cx)
4407 })
4408 })?
4409 .await?
4410 .await;
4411
4412 let item = items.pop().flatten();
4413 item.ok_or_else(|| anyhow!("path {path:?} is not a file"))?
4414 })
4415}
4416
4417pub fn join_remote_project(
4418 project_id: u64,
4419 follow_user_id: u64,
4420 app_state: Arc<AppState>,
4421 cx: &mut AppContext,
4422) -> Task<Result<()>> {
4423 cx.spawn(|mut cx| async move {
4424 let windows = cx.windows();
4425 let existing_workspace = windows.into_iter().find_map(|window| {
4426 window.downcast::<Workspace>().and_then(|window| {
4427 window
4428 .read_root_with(&cx, |workspace, cx| {
4429 if workspace.project().read(cx).remote_id() == Some(project_id) {
4430 Some(cx.handle().downgrade())
4431 } else {
4432 None
4433 }
4434 })
4435 .unwrap_or(None)
4436 })
4437 });
4438
4439 let workspace = if let Some(existing_workspace) = existing_workspace {
4440 existing_workspace
4441 } else {
4442 let active_call = cx.read(ActiveCall::global);
4443 let room = active_call
4444 .read_with(&cx, |call, _| call.room().cloned())
4445 .ok_or_else(|| anyhow!("not in a call"))?;
4446 let project = room
4447 .update(&mut cx, |room, cx| {
4448 room.join_project(
4449 project_id,
4450 app_state.languages.clone(),
4451 app_state.fs.clone(),
4452 cx,
4453 )
4454 })
4455 .await?;
4456
4457 let window_bounds_override = window_bounds_env_override(&cx);
4458 let window = cx.add_window(
4459 (app_state.build_window_options)(
4460 window_bounds_override,
4461 None,
4462 cx.platform().as_ref(),
4463 ),
4464 |cx| Workspace::new(0, project, app_state.clone(), cx),
4465 );
4466 let workspace = window.root(&cx).unwrap();
4467 (app_state.initialize_workspace)(
4468 workspace.downgrade(),
4469 false,
4470 app_state.clone(),
4471 cx.clone(),
4472 )
4473 .await
4474 .log_err();
4475
4476 workspace.downgrade()
4477 };
4478
4479 workspace.window().activate(&mut cx);
4480 cx.platform().activate(true);
4481
4482 workspace.update(&mut cx, |workspace, cx| {
4483 if let Some(room) = ActiveCall::global(cx).read(cx).room().cloned() {
4484 let follow_peer_id = room
4485 .read(cx)
4486 .remote_participants()
4487 .iter()
4488 .find(|(_, participant)| participant.user.id == follow_user_id)
4489 .map(|(_, p)| p.peer_id)
4490 .or_else(|| {
4491 // If we couldn't follow the given user, follow the host instead.
4492 let collaborator = workspace
4493 .project()
4494 .read(cx)
4495 .collaborators()
4496 .values()
4497 .find(|collaborator| collaborator.replica_id == 0)?;
4498 Some(collaborator.peer_id)
4499 });
4500
4501 if let Some(follow_peer_id) = follow_peer_id {
4502 workspace
4503 .follow(follow_peer_id, cx)
4504 .map(|follow| follow.detach_and_log_err(cx));
4505 }
4506 }
4507 })?;
4508
4509 anyhow::Ok(())
4510 })
4511}
4512
4513pub fn restart(_: &Restart, cx: &mut AppContext) {
4514 let should_confirm = settings::get::<WorkspaceSettings>(cx).confirm_quit;
4515 cx.spawn(|mut cx| async move {
4516 let mut workspace_windows = cx
4517 .windows()
4518 .into_iter()
4519 .filter_map(|window| window.downcast::<Workspace>())
4520 .collect::<Vec<_>>();
4521
4522 // If multiple windows have unsaved changes, and need a save prompt,
4523 // prompt in the active window before switching to a different window.
4524 workspace_windows.sort_by_key(|window| window.is_active(&cx) == Some(false));
4525
4526 if let (true, Some(window)) = (should_confirm, workspace_windows.first()) {
4527 let answer = window.prompt(
4528 PromptLevel::Info,
4529 "Are you sure you want to restart?",
4530 &["Restart", "Cancel"],
4531 &mut cx,
4532 );
4533
4534 if let Some(mut answer) = answer {
4535 let answer = answer.next().await;
4536 if answer != Some(0) {
4537 return Ok(());
4538 }
4539 }
4540 }
4541
4542 // If the user cancels any save prompt, then keep the app open.
4543 for window in workspace_windows {
4544 if let Some(should_close) = window.update_root(&mut cx, |workspace, cx| {
4545 workspace.prepare_to_close(true, cx)
4546 }) {
4547 if !should_close.await? {
4548 return Ok(());
4549 }
4550 }
4551 }
4552 cx.platform().restart();
4553 anyhow::Ok(())
4554 })
4555 .detach_and_log_err(cx);
4556}
4557
4558fn parse_pixel_position_env_var(value: &str) -> Option<Vector2F> {
4559 let mut parts = value.split(',');
4560 let width: usize = parts.next()?.parse().ok()?;
4561 let height: usize = parts.next()?.parse().ok()?;
4562 Some(vec2f(width as f32, height as f32))
4563}
4564
4565#[cfg(test)]
4566mod tests {
4567 use super::*;
4568 use crate::{
4569 dock::test::{TestPanel, TestPanelEvent},
4570 item::test::{TestItem, TestItemEvent, TestProjectItem},
4571 };
4572 use fs::FakeFs;
4573 use gpui::{executor::Deterministic, test::EmptyView, TestAppContext};
4574 use project::{Project, ProjectEntryId};
4575 use serde_json::json;
4576 use settings::SettingsStore;
4577 use std::{cell::RefCell, rc::Rc};
4578
4579 #[gpui::test]
4580 async fn test_tab_disambiguation(cx: &mut TestAppContext) {
4581 init_test(cx);
4582
4583 let fs = FakeFs::new(cx.background());
4584 let project = Project::test(fs, [], cx).await;
4585 let window = cx.add_window(|cx| Workspace::test_new(project.clone(), cx));
4586 let workspace = window.root(cx);
4587
4588 // Adding an item with no ambiguity renders the tab without detail.
4589 let item1 = window.add_view(cx, |_| {
4590 let mut item = TestItem::new();
4591 item.tab_descriptions = Some(vec!["c", "b1/c", "a/b1/c"]);
4592 item
4593 });
4594 workspace.update(cx, |workspace, cx| {
4595 workspace.add_item(Box::new(item1.clone()), cx);
4596 });
4597 item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), None));
4598
4599 // Adding an item that creates ambiguity increases the level of detail on
4600 // both tabs.
4601 let item2 = window.add_view(cx, |_| {
4602 let mut item = TestItem::new();
4603 item.tab_descriptions = Some(vec!["c", "b2/c", "a/b2/c"]);
4604 item
4605 });
4606 workspace.update(cx, |workspace, cx| {
4607 workspace.add_item(Box::new(item2.clone()), cx);
4608 });
4609 item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1)));
4610 item2.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1)));
4611
4612 // Adding an item that creates ambiguity increases the level of detail only
4613 // on the ambiguous tabs. In this case, the ambiguity can't be resolved so
4614 // we stop at the highest detail available.
4615 let item3 = window.add_view(cx, |_| {
4616 let mut item = TestItem::new();
4617 item.tab_descriptions = Some(vec!["c", "b2/c", "a/b2/c"]);
4618 item
4619 });
4620 workspace.update(cx, |workspace, cx| {
4621 workspace.add_item(Box::new(item3.clone()), cx);
4622 });
4623 item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1)));
4624 item2.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(3)));
4625 item3.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(3)));
4626 }
4627
4628 #[gpui::test]
4629 async fn test_tracking_active_path(cx: &mut TestAppContext) {
4630 init_test(cx);
4631
4632 let fs = FakeFs::new(cx.background());
4633 fs.insert_tree(
4634 "/root1",
4635 json!({
4636 "one.txt": "",
4637 "two.txt": "",
4638 }),
4639 )
4640 .await;
4641 fs.insert_tree(
4642 "/root2",
4643 json!({
4644 "three.txt": "",
4645 }),
4646 )
4647 .await;
4648
4649 let project = Project::test(fs, ["root1".as_ref()], cx).await;
4650 let window = cx.add_window(|cx| Workspace::test_new(project.clone(), cx));
4651 let workspace = window.root(cx);
4652 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
4653 let worktree_id = project.read_with(cx, |project, cx| {
4654 project.worktrees(cx).next().unwrap().read(cx).id()
4655 });
4656
4657 let item1 = window.add_view(cx, |cx| {
4658 TestItem::new().with_project_items(&[TestProjectItem::new(1, "one.txt", cx)])
4659 });
4660 let item2 = window.add_view(cx, |cx| {
4661 TestItem::new().with_project_items(&[TestProjectItem::new(2, "two.txt", cx)])
4662 });
4663
4664 // Add an item to an empty pane
4665 workspace.update(cx, |workspace, cx| workspace.add_item(Box::new(item1), cx));
4666 project.read_with(cx, |project, cx| {
4667 assert_eq!(
4668 project.active_entry(),
4669 project
4670 .entry_for_path(&(worktree_id, "one.txt").into(), cx)
4671 .map(|e| e.id)
4672 );
4673 });
4674 assert_eq!(window.current_title(cx).as_deref(), Some("one.txt — root1"));
4675
4676 // Add a second item to a non-empty pane
4677 workspace.update(cx, |workspace, cx| workspace.add_item(Box::new(item2), cx));
4678 assert_eq!(window.current_title(cx).as_deref(), Some("two.txt — root1"));
4679 project.read_with(cx, |project, cx| {
4680 assert_eq!(
4681 project.active_entry(),
4682 project
4683 .entry_for_path(&(worktree_id, "two.txt").into(), cx)
4684 .map(|e| e.id)
4685 );
4686 });
4687
4688 // Close the active item
4689 pane.update(cx, |pane, cx| {
4690 pane.close_active_item(&Default::default(), cx).unwrap()
4691 })
4692 .await
4693 .unwrap();
4694 assert_eq!(window.current_title(cx).as_deref(), Some("one.txt — root1"));
4695 project.read_with(cx, |project, cx| {
4696 assert_eq!(
4697 project.active_entry(),
4698 project
4699 .entry_for_path(&(worktree_id, "one.txt").into(), cx)
4700 .map(|e| e.id)
4701 );
4702 });
4703
4704 // Add a project folder
4705 project
4706 .update(cx, |project, cx| {
4707 project.find_or_create_local_worktree("/root2", true, cx)
4708 })
4709 .await
4710 .unwrap();
4711 assert_eq!(
4712 window.current_title(cx).as_deref(),
4713 Some("one.txt — root1, root2")
4714 );
4715
4716 // Remove a project folder
4717 project.update(cx, |project, cx| project.remove_worktree(worktree_id, cx));
4718 assert_eq!(window.current_title(cx).as_deref(), Some("one.txt — root2"));
4719 }
4720
4721 #[gpui::test]
4722 async fn test_close_window(cx: &mut TestAppContext) {
4723 init_test(cx);
4724
4725 let fs = FakeFs::new(cx.background());
4726 fs.insert_tree("/root", json!({ "one": "" })).await;
4727
4728 let project = Project::test(fs, ["root".as_ref()], cx).await;
4729 let window = cx.add_window(|cx| Workspace::test_new(project.clone(), cx));
4730 let workspace = window.root(cx);
4731
4732 // When there are no dirty items, there's nothing to do.
4733 let item1 = window.add_view(cx, |_| TestItem::new());
4734 workspace.update(cx, |w, cx| w.add_item(Box::new(item1.clone()), cx));
4735 let task = workspace.update(cx, |w, cx| w.prepare_to_close(false, cx));
4736 assert!(task.await.unwrap());
4737
4738 // When there are dirty untitled items, prompt to save each one. If the user
4739 // cancels any prompt, then abort.
4740 let item2 = window.add_view(cx, |_| TestItem::new().with_dirty(true));
4741 let item3 = window.add_view(cx, |cx| {
4742 TestItem::new()
4743 .with_dirty(true)
4744 .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
4745 });
4746 workspace.update(cx, |w, cx| {
4747 w.add_item(Box::new(item2.clone()), cx);
4748 w.add_item(Box::new(item3.clone()), cx);
4749 });
4750 let task = workspace.update(cx, |w, cx| w.prepare_to_close(false, cx));
4751 cx.foreground().run_until_parked();
4752 window.simulate_prompt_answer(2, cx); // cancel save all
4753 cx.foreground().run_until_parked();
4754 window.simulate_prompt_answer(2, cx); // cancel save all
4755 cx.foreground().run_until_parked();
4756 assert!(!window.has_pending_prompt(cx));
4757 assert!(!task.await.unwrap());
4758 }
4759
4760 #[gpui::test]
4761 async fn test_close_pane_items(cx: &mut TestAppContext) {
4762 init_test(cx);
4763
4764 let fs = FakeFs::new(cx.background());
4765
4766 let project = Project::test(fs, None, cx).await;
4767 let window = cx.add_window(|cx| Workspace::test_new(project, cx));
4768 let workspace = window.root(cx);
4769
4770 let item1 = window.add_view(cx, |cx| {
4771 TestItem::new()
4772 .with_dirty(true)
4773 .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
4774 });
4775 let item2 = window.add_view(cx, |cx| {
4776 TestItem::new()
4777 .with_dirty(true)
4778 .with_conflict(true)
4779 .with_project_items(&[TestProjectItem::new(2, "2.txt", cx)])
4780 });
4781 let item3 = window.add_view(cx, |cx| {
4782 TestItem::new()
4783 .with_dirty(true)
4784 .with_conflict(true)
4785 .with_project_items(&[TestProjectItem::new(3, "3.txt", cx)])
4786 });
4787 let item4 = window.add_view(cx, |cx| {
4788 TestItem::new()
4789 .with_dirty(true)
4790 .with_project_items(&[TestProjectItem::new_untitled(cx)])
4791 });
4792 let pane = workspace.update(cx, |workspace, cx| {
4793 workspace.add_item(Box::new(item1.clone()), cx);
4794 workspace.add_item(Box::new(item2.clone()), cx);
4795 workspace.add_item(Box::new(item3.clone()), cx);
4796 workspace.add_item(Box::new(item4.clone()), cx);
4797 workspace.active_pane().clone()
4798 });
4799
4800 let close_items = pane.update(cx, |pane, cx| {
4801 pane.activate_item(1, true, true, cx);
4802 assert_eq!(pane.active_item().unwrap().id(), item2.id());
4803 let item1_id = item1.id();
4804 let item3_id = item3.id();
4805 let item4_id = item4.id();
4806 pane.close_items(cx, SaveIntent::Close, move |id| {
4807 [item1_id, item3_id, item4_id].contains(&id)
4808 })
4809 });
4810 cx.foreground().run_until_parked();
4811
4812 assert!(window.has_pending_prompt(cx));
4813 // Ignore "Save all" prompt
4814 window.simulate_prompt_answer(2, cx);
4815 cx.foreground().run_until_parked();
4816 // There's a prompt to save item 1.
4817 pane.read_with(cx, |pane, _| {
4818 assert_eq!(pane.items_len(), 4);
4819 assert_eq!(pane.active_item().unwrap().id(), item1.id());
4820 });
4821 // Confirm saving item 1.
4822 window.simulate_prompt_answer(0, cx);
4823 cx.foreground().run_until_parked();
4824
4825 // Item 1 is saved. There's a prompt to save item 3.
4826 pane.read_with(cx, |pane, cx| {
4827 assert_eq!(item1.read(cx).save_count, 1);
4828 assert_eq!(item1.read(cx).save_as_count, 0);
4829 assert_eq!(item1.read(cx).reload_count, 0);
4830 assert_eq!(pane.items_len(), 3);
4831 assert_eq!(pane.active_item().unwrap().id(), item3.id());
4832 });
4833 assert!(window.has_pending_prompt(cx));
4834
4835 // Cancel saving item 3.
4836 window.simulate_prompt_answer(1, cx);
4837 cx.foreground().run_until_parked();
4838
4839 // Item 3 is reloaded. There's a prompt to save item 4.
4840 pane.read_with(cx, |pane, cx| {
4841 assert_eq!(item3.read(cx).save_count, 0);
4842 assert_eq!(item3.read(cx).save_as_count, 0);
4843 assert_eq!(item3.read(cx).reload_count, 1);
4844 assert_eq!(pane.items_len(), 2);
4845 assert_eq!(pane.active_item().unwrap().id(), item4.id());
4846 });
4847 assert!(window.has_pending_prompt(cx));
4848
4849 // Confirm saving item 4.
4850 window.simulate_prompt_answer(0, cx);
4851 cx.foreground().run_until_parked();
4852
4853 // There's a prompt for a path for item 4.
4854 cx.simulate_new_path_selection(|_| Some(Default::default()));
4855 close_items.await.unwrap();
4856
4857 // The requested items are closed.
4858 pane.read_with(cx, |pane, cx| {
4859 assert_eq!(item4.read(cx).save_count, 0);
4860 assert_eq!(item4.read(cx).save_as_count, 1);
4861 assert_eq!(item4.read(cx).reload_count, 0);
4862 assert_eq!(pane.items_len(), 1);
4863 assert_eq!(pane.active_item().unwrap().id(), item2.id());
4864 });
4865 }
4866
4867 #[gpui::test]
4868 async fn test_prompting_to_save_only_on_last_item_for_entry(cx: &mut TestAppContext) {
4869 init_test(cx);
4870
4871 let fs = FakeFs::new(cx.background());
4872
4873 let project = Project::test(fs, [], cx).await;
4874 let window = cx.add_window(|cx| Workspace::test_new(project, cx));
4875 let workspace = window.root(cx);
4876
4877 // Create several workspace items with single project entries, and two
4878 // workspace items with multiple project entries.
4879 let single_entry_items = (0..=4)
4880 .map(|project_entry_id| {
4881 window.add_view(cx, |cx| {
4882 TestItem::new()
4883 .with_dirty(true)
4884 .with_project_items(&[TestProjectItem::new(
4885 project_entry_id,
4886 &format!("{project_entry_id}.txt"),
4887 cx,
4888 )])
4889 })
4890 })
4891 .collect::<Vec<_>>();
4892 let item_2_3 = window.add_view(cx, |cx| {
4893 TestItem::new()
4894 .with_dirty(true)
4895 .with_singleton(false)
4896 .with_project_items(&[
4897 single_entry_items[2].read(cx).project_items[0].clone(),
4898 single_entry_items[3].read(cx).project_items[0].clone(),
4899 ])
4900 });
4901 let item_3_4 = window.add_view(cx, |cx| {
4902 TestItem::new()
4903 .with_dirty(true)
4904 .with_singleton(false)
4905 .with_project_items(&[
4906 single_entry_items[3].read(cx).project_items[0].clone(),
4907 single_entry_items[4].read(cx).project_items[0].clone(),
4908 ])
4909 });
4910
4911 // Create two panes that contain the following project entries:
4912 // left pane:
4913 // multi-entry items: (2, 3)
4914 // single-entry items: 0, 1, 2, 3, 4
4915 // right pane:
4916 // single-entry items: 1
4917 // multi-entry items: (3, 4)
4918 let left_pane = workspace.update(cx, |workspace, cx| {
4919 let left_pane = workspace.active_pane().clone();
4920 workspace.add_item(Box::new(item_2_3.clone()), cx);
4921 for item in single_entry_items {
4922 workspace.add_item(Box::new(item), cx);
4923 }
4924 left_pane.update(cx, |pane, cx| {
4925 pane.activate_item(2, true, true, cx);
4926 });
4927
4928 workspace
4929 .split_and_clone(left_pane.clone(), SplitDirection::Right, cx)
4930 .unwrap();
4931
4932 left_pane
4933 });
4934
4935 //Need to cause an effect flush in order to respect new focus
4936 workspace.update(cx, |workspace, cx| {
4937 workspace.add_item(Box::new(item_3_4.clone()), cx);
4938 cx.focus(&left_pane);
4939 });
4940
4941 // When closing all of the items in the left pane, we should be prompted twice:
4942 // once for project entry 0, and once for project entry 2. After those two
4943 // prompts, the task should complete.
4944
4945 let close = left_pane.update(cx, |pane, cx| {
4946 pane.close_items(cx, SaveIntent::Close, move |_| true)
4947 });
4948 cx.foreground().run_until_parked();
4949 // Discard "Save all" prompt
4950 window.simulate_prompt_answer(2, cx);
4951
4952 cx.foreground().run_until_parked();
4953 left_pane.read_with(cx, |pane, cx| {
4954 assert_eq!(
4955 pane.active_item().unwrap().project_entry_ids(cx).as_slice(),
4956 &[ProjectEntryId::from_proto(0)]
4957 );
4958 });
4959 window.simulate_prompt_answer(0, cx);
4960
4961 cx.foreground().run_until_parked();
4962 left_pane.read_with(cx, |pane, cx| {
4963 assert_eq!(
4964 pane.active_item().unwrap().project_entry_ids(cx).as_slice(),
4965 &[ProjectEntryId::from_proto(2)]
4966 );
4967 });
4968 window.simulate_prompt_answer(0, cx);
4969
4970 cx.foreground().run_until_parked();
4971 close.await.unwrap();
4972 left_pane.read_with(cx, |pane, _| {
4973 assert_eq!(pane.items_len(), 0);
4974 });
4975 }
4976
4977 #[gpui::test]
4978 async fn test_autosave(deterministic: Arc<Deterministic>, cx: &mut gpui::TestAppContext) {
4979 init_test(cx);
4980
4981 let fs = FakeFs::new(cx.background());
4982
4983 let project = Project::test(fs, [], cx).await;
4984 let window = cx.add_window(|cx| Workspace::test_new(project, cx));
4985 let workspace = window.root(cx);
4986 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
4987
4988 let item = window.add_view(cx, |cx| {
4989 TestItem::new().with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
4990 });
4991 let item_id = item.id();
4992 workspace.update(cx, |workspace, cx| {
4993 workspace.add_item(Box::new(item.clone()), cx);
4994 });
4995
4996 // Autosave on window change.
4997 item.update(cx, |item, cx| {
4998 cx.update_global(|settings: &mut SettingsStore, cx| {
4999 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
5000 settings.autosave = Some(AutosaveSetting::OnWindowChange);
5001 })
5002 });
5003 item.is_dirty = true;
5004 });
5005
5006 // Deactivating the window saves the file.
5007 window.simulate_deactivation(cx);
5008 deterministic.run_until_parked();
5009 item.read_with(cx, |item, _| assert_eq!(item.save_count, 1));
5010
5011 // Autosave on focus change.
5012 item.update(cx, |item, cx| {
5013 cx.focus_self();
5014 cx.update_global(|settings: &mut SettingsStore, cx| {
5015 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
5016 settings.autosave = Some(AutosaveSetting::OnFocusChange);
5017 })
5018 });
5019 item.is_dirty = true;
5020 });
5021
5022 // Blurring the item saves the file.
5023 item.update(cx, |_, cx| cx.blur());
5024 deterministic.run_until_parked();
5025 item.read_with(cx, |item, _| assert_eq!(item.save_count, 2));
5026
5027 // Deactivating the window still saves the file.
5028 window.simulate_activation(cx);
5029 item.update(cx, |item, cx| {
5030 cx.focus_self();
5031 item.is_dirty = true;
5032 });
5033 window.simulate_deactivation(cx);
5034
5035 deterministic.run_until_parked();
5036 item.read_with(cx, |item, _| assert_eq!(item.save_count, 3));
5037
5038 // Autosave after delay.
5039 item.update(cx, |item, cx| {
5040 cx.update_global(|settings: &mut SettingsStore, cx| {
5041 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
5042 settings.autosave = Some(AutosaveSetting::AfterDelay { milliseconds: 500 });
5043 })
5044 });
5045 item.is_dirty = true;
5046 cx.emit(TestItemEvent::Edit);
5047 });
5048
5049 // Delay hasn't fully expired, so the file is still dirty and unsaved.
5050 deterministic.advance_clock(Duration::from_millis(250));
5051 item.read_with(cx, |item, _| assert_eq!(item.save_count, 3));
5052
5053 // After delay expires, the file is saved.
5054 deterministic.advance_clock(Duration::from_millis(250));
5055 item.read_with(cx, |item, _| assert_eq!(item.save_count, 4));
5056
5057 // Autosave on focus change, ensuring closing the tab counts as such.
5058 item.update(cx, |item, cx| {
5059 cx.update_global(|settings: &mut SettingsStore, cx| {
5060 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
5061 settings.autosave = Some(AutosaveSetting::OnFocusChange);
5062 })
5063 });
5064 item.is_dirty = true;
5065 });
5066
5067 pane.update(cx, |pane, cx| {
5068 pane.close_items(cx, SaveIntent::Close, move |id| id == item_id)
5069 })
5070 .await
5071 .unwrap();
5072 assert!(!window.has_pending_prompt(cx));
5073 item.read_with(cx, |item, _| assert_eq!(item.save_count, 5));
5074
5075 // Add the item again, ensuring autosave is prevented if the underlying file has been deleted.
5076 workspace.update(cx, |workspace, cx| {
5077 workspace.add_item(Box::new(item.clone()), cx);
5078 });
5079 item.update(cx, |item, cx| {
5080 item.project_items[0].update(cx, |item, _| {
5081 item.entry_id = None;
5082 });
5083 item.is_dirty = true;
5084 cx.blur();
5085 });
5086 deterministic.run_until_parked();
5087 item.read_with(cx, |item, _| assert_eq!(item.save_count, 5));
5088
5089 // Ensure autosave is prevented for deleted files also when closing the buffer.
5090 let _close_items = pane.update(cx, |pane, cx| {
5091 pane.close_items(cx, SaveIntent::Close, move |id| id == item_id)
5092 });
5093 deterministic.run_until_parked();
5094 assert!(window.has_pending_prompt(cx));
5095 item.read_with(cx, |item, _| assert_eq!(item.save_count, 5));
5096 }
5097
5098 #[gpui::test]
5099 async fn test_pane_navigation(cx: &mut gpui::TestAppContext) {
5100 init_test(cx);
5101
5102 let fs = FakeFs::new(cx.background());
5103
5104 let project = Project::test(fs, [], cx).await;
5105 let window = cx.add_window(|cx| Workspace::test_new(project, cx));
5106 let workspace = window.root(cx);
5107
5108 let item = window.add_view(cx, |cx| {
5109 TestItem::new().with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
5110 });
5111 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
5112 let toolbar = pane.read_with(cx, |pane, _| pane.toolbar().clone());
5113 let toolbar_notify_count = Rc::new(RefCell::new(0));
5114
5115 workspace.update(cx, |workspace, cx| {
5116 workspace.add_item(Box::new(item.clone()), cx);
5117 let toolbar_notification_count = toolbar_notify_count.clone();
5118 cx.observe(&toolbar, move |_, _, _| {
5119 *toolbar_notification_count.borrow_mut() += 1
5120 })
5121 .detach();
5122 });
5123
5124 pane.read_with(cx, |pane, _| {
5125 assert!(!pane.can_navigate_backward());
5126 assert!(!pane.can_navigate_forward());
5127 });
5128
5129 item.update(cx, |item, cx| {
5130 item.set_state("one".to_string(), cx);
5131 });
5132
5133 // Toolbar must be notified to re-render the navigation buttons
5134 assert_eq!(*toolbar_notify_count.borrow(), 1);
5135
5136 pane.read_with(cx, |pane, _| {
5137 assert!(pane.can_navigate_backward());
5138 assert!(!pane.can_navigate_forward());
5139 });
5140
5141 workspace
5142 .update(cx, |workspace, cx| workspace.go_back(pane.downgrade(), cx))
5143 .await
5144 .unwrap();
5145
5146 assert_eq!(*toolbar_notify_count.borrow(), 3);
5147 pane.read_with(cx, |pane, _| {
5148 assert!(!pane.can_navigate_backward());
5149 assert!(pane.can_navigate_forward());
5150 });
5151 }
5152
5153 #[gpui::test]
5154 async fn test_toggle_docks_and_panels(cx: &mut gpui::TestAppContext) {
5155 init_test(cx);
5156 let fs = FakeFs::new(cx.background());
5157
5158 let project = Project::test(fs, [], cx).await;
5159 let window = cx.add_window(|cx| Workspace::test_new(project, cx));
5160 let workspace = window.root(cx);
5161
5162 let panel = workspace.update(cx, |workspace, cx| {
5163 let panel = cx.add_view(|_| TestPanel::new(DockPosition::Right));
5164 workspace.add_panel(panel.clone(), cx);
5165
5166 workspace
5167 .right_dock()
5168 .update(cx, |right_dock, cx| right_dock.set_open(true, cx));
5169
5170 panel
5171 });
5172
5173 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
5174 pane.update(cx, |pane, cx| {
5175 let item = cx.add_view(|_| TestItem::new());
5176 pane.add_item(Box::new(item), true, true, None, cx);
5177 });
5178
5179 // Transfer focus from center to panel
5180 workspace.update(cx, |workspace, cx| {
5181 workspace.toggle_panel_focus::<TestPanel>(cx);
5182 });
5183
5184 workspace.read_with(cx, |workspace, cx| {
5185 assert!(workspace.right_dock().read(cx).is_open());
5186 assert!(!panel.is_zoomed(cx));
5187 assert!(panel.has_focus(cx));
5188 });
5189
5190 // Transfer focus from panel to center
5191 workspace.update(cx, |workspace, cx| {
5192 workspace.toggle_panel_focus::<TestPanel>(cx);
5193 });
5194
5195 workspace.read_with(cx, |workspace, cx| {
5196 assert!(workspace.right_dock().read(cx).is_open());
5197 assert!(!panel.is_zoomed(cx));
5198 assert!(!panel.has_focus(cx));
5199 });
5200
5201 // Close the dock
5202 workspace.update(cx, |workspace, cx| {
5203 workspace.toggle_dock(DockPosition::Right, cx);
5204 });
5205
5206 workspace.read_with(cx, |workspace, cx| {
5207 assert!(!workspace.right_dock().read(cx).is_open());
5208 assert!(!panel.is_zoomed(cx));
5209 assert!(!panel.has_focus(cx));
5210 });
5211
5212 // Open the dock
5213 workspace.update(cx, |workspace, cx| {
5214 workspace.toggle_dock(DockPosition::Right, cx);
5215 });
5216
5217 workspace.read_with(cx, |workspace, cx| {
5218 assert!(workspace.right_dock().read(cx).is_open());
5219 assert!(!panel.is_zoomed(cx));
5220 assert!(panel.has_focus(cx));
5221 });
5222
5223 // Focus and zoom panel
5224 panel.update(cx, |panel, cx| {
5225 cx.focus_self();
5226 panel.set_zoomed(true, cx)
5227 });
5228
5229 workspace.read_with(cx, |workspace, cx| {
5230 assert!(workspace.right_dock().read(cx).is_open());
5231 assert!(panel.is_zoomed(cx));
5232 assert!(panel.has_focus(cx));
5233 });
5234
5235 // Transfer focus to the center closes the dock
5236 workspace.update(cx, |workspace, cx| {
5237 workspace.toggle_panel_focus::<TestPanel>(cx);
5238 });
5239
5240 workspace.read_with(cx, |workspace, cx| {
5241 assert!(!workspace.right_dock().read(cx).is_open());
5242 assert!(panel.is_zoomed(cx));
5243 assert!(!panel.has_focus(cx));
5244 });
5245
5246 // Transferring focus back to the panel keeps it zoomed
5247 workspace.update(cx, |workspace, cx| {
5248 workspace.toggle_panel_focus::<TestPanel>(cx);
5249 });
5250
5251 workspace.read_with(cx, |workspace, cx| {
5252 assert!(workspace.right_dock().read(cx).is_open());
5253 assert!(panel.is_zoomed(cx));
5254 assert!(panel.has_focus(cx));
5255 });
5256
5257 // Close the dock while it is zoomed
5258 workspace.update(cx, |workspace, cx| {
5259 workspace.toggle_dock(DockPosition::Right, cx)
5260 });
5261
5262 workspace.read_with(cx, |workspace, cx| {
5263 assert!(!workspace.right_dock().read(cx).is_open());
5264 assert!(panel.is_zoomed(cx));
5265 assert!(workspace.zoomed.is_none());
5266 assert!(!panel.has_focus(cx));
5267 });
5268
5269 // Opening the dock, when it's zoomed, retains focus
5270 workspace.update(cx, |workspace, cx| {
5271 workspace.toggle_dock(DockPosition::Right, cx)
5272 });
5273
5274 workspace.read_with(cx, |workspace, cx| {
5275 assert!(workspace.right_dock().read(cx).is_open());
5276 assert!(panel.is_zoomed(cx));
5277 assert!(workspace.zoomed.is_some());
5278 assert!(panel.has_focus(cx));
5279 });
5280
5281 // Unzoom and close the panel, zoom the active pane.
5282 panel.update(cx, |panel, cx| panel.set_zoomed(false, cx));
5283 workspace.update(cx, |workspace, cx| {
5284 workspace.toggle_dock(DockPosition::Right, cx)
5285 });
5286 pane.update(cx, |pane, cx| pane.toggle_zoom(&Default::default(), cx));
5287
5288 // Opening a dock unzooms the pane.
5289 workspace.update(cx, |workspace, cx| {
5290 workspace.toggle_dock(DockPosition::Right, cx)
5291 });
5292 workspace.read_with(cx, |workspace, cx| {
5293 let pane = pane.read(cx);
5294 assert!(!pane.is_zoomed());
5295 assert!(!pane.has_focus());
5296 assert!(workspace.right_dock().read(cx).is_open());
5297 assert!(workspace.zoomed.is_none());
5298 });
5299 }
5300
5301 #[gpui::test]
5302 async fn test_panels(cx: &mut gpui::TestAppContext) {
5303 init_test(cx);
5304 let fs = FakeFs::new(cx.background());
5305
5306 let project = Project::test(fs, [], cx).await;
5307 let window = cx.add_window(|cx| Workspace::test_new(project, cx));
5308 let workspace = window.root(cx);
5309
5310 let (panel_1, panel_2) = workspace.update(cx, |workspace, cx| {
5311 // Add panel_1 on the left, panel_2 on the right.
5312 let panel_1 = cx.add_view(|_| TestPanel::new(DockPosition::Left));
5313 workspace.add_panel(panel_1.clone(), cx);
5314 workspace
5315 .left_dock()
5316 .update(cx, |left_dock, cx| left_dock.set_open(true, cx));
5317 let panel_2 = cx.add_view(|_| TestPanel::new(DockPosition::Right));
5318 workspace.add_panel(panel_2.clone(), cx);
5319 workspace
5320 .right_dock()
5321 .update(cx, |right_dock, cx| right_dock.set_open(true, cx));
5322
5323 let left_dock = workspace.left_dock();
5324 assert_eq!(
5325 left_dock.read(cx).visible_panel().unwrap().id(),
5326 panel_1.id()
5327 );
5328 assert_eq!(
5329 left_dock.read(cx).active_panel_size(cx).unwrap(),
5330 panel_1.size(cx)
5331 );
5332
5333 left_dock.update(cx, |left_dock, cx| {
5334 left_dock.resize_active_panel(Some(1337.), cx)
5335 });
5336 assert_eq!(
5337 workspace
5338 .right_dock()
5339 .read(cx)
5340 .visible_panel()
5341 .unwrap()
5342 .id(),
5343 panel_2.id()
5344 );
5345
5346 (panel_1, panel_2)
5347 });
5348
5349 // Move panel_1 to the right
5350 panel_1.update(cx, |panel_1, cx| {
5351 panel_1.set_position(DockPosition::Right, cx)
5352 });
5353
5354 workspace.update(cx, |workspace, cx| {
5355 // Since panel_1 was visible on the left, it should now be visible now that it's been moved to the right.
5356 // Since it was the only panel on the left, the left dock should now be closed.
5357 assert!(!workspace.left_dock().read(cx).is_open());
5358 assert!(workspace.left_dock().read(cx).visible_panel().is_none());
5359 let right_dock = workspace.right_dock();
5360 assert_eq!(
5361 right_dock.read(cx).visible_panel().unwrap().id(),
5362 panel_1.id()
5363 );
5364 assert_eq!(right_dock.read(cx).active_panel_size(cx).unwrap(), 1337.);
5365
5366 // Now we move panel_2 to the left
5367 panel_2.set_position(DockPosition::Left, cx);
5368 });
5369
5370 workspace.update(cx, |workspace, cx| {
5371 // Since panel_2 was not visible on the right, we don't open the left dock.
5372 assert!(!workspace.left_dock().read(cx).is_open());
5373 // And the right dock is unaffected in it's displaying of panel_1
5374 assert!(workspace.right_dock().read(cx).is_open());
5375 assert_eq!(
5376 workspace
5377 .right_dock()
5378 .read(cx)
5379 .visible_panel()
5380 .unwrap()
5381 .id(),
5382 panel_1.id()
5383 );
5384 });
5385
5386 // Move panel_1 back to the left
5387 panel_1.update(cx, |panel_1, cx| {
5388 panel_1.set_position(DockPosition::Left, cx)
5389 });
5390
5391 workspace.update(cx, |workspace, cx| {
5392 // Since panel_1 was visible on the right, we open the left dock and make panel_1 active.
5393 let left_dock = workspace.left_dock();
5394 assert!(left_dock.read(cx).is_open());
5395 assert_eq!(
5396 left_dock.read(cx).visible_panel().unwrap().id(),
5397 panel_1.id()
5398 );
5399 assert_eq!(left_dock.read(cx).active_panel_size(cx).unwrap(), 1337.);
5400 // And right the dock should be closed as it no longer has any panels.
5401 assert!(!workspace.right_dock().read(cx).is_open());
5402
5403 // Now we move panel_1 to the bottom
5404 panel_1.set_position(DockPosition::Bottom, cx);
5405 });
5406
5407 workspace.update(cx, |workspace, cx| {
5408 // Since panel_1 was visible on the left, we close the left dock.
5409 assert!(!workspace.left_dock().read(cx).is_open());
5410 // The bottom dock is sized based on the panel's default size,
5411 // since the panel orientation changed from vertical to horizontal.
5412 let bottom_dock = workspace.bottom_dock();
5413 assert_eq!(
5414 bottom_dock.read(cx).active_panel_size(cx).unwrap(),
5415 panel_1.size(cx),
5416 );
5417 // Close bottom dock and move panel_1 back to the left.
5418 bottom_dock.update(cx, |bottom_dock, cx| bottom_dock.set_open(false, cx));
5419 panel_1.set_position(DockPosition::Left, cx);
5420 });
5421
5422 // Emit activated event on panel 1
5423 panel_1.update(cx, |_, cx| cx.emit(TestPanelEvent::Activated));
5424
5425 // Now the left dock is open and panel_1 is active and focused.
5426 workspace.read_with(cx, |workspace, cx| {
5427 let left_dock = workspace.left_dock();
5428 assert!(left_dock.read(cx).is_open());
5429 assert_eq!(
5430 left_dock.read(cx).visible_panel().unwrap().id(),
5431 panel_1.id()
5432 );
5433 assert!(panel_1.is_focused(cx));
5434 });
5435
5436 // Emit closed event on panel 2, which is not active
5437 panel_2.update(cx, |_, cx| cx.emit(TestPanelEvent::Closed));
5438
5439 // Wo don't close the left dock, because panel_2 wasn't the active panel
5440 workspace.read_with(cx, |workspace, cx| {
5441 let left_dock = workspace.left_dock();
5442 assert!(left_dock.read(cx).is_open());
5443 assert_eq!(
5444 left_dock.read(cx).visible_panel().unwrap().id(),
5445 panel_1.id()
5446 );
5447 });
5448
5449 // Emitting a ZoomIn event shows the panel as zoomed.
5450 panel_1.update(cx, |_, cx| cx.emit(TestPanelEvent::ZoomIn));
5451 workspace.read_with(cx, |workspace, _| {
5452 assert_eq!(workspace.zoomed, Some(panel_1.downgrade().into_any()));
5453 assert_eq!(workspace.zoomed_position, Some(DockPosition::Left));
5454 });
5455
5456 // Move panel to another dock while it is zoomed
5457 panel_1.update(cx, |panel, cx| panel.set_position(DockPosition::Right, cx));
5458 workspace.read_with(cx, |workspace, _| {
5459 assert_eq!(workspace.zoomed, Some(panel_1.downgrade().into_any()));
5460 assert_eq!(workspace.zoomed_position, Some(DockPosition::Right));
5461 });
5462
5463 // If focus is transferred to another view that's not a panel or another pane, we still show
5464 // the panel as zoomed.
5465 let focus_receiver = window.add_view(cx, |_| EmptyView);
5466 focus_receiver.update(cx, |_, cx| cx.focus_self());
5467 workspace.read_with(cx, |workspace, _| {
5468 assert_eq!(workspace.zoomed, Some(panel_1.downgrade().into_any()));
5469 assert_eq!(workspace.zoomed_position, Some(DockPosition::Right));
5470 });
5471
5472 // If focus is transferred elsewhere in the workspace, the panel is no longer zoomed.
5473 workspace.update(cx, |_, cx| cx.focus_self());
5474 workspace.read_with(cx, |workspace, _| {
5475 assert_eq!(workspace.zoomed, None);
5476 assert_eq!(workspace.zoomed_position, None);
5477 });
5478
5479 // If focus is transferred again to another view that's not a panel or a pane, we won't
5480 // show the panel as zoomed because it wasn't zoomed before.
5481 focus_receiver.update(cx, |_, cx| cx.focus_self());
5482 workspace.read_with(cx, |workspace, _| {
5483 assert_eq!(workspace.zoomed, None);
5484 assert_eq!(workspace.zoomed_position, None);
5485 });
5486
5487 // When focus is transferred back to the panel, it is zoomed again.
5488 panel_1.update(cx, |_, cx| cx.focus_self());
5489 workspace.read_with(cx, |workspace, _| {
5490 assert_eq!(workspace.zoomed, Some(panel_1.downgrade().into_any()));
5491 assert_eq!(workspace.zoomed_position, Some(DockPosition::Right));
5492 });
5493
5494 // Emitting a ZoomOut event unzooms the panel.
5495 panel_1.update(cx, |_, cx| cx.emit(TestPanelEvent::ZoomOut));
5496 workspace.read_with(cx, |workspace, _| {
5497 assert_eq!(workspace.zoomed, None);
5498 assert_eq!(workspace.zoomed_position, None);
5499 });
5500
5501 // Emit closed event on panel 1, which is active
5502 panel_1.update(cx, |_, cx| cx.emit(TestPanelEvent::Closed));
5503
5504 // Now the left dock is closed, because panel_1 was the active panel
5505 workspace.read_with(cx, |workspace, cx| {
5506 let right_dock = workspace.right_dock();
5507 assert!(!right_dock.read(cx).is_open());
5508 });
5509 }
5510
5511 pub fn init_test(cx: &mut TestAppContext) {
5512 cx.foreground().forbid_parking();
5513 cx.update(|cx| {
5514 cx.set_global(SettingsStore::test(cx));
5515 theme::init((), cx);
5516 language::init(cx);
5517 crate::init_settings(cx);
5518 Project::init_settings(cx);
5519 });
5520 }
5521}