1pub mod dock;
2pub mod history_manager;
3pub mod item;
4mod modal_layer;
5pub mod notifications;
6pub mod pane;
7pub mod pane_group;
8mod persistence;
9pub mod searchable;
10pub mod shared_screen;
11mod status_bar;
12pub mod tasks;
13mod theme_preview;
14mod toast_layer;
15mod toolbar;
16mod workspace_settings;
17
18pub use crate::notifications::NotificationFrame;
19pub use dock::Panel;
20pub use toast_layer::{ToastAction, ToastLayer, ToastView};
21
22use anyhow::{Context as _, Result, anyhow};
23use call::{ActiveCall, call_settings::CallSettings};
24use client::{
25 ChannelId, Client, ErrorExt, Status, TypedEnvelope, UserStore,
26 proto::{self, ErrorCode, PanelId, PeerId},
27};
28use collections::{HashMap, HashSet, hash_map};
29use dock::{Dock, DockPosition, PanelButtons, PanelHandle, RESIZE_HANDLE_SIZE};
30use futures::{
31 Future, FutureExt, StreamExt,
32 channel::{
33 mpsc::{self, UnboundedReceiver, UnboundedSender},
34 oneshot,
35 },
36 future::{Shared, try_join_all},
37};
38use gpui::{
39 Action, AnyEntity, AnyView, AnyWeakView, App, AsyncApp, AsyncWindowContext, Bounds, Context,
40 CursorStyle, Decorations, DragMoveEvent, Entity, EntityId, EventEmitter, FocusHandle,
41 Focusable, Global, HitboxBehavior, Hsla, KeyContext, Keystroke, ManagedView, MouseButton,
42 PathPromptOptions, Point, PromptLevel, Render, ResizeEdge, Size, Stateful, Subscription, Task,
43 Tiling, WeakEntity, WindowBounds, WindowHandle, WindowId, WindowOptions, actions, canvas,
44 point, relative, size, transparent_black,
45};
46pub use history_manager::*;
47pub use item::{
48 FollowableItem, FollowableItemHandle, Item, ItemHandle, ItemSettings, PreviewTabsSettings,
49 ProjectItem, SerializableItem, SerializableItemHandle, WeakItemHandle,
50};
51use itertools::Itertools;
52use language::{
53 Buffer, LanguageRegistry, Rope,
54 language_settings::{AllLanguageSettings, all_language_settings},
55};
56pub use modal_layer::*;
57use node_runtime::NodeRuntime;
58use notifications::{
59 DetachAndPromptErr, Notifications, dismiss_app_notification,
60 simple_message_notification::MessageNotification,
61};
62pub use pane::*;
63pub use pane_group::*;
64use persistence::{
65 DB, SerializedWindowBounds,
66 model::{SerializedSshProject, SerializedWorkspace},
67};
68pub use persistence::{
69 DB as WORKSPACE_DB, WorkspaceDb, delete_unloaded_items,
70 model::{ItemId, LocalPaths, SerializedWorkspaceLocation},
71};
72use postage::stream::Stream;
73use project::{
74 DirectoryLister, Project, ProjectEntryId, ProjectPath, ResolvedPath, Worktree, WorktreeId,
75 debugger::{breakpoint_store::BreakpointStoreEvent, session::ThreadStatus},
76};
77use remote::{SshClientDelegate, SshConnectionOptions, ssh_session::ConnectionIdentifier};
78use schemars::JsonSchema;
79use serde::Deserialize;
80use session::AppSession;
81use settings::{Settings, update_settings_file};
82use shared_screen::SharedScreen;
83use sqlez::{
84 bindable::{Bind, Column, StaticColumnCount},
85 statement::Statement,
86};
87use status_bar::StatusBar;
88pub use status_bar::StatusItemView;
89use std::{
90 any::TypeId,
91 borrow::Cow,
92 cell::RefCell,
93 cmp,
94 collections::{VecDeque, hash_map::DefaultHasher},
95 env,
96 hash::{Hash, Hasher},
97 path::{Path, PathBuf},
98 process::ExitStatus,
99 rc::Rc,
100 sync::{Arc, LazyLock, Weak, atomic::AtomicUsize},
101 time::Duration,
102};
103use task::{DebugScenario, SpawnInTerminal, TaskContext};
104use theme::{ActiveTheme, SystemAppearance, ThemeSettings};
105pub use toolbar::{Toolbar, ToolbarItemEvent, ToolbarItemLocation, ToolbarItemView};
106pub use ui;
107use ui::{Window, prelude::*};
108use util::{ResultExt, TryFutureExt, paths::SanitizedPath, serde::default_true};
109use uuid::Uuid;
110pub use workspace_settings::{
111 AutosaveSetting, BottomDockLayout, RestoreOnStartupBehavior, TabBarSettings, WorkspaceSettings,
112};
113use zed_actions::{Spawn, feedback::FileBugReport};
114
115use crate::notifications::NotificationId;
116use crate::persistence::{
117 SerializedAxis,
118 model::{DockData, DockStructure, SerializedItem, SerializedPane, SerializedPaneGroup},
119};
120
121pub const SERIALIZATION_THROTTLE_TIME: Duration = Duration::from_millis(200);
122
123static ZED_WINDOW_SIZE: LazyLock<Option<Size<Pixels>>> = LazyLock::new(|| {
124 env::var("ZED_WINDOW_SIZE")
125 .ok()
126 .as_deref()
127 .and_then(parse_pixel_size_env_var)
128});
129
130static ZED_WINDOW_POSITION: LazyLock<Option<Point<Pixels>>> = LazyLock::new(|| {
131 env::var("ZED_WINDOW_POSITION")
132 .ok()
133 .as_deref()
134 .and_then(parse_pixel_position_env_var)
135});
136
137pub trait TerminalProvider {
138 fn spawn(
139 &self,
140 task: SpawnInTerminal,
141 window: &mut Window,
142 cx: &mut App,
143 ) -> Task<Option<Result<ExitStatus>>>;
144}
145
146pub trait DebuggerProvider {
147 // `active_buffer` is used to resolve build task's name against language-specific tasks.
148 fn start_session(
149 &self,
150 definition: DebugScenario,
151 task_context: TaskContext,
152 active_buffer: Option<Entity<Buffer>>,
153 worktree_id: Option<WorktreeId>,
154 window: &mut Window,
155 cx: &mut App,
156 );
157
158 fn spawn_task_or_modal(
159 &self,
160 workspace: &mut Workspace,
161 action: &Spawn,
162 window: &mut Window,
163 cx: &mut Context<Workspace>,
164 );
165
166 fn task_scheduled(&self, cx: &mut App);
167 fn debug_scenario_scheduled(&self, cx: &mut App);
168 fn debug_scenario_scheduled_last(&self, cx: &App) -> bool;
169
170 fn active_thread_state(&self, cx: &App) -> Option<ThreadStatus>;
171}
172
173actions!(
174 workspace,
175 [
176 /// Activates the next pane in the workspace.
177 ActivateNextPane,
178 /// Activates the previous pane in the workspace.
179 ActivatePreviousPane,
180 /// Switches to the next window.
181 ActivateNextWindow,
182 /// Switches to the previous window.
183 ActivatePreviousWindow,
184 /// Adds a folder to the current project.
185 AddFolderToProject,
186 /// Clears all notifications.
187 ClearAllNotifications,
188 /// Closes the active dock.
189 CloseActiveDock,
190 /// Closes all docks.
191 CloseAllDocks,
192 /// Closes the current window.
193 CloseWindow,
194 /// Opens the feedback dialog.
195 Feedback,
196 /// Follows the next collaborator in the session.
197 FollowNextCollaborator,
198 /// Moves the focused panel to the next position.
199 MoveFocusedPanelToNextPosition,
200 /// Opens a new terminal in the center.
201 NewCenterTerminal,
202 /// Creates a new file.
203 NewFile,
204 /// Creates a new file in a vertical split.
205 NewFileSplitVertical,
206 /// Creates a new file in a horizontal split.
207 NewFileSplitHorizontal,
208 /// Opens a new search.
209 NewSearch,
210 /// Opens a new terminal.
211 NewTerminal,
212 /// Opens a new window.
213 NewWindow,
214 /// Opens a file or directory.
215 Open,
216 /// Opens multiple files.
217 OpenFiles,
218 /// Opens the current location in terminal.
219 OpenInTerminal,
220 /// Opens the component preview.
221 OpenComponentPreview,
222 /// Reloads the active item.
223 ReloadActiveItem,
224 /// Resets the active dock to its default size.
225 ResetActiveDockSize,
226 /// Resets all open docks to their default sizes.
227 ResetOpenDocksSize,
228 /// Reloads the application
229 Reload,
230 /// Saves the current file with a new name.
231 SaveAs,
232 /// Saves without formatting.
233 SaveWithoutFormat,
234 /// Shuts down all debug adapters.
235 ShutdownDebugAdapters,
236 /// Suppresses the current notification.
237 SuppressNotification,
238 /// Toggles the bottom dock.
239 ToggleBottomDock,
240 /// Toggles centered layout mode.
241 ToggleCenteredLayout,
242 /// Toggles edit prediction feature globally for all files.
243 ToggleEditPrediction,
244 /// Toggles the left dock.
245 ToggleLeftDock,
246 /// Toggles the right dock.
247 ToggleRightDock,
248 /// Toggles zoom on the active pane.
249 ToggleZoom,
250 /// Stops following a collaborator.
251 Unfollow,
252 /// Restores the banner.
253 RestoreBanner,
254 /// Toggles expansion of the selected item.
255 ToggleExpandItem,
256 ]
257);
258
259/// Activates a specific pane by its index.
260#[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)]
261#[action(namespace = workspace)]
262pub struct ActivatePane(pub usize);
263
264/// Moves an item to a specific pane by index.
265#[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)]
266#[action(namespace = workspace)]
267#[serde(deny_unknown_fields)]
268pub struct MoveItemToPane {
269 #[serde(default = "default_1")]
270 pub destination: usize,
271 #[serde(default = "default_true")]
272 pub focus: bool,
273 #[serde(default)]
274 pub clone: bool,
275}
276
277fn default_1() -> usize {
278 1
279}
280
281/// Moves an item to a pane in the specified direction.
282#[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)]
283#[action(namespace = workspace)]
284#[serde(deny_unknown_fields)]
285pub struct MoveItemToPaneInDirection {
286 #[serde(default = "default_right")]
287 pub direction: SplitDirection,
288 #[serde(default = "default_true")]
289 pub focus: bool,
290 #[serde(default)]
291 pub clone: bool,
292}
293
294fn default_right() -> SplitDirection {
295 SplitDirection::Right
296}
297
298/// Saves all open files in the workspace.
299#[derive(Clone, PartialEq, Debug, Deserialize, JsonSchema, Action)]
300#[action(namespace = workspace)]
301#[serde(deny_unknown_fields)]
302pub struct SaveAll {
303 #[serde(default)]
304 pub save_intent: Option<SaveIntent>,
305}
306
307/// Saves the current file with the specified options.
308#[derive(Clone, PartialEq, Debug, Deserialize, JsonSchema, Action)]
309#[action(namespace = workspace)]
310#[serde(deny_unknown_fields)]
311pub struct Save {
312 #[serde(default)]
313 pub save_intent: Option<SaveIntent>,
314}
315
316/// Closes all items and panes in the workspace.
317#[derive(Clone, PartialEq, Debug, Deserialize, Default, JsonSchema, Action)]
318#[action(namespace = workspace)]
319#[serde(deny_unknown_fields)]
320pub struct CloseAllItemsAndPanes {
321 #[serde(default)]
322 pub save_intent: Option<SaveIntent>,
323}
324
325/// Closes all inactive tabs and panes in the workspace.
326#[derive(Clone, PartialEq, Debug, Deserialize, Default, JsonSchema, Action)]
327#[action(namespace = workspace)]
328#[serde(deny_unknown_fields)]
329pub struct CloseInactiveTabsAndPanes {
330 #[serde(default)]
331 pub save_intent: Option<SaveIntent>,
332}
333
334/// Sends a sequence of keystrokes to the active element.
335#[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)]
336#[action(namespace = workspace)]
337pub struct SendKeystrokes(pub String);
338
339actions!(
340 project_symbols,
341 [
342 /// Toggles the project symbols search.
343 #[action(name = "Toggle")]
344 ToggleProjectSymbols
345 ]
346);
347
348/// Toggles the file finder interface.
349#[derive(Default, PartialEq, Eq, Clone, Deserialize, JsonSchema, Action)]
350#[action(namespace = file_finder, name = "Toggle")]
351#[serde(deny_unknown_fields)]
352pub struct ToggleFileFinder {
353 #[serde(default)]
354 pub separate_history: bool,
355}
356
357/// Increases size of a currently focused dock by a given amount of pixels.
358#[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)]
359#[action(namespace = workspace)]
360#[serde(deny_unknown_fields)]
361pub struct IncreaseActiveDockSize {
362 /// For 0px parameter, uses UI font size value.
363 #[serde(default)]
364 pub px: u32,
365}
366
367/// Decreases size of a currently focused dock by a given amount of pixels.
368#[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)]
369#[action(namespace = workspace)]
370#[serde(deny_unknown_fields)]
371pub struct DecreaseActiveDockSize {
372 /// For 0px parameter, uses UI font size value.
373 #[serde(default)]
374 pub px: u32,
375}
376
377/// Increases size of all currently visible docks uniformly, by a given amount of pixels.
378#[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)]
379#[action(namespace = workspace)]
380#[serde(deny_unknown_fields)]
381pub struct IncreaseOpenDocksSize {
382 /// For 0px parameter, uses UI font size value.
383 #[serde(default)]
384 pub px: u32,
385}
386
387/// Decreases size of all currently visible docks uniformly, by a given amount of pixels.
388#[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)]
389#[action(namespace = workspace)]
390#[serde(deny_unknown_fields)]
391pub struct DecreaseOpenDocksSize {
392 /// For 0px parameter, uses UI font size value.
393 #[serde(default)]
394 pub px: u32,
395}
396
397actions!(
398 workspace,
399 [
400 /// Activates the pane to the left.
401 ActivatePaneLeft,
402 /// Activates the pane to the right.
403 ActivatePaneRight,
404 /// Activates the pane above.
405 ActivatePaneUp,
406 /// Activates the pane below.
407 ActivatePaneDown,
408 /// Swaps the current pane with the one to the left.
409 SwapPaneLeft,
410 /// Swaps the current pane with the one to the right.
411 SwapPaneRight,
412 /// Swaps the current pane with the one above.
413 SwapPaneUp,
414 /// Swaps the current pane with the one below.
415 SwapPaneDown,
416 ]
417);
418
419#[derive(PartialEq, Eq, Debug)]
420pub enum CloseIntent {
421 /// Quit the program entirely.
422 Quit,
423 /// Close a window.
424 CloseWindow,
425 /// Replace the workspace in an existing window.
426 ReplaceWindow,
427}
428
429#[derive(Clone)]
430pub struct Toast {
431 id: NotificationId,
432 msg: Cow<'static, str>,
433 autohide: bool,
434 on_click: Option<(Cow<'static, str>, Arc<dyn Fn(&mut Window, &mut App)>)>,
435}
436
437impl Toast {
438 pub fn new<I: Into<Cow<'static, str>>>(id: NotificationId, msg: I) -> Self {
439 Toast {
440 id,
441 msg: msg.into(),
442 on_click: None,
443 autohide: false,
444 }
445 }
446
447 pub fn on_click<F, M>(mut self, message: M, on_click: F) -> Self
448 where
449 M: Into<Cow<'static, str>>,
450 F: Fn(&mut Window, &mut App) + 'static,
451 {
452 self.on_click = Some((message.into(), Arc::new(on_click)));
453 self
454 }
455
456 pub fn autohide(mut self) -> Self {
457 self.autohide = true;
458 self
459 }
460}
461
462impl PartialEq for Toast {
463 fn eq(&self, other: &Self) -> bool {
464 self.id == other.id
465 && self.msg == other.msg
466 && self.on_click.is_some() == other.on_click.is_some()
467 }
468}
469
470/// Opens a new terminal with the specified working directory.
471#[derive(Debug, Default, Clone, Deserialize, PartialEq, JsonSchema, Action)]
472#[action(namespace = workspace)]
473#[serde(deny_unknown_fields)]
474pub struct OpenTerminal {
475 pub working_directory: PathBuf,
476}
477
478#[derive(Clone, Copy, Debug, Default, Hash, PartialEq, Eq, PartialOrd, Ord)]
479pub struct WorkspaceId(i64);
480
481impl StaticColumnCount for WorkspaceId {}
482impl Bind for WorkspaceId {
483 fn bind(&self, statement: &Statement, start_index: i32) -> Result<i32> {
484 self.0.bind(statement, start_index)
485 }
486}
487impl Column for WorkspaceId {
488 fn column(statement: &mut Statement, start_index: i32) -> Result<(Self, i32)> {
489 i64::column(statement, start_index)
490 .map(|(i, next_index)| (Self(i), next_index))
491 .with_context(|| format!("Failed to read WorkspaceId at index {start_index}"))
492 }
493}
494impl From<WorkspaceId> for i64 {
495 fn from(val: WorkspaceId) -> Self {
496 val.0
497 }
498}
499
500pub fn init_settings(cx: &mut App) {
501 WorkspaceSettings::register(cx);
502 ItemSettings::register(cx);
503 PreviewTabsSettings::register(cx);
504 TabBarSettings::register(cx);
505}
506
507fn prompt_and_open_paths(app_state: Arc<AppState>, options: PathPromptOptions, cx: &mut App) {
508 let paths = cx.prompt_for_paths(options);
509 cx.spawn(
510 async move |cx| match paths.await.anyhow().and_then(|res| res) {
511 Ok(Some(paths)) => {
512 cx.update(|cx| {
513 open_paths(&paths, app_state, OpenOptions::default(), cx).detach_and_log_err(cx)
514 })
515 .ok();
516 }
517 Ok(None) => {}
518 Err(err) => {
519 util::log_err(&err);
520 cx.update(|cx| {
521 if let Some(workspace_window) = cx
522 .active_window()
523 .and_then(|window| window.downcast::<Workspace>())
524 {
525 workspace_window
526 .update(cx, |workspace, _, cx| {
527 workspace.show_portal_error(err.to_string(), cx);
528 })
529 .ok();
530 }
531 })
532 .ok();
533 }
534 },
535 )
536 .detach();
537}
538
539pub fn init(app_state: Arc<AppState>, cx: &mut App) {
540 init_settings(cx);
541 component::init();
542 theme_preview::init(cx);
543 toast_layer::init(cx);
544 history_manager::init(cx);
545
546 cx.on_action(|_: &CloseWindow, cx| Workspace::close_global(cx));
547 cx.on_action(|_: &Reload, cx| reload(cx));
548
549 cx.on_action({
550 let app_state = Arc::downgrade(&app_state);
551 move |_: &Open, cx: &mut App| {
552 if let Some(app_state) = app_state.upgrade() {
553 prompt_and_open_paths(
554 app_state,
555 PathPromptOptions {
556 files: true,
557 directories: true,
558 multiple: true,
559 prompt: None,
560 },
561 cx,
562 );
563 }
564 }
565 });
566 cx.on_action({
567 let app_state = Arc::downgrade(&app_state);
568 move |_: &OpenFiles, cx: &mut App| {
569 let directories = cx.can_select_mixed_files_and_dirs();
570 if let Some(app_state) = app_state.upgrade() {
571 prompt_and_open_paths(
572 app_state,
573 PathPromptOptions {
574 files: true,
575 directories,
576 multiple: true,
577 prompt: None,
578 },
579 cx,
580 );
581 }
582 }
583 });
584}
585
586type BuildProjectItemFn =
587 fn(AnyEntity, Entity<Project>, Option<&Pane>, &mut Window, &mut App) -> Box<dyn ItemHandle>;
588
589type BuildProjectItemForPathFn =
590 fn(
591 &Entity<Project>,
592 &ProjectPath,
593 &mut Window,
594 &mut App,
595 ) -> Option<Task<Result<(Option<ProjectEntryId>, WorkspaceItemBuilder)>>>;
596
597#[derive(Clone, Default)]
598struct ProjectItemRegistry {
599 build_project_item_fns_by_type: HashMap<TypeId, BuildProjectItemFn>,
600 build_project_item_for_path_fns: Vec<BuildProjectItemForPathFn>,
601}
602
603impl ProjectItemRegistry {
604 fn register<T: ProjectItem>(&mut self) {
605 self.build_project_item_fns_by_type.insert(
606 TypeId::of::<T::Item>(),
607 |item, project, pane, window, cx| {
608 let item = item.downcast().unwrap();
609 Box::new(cx.new(|cx| T::for_project_item(project, pane, item, window, cx)))
610 as Box<dyn ItemHandle>
611 },
612 );
613 self.build_project_item_for_path_fns
614 .push(|project, project_path, window, cx| {
615 let project_item =
616 <T::Item as project::ProjectItem>::try_open(project, project_path, cx)?;
617 let project = project.clone();
618 Some(window.spawn(cx, async move |cx| {
619 let project_item = project_item.await?;
620 let project_entry_id: Option<ProjectEntryId> =
621 project_item.read_with(cx, project::ProjectItem::entry_id)?;
622 let build_workspace_item = Box::new(
623 |pane: &mut Pane, window: &mut Window, cx: &mut Context<Pane>| {
624 Box::new(cx.new(|cx| {
625 T::for_project_item(project, Some(pane), project_item, window, cx)
626 })) as Box<dyn ItemHandle>
627 },
628 ) as Box<_>;
629 Ok((project_entry_id, build_workspace_item))
630 }))
631 });
632 }
633
634 fn open_path(
635 &self,
636 project: &Entity<Project>,
637 path: &ProjectPath,
638 window: &mut Window,
639 cx: &mut App,
640 ) -> Task<Result<(Option<ProjectEntryId>, WorkspaceItemBuilder)>> {
641 let Some(open_project_item) = self
642 .build_project_item_for_path_fns
643 .iter()
644 .rev()
645 .find_map(|open_project_item| open_project_item(project, path, window, cx))
646 else {
647 return Task::ready(Err(anyhow!("cannot open file {:?}", path.path)));
648 };
649 open_project_item
650 }
651
652 fn build_item<T: project::ProjectItem>(
653 &self,
654 item: Entity<T>,
655 project: Entity<Project>,
656 pane: Option<&Pane>,
657 window: &mut Window,
658 cx: &mut App,
659 ) -> Option<Box<dyn ItemHandle>> {
660 let build = self
661 .build_project_item_fns_by_type
662 .get(&TypeId::of::<T>())?;
663 Some(build(item.into_any(), project, pane, window, cx))
664 }
665}
666
667type WorkspaceItemBuilder =
668 Box<dyn FnOnce(&mut Pane, &mut Window, &mut Context<Pane>) -> Box<dyn ItemHandle>>;
669
670impl Global for ProjectItemRegistry {}
671
672/// Registers a [ProjectItem] for the app. When opening a file, all the registered
673/// items will get a chance to open the file, starting from the project item that
674/// was added last.
675pub fn register_project_item<I: ProjectItem>(cx: &mut App) {
676 cx.default_global::<ProjectItemRegistry>().register::<I>();
677}
678
679#[derive(Default)]
680pub struct FollowableViewRegistry(HashMap<TypeId, FollowableViewDescriptor>);
681
682struct FollowableViewDescriptor {
683 from_state_proto: fn(
684 Entity<Workspace>,
685 ViewId,
686 &mut Option<proto::view::Variant>,
687 &mut Window,
688 &mut App,
689 ) -> Option<Task<Result<Box<dyn FollowableItemHandle>>>>,
690 to_followable_view: fn(&AnyView) -> Box<dyn FollowableItemHandle>,
691}
692
693impl Global for FollowableViewRegistry {}
694
695impl FollowableViewRegistry {
696 pub fn register<I: FollowableItem>(cx: &mut App) {
697 cx.default_global::<Self>().0.insert(
698 TypeId::of::<I>(),
699 FollowableViewDescriptor {
700 from_state_proto: |workspace, id, state, window, cx| {
701 I::from_state_proto(workspace, id, state, window, cx).map(|task| {
702 cx.foreground_executor()
703 .spawn(async move { Ok(Box::new(task.await?) as Box<_>) })
704 })
705 },
706 to_followable_view: |view| Box::new(view.clone().downcast::<I>().unwrap()),
707 },
708 );
709 }
710
711 pub fn from_state_proto(
712 workspace: Entity<Workspace>,
713 view_id: ViewId,
714 mut state: Option<proto::view::Variant>,
715 window: &mut Window,
716 cx: &mut App,
717 ) -> Option<Task<Result<Box<dyn FollowableItemHandle>>>> {
718 cx.update_default_global(|this: &mut Self, cx| {
719 this.0.values().find_map(|descriptor| {
720 (descriptor.from_state_proto)(workspace.clone(), view_id, &mut state, window, cx)
721 })
722 })
723 }
724
725 pub fn to_followable_view(
726 view: impl Into<AnyView>,
727 cx: &App,
728 ) -> Option<Box<dyn FollowableItemHandle>> {
729 let this = cx.try_global::<Self>()?;
730 let view = view.into();
731 let descriptor = this.0.get(&view.entity_type())?;
732 Some((descriptor.to_followable_view)(&view))
733 }
734}
735
736#[derive(Copy, Clone)]
737struct SerializableItemDescriptor {
738 deserialize: fn(
739 Entity<Project>,
740 WeakEntity<Workspace>,
741 WorkspaceId,
742 ItemId,
743 &mut Window,
744 &mut Context<Pane>,
745 ) -> Task<Result<Box<dyn ItemHandle>>>,
746 cleanup: fn(WorkspaceId, Vec<ItemId>, &mut Window, &mut App) -> Task<Result<()>>,
747 view_to_serializable_item: fn(AnyView) -> Box<dyn SerializableItemHandle>,
748}
749
750#[derive(Default)]
751struct SerializableItemRegistry {
752 descriptors_by_kind: HashMap<Arc<str>, SerializableItemDescriptor>,
753 descriptors_by_type: HashMap<TypeId, SerializableItemDescriptor>,
754}
755
756impl Global for SerializableItemRegistry {}
757
758impl SerializableItemRegistry {
759 fn deserialize(
760 item_kind: &str,
761 project: Entity<Project>,
762 workspace: WeakEntity<Workspace>,
763 workspace_id: WorkspaceId,
764 item_item: ItemId,
765 window: &mut Window,
766 cx: &mut Context<Pane>,
767 ) -> Task<Result<Box<dyn ItemHandle>>> {
768 let Some(descriptor) = Self::descriptor(item_kind, cx) else {
769 return Task::ready(Err(anyhow!(
770 "cannot deserialize {}, descriptor not found",
771 item_kind
772 )));
773 };
774
775 (descriptor.deserialize)(project, workspace, workspace_id, item_item, window, cx)
776 }
777
778 fn cleanup(
779 item_kind: &str,
780 workspace_id: WorkspaceId,
781 loaded_items: Vec<ItemId>,
782 window: &mut Window,
783 cx: &mut App,
784 ) -> Task<Result<()>> {
785 let Some(descriptor) = Self::descriptor(item_kind, cx) else {
786 return Task::ready(Err(anyhow!(
787 "cannot cleanup {}, descriptor not found",
788 item_kind
789 )));
790 };
791
792 (descriptor.cleanup)(workspace_id, loaded_items, window, cx)
793 }
794
795 fn view_to_serializable_item_handle(
796 view: AnyView,
797 cx: &App,
798 ) -> Option<Box<dyn SerializableItemHandle>> {
799 let this = cx.try_global::<Self>()?;
800 let descriptor = this.descriptors_by_type.get(&view.entity_type())?;
801 Some((descriptor.view_to_serializable_item)(view))
802 }
803
804 fn descriptor(item_kind: &str, cx: &App) -> Option<SerializableItemDescriptor> {
805 let this = cx.try_global::<Self>()?;
806 this.descriptors_by_kind.get(item_kind).copied()
807 }
808}
809
810pub fn register_serializable_item<I: SerializableItem>(cx: &mut App) {
811 let serialized_item_kind = I::serialized_item_kind();
812
813 let registry = cx.default_global::<SerializableItemRegistry>();
814 let descriptor = SerializableItemDescriptor {
815 deserialize: |project, workspace, workspace_id, item_id, window, cx| {
816 let task = I::deserialize(project, workspace, workspace_id, item_id, window, cx);
817 cx.foreground_executor()
818 .spawn(async { Ok(Box::new(task.await?) as Box<_>) })
819 },
820 cleanup: |workspace_id, loaded_items, window, cx| {
821 I::cleanup(workspace_id, loaded_items, window, cx)
822 },
823 view_to_serializable_item: |view| Box::new(view.downcast::<I>().unwrap()),
824 };
825 registry
826 .descriptors_by_kind
827 .insert(Arc::from(serialized_item_kind), descriptor);
828 registry
829 .descriptors_by_type
830 .insert(TypeId::of::<I>(), descriptor);
831}
832
833pub struct AppState {
834 pub languages: Arc<LanguageRegistry>,
835 pub client: Arc<Client>,
836 pub user_store: Entity<UserStore>,
837 pub workspace_store: Entity<WorkspaceStore>,
838 pub fs: Arc<dyn fs::Fs>,
839 pub build_window_options: fn(Option<Uuid>, &mut App) -> WindowOptions,
840 pub node_runtime: NodeRuntime,
841 pub session: Entity<AppSession>,
842}
843
844struct GlobalAppState(Weak<AppState>);
845
846impl Global for GlobalAppState {}
847
848pub struct WorkspaceStore {
849 workspaces: HashSet<WindowHandle<Workspace>>,
850 client: Arc<Client>,
851 _subscriptions: Vec<client::Subscription>,
852}
853
854#[derive(Copy, Clone, Debug, Hash, Eq, PartialEq, PartialOrd, Ord)]
855pub enum CollaboratorId {
856 PeerId(PeerId),
857 Agent,
858}
859
860impl From<PeerId> for CollaboratorId {
861 fn from(peer_id: PeerId) -> Self {
862 CollaboratorId::PeerId(peer_id)
863 }
864}
865
866impl From<&PeerId> for CollaboratorId {
867 fn from(peer_id: &PeerId) -> Self {
868 CollaboratorId::PeerId(*peer_id)
869 }
870}
871
872#[derive(PartialEq, Eq, PartialOrd, Ord, Debug)]
873struct Follower {
874 project_id: Option<u64>,
875 peer_id: PeerId,
876}
877
878impl AppState {
879 #[track_caller]
880 pub fn global(cx: &App) -> Weak<Self> {
881 cx.global::<GlobalAppState>().0.clone()
882 }
883 pub fn try_global(cx: &App) -> Option<Weak<Self>> {
884 cx.try_global::<GlobalAppState>()
885 .map(|state| state.0.clone())
886 }
887 pub fn set_global(state: Weak<AppState>, cx: &mut App) {
888 cx.set_global(GlobalAppState(state));
889 }
890
891 #[cfg(any(test, feature = "test-support"))]
892 pub fn test(cx: &mut App) -> Arc<Self> {
893 use node_runtime::NodeRuntime;
894 use session::Session;
895 use settings::SettingsStore;
896
897 if !cx.has_global::<SettingsStore>() {
898 let settings_store = SettingsStore::test(cx);
899 cx.set_global(settings_store);
900 }
901
902 let fs = fs::FakeFs::new(cx.background_executor().clone());
903 let languages = Arc::new(LanguageRegistry::test(cx.background_executor().clone()));
904 let clock = Arc::new(clock::FakeSystemClock::new());
905 let http_client = http_client::FakeHttpClient::with_404_response();
906 let client = Client::new(clock, http_client, cx);
907 let session = cx.new(|cx| AppSession::new(Session::test(), cx));
908 let user_store = cx.new(|cx| UserStore::new(client.clone(), cx));
909 let workspace_store = cx.new(|cx| WorkspaceStore::new(client.clone(), cx));
910
911 theme::init(theme::LoadThemes::JustBase, cx);
912 client::init(&client, cx);
913 crate::init_settings(cx);
914
915 Arc::new(Self {
916 client,
917 fs,
918 languages,
919 user_store,
920 workspace_store,
921 node_runtime: NodeRuntime::unavailable(),
922 build_window_options: |_, _| Default::default(),
923 session,
924 })
925 }
926}
927
928struct DelayedDebouncedEditAction {
929 task: Option<Task<()>>,
930 cancel_channel: Option<oneshot::Sender<()>>,
931}
932
933impl DelayedDebouncedEditAction {
934 fn new() -> DelayedDebouncedEditAction {
935 DelayedDebouncedEditAction {
936 task: None,
937 cancel_channel: None,
938 }
939 }
940
941 fn fire_new<F>(
942 &mut self,
943 delay: Duration,
944 window: &mut Window,
945 cx: &mut Context<Workspace>,
946 func: F,
947 ) where
948 F: 'static
949 + Send
950 + FnOnce(&mut Workspace, &mut Window, &mut Context<Workspace>) -> Task<Result<()>>,
951 {
952 if let Some(channel) = self.cancel_channel.take() {
953 _ = channel.send(());
954 }
955
956 let (sender, mut receiver) = oneshot::channel::<()>();
957 self.cancel_channel = Some(sender);
958
959 let previous_task = self.task.take();
960 self.task = Some(cx.spawn_in(window, async move |workspace, cx| {
961 let mut timer = cx.background_executor().timer(delay).fuse();
962 if let Some(previous_task) = previous_task {
963 previous_task.await;
964 }
965
966 futures::select_biased! {
967 _ = receiver => return,
968 _ = timer => {}
969 }
970
971 if let Some(result) = workspace
972 .update_in(cx, |workspace, window, cx| (func)(workspace, window, cx))
973 .log_err()
974 {
975 result.await.log_err();
976 }
977 }));
978 }
979}
980
981pub enum Event {
982 PaneAdded(Entity<Pane>),
983 PaneRemoved,
984 ItemAdded {
985 item: Box<dyn ItemHandle>,
986 },
987 ItemRemoved,
988 ActiveItemChanged,
989 UserSavedItem {
990 pane: WeakEntity<Pane>,
991 item: Box<dyn WeakItemHandle>,
992 save_intent: SaveIntent,
993 },
994 ContactRequestedJoin(u64),
995 WorkspaceCreated(WeakEntity<Workspace>),
996 OpenBundledFile {
997 text: Cow<'static, str>,
998 title: &'static str,
999 language: &'static str,
1000 },
1001 ZoomChanged,
1002 ModalOpened,
1003 ClearActivityIndicator,
1004}
1005
1006#[derive(Debug)]
1007pub enum OpenVisible {
1008 All,
1009 None,
1010 OnlyFiles,
1011 OnlyDirectories,
1012}
1013
1014enum WorkspaceLocation {
1015 // Valid local paths or SSH project to serialize
1016 Location(SerializedWorkspaceLocation),
1017 // No valid location found hence clear session id
1018 DetachFromSession,
1019 // No valid location found to serialize
1020 None,
1021}
1022
1023type PromptForNewPath = Box<
1024 dyn Fn(
1025 &mut Workspace,
1026 DirectoryLister,
1027 &mut Window,
1028 &mut Context<Workspace>,
1029 ) -> oneshot::Receiver<Option<Vec<PathBuf>>>,
1030>;
1031
1032type PromptForOpenPath = Box<
1033 dyn Fn(
1034 &mut Workspace,
1035 DirectoryLister,
1036 &mut Window,
1037 &mut Context<Workspace>,
1038 ) -> oneshot::Receiver<Option<Vec<PathBuf>>>,
1039>;
1040
1041#[derive(Default)]
1042struct DispatchingKeystrokes {
1043 dispatched: HashSet<Vec<Keystroke>>,
1044 queue: VecDeque<Keystroke>,
1045 task: Option<Shared<Task<()>>>,
1046}
1047
1048/// Collects everything project-related for a certain window opened.
1049/// In some way, is a counterpart of a window, as the [`WindowHandle`] could be downcast into `Workspace`.
1050///
1051/// A `Workspace` usually consists of 1 or more projects, a central pane group, 3 docks and a status bar.
1052/// The `Workspace` owns everybody's state and serves as a default, "global context",
1053/// that can be used to register a global action to be triggered from any place in the window.
1054pub struct Workspace {
1055 weak_self: WeakEntity<Self>,
1056 workspace_actions: Vec<Box<dyn Fn(Div, &Workspace, &mut Window, &mut Context<Self>) -> Div>>,
1057 zoomed: Option<AnyWeakView>,
1058 previous_dock_drag_coordinates: Option<Point<Pixels>>,
1059 zoomed_position: Option<DockPosition>,
1060 center: PaneGroup,
1061 left_dock: Entity<Dock>,
1062 bottom_dock: Entity<Dock>,
1063 right_dock: Entity<Dock>,
1064 panes: Vec<Entity<Pane>>,
1065 panes_by_item: HashMap<EntityId, WeakEntity<Pane>>,
1066 active_pane: Entity<Pane>,
1067 last_active_center_pane: Option<WeakEntity<Pane>>,
1068 last_active_view_id: Option<proto::ViewId>,
1069 status_bar: Entity<StatusBar>,
1070 modal_layer: Entity<ModalLayer>,
1071 toast_layer: Entity<ToastLayer>,
1072 titlebar_item: Option<AnyView>,
1073 notifications: Notifications,
1074 suppressed_notifications: HashSet<NotificationId>,
1075 project: Entity<Project>,
1076 follower_states: HashMap<CollaboratorId, FollowerState>,
1077 last_leaders_by_pane: HashMap<WeakEntity<Pane>, CollaboratorId>,
1078 window_edited: bool,
1079 last_window_title: Option<String>,
1080 dirty_items: HashMap<EntityId, Subscription>,
1081 active_call: Option<(Entity<ActiveCall>, Vec<Subscription>)>,
1082 leader_updates_tx: mpsc::UnboundedSender<(PeerId, proto::UpdateFollowers)>,
1083 database_id: Option<WorkspaceId>,
1084 app_state: Arc<AppState>,
1085 dispatching_keystrokes: Rc<RefCell<DispatchingKeystrokes>>,
1086 _subscriptions: Vec<Subscription>,
1087 _apply_leader_updates: Task<Result<()>>,
1088 _observe_current_user: Task<Result<()>>,
1089 _schedule_serialize_workspace: Option<Task<()>>,
1090 _schedule_serialize_ssh_paths: Option<Task<()>>,
1091 pane_history_timestamp: Arc<AtomicUsize>,
1092 bounds: Bounds<Pixels>,
1093 pub centered_layout: bool,
1094 bounds_save_task_queued: Option<Task<()>>,
1095 on_prompt_for_new_path: Option<PromptForNewPath>,
1096 on_prompt_for_open_path: Option<PromptForOpenPath>,
1097 terminal_provider: Option<Box<dyn TerminalProvider>>,
1098 debugger_provider: Option<Arc<dyn DebuggerProvider>>,
1099 serializable_items_tx: UnboundedSender<Box<dyn SerializableItemHandle>>,
1100 serialized_ssh_project: Option<SerializedSshProject>,
1101 _items_serializer: Task<Result<()>>,
1102 session_id: Option<String>,
1103 scheduled_tasks: Vec<Task<()>>,
1104}
1105
1106impl EventEmitter<Event> for Workspace {}
1107
1108#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
1109pub struct ViewId {
1110 pub creator: CollaboratorId,
1111 pub id: u64,
1112}
1113
1114pub struct FollowerState {
1115 center_pane: Entity<Pane>,
1116 dock_pane: Option<Entity<Pane>>,
1117 active_view_id: Option<ViewId>,
1118 items_by_leader_view_id: HashMap<ViewId, FollowerView>,
1119}
1120
1121struct FollowerView {
1122 view: Box<dyn FollowableItemHandle>,
1123 location: Option<proto::PanelId>,
1124}
1125
1126impl Workspace {
1127 const DEFAULT_PADDING: f32 = 0.2;
1128 const MAX_PADDING: f32 = 0.4;
1129
1130 pub fn new(
1131 workspace_id: Option<WorkspaceId>,
1132 project: Entity<Project>,
1133 app_state: Arc<AppState>,
1134 window: &mut Window,
1135 cx: &mut Context<Self>,
1136 ) -> Self {
1137 cx.subscribe_in(&project, window, move |this, _, event, window, cx| {
1138 match event {
1139 project::Event::RemoteIdChanged(_) => {
1140 this.update_window_title(window, cx);
1141 }
1142
1143 project::Event::CollaboratorLeft(peer_id) => {
1144 this.collaborator_left(*peer_id, window, cx);
1145 }
1146
1147 project::Event::WorktreeRemoved(_) | project::Event::WorktreeAdded(_) => {
1148 this.update_window_title(window, cx);
1149 this.update_ssh_paths(cx);
1150 this.serialize_ssh_paths(window, cx);
1151 this.serialize_workspace(window, cx);
1152 // This event could be triggered by `AddFolderToProject` or `RemoveFromProject`.
1153 this.update_history(cx);
1154 }
1155
1156 project::Event::DisconnectedFromHost => {
1157 this.update_window_edited(window, cx);
1158 let leaders_to_unfollow =
1159 this.follower_states.keys().copied().collect::<Vec<_>>();
1160 for leader_id in leaders_to_unfollow {
1161 this.unfollow(leader_id, window, cx);
1162 }
1163 }
1164
1165 project::Event::DisconnectedFromSshRemote => {
1166 this.update_window_edited(window, cx);
1167 }
1168
1169 project::Event::Closed => {
1170 window.remove_window();
1171 }
1172
1173 project::Event::DeletedEntry(_, entry_id) => {
1174 for pane in this.panes.iter() {
1175 pane.update(cx, |pane, cx| {
1176 pane.handle_deleted_project_item(*entry_id, window, cx)
1177 });
1178 }
1179 }
1180
1181 project::Event::Toast {
1182 notification_id,
1183 message,
1184 } => this.show_notification(
1185 NotificationId::named(notification_id.clone()),
1186 cx,
1187 |cx| cx.new(|cx| MessageNotification::new(message.clone(), cx)),
1188 ),
1189
1190 project::Event::HideToast { notification_id } => {
1191 this.dismiss_notification(&NotificationId::named(notification_id.clone()), cx)
1192 }
1193
1194 project::Event::LanguageServerPrompt(request) => {
1195 struct LanguageServerPrompt;
1196
1197 let mut hasher = DefaultHasher::new();
1198 request.lsp_name.as_str().hash(&mut hasher);
1199 let id = hasher.finish();
1200
1201 this.show_notification(
1202 NotificationId::composite::<LanguageServerPrompt>(id as usize),
1203 cx,
1204 |cx| {
1205 cx.new(|cx| {
1206 notifications::LanguageServerPrompt::new(request.clone(), cx)
1207 })
1208 },
1209 );
1210 }
1211
1212 project::Event::AgentLocationChanged => {
1213 this.handle_agent_location_changed(window, cx)
1214 }
1215
1216 _ => {}
1217 }
1218 cx.notify()
1219 })
1220 .detach();
1221
1222 cx.subscribe_in(
1223 &project.read(cx).breakpoint_store(),
1224 window,
1225 |workspace, _, event, window, cx| match event {
1226 BreakpointStoreEvent::BreakpointsUpdated(_, _)
1227 | BreakpointStoreEvent::BreakpointsCleared(_) => {
1228 workspace.serialize_workspace(window, cx);
1229 }
1230 BreakpointStoreEvent::SetDebugLine | BreakpointStoreEvent::ClearDebugLines => {}
1231 },
1232 )
1233 .detach();
1234
1235 cx.on_focus_lost(window, |this, window, cx| {
1236 let focus_handle = this.focus_handle(cx);
1237 window.focus(&focus_handle);
1238 })
1239 .detach();
1240
1241 let weak_handle = cx.entity().downgrade();
1242 let pane_history_timestamp = Arc::new(AtomicUsize::new(0));
1243
1244 let center_pane = cx.new(|cx| {
1245 let mut center_pane = Pane::new(
1246 weak_handle.clone(),
1247 project.clone(),
1248 pane_history_timestamp.clone(),
1249 None,
1250 NewFile.boxed_clone(),
1251 window,
1252 cx,
1253 );
1254 center_pane.set_can_split(Some(Arc::new(|_, _, _, _| true)));
1255 center_pane
1256 });
1257 cx.subscribe_in(¢er_pane, window, Self::handle_pane_event)
1258 .detach();
1259
1260 window.focus(¢er_pane.focus_handle(cx));
1261
1262 cx.emit(Event::PaneAdded(center_pane.clone()));
1263
1264 let window_handle = window.window_handle().downcast::<Workspace>().unwrap();
1265 app_state.workspace_store.update(cx, |store, _| {
1266 store.workspaces.insert(window_handle);
1267 });
1268
1269 let mut current_user = app_state.user_store.read(cx).watch_current_user();
1270 let mut connection_status = app_state.client.status();
1271 let _observe_current_user = cx.spawn_in(window, async move |this, cx| {
1272 current_user.next().await;
1273 connection_status.next().await;
1274 let mut stream =
1275 Stream::map(current_user, drop).merge(Stream::map(connection_status, drop));
1276
1277 while stream.recv().await.is_some() {
1278 this.update(cx, |_, cx| cx.notify())?;
1279 }
1280 anyhow::Ok(())
1281 });
1282
1283 // All leader updates are enqueued and then processed in a single task, so
1284 // that each asynchronous operation can be run in order.
1285 let (leader_updates_tx, mut leader_updates_rx) =
1286 mpsc::unbounded::<(PeerId, proto::UpdateFollowers)>();
1287 let _apply_leader_updates = cx.spawn_in(window, async move |this, cx| {
1288 while let Some((leader_id, update)) = leader_updates_rx.next().await {
1289 Self::process_leader_update(&this, leader_id, update, cx)
1290 .await
1291 .log_err();
1292 }
1293
1294 Ok(())
1295 });
1296
1297 cx.emit(Event::WorkspaceCreated(weak_handle.clone()));
1298 let modal_layer = cx.new(|_| ModalLayer::new());
1299 let toast_layer = cx.new(|_| ToastLayer::new());
1300 cx.subscribe(
1301 &modal_layer,
1302 |_, _, _: &modal_layer::ModalOpenedEvent, cx| {
1303 cx.emit(Event::ModalOpened);
1304 },
1305 )
1306 .detach();
1307
1308 let left_dock = Dock::new(DockPosition::Left, modal_layer.clone(), window, cx);
1309 let bottom_dock = Dock::new(DockPosition::Bottom, modal_layer.clone(), window, cx);
1310 let right_dock = Dock::new(DockPosition::Right, modal_layer.clone(), window, cx);
1311 let left_dock_buttons = cx.new(|cx| PanelButtons::new(left_dock.clone(), cx));
1312 let bottom_dock_buttons = cx.new(|cx| PanelButtons::new(bottom_dock.clone(), cx));
1313 let right_dock_buttons = cx.new(|cx| PanelButtons::new(right_dock.clone(), cx));
1314 let status_bar = cx.new(|cx| {
1315 let mut status_bar = StatusBar::new(¢er_pane.clone(), window, cx);
1316 status_bar.add_left_item(left_dock_buttons, window, cx);
1317 status_bar.add_right_item(right_dock_buttons, window, cx);
1318 status_bar.add_right_item(bottom_dock_buttons, window, cx);
1319 status_bar
1320 });
1321
1322 let session_id = app_state.session.read(cx).id().to_owned();
1323
1324 let mut active_call = None;
1325 if let Some(call) = ActiveCall::try_global(cx) {
1326 let subscriptions = vec![cx.subscribe_in(&call, window, Self::on_active_call_event)];
1327 active_call = Some((call, subscriptions));
1328 }
1329
1330 let (serializable_items_tx, serializable_items_rx) =
1331 mpsc::unbounded::<Box<dyn SerializableItemHandle>>();
1332 let _items_serializer = cx.spawn_in(window, async move |this, cx| {
1333 Self::serialize_items(&this, serializable_items_rx, cx).await
1334 });
1335
1336 let subscriptions = vec![
1337 cx.observe_window_activation(window, Self::on_window_activation_changed),
1338 cx.observe_window_bounds(window, move |this, window, cx| {
1339 if this.bounds_save_task_queued.is_some() {
1340 return;
1341 }
1342 this.bounds_save_task_queued = Some(cx.spawn_in(window, async move |this, cx| {
1343 cx.background_executor()
1344 .timer(Duration::from_millis(100))
1345 .await;
1346 this.update_in(cx, |this, window, cx| {
1347 if let Some(display) = window.display(cx)
1348 && let Ok(display_uuid) = display.uuid()
1349 {
1350 let window_bounds = window.inner_window_bounds();
1351 if let Some(database_id) = workspace_id {
1352 cx.background_executor()
1353 .spawn(DB.set_window_open_status(
1354 database_id,
1355 SerializedWindowBounds(window_bounds),
1356 display_uuid,
1357 ))
1358 .detach_and_log_err(cx);
1359 }
1360 }
1361 this.bounds_save_task_queued.take();
1362 })
1363 .ok();
1364 }));
1365 cx.notify();
1366 }),
1367 cx.observe_window_appearance(window, |_, window, cx| {
1368 let window_appearance = window.appearance();
1369
1370 *SystemAppearance::global_mut(cx) = SystemAppearance(window_appearance.into());
1371
1372 ThemeSettings::reload_current_theme(cx);
1373 ThemeSettings::reload_current_icon_theme(cx);
1374 }),
1375 cx.on_release(move |this, cx| {
1376 this.app_state.workspace_store.update(cx, move |store, _| {
1377 store.workspaces.remove(&window_handle.clone());
1378 })
1379 }),
1380 ];
1381
1382 cx.defer_in(window, |this, window, cx| {
1383 this.update_window_title(window, cx);
1384 this.show_initial_notifications(cx);
1385 });
1386 Workspace {
1387 weak_self: weak_handle.clone(),
1388 zoomed: None,
1389 zoomed_position: None,
1390 previous_dock_drag_coordinates: None,
1391 center: PaneGroup::new(center_pane.clone()),
1392 panes: vec![center_pane.clone()],
1393 panes_by_item: Default::default(),
1394 active_pane: center_pane.clone(),
1395 last_active_center_pane: Some(center_pane.downgrade()),
1396 last_active_view_id: None,
1397 status_bar,
1398 modal_layer,
1399 toast_layer,
1400 titlebar_item: None,
1401 notifications: Notifications::default(),
1402 suppressed_notifications: HashSet::default(),
1403 left_dock,
1404 bottom_dock,
1405 right_dock,
1406 project: project.clone(),
1407 follower_states: Default::default(),
1408 last_leaders_by_pane: Default::default(),
1409 dispatching_keystrokes: Default::default(),
1410 window_edited: false,
1411 last_window_title: None,
1412 dirty_items: Default::default(),
1413 active_call,
1414 database_id: workspace_id,
1415 app_state,
1416 _observe_current_user,
1417 _apply_leader_updates,
1418 _schedule_serialize_workspace: None,
1419 _schedule_serialize_ssh_paths: None,
1420 leader_updates_tx,
1421 _subscriptions: subscriptions,
1422 pane_history_timestamp,
1423 workspace_actions: Default::default(),
1424 // This data will be incorrect, but it will be overwritten by the time it needs to be used.
1425 bounds: Default::default(),
1426 centered_layout: false,
1427 bounds_save_task_queued: None,
1428 on_prompt_for_new_path: None,
1429 on_prompt_for_open_path: None,
1430 terminal_provider: None,
1431 debugger_provider: None,
1432 serializable_items_tx,
1433 _items_serializer,
1434 session_id: Some(session_id),
1435 serialized_ssh_project: None,
1436 scheduled_tasks: Vec::new(),
1437 }
1438 }
1439
1440 pub fn new_local(
1441 abs_paths: Vec<PathBuf>,
1442 app_state: Arc<AppState>,
1443 requesting_window: Option<WindowHandle<Workspace>>,
1444 env: Option<HashMap<String, String>>,
1445 cx: &mut App,
1446 ) -> Task<
1447 anyhow::Result<(
1448 WindowHandle<Workspace>,
1449 Vec<Option<anyhow::Result<Box<dyn ItemHandle>>>>,
1450 )>,
1451 > {
1452 let project_handle = Project::local(
1453 app_state.client.clone(),
1454 app_state.node_runtime.clone(),
1455 app_state.user_store.clone(),
1456 app_state.languages.clone(),
1457 app_state.fs.clone(),
1458 env,
1459 cx,
1460 );
1461
1462 cx.spawn(async move |cx| {
1463 let mut paths_to_open = Vec::with_capacity(abs_paths.len());
1464 for path in abs_paths.into_iter() {
1465 if let Some(canonical) = app_state.fs.canonicalize(&path).await.ok() {
1466 paths_to_open.push(canonical)
1467 } else {
1468 paths_to_open.push(path)
1469 }
1470 }
1471
1472 let serialized_workspace =
1473 persistence::DB.workspace_for_roots(paths_to_open.as_slice());
1474
1475 let workspace_location = serialized_workspace
1476 .as_ref()
1477 .map(|ws| &ws.location)
1478 .and_then(|loc| match loc {
1479 SerializedWorkspaceLocation::Local(_, order) => {
1480 Some((loc.sorted_paths(), order.order()))
1481 }
1482 _ => None,
1483 });
1484
1485 if let Some((paths, order)) = workspace_location {
1486 paths_to_open = paths.iter().cloned().collect();
1487
1488 if order.iter().enumerate().any(|(i, &j)| i != j) {
1489 project_handle
1490 .update(cx, |project, cx| {
1491 project.set_worktrees_reordered(true, cx);
1492 })
1493 .log_err();
1494 }
1495 }
1496
1497 // Get project paths for all of the abs_paths
1498 let mut project_paths: Vec<(PathBuf, Option<ProjectPath>)> =
1499 Vec::with_capacity(paths_to_open.len());
1500
1501 for path in paths_to_open.into_iter() {
1502 if let Some((_, project_entry)) = cx
1503 .update(|cx| {
1504 Workspace::project_path_for_path(project_handle.clone(), &path, true, cx)
1505 })?
1506 .await
1507 .log_err()
1508 {
1509 project_paths.push((path, Some(project_entry)));
1510 } else {
1511 project_paths.push((path, None));
1512 }
1513 }
1514
1515 let workspace_id = if let Some(serialized_workspace) = serialized_workspace.as_ref() {
1516 serialized_workspace.id
1517 } else {
1518 DB.next_id().await.unwrap_or_else(|_| Default::default())
1519 };
1520
1521 let toolchains = DB.toolchains(workspace_id).await?;
1522
1523 for (toolchain, worktree_id, path) in toolchains {
1524 let toolchain_path = PathBuf::from(toolchain.path.clone().to_string());
1525 if !app_state.fs.is_file(toolchain_path.as_path()).await {
1526 continue;
1527 }
1528
1529 project_handle
1530 .update(cx, |this, cx| {
1531 this.activate_toolchain(ProjectPath { worktree_id, path }, toolchain, cx)
1532 })?
1533 .await;
1534 }
1535 let window = if let Some(window) = requesting_window {
1536 let centered_layout = serialized_workspace
1537 .as_ref()
1538 .map(|w| w.centered_layout)
1539 .unwrap_or(false);
1540
1541 cx.update_window(window.into(), |_, window, cx| {
1542 window.replace_root(cx, |window, cx| {
1543 let mut workspace = Workspace::new(
1544 Some(workspace_id),
1545 project_handle.clone(),
1546 app_state.clone(),
1547 window,
1548 cx,
1549 );
1550
1551 workspace.centered_layout = centered_layout;
1552 workspace
1553 });
1554 })?;
1555 window
1556 } else {
1557 let window_bounds_override = window_bounds_env_override();
1558
1559 let (window_bounds, display) = if let Some(bounds) = window_bounds_override {
1560 (Some(WindowBounds::Windowed(bounds)), None)
1561 } else {
1562 let restorable_bounds = serialized_workspace
1563 .as_ref()
1564 .and_then(|workspace| Some((workspace.display?, workspace.window_bounds?)))
1565 .or_else(|| {
1566 let (display, window_bounds) = DB.last_window().log_err()?;
1567 Some((display?, window_bounds?))
1568 });
1569
1570 if let Some((serialized_display, serialized_status)) = restorable_bounds {
1571 (Some(serialized_status.0), Some(serialized_display))
1572 } else {
1573 (None, None)
1574 }
1575 };
1576
1577 // Use the serialized workspace to construct the new window
1578 let mut options = cx.update(|cx| (app_state.build_window_options)(display, cx))?;
1579 options.window_bounds = window_bounds;
1580 let centered_layout = serialized_workspace
1581 .as_ref()
1582 .map(|w| w.centered_layout)
1583 .unwrap_or(false);
1584 cx.open_window(options, {
1585 let app_state = app_state.clone();
1586 let project_handle = project_handle.clone();
1587 move |window, cx| {
1588 cx.new(|cx| {
1589 let mut workspace = Workspace::new(
1590 Some(workspace_id),
1591 project_handle,
1592 app_state,
1593 window,
1594 cx,
1595 );
1596 workspace.centered_layout = centered_layout;
1597 workspace
1598 })
1599 }
1600 })?
1601 };
1602
1603 notify_if_database_failed(window, cx);
1604 let opened_items = window
1605 .update(cx, |_workspace, window, cx| {
1606 open_items(serialized_workspace, project_paths, window, cx)
1607 })?
1608 .await
1609 .unwrap_or_default();
1610
1611 window
1612 .update(cx, |workspace, window, cx| {
1613 window.activate_window();
1614 workspace.update_history(cx);
1615 })
1616 .log_err();
1617 Ok((window, opened_items))
1618 })
1619 }
1620
1621 pub fn weak_handle(&self) -> WeakEntity<Self> {
1622 self.weak_self.clone()
1623 }
1624
1625 pub fn left_dock(&self) -> &Entity<Dock> {
1626 &self.left_dock
1627 }
1628
1629 pub fn bottom_dock(&self) -> &Entity<Dock> {
1630 &self.bottom_dock
1631 }
1632
1633 pub fn set_bottom_dock_layout(
1634 &mut self,
1635 layout: BottomDockLayout,
1636 window: &mut Window,
1637 cx: &mut Context<Self>,
1638 ) {
1639 let fs = self.project().read(cx).fs();
1640 settings::update_settings_file::<WorkspaceSettings>(fs.clone(), cx, move |content, _cx| {
1641 content.bottom_dock_layout = Some(layout);
1642 });
1643
1644 cx.notify();
1645 self.serialize_workspace(window, cx);
1646 }
1647
1648 pub fn right_dock(&self) -> &Entity<Dock> {
1649 &self.right_dock
1650 }
1651
1652 pub fn all_docks(&self) -> [&Entity<Dock>; 3] {
1653 [&self.left_dock, &self.bottom_dock, &self.right_dock]
1654 }
1655
1656 pub fn dock_at_position(&self, position: DockPosition) -> &Entity<Dock> {
1657 match position {
1658 DockPosition::Left => &self.left_dock,
1659 DockPosition::Bottom => &self.bottom_dock,
1660 DockPosition::Right => &self.right_dock,
1661 }
1662 }
1663
1664 pub fn is_edited(&self) -> bool {
1665 self.window_edited
1666 }
1667
1668 pub fn add_panel<T: Panel>(
1669 &mut self,
1670 panel: Entity<T>,
1671 window: &mut Window,
1672 cx: &mut Context<Self>,
1673 ) {
1674 let focus_handle = panel.panel_focus_handle(cx);
1675 cx.on_focus_in(&focus_handle, window, Self::handle_panel_focused)
1676 .detach();
1677
1678 let dock_position = panel.position(window, cx);
1679 let dock = self.dock_at_position(dock_position);
1680
1681 dock.update(cx, |dock, cx| {
1682 dock.add_panel(panel, self.weak_self.clone(), window, cx)
1683 });
1684 }
1685
1686 pub fn status_bar(&self) -> &Entity<StatusBar> {
1687 &self.status_bar
1688 }
1689
1690 pub fn app_state(&self) -> &Arc<AppState> {
1691 &self.app_state
1692 }
1693
1694 pub fn user_store(&self) -> &Entity<UserStore> {
1695 &self.app_state.user_store
1696 }
1697
1698 pub fn project(&self) -> &Entity<Project> {
1699 &self.project
1700 }
1701
1702 pub fn recently_activated_items(&self, cx: &App) -> HashMap<EntityId, usize> {
1703 let mut history: HashMap<EntityId, usize> = HashMap::default();
1704
1705 for pane_handle in &self.panes {
1706 let pane = pane_handle.read(cx);
1707
1708 for entry in pane.activation_history() {
1709 history.insert(
1710 entry.entity_id,
1711 history
1712 .get(&entry.entity_id)
1713 .cloned()
1714 .unwrap_or(0)
1715 .max(entry.timestamp),
1716 );
1717 }
1718 }
1719
1720 history
1721 }
1722
1723 pub fn recent_active_item_by_type<T: 'static>(&self, cx: &App) -> Option<Entity<T>> {
1724 let mut recent_item: Option<Entity<T>> = None;
1725 let mut recent_timestamp = 0;
1726 for pane_handle in &self.panes {
1727 let pane = pane_handle.read(cx);
1728 let item_map: HashMap<EntityId, &Box<dyn ItemHandle>> =
1729 pane.items().map(|item| (item.item_id(), item)).collect();
1730 for entry in pane.activation_history() {
1731 if entry.timestamp > recent_timestamp
1732 && let Some(&item) = item_map.get(&entry.entity_id)
1733 && let Some(typed_item) = item.act_as::<T>(cx)
1734 {
1735 recent_timestamp = entry.timestamp;
1736 recent_item = Some(typed_item);
1737 }
1738 }
1739 }
1740 recent_item
1741 }
1742
1743 pub fn recent_navigation_history_iter(
1744 &self,
1745 cx: &App,
1746 ) -> impl Iterator<Item = (ProjectPath, Option<PathBuf>)> {
1747 let mut abs_paths_opened: HashMap<PathBuf, HashSet<ProjectPath>> = HashMap::default();
1748 let mut history: HashMap<ProjectPath, (Option<PathBuf>, usize)> = HashMap::default();
1749
1750 for pane in &self.panes {
1751 let pane = pane.read(cx);
1752
1753 pane.nav_history()
1754 .for_each_entry(cx, |entry, (project_path, fs_path)| {
1755 if let Some(fs_path) = &fs_path {
1756 abs_paths_opened
1757 .entry(fs_path.clone())
1758 .or_default()
1759 .insert(project_path.clone());
1760 }
1761 let timestamp = entry.timestamp;
1762 match history.entry(project_path) {
1763 hash_map::Entry::Occupied(mut entry) => {
1764 let (_, old_timestamp) = entry.get();
1765 if ×tamp > old_timestamp {
1766 entry.insert((fs_path, timestamp));
1767 }
1768 }
1769 hash_map::Entry::Vacant(entry) => {
1770 entry.insert((fs_path, timestamp));
1771 }
1772 }
1773 });
1774
1775 if let Some(item) = pane.active_item()
1776 && let Some(project_path) = item.project_path(cx)
1777 {
1778 let fs_path = self.project.read(cx).absolute_path(&project_path, cx);
1779
1780 if let Some(fs_path) = &fs_path {
1781 abs_paths_opened
1782 .entry(fs_path.clone())
1783 .or_default()
1784 .insert(project_path.clone());
1785 }
1786
1787 history.insert(project_path, (fs_path, std::usize::MAX));
1788 }
1789 }
1790
1791 history
1792 .into_iter()
1793 .sorted_by_key(|(_, (_, order))| *order)
1794 .map(|(project_path, (fs_path, _))| (project_path, fs_path))
1795 .rev()
1796 .filter(move |(history_path, abs_path)| {
1797 let latest_project_path_opened = abs_path
1798 .as_ref()
1799 .and_then(|abs_path| abs_paths_opened.get(abs_path))
1800 .and_then(|project_paths| {
1801 project_paths
1802 .iter()
1803 .max_by(|b1, b2| b1.worktree_id.cmp(&b2.worktree_id))
1804 });
1805
1806 latest_project_path_opened.is_none_or(|path| path == history_path)
1807 })
1808 }
1809
1810 pub fn recent_navigation_history(
1811 &self,
1812 limit: Option<usize>,
1813 cx: &App,
1814 ) -> Vec<(ProjectPath, Option<PathBuf>)> {
1815 self.recent_navigation_history_iter(cx)
1816 .take(limit.unwrap_or(usize::MAX))
1817 .collect()
1818 }
1819
1820 fn navigate_history(
1821 &mut self,
1822 pane: WeakEntity<Pane>,
1823 mode: NavigationMode,
1824 window: &mut Window,
1825 cx: &mut Context<Workspace>,
1826 ) -> Task<Result<()>> {
1827 let to_load = if let Some(pane) = pane.upgrade() {
1828 pane.update(cx, |pane, cx| {
1829 window.focus(&pane.focus_handle(cx));
1830 loop {
1831 // Retrieve the weak item handle from the history.
1832 let entry = pane.nav_history_mut().pop(mode, cx)?;
1833
1834 // If the item is still present in this pane, then activate it.
1835 if let Some(index) = entry
1836 .item
1837 .upgrade()
1838 .and_then(|v| pane.index_for_item(v.as_ref()))
1839 {
1840 let prev_active_item_index = pane.active_item_index();
1841 pane.nav_history_mut().set_mode(mode);
1842 pane.activate_item(index, true, true, window, cx);
1843 pane.nav_history_mut().set_mode(NavigationMode::Normal);
1844
1845 let mut navigated = prev_active_item_index != pane.active_item_index();
1846 if let Some(data) = entry.data {
1847 navigated |= pane.active_item()?.navigate(data, window, cx);
1848 }
1849
1850 if navigated {
1851 break None;
1852 }
1853 } else {
1854 // If the item is no longer present in this pane, then retrieve its
1855 // path info in order to reopen it.
1856 break pane
1857 .nav_history()
1858 .path_for_item(entry.item.id())
1859 .map(|(project_path, abs_path)| (project_path, abs_path, entry));
1860 }
1861 }
1862 })
1863 } else {
1864 None
1865 };
1866
1867 if let Some((project_path, abs_path, entry)) = to_load {
1868 // If the item was no longer present, then load it again from its previous path, first try the local path
1869 let open_by_project_path = self.load_path(project_path.clone(), window, cx);
1870
1871 cx.spawn_in(window, async move |workspace, cx| {
1872 let open_by_project_path = open_by_project_path.await;
1873 let mut navigated = false;
1874 match open_by_project_path
1875 .with_context(|| format!("Navigating to {project_path:?}"))
1876 {
1877 Ok((project_entry_id, build_item)) => {
1878 let prev_active_item_id = pane.update(cx, |pane, _| {
1879 pane.nav_history_mut().set_mode(mode);
1880 pane.active_item().map(|p| p.item_id())
1881 })?;
1882
1883 pane.update_in(cx, |pane, window, cx| {
1884 let item = pane.open_item(
1885 project_entry_id,
1886 project_path,
1887 true,
1888 entry.is_preview,
1889 true,
1890 None,
1891 window, cx,
1892 build_item,
1893 );
1894 navigated |= Some(item.item_id()) != prev_active_item_id;
1895 pane.nav_history_mut().set_mode(NavigationMode::Normal);
1896 if let Some(data) = entry.data {
1897 navigated |= item.navigate(data, window, cx);
1898 }
1899 })?;
1900 }
1901 Err(open_by_project_path_e) => {
1902 // Fall back to opening by abs path, in case an external file was opened and closed,
1903 // and its worktree is now dropped
1904 if let Some(abs_path) = abs_path {
1905 let prev_active_item_id = pane.update(cx, |pane, _| {
1906 pane.nav_history_mut().set_mode(mode);
1907 pane.active_item().map(|p| p.item_id())
1908 })?;
1909 let open_by_abs_path = workspace.update_in(cx, |workspace, window, cx| {
1910 workspace.open_abs_path(abs_path.clone(), OpenOptions { visible: Some(OpenVisible::None), ..Default::default() }, window, cx)
1911 })?;
1912 match open_by_abs_path
1913 .await
1914 .with_context(|| format!("Navigating to {abs_path:?}"))
1915 {
1916 Ok(item) => {
1917 pane.update_in(cx, |pane, window, cx| {
1918 navigated |= Some(item.item_id()) != prev_active_item_id;
1919 pane.nav_history_mut().set_mode(NavigationMode::Normal);
1920 if let Some(data) = entry.data {
1921 navigated |= item.navigate(data, window, cx);
1922 }
1923 })?;
1924 }
1925 Err(open_by_abs_path_e) => {
1926 log::error!("Failed to navigate history: {open_by_project_path_e:#} and {open_by_abs_path_e:#}");
1927 }
1928 }
1929 }
1930 }
1931 }
1932
1933 if !navigated {
1934 workspace
1935 .update_in(cx, |workspace, window, cx| {
1936 Self::navigate_history(workspace, pane, mode, window, cx)
1937 })?
1938 .await?;
1939 }
1940
1941 Ok(())
1942 })
1943 } else {
1944 Task::ready(Ok(()))
1945 }
1946 }
1947
1948 pub fn go_back(
1949 &mut self,
1950 pane: WeakEntity<Pane>,
1951 window: &mut Window,
1952 cx: &mut Context<Workspace>,
1953 ) -> Task<Result<()>> {
1954 self.navigate_history(pane, NavigationMode::GoingBack, window, cx)
1955 }
1956
1957 pub fn go_forward(
1958 &mut self,
1959 pane: WeakEntity<Pane>,
1960 window: &mut Window,
1961 cx: &mut Context<Workspace>,
1962 ) -> Task<Result<()>> {
1963 self.navigate_history(pane, NavigationMode::GoingForward, window, cx)
1964 }
1965
1966 pub fn reopen_closed_item(
1967 &mut self,
1968 window: &mut Window,
1969 cx: &mut Context<Workspace>,
1970 ) -> Task<Result<()>> {
1971 self.navigate_history(
1972 self.active_pane().downgrade(),
1973 NavigationMode::ReopeningClosedItem,
1974 window,
1975 cx,
1976 )
1977 }
1978
1979 pub fn client(&self) -> &Arc<Client> {
1980 &self.app_state.client
1981 }
1982
1983 pub fn set_titlebar_item(&mut self, item: AnyView, _: &mut Window, cx: &mut Context<Self>) {
1984 self.titlebar_item = Some(item);
1985 cx.notify();
1986 }
1987
1988 pub fn set_prompt_for_new_path(&mut self, prompt: PromptForNewPath) {
1989 self.on_prompt_for_new_path = Some(prompt)
1990 }
1991
1992 pub fn set_prompt_for_open_path(&mut self, prompt: PromptForOpenPath) {
1993 self.on_prompt_for_open_path = Some(prompt)
1994 }
1995
1996 pub fn set_terminal_provider(&mut self, provider: impl TerminalProvider + 'static) {
1997 self.terminal_provider = Some(Box::new(provider));
1998 }
1999
2000 pub fn set_debugger_provider(&mut self, provider: impl DebuggerProvider + 'static) {
2001 self.debugger_provider = Some(Arc::new(provider));
2002 }
2003
2004 pub fn debugger_provider(&self) -> Option<Arc<dyn DebuggerProvider>> {
2005 self.debugger_provider.clone()
2006 }
2007
2008 pub fn serialized_ssh_project(&self) -> Option<SerializedSshProject> {
2009 self.serialized_ssh_project.clone()
2010 }
2011
2012 pub fn set_serialized_ssh_project(&mut self, serialized_ssh_project: SerializedSshProject) {
2013 self.serialized_ssh_project = Some(serialized_ssh_project);
2014 }
2015
2016 pub fn prompt_for_open_path(
2017 &mut self,
2018 path_prompt_options: PathPromptOptions,
2019 lister: DirectoryLister,
2020 window: &mut Window,
2021 cx: &mut Context<Self>,
2022 ) -> oneshot::Receiver<Option<Vec<PathBuf>>> {
2023 if !lister.is_local(cx) || !WorkspaceSettings::get_global(cx).use_system_path_prompts {
2024 let prompt = self.on_prompt_for_open_path.take().unwrap();
2025 let rx = prompt(self, lister, window, cx);
2026 self.on_prompt_for_open_path = Some(prompt);
2027 rx
2028 } else {
2029 let (tx, rx) = oneshot::channel();
2030 let abs_path = cx.prompt_for_paths(path_prompt_options);
2031
2032 cx.spawn_in(window, async move |workspace, cx| {
2033 let Ok(result) = abs_path.await else {
2034 return Ok(());
2035 };
2036
2037 match result {
2038 Ok(result) => {
2039 tx.send(result).ok();
2040 }
2041 Err(err) => {
2042 let rx = workspace.update_in(cx, |workspace, window, cx| {
2043 workspace.show_portal_error(err.to_string(), cx);
2044 let prompt = workspace.on_prompt_for_open_path.take().unwrap();
2045 let rx = prompt(workspace, lister, window, cx);
2046 workspace.on_prompt_for_open_path = Some(prompt);
2047 rx
2048 })?;
2049 if let Ok(path) = rx.await {
2050 tx.send(path).ok();
2051 }
2052 }
2053 };
2054 anyhow::Ok(())
2055 })
2056 .detach();
2057
2058 rx
2059 }
2060 }
2061
2062 pub fn prompt_for_new_path(
2063 &mut self,
2064 lister: DirectoryLister,
2065 suggested_name: Option<String>,
2066 window: &mut Window,
2067 cx: &mut Context<Self>,
2068 ) -> oneshot::Receiver<Option<Vec<PathBuf>>> {
2069 if self.project.read(cx).is_via_collab()
2070 || self.project.read(cx).is_via_ssh()
2071 || !WorkspaceSettings::get_global(cx).use_system_path_prompts
2072 {
2073 let prompt = self.on_prompt_for_new_path.take().unwrap();
2074 let rx = prompt(self, lister, window, cx);
2075 self.on_prompt_for_new_path = Some(prompt);
2076 return rx;
2077 }
2078
2079 let (tx, rx) = oneshot::channel();
2080 cx.spawn_in(window, async move |workspace, cx| {
2081 let abs_path = workspace.update(cx, |workspace, cx| {
2082 let relative_to = workspace
2083 .most_recent_active_path(cx)
2084 .and_then(|p| p.parent().map(|p| p.to_path_buf()))
2085 .or_else(|| {
2086 let project = workspace.project.read(cx);
2087 project.visible_worktrees(cx).find_map(|worktree| {
2088 Some(worktree.read(cx).as_local()?.abs_path().to_path_buf())
2089 })
2090 })
2091 .or_else(std::env::home_dir)
2092 .unwrap_or_else(|| PathBuf::from(""));
2093 cx.prompt_for_new_path(&relative_to, suggested_name.as_deref())
2094 })?;
2095 let abs_path = match abs_path.await? {
2096 Ok(path) => path,
2097 Err(err) => {
2098 let rx = workspace.update_in(cx, |workspace, window, cx| {
2099 workspace.show_portal_error(err.to_string(), cx);
2100
2101 let prompt = workspace.on_prompt_for_new_path.take().unwrap();
2102 let rx = prompt(workspace, lister, window, cx);
2103 workspace.on_prompt_for_new_path = Some(prompt);
2104 rx
2105 })?;
2106 if let Ok(path) = rx.await {
2107 tx.send(path).ok();
2108 }
2109 return anyhow::Ok(());
2110 }
2111 };
2112
2113 tx.send(abs_path.map(|path| vec![path])).ok();
2114 anyhow::Ok(())
2115 })
2116 .detach();
2117
2118 rx
2119 }
2120
2121 pub fn titlebar_item(&self) -> Option<AnyView> {
2122 self.titlebar_item.clone()
2123 }
2124
2125 /// Call the given callback with a workspace whose project is local.
2126 ///
2127 /// If the given workspace has a local project, then it will be passed
2128 /// to the callback. Otherwise, a new empty window will be created.
2129 pub fn with_local_workspace<T, F>(
2130 &mut self,
2131 window: &mut Window,
2132 cx: &mut Context<Self>,
2133 callback: F,
2134 ) -> Task<Result<T>>
2135 where
2136 T: 'static,
2137 F: 'static + FnOnce(&mut Workspace, &mut Window, &mut Context<Workspace>) -> T,
2138 {
2139 if self.project.read(cx).is_local() {
2140 Task::ready(Ok(callback(self, window, cx)))
2141 } else {
2142 let env = self.project.read(cx).cli_environment(cx);
2143 let task = Self::new_local(Vec::new(), self.app_state.clone(), None, env, cx);
2144 cx.spawn_in(window, async move |_vh, cx| {
2145 let (workspace, _) = task.await?;
2146 workspace.update(cx, callback)
2147 })
2148 }
2149 }
2150
2151 pub fn worktrees<'a>(&self, cx: &'a App) -> impl 'a + Iterator<Item = Entity<Worktree>> {
2152 self.project.read(cx).worktrees(cx)
2153 }
2154
2155 pub fn visible_worktrees<'a>(
2156 &self,
2157 cx: &'a App,
2158 ) -> impl 'a + Iterator<Item = Entity<Worktree>> {
2159 self.project.read(cx).visible_worktrees(cx)
2160 }
2161
2162 #[cfg(any(test, feature = "test-support"))]
2163 pub fn worktree_scans_complete(&self, cx: &App) -> impl Future<Output = ()> + 'static + use<> {
2164 let futures = self
2165 .worktrees(cx)
2166 .filter_map(|worktree| worktree.read(cx).as_local())
2167 .map(|worktree| worktree.scan_complete())
2168 .collect::<Vec<_>>();
2169 async move {
2170 for future in futures {
2171 future.await;
2172 }
2173 }
2174 }
2175
2176 pub fn close_global(cx: &mut App) {
2177 cx.defer(|cx| {
2178 cx.windows().iter().find(|window| {
2179 window
2180 .update(cx, |_, window, _| {
2181 if window.is_window_active() {
2182 //This can only get called when the window's project connection has been lost
2183 //so we don't need to prompt the user for anything and instead just close the window
2184 window.remove_window();
2185 true
2186 } else {
2187 false
2188 }
2189 })
2190 .unwrap_or(false)
2191 });
2192 });
2193 }
2194
2195 pub fn close_window(&mut self, _: &CloseWindow, window: &mut Window, cx: &mut Context<Self>) {
2196 let prepare = self.prepare_to_close(CloseIntent::CloseWindow, window, cx);
2197 cx.spawn_in(window, async move |_, cx| {
2198 if prepare.await? {
2199 cx.update(|window, _cx| window.remove_window())?;
2200 }
2201 anyhow::Ok(())
2202 })
2203 .detach_and_log_err(cx)
2204 }
2205
2206 pub fn move_focused_panel_to_next_position(
2207 &mut self,
2208 _: &MoveFocusedPanelToNextPosition,
2209 window: &mut Window,
2210 cx: &mut Context<Self>,
2211 ) {
2212 let docks = self.all_docks();
2213 let active_dock = docks
2214 .into_iter()
2215 .find(|dock| dock.focus_handle(cx).contains_focused(window, cx));
2216
2217 if let Some(dock) = active_dock {
2218 dock.update(cx, |dock, cx| {
2219 let active_panel = dock
2220 .active_panel()
2221 .filter(|panel| panel.panel_focus_handle(cx).contains_focused(window, cx));
2222
2223 if let Some(panel) = active_panel {
2224 panel.move_to_next_position(window, cx);
2225 }
2226 })
2227 }
2228 }
2229
2230 pub fn prepare_to_close(
2231 &mut self,
2232 close_intent: CloseIntent,
2233 window: &mut Window,
2234 cx: &mut Context<Self>,
2235 ) -> Task<Result<bool>> {
2236 let active_call = self.active_call().cloned();
2237
2238 // On Linux and Windows, closing the last window should restore the last workspace.
2239 let save_last_workspace = cfg!(not(target_os = "macos"))
2240 && close_intent != CloseIntent::ReplaceWindow
2241 && cx.windows().len() == 1;
2242
2243 cx.spawn_in(window, async move |this, cx| {
2244 let workspace_count = cx.update(|_window, cx| {
2245 cx.windows()
2246 .iter()
2247 .filter(|window| window.downcast::<Workspace>().is_some())
2248 .count()
2249 })?;
2250
2251 if let Some(active_call) = active_call
2252 && close_intent != CloseIntent::Quit
2253 && workspace_count == 1
2254 && active_call.read_with(cx, |call, _| call.room().is_some())?
2255 {
2256 let answer = cx.update(|window, cx| {
2257 window.prompt(
2258 PromptLevel::Warning,
2259 "Do you want to leave the current call?",
2260 None,
2261 &["Close window and hang up", "Cancel"],
2262 cx,
2263 )
2264 })?;
2265
2266 if answer.await.log_err() == Some(1) {
2267 return anyhow::Ok(false);
2268 } else {
2269 active_call
2270 .update(cx, |call, cx| call.hang_up(cx))?
2271 .await
2272 .log_err();
2273 }
2274 }
2275
2276 let save_result = this
2277 .update_in(cx, |this, window, cx| {
2278 this.save_all_internal(SaveIntent::Close, window, cx)
2279 })?
2280 .await;
2281
2282 // If we're not quitting, but closing, we remove the workspace from
2283 // the current session.
2284 if close_intent != CloseIntent::Quit
2285 && !save_last_workspace
2286 && save_result.as_ref().is_ok_and(|&res| res)
2287 {
2288 this.update_in(cx, |this, window, cx| this.remove_from_session(window, cx))?
2289 .await;
2290 }
2291
2292 save_result
2293 })
2294 }
2295
2296 fn save_all(&mut self, action: &SaveAll, window: &mut Window, cx: &mut Context<Self>) {
2297 self.save_all_internal(
2298 action.save_intent.unwrap_or(SaveIntent::SaveAll),
2299 window,
2300 cx,
2301 )
2302 .detach_and_log_err(cx);
2303 }
2304
2305 fn send_keystrokes(
2306 &mut self,
2307 action: &SendKeystrokes,
2308 window: &mut Window,
2309 cx: &mut Context<Self>,
2310 ) {
2311 let keystrokes: Vec<Keystroke> = action
2312 .0
2313 .split(' ')
2314 .flat_map(|k| Keystroke::parse(k).log_err())
2315 .collect();
2316 let _ = self.send_keystrokes_impl(keystrokes, window, cx);
2317 }
2318
2319 pub fn send_keystrokes_impl(
2320 &mut self,
2321 keystrokes: Vec<Keystroke>,
2322 window: &mut Window,
2323 cx: &mut Context<Self>,
2324 ) -> Shared<Task<()>> {
2325 let mut state = self.dispatching_keystrokes.borrow_mut();
2326 if !state.dispatched.insert(keystrokes.clone()) {
2327 cx.propagate();
2328 return state.task.clone().unwrap();
2329 }
2330
2331 state.queue.extend(keystrokes);
2332
2333 let keystrokes = self.dispatching_keystrokes.clone();
2334 if state.task.is_none() {
2335 state.task = Some(
2336 window
2337 .spawn(cx, async move |cx| {
2338 // limit to 100 keystrokes to avoid infinite recursion.
2339 for _ in 0..100 {
2340 let mut state = keystrokes.borrow_mut();
2341 let Some(keystroke) = state.queue.pop_front() else {
2342 state.dispatched.clear();
2343 state.task.take();
2344 return;
2345 };
2346 drop(state);
2347 cx.update(|window, cx| {
2348 let focused = window.focused(cx);
2349 window.dispatch_keystroke(keystroke.clone(), cx);
2350 if window.focused(cx) != focused {
2351 // dispatch_keystroke may cause the focus to change.
2352 // draw's side effect is to schedule the FocusChanged events in the current flush effect cycle
2353 // And we need that to happen before the next keystroke to keep vim mode happy...
2354 // (Note that the tests always do this implicitly, so you must manually test with something like:
2355 // "bindings": { "g z": ["workspace::SendKeystrokes", ": j <enter> u"]}
2356 // )
2357 window.draw(cx).clear();
2358 }
2359 })
2360 .ok();
2361 }
2362
2363 *keystrokes.borrow_mut() = Default::default();
2364 log::error!("over 100 keystrokes passed to send_keystrokes");
2365 })
2366 .shared(),
2367 );
2368 }
2369 state.task.clone().unwrap()
2370 }
2371
2372 fn save_all_internal(
2373 &mut self,
2374 mut save_intent: SaveIntent,
2375 window: &mut Window,
2376 cx: &mut Context<Self>,
2377 ) -> Task<Result<bool>> {
2378 if self.project.read(cx).is_disconnected(cx) {
2379 return Task::ready(Ok(true));
2380 }
2381 let dirty_items = self
2382 .panes
2383 .iter()
2384 .flat_map(|pane| {
2385 pane.read(cx).items().filter_map(|item| {
2386 if item.is_dirty(cx) {
2387 item.tab_content_text(0, cx);
2388 Some((pane.downgrade(), item.boxed_clone()))
2389 } else {
2390 None
2391 }
2392 })
2393 })
2394 .collect::<Vec<_>>();
2395
2396 let project = self.project.clone();
2397 cx.spawn_in(window, async move |workspace, cx| {
2398 let dirty_items = if save_intent == SaveIntent::Close && !dirty_items.is_empty() {
2399 let (serialize_tasks, remaining_dirty_items) =
2400 workspace.update_in(cx, |workspace, window, cx| {
2401 let mut remaining_dirty_items = Vec::new();
2402 let mut serialize_tasks = Vec::new();
2403 for (pane, item) in dirty_items {
2404 if let Some(task) = item
2405 .to_serializable_item_handle(cx)
2406 .and_then(|handle| handle.serialize(workspace, true, window, cx))
2407 {
2408 serialize_tasks.push(task);
2409 } else {
2410 remaining_dirty_items.push((pane, item));
2411 }
2412 }
2413 (serialize_tasks, remaining_dirty_items)
2414 })?;
2415
2416 futures::future::try_join_all(serialize_tasks).await?;
2417
2418 if remaining_dirty_items.len() > 1 {
2419 let answer = workspace.update_in(cx, |_, window, cx| {
2420 let detail = Pane::file_names_for_prompt(
2421 &mut remaining_dirty_items.iter().map(|(_, handle)| handle),
2422 cx,
2423 );
2424 window.prompt(
2425 PromptLevel::Warning,
2426 "Do you want to save all changes in the following files?",
2427 Some(&detail),
2428 &["Save all", "Discard all", "Cancel"],
2429 cx,
2430 )
2431 })?;
2432 match answer.await.log_err() {
2433 Some(0) => save_intent = SaveIntent::SaveAll,
2434 Some(1) => save_intent = SaveIntent::Skip,
2435 Some(2) => return Ok(false),
2436 _ => {}
2437 }
2438 }
2439
2440 remaining_dirty_items
2441 } else {
2442 dirty_items
2443 };
2444
2445 for (pane, item) in dirty_items {
2446 let (singleton, project_entry_ids) =
2447 cx.update(|_, cx| (item.is_singleton(cx), item.project_entry_ids(cx)))?;
2448 if (singleton || !project_entry_ids.is_empty())
2449 && !Pane::save_item(project.clone(), &pane, &*item, save_intent, cx).await?
2450 {
2451 return Ok(false);
2452 }
2453 }
2454 Ok(true)
2455 })
2456 }
2457
2458 pub fn open_workspace_for_paths(
2459 &mut self,
2460 replace_current_window: bool,
2461 paths: Vec<PathBuf>,
2462 window: &mut Window,
2463 cx: &mut Context<Self>,
2464 ) -> Task<Result<()>> {
2465 let window_handle = window.window_handle().downcast::<Self>();
2466 let is_remote = self.project.read(cx).is_via_collab();
2467 let has_worktree = self.project.read(cx).worktrees(cx).next().is_some();
2468 let has_dirty_items = self.items(cx).any(|item| item.is_dirty(cx));
2469
2470 let window_to_replace = if replace_current_window {
2471 window_handle
2472 } else if is_remote || has_worktree || has_dirty_items {
2473 None
2474 } else {
2475 window_handle
2476 };
2477 let app_state = self.app_state.clone();
2478
2479 cx.spawn(async move |_, cx| {
2480 cx.update(|cx| {
2481 open_paths(
2482 &paths,
2483 app_state,
2484 OpenOptions {
2485 replace_window: window_to_replace,
2486 ..Default::default()
2487 },
2488 cx,
2489 )
2490 })?
2491 .await?;
2492 Ok(())
2493 })
2494 }
2495
2496 #[allow(clippy::type_complexity)]
2497 pub fn open_paths(
2498 &mut self,
2499 mut abs_paths: Vec<PathBuf>,
2500 options: OpenOptions,
2501 pane: Option<WeakEntity<Pane>>,
2502 window: &mut Window,
2503 cx: &mut Context<Self>,
2504 ) -> Task<Vec<Option<anyhow::Result<Box<dyn ItemHandle>>>>> {
2505 let fs = self.app_state.fs.clone();
2506
2507 // Sort the paths to ensure we add worktrees for parents before their children.
2508 abs_paths.sort_unstable();
2509 cx.spawn_in(window, async move |this, cx| {
2510 let mut tasks = Vec::with_capacity(abs_paths.len());
2511
2512 for abs_path in &abs_paths {
2513 let visible = match options.visible.as_ref().unwrap_or(&OpenVisible::None) {
2514 OpenVisible::All => Some(true),
2515 OpenVisible::None => Some(false),
2516 OpenVisible::OnlyFiles => match fs.metadata(abs_path).await.log_err() {
2517 Some(Some(metadata)) => Some(!metadata.is_dir),
2518 Some(None) => Some(true),
2519 None => None,
2520 },
2521 OpenVisible::OnlyDirectories => match fs.metadata(abs_path).await.log_err() {
2522 Some(Some(metadata)) => Some(metadata.is_dir),
2523 Some(None) => Some(false),
2524 None => None,
2525 },
2526 };
2527 let project_path = match visible {
2528 Some(visible) => match this
2529 .update(cx, |this, cx| {
2530 Workspace::project_path_for_path(
2531 this.project.clone(),
2532 abs_path,
2533 visible,
2534 cx,
2535 )
2536 })
2537 .log_err()
2538 {
2539 Some(project_path) => project_path.await.log_err(),
2540 None => None,
2541 },
2542 None => None,
2543 };
2544
2545 let this = this.clone();
2546 let abs_path: Arc<Path> = SanitizedPath::from(abs_path.clone()).into();
2547 let fs = fs.clone();
2548 let pane = pane.clone();
2549 let task = cx.spawn(async move |cx| {
2550 let (worktree, project_path) = project_path?;
2551 if fs.is_dir(&abs_path).await {
2552 this.update(cx, |workspace, cx| {
2553 let worktree = worktree.read(cx);
2554 let worktree_abs_path = worktree.abs_path();
2555 let entry_id = if abs_path.as_ref() == worktree_abs_path.as_ref() {
2556 worktree.root_entry()
2557 } else {
2558 abs_path
2559 .strip_prefix(worktree_abs_path.as_ref())
2560 .ok()
2561 .and_then(|relative_path| {
2562 worktree.entry_for_path(relative_path)
2563 })
2564 }
2565 .map(|entry| entry.id);
2566 if let Some(entry_id) = entry_id {
2567 workspace.project.update(cx, |_, cx| {
2568 cx.emit(project::Event::ActiveEntryChanged(Some(entry_id)));
2569 })
2570 }
2571 })
2572 .ok()?;
2573 None
2574 } else {
2575 Some(
2576 this.update_in(cx, |this, window, cx| {
2577 this.open_path(
2578 project_path,
2579 pane,
2580 options.focus.unwrap_or(true),
2581 window,
2582 cx,
2583 )
2584 })
2585 .ok()?
2586 .await,
2587 )
2588 }
2589 });
2590 tasks.push(task);
2591 }
2592
2593 futures::future::join_all(tasks).await
2594 })
2595 }
2596
2597 pub fn open_resolved_path(
2598 &mut self,
2599 path: ResolvedPath,
2600 window: &mut Window,
2601 cx: &mut Context<Self>,
2602 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
2603 match path {
2604 ResolvedPath::ProjectPath { project_path, .. } => {
2605 self.open_path(project_path, None, true, window, cx)
2606 }
2607 ResolvedPath::AbsPath { path, .. } => self.open_abs_path(
2608 path,
2609 OpenOptions {
2610 visible: Some(OpenVisible::None),
2611 ..Default::default()
2612 },
2613 window,
2614 cx,
2615 ),
2616 }
2617 }
2618
2619 pub fn absolute_path_of_worktree(
2620 &self,
2621 worktree_id: WorktreeId,
2622 cx: &mut Context<Self>,
2623 ) -> Option<PathBuf> {
2624 self.project
2625 .read(cx)
2626 .worktree_for_id(worktree_id, cx)
2627 // TODO: use `abs_path` or `root_dir`
2628 .map(|wt| wt.read(cx).abs_path().as_ref().to_path_buf())
2629 }
2630
2631 fn add_folder_to_project(
2632 &mut self,
2633 _: &AddFolderToProject,
2634 window: &mut Window,
2635 cx: &mut Context<Self>,
2636 ) {
2637 let project = self.project.read(cx);
2638 if project.is_via_collab() {
2639 self.show_error(
2640 &anyhow!("You cannot add folders to someone else's project"),
2641 cx,
2642 );
2643 return;
2644 }
2645 let paths = self.prompt_for_open_path(
2646 PathPromptOptions {
2647 files: false,
2648 directories: true,
2649 multiple: true,
2650 prompt: None,
2651 },
2652 DirectoryLister::Project(self.project.clone()),
2653 window,
2654 cx,
2655 );
2656 cx.spawn_in(window, async move |this, cx| {
2657 if let Some(paths) = paths.await.log_err().flatten() {
2658 let results = this
2659 .update_in(cx, |this, window, cx| {
2660 this.open_paths(
2661 paths,
2662 OpenOptions {
2663 visible: Some(OpenVisible::All),
2664 ..Default::default()
2665 },
2666 None,
2667 window,
2668 cx,
2669 )
2670 })?
2671 .await;
2672 for result in results.into_iter().flatten() {
2673 result.log_err();
2674 }
2675 }
2676 anyhow::Ok(())
2677 })
2678 .detach_and_log_err(cx);
2679 }
2680
2681 pub fn project_path_for_path(
2682 project: Entity<Project>,
2683 abs_path: &Path,
2684 visible: bool,
2685 cx: &mut App,
2686 ) -> Task<Result<(Entity<Worktree>, ProjectPath)>> {
2687 let entry = project.update(cx, |project, cx| {
2688 project.find_or_create_worktree(abs_path, visible, cx)
2689 });
2690 cx.spawn(async move |cx| {
2691 let (worktree, path) = entry.await?;
2692 let worktree_id = worktree.read_with(cx, |t, _| t.id())?;
2693 Ok((
2694 worktree,
2695 ProjectPath {
2696 worktree_id,
2697 path: path.into(),
2698 },
2699 ))
2700 })
2701 }
2702
2703 pub fn items<'a>(&'a self, cx: &'a App) -> impl 'a + Iterator<Item = &'a Box<dyn ItemHandle>> {
2704 self.panes.iter().flat_map(|pane| pane.read(cx).items())
2705 }
2706
2707 pub fn item_of_type<T: Item>(&self, cx: &App) -> Option<Entity<T>> {
2708 self.items_of_type(cx).max_by_key(|item| item.item_id())
2709 }
2710
2711 pub fn items_of_type<'a, T: Item>(
2712 &'a self,
2713 cx: &'a App,
2714 ) -> impl 'a + Iterator<Item = Entity<T>> {
2715 self.panes
2716 .iter()
2717 .flat_map(|pane| pane.read(cx).items_of_type())
2718 }
2719
2720 pub fn active_item(&self, cx: &App) -> Option<Box<dyn ItemHandle>> {
2721 self.active_pane().read(cx).active_item()
2722 }
2723
2724 pub fn active_item_as<I: 'static>(&self, cx: &App) -> Option<Entity<I>> {
2725 let item = self.active_item(cx)?;
2726 item.to_any().downcast::<I>().ok()
2727 }
2728
2729 fn active_project_path(&self, cx: &App) -> Option<ProjectPath> {
2730 self.active_item(cx).and_then(|item| item.project_path(cx))
2731 }
2732
2733 pub fn most_recent_active_path(&self, cx: &App) -> Option<PathBuf> {
2734 self.recent_navigation_history_iter(cx)
2735 .filter_map(|(path, abs_path)| {
2736 let worktree = self
2737 .project
2738 .read(cx)
2739 .worktree_for_id(path.worktree_id, cx)?;
2740 if worktree.read(cx).is_visible() {
2741 abs_path
2742 } else {
2743 None
2744 }
2745 })
2746 .next()
2747 }
2748
2749 pub fn save_active_item(
2750 &mut self,
2751 save_intent: SaveIntent,
2752 window: &mut Window,
2753 cx: &mut App,
2754 ) -> Task<Result<()>> {
2755 let project = self.project.clone();
2756 let pane = self.active_pane();
2757 let item = pane.read(cx).active_item();
2758 let pane = pane.downgrade();
2759
2760 window.spawn(cx, async move |cx| {
2761 if let Some(item) = item {
2762 Pane::save_item(project, &pane, item.as_ref(), save_intent, cx)
2763 .await
2764 .map(|_| ())
2765 } else {
2766 Ok(())
2767 }
2768 })
2769 }
2770
2771 pub fn close_inactive_items_and_panes(
2772 &mut self,
2773 action: &CloseInactiveTabsAndPanes,
2774 window: &mut Window,
2775 cx: &mut Context<Self>,
2776 ) {
2777 if let Some(task) = self.close_all_internal(
2778 true,
2779 action.save_intent.unwrap_or(SaveIntent::Close),
2780 window,
2781 cx,
2782 ) {
2783 task.detach_and_log_err(cx)
2784 }
2785 }
2786
2787 pub fn close_all_items_and_panes(
2788 &mut self,
2789 action: &CloseAllItemsAndPanes,
2790 window: &mut Window,
2791 cx: &mut Context<Self>,
2792 ) {
2793 if let Some(task) = self.close_all_internal(
2794 false,
2795 action.save_intent.unwrap_or(SaveIntent::Close),
2796 window,
2797 cx,
2798 ) {
2799 task.detach_and_log_err(cx)
2800 }
2801 }
2802
2803 fn close_all_internal(
2804 &mut self,
2805 retain_active_pane: bool,
2806 save_intent: SaveIntent,
2807 window: &mut Window,
2808 cx: &mut Context<Self>,
2809 ) -> Option<Task<Result<()>>> {
2810 let current_pane = self.active_pane();
2811
2812 let mut tasks = Vec::new();
2813
2814 if retain_active_pane {
2815 let current_pane_close = current_pane.update(cx, |pane, cx| {
2816 pane.close_other_items(
2817 &CloseOtherItems {
2818 save_intent: None,
2819 close_pinned: false,
2820 },
2821 None,
2822 window,
2823 cx,
2824 )
2825 });
2826
2827 tasks.push(current_pane_close);
2828 }
2829
2830 for pane in self.panes() {
2831 if retain_active_pane && pane.entity_id() == current_pane.entity_id() {
2832 continue;
2833 }
2834
2835 let close_pane_items = pane.update(cx, |pane: &mut Pane, cx| {
2836 pane.close_all_items(
2837 &CloseAllItems {
2838 save_intent: Some(save_intent),
2839 close_pinned: false,
2840 },
2841 window,
2842 cx,
2843 )
2844 });
2845
2846 tasks.push(close_pane_items)
2847 }
2848
2849 if tasks.is_empty() {
2850 None
2851 } else {
2852 Some(cx.spawn_in(window, async move |_, _| {
2853 for task in tasks {
2854 task.await?
2855 }
2856 Ok(())
2857 }))
2858 }
2859 }
2860
2861 pub fn is_dock_at_position_open(&self, position: DockPosition, cx: &mut Context<Self>) -> bool {
2862 self.dock_at_position(position).read(cx).is_open()
2863 }
2864
2865 pub fn toggle_dock(
2866 &mut self,
2867 dock_side: DockPosition,
2868 window: &mut Window,
2869 cx: &mut Context<Self>,
2870 ) {
2871 let dock = self.dock_at_position(dock_side);
2872 let mut focus_center = false;
2873 let mut reveal_dock = false;
2874 dock.update(cx, |dock, cx| {
2875 let other_is_zoomed = self.zoomed.is_some() && self.zoomed_position != Some(dock_side);
2876 let was_visible = dock.is_open() && !other_is_zoomed;
2877 dock.set_open(!was_visible, window, cx);
2878
2879 if dock.active_panel().is_none() {
2880 let Some(panel_ix) = dock
2881 .first_enabled_panel_idx(cx)
2882 .log_with_level(log::Level::Info)
2883 else {
2884 return;
2885 };
2886 dock.activate_panel(panel_ix, window, cx);
2887 }
2888
2889 if let Some(active_panel) = dock.active_panel() {
2890 if was_visible {
2891 if active_panel
2892 .panel_focus_handle(cx)
2893 .contains_focused(window, cx)
2894 {
2895 focus_center = true;
2896 }
2897 } else {
2898 let focus_handle = &active_panel.panel_focus_handle(cx);
2899 window.focus(focus_handle);
2900 reveal_dock = true;
2901 }
2902 }
2903 });
2904
2905 if reveal_dock {
2906 self.dismiss_zoomed_items_to_reveal(Some(dock_side), window, cx);
2907 }
2908
2909 if focus_center {
2910 self.active_pane
2911 .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx)))
2912 }
2913
2914 cx.notify();
2915 self.serialize_workspace(window, cx);
2916 }
2917
2918 fn active_dock(&self, window: &Window, cx: &Context<Self>) -> Option<&Entity<Dock>> {
2919 self.all_docks().into_iter().find(|&dock| {
2920 dock.read(cx).is_open() && dock.focus_handle(cx).contains_focused(window, cx)
2921 })
2922 }
2923
2924 fn close_active_dock(&mut self, window: &mut Window, cx: &mut Context<Self>) -> bool {
2925 if let Some(dock) = self.active_dock(window, cx) {
2926 dock.update(cx, |dock, cx| {
2927 dock.set_open(false, window, cx);
2928 });
2929 return true;
2930 }
2931 false
2932 }
2933
2934 pub fn close_all_docks(&mut self, window: &mut Window, cx: &mut Context<Self>) {
2935 for dock in self.all_docks() {
2936 dock.update(cx, |dock, cx| {
2937 dock.set_open(false, window, cx);
2938 });
2939 }
2940
2941 cx.focus_self(window);
2942 cx.notify();
2943 self.serialize_workspace(window, cx);
2944 }
2945
2946 /// Transfer focus to the panel of the given type.
2947 pub fn focus_panel<T: Panel>(
2948 &mut self,
2949 window: &mut Window,
2950 cx: &mut Context<Self>,
2951 ) -> Option<Entity<T>> {
2952 let panel = self.focus_or_unfocus_panel::<T>(window, cx, |_, _, _| true)?;
2953 panel.to_any().downcast().ok()
2954 }
2955
2956 /// Focus the panel of the given type if it isn't already focused. If it is
2957 /// already focused, then transfer focus back to the workspace center.
2958 pub fn toggle_panel_focus<T: Panel>(
2959 &mut self,
2960 window: &mut Window,
2961 cx: &mut Context<Self>,
2962 ) -> bool {
2963 let mut did_focus_panel = false;
2964 self.focus_or_unfocus_panel::<T>(window, cx, |panel, window, cx| {
2965 did_focus_panel = !panel.panel_focus_handle(cx).contains_focused(window, cx);
2966 did_focus_panel
2967 });
2968 did_focus_panel
2969 }
2970
2971 pub fn activate_panel_for_proto_id(
2972 &mut self,
2973 panel_id: PanelId,
2974 window: &mut Window,
2975 cx: &mut Context<Self>,
2976 ) -> Option<Arc<dyn PanelHandle>> {
2977 let mut panel = None;
2978 for dock in self.all_docks() {
2979 if let Some(panel_index) = dock.read(cx).panel_index_for_proto_id(panel_id) {
2980 panel = dock.update(cx, |dock, cx| {
2981 dock.activate_panel(panel_index, window, cx);
2982 dock.set_open(true, window, cx);
2983 dock.active_panel().cloned()
2984 });
2985 break;
2986 }
2987 }
2988
2989 if panel.is_some() {
2990 cx.notify();
2991 self.serialize_workspace(window, cx);
2992 }
2993
2994 panel
2995 }
2996
2997 /// Focus or unfocus the given panel type, depending on the given callback.
2998 fn focus_or_unfocus_panel<T: Panel>(
2999 &mut self,
3000 window: &mut Window,
3001 cx: &mut Context<Self>,
3002 mut should_focus: impl FnMut(&dyn PanelHandle, &mut Window, &mut Context<Dock>) -> bool,
3003 ) -> Option<Arc<dyn PanelHandle>> {
3004 let mut result_panel = None;
3005 let mut serialize = false;
3006 for dock in self.all_docks() {
3007 if let Some(panel_index) = dock.read(cx).panel_index_for_type::<T>() {
3008 let mut focus_center = false;
3009 let panel = dock.update(cx, |dock, cx| {
3010 dock.activate_panel(panel_index, window, cx);
3011
3012 let panel = dock.active_panel().cloned();
3013 if let Some(panel) = panel.as_ref() {
3014 if should_focus(&**panel, window, cx) {
3015 dock.set_open(true, window, cx);
3016 panel.panel_focus_handle(cx).focus(window);
3017 } else {
3018 focus_center = true;
3019 }
3020 }
3021 panel
3022 });
3023
3024 if focus_center {
3025 self.active_pane
3026 .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx)))
3027 }
3028
3029 result_panel = panel;
3030 serialize = true;
3031 break;
3032 }
3033 }
3034
3035 if serialize {
3036 self.serialize_workspace(window, cx);
3037 }
3038
3039 cx.notify();
3040 result_panel
3041 }
3042
3043 /// Open the panel of the given type
3044 pub fn open_panel<T: Panel>(&mut self, window: &mut Window, cx: &mut Context<Self>) {
3045 for dock in self.all_docks() {
3046 if let Some(panel_index) = dock.read(cx).panel_index_for_type::<T>() {
3047 dock.update(cx, |dock, cx| {
3048 dock.activate_panel(panel_index, window, cx);
3049 dock.set_open(true, window, cx);
3050 });
3051 }
3052 }
3053 }
3054
3055 pub fn panel<T: Panel>(&self, cx: &App) -> Option<Entity<T>> {
3056 self.all_docks()
3057 .iter()
3058 .find_map(|dock| dock.read(cx).panel::<T>())
3059 }
3060
3061 fn dismiss_zoomed_items_to_reveal(
3062 &mut self,
3063 dock_to_reveal: Option<DockPosition>,
3064 window: &mut Window,
3065 cx: &mut Context<Self>,
3066 ) {
3067 // If a center pane is zoomed, unzoom it.
3068 for pane in &self.panes {
3069 if pane != &self.active_pane || dock_to_reveal.is_some() {
3070 pane.update(cx, |pane, cx| pane.set_zoomed(false, cx));
3071 }
3072 }
3073
3074 // If another dock is zoomed, hide it.
3075 let mut focus_center = false;
3076 for dock in self.all_docks() {
3077 dock.update(cx, |dock, cx| {
3078 if Some(dock.position()) != dock_to_reveal
3079 && let Some(panel) = dock.active_panel()
3080 && panel.is_zoomed(window, cx)
3081 {
3082 focus_center |= panel.panel_focus_handle(cx).contains_focused(window, cx);
3083 dock.set_open(false, window, cx);
3084 }
3085 });
3086 }
3087
3088 if focus_center {
3089 self.active_pane
3090 .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx)))
3091 }
3092
3093 if self.zoomed_position != dock_to_reveal {
3094 self.zoomed = None;
3095 self.zoomed_position = None;
3096 cx.emit(Event::ZoomChanged);
3097 }
3098
3099 cx.notify();
3100 }
3101
3102 fn add_pane(&mut self, window: &mut Window, cx: &mut Context<Self>) -> Entity<Pane> {
3103 let pane = cx.new(|cx| {
3104 let mut pane = Pane::new(
3105 self.weak_handle(),
3106 self.project.clone(),
3107 self.pane_history_timestamp.clone(),
3108 None,
3109 NewFile.boxed_clone(),
3110 window,
3111 cx,
3112 );
3113 pane.set_can_split(Some(Arc::new(|_, _, _, _| true)));
3114 pane
3115 });
3116 cx.subscribe_in(&pane, window, Self::handle_pane_event)
3117 .detach();
3118 self.panes.push(pane.clone());
3119
3120 window.focus(&pane.focus_handle(cx));
3121
3122 cx.emit(Event::PaneAdded(pane.clone()));
3123 pane
3124 }
3125
3126 pub fn add_item_to_center(
3127 &mut self,
3128 item: Box<dyn ItemHandle>,
3129 window: &mut Window,
3130 cx: &mut Context<Self>,
3131 ) -> bool {
3132 if let Some(center_pane) = self.last_active_center_pane.clone() {
3133 if let Some(center_pane) = center_pane.upgrade() {
3134 center_pane.update(cx, |pane, cx| {
3135 pane.add_item(item, true, true, None, window, cx)
3136 });
3137 true
3138 } else {
3139 false
3140 }
3141 } else {
3142 false
3143 }
3144 }
3145
3146 pub fn add_item_to_active_pane(
3147 &mut self,
3148 item: Box<dyn ItemHandle>,
3149 destination_index: Option<usize>,
3150 focus_item: bool,
3151 window: &mut Window,
3152 cx: &mut App,
3153 ) {
3154 self.add_item(
3155 self.active_pane.clone(),
3156 item,
3157 destination_index,
3158 false,
3159 focus_item,
3160 window,
3161 cx,
3162 )
3163 }
3164
3165 pub fn add_item(
3166 &mut self,
3167 pane: Entity<Pane>,
3168 item: Box<dyn ItemHandle>,
3169 destination_index: Option<usize>,
3170 activate_pane: bool,
3171 focus_item: bool,
3172 window: &mut Window,
3173 cx: &mut App,
3174 ) {
3175 if let Some(text) = item.telemetry_event_text(cx) {
3176 telemetry::event!(text);
3177 }
3178
3179 pane.update(cx, |pane, cx| {
3180 pane.add_item(
3181 item,
3182 activate_pane,
3183 focus_item,
3184 destination_index,
3185 window,
3186 cx,
3187 )
3188 });
3189 }
3190
3191 pub fn split_item(
3192 &mut self,
3193 split_direction: SplitDirection,
3194 item: Box<dyn ItemHandle>,
3195 window: &mut Window,
3196 cx: &mut Context<Self>,
3197 ) {
3198 let new_pane = self.split_pane(self.active_pane.clone(), split_direction, window, cx);
3199 self.add_item(new_pane, item, None, true, true, window, cx);
3200 }
3201
3202 pub fn open_abs_path(
3203 &mut self,
3204 abs_path: PathBuf,
3205 options: OpenOptions,
3206 window: &mut Window,
3207 cx: &mut Context<Self>,
3208 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
3209 cx.spawn_in(window, async move |workspace, cx| {
3210 let open_paths_task_result = workspace
3211 .update_in(cx, |workspace, window, cx| {
3212 workspace.open_paths(vec![abs_path.clone()], options, None, window, cx)
3213 })
3214 .with_context(|| format!("open abs path {abs_path:?} task spawn"))?
3215 .await;
3216 anyhow::ensure!(
3217 open_paths_task_result.len() == 1,
3218 "open abs path {abs_path:?} task returned incorrect number of results"
3219 );
3220 match open_paths_task_result
3221 .into_iter()
3222 .next()
3223 .expect("ensured single task result")
3224 {
3225 Some(open_result) => {
3226 open_result.with_context(|| format!("open abs path {abs_path:?} task join"))
3227 }
3228 None => anyhow::bail!("open abs path {abs_path:?} task returned None"),
3229 }
3230 })
3231 }
3232
3233 pub fn split_abs_path(
3234 &mut self,
3235 abs_path: PathBuf,
3236 visible: bool,
3237 window: &mut Window,
3238 cx: &mut Context<Self>,
3239 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
3240 let project_path_task =
3241 Workspace::project_path_for_path(self.project.clone(), &abs_path, visible, cx);
3242 cx.spawn_in(window, async move |this, cx| {
3243 let (_, path) = project_path_task.await?;
3244 this.update_in(cx, |this, window, cx| this.split_path(path, window, cx))?
3245 .await
3246 })
3247 }
3248
3249 pub fn open_path(
3250 &mut self,
3251 path: impl Into<ProjectPath>,
3252 pane: Option<WeakEntity<Pane>>,
3253 focus_item: bool,
3254 window: &mut Window,
3255 cx: &mut App,
3256 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
3257 self.open_path_preview(path, pane, focus_item, false, true, window, cx)
3258 }
3259
3260 pub fn open_path_preview(
3261 &mut self,
3262 path: impl Into<ProjectPath>,
3263 pane: Option<WeakEntity<Pane>>,
3264 focus_item: bool,
3265 allow_preview: bool,
3266 activate: bool,
3267 window: &mut Window,
3268 cx: &mut App,
3269 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
3270 let pane = pane.unwrap_or_else(|| {
3271 self.last_active_center_pane.clone().unwrap_or_else(|| {
3272 self.panes
3273 .first()
3274 .expect("There must be an active pane")
3275 .downgrade()
3276 })
3277 });
3278
3279 let project_path = path.into();
3280 let task = self.load_path(project_path.clone(), window, cx);
3281 window.spawn(cx, async move |cx| {
3282 let (project_entry_id, build_item) = task.await?;
3283
3284 pane.update_in(cx, |pane, window, cx| {
3285 pane.open_item(
3286 project_entry_id,
3287 project_path,
3288 focus_item,
3289 allow_preview,
3290 activate,
3291 None,
3292 window,
3293 cx,
3294 build_item,
3295 )
3296 })
3297 })
3298 }
3299
3300 pub fn split_path(
3301 &mut self,
3302 path: impl Into<ProjectPath>,
3303 window: &mut Window,
3304 cx: &mut Context<Self>,
3305 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
3306 self.split_path_preview(path, false, None, window, cx)
3307 }
3308
3309 pub fn split_path_preview(
3310 &mut self,
3311 path: impl Into<ProjectPath>,
3312 allow_preview: bool,
3313 split_direction: Option<SplitDirection>,
3314 window: &mut Window,
3315 cx: &mut Context<Self>,
3316 ) -> Task<anyhow::Result<Box<dyn ItemHandle>>> {
3317 let pane = self.last_active_center_pane.clone().unwrap_or_else(|| {
3318 self.panes
3319 .first()
3320 .expect("There must be an active pane")
3321 .downgrade()
3322 });
3323
3324 if let Member::Pane(center_pane) = &self.center.root
3325 && center_pane.read(cx).items_len() == 0
3326 {
3327 return self.open_path(path, Some(pane), true, window, cx);
3328 }
3329
3330 let project_path = path.into();
3331 let task = self.load_path(project_path.clone(), window, cx);
3332 cx.spawn_in(window, async move |this, cx| {
3333 let (project_entry_id, build_item) = task.await?;
3334 this.update_in(cx, move |this, window, cx| -> Option<_> {
3335 let pane = pane.upgrade()?;
3336 let new_pane = this.split_pane(
3337 pane,
3338 split_direction.unwrap_or(SplitDirection::Right),
3339 window,
3340 cx,
3341 );
3342 new_pane.update(cx, |new_pane, cx| {
3343 Some(new_pane.open_item(
3344 project_entry_id,
3345 project_path,
3346 true,
3347 allow_preview,
3348 true,
3349 None,
3350 window,
3351 cx,
3352 build_item,
3353 ))
3354 })
3355 })
3356 .map(|option| option.context("pane was dropped"))?
3357 })
3358 }
3359
3360 fn load_path(
3361 &mut self,
3362 path: ProjectPath,
3363 window: &mut Window,
3364 cx: &mut App,
3365 ) -> Task<Result<(Option<ProjectEntryId>, WorkspaceItemBuilder)>> {
3366 let project = self.project().clone();
3367 let registry = cx.default_global::<ProjectItemRegistry>().clone();
3368 registry.open_path(&project, &path, window, cx)
3369 }
3370
3371 pub fn find_project_item<T>(
3372 &self,
3373 pane: &Entity<Pane>,
3374 project_item: &Entity<T::Item>,
3375 cx: &App,
3376 ) -> Option<Entity<T>>
3377 where
3378 T: ProjectItem,
3379 {
3380 use project::ProjectItem as _;
3381 let project_item = project_item.read(cx);
3382 let entry_id = project_item.entry_id(cx);
3383 let project_path = project_item.project_path(cx);
3384
3385 let mut item = None;
3386 if let Some(entry_id) = entry_id {
3387 item = pane.read(cx).item_for_entry(entry_id, cx);
3388 }
3389 if item.is_none()
3390 && let Some(project_path) = project_path
3391 {
3392 item = pane.read(cx).item_for_path(project_path, cx);
3393 }
3394
3395 item.and_then(|item| item.downcast::<T>())
3396 }
3397
3398 pub fn is_project_item_open<T>(
3399 &self,
3400 pane: &Entity<Pane>,
3401 project_item: &Entity<T::Item>,
3402 cx: &App,
3403 ) -> bool
3404 where
3405 T: ProjectItem,
3406 {
3407 self.find_project_item::<T>(pane, project_item, cx)
3408 .is_some()
3409 }
3410
3411 pub fn open_project_item<T>(
3412 &mut self,
3413 pane: Entity<Pane>,
3414 project_item: Entity<T::Item>,
3415 activate_pane: bool,
3416 focus_item: bool,
3417 window: &mut Window,
3418 cx: &mut Context<Self>,
3419 ) -> Entity<T>
3420 where
3421 T: ProjectItem,
3422 {
3423 if let Some(item) = self.find_project_item(&pane, &project_item, cx) {
3424 self.activate_item(&item, activate_pane, focus_item, window, cx);
3425 return item;
3426 }
3427
3428 let item = pane.update(cx, |pane, cx| {
3429 cx.new(|cx| {
3430 T::for_project_item(self.project().clone(), Some(pane), project_item, window, cx)
3431 })
3432 });
3433 let item_id = item.item_id();
3434 let mut destination_index = None;
3435 pane.update(cx, |pane, cx| {
3436 if PreviewTabsSettings::get_global(cx).enable_preview_from_code_navigation
3437 && let Some(preview_item_id) = pane.preview_item_id()
3438 && preview_item_id != item_id
3439 {
3440 destination_index = pane.close_current_preview_item(window, cx);
3441 }
3442 pane.set_preview_item_id(Some(item.item_id()), cx)
3443 });
3444
3445 self.add_item(
3446 pane,
3447 Box::new(item.clone()),
3448 destination_index,
3449 activate_pane,
3450 focus_item,
3451 window,
3452 cx,
3453 );
3454 item
3455 }
3456
3457 pub fn open_shared_screen(
3458 &mut self,
3459 peer_id: PeerId,
3460 window: &mut Window,
3461 cx: &mut Context<Self>,
3462 ) {
3463 if let Some(shared_screen) =
3464 self.shared_screen_for_peer(peer_id, &self.active_pane, window, cx)
3465 {
3466 self.active_pane.update(cx, |pane, cx| {
3467 pane.add_item(Box::new(shared_screen), false, true, None, window, cx)
3468 });
3469 }
3470 }
3471
3472 pub fn activate_item(
3473 &mut self,
3474 item: &dyn ItemHandle,
3475 activate_pane: bool,
3476 focus_item: bool,
3477 window: &mut Window,
3478 cx: &mut App,
3479 ) -> bool {
3480 let result = self.panes.iter().find_map(|pane| {
3481 pane.read(cx)
3482 .index_for_item(item)
3483 .map(|ix| (pane.clone(), ix))
3484 });
3485 if let Some((pane, ix)) = result {
3486 pane.update(cx, |pane, cx| {
3487 pane.activate_item(ix, activate_pane, focus_item, window, cx)
3488 });
3489 true
3490 } else {
3491 false
3492 }
3493 }
3494
3495 fn activate_pane_at_index(
3496 &mut self,
3497 action: &ActivatePane,
3498 window: &mut Window,
3499 cx: &mut Context<Self>,
3500 ) {
3501 let panes = self.center.panes();
3502 if let Some(pane) = panes.get(action.0).map(|p| (*p).clone()) {
3503 window.focus(&pane.focus_handle(cx));
3504 } else {
3505 self.split_and_clone(self.active_pane.clone(), SplitDirection::Right, window, cx);
3506 }
3507 }
3508
3509 fn move_item_to_pane_at_index(
3510 &mut self,
3511 action: &MoveItemToPane,
3512 window: &mut Window,
3513 cx: &mut Context<Self>,
3514 ) {
3515 let panes = self.center.panes();
3516 let destination = match panes.get(action.destination) {
3517 Some(&destination) => destination.clone(),
3518 None => {
3519 if !action.clone && self.active_pane.read(cx).items_len() < 2 {
3520 return;
3521 }
3522 let direction = SplitDirection::Right;
3523 let split_off_pane = self
3524 .find_pane_in_direction(direction, cx)
3525 .unwrap_or_else(|| self.active_pane.clone());
3526 let new_pane = self.add_pane(window, cx);
3527 if self
3528 .center
3529 .split(&split_off_pane, &new_pane, direction)
3530 .log_err()
3531 .is_none()
3532 {
3533 return;
3534 };
3535 new_pane
3536 }
3537 };
3538
3539 if action.clone {
3540 clone_active_item(
3541 self.database_id(),
3542 &self.active_pane,
3543 &destination,
3544 action.focus,
3545 window,
3546 cx,
3547 )
3548 } else {
3549 move_active_item(
3550 &self.active_pane,
3551 &destination,
3552 action.focus,
3553 true,
3554 window,
3555 cx,
3556 )
3557 }
3558 }
3559
3560 pub fn activate_next_pane(&mut self, window: &mut Window, cx: &mut App) {
3561 let panes = self.center.panes();
3562 if let Some(ix) = panes.iter().position(|pane| **pane == self.active_pane) {
3563 let next_ix = (ix + 1) % panes.len();
3564 let next_pane = panes[next_ix].clone();
3565 window.focus(&next_pane.focus_handle(cx));
3566 }
3567 }
3568
3569 pub fn activate_previous_pane(&mut self, window: &mut Window, cx: &mut App) {
3570 let panes = self.center.panes();
3571 if let Some(ix) = panes.iter().position(|pane| **pane == self.active_pane) {
3572 let prev_ix = cmp::min(ix.wrapping_sub(1), panes.len() - 1);
3573 let prev_pane = panes[prev_ix].clone();
3574 window.focus(&prev_pane.focus_handle(cx));
3575 }
3576 }
3577
3578 pub fn activate_pane_in_direction(
3579 &mut self,
3580 direction: SplitDirection,
3581 window: &mut Window,
3582 cx: &mut App,
3583 ) {
3584 use ActivateInDirectionTarget as Target;
3585 enum Origin {
3586 LeftDock,
3587 RightDock,
3588 BottomDock,
3589 Center,
3590 }
3591
3592 let origin: Origin = [
3593 (&self.left_dock, Origin::LeftDock),
3594 (&self.right_dock, Origin::RightDock),
3595 (&self.bottom_dock, Origin::BottomDock),
3596 ]
3597 .into_iter()
3598 .find_map(|(dock, origin)| {
3599 if dock.focus_handle(cx).contains_focused(window, cx) && dock.read(cx).is_open() {
3600 Some(origin)
3601 } else {
3602 None
3603 }
3604 })
3605 .unwrap_or(Origin::Center);
3606
3607 let get_last_active_pane = || {
3608 let pane = self
3609 .last_active_center_pane
3610 .clone()
3611 .unwrap_or_else(|| {
3612 self.panes
3613 .first()
3614 .expect("There must be an active pane")
3615 .downgrade()
3616 })
3617 .upgrade()?;
3618 (pane.read(cx).items_len() != 0).then_some(pane)
3619 };
3620
3621 let try_dock =
3622 |dock: &Entity<Dock>| dock.read(cx).is_open().then(|| Target::Dock(dock.clone()));
3623
3624 let target = match (origin, direction) {
3625 // We're in the center, so we first try to go to a different pane,
3626 // otherwise try to go to a dock.
3627 (Origin::Center, direction) => {
3628 if let Some(pane) = self.find_pane_in_direction(direction, cx) {
3629 Some(Target::Pane(pane))
3630 } else {
3631 match direction {
3632 SplitDirection::Up => None,
3633 SplitDirection::Down => try_dock(&self.bottom_dock),
3634 SplitDirection::Left => try_dock(&self.left_dock),
3635 SplitDirection::Right => try_dock(&self.right_dock),
3636 }
3637 }
3638 }
3639
3640 (Origin::LeftDock, SplitDirection::Right) => {
3641 if let Some(last_active_pane) = get_last_active_pane() {
3642 Some(Target::Pane(last_active_pane))
3643 } else {
3644 try_dock(&self.bottom_dock).or_else(|| try_dock(&self.right_dock))
3645 }
3646 }
3647
3648 (Origin::LeftDock, SplitDirection::Down)
3649 | (Origin::RightDock, SplitDirection::Down) => try_dock(&self.bottom_dock),
3650
3651 (Origin::BottomDock, SplitDirection::Up) => get_last_active_pane().map(Target::Pane),
3652 (Origin::BottomDock, SplitDirection::Left) => try_dock(&self.left_dock),
3653 (Origin::BottomDock, SplitDirection::Right) => try_dock(&self.right_dock),
3654
3655 (Origin::RightDock, SplitDirection::Left) => {
3656 if let Some(last_active_pane) = get_last_active_pane() {
3657 Some(Target::Pane(last_active_pane))
3658 } else {
3659 try_dock(&self.bottom_dock).or_else(|| try_dock(&self.left_dock))
3660 }
3661 }
3662
3663 _ => None,
3664 };
3665
3666 match target {
3667 Some(ActivateInDirectionTarget::Pane(pane)) => {
3668 let pane = pane.read(cx);
3669 if let Some(item) = pane.active_item() {
3670 item.item_focus_handle(cx).focus(window);
3671 } else {
3672 log::error!(
3673 "Could not find a focus target when in switching focus in {direction} direction for a pane",
3674 );
3675 }
3676 }
3677 Some(ActivateInDirectionTarget::Dock(dock)) => {
3678 // Defer this to avoid a panic when the dock's active panel is already on the stack.
3679 window.defer(cx, move |window, cx| {
3680 let dock = dock.read(cx);
3681 if let Some(panel) = dock.active_panel() {
3682 panel.panel_focus_handle(cx).focus(window);
3683 } else {
3684 log::error!("Could not find a focus target when in switching focus in {direction} direction for a {:?} dock", dock.position());
3685 }
3686 })
3687 }
3688 None => {}
3689 }
3690 }
3691
3692 pub fn move_item_to_pane_in_direction(
3693 &mut self,
3694 action: &MoveItemToPaneInDirection,
3695 window: &mut Window,
3696 cx: &mut Context<Self>,
3697 ) {
3698 let destination = match self.find_pane_in_direction(action.direction, cx) {
3699 Some(destination) => destination,
3700 None => {
3701 if !action.clone && self.active_pane.read(cx).items_len() < 2 {
3702 return;
3703 }
3704 let new_pane = self.add_pane(window, cx);
3705 if self
3706 .center
3707 .split(&self.active_pane, &new_pane, action.direction)
3708 .log_err()
3709 .is_none()
3710 {
3711 return;
3712 };
3713 new_pane
3714 }
3715 };
3716
3717 if action.clone {
3718 clone_active_item(
3719 self.database_id(),
3720 &self.active_pane,
3721 &destination,
3722 action.focus,
3723 window,
3724 cx,
3725 )
3726 } else {
3727 move_active_item(
3728 &self.active_pane,
3729 &destination,
3730 action.focus,
3731 true,
3732 window,
3733 cx,
3734 );
3735 }
3736 }
3737
3738 pub fn bounding_box_for_pane(&self, pane: &Entity<Pane>) -> Option<Bounds<Pixels>> {
3739 self.center.bounding_box_for_pane(pane)
3740 }
3741
3742 pub fn find_pane_in_direction(
3743 &mut self,
3744 direction: SplitDirection,
3745 cx: &App,
3746 ) -> Option<Entity<Pane>> {
3747 self.center
3748 .find_pane_in_direction(&self.active_pane, direction, cx)
3749 .cloned()
3750 }
3751
3752 pub fn swap_pane_in_direction(&mut self, direction: SplitDirection, cx: &mut Context<Self>) {
3753 if let Some(to) = self.find_pane_in_direction(direction, cx) {
3754 self.center.swap(&self.active_pane, &to);
3755 cx.notify();
3756 }
3757 }
3758
3759 pub fn resize_pane(
3760 &mut self,
3761 axis: gpui::Axis,
3762 amount: Pixels,
3763 window: &mut Window,
3764 cx: &mut Context<Self>,
3765 ) {
3766 let docks = self.all_docks();
3767 let active_dock = docks
3768 .into_iter()
3769 .find(|dock| dock.focus_handle(cx).contains_focused(window, cx));
3770
3771 if let Some(dock) = active_dock {
3772 let Some(panel_size) = dock.read(cx).active_panel_size(window, cx) else {
3773 return;
3774 };
3775 match dock.read(cx).position() {
3776 DockPosition::Left => self.resize_left_dock(panel_size + amount, window, cx),
3777 DockPosition::Bottom => self.resize_bottom_dock(panel_size + amount, window, cx),
3778 DockPosition::Right => self.resize_right_dock(panel_size + amount, window, cx),
3779 }
3780 } else {
3781 self.center
3782 .resize(&self.active_pane, axis, amount, &self.bounds);
3783 }
3784 cx.notify();
3785 }
3786
3787 pub fn reset_pane_sizes(&mut self, cx: &mut Context<Self>) {
3788 self.center.reset_pane_sizes();
3789 cx.notify();
3790 }
3791
3792 fn handle_pane_focused(
3793 &mut self,
3794 pane: Entity<Pane>,
3795 window: &mut Window,
3796 cx: &mut Context<Self>,
3797 ) {
3798 // This is explicitly hoisted out of the following check for pane identity as
3799 // terminal panel panes are not registered as a center panes.
3800 self.status_bar.update(cx, |status_bar, cx| {
3801 status_bar.set_active_pane(&pane, window, cx);
3802 });
3803 if self.active_pane != pane {
3804 self.set_active_pane(&pane, window, cx);
3805 }
3806
3807 if self.last_active_center_pane.is_none() {
3808 self.last_active_center_pane = Some(pane.downgrade());
3809 }
3810
3811 self.dismiss_zoomed_items_to_reveal(None, window, cx);
3812 if pane.read(cx).is_zoomed() {
3813 self.zoomed = Some(pane.downgrade().into());
3814 } else {
3815 self.zoomed = None;
3816 }
3817 self.zoomed_position = None;
3818 cx.emit(Event::ZoomChanged);
3819 self.update_active_view_for_followers(window, cx);
3820 pane.update(cx, |pane, _| {
3821 pane.track_alternate_file_items();
3822 });
3823
3824 cx.notify();
3825 }
3826
3827 fn set_active_pane(
3828 &mut self,
3829 pane: &Entity<Pane>,
3830 window: &mut Window,
3831 cx: &mut Context<Self>,
3832 ) {
3833 self.active_pane = pane.clone();
3834 self.active_item_path_changed(window, cx);
3835 self.last_active_center_pane = Some(pane.downgrade());
3836 }
3837
3838 fn handle_panel_focused(&mut self, window: &mut Window, cx: &mut Context<Self>) {
3839 self.update_active_view_for_followers(window, cx);
3840 }
3841
3842 fn handle_pane_event(
3843 &mut self,
3844 pane: &Entity<Pane>,
3845 event: &pane::Event,
3846 window: &mut Window,
3847 cx: &mut Context<Self>,
3848 ) {
3849 let mut serialize_workspace = true;
3850 match event {
3851 pane::Event::AddItem { item } => {
3852 item.added_to_pane(self, pane.clone(), window, cx);
3853 cx.emit(Event::ItemAdded {
3854 item: item.boxed_clone(),
3855 });
3856 }
3857 pane::Event::Split(direction) => {
3858 self.split_and_clone(pane.clone(), *direction, window, cx);
3859 }
3860 pane::Event::JoinIntoNext => {
3861 self.join_pane_into_next(pane.clone(), window, cx);
3862 }
3863 pane::Event::JoinAll => {
3864 self.join_all_panes(window, cx);
3865 }
3866 pane::Event::Remove { focus_on_pane } => {
3867 self.remove_pane(pane.clone(), focus_on_pane.clone(), window, cx);
3868 }
3869 pane::Event::ActivateItem {
3870 local,
3871 focus_changed,
3872 } => {
3873 window.invalidate_character_coordinates();
3874
3875 pane.update(cx, |pane, _| {
3876 pane.track_alternate_file_items();
3877 });
3878 if *local {
3879 self.unfollow_in_pane(pane, window, cx);
3880 }
3881 serialize_workspace = *focus_changed || pane != self.active_pane();
3882 if pane == self.active_pane() {
3883 self.active_item_path_changed(window, cx);
3884 self.update_active_view_for_followers(window, cx);
3885 } else if *local {
3886 self.set_active_pane(pane, window, cx);
3887 }
3888 }
3889 pane::Event::UserSavedItem { item, save_intent } => {
3890 cx.emit(Event::UserSavedItem {
3891 pane: pane.downgrade(),
3892 item: item.boxed_clone(),
3893 save_intent: *save_intent,
3894 });
3895 serialize_workspace = false;
3896 }
3897 pane::Event::ChangeItemTitle => {
3898 if *pane == self.active_pane {
3899 self.active_item_path_changed(window, cx);
3900 }
3901 serialize_workspace = false;
3902 }
3903 pane::Event::RemoveItem { .. } => {}
3904 pane::Event::RemovedItem { item } => {
3905 cx.emit(Event::ActiveItemChanged);
3906 self.update_window_edited(window, cx);
3907 if let hash_map::Entry::Occupied(entry) = self.panes_by_item.entry(item.item_id())
3908 && entry.get().entity_id() == pane.entity_id()
3909 {
3910 entry.remove();
3911 }
3912 }
3913 pane::Event::Focus => {
3914 window.invalidate_character_coordinates();
3915 self.handle_pane_focused(pane.clone(), window, cx);
3916 }
3917 pane::Event::ZoomIn => {
3918 if *pane == self.active_pane {
3919 pane.update(cx, |pane, cx| pane.set_zoomed(true, cx));
3920 if pane.read(cx).has_focus(window, cx) {
3921 self.zoomed = Some(pane.downgrade().into());
3922 self.zoomed_position = None;
3923 cx.emit(Event::ZoomChanged);
3924 }
3925 cx.notify();
3926 }
3927 }
3928 pane::Event::ZoomOut => {
3929 pane.update(cx, |pane, cx| pane.set_zoomed(false, cx));
3930 if self.zoomed_position.is_none() {
3931 self.zoomed = None;
3932 cx.emit(Event::ZoomChanged);
3933 }
3934 cx.notify();
3935 }
3936 pane::Event::ItemPinned | pane::Event::ItemUnpinned => {}
3937 }
3938
3939 if serialize_workspace {
3940 self.serialize_workspace(window, cx);
3941 }
3942 }
3943
3944 pub fn unfollow_in_pane(
3945 &mut self,
3946 pane: &Entity<Pane>,
3947 window: &mut Window,
3948 cx: &mut Context<Workspace>,
3949 ) -> Option<CollaboratorId> {
3950 let leader_id = self.leader_for_pane(pane)?;
3951 self.unfollow(leader_id, window, cx);
3952 Some(leader_id)
3953 }
3954
3955 pub fn split_pane(
3956 &mut self,
3957 pane_to_split: Entity<Pane>,
3958 split_direction: SplitDirection,
3959 window: &mut Window,
3960 cx: &mut Context<Self>,
3961 ) -> Entity<Pane> {
3962 let new_pane = self.add_pane(window, cx);
3963 self.center
3964 .split(&pane_to_split, &new_pane, split_direction)
3965 .unwrap();
3966 cx.notify();
3967 new_pane
3968 }
3969
3970 pub fn split_and_clone(
3971 &mut self,
3972 pane: Entity<Pane>,
3973 direction: SplitDirection,
3974 window: &mut Window,
3975 cx: &mut Context<Self>,
3976 ) -> Option<Entity<Pane>> {
3977 let item = pane.read(cx).active_item()?;
3978 let maybe_pane_handle =
3979 if let Some(clone) = item.clone_on_split(self.database_id(), window, cx) {
3980 let new_pane = self.add_pane(window, cx);
3981 new_pane.update(cx, |pane, cx| {
3982 pane.add_item(clone, true, true, None, window, cx)
3983 });
3984 self.center.split(&pane, &new_pane, direction).unwrap();
3985 Some(new_pane)
3986 } else {
3987 None
3988 };
3989 cx.notify();
3990 maybe_pane_handle
3991 }
3992
3993 pub fn split_pane_with_item(
3994 &mut self,
3995 pane_to_split: WeakEntity<Pane>,
3996 split_direction: SplitDirection,
3997 from: WeakEntity<Pane>,
3998 item_id_to_move: EntityId,
3999 window: &mut Window,
4000 cx: &mut Context<Self>,
4001 ) {
4002 let Some(pane_to_split) = pane_to_split.upgrade() else {
4003 return;
4004 };
4005 let Some(from) = from.upgrade() else {
4006 return;
4007 };
4008
4009 let new_pane = self.add_pane(window, cx);
4010 move_item(&from, &new_pane, item_id_to_move, 0, true, window, cx);
4011 self.center
4012 .split(&pane_to_split, &new_pane, split_direction)
4013 .unwrap();
4014 cx.notify();
4015 }
4016
4017 pub fn split_pane_with_project_entry(
4018 &mut self,
4019 pane_to_split: WeakEntity<Pane>,
4020 split_direction: SplitDirection,
4021 project_entry: ProjectEntryId,
4022 window: &mut Window,
4023 cx: &mut Context<Self>,
4024 ) -> Option<Task<Result<()>>> {
4025 let pane_to_split = pane_to_split.upgrade()?;
4026 let new_pane = self.add_pane(window, cx);
4027 self.center
4028 .split(&pane_to_split, &new_pane, split_direction)
4029 .unwrap();
4030
4031 let path = self.project.read(cx).path_for_entry(project_entry, cx)?;
4032 let task = self.open_path(path, Some(new_pane.downgrade()), true, window, cx);
4033 Some(cx.foreground_executor().spawn(async move {
4034 task.await?;
4035 Ok(())
4036 }))
4037 }
4038
4039 pub fn join_all_panes(&mut self, window: &mut Window, cx: &mut Context<Self>) {
4040 let active_item = self.active_pane.read(cx).active_item();
4041 for pane in &self.panes {
4042 join_pane_into_active(&self.active_pane, pane, window, cx);
4043 }
4044 if let Some(active_item) = active_item {
4045 self.activate_item(active_item.as_ref(), true, true, window, cx);
4046 }
4047 cx.notify();
4048 }
4049
4050 pub fn join_pane_into_next(
4051 &mut self,
4052 pane: Entity<Pane>,
4053 window: &mut Window,
4054 cx: &mut Context<Self>,
4055 ) {
4056 let next_pane = self
4057 .find_pane_in_direction(SplitDirection::Right, cx)
4058 .or_else(|| self.find_pane_in_direction(SplitDirection::Down, cx))
4059 .or_else(|| self.find_pane_in_direction(SplitDirection::Left, cx))
4060 .or_else(|| self.find_pane_in_direction(SplitDirection::Up, cx));
4061 let Some(next_pane) = next_pane else {
4062 return;
4063 };
4064 move_all_items(&pane, &next_pane, window, cx);
4065 cx.notify();
4066 }
4067
4068 fn remove_pane(
4069 &mut self,
4070 pane: Entity<Pane>,
4071 focus_on: Option<Entity<Pane>>,
4072 window: &mut Window,
4073 cx: &mut Context<Self>,
4074 ) {
4075 if self.center.remove(&pane).unwrap() {
4076 self.force_remove_pane(&pane, &focus_on, window, cx);
4077 self.unfollow_in_pane(&pane, window, cx);
4078 self.last_leaders_by_pane.remove(&pane.downgrade());
4079 for removed_item in pane.read(cx).items() {
4080 self.panes_by_item.remove(&removed_item.item_id());
4081 }
4082
4083 cx.notify();
4084 } else {
4085 self.active_item_path_changed(window, cx);
4086 }
4087 cx.emit(Event::PaneRemoved);
4088 }
4089
4090 pub fn panes(&self) -> &[Entity<Pane>] {
4091 &self.panes
4092 }
4093
4094 pub fn active_pane(&self) -> &Entity<Pane> {
4095 &self.active_pane
4096 }
4097
4098 pub fn focused_pane(&self, window: &Window, cx: &App) -> Entity<Pane> {
4099 for dock in self.all_docks() {
4100 if dock.focus_handle(cx).contains_focused(window, cx)
4101 && let Some(pane) = dock
4102 .read(cx)
4103 .active_panel()
4104 .and_then(|panel| panel.pane(cx))
4105 {
4106 return pane;
4107 }
4108 }
4109 self.active_pane().clone()
4110 }
4111
4112 pub fn adjacent_pane(&mut self, window: &mut Window, cx: &mut Context<Self>) -> Entity<Pane> {
4113 self.find_pane_in_direction(SplitDirection::Right, cx)
4114 .or_else(|| self.find_pane_in_direction(SplitDirection::Left, cx))
4115 .unwrap_or_else(|| {
4116 self.split_pane(self.active_pane.clone(), SplitDirection::Right, window, cx)
4117 })
4118 }
4119
4120 pub fn pane_for(&self, handle: &dyn ItemHandle) -> Option<Entity<Pane>> {
4121 let weak_pane = self.panes_by_item.get(&handle.item_id())?;
4122 weak_pane.upgrade()
4123 }
4124
4125 fn collaborator_left(&mut self, peer_id: PeerId, window: &mut Window, cx: &mut Context<Self>) {
4126 self.follower_states.retain(|leader_id, state| {
4127 if *leader_id == CollaboratorId::PeerId(peer_id) {
4128 for item in state.items_by_leader_view_id.values() {
4129 item.view.set_leader_id(None, window, cx);
4130 }
4131 false
4132 } else {
4133 true
4134 }
4135 });
4136 cx.notify();
4137 }
4138
4139 pub fn start_following(
4140 &mut self,
4141 leader_id: impl Into<CollaboratorId>,
4142 window: &mut Window,
4143 cx: &mut Context<Self>,
4144 ) -> Option<Task<Result<()>>> {
4145 let leader_id = leader_id.into();
4146 let pane = self.active_pane().clone();
4147
4148 self.last_leaders_by_pane
4149 .insert(pane.downgrade(), leader_id);
4150 self.unfollow(leader_id, window, cx);
4151 self.unfollow_in_pane(&pane, window, cx);
4152 self.follower_states.insert(
4153 leader_id,
4154 FollowerState {
4155 center_pane: pane.clone(),
4156 dock_pane: None,
4157 active_view_id: None,
4158 items_by_leader_view_id: Default::default(),
4159 },
4160 );
4161 cx.notify();
4162
4163 match leader_id {
4164 CollaboratorId::PeerId(leader_peer_id) => {
4165 let room_id = self.active_call()?.read(cx).room()?.read(cx).id();
4166 let project_id = self.project.read(cx).remote_id();
4167 let request = self.app_state.client.request(proto::Follow {
4168 room_id,
4169 project_id,
4170 leader_id: Some(leader_peer_id),
4171 });
4172
4173 Some(cx.spawn_in(window, async move |this, cx| {
4174 let response = request.await?;
4175 this.update(cx, |this, _| {
4176 let state = this
4177 .follower_states
4178 .get_mut(&leader_id)
4179 .context("following interrupted")?;
4180 state.active_view_id = response
4181 .active_view
4182 .as_ref()
4183 .and_then(|view| ViewId::from_proto(view.id.clone()?).ok());
4184 anyhow::Ok(())
4185 })??;
4186 if let Some(view) = response.active_view {
4187 Self::add_view_from_leader(this.clone(), leader_peer_id, &view, cx).await?;
4188 }
4189 this.update_in(cx, |this, window, cx| {
4190 this.leader_updated(leader_id, window, cx)
4191 })?;
4192 Ok(())
4193 }))
4194 }
4195 CollaboratorId::Agent => {
4196 self.leader_updated(leader_id, window, cx)?;
4197 Some(Task::ready(Ok(())))
4198 }
4199 }
4200 }
4201
4202 pub fn follow_next_collaborator(
4203 &mut self,
4204 _: &FollowNextCollaborator,
4205 window: &mut Window,
4206 cx: &mut Context<Self>,
4207 ) {
4208 let collaborators = self.project.read(cx).collaborators();
4209 let next_leader_id = if let Some(leader_id) = self.leader_for_pane(&self.active_pane) {
4210 let mut collaborators = collaborators.keys().copied();
4211 for peer_id in collaborators.by_ref() {
4212 if CollaboratorId::PeerId(peer_id) == leader_id {
4213 break;
4214 }
4215 }
4216 collaborators.next().map(CollaboratorId::PeerId)
4217 } else if let Some(last_leader_id) =
4218 self.last_leaders_by_pane.get(&self.active_pane.downgrade())
4219 {
4220 match last_leader_id {
4221 CollaboratorId::PeerId(peer_id) => {
4222 if collaborators.contains_key(peer_id) {
4223 Some(*last_leader_id)
4224 } else {
4225 None
4226 }
4227 }
4228 CollaboratorId::Agent => Some(CollaboratorId::Agent),
4229 }
4230 } else {
4231 None
4232 };
4233
4234 let pane = self.active_pane.clone();
4235 let Some(leader_id) = next_leader_id.or_else(|| {
4236 Some(CollaboratorId::PeerId(
4237 collaborators.keys().copied().next()?,
4238 ))
4239 }) else {
4240 return;
4241 };
4242 if self.unfollow_in_pane(&pane, window, cx) == Some(leader_id) {
4243 return;
4244 }
4245 if let Some(task) = self.start_following(leader_id, window, cx) {
4246 task.detach_and_log_err(cx)
4247 }
4248 }
4249
4250 pub fn follow(
4251 &mut self,
4252 leader_id: impl Into<CollaboratorId>,
4253 window: &mut Window,
4254 cx: &mut Context<Self>,
4255 ) {
4256 let leader_id = leader_id.into();
4257
4258 if let CollaboratorId::PeerId(peer_id) = leader_id {
4259 let Some(room) = ActiveCall::global(cx).read(cx).room() else {
4260 return;
4261 };
4262 let room = room.read(cx);
4263 let Some(remote_participant) = room.remote_participant_for_peer_id(peer_id) else {
4264 return;
4265 };
4266
4267 let project = self.project.read(cx);
4268
4269 let other_project_id = match remote_participant.location {
4270 call::ParticipantLocation::External => None,
4271 call::ParticipantLocation::UnsharedProject => None,
4272 call::ParticipantLocation::SharedProject { project_id } => {
4273 if Some(project_id) == project.remote_id() {
4274 None
4275 } else {
4276 Some(project_id)
4277 }
4278 }
4279 };
4280
4281 // if they are active in another project, follow there.
4282 if let Some(project_id) = other_project_id {
4283 let app_state = self.app_state.clone();
4284 crate::join_in_room_project(project_id, remote_participant.user.id, app_state, cx)
4285 .detach_and_log_err(cx);
4286 }
4287 }
4288
4289 // if you're already following, find the right pane and focus it.
4290 if let Some(follower_state) = self.follower_states.get(&leader_id) {
4291 window.focus(&follower_state.pane().focus_handle(cx));
4292
4293 return;
4294 }
4295
4296 // Otherwise, follow.
4297 if let Some(task) = self.start_following(leader_id, window, cx) {
4298 task.detach_and_log_err(cx)
4299 }
4300 }
4301
4302 pub fn unfollow(
4303 &mut self,
4304 leader_id: impl Into<CollaboratorId>,
4305 window: &mut Window,
4306 cx: &mut Context<Self>,
4307 ) -> Option<()> {
4308 cx.notify();
4309
4310 let leader_id = leader_id.into();
4311 let state = self.follower_states.remove(&leader_id)?;
4312 for (_, item) in state.items_by_leader_view_id {
4313 item.view.set_leader_id(None, window, cx);
4314 }
4315
4316 if let CollaboratorId::PeerId(leader_peer_id) = leader_id {
4317 let project_id = self.project.read(cx).remote_id();
4318 let room_id = self.active_call()?.read(cx).room()?.read(cx).id();
4319 self.app_state
4320 .client
4321 .send(proto::Unfollow {
4322 room_id,
4323 project_id,
4324 leader_id: Some(leader_peer_id),
4325 })
4326 .log_err();
4327 }
4328
4329 Some(())
4330 }
4331
4332 pub fn is_being_followed(&self, id: impl Into<CollaboratorId>) -> bool {
4333 self.follower_states.contains_key(&id.into())
4334 }
4335
4336 fn active_item_path_changed(&mut self, window: &mut Window, cx: &mut Context<Self>) {
4337 cx.emit(Event::ActiveItemChanged);
4338 let active_entry = self.active_project_path(cx);
4339 self.project
4340 .update(cx, |project, cx| project.set_active_path(active_entry, cx));
4341
4342 self.update_window_title(window, cx);
4343 }
4344
4345 fn update_window_title(&mut self, window: &mut Window, cx: &mut App) {
4346 let project = self.project().read(cx);
4347 let mut title = String::new();
4348
4349 for (i, name) in project.worktree_root_names(cx).enumerate() {
4350 if i > 0 {
4351 title.push_str(", ");
4352 }
4353 title.push_str(name);
4354 }
4355
4356 if title.is_empty() {
4357 title = "empty project".to_string();
4358 }
4359
4360 if let Some(path) = self.active_item(cx).and_then(|item| item.project_path(cx)) {
4361 let filename = path
4362 .path
4363 .file_name()
4364 .map(|s| s.to_string_lossy())
4365 .or_else(|| {
4366 Some(Cow::Borrowed(
4367 project
4368 .worktree_for_id(path.worktree_id, cx)?
4369 .read(cx)
4370 .root_name(),
4371 ))
4372 });
4373
4374 if let Some(filename) = filename {
4375 title.push_str(" — ");
4376 title.push_str(filename.as_ref());
4377 }
4378 }
4379
4380 if project.is_via_collab() {
4381 title.push_str(" ↙");
4382 } else if project.is_shared() {
4383 title.push_str(" ↗");
4384 }
4385
4386 if let Some(last_title) = self.last_window_title.as_ref()
4387 && &title == last_title
4388 {
4389 return;
4390 }
4391 window.set_window_title(&title);
4392 self.last_window_title = Some(title);
4393 }
4394
4395 fn update_window_edited(&mut self, window: &mut Window, cx: &mut App) {
4396 let is_edited = !self.project.read(cx).is_disconnected(cx) && !self.dirty_items.is_empty();
4397 if is_edited != self.window_edited {
4398 self.window_edited = is_edited;
4399 window.set_window_edited(self.window_edited)
4400 }
4401 }
4402
4403 fn update_item_dirty_state(
4404 &mut self,
4405 item: &dyn ItemHandle,
4406 window: &mut Window,
4407 cx: &mut App,
4408 ) {
4409 let is_dirty = item.is_dirty(cx);
4410 let item_id = item.item_id();
4411 let was_dirty = self.dirty_items.contains_key(&item_id);
4412 if is_dirty == was_dirty {
4413 return;
4414 }
4415 if was_dirty {
4416 self.dirty_items.remove(&item_id);
4417 self.update_window_edited(window, cx);
4418 return;
4419 }
4420 if let Some(window_handle) = window.window_handle().downcast::<Self>() {
4421 let s = item.on_release(
4422 cx,
4423 Box::new(move |cx| {
4424 window_handle
4425 .update(cx, |this, window, cx| {
4426 this.dirty_items.remove(&item_id);
4427 this.update_window_edited(window, cx)
4428 })
4429 .ok();
4430 }),
4431 );
4432 self.dirty_items.insert(item_id, s);
4433 self.update_window_edited(window, cx);
4434 }
4435 }
4436
4437 fn render_notifications(&self, _window: &mut Window, _cx: &mut Context<Self>) -> Option<Div> {
4438 if self.notifications.is_empty() {
4439 None
4440 } else {
4441 Some(
4442 div()
4443 .absolute()
4444 .right_3()
4445 .bottom_3()
4446 .w_112()
4447 .h_full()
4448 .flex()
4449 .flex_col()
4450 .justify_end()
4451 .gap_2()
4452 .children(
4453 self.notifications
4454 .iter()
4455 .map(|(_, notification)| notification.clone().into_any()),
4456 ),
4457 )
4458 }
4459 }
4460
4461 // RPC handlers
4462
4463 fn active_view_for_follower(
4464 &self,
4465 follower_project_id: Option<u64>,
4466 window: &mut Window,
4467 cx: &mut Context<Self>,
4468 ) -> Option<proto::View> {
4469 let (item, panel_id) = self.active_item_for_followers(window, cx);
4470 let item = item?;
4471 let leader_id = self
4472 .pane_for(&*item)
4473 .and_then(|pane| self.leader_for_pane(&pane));
4474 let leader_peer_id = match leader_id {
4475 Some(CollaboratorId::PeerId(peer_id)) => Some(peer_id),
4476 Some(CollaboratorId::Agent) | None => None,
4477 };
4478
4479 let item_handle = item.to_followable_item_handle(cx)?;
4480 let id = item_handle.remote_id(&self.app_state.client, window, cx)?;
4481 let variant = item_handle.to_state_proto(window, cx)?;
4482
4483 if item_handle.is_project_item(window, cx)
4484 && (follower_project_id.is_none()
4485 || follower_project_id != self.project.read(cx).remote_id())
4486 {
4487 return None;
4488 }
4489
4490 Some(proto::View {
4491 id: id.to_proto(),
4492 leader_id: leader_peer_id,
4493 variant: Some(variant),
4494 panel_id: panel_id.map(|id| id as i32),
4495 })
4496 }
4497
4498 fn handle_follow(
4499 &mut self,
4500 follower_project_id: Option<u64>,
4501 window: &mut Window,
4502 cx: &mut Context<Self>,
4503 ) -> proto::FollowResponse {
4504 let active_view = self.active_view_for_follower(follower_project_id, window, cx);
4505
4506 cx.notify();
4507 proto::FollowResponse {
4508 // TODO: Remove after version 0.145.x stabilizes.
4509 active_view_id: active_view.as_ref().and_then(|view| view.id.clone()),
4510 views: active_view.iter().cloned().collect(),
4511 active_view,
4512 }
4513 }
4514
4515 fn handle_update_followers(
4516 &mut self,
4517 leader_id: PeerId,
4518 message: proto::UpdateFollowers,
4519 _window: &mut Window,
4520 _cx: &mut Context<Self>,
4521 ) {
4522 self.leader_updates_tx
4523 .unbounded_send((leader_id, message))
4524 .ok();
4525 }
4526
4527 async fn process_leader_update(
4528 this: &WeakEntity<Self>,
4529 leader_id: PeerId,
4530 update: proto::UpdateFollowers,
4531 cx: &mut AsyncWindowContext,
4532 ) -> Result<()> {
4533 match update.variant.context("invalid update")? {
4534 proto::update_followers::Variant::CreateView(view) => {
4535 let view_id = ViewId::from_proto(view.id.clone().context("invalid view id")?)?;
4536 let should_add_view = this.update(cx, |this, _| {
4537 if let Some(state) = this.follower_states.get_mut(&leader_id.into()) {
4538 anyhow::Ok(!state.items_by_leader_view_id.contains_key(&view_id))
4539 } else {
4540 anyhow::Ok(false)
4541 }
4542 })??;
4543
4544 if should_add_view {
4545 Self::add_view_from_leader(this.clone(), leader_id, &view, cx).await?
4546 }
4547 }
4548 proto::update_followers::Variant::UpdateActiveView(update_active_view) => {
4549 let should_add_view = this.update(cx, |this, _| {
4550 if let Some(state) = this.follower_states.get_mut(&leader_id.into()) {
4551 state.active_view_id = update_active_view
4552 .view
4553 .as_ref()
4554 .and_then(|view| ViewId::from_proto(view.id.clone()?).ok());
4555
4556 if state.active_view_id.is_some_and(|view_id| {
4557 !state.items_by_leader_view_id.contains_key(&view_id)
4558 }) {
4559 anyhow::Ok(true)
4560 } else {
4561 anyhow::Ok(false)
4562 }
4563 } else {
4564 anyhow::Ok(false)
4565 }
4566 })??;
4567
4568 if should_add_view && let Some(view) = update_active_view.view {
4569 Self::add_view_from_leader(this.clone(), leader_id, &view, cx).await?
4570 }
4571 }
4572 proto::update_followers::Variant::UpdateView(update_view) => {
4573 let variant = update_view.variant.context("missing update view variant")?;
4574 let id = update_view.id.context("missing update view id")?;
4575 let mut tasks = Vec::new();
4576 this.update_in(cx, |this, window, cx| {
4577 let project = this.project.clone();
4578 if let Some(state) = this.follower_states.get(&leader_id.into()) {
4579 let view_id = ViewId::from_proto(id.clone())?;
4580 if let Some(item) = state.items_by_leader_view_id.get(&view_id) {
4581 tasks.push(item.view.apply_update_proto(
4582 &project,
4583 variant.clone(),
4584 window,
4585 cx,
4586 ));
4587 }
4588 }
4589 anyhow::Ok(())
4590 })??;
4591 try_join_all(tasks).await.log_err();
4592 }
4593 }
4594 this.update_in(cx, |this, window, cx| {
4595 this.leader_updated(leader_id, window, cx)
4596 })?;
4597 Ok(())
4598 }
4599
4600 async fn add_view_from_leader(
4601 this: WeakEntity<Self>,
4602 leader_id: PeerId,
4603 view: &proto::View,
4604 cx: &mut AsyncWindowContext,
4605 ) -> Result<()> {
4606 let this = this.upgrade().context("workspace dropped")?;
4607
4608 let Some(id) = view.id.clone() else {
4609 anyhow::bail!("no id for view");
4610 };
4611 let id = ViewId::from_proto(id)?;
4612 let panel_id = view.panel_id.and_then(proto::PanelId::from_i32);
4613
4614 let pane = this.update(cx, |this, _cx| {
4615 let state = this
4616 .follower_states
4617 .get(&leader_id.into())
4618 .context("stopped following")?;
4619 anyhow::Ok(state.pane().clone())
4620 })??;
4621 let existing_item = pane.update_in(cx, |pane, window, cx| {
4622 let client = this.read(cx).client().clone();
4623 pane.items().find_map(|item| {
4624 let item = item.to_followable_item_handle(cx)?;
4625 if item.remote_id(&client, window, cx) == Some(id) {
4626 Some(item)
4627 } else {
4628 None
4629 }
4630 })
4631 })?;
4632 let item = if let Some(existing_item) = existing_item {
4633 existing_item
4634 } else {
4635 let variant = view.variant.clone();
4636 anyhow::ensure!(variant.is_some(), "missing view variant");
4637
4638 let task = cx.update(|window, cx| {
4639 FollowableViewRegistry::from_state_proto(this.clone(), id, variant, window, cx)
4640 })?;
4641
4642 let Some(task) = task else {
4643 anyhow::bail!(
4644 "failed to construct view from leader (maybe from a different version of zed?)"
4645 );
4646 };
4647
4648 let mut new_item = task.await?;
4649 pane.update_in(cx, |pane, window, cx| {
4650 let mut item_to_remove = None;
4651 for (ix, item) in pane.items().enumerate() {
4652 if let Some(item) = item.to_followable_item_handle(cx) {
4653 match new_item.dedup(item.as_ref(), window, cx) {
4654 Some(item::Dedup::KeepExisting) => {
4655 new_item =
4656 item.boxed_clone().to_followable_item_handle(cx).unwrap();
4657 break;
4658 }
4659 Some(item::Dedup::ReplaceExisting) => {
4660 item_to_remove = Some((ix, item.item_id()));
4661 break;
4662 }
4663 None => {}
4664 }
4665 }
4666 }
4667
4668 if let Some((ix, id)) = item_to_remove {
4669 pane.remove_item(id, false, false, window, cx);
4670 pane.add_item(new_item.boxed_clone(), false, false, Some(ix), window, cx);
4671 }
4672 })?;
4673
4674 new_item
4675 };
4676
4677 this.update_in(cx, |this, window, cx| {
4678 let state = this.follower_states.get_mut(&leader_id.into())?;
4679 item.set_leader_id(Some(leader_id.into()), window, cx);
4680 state.items_by_leader_view_id.insert(
4681 id,
4682 FollowerView {
4683 view: item,
4684 location: panel_id,
4685 },
4686 );
4687
4688 Some(())
4689 })?;
4690
4691 Ok(())
4692 }
4693
4694 fn handle_agent_location_changed(&mut self, window: &mut Window, cx: &mut Context<Self>) {
4695 let Some(follower_state) = self.follower_states.get_mut(&CollaboratorId::Agent) else {
4696 return;
4697 };
4698
4699 if let Some(agent_location) = self.project.read(cx).agent_location() {
4700 let buffer_entity_id = agent_location.buffer.entity_id();
4701 let view_id = ViewId {
4702 creator: CollaboratorId::Agent,
4703 id: buffer_entity_id.as_u64(),
4704 };
4705 follower_state.active_view_id = Some(view_id);
4706
4707 let item = match follower_state.items_by_leader_view_id.entry(view_id) {
4708 hash_map::Entry::Occupied(entry) => Some(entry.into_mut()),
4709 hash_map::Entry::Vacant(entry) => {
4710 let existing_view =
4711 follower_state
4712 .center_pane
4713 .read(cx)
4714 .items()
4715 .find_map(|item| {
4716 let item = item.to_followable_item_handle(cx)?;
4717 if item.is_singleton(cx)
4718 && item.project_item_model_ids(cx).as_slice()
4719 == [buffer_entity_id]
4720 {
4721 Some(item)
4722 } else {
4723 None
4724 }
4725 });
4726 let view = existing_view.or_else(|| {
4727 agent_location.buffer.upgrade().and_then(|buffer| {
4728 cx.update_default_global(|registry: &mut ProjectItemRegistry, cx| {
4729 registry.build_item(buffer, self.project.clone(), None, window, cx)
4730 })?
4731 .to_followable_item_handle(cx)
4732 })
4733 });
4734
4735 if let Some(view) = view {
4736 Some(entry.insert(FollowerView {
4737 view,
4738 location: None,
4739 }))
4740 } else {
4741 None
4742 }
4743 }
4744 };
4745
4746 if let Some(item) = item {
4747 item.view
4748 .set_leader_id(Some(CollaboratorId::Agent), window, cx);
4749 item.view
4750 .update_agent_location(agent_location.position, window, cx);
4751 }
4752 } else {
4753 follower_state.active_view_id = None;
4754 }
4755
4756 self.leader_updated(CollaboratorId::Agent, window, cx);
4757 }
4758
4759 pub fn update_active_view_for_followers(&mut self, window: &mut Window, cx: &mut App) {
4760 let mut is_project_item = true;
4761 let mut update = proto::UpdateActiveView::default();
4762 if window.is_window_active() {
4763 let (active_item, panel_id) = self.active_item_for_followers(window, cx);
4764
4765 if let Some(item) = active_item
4766 && item.item_focus_handle(cx).contains_focused(window, cx)
4767 {
4768 let leader_id = self
4769 .pane_for(&*item)
4770 .and_then(|pane| self.leader_for_pane(&pane));
4771 let leader_peer_id = match leader_id {
4772 Some(CollaboratorId::PeerId(peer_id)) => Some(peer_id),
4773 Some(CollaboratorId::Agent) | None => None,
4774 };
4775
4776 if let Some(item) = item.to_followable_item_handle(cx) {
4777 let id = item
4778 .remote_id(&self.app_state.client, window, cx)
4779 .map(|id| id.to_proto());
4780
4781 if let Some(id) = id
4782 && let Some(variant) = item.to_state_proto(window, cx)
4783 {
4784 let view = Some(proto::View {
4785 id: id.clone(),
4786 leader_id: leader_peer_id,
4787 variant: Some(variant),
4788 panel_id: panel_id.map(|id| id as i32),
4789 });
4790
4791 is_project_item = item.is_project_item(window, cx);
4792 update = proto::UpdateActiveView {
4793 view,
4794 // TODO: Remove after version 0.145.x stabilizes.
4795 id,
4796 leader_id: leader_peer_id,
4797 };
4798 };
4799 }
4800 }
4801 }
4802
4803 let active_view_id = update.view.as_ref().and_then(|view| view.id.as_ref());
4804 if active_view_id != self.last_active_view_id.as_ref() {
4805 self.last_active_view_id = active_view_id.cloned();
4806 self.update_followers(
4807 is_project_item,
4808 proto::update_followers::Variant::UpdateActiveView(update),
4809 window,
4810 cx,
4811 );
4812 }
4813 }
4814
4815 fn active_item_for_followers(
4816 &self,
4817 window: &mut Window,
4818 cx: &mut App,
4819 ) -> (Option<Box<dyn ItemHandle>>, Option<proto::PanelId>) {
4820 let mut active_item = None;
4821 let mut panel_id = None;
4822 for dock in self.all_docks() {
4823 if dock.focus_handle(cx).contains_focused(window, cx)
4824 && let Some(panel) = dock.read(cx).active_panel()
4825 && let Some(pane) = panel.pane(cx)
4826 && let Some(item) = pane.read(cx).active_item()
4827 {
4828 active_item = Some(item);
4829 panel_id = panel.remote_id();
4830 break;
4831 }
4832 }
4833
4834 if active_item.is_none() {
4835 active_item = self.active_pane().read(cx).active_item();
4836 }
4837 (active_item, panel_id)
4838 }
4839
4840 fn update_followers(
4841 &self,
4842 project_only: bool,
4843 update: proto::update_followers::Variant,
4844 _: &mut Window,
4845 cx: &mut App,
4846 ) -> Option<()> {
4847 // If this update only applies to for followers in the current project,
4848 // then skip it unless this project is shared. If it applies to all
4849 // followers, regardless of project, then set `project_id` to none,
4850 // indicating that it goes to all followers.
4851 let project_id = if project_only {
4852 Some(self.project.read(cx).remote_id()?)
4853 } else {
4854 None
4855 };
4856 self.app_state().workspace_store.update(cx, |store, cx| {
4857 store.update_followers(project_id, update, cx)
4858 })
4859 }
4860
4861 pub fn leader_for_pane(&self, pane: &Entity<Pane>) -> Option<CollaboratorId> {
4862 self.follower_states.iter().find_map(|(leader_id, state)| {
4863 if state.center_pane == *pane || state.dock_pane.as_ref() == Some(pane) {
4864 Some(*leader_id)
4865 } else {
4866 None
4867 }
4868 })
4869 }
4870
4871 fn leader_updated(
4872 &mut self,
4873 leader_id: impl Into<CollaboratorId>,
4874 window: &mut Window,
4875 cx: &mut Context<Self>,
4876 ) -> Option<Box<dyn ItemHandle>> {
4877 cx.notify();
4878
4879 let leader_id = leader_id.into();
4880 let (panel_id, item) = match leader_id {
4881 CollaboratorId::PeerId(peer_id) => self.active_item_for_peer(peer_id, window, cx)?,
4882 CollaboratorId::Agent => (None, self.active_item_for_agent()?),
4883 };
4884
4885 let state = self.follower_states.get(&leader_id)?;
4886 let mut transfer_focus = state.center_pane.read(cx).has_focus(window, cx);
4887 let pane;
4888 if let Some(panel_id) = panel_id {
4889 pane = self
4890 .activate_panel_for_proto_id(panel_id, window, cx)?
4891 .pane(cx)?;
4892 let state = self.follower_states.get_mut(&leader_id)?;
4893 state.dock_pane = Some(pane.clone());
4894 } else {
4895 pane = state.center_pane.clone();
4896 let state = self.follower_states.get_mut(&leader_id)?;
4897 if let Some(dock_pane) = state.dock_pane.take() {
4898 transfer_focus |= dock_pane.focus_handle(cx).contains_focused(window, cx);
4899 }
4900 }
4901
4902 pane.update(cx, |pane, cx| {
4903 let focus_active_item = pane.has_focus(window, cx) || transfer_focus;
4904 if let Some(index) = pane.index_for_item(item.as_ref()) {
4905 pane.activate_item(index, false, false, window, cx);
4906 } else {
4907 pane.add_item(item.boxed_clone(), false, false, None, window, cx)
4908 }
4909
4910 if focus_active_item {
4911 pane.focus_active_item(window, cx)
4912 }
4913 });
4914
4915 Some(item)
4916 }
4917
4918 fn active_item_for_agent(&self) -> Option<Box<dyn ItemHandle>> {
4919 let state = self.follower_states.get(&CollaboratorId::Agent)?;
4920 let active_view_id = state.active_view_id?;
4921 Some(
4922 state
4923 .items_by_leader_view_id
4924 .get(&active_view_id)?
4925 .view
4926 .boxed_clone(),
4927 )
4928 }
4929
4930 fn active_item_for_peer(
4931 &self,
4932 peer_id: PeerId,
4933 window: &mut Window,
4934 cx: &mut Context<Self>,
4935 ) -> Option<(Option<PanelId>, Box<dyn ItemHandle>)> {
4936 let call = self.active_call()?;
4937 let room = call.read(cx).room()?.read(cx);
4938 let participant = room.remote_participant_for_peer_id(peer_id)?;
4939 let leader_in_this_app;
4940 let leader_in_this_project;
4941 match participant.location {
4942 call::ParticipantLocation::SharedProject { project_id } => {
4943 leader_in_this_app = true;
4944 leader_in_this_project = Some(project_id) == self.project.read(cx).remote_id();
4945 }
4946 call::ParticipantLocation::UnsharedProject => {
4947 leader_in_this_app = true;
4948 leader_in_this_project = false;
4949 }
4950 call::ParticipantLocation::External => {
4951 leader_in_this_app = false;
4952 leader_in_this_project = false;
4953 }
4954 };
4955 let state = self.follower_states.get(&peer_id.into())?;
4956 let mut item_to_activate = None;
4957 if let (Some(active_view_id), true) = (state.active_view_id, leader_in_this_app) {
4958 if let Some(item) = state.items_by_leader_view_id.get(&active_view_id)
4959 && (leader_in_this_project || !item.view.is_project_item(window, cx))
4960 {
4961 item_to_activate = Some((item.location, item.view.boxed_clone()));
4962 }
4963 } else if let Some(shared_screen) =
4964 self.shared_screen_for_peer(peer_id, &state.center_pane, window, cx)
4965 {
4966 item_to_activate = Some((None, Box::new(shared_screen)));
4967 }
4968 item_to_activate
4969 }
4970
4971 fn shared_screen_for_peer(
4972 &self,
4973 peer_id: PeerId,
4974 pane: &Entity<Pane>,
4975 window: &mut Window,
4976 cx: &mut App,
4977 ) -> Option<Entity<SharedScreen>> {
4978 let call = self.active_call()?;
4979 let room = call.read(cx).room()?.clone();
4980 let participant = room.read(cx).remote_participant_for_peer_id(peer_id)?;
4981 let track = participant.video_tracks.values().next()?.clone();
4982 let user = participant.user.clone();
4983
4984 for item in pane.read(cx).items_of_type::<SharedScreen>() {
4985 if item.read(cx).peer_id == peer_id {
4986 return Some(item);
4987 }
4988 }
4989
4990 Some(cx.new(|cx| SharedScreen::new(track, peer_id, user.clone(), room.clone(), window, cx)))
4991 }
4992
4993 pub fn on_window_activation_changed(&mut self, window: &mut Window, cx: &mut Context<Self>) {
4994 if window.is_window_active() {
4995 self.update_active_view_for_followers(window, cx);
4996
4997 if let Some(database_id) = self.database_id {
4998 cx.background_spawn(persistence::DB.update_timestamp(database_id))
4999 .detach();
5000 }
5001 } else {
5002 for pane in &self.panes {
5003 pane.update(cx, |pane, cx| {
5004 if let Some(item) = pane.active_item() {
5005 item.workspace_deactivated(window, cx);
5006 }
5007 for item in pane.items() {
5008 if matches!(
5009 item.workspace_settings(cx).autosave,
5010 AutosaveSetting::OnWindowChange | AutosaveSetting::OnFocusChange
5011 ) {
5012 Pane::autosave_item(item.as_ref(), self.project.clone(), window, cx)
5013 .detach_and_log_err(cx);
5014 }
5015 }
5016 });
5017 }
5018 }
5019 }
5020
5021 pub fn active_call(&self) -> Option<&Entity<ActiveCall>> {
5022 self.active_call.as_ref().map(|(call, _)| call)
5023 }
5024
5025 fn on_active_call_event(
5026 &mut self,
5027 _: &Entity<ActiveCall>,
5028 event: &call::room::Event,
5029 window: &mut Window,
5030 cx: &mut Context<Self>,
5031 ) {
5032 match event {
5033 call::room::Event::ParticipantLocationChanged { participant_id }
5034 | call::room::Event::RemoteVideoTracksChanged { participant_id } => {
5035 self.leader_updated(participant_id, window, cx);
5036 }
5037 _ => {}
5038 }
5039 }
5040
5041 pub fn database_id(&self) -> Option<WorkspaceId> {
5042 self.database_id
5043 }
5044
5045 pub fn session_id(&self) -> Option<String> {
5046 self.session_id.clone()
5047 }
5048
5049 fn local_paths(&self, cx: &App) -> Option<Vec<Arc<Path>>> {
5050 let project = self.project().read(cx);
5051
5052 if project.is_local() {
5053 Some(
5054 project
5055 .visible_worktrees(cx)
5056 .map(|worktree| worktree.read(cx).abs_path())
5057 .collect::<Vec<_>>(),
5058 )
5059 } else {
5060 None
5061 }
5062 }
5063
5064 fn update_ssh_paths(&mut self, cx: &App) {
5065 let project = self.project().read(cx);
5066 if !project.is_local() {
5067 let paths: Vec<String> = project
5068 .visible_worktrees(cx)
5069 .map(|worktree| worktree.read(cx).abs_path().to_string_lossy().to_string())
5070 .collect();
5071 if let Some(ssh_project) = &mut self.serialized_ssh_project {
5072 ssh_project.paths = paths;
5073 }
5074 }
5075 }
5076
5077 fn serialize_ssh_paths(&mut self, window: &mut Window, cx: &mut Context<Workspace>) {
5078 if self._schedule_serialize_ssh_paths.is_none() {
5079 self._schedule_serialize_ssh_paths =
5080 Some(cx.spawn_in(window, async move |this, cx| {
5081 cx.background_executor()
5082 .timer(SERIALIZATION_THROTTLE_TIME)
5083 .await;
5084 this.update_in(cx, |this, window, cx| {
5085 let task = if let Some(ssh_project) = &this.serialized_ssh_project {
5086 let ssh_project_id = ssh_project.id;
5087 let ssh_project_paths = ssh_project.paths.clone();
5088 window.spawn(cx, async move |_| {
5089 persistence::DB
5090 .update_ssh_project_paths(ssh_project_id, ssh_project_paths)
5091 .await
5092 })
5093 } else {
5094 Task::ready(Err(anyhow::anyhow!("No SSH project to serialize")))
5095 };
5096 task.detach();
5097 this._schedule_serialize_ssh_paths.take();
5098 })
5099 .log_err();
5100 }));
5101 }
5102 }
5103
5104 fn remove_panes(&mut self, member: Member, window: &mut Window, cx: &mut Context<Workspace>) {
5105 match member {
5106 Member::Axis(PaneAxis { members, .. }) => {
5107 for child in members.iter() {
5108 self.remove_panes(child.clone(), window, cx)
5109 }
5110 }
5111 Member::Pane(pane) => {
5112 self.force_remove_pane(&pane, &None, window, cx);
5113 }
5114 }
5115 }
5116
5117 fn remove_from_session(&mut self, window: &mut Window, cx: &mut App) -> Task<()> {
5118 self.session_id.take();
5119 self.serialize_workspace_internal(window, cx)
5120 }
5121
5122 fn force_remove_pane(
5123 &mut self,
5124 pane: &Entity<Pane>,
5125 focus_on: &Option<Entity<Pane>>,
5126 window: &mut Window,
5127 cx: &mut Context<Workspace>,
5128 ) {
5129 self.panes.retain(|p| p != pane);
5130 if let Some(focus_on) = focus_on {
5131 focus_on.update(cx, |pane, cx| window.focus(&pane.focus_handle(cx)));
5132 } else if self.active_pane() == pane {
5133 self.panes
5134 .last()
5135 .unwrap()
5136 .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx)));
5137 }
5138 if self.last_active_center_pane == Some(pane.downgrade()) {
5139 self.last_active_center_pane = None;
5140 }
5141 cx.notify();
5142 }
5143
5144 fn serialize_workspace(&mut self, window: &mut Window, cx: &mut Context<Self>) {
5145 if self._schedule_serialize_workspace.is_none() {
5146 self._schedule_serialize_workspace =
5147 Some(cx.spawn_in(window, async move |this, cx| {
5148 cx.background_executor()
5149 .timer(SERIALIZATION_THROTTLE_TIME)
5150 .await;
5151 this.update_in(cx, |this, window, cx| {
5152 this.serialize_workspace_internal(window, cx).detach();
5153 this._schedule_serialize_workspace.take();
5154 })
5155 .log_err();
5156 }));
5157 }
5158 }
5159
5160 fn serialize_workspace_internal(&self, window: &mut Window, cx: &mut App) -> Task<()> {
5161 let Some(database_id) = self.database_id() else {
5162 return Task::ready(());
5163 };
5164
5165 fn serialize_pane_handle(
5166 pane_handle: &Entity<Pane>,
5167 window: &mut Window,
5168 cx: &mut App,
5169 ) -> SerializedPane {
5170 let (items, active, pinned_count) = {
5171 let pane = pane_handle.read(cx);
5172 let active_item_id = pane.active_item().map(|item| item.item_id());
5173 (
5174 pane.items()
5175 .filter_map(|handle| {
5176 let handle = handle.to_serializable_item_handle(cx)?;
5177
5178 Some(SerializedItem {
5179 kind: Arc::from(handle.serialized_item_kind()),
5180 item_id: handle.item_id().as_u64(),
5181 active: Some(handle.item_id()) == active_item_id,
5182 preview: pane.is_active_preview_item(handle.item_id()),
5183 })
5184 })
5185 .collect::<Vec<_>>(),
5186 pane.has_focus(window, cx),
5187 pane.pinned_count(),
5188 )
5189 };
5190
5191 SerializedPane::new(items, active, pinned_count)
5192 }
5193
5194 fn build_serialized_pane_group(
5195 pane_group: &Member,
5196 window: &mut Window,
5197 cx: &mut App,
5198 ) -> SerializedPaneGroup {
5199 match pane_group {
5200 Member::Axis(PaneAxis {
5201 axis,
5202 members,
5203 flexes,
5204 bounding_boxes: _,
5205 }) => SerializedPaneGroup::Group {
5206 axis: SerializedAxis(*axis),
5207 children: members
5208 .iter()
5209 .map(|member| build_serialized_pane_group(member, window, cx))
5210 .collect::<Vec<_>>(),
5211 flexes: Some(flexes.lock().clone()),
5212 },
5213 Member::Pane(pane_handle) => {
5214 SerializedPaneGroup::Pane(serialize_pane_handle(pane_handle, window, cx))
5215 }
5216 }
5217 }
5218
5219 fn build_serialized_docks(
5220 this: &Workspace,
5221 window: &mut Window,
5222 cx: &mut App,
5223 ) -> DockStructure {
5224 let left_dock = this.left_dock.read(cx);
5225 let left_visible = left_dock.is_open();
5226 let left_active_panel = left_dock
5227 .active_panel()
5228 .map(|panel| panel.persistent_name().to_string());
5229 let left_dock_zoom = left_dock
5230 .active_panel()
5231 .map(|panel| panel.is_zoomed(window, cx))
5232 .unwrap_or(false);
5233
5234 let right_dock = this.right_dock.read(cx);
5235 let right_visible = right_dock.is_open();
5236 let right_active_panel = right_dock
5237 .active_panel()
5238 .map(|panel| panel.persistent_name().to_string());
5239 let right_dock_zoom = right_dock
5240 .active_panel()
5241 .map(|panel| panel.is_zoomed(window, cx))
5242 .unwrap_or(false);
5243
5244 let bottom_dock = this.bottom_dock.read(cx);
5245 let bottom_visible = bottom_dock.is_open();
5246 let bottom_active_panel = bottom_dock
5247 .active_panel()
5248 .map(|panel| panel.persistent_name().to_string());
5249 let bottom_dock_zoom = bottom_dock
5250 .active_panel()
5251 .map(|panel| panel.is_zoomed(window, cx))
5252 .unwrap_or(false);
5253
5254 DockStructure {
5255 left: DockData {
5256 visible: left_visible,
5257 active_panel: left_active_panel,
5258 zoom: left_dock_zoom,
5259 },
5260 right: DockData {
5261 visible: right_visible,
5262 active_panel: right_active_panel,
5263 zoom: right_dock_zoom,
5264 },
5265 bottom: DockData {
5266 visible: bottom_visible,
5267 active_panel: bottom_active_panel,
5268 zoom: bottom_dock_zoom,
5269 },
5270 }
5271 }
5272
5273 match self.serialize_workspace_location(cx) {
5274 WorkspaceLocation::Location(location) => {
5275 let breakpoints = self.project.update(cx, |project, cx| {
5276 project
5277 .breakpoint_store()
5278 .read(cx)
5279 .all_source_breakpoints(cx)
5280 });
5281
5282 let center_group = build_serialized_pane_group(&self.center.root, window, cx);
5283 let docks = build_serialized_docks(self, window, cx);
5284 let window_bounds = Some(SerializedWindowBounds(window.window_bounds()));
5285 let serialized_workspace = SerializedWorkspace {
5286 id: database_id,
5287 location,
5288 center_group,
5289 window_bounds,
5290 display: Default::default(),
5291 docks,
5292 centered_layout: self.centered_layout,
5293 session_id: self.session_id.clone(),
5294 breakpoints,
5295 window_id: Some(window.window_handle().window_id().as_u64()),
5296 };
5297
5298 window.spawn(cx, async move |_| {
5299 persistence::DB.save_workspace(serialized_workspace).await;
5300 })
5301 }
5302 WorkspaceLocation::DetachFromSession => window.spawn(cx, async move |_| {
5303 persistence::DB
5304 .set_session_id(database_id, None)
5305 .await
5306 .log_err();
5307 }),
5308 WorkspaceLocation::None => Task::ready(()),
5309 }
5310 }
5311
5312 fn serialize_workspace_location(&self, cx: &App) -> WorkspaceLocation {
5313 if let Some(ssh_project) = &self.serialized_ssh_project {
5314 WorkspaceLocation::Location(SerializedWorkspaceLocation::Ssh(ssh_project.clone()))
5315 } else if let Some(local_paths) = self.local_paths(cx) {
5316 if !local_paths.is_empty() {
5317 WorkspaceLocation::Location(SerializedWorkspaceLocation::from_local_paths(
5318 local_paths,
5319 ))
5320 } else {
5321 WorkspaceLocation::DetachFromSession
5322 }
5323 } else {
5324 WorkspaceLocation::None
5325 }
5326 }
5327
5328 fn update_history(&self, cx: &mut App) {
5329 let Some(id) = self.database_id() else {
5330 return;
5331 };
5332 let location = match self.serialize_workspace_location(cx) {
5333 WorkspaceLocation::Location(location) => location,
5334 _ => return,
5335 };
5336 if let Some(manager) = HistoryManager::global(cx) {
5337 manager.update(cx, |this, cx| {
5338 this.update_history(id, HistoryManagerEntry::new(id, &location), cx);
5339 });
5340 }
5341 }
5342
5343 async fn serialize_items(
5344 this: &WeakEntity<Self>,
5345 items_rx: UnboundedReceiver<Box<dyn SerializableItemHandle>>,
5346 cx: &mut AsyncWindowContext,
5347 ) -> Result<()> {
5348 const CHUNK_SIZE: usize = 200;
5349
5350 let mut serializable_items = items_rx.ready_chunks(CHUNK_SIZE);
5351
5352 while let Some(items_received) = serializable_items.next().await {
5353 let unique_items =
5354 items_received
5355 .into_iter()
5356 .fold(HashMap::default(), |mut acc, item| {
5357 acc.entry(item.item_id()).or_insert(item);
5358 acc
5359 });
5360
5361 // We use into_iter() here so that the references to the items are moved into
5362 // the tasks and not kept alive while we're sleeping.
5363 for (_, item) in unique_items.into_iter() {
5364 if let Ok(Some(task)) = this.update_in(cx, |workspace, window, cx| {
5365 item.serialize(workspace, false, window, cx)
5366 }) {
5367 cx.background_spawn(async move { task.await.log_err() })
5368 .detach();
5369 }
5370 }
5371
5372 cx.background_executor()
5373 .timer(SERIALIZATION_THROTTLE_TIME)
5374 .await;
5375 }
5376
5377 Ok(())
5378 }
5379
5380 pub(crate) fn enqueue_item_serialization(
5381 &mut self,
5382 item: Box<dyn SerializableItemHandle>,
5383 ) -> Result<()> {
5384 self.serializable_items_tx
5385 .unbounded_send(item)
5386 .map_err(|err| anyhow!("failed to send serializable item over channel: {err}"))
5387 }
5388
5389 pub(crate) fn load_workspace(
5390 serialized_workspace: SerializedWorkspace,
5391 paths_to_open: Vec<Option<ProjectPath>>,
5392 window: &mut Window,
5393 cx: &mut Context<Workspace>,
5394 ) -> Task<Result<Vec<Option<Box<dyn ItemHandle>>>>> {
5395 cx.spawn_in(window, async move |workspace, cx| {
5396 let project = workspace.read_with(cx, |workspace, _| workspace.project().clone())?;
5397
5398 let mut center_group = None;
5399 let mut center_items = None;
5400
5401 // Traverse the splits tree and add to things
5402 if let Some((group, active_pane, items)) = serialized_workspace
5403 .center_group
5404 .deserialize(&project, serialized_workspace.id, workspace.clone(), cx)
5405 .await
5406 {
5407 center_items = Some(items);
5408 center_group = Some((group, active_pane))
5409 }
5410
5411 let mut items_by_project_path = HashMap::default();
5412 let mut item_ids_by_kind = HashMap::default();
5413 let mut all_deserialized_items = Vec::default();
5414 cx.update(|_, cx| {
5415 for item in center_items.unwrap_or_default().into_iter().flatten() {
5416 if let Some(serializable_item_handle) = item.to_serializable_item_handle(cx) {
5417 item_ids_by_kind
5418 .entry(serializable_item_handle.serialized_item_kind())
5419 .or_insert(Vec::new())
5420 .push(item.item_id().as_u64() as ItemId);
5421 }
5422
5423 if let Some(project_path) = item.project_path(cx) {
5424 items_by_project_path.insert(project_path, item.clone());
5425 }
5426 all_deserialized_items.push(item);
5427 }
5428 })?;
5429
5430 let opened_items = paths_to_open
5431 .into_iter()
5432 .map(|path_to_open| {
5433 path_to_open
5434 .and_then(|path_to_open| items_by_project_path.remove(&path_to_open))
5435 })
5436 .collect::<Vec<_>>();
5437
5438 // Remove old panes from workspace panes list
5439 workspace.update_in(cx, |workspace, window, cx| {
5440 if let Some((center_group, active_pane)) = center_group {
5441 workspace.remove_panes(workspace.center.root.clone(), window, cx);
5442
5443 // Swap workspace center group
5444 workspace.center = PaneGroup::with_root(center_group);
5445 if let Some(active_pane) = active_pane {
5446 workspace.set_active_pane(&active_pane, window, cx);
5447 cx.focus_self(window);
5448 } else {
5449 workspace.set_active_pane(&workspace.center.first_pane(), window, cx);
5450 }
5451 }
5452
5453 let docks = serialized_workspace.docks;
5454
5455 for (dock, serialized_dock) in [
5456 (&mut workspace.right_dock, docks.right),
5457 (&mut workspace.left_dock, docks.left),
5458 (&mut workspace.bottom_dock, docks.bottom),
5459 ]
5460 .iter_mut()
5461 {
5462 dock.update(cx, |dock, cx| {
5463 dock.serialized_dock = Some(serialized_dock.clone());
5464 dock.restore_state(window, cx);
5465 });
5466 }
5467
5468 cx.notify();
5469 })?;
5470
5471 let _ = project
5472 .update(cx, |project, cx| {
5473 project
5474 .breakpoint_store()
5475 .update(cx, |breakpoint_store, cx| {
5476 breakpoint_store
5477 .with_serialized_breakpoints(serialized_workspace.breakpoints, cx)
5478 })
5479 })?
5480 .await;
5481
5482 // Clean up all the items that have _not_ been loaded. Our ItemIds aren't stable. That means
5483 // after loading the items, we might have different items and in order to avoid
5484 // the database filling up, we delete items that haven't been loaded now.
5485 //
5486 // The items that have been loaded, have been saved after they've been added to the workspace.
5487 let clean_up_tasks = workspace.update_in(cx, |_, window, cx| {
5488 item_ids_by_kind
5489 .into_iter()
5490 .map(|(item_kind, loaded_items)| {
5491 SerializableItemRegistry::cleanup(
5492 item_kind,
5493 serialized_workspace.id,
5494 loaded_items,
5495 window,
5496 cx,
5497 )
5498 .log_err()
5499 })
5500 .collect::<Vec<_>>()
5501 })?;
5502
5503 futures::future::join_all(clean_up_tasks).await;
5504
5505 workspace
5506 .update_in(cx, |workspace, window, cx| {
5507 // Serialize ourself to make sure our timestamps and any pane / item changes are replicated
5508 workspace.serialize_workspace_internal(window, cx).detach();
5509
5510 // Ensure that we mark the window as edited if we did load dirty items
5511 workspace.update_window_edited(window, cx);
5512 })
5513 .ok();
5514
5515 Ok(opened_items)
5516 })
5517 }
5518
5519 fn actions(&self, div: Div, window: &mut Window, cx: &mut Context<Self>) -> Div {
5520 self.add_workspace_actions_listeners(div, window, cx)
5521 .on_action(cx.listener(Self::close_inactive_items_and_panes))
5522 .on_action(cx.listener(Self::close_all_items_and_panes))
5523 .on_action(cx.listener(Self::save_all))
5524 .on_action(cx.listener(Self::send_keystrokes))
5525 .on_action(cx.listener(Self::add_folder_to_project))
5526 .on_action(cx.listener(Self::follow_next_collaborator))
5527 .on_action(cx.listener(Self::close_window))
5528 .on_action(cx.listener(Self::activate_pane_at_index))
5529 .on_action(cx.listener(Self::move_item_to_pane_at_index))
5530 .on_action(cx.listener(Self::move_focused_panel_to_next_position))
5531 .on_action(cx.listener(Self::toggle_edit_predictions_all_files))
5532 .on_action(cx.listener(|workspace, _: &Unfollow, window, cx| {
5533 let pane = workspace.active_pane().clone();
5534 workspace.unfollow_in_pane(&pane, window, cx);
5535 }))
5536 .on_action(cx.listener(|workspace, action: &Save, window, cx| {
5537 workspace
5538 .save_active_item(action.save_intent.unwrap_or(SaveIntent::Save), window, cx)
5539 .detach_and_prompt_err("Failed to save", window, cx, |_, _, _| None);
5540 }))
5541 .on_action(cx.listener(|workspace, _: &SaveWithoutFormat, window, cx| {
5542 workspace
5543 .save_active_item(SaveIntent::SaveWithoutFormat, window, cx)
5544 .detach_and_prompt_err("Failed to save", window, cx, |_, _, _| None);
5545 }))
5546 .on_action(cx.listener(|workspace, _: &SaveAs, window, cx| {
5547 workspace
5548 .save_active_item(SaveIntent::SaveAs, window, cx)
5549 .detach_and_prompt_err("Failed to save", window, cx, |_, _, _| None);
5550 }))
5551 .on_action(
5552 cx.listener(|workspace, _: &ActivatePreviousPane, window, cx| {
5553 workspace.activate_previous_pane(window, cx)
5554 }),
5555 )
5556 .on_action(cx.listener(|workspace, _: &ActivateNextPane, window, cx| {
5557 workspace.activate_next_pane(window, cx)
5558 }))
5559 .on_action(
5560 cx.listener(|workspace, _: &ActivateNextWindow, _window, cx| {
5561 workspace.activate_next_window(cx)
5562 }),
5563 )
5564 .on_action(
5565 cx.listener(|workspace, _: &ActivatePreviousWindow, _window, cx| {
5566 workspace.activate_previous_window(cx)
5567 }),
5568 )
5569 .on_action(cx.listener(|workspace, _: &ActivatePaneLeft, window, cx| {
5570 workspace.activate_pane_in_direction(SplitDirection::Left, window, cx)
5571 }))
5572 .on_action(cx.listener(|workspace, _: &ActivatePaneRight, window, cx| {
5573 workspace.activate_pane_in_direction(SplitDirection::Right, window, cx)
5574 }))
5575 .on_action(cx.listener(|workspace, _: &ActivatePaneUp, window, cx| {
5576 workspace.activate_pane_in_direction(SplitDirection::Up, window, cx)
5577 }))
5578 .on_action(cx.listener(|workspace, _: &ActivatePaneDown, window, cx| {
5579 workspace.activate_pane_in_direction(SplitDirection::Down, window, cx)
5580 }))
5581 .on_action(cx.listener(|workspace, _: &ActivateNextPane, window, cx| {
5582 workspace.activate_next_pane(window, cx)
5583 }))
5584 .on_action(cx.listener(
5585 |workspace, action: &MoveItemToPaneInDirection, window, cx| {
5586 workspace.move_item_to_pane_in_direction(action, window, cx)
5587 },
5588 ))
5589 .on_action(cx.listener(|workspace, _: &SwapPaneLeft, _, cx| {
5590 workspace.swap_pane_in_direction(SplitDirection::Left, cx)
5591 }))
5592 .on_action(cx.listener(|workspace, _: &SwapPaneRight, _, cx| {
5593 workspace.swap_pane_in_direction(SplitDirection::Right, cx)
5594 }))
5595 .on_action(cx.listener(|workspace, _: &SwapPaneUp, _, cx| {
5596 workspace.swap_pane_in_direction(SplitDirection::Up, cx)
5597 }))
5598 .on_action(cx.listener(|workspace, _: &SwapPaneDown, _, cx| {
5599 workspace.swap_pane_in_direction(SplitDirection::Down, cx)
5600 }))
5601 .on_action(cx.listener(|this, _: &ToggleLeftDock, window, cx| {
5602 this.toggle_dock(DockPosition::Left, window, cx);
5603 }))
5604 .on_action(cx.listener(
5605 |workspace: &mut Workspace, _: &ToggleRightDock, window, cx| {
5606 workspace.toggle_dock(DockPosition::Right, window, cx);
5607 },
5608 ))
5609 .on_action(cx.listener(
5610 |workspace: &mut Workspace, _: &ToggleBottomDock, window, cx| {
5611 workspace.toggle_dock(DockPosition::Bottom, window, cx);
5612 },
5613 ))
5614 .on_action(cx.listener(
5615 |workspace: &mut Workspace, _: &CloseActiveDock, window, cx| {
5616 if !workspace.close_active_dock(window, cx) {
5617 cx.propagate();
5618 }
5619 },
5620 ))
5621 .on_action(
5622 cx.listener(|workspace: &mut Workspace, _: &CloseAllDocks, window, cx| {
5623 workspace.close_all_docks(window, cx);
5624 }),
5625 )
5626 .on_action(cx.listener(
5627 |workspace: &mut Workspace, _: &ClearAllNotifications, _, cx| {
5628 workspace.clear_all_notifications(cx);
5629 },
5630 ))
5631 .on_action(cx.listener(
5632 |workspace: &mut Workspace, _: &SuppressNotification, _, cx| {
5633 if let Some((notification_id, _)) = workspace.notifications.pop() {
5634 workspace.suppress_notification(¬ification_id, cx);
5635 }
5636 },
5637 ))
5638 .on_action(cx.listener(
5639 |workspace: &mut Workspace, _: &ReopenClosedItem, window, cx| {
5640 workspace.reopen_closed_item(window, cx).detach();
5641 },
5642 ))
5643 .on_action(cx.listener(
5644 |workspace: &mut Workspace, _: &ResetActiveDockSize, window, cx| {
5645 for dock in workspace.all_docks() {
5646 if dock.focus_handle(cx).contains_focused(window, cx) {
5647 let Some(panel) = dock.read(cx).active_panel() else {
5648 return;
5649 };
5650
5651 // Set to `None`, then the size will fall back to the default.
5652 panel.clone().set_size(None, window, cx);
5653
5654 return;
5655 }
5656 }
5657 },
5658 ))
5659 .on_action(cx.listener(
5660 |workspace: &mut Workspace, _: &ResetOpenDocksSize, window, cx| {
5661 for dock in workspace.all_docks() {
5662 if let Some(panel) = dock.read(cx).visible_panel() {
5663 // Set to `None`, then the size will fall back to the default.
5664 panel.clone().set_size(None, window, cx);
5665 }
5666 }
5667 },
5668 ))
5669 .on_action(cx.listener(
5670 |workspace: &mut Workspace, act: &IncreaseActiveDockSize, window, cx| {
5671 adjust_active_dock_size_by_px(
5672 px_with_ui_font_fallback(act.px, cx),
5673 workspace,
5674 window,
5675 cx,
5676 );
5677 },
5678 ))
5679 .on_action(cx.listener(
5680 |workspace: &mut Workspace, act: &DecreaseActiveDockSize, window, cx| {
5681 adjust_active_dock_size_by_px(
5682 px_with_ui_font_fallback(act.px, cx) * -1.,
5683 workspace,
5684 window,
5685 cx,
5686 );
5687 },
5688 ))
5689 .on_action(cx.listener(
5690 |workspace: &mut Workspace, act: &IncreaseOpenDocksSize, window, cx| {
5691 adjust_open_docks_size_by_px(
5692 px_with_ui_font_fallback(act.px, cx),
5693 workspace,
5694 window,
5695 cx,
5696 );
5697 },
5698 ))
5699 .on_action(cx.listener(
5700 |workspace: &mut Workspace, act: &DecreaseOpenDocksSize, window, cx| {
5701 adjust_open_docks_size_by_px(
5702 px_with_ui_font_fallback(act.px, cx) * -1.,
5703 workspace,
5704 window,
5705 cx,
5706 );
5707 },
5708 ))
5709 .on_action(cx.listener(Workspace::toggle_centered_layout))
5710 .on_action(cx.listener(Workspace::cancel))
5711 }
5712
5713 #[cfg(any(test, feature = "test-support"))]
5714 pub fn test_new(project: Entity<Project>, window: &mut Window, cx: &mut Context<Self>) -> Self {
5715 use node_runtime::NodeRuntime;
5716 use session::Session;
5717
5718 let client = project.read(cx).client();
5719 let user_store = project.read(cx).user_store();
5720 let workspace_store = cx.new(|cx| WorkspaceStore::new(client.clone(), cx));
5721 let session = cx.new(|cx| AppSession::new(Session::test(), cx));
5722 window.activate_window();
5723 let app_state = Arc::new(AppState {
5724 languages: project.read(cx).languages().clone(),
5725 workspace_store,
5726 client,
5727 user_store,
5728 fs: project.read(cx).fs().clone(),
5729 build_window_options: |_, _| Default::default(),
5730 node_runtime: NodeRuntime::unavailable(),
5731 session,
5732 });
5733 let workspace = Self::new(Default::default(), project, app_state, window, cx);
5734 workspace
5735 .active_pane
5736 .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx)));
5737 workspace
5738 }
5739
5740 pub fn register_action<A: Action>(
5741 &mut self,
5742 callback: impl Fn(&mut Self, &A, &mut Window, &mut Context<Self>) + 'static,
5743 ) -> &mut Self {
5744 let callback = Arc::new(callback);
5745
5746 self.workspace_actions.push(Box::new(move |div, _, _, cx| {
5747 let callback = callback.clone();
5748 div.on_action(cx.listener(move |workspace, event, window, cx| {
5749 (callback)(workspace, event, window, cx)
5750 }))
5751 }));
5752 self
5753 }
5754 pub fn register_action_renderer(
5755 &mut self,
5756 callback: impl Fn(Div, &Workspace, &mut Window, &mut Context<Self>) -> Div + 'static,
5757 ) -> &mut Self {
5758 self.workspace_actions.push(Box::new(callback));
5759 self
5760 }
5761
5762 fn add_workspace_actions_listeners(
5763 &self,
5764 mut div: Div,
5765 window: &mut Window,
5766 cx: &mut Context<Self>,
5767 ) -> Div {
5768 for action in self.workspace_actions.iter() {
5769 div = (action)(div, self, window, cx)
5770 }
5771 div
5772 }
5773
5774 pub fn has_active_modal(&self, _: &mut Window, cx: &mut App) -> bool {
5775 self.modal_layer.read(cx).has_active_modal()
5776 }
5777
5778 pub fn active_modal<V: ManagedView + 'static>(&self, cx: &App) -> Option<Entity<V>> {
5779 self.modal_layer.read(cx).active_modal()
5780 }
5781
5782 pub fn toggle_modal<V: ModalView, B>(&mut self, window: &mut Window, cx: &mut App, build: B)
5783 where
5784 B: FnOnce(&mut Window, &mut Context<V>) -> V,
5785 {
5786 self.modal_layer.update(cx, |modal_layer, cx| {
5787 modal_layer.toggle_modal(window, cx, build)
5788 })
5789 }
5790
5791 pub fn toggle_status_toast<V: ToastView>(&mut self, entity: Entity<V>, cx: &mut App) {
5792 self.toast_layer
5793 .update(cx, |toast_layer, cx| toast_layer.toggle_toast(cx, entity))
5794 }
5795
5796 pub fn toggle_centered_layout(
5797 &mut self,
5798 _: &ToggleCenteredLayout,
5799 _: &mut Window,
5800 cx: &mut Context<Self>,
5801 ) {
5802 self.centered_layout = !self.centered_layout;
5803 if let Some(database_id) = self.database_id() {
5804 cx.background_spawn(DB.set_centered_layout(database_id, self.centered_layout))
5805 .detach_and_log_err(cx);
5806 }
5807 cx.notify();
5808 }
5809
5810 fn adjust_padding(padding: Option<f32>) -> f32 {
5811 padding
5812 .unwrap_or(Self::DEFAULT_PADDING)
5813 .clamp(0.0, Self::MAX_PADDING)
5814 }
5815
5816 fn render_dock(
5817 &self,
5818 position: DockPosition,
5819 dock: &Entity<Dock>,
5820 window: &mut Window,
5821 cx: &mut App,
5822 ) -> Option<Div> {
5823 if self.zoomed_position == Some(position) {
5824 return None;
5825 }
5826
5827 let leader_border = dock.read(cx).active_panel().and_then(|panel| {
5828 let pane = panel.pane(cx)?;
5829 let follower_states = &self.follower_states;
5830 leader_border_for_pane(follower_states, &pane, window, cx)
5831 });
5832
5833 Some(
5834 div()
5835 .flex()
5836 .flex_none()
5837 .overflow_hidden()
5838 .child(dock.clone())
5839 .children(leader_border),
5840 )
5841 }
5842
5843 pub fn for_window(window: &mut Window, _: &mut App) -> Option<Entity<Workspace>> {
5844 window.root().flatten()
5845 }
5846
5847 pub fn zoomed_item(&self) -> Option<&AnyWeakView> {
5848 self.zoomed.as_ref()
5849 }
5850
5851 pub fn activate_next_window(&mut self, cx: &mut Context<Self>) {
5852 let Some(current_window_id) = cx.active_window().map(|a| a.window_id()) else {
5853 return;
5854 };
5855 let windows = cx.windows();
5856 let Some(next_window) = windows
5857 .iter()
5858 .cycle()
5859 .skip_while(|window| window.window_id() != current_window_id)
5860 .nth(1)
5861 else {
5862 return;
5863 };
5864 next_window
5865 .update(cx, |_, window, _| window.activate_window())
5866 .ok();
5867 }
5868
5869 pub fn activate_previous_window(&mut self, cx: &mut Context<Self>) {
5870 let Some(current_window_id) = cx.active_window().map(|a| a.window_id()) else {
5871 return;
5872 };
5873 let windows = cx.windows();
5874 let Some(prev_window) = windows
5875 .iter()
5876 .rev()
5877 .cycle()
5878 .skip_while(|window| window.window_id() != current_window_id)
5879 .nth(1)
5880 else {
5881 return;
5882 };
5883 prev_window
5884 .update(cx, |_, window, _| window.activate_window())
5885 .ok();
5886 }
5887
5888 pub fn cancel(&mut self, _: &menu::Cancel, window: &mut Window, cx: &mut Context<Self>) {
5889 if cx.stop_active_drag(window) {
5890 } else if let Some((notification_id, _)) = self.notifications.pop() {
5891 dismiss_app_notification(¬ification_id, cx);
5892 } else {
5893 cx.propagate();
5894 }
5895 }
5896
5897 fn adjust_dock_size_by_px(
5898 &mut self,
5899 panel_size: Pixels,
5900 dock_pos: DockPosition,
5901 px: Pixels,
5902 window: &mut Window,
5903 cx: &mut Context<Self>,
5904 ) {
5905 match dock_pos {
5906 DockPosition::Left => self.resize_left_dock(panel_size + px, window, cx),
5907 DockPosition::Right => self.resize_right_dock(panel_size + px, window, cx),
5908 DockPosition::Bottom => self.resize_bottom_dock(panel_size + px, window, cx),
5909 }
5910 }
5911
5912 fn resize_left_dock(&mut self, new_size: Pixels, window: &mut Window, cx: &mut App) {
5913 let size = new_size.min(self.bounds.right() - RESIZE_HANDLE_SIZE);
5914
5915 self.left_dock.update(cx, |left_dock, cx| {
5916 if WorkspaceSettings::get_global(cx)
5917 .resize_all_panels_in_dock
5918 .contains(&DockPosition::Left)
5919 {
5920 left_dock.resize_all_panels(Some(size), window, cx);
5921 } else {
5922 left_dock.resize_active_panel(Some(size), window, cx);
5923 }
5924 });
5925 }
5926
5927 fn resize_right_dock(&mut self, new_size: Pixels, window: &mut Window, cx: &mut App) {
5928 let mut size = new_size.max(self.bounds.left() - RESIZE_HANDLE_SIZE);
5929 self.left_dock.read_with(cx, |left_dock, cx| {
5930 let left_dock_size = left_dock
5931 .active_panel_size(window, cx)
5932 .unwrap_or(Pixels(0.0));
5933 if left_dock_size + size > self.bounds.right() {
5934 size = self.bounds.right() - left_dock_size
5935 }
5936 });
5937 self.right_dock.update(cx, |right_dock, cx| {
5938 if WorkspaceSettings::get_global(cx)
5939 .resize_all_panels_in_dock
5940 .contains(&DockPosition::Right)
5941 {
5942 right_dock.resize_all_panels(Some(size), window, cx);
5943 } else {
5944 right_dock.resize_active_panel(Some(size), window, cx);
5945 }
5946 });
5947 }
5948
5949 fn resize_bottom_dock(&mut self, new_size: Pixels, window: &mut Window, cx: &mut App) {
5950 let size = new_size.min(self.bounds.bottom() - RESIZE_HANDLE_SIZE - self.bounds.top());
5951 self.bottom_dock.update(cx, |bottom_dock, cx| {
5952 if WorkspaceSettings::get_global(cx)
5953 .resize_all_panels_in_dock
5954 .contains(&DockPosition::Bottom)
5955 {
5956 bottom_dock.resize_all_panels(Some(size), window, cx);
5957 } else {
5958 bottom_dock.resize_active_panel(Some(size), window, cx);
5959 }
5960 });
5961 }
5962
5963 fn toggle_edit_predictions_all_files(
5964 &mut self,
5965 _: &ToggleEditPrediction,
5966 _window: &mut Window,
5967 cx: &mut Context<Self>,
5968 ) {
5969 let fs = self.project().read(cx).fs().clone();
5970 let show_edit_predictions = all_language_settings(None, cx).show_edit_predictions(None, cx);
5971 update_settings_file::<AllLanguageSettings>(fs, cx, move |file, _| {
5972 file.defaults.show_edit_predictions = Some(!show_edit_predictions)
5973 });
5974 }
5975}
5976
5977fn leader_border_for_pane(
5978 follower_states: &HashMap<CollaboratorId, FollowerState>,
5979 pane: &Entity<Pane>,
5980 _: &Window,
5981 cx: &App,
5982) -> Option<Div> {
5983 let (leader_id, _follower_state) = follower_states.iter().find_map(|(leader_id, state)| {
5984 if state.pane() == pane {
5985 Some((*leader_id, state))
5986 } else {
5987 None
5988 }
5989 })?;
5990
5991 let mut leader_color = match leader_id {
5992 CollaboratorId::PeerId(leader_peer_id) => {
5993 let room = ActiveCall::try_global(cx)?.read(cx).room()?.read(cx);
5994 let leader = room.remote_participant_for_peer_id(leader_peer_id)?;
5995
5996 cx.theme()
5997 .players()
5998 .color_for_participant(leader.participant_index.0)
5999 .cursor
6000 }
6001 CollaboratorId::Agent => cx.theme().players().agent().cursor,
6002 };
6003 leader_color.fade_out(0.3);
6004 Some(
6005 div()
6006 .absolute()
6007 .size_full()
6008 .left_0()
6009 .top_0()
6010 .border_2()
6011 .border_color(leader_color),
6012 )
6013}
6014
6015fn window_bounds_env_override() -> Option<Bounds<Pixels>> {
6016 ZED_WINDOW_POSITION
6017 .zip(*ZED_WINDOW_SIZE)
6018 .map(|(position, size)| Bounds {
6019 origin: position,
6020 size,
6021 })
6022}
6023
6024fn open_items(
6025 serialized_workspace: Option<SerializedWorkspace>,
6026 mut project_paths_to_open: Vec<(PathBuf, Option<ProjectPath>)>,
6027 window: &mut Window,
6028 cx: &mut Context<Workspace>,
6029) -> impl 'static + Future<Output = Result<Vec<Option<Result<Box<dyn ItemHandle>>>>>> + use<> {
6030 let restored_items = serialized_workspace.map(|serialized_workspace| {
6031 Workspace::load_workspace(
6032 serialized_workspace,
6033 project_paths_to_open
6034 .iter()
6035 .map(|(_, project_path)| project_path)
6036 .cloned()
6037 .collect(),
6038 window,
6039 cx,
6040 )
6041 });
6042
6043 cx.spawn_in(window, async move |workspace, cx| {
6044 let mut opened_items = Vec::with_capacity(project_paths_to_open.len());
6045
6046 if let Some(restored_items) = restored_items {
6047 let restored_items = restored_items.await?;
6048
6049 let restored_project_paths = restored_items
6050 .iter()
6051 .filter_map(|item| {
6052 cx.update(|_, cx| item.as_ref()?.project_path(cx))
6053 .ok()
6054 .flatten()
6055 })
6056 .collect::<HashSet<_>>();
6057
6058 for restored_item in restored_items {
6059 opened_items.push(restored_item.map(Ok));
6060 }
6061
6062 project_paths_to_open
6063 .iter_mut()
6064 .for_each(|(_, project_path)| {
6065 if let Some(project_path_to_open) = project_path
6066 && restored_project_paths.contains(project_path_to_open)
6067 {
6068 *project_path = None;
6069 }
6070 });
6071 } else {
6072 for _ in 0..project_paths_to_open.len() {
6073 opened_items.push(None);
6074 }
6075 }
6076 assert!(opened_items.len() == project_paths_to_open.len());
6077
6078 let tasks =
6079 project_paths_to_open
6080 .into_iter()
6081 .enumerate()
6082 .map(|(ix, (abs_path, project_path))| {
6083 let workspace = workspace.clone();
6084 cx.spawn(async move |cx| {
6085 let file_project_path = project_path?;
6086 let abs_path_task = workspace.update(cx, |workspace, cx| {
6087 workspace.project().update(cx, |project, cx| {
6088 project.resolve_abs_path(abs_path.to_string_lossy().as_ref(), cx)
6089 })
6090 });
6091
6092 // We only want to open file paths here. If one of the items
6093 // here is a directory, it was already opened further above
6094 // with a `find_or_create_worktree`.
6095 if let Ok(task) = abs_path_task
6096 && task.await.is_none_or(|p| p.is_file())
6097 {
6098 return Some((
6099 ix,
6100 workspace
6101 .update_in(cx, |workspace, window, cx| {
6102 workspace.open_path(
6103 file_project_path,
6104 None,
6105 true,
6106 window,
6107 cx,
6108 )
6109 })
6110 .log_err()?
6111 .await,
6112 ));
6113 }
6114 None
6115 })
6116 });
6117
6118 let tasks = tasks.collect::<Vec<_>>();
6119
6120 let tasks = futures::future::join_all(tasks);
6121 for (ix, path_open_result) in tasks.await.into_iter().flatten() {
6122 opened_items[ix] = Some(path_open_result);
6123 }
6124
6125 Ok(opened_items)
6126 })
6127}
6128
6129enum ActivateInDirectionTarget {
6130 Pane(Entity<Pane>),
6131 Dock(Entity<Dock>),
6132}
6133
6134fn notify_if_database_failed(workspace: WindowHandle<Workspace>, cx: &mut AsyncApp) {
6135 workspace
6136 .update(cx, |workspace, _, cx| {
6137 if (*db::ALL_FILE_DB_FAILED).load(std::sync::atomic::Ordering::Acquire) {
6138 struct DatabaseFailedNotification;
6139
6140 workspace.show_notification(
6141 NotificationId::unique::<DatabaseFailedNotification>(),
6142 cx,
6143 |cx| {
6144 cx.new(|cx| {
6145 MessageNotification::new("Failed to load the database file.", cx)
6146 .primary_message("File an Issue")
6147 .primary_icon(IconName::Plus)
6148 .primary_on_click(|window, cx| {
6149 window.dispatch_action(Box::new(FileBugReport), cx)
6150 })
6151 })
6152 },
6153 );
6154 }
6155 })
6156 .log_err();
6157}
6158
6159fn px_with_ui_font_fallback(val: u32, cx: &Context<Workspace>) -> Pixels {
6160 if val == 0 {
6161 ThemeSettings::get_global(cx).ui_font_size(cx)
6162 } else {
6163 px(val as f32)
6164 }
6165}
6166
6167fn adjust_active_dock_size_by_px(
6168 px: Pixels,
6169 workspace: &mut Workspace,
6170 window: &mut Window,
6171 cx: &mut Context<Workspace>,
6172) {
6173 let Some(active_dock) = workspace
6174 .all_docks()
6175 .into_iter()
6176 .find(|dock| dock.focus_handle(cx).contains_focused(window, cx))
6177 else {
6178 return;
6179 };
6180 let dock = active_dock.read(cx);
6181 let Some(panel_size) = dock.active_panel_size(window, cx) else {
6182 return;
6183 };
6184 let dock_pos = dock.position();
6185 workspace.adjust_dock_size_by_px(panel_size, dock_pos, px, window, cx);
6186}
6187
6188fn adjust_open_docks_size_by_px(
6189 px: Pixels,
6190 workspace: &mut Workspace,
6191 window: &mut Window,
6192 cx: &mut Context<Workspace>,
6193) {
6194 let docks = workspace
6195 .all_docks()
6196 .into_iter()
6197 .filter_map(|dock| {
6198 if dock.read(cx).is_open() {
6199 let dock = dock.read(cx);
6200 let panel_size = dock.active_panel_size(window, cx)?;
6201 let dock_pos = dock.position();
6202 Some((panel_size, dock_pos, px))
6203 } else {
6204 None
6205 }
6206 })
6207 .collect::<Vec<_>>();
6208
6209 docks
6210 .into_iter()
6211 .for_each(|(panel_size, dock_pos, offset)| {
6212 workspace.adjust_dock_size_by_px(panel_size, dock_pos, offset, window, cx);
6213 });
6214}
6215
6216impl Focusable for Workspace {
6217 fn focus_handle(&self, cx: &App) -> FocusHandle {
6218 self.active_pane.focus_handle(cx)
6219 }
6220}
6221
6222#[derive(Clone)]
6223struct DraggedDock(DockPosition);
6224
6225impl Render for DraggedDock {
6226 fn render(&mut self, _window: &mut Window, _cx: &mut Context<Self>) -> impl IntoElement {
6227 gpui::Empty
6228 }
6229}
6230
6231impl Render for Workspace {
6232 fn render(&mut self, window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
6233 let mut context = KeyContext::new_with_defaults();
6234 context.add("Workspace");
6235 context.set("keyboard_layout", cx.keyboard_layout().name().to_string());
6236 if let Some(status) = self
6237 .debugger_provider
6238 .as_ref()
6239 .and_then(|provider| provider.active_thread_state(cx))
6240 {
6241 match status {
6242 ThreadStatus::Running | ThreadStatus::Stepping => {
6243 context.add("debugger_running");
6244 }
6245 ThreadStatus::Stopped => context.add("debugger_stopped"),
6246 ThreadStatus::Exited | ThreadStatus::Ended => {}
6247 }
6248 }
6249
6250 let centered_layout = self.centered_layout
6251 && self.center.panes().len() == 1
6252 && self.active_item(cx).is_some();
6253 let render_padding = |size| {
6254 (size > 0.0).then(|| {
6255 div()
6256 .h_full()
6257 .w(relative(size))
6258 .bg(cx.theme().colors().editor_background)
6259 .border_color(cx.theme().colors().pane_group_border)
6260 })
6261 };
6262 let paddings = if centered_layout {
6263 let settings = WorkspaceSettings::get_global(cx).centered_layout;
6264 (
6265 render_padding(Self::adjust_padding(settings.left_padding)),
6266 render_padding(Self::adjust_padding(settings.right_padding)),
6267 )
6268 } else {
6269 (None, None)
6270 };
6271 let ui_font = theme::setup_ui_font(window, cx);
6272
6273 let theme = cx.theme().clone();
6274 let colors = theme.colors();
6275 let notification_entities = self
6276 .notifications
6277 .iter()
6278 .map(|(_, notification)| notification.entity_id())
6279 .collect::<Vec<_>>();
6280 let bottom_dock_layout = WorkspaceSettings::get_global(cx).bottom_dock_layout;
6281
6282 client_side_decorations(
6283 self.actions(div(), window, cx)
6284 .key_context(context)
6285 .relative()
6286 .size_full()
6287 .flex()
6288 .flex_col()
6289 .font(ui_font)
6290 .gap_0()
6291 .justify_start()
6292 .items_start()
6293 .text_color(colors.text)
6294 .overflow_hidden()
6295 .children(self.titlebar_item.clone())
6296 .on_modifiers_changed(move |_, _, cx| {
6297 for &id in ¬ification_entities {
6298 cx.notify(id);
6299 }
6300 })
6301 .child(
6302 div()
6303 .size_full()
6304 .relative()
6305 .flex_1()
6306 .flex()
6307 .flex_col()
6308 .child(
6309 div()
6310 .id("workspace")
6311 .bg(colors.background)
6312 .relative()
6313 .flex_1()
6314 .w_full()
6315 .flex()
6316 .flex_col()
6317 .overflow_hidden()
6318 .border_t_1()
6319 .border_b_1()
6320 .border_color(colors.border)
6321 .child({
6322 let this = cx.entity();
6323 canvas(
6324 move |bounds, window, cx| {
6325 this.update(cx, |this, cx| {
6326 let bounds_changed = this.bounds != bounds;
6327 this.bounds = bounds;
6328
6329 if bounds_changed {
6330 this.left_dock.update(cx, |dock, cx| {
6331 dock.clamp_panel_size(
6332 bounds.size.width,
6333 window,
6334 cx,
6335 )
6336 });
6337
6338 this.right_dock.update(cx, |dock, cx| {
6339 dock.clamp_panel_size(
6340 bounds.size.width,
6341 window,
6342 cx,
6343 )
6344 });
6345
6346 this.bottom_dock.update(cx, |dock, cx| {
6347 dock.clamp_panel_size(
6348 bounds.size.height,
6349 window,
6350 cx,
6351 )
6352 });
6353 }
6354 })
6355 },
6356 |_, _, _, _| {},
6357 )
6358 .absolute()
6359 .size_full()
6360 })
6361 .when(self.zoomed.is_none(), |this| {
6362 this.on_drag_move(cx.listener(
6363 move |workspace,
6364 e: &DragMoveEvent<DraggedDock>,
6365 window,
6366 cx| {
6367 if workspace.previous_dock_drag_coordinates
6368 != Some(e.event.position)
6369 {
6370 workspace.previous_dock_drag_coordinates =
6371 Some(e.event.position);
6372 match e.drag(cx).0 {
6373 DockPosition::Left => {
6374 workspace.resize_left_dock(
6375 e.event.position.x
6376 - workspace.bounds.left(),
6377 window,
6378 cx,
6379 );
6380 }
6381 DockPosition::Right => {
6382 workspace.resize_right_dock(
6383 workspace.bounds.right()
6384 - e.event.position.x,
6385 window,
6386 cx,
6387 );
6388 }
6389 DockPosition::Bottom => {
6390 workspace.resize_bottom_dock(
6391 workspace.bounds.bottom()
6392 - e.event.position.y,
6393 window,
6394 cx,
6395 );
6396 }
6397 };
6398 workspace.serialize_workspace(window, cx);
6399 }
6400 },
6401 ))
6402 })
6403 .child({
6404 match bottom_dock_layout {
6405 BottomDockLayout::Full => div()
6406 .flex()
6407 .flex_col()
6408 .h_full()
6409 .child(
6410 div()
6411 .flex()
6412 .flex_row()
6413 .flex_1()
6414 .overflow_hidden()
6415 .children(self.render_dock(
6416 DockPosition::Left,
6417 &self.left_dock,
6418 window,
6419 cx,
6420 ))
6421 .child(
6422 div()
6423 .flex()
6424 .flex_col()
6425 .flex_1()
6426 .overflow_hidden()
6427 .child(
6428 h_flex()
6429 .flex_1()
6430 .when_some(
6431 paddings.0,
6432 |this, p| {
6433 this.child(
6434 p.border_r_1(),
6435 )
6436 },
6437 )
6438 .child(self.center.render(
6439 self.zoomed.as_ref(),
6440 &PaneRenderContext {
6441 follower_states:
6442 &self.follower_states,
6443 active_call: self.active_call(),
6444 active_pane: &self.active_pane,
6445 app_state: &self.app_state,
6446 project: &self.project,
6447 workspace: &self.weak_self,
6448 },
6449 window,
6450 cx,
6451 ))
6452 .when_some(
6453 paddings.1,
6454 |this, p| {
6455 this.child(
6456 p.border_l_1(),
6457 )
6458 },
6459 ),
6460 ),
6461 )
6462 .children(self.render_dock(
6463 DockPosition::Right,
6464 &self.right_dock,
6465 window,
6466 cx,
6467 )),
6468 )
6469 .child(div().w_full().children(self.render_dock(
6470 DockPosition::Bottom,
6471 &self.bottom_dock,
6472 window,
6473 cx
6474 ))),
6475
6476 BottomDockLayout::LeftAligned => div()
6477 .flex()
6478 .flex_row()
6479 .h_full()
6480 .child(
6481 div()
6482 .flex()
6483 .flex_col()
6484 .flex_1()
6485 .h_full()
6486 .child(
6487 div()
6488 .flex()
6489 .flex_row()
6490 .flex_1()
6491 .children(self.render_dock(DockPosition::Left, &self.left_dock, window, cx))
6492 .child(
6493 div()
6494 .flex()
6495 .flex_col()
6496 .flex_1()
6497 .overflow_hidden()
6498 .child(
6499 h_flex()
6500 .flex_1()
6501 .when_some(paddings.0, |this, p| this.child(p.border_r_1()))
6502 .child(self.center.render(
6503 self.zoomed.as_ref(),
6504 &PaneRenderContext {
6505 follower_states:
6506 &self.follower_states,
6507 active_call: self.active_call(),
6508 active_pane: &self.active_pane,
6509 app_state: &self.app_state,
6510 project: &self.project,
6511 workspace: &self.weak_self,
6512 },
6513 window,
6514 cx,
6515 ))
6516 .when_some(paddings.1, |this, p| this.child(p.border_l_1())),
6517 )
6518 )
6519 )
6520 .child(
6521 div()
6522 .w_full()
6523 .children(self.render_dock(DockPosition::Bottom, &self.bottom_dock, window, cx))
6524 ),
6525 )
6526 .children(self.render_dock(
6527 DockPosition::Right,
6528 &self.right_dock,
6529 window,
6530 cx,
6531 )),
6532
6533 BottomDockLayout::RightAligned => div()
6534 .flex()
6535 .flex_row()
6536 .h_full()
6537 .children(self.render_dock(
6538 DockPosition::Left,
6539 &self.left_dock,
6540 window,
6541 cx,
6542 ))
6543 .child(
6544 div()
6545 .flex()
6546 .flex_col()
6547 .flex_1()
6548 .h_full()
6549 .child(
6550 div()
6551 .flex()
6552 .flex_row()
6553 .flex_1()
6554 .child(
6555 div()
6556 .flex()
6557 .flex_col()
6558 .flex_1()
6559 .overflow_hidden()
6560 .child(
6561 h_flex()
6562 .flex_1()
6563 .when_some(paddings.0, |this, p| this.child(p.border_r_1()))
6564 .child(self.center.render(
6565 self.zoomed.as_ref(),
6566 &PaneRenderContext {
6567 follower_states:
6568 &self.follower_states,
6569 active_call: self.active_call(),
6570 active_pane: &self.active_pane,
6571 app_state: &self.app_state,
6572 project: &self.project,
6573 workspace: &self.weak_self,
6574 },
6575 window,
6576 cx,
6577 ))
6578 .when_some(paddings.1, |this, p| this.child(p.border_l_1())),
6579 )
6580 )
6581 .children(self.render_dock(DockPosition::Right, &self.right_dock, window, cx))
6582 )
6583 .child(
6584 div()
6585 .w_full()
6586 .children(self.render_dock(DockPosition::Bottom, &self.bottom_dock, window, cx))
6587 ),
6588 ),
6589
6590 BottomDockLayout::Contained => div()
6591 .flex()
6592 .flex_row()
6593 .h_full()
6594 .children(self.render_dock(
6595 DockPosition::Left,
6596 &self.left_dock,
6597 window,
6598 cx,
6599 ))
6600 .child(
6601 div()
6602 .flex()
6603 .flex_col()
6604 .flex_1()
6605 .overflow_hidden()
6606 .child(
6607 h_flex()
6608 .flex_1()
6609 .when_some(paddings.0, |this, p| {
6610 this.child(p.border_r_1())
6611 })
6612 .child(self.center.render(
6613 self.zoomed.as_ref(),
6614 &PaneRenderContext {
6615 follower_states:
6616 &self.follower_states,
6617 active_call: self.active_call(),
6618 active_pane: &self.active_pane,
6619 app_state: &self.app_state,
6620 project: &self.project,
6621 workspace: &self.weak_self,
6622 },
6623 window,
6624 cx,
6625 ))
6626 .when_some(paddings.1, |this, p| {
6627 this.child(p.border_l_1())
6628 }),
6629 )
6630 .children(self.render_dock(
6631 DockPosition::Bottom,
6632 &self.bottom_dock,
6633 window,
6634 cx,
6635 )),
6636 )
6637 .children(self.render_dock(
6638 DockPosition::Right,
6639 &self.right_dock,
6640 window,
6641 cx,
6642 )),
6643 }
6644 })
6645 .children(self.zoomed.as_ref().and_then(|view| {
6646 Some(div()
6647 .occlude()
6648 .absolute()
6649 .overflow_hidden()
6650 .border_color(colors.border)
6651 .bg(colors.background)
6652 .child(view.upgrade()?)
6653 .inset_0()
6654 .shadow_lg())
6655 }))
6656 .children(self.render_notifications(window, cx)),
6657 )
6658 .child(self.status_bar.clone())
6659 .child(self.modal_layer.clone())
6660 .child(self.toast_layer.clone()),
6661 ),
6662 window,
6663 cx,
6664 )
6665 }
6666}
6667
6668impl WorkspaceStore {
6669 pub fn new(client: Arc<Client>, cx: &mut Context<Self>) -> Self {
6670 Self {
6671 workspaces: Default::default(),
6672 _subscriptions: vec![
6673 client.add_request_handler(cx.weak_entity(), Self::handle_follow),
6674 client.add_message_handler(cx.weak_entity(), Self::handle_update_followers),
6675 ],
6676 client,
6677 }
6678 }
6679
6680 pub fn update_followers(
6681 &self,
6682 project_id: Option<u64>,
6683 update: proto::update_followers::Variant,
6684 cx: &App,
6685 ) -> Option<()> {
6686 let active_call = ActiveCall::try_global(cx)?;
6687 let room_id = active_call.read(cx).room()?.read(cx).id();
6688 self.client
6689 .send(proto::UpdateFollowers {
6690 room_id,
6691 project_id,
6692 variant: Some(update),
6693 })
6694 .log_err()
6695 }
6696
6697 pub async fn handle_follow(
6698 this: Entity<Self>,
6699 envelope: TypedEnvelope<proto::Follow>,
6700 mut cx: AsyncApp,
6701 ) -> Result<proto::FollowResponse> {
6702 this.update(&mut cx, |this, cx| {
6703 let follower = Follower {
6704 project_id: envelope.payload.project_id,
6705 peer_id: envelope.original_sender_id()?,
6706 };
6707
6708 let mut response = proto::FollowResponse::default();
6709 this.workspaces.retain(|workspace| {
6710 workspace
6711 .update(cx, |workspace, window, cx| {
6712 let handler_response =
6713 workspace.handle_follow(follower.project_id, window, cx);
6714 if let Some(active_view) = handler_response.active_view
6715 && workspace.project.read(cx).remote_id() == follower.project_id
6716 {
6717 response.active_view = Some(active_view)
6718 }
6719 })
6720 .is_ok()
6721 });
6722
6723 Ok(response)
6724 })?
6725 }
6726
6727 async fn handle_update_followers(
6728 this: Entity<Self>,
6729 envelope: TypedEnvelope<proto::UpdateFollowers>,
6730 mut cx: AsyncApp,
6731 ) -> Result<()> {
6732 let leader_id = envelope.original_sender_id()?;
6733 let update = envelope.payload;
6734
6735 this.update(&mut cx, |this, cx| {
6736 this.workspaces.retain(|workspace| {
6737 workspace
6738 .update(cx, |workspace, window, cx| {
6739 let project_id = workspace.project.read(cx).remote_id();
6740 if update.project_id != project_id && update.project_id.is_some() {
6741 return;
6742 }
6743 workspace.handle_update_followers(leader_id, update.clone(), window, cx);
6744 })
6745 .is_ok()
6746 });
6747 Ok(())
6748 })?
6749 }
6750
6751 pub fn workspaces(&self) -> &HashSet<WindowHandle<Workspace>> {
6752 &self.workspaces
6753 }
6754}
6755
6756impl ViewId {
6757 pub(crate) fn from_proto(message: proto::ViewId) -> Result<Self> {
6758 Ok(Self {
6759 creator: message
6760 .creator
6761 .map(CollaboratorId::PeerId)
6762 .context("creator is missing")?,
6763 id: message.id,
6764 })
6765 }
6766
6767 pub(crate) fn to_proto(self) -> Option<proto::ViewId> {
6768 if let CollaboratorId::PeerId(peer_id) = self.creator {
6769 Some(proto::ViewId {
6770 creator: Some(peer_id),
6771 id: self.id,
6772 })
6773 } else {
6774 None
6775 }
6776 }
6777}
6778
6779impl FollowerState {
6780 fn pane(&self) -> &Entity<Pane> {
6781 self.dock_pane.as_ref().unwrap_or(&self.center_pane)
6782 }
6783}
6784
6785pub trait WorkspaceHandle {
6786 fn file_project_paths(&self, cx: &App) -> Vec<ProjectPath>;
6787}
6788
6789impl WorkspaceHandle for Entity<Workspace> {
6790 fn file_project_paths(&self, cx: &App) -> Vec<ProjectPath> {
6791 self.read(cx)
6792 .worktrees(cx)
6793 .flat_map(|worktree| {
6794 let worktree_id = worktree.read(cx).id();
6795 worktree.read(cx).files(true, 0).map(move |f| ProjectPath {
6796 worktree_id,
6797 path: f.path.clone(),
6798 })
6799 })
6800 .collect::<Vec<_>>()
6801 }
6802}
6803
6804pub async fn last_opened_workspace_location() -> Option<SerializedWorkspaceLocation> {
6805 DB.last_workspace().await.log_err().flatten()
6806}
6807
6808pub fn last_session_workspace_locations(
6809 last_session_id: &str,
6810 last_session_window_stack: Option<Vec<WindowId>>,
6811) -> Option<Vec<SerializedWorkspaceLocation>> {
6812 DB.last_session_workspace_locations(last_session_id, last_session_window_stack)
6813 .log_err()
6814}
6815
6816actions!(
6817 collab,
6818 [
6819 /// Opens the channel notes for the current call.
6820 ///
6821 /// If you want to open a specific channel, use `zed::OpenZedUrl` with a channel notes URL -
6822 /// can be copied via "Copy link to section" in the context menu of the channel notes
6823 /// buffer. These URLs look like `https://zed.dev/channel/channel-name-CHANNEL_ID/notes`.
6824 OpenChannelNotes,
6825 /// Mutes your microphone.
6826 Mute,
6827 /// Deafens yourself (mute both microphone and speakers).
6828 Deafen,
6829 /// Leaves the current call.
6830 LeaveCall,
6831 /// Shares the current project with collaborators.
6832 ShareProject,
6833 /// Shares your screen with collaborators.
6834 ScreenShare
6835 ]
6836);
6837actions!(
6838 zed,
6839 [
6840 /// Opens the Zed log file.
6841 OpenLog
6842 ]
6843);
6844
6845async fn join_channel_internal(
6846 channel_id: ChannelId,
6847 app_state: &Arc<AppState>,
6848 requesting_window: Option<WindowHandle<Workspace>>,
6849 active_call: &Entity<ActiveCall>,
6850 cx: &mut AsyncApp,
6851) -> Result<bool> {
6852 let (should_prompt, open_room) = active_call.update(cx, |active_call, cx| {
6853 let Some(room) = active_call.room().map(|room| room.read(cx)) else {
6854 return (false, None);
6855 };
6856
6857 let already_in_channel = room.channel_id() == Some(channel_id);
6858 let should_prompt = room.is_sharing_project()
6859 && !room.remote_participants().is_empty()
6860 && !already_in_channel;
6861 let open_room = if already_in_channel {
6862 active_call.room().cloned()
6863 } else {
6864 None
6865 };
6866 (should_prompt, open_room)
6867 })?;
6868
6869 if let Some(room) = open_room {
6870 let task = room.update(cx, |room, cx| {
6871 if let Some((project, host)) = room.most_active_project(cx) {
6872 return Some(join_in_room_project(project, host, app_state.clone(), cx));
6873 }
6874
6875 None
6876 })?;
6877 if let Some(task) = task {
6878 task.await?;
6879 }
6880 return anyhow::Ok(true);
6881 }
6882
6883 if should_prompt {
6884 if let Some(workspace) = requesting_window {
6885 let answer = workspace
6886 .update(cx, |_, window, cx| {
6887 window.prompt(
6888 PromptLevel::Warning,
6889 "Do you want to switch channels?",
6890 Some("Leaving this call will unshare your current project."),
6891 &["Yes, Join Channel", "Cancel"],
6892 cx,
6893 )
6894 })?
6895 .await;
6896
6897 if answer == Ok(1) {
6898 return Ok(false);
6899 }
6900 } else {
6901 return Ok(false); // unreachable!() hopefully
6902 }
6903 }
6904
6905 let client = cx.update(|cx| active_call.read(cx).client())?;
6906
6907 let mut client_status = client.status();
6908
6909 // this loop will terminate within client::CONNECTION_TIMEOUT seconds.
6910 'outer: loop {
6911 let Some(status) = client_status.recv().await else {
6912 anyhow::bail!("error connecting");
6913 };
6914
6915 match status {
6916 Status::Connecting
6917 | Status::Authenticating
6918 | Status::Authenticated
6919 | Status::Reconnecting
6920 | Status::Reauthenticating => continue,
6921 Status::Connected { .. } => break 'outer,
6922 Status::SignedOut | Status::AuthenticationError => {
6923 return Err(ErrorCode::SignedOut.into());
6924 }
6925 Status::UpgradeRequired => return Err(ErrorCode::UpgradeRequired.into()),
6926 Status::ConnectionError | Status::ConnectionLost | Status::ReconnectionError { .. } => {
6927 return Err(ErrorCode::Disconnected.into());
6928 }
6929 }
6930 }
6931
6932 let room = active_call
6933 .update(cx, |active_call, cx| {
6934 active_call.join_channel(channel_id, cx)
6935 })?
6936 .await?;
6937
6938 let Some(room) = room else {
6939 return anyhow::Ok(true);
6940 };
6941
6942 room.update(cx, |room, _| room.room_update_completed())?
6943 .await;
6944
6945 let task = room.update(cx, |room, cx| {
6946 if let Some((project, host)) = room.most_active_project(cx) {
6947 return Some(join_in_room_project(project, host, app_state.clone(), cx));
6948 }
6949
6950 // If you are the first to join a channel, see if you should share your project.
6951 if room.remote_participants().is_empty()
6952 && !room.local_participant_is_guest()
6953 && let Some(workspace) = requesting_window
6954 {
6955 let project = workspace.update(cx, |workspace, _, cx| {
6956 let project = workspace.project.read(cx);
6957
6958 if !CallSettings::get_global(cx).share_on_join {
6959 return None;
6960 }
6961
6962 if (project.is_local() || project.is_via_ssh())
6963 && project.visible_worktrees(cx).any(|tree| {
6964 tree.read(cx)
6965 .root_entry()
6966 .is_some_and(|entry| entry.is_dir())
6967 })
6968 {
6969 Some(workspace.project.clone())
6970 } else {
6971 None
6972 }
6973 });
6974 if let Ok(Some(project)) = project {
6975 return Some(cx.spawn(async move |room, cx| {
6976 room.update(cx, |room, cx| room.share_project(project, cx))?
6977 .await?;
6978 Ok(())
6979 }));
6980 }
6981 }
6982
6983 None
6984 })?;
6985 if let Some(task) = task {
6986 task.await?;
6987 return anyhow::Ok(true);
6988 }
6989 anyhow::Ok(false)
6990}
6991
6992pub fn join_channel(
6993 channel_id: ChannelId,
6994 app_state: Arc<AppState>,
6995 requesting_window: Option<WindowHandle<Workspace>>,
6996 cx: &mut App,
6997) -> Task<Result<()>> {
6998 let active_call = ActiveCall::global(cx);
6999 cx.spawn(async move |cx| {
7000 let result = join_channel_internal(
7001 channel_id,
7002 &app_state,
7003 requesting_window,
7004 &active_call,
7005 cx,
7006 )
7007 .await;
7008
7009 // join channel succeeded, and opened a window
7010 if matches!(result, Ok(true)) {
7011 return anyhow::Ok(());
7012 }
7013
7014 // find an existing workspace to focus and show call controls
7015 let mut active_window =
7016 requesting_window.or_else(|| activate_any_workspace_window( cx));
7017 if active_window.is_none() {
7018 // no open workspaces, make one to show the error in (blergh)
7019 let (window_handle, _) = cx
7020 .update(|cx| {
7021 Workspace::new_local(vec![], app_state.clone(), requesting_window, None, cx)
7022 })?
7023 .await?;
7024
7025 if result.is_ok() {
7026 cx.update(|cx| {
7027 cx.dispatch_action(&OpenChannelNotes);
7028 }).log_err();
7029 }
7030
7031 active_window = Some(window_handle);
7032 }
7033
7034 if let Err(err) = result {
7035 log::error!("failed to join channel: {}", err);
7036 if let Some(active_window) = active_window {
7037 active_window
7038 .update(cx, |_, window, cx| {
7039 let detail: SharedString = match err.error_code() {
7040 ErrorCode::SignedOut => {
7041 "Please sign in to continue.".into()
7042 }
7043 ErrorCode::UpgradeRequired => {
7044 "Your are running an unsupported version of Zed. Please update to continue.".into()
7045 }
7046 ErrorCode::NoSuchChannel => {
7047 "No matching channel was found. Please check the link and try again.".into()
7048 }
7049 ErrorCode::Forbidden => {
7050 "This channel is private, and you do not have access. Please ask someone to add you and try again.".into()
7051 }
7052 ErrorCode::Disconnected => "Please check your internet connection and try again.".into(),
7053 _ => format!("{}\n\nPlease try again.", err).into(),
7054 };
7055 window.prompt(
7056 PromptLevel::Critical,
7057 "Failed to join channel",
7058 Some(&detail),
7059 &["Ok"],
7060 cx)
7061 })?
7062 .await
7063 .ok();
7064 }
7065 }
7066
7067 // return ok, we showed the error to the user.
7068 anyhow::Ok(())
7069 })
7070}
7071
7072pub async fn get_any_active_workspace(
7073 app_state: Arc<AppState>,
7074 mut cx: AsyncApp,
7075) -> anyhow::Result<WindowHandle<Workspace>> {
7076 // find an existing workspace to focus and show call controls
7077 let active_window = activate_any_workspace_window(&mut cx);
7078 if active_window.is_none() {
7079 cx.update(|cx| Workspace::new_local(vec![], app_state.clone(), None, None, cx))?
7080 .await?;
7081 }
7082 activate_any_workspace_window(&mut cx).context("could not open zed")
7083}
7084
7085fn activate_any_workspace_window(cx: &mut AsyncApp) -> Option<WindowHandle<Workspace>> {
7086 cx.update(|cx| {
7087 if let Some(workspace_window) = cx
7088 .active_window()
7089 .and_then(|window| window.downcast::<Workspace>())
7090 {
7091 return Some(workspace_window);
7092 }
7093
7094 for window in cx.windows() {
7095 if let Some(workspace_window) = window.downcast::<Workspace>() {
7096 workspace_window
7097 .update(cx, |_, window, _| window.activate_window())
7098 .ok();
7099 return Some(workspace_window);
7100 }
7101 }
7102 None
7103 })
7104 .ok()
7105 .flatten()
7106}
7107
7108pub fn local_workspace_windows(cx: &App) -> Vec<WindowHandle<Workspace>> {
7109 cx.windows()
7110 .into_iter()
7111 .filter_map(|window| window.downcast::<Workspace>())
7112 .filter(|workspace| {
7113 workspace
7114 .read(cx)
7115 .is_ok_and(|workspace| workspace.project.read(cx).is_local())
7116 })
7117 .collect()
7118}
7119
7120#[derive(Default)]
7121pub struct OpenOptions {
7122 pub visible: Option<OpenVisible>,
7123 pub focus: Option<bool>,
7124 pub open_new_workspace: Option<bool>,
7125 pub replace_window: Option<WindowHandle<Workspace>>,
7126 pub env: Option<HashMap<String, String>>,
7127}
7128
7129#[allow(clippy::type_complexity)]
7130pub fn open_paths(
7131 abs_paths: &[PathBuf],
7132 app_state: Arc<AppState>,
7133 open_options: OpenOptions,
7134 cx: &mut App,
7135) -> Task<
7136 anyhow::Result<(
7137 WindowHandle<Workspace>,
7138 Vec<Option<anyhow::Result<Box<dyn ItemHandle>>>>,
7139 )>,
7140> {
7141 let abs_paths = abs_paths.to_vec();
7142 let mut existing = None;
7143 let mut best_match = None;
7144 let mut open_visible = OpenVisible::All;
7145
7146 cx.spawn(async move |cx| {
7147 if open_options.open_new_workspace != Some(true) {
7148 let all_paths = abs_paths.iter().map(|path| app_state.fs.metadata(path));
7149 let all_metadatas = futures::future::join_all(all_paths)
7150 .await
7151 .into_iter()
7152 .filter_map(|result| result.ok().flatten())
7153 .collect::<Vec<_>>();
7154
7155 cx.update(|cx| {
7156 for window in local_workspace_windows(cx) {
7157 if let Ok(workspace) = window.read(cx) {
7158 let m = workspace.project.read(cx).visibility_for_paths(
7159 &abs_paths,
7160 &all_metadatas,
7161 open_options.open_new_workspace == None,
7162 cx,
7163 );
7164 if m > best_match {
7165 existing = Some(window);
7166 best_match = m;
7167 } else if best_match.is_none()
7168 && open_options.open_new_workspace == Some(false)
7169 {
7170 existing = Some(window)
7171 }
7172 }
7173 }
7174 })?;
7175
7176 if open_options.open_new_workspace.is_none()
7177 && existing.is_none()
7178 && all_metadatas.iter().all(|file| !file.is_dir)
7179 {
7180 cx.update(|cx| {
7181 if let Some(window) = cx
7182 .active_window()
7183 .and_then(|window| window.downcast::<Workspace>())
7184 && let Ok(workspace) = window.read(cx)
7185 {
7186 let project = workspace.project().read(cx);
7187 if project.is_local() && !project.is_via_collab() {
7188 existing = Some(window);
7189 open_visible = OpenVisible::None;
7190 return;
7191 }
7192 }
7193 for window in local_workspace_windows(cx) {
7194 if let Ok(workspace) = window.read(cx) {
7195 let project = workspace.project().read(cx);
7196 if project.is_via_collab() {
7197 continue;
7198 }
7199 existing = Some(window);
7200 open_visible = OpenVisible::None;
7201 break;
7202 }
7203 }
7204 })?;
7205 }
7206 }
7207
7208 if let Some(existing) = existing {
7209 let open_task = existing
7210 .update(cx, |workspace, window, cx| {
7211 window.activate_window();
7212 workspace.open_paths(
7213 abs_paths,
7214 OpenOptions {
7215 visible: Some(open_visible),
7216 ..Default::default()
7217 },
7218 None,
7219 window,
7220 cx,
7221 )
7222 })?
7223 .await;
7224
7225 _ = existing.update(cx, |workspace, _, cx| {
7226 for item in open_task.iter().flatten() {
7227 if let Err(e) = item {
7228 workspace.show_error(&e, cx);
7229 }
7230 }
7231 });
7232
7233 Ok((existing, open_task))
7234 } else {
7235 cx.update(move |cx| {
7236 Workspace::new_local(
7237 abs_paths,
7238 app_state.clone(),
7239 open_options.replace_window,
7240 open_options.env,
7241 cx,
7242 )
7243 })?
7244 .await
7245 }
7246 })
7247}
7248
7249pub fn open_new(
7250 open_options: OpenOptions,
7251 app_state: Arc<AppState>,
7252 cx: &mut App,
7253 init: impl FnOnce(&mut Workspace, &mut Window, &mut Context<Workspace>) + 'static + Send,
7254) -> Task<anyhow::Result<()>> {
7255 let task = Workspace::new_local(Vec::new(), app_state, None, open_options.env, cx);
7256 cx.spawn(async move |cx| {
7257 let (workspace, opened_paths) = task.await?;
7258 workspace.update(cx, |workspace, window, cx| {
7259 if opened_paths.is_empty() {
7260 init(workspace, window, cx)
7261 }
7262 })?;
7263 Ok(())
7264 })
7265}
7266
7267pub fn create_and_open_local_file(
7268 path: &'static Path,
7269 window: &mut Window,
7270 cx: &mut Context<Workspace>,
7271 default_content: impl 'static + Send + FnOnce() -> Rope,
7272) -> Task<Result<Box<dyn ItemHandle>>> {
7273 cx.spawn_in(window, async move |workspace, cx| {
7274 let fs = workspace.read_with(cx, |workspace, _| workspace.app_state().fs.clone())?;
7275 if !fs.is_file(path).await {
7276 fs.create_file(path, Default::default()).await?;
7277 fs.save(path, &default_content(), Default::default())
7278 .await?;
7279 }
7280
7281 let mut items = workspace
7282 .update_in(cx, |workspace, window, cx| {
7283 workspace.with_local_workspace(window, cx, |workspace, window, cx| {
7284 workspace.open_paths(
7285 vec![path.to_path_buf()],
7286 OpenOptions {
7287 visible: Some(OpenVisible::None),
7288 ..Default::default()
7289 },
7290 None,
7291 window,
7292 cx,
7293 )
7294 })
7295 })?
7296 .await?
7297 .await;
7298
7299 let item = items.pop().flatten();
7300 item.with_context(|| format!("path {path:?} is not a file"))?
7301 })
7302}
7303
7304pub fn open_ssh_project_with_new_connection(
7305 window: WindowHandle<Workspace>,
7306 connection_options: SshConnectionOptions,
7307 cancel_rx: oneshot::Receiver<()>,
7308 delegate: Arc<dyn SshClientDelegate>,
7309 app_state: Arc<AppState>,
7310 paths: Vec<PathBuf>,
7311 cx: &mut App,
7312) -> Task<Result<()>> {
7313 cx.spawn(async move |cx| {
7314 let (serialized_ssh_project, workspace_id, serialized_workspace) =
7315 serialize_ssh_project(connection_options.clone(), paths.clone(), cx).await?;
7316
7317 let session = match cx
7318 .update(|cx| {
7319 remote::SshRemoteClient::new(
7320 ConnectionIdentifier::Workspace(workspace_id.0),
7321 connection_options,
7322 cancel_rx,
7323 delegate,
7324 cx,
7325 )
7326 })?
7327 .await?
7328 {
7329 Some(result) => result,
7330 None => return Ok(()),
7331 };
7332
7333 let project = cx.update(|cx| {
7334 project::Project::ssh(
7335 session,
7336 app_state.client.clone(),
7337 app_state.node_runtime.clone(),
7338 app_state.user_store.clone(),
7339 app_state.languages.clone(),
7340 app_state.fs.clone(),
7341 cx,
7342 )
7343 })?;
7344
7345 open_ssh_project_inner(
7346 project,
7347 paths,
7348 serialized_ssh_project,
7349 workspace_id,
7350 serialized_workspace,
7351 app_state,
7352 window,
7353 cx,
7354 )
7355 .await
7356 })
7357}
7358
7359pub fn open_ssh_project_with_existing_connection(
7360 connection_options: SshConnectionOptions,
7361 project: Entity<Project>,
7362 paths: Vec<PathBuf>,
7363 app_state: Arc<AppState>,
7364 window: WindowHandle<Workspace>,
7365 cx: &mut AsyncApp,
7366) -> Task<Result<()>> {
7367 cx.spawn(async move |cx| {
7368 let (serialized_ssh_project, workspace_id, serialized_workspace) =
7369 serialize_ssh_project(connection_options.clone(), paths.clone(), cx).await?;
7370
7371 open_ssh_project_inner(
7372 project,
7373 paths,
7374 serialized_ssh_project,
7375 workspace_id,
7376 serialized_workspace,
7377 app_state,
7378 window,
7379 cx,
7380 )
7381 .await
7382 })
7383}
7384
7385async fn open_ssh_project_inner(
7386 project: Entity<Project>,
7387 paths: Vec<PathBuf>,
7388 serialized_ssh_project: SerializedSshProject,
7389 workspace_id: WorkspaceId,
7390 serialized_workspace: Option<SerializedWorkspace>,
7391 app_state: Arc<AppState>,
7392 window: WindowHandle<Workspace>,
7393 cx: &mut AsyncApp,
7394) -> Result<()> {
7395 let toolchains = DB.toolchains(workspace_id).await?;
7396 for (toolchain, worktree_id, path) in toolchains {
7397 project
7398 .update(cx, |this, cx| {
7399 this.activate_toolchain(ProjectPath { worktree_id, path }, toolchain, cx)
7400 })?
7401 .await;
7402 }
7403 let mut project_paths_to_open = vec![];
7404 let mut project_path_errors = vec![];
7405
7406 for path in paths {
7407 let result = cx
7408 .update(|cx| Workspace::project_path_for_path(project.clone(), &path, true, cx))?
7409 .await;
7410 match result {
7411 Ok((_, project_path)) => {
7412 project_paths_to_open.push((path.clone(), Some(project_path)));
7413 }
7414 Err(error) => {
7415 project_path_errors.push(error);
7416 }
7417 };
7418 }
7419
7420 if project_paths_to_open.is_empty() {
7421 return Err(project_path_errors.pop().context("no paths given")?);
7422 }
7423
7424 if let Some(detach_session_task) = window
7425 .update(cx, |_workspace, window, cx| {
7426 cx.spawn_in(window, async move |this, cx| {
7427 this.update_in(cx, |this, window, cx| this.remove_from_session(window, cx))
7428 })
7429 })
7430 .ok()
7431 {
7432 detach_session_task.await.ok();
7433 }
7434
7435 cx.update_window(window.into(), |_, window, cx| {
7436 window.replace_root(cx, |window, cx| {
7437 telemetry::event!("SSH Project Opened");
7438
7439 let mut workspace =
7440 Workspace::new(Some(workspace_id), project, app_state.clone(), window, cx);
7441 workspace.set_serialized_ssh_project(serialized_ssh_project);
7442 workspace.update_history(cx);
7443
7444 if let Some(ref serialized) = serialized_workspace {
7445 workspace.centered_layout = serialized.centered_layout;
7446 }
7447
7448 workspace
7449 });
7450 })?;
7451
7452 window
7453 .update(cx, |_, window, cx| {
7454 window.activate_window();
7455 open_items(serialized_workspace, project_paths_to_open, window, cx)
7456 })?
7457 .await?;
7458
7459 window.update(cx, |workspace, _, cx| {
7460 for error in project_path_errors {
7461 if error.error_code() == proto::ErrorCode::DevServerProjectPathDoesNotExist {
7462 if let Some(path) = error.error_tag("path") {
7463 workspace.show_error(&anyhow!("'{path}' does not exist"), cx)
7464 }
7465 } else {
7466 workspace.show_error(&error, cx)
7467 }
7468 }
7469 })?;
7470
7471 Ok(())
7472}
7473
7474fn serialize_ssh_project(
7475 connection_options: SshConnectionOptions,
7476 paths: Vec<PathBuf>,
7477 cx: &AsyncApp,
7478) -> Task<
7479 Result<(
7480 SerializedSshProject,
7481 WorkspaceId,
7482 Option<SerializedWorkspace>,
7483 )>,
7484> {
7485 cx.background_spawn(async move {
7486 let serialized_ssh_project = persistence::DB
7487 .get_or_create_ssh_project(
7488 connection_options.host.clone(),
7489 connection_options.port,
7490 paths
7491 .iter()
7492 .map(|path| path.to_string_lossy().to_string())
7493 .collect::<Vec<_>>(),
7494 connection_options.username.clone(),
7495 )
7496 .await?;
7497
7498 let serialized_workspace =
7499 persistence::DB.workspace_for_ssh_project(&serialized_ssh_project);
7500
7501 let workspace_id = if let Some(workspace_id) =
7502 serialized_workspace.as_ref().map(|workspace| workspace.id)
7503 {
7504 workspace_id
7505 } else {
7506 persistence::DB.next_id().await?
7507 };
7508
7509 Ok((serialized_ssh_project, workspace_id, serialized_workspace))
7510 })
7511}
7512
7513pub fn join_in_room_project(
7514 project_id: u64,
7515 follow_user_id: u64,
7516 app_state: Arc<AppState>,
7517 cx: &mut App,
7518) -> Task<Result<()>> {
7519 let windows = cx.windows();
7520 cx.spawn(async move |cx| {
7521 let existing_workspace = windows.into_iter().find_map(|window_handle| {
7522 window_handle
7523 .downcast::<Workspace>()
7524 .and_then(|window_handle| {
7525 window_handle
7526 .update(cx, |workspace, _window, cx| {
7527 if workspace.project().read(cx).remote_id() == Some(project_id) {
7528 Some(window_handle)
7529 } else {
7530 None
7531 }
7532 })
7533 .unwrap_or(None)
7534 })
7535 });
7536
7537 let workspace = if let Some(existing_workspace) = existing_workspace {
7538 existing_workspace
7539 } else {
7540 let active_call = cx.update(|cx| ActiveCall::global(cx))?;
7541 let room = active_call
7542 .read_with(cx, |call, _| call.room().cloned())?
7543 .context("not in a call")?;
7544 let project = room
7545 .update(cx, |room, cx| {
7546 room.join_project(
7547 project_id,
7548 app_state.languages.clone(),
7549 app_state.fs.clone(),
7550 cx,
7551 )
7552 })?
7553 .await?;
7554
7555 let window_bounds_override = window_bounds_env_override();
7556 cx.update(|cx| {
7557 let mut options = (app_state.build_window_options)(None, cx);
7558 options.window_bounds = window_bounds_override.map(WindowBounds::Windowed);
7559 cx.open_window(options, |window, cx| {
7560 cx.new(|cx| {
7561 Workspace::new(Default::default(), project, app_state.clone(), window, cx)
7562 })
7563 })
7564 })??
7565 };
7566
7567 workspace.update(cx, |workspace, window, cx| {
7568 cx.activate(true);
7569 window.activate_window();
7570
7571 if let Some(room) = ActiveCall::global(cx).read(cx).room().cloned() {
7572 let follow_peer_id = room
7573 .read(cx)
7574 .remote_participants()
7575 .iter()
7576 .find(|(_, participant)| participant.user.id == follow_user_id)
7577 .map(|(_, p)| p.peer_id)
7578 .or_else(|| {
7579 // If we couldn't follow the given user, follow the host instead.
7580 let collaborator = workspace
7581 .project()
7582 .read(cx)
7583 .collaborators()
7584 .values()
7585 .find(|collaborator| collaborator.is_host)?;
7586 Some(collaborator.peer_id)
7587 });
7588
7589 if let Some(follow_peer_id) = follow_peer_id {
7590 workspace.follow(follow_peer_id, window, cx);
7591 }
7592 }
7593 })?;
7594
7595 anyhow::Ok(())
7596 })
7597}
7598
7599pub fn reload(cx: &mut App) {
7600 let should_confirm = WorkspaceSettings::get_global(cx).confirm_quit;
7601 let mut workspace_windows = cx
7602 .windows()
7603 .into_iter()
7604 .filter_map(|window| window.downcast::<Workspace>())
7605 .collect::<Vec<_>>();
7606
7607 // If multiple windows have unsaved changes, and need a save prompt,
7608 // prompt in the active window before switching to a different window.
7609 workspace_windows.sort_by_key(|window| window.is_active(cx) == Some(false));
7610
7611 let mut prompt = None;
7612 if let (true, Some(window)) = (should_confirm, workspace_windows.first()) {
7613 prompt = window
7614 .update(cx, |_, window, cx| {
7615 window.prompt(
7616 PromptLevel::Info,
7617 "Are you sure you want to restart?",
7618 None,
7619 &["Restart", "Cancel"],
7620 cx,
7621 )
7622 })
7623 .ok();
7624 }
7625
7626 cx.spawn(async move |cx| {
7627 if let Some(prompt) = prompt {
7628 let answer = prompt.await?;
7629 if answer != 0 {
7630 return Ok(());
7631 }
7632 }
7633
7634 // If the user cancels any save prompt, then keep the app open.
7635 for window in workspace_windows {
7636 if let Ok(should_close) = window.update(cx, |workspace, window, cx| {
7637 workspace.prepare_to_close(CloseIntent::Quit, window, cx)
7638 }) && !should_close.await?
7639 {
7640 return Ok(());
7641 }
7642 }
7643 cx.update(|cx| cx.restart())
7644 })
7645 .detach_and_log_err(cx);
7646}
7647
7648fn parse_pixel_position_env_var(value: &str) -> Option<Point<Pixels>> {
7649 let mut parts = value.split(',');
7650 let x: usize = parts.next()?.parse().ok()?;
7651 let y: usize = parts.next()?.parse().ok()?;
7652 Some(point(px(x as f32), px(y as f32)))
7653}
7654
7655fn parse_pixel_size_env_var(value: &str) -> Option<Size<Pixels>> {
7656 let mut parts = value.split(',');
7657 let width: usize = parts.next()?.parse().ok()?;
7658 let height: usize = parts.next()?.parse().ok()?;
7659 Some(size(px(width as f32), px(height as f32)))
7660}
7661
7662/// Add client-side decorations (rounded corners, shadows, resize handling) when appropriate.
7663pub fn client_side_decorations(
7664 element: impl IntoElement,
7665 window: &mut Window,
7666 cx: &mut App,
7667) -> Stateful<Div> {
7668 const BORDER_SIZE: Pixels = px(1.0);
7669 let decorations = window.window_decorations();
7670
7671 match decorations {
7672 Decorations::Client { .. } => window.set_client_inset(theme::CLIENT_SIDE_DECORATION_SHADOW),
7673 Decorations::Server => window.set_client_inset(px(0.0)),
7674 }
7675
7676 struct GlobalResizeEdge(ResizeEdge);
7677 impl Global for GlobalResizeEdge {}
7678
7679 div()
7680 .id("window-backdrop")
7681 .bg(transparent_black())
7682 .map(|div| match decorations {
7683 Decorations::Server => div,
7684 Decorations::Client { tiling, .. } => div
7685 .when(!(tiling.top || tiling.right), |div| {
7686 div.rounded_tr(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7687 })
7688 .when(!(tiling.top || tiling.left), |div| {
7689 div.rounded_tl(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7690 })
7691 .when(!(tiling.bottom || tiling.right), |div| {
7692 div.rounded_br(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7693 })
7694 .when(!(tiling.bottom || tiling.left), |div| {
7695 div.rounded_bl(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7696 })
7697 .when(!tiling.top, |div| {
7698 div.pt(theme::CLIENT_SIDE_DECORATION_SHADOW)
7699 })
7700 .when(!tiling.bottom, |div| {
7701 div.pb(theme::CLIENT_SIDE_DECORATION_SHADOW)
7702 })
7703 .when(!tiling.left, |div| {
7704 div.pl(theme::CLIENT_SIDE_DECORATION_SHADOW)
7705 })
7706 .when(!tiling.right, |div| {
7707 div.pr(theme::CLIENT_SIDE_DECORATION_SHADOW)
7708 })
7709 .on_mouse_move(move |e, window, cx| {
7710 let size = window.window_bounds().get_bounds().size;
7711 let pos = e.position;
7712
7713 let new_edge =
7714 resize_edge(pos, theme::CLIENT_SIDE_DECORATION_SHADOW, size, tiling);
7715
7716 let edge = cx.try_global::<GlobalResizeEdge>();
7717 if new_edge != edge.map(|edge| edge.0) {
7718 window
7719 .window_handle()
7720 .update(cx, |workspace, _, cx| {
7721 cx.notify(workspace.entity_id());
7722 })
7723 .ok();
7724 }
7725 })
7726 .on_mouse_down(MouseButton::Left, move |e, window, _| {
7727 let size = window.window_bounds().get_bounds().size;
7728 let pos = e.position;
7729
7730 let edge = match resize_edge(
7731 pos,
7732 theme::CLIENT_SIDE_DECORATION_SHADOW,
7733 size,
7734 tiling,
7735 ) {
7736 Some(value) => value,
7737 None => return,
7738 };
7739
7740 window.start_window_resize(edge);
7741 }),
7742 })
7743 .size_full()
7744 .child(
7745 div()
7746 .cursor(CursorStyle::Arrow)
7747 .map(|div| match decorations {
7748 Decorations::Server => div,
7749 Decorations::Client { tiling } => div
7750 .border_color(cx.theme().colors().border)
7751 .when(!(tiling.top || tiling.right), |div| {
7752 div.rounded_tr(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7753 })
7754 .when(!(tiling.top || tiling.left), |div| {
7755 div.rounded_tl(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7756 })
7757 .when(!(tiling.bottom || tiling.right), |div| {
7758 div.rounded_br(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7759 })
7760 .when(!(tiling.bottom || tiling.left), |div| {
7761 div.rounded_bl(theme::CLIENT_SIDE_DECORATION_ROUNDING)
7762 })
7763 .when(!tiling.top, |div| div.border_t(BORDER_SIZE))
7764 .when(!tiling.bottom, |div| div.border_b(BORDER_SIZE))
7765 .when(!tiling.left, |div| div.border_l(BORDER_SIZE))
7766 .when(!tiling.right, |div| div.border_r(BORDER_SIZE))
7767 .when(!tiling.is_tiled(), |div| {
7768 div.shadow(vec![gpui::BoxShadow {
7769 color: Hsla {
7770 h: 0.,
7771 s: 0.,
7772 l: 0.,
7773 a: 0.4,
7774 },
7775 blur_radius: theme::CLIENT_SIDE_DECORATION_SHADOW / 2.,
7776 spread_radius: px(0.),
7777 offset: point(px(0.0), px(0.0)),
7778 }])
7779 }),
7780 })
7781 .on_mouse_move(|_e, _, cx| {
7782 cx.stop_propagation();
7783 })
7784 .size_full()
7785 .child(element),
7786 )
7787 .map(|div| match decorations {
7788 Decorations::Server => div,
7789 Decorations::Client { tiling, .. } => div.child(
7790 canvas(
7791 |_bounds, window, _| {
7792 window.insert_hitbox(
7793 Bounds::new(
7794 point(px(0.0), px(0.0)),
7795 window.window_bounds().get_bounds().size,
7796 ),
7797 HitboxBehavior::Normal,
7798 )
7799 },
7800 move |_bounds, hitbox, window, cx| {
7801 let mouse = window.mouse_position();
7802 let size = window.window_bounds().get_bounds().size;
7803 let Some(edge) =
7804 resize_edge(mouse, theme::CLIENT_SIDE_DECORATION_SHADOW, size, tiling)
7805 else {
7806 return;
7807 };
7808 cx.set_global(GlobalResizeEdge(edge));
7809 window.set_cursor_style(
7810 match edge {
7811 ResizeEdge::Top | ResizeEdge::Bottom => CursorStyle::ResizeUpDown,
7812 ResizeEdge::Left | ResizeEdge::Right => {
7813 CursorStyle::ResizeLeftRight
7814 }
7815 ResizeEdge::TopLeft | ResizeEdge::BottomRight => {
7816 CursorStyle::ResizeUpLeftDownRight
7817 }
7818 ResizeEdge::TopRight | ResizeEdge::BottomLeft => {
7819 CursorStyle::ResizeUpRightDownLeft
7820 }
7821 },
7822 &hitbox,
7823 );
7824 },
7825 )
7826 .size_full()
7827 .absolute(),
7828 ),
7829 })
7830}
7831
7832fn resize_edge(
7833 pos: Point<Pixels>,
7834 shadow_size: Pixels,
7835 window_size: Size<Pixels>,
7836 tiling: Tiling,
7837) -> Option<ResizeEdge> {
7838 let bounds = Bounds::new(Point::default(), window_size).inset(shadow_size * 1.5);
7839 if bounds.contains(&pos) {
7840 return None;
7841 }
7842
7843 let corner_size = size(shadow_size * 1.5, shadow_size * 1.5);
7844 let top_left_bounds = Bounds::new(Point::new(px(0.), px(0.)), corner_size);
7845 if !tiling.top && top_left_bounds.contains(&pos) {
7846 return Some(ResizeEdge::TopLeft);
7847 }
7848
7849 let top_right_bounds = Bounds::new(
7850 Point::new(window_size.width - corner_size.width, px(0.)),
7851 corner_size,
7852 );
7853 if !tiling.top && top_right_bounds.contains(&pos) {
7854 return Some(ResizeEdge::TopRight);
7855 }
7856
7857 let bottom_left_bounds = Bounds::new(
7858 Point::new(px(0.), window_size.height - corner_size.height),
7859 corner_size,
7860 );
7861 if !tiling.bottom && bottom_left_bounds.contains(&pos) {
7862 return Some(ResizeEdge::BottomLeft);
7863 }
7864
7865 let bottom_right_bounds = Bounds::new(
7866 Point::new(
7867 window_size.width - corner_size.width,
7868 window_size.height - corner_size.height,
7869 ),
7870 corner_size,
7871 );
7872 if !tiling.bottom && bottom_right_bounds.contains(&pos) {
7873 return Some(ResizeEdge::BottomRight);
7874 }
7875
7876 if !tiling.top && pos.y < shadow_size {
7877 Some(ResizeEdge::Top)
7878 } else if !tiling.bottom && pos.y > window_size.height - shadow_size {
7879 Some(ResizeEdge::Bottom)
7880 } else if !tiling.left && pos.x < shadow_size {
7881 Some(ResizeEdge::Left)
7882 } else if !tiling.right && pos.x > window_size.width - shadow_size {
7883 Some(ResizeEdge::Right)
7884 } else {
7885 None
7886 }
7887}
7888
7889fn join_pane_into_active(
7890 active_pane: &Entity<Pane>,
7891 pane: &Entity<Pane>,
7892 window: &mut Window,
7893 cx: &mut App,
7894) {
7895 if pane == active_pane {
7896 } else if pane.read(cx).items_len() == 0 {
7897 pane.update(cx, |_, cx| {
7898 cx.emit(pane::Event::Remove {
7899 focus_on_pane: None,
7900 });
7901 })
7902 } else {
7903 move_all_items(pane, active_pane, window, cx);
7904 }
7905}
7906
7907fn move_all_items(
7908 from_pane: &Entity<Pane>,
7909 to_pane: &Entity<Pane>,
7910 window: &mut Window,
7911 cx: &mut App,
7912) {
7913 let destination_is_different = from_pane != to_pane;
7914 let mut moved_items = 0;
7915 for (item_ix, item_handle) in from_pane
7916 .read(cx)
7917 .items()
7918 .enumerate()
7919 .map(|(ix, item)| (ix, item.clone()))
7920 .collect::<Vec<_>>()
7921 {
7922 let ix = item_ix - moved_items;
7923 if destination_is_different {
7924 // Close item from previous pane
7925 from_pane.update(cx, |source, cx| {
7926 source.remove_item_and_focus_on_pane(ix, false, to_pane.clone(), window, cx);
7927 });
7928 moved_items += 1;
7929 }
7930
7931 // This automatically removes duplicate items in the pane
7932 to_pane.update(cx, |destination, cx| {
7933 destination.add_item(item_handle, true, true, None, window, cx);
7934 window.focus(&destination.focus_handle(cx))
7935 });
7936 }
7937}
7938
7939pub fn move_item(
7940 source: &Entity<Pane>,
7941 destination: &Entity<Pane>,
7942 item_id_to_move: EntityId,
7943 destination_index: usize,
7944 activate: bool,
7945 window: &mut Window,
7946 cx: &mut App,
7947) {
7948 let Some((item_ix, item_handle)) = source
7949 .read(cx)
7950 .items()
7951 .enumerate()
7952 .find(|(_, item_handle)| item_handle.item_id() == item_id_to_move)
7953 .map(|(ix, item)| (ix, item.clone()))
7954 else {
7955 // Tab was closed during drag
7956 return;
7957 };
7958
7959 if source != destination {
7960 // Close item from previous pane
7961 source.update(cx, |source, cx| {
7962 source.remove_item_and_focus_on_pane(item_ix, false, destination.clone(), window, cx);
7963 });
7964 }
7965
7966 // This automatically removes duplicate items in the pane
7967 destination.update(cx, |destination, cx| {
7968 destination.add_item_inner(
7969 item_handle,
7970 activate,
7971 activate,
7972 activate,
7973 Some(destination_index),
7974 window,
7975 cx,
7976 );
7977 if activate {
7978 window.focus(&destination.focus_handle(cx))
7979 }
7980 });
7981}
7982
7983pub fn move_active_item(
7984 source: &Entity<Pane>,
7985 destination: &Entity<Pane>,
7986 focus_destination: bool,
7987 close_if_empty: bool,
7988 window: &mut Window,
7989 cx: &mut App,
7990) {
7991 if source == destination {
7992 return;
7993 }
7994 let Some(active_item) = source.read(cx).active_item() else {
7995 return;
7996 };
7997 source.update(cx, |source_pane, cx| {
7998 let item_id = active_item.item_id();
7999 source_pane.remove_item(item_id, false, close_if_empty, window, cx);
8000 destination.update(cx, |target_pane, cx| {
8001 target_pane.add_item(
8002 active_item,
8003 focus_destination,
8004 focus_destination,
8005 Some(target_pane.items_len()),
8006 window,
8007 cx,
8008 );
8009 });
8010 });
8011}
8012
8013pub fn clone_active_item(
8014 workspace_id: Option<WorkspaceId>,
8015 source: &Entity<Pane>,
8016 destination: &Entity<Pane>,
8017 focus_destination: bool,
8018 window: &mut Window,
8019 cx: &mut App,
8020) {
8021 if source == destination {
8022 return;
8023 }
8024 let Some(active_item) = source.read(cx).active_item() else {
8025 return;
8026 };
8027 destination.update(cx, |target_pane, cx| {
8028 let Some(clone) = active_item.clone_on_split(workspace_id, window, cx) else {
8029 return;
8030 };
8031 target_pane.add_item(
8032 clone,
8033 focus_destination,
8034 focus_destination,
8035 Some(target_pane.items_len()),
8036 window,
8037 cx,
8038 );
8039 });
8040}
8041
8042#[derive(Debug)]
8043pub struct WorkspacePosition {
8044 pub window_bounds: Option<WindowBounds>,
8045 pub display: Option<Uuid>,
8046 pub centered_layout: bool,
8047}
8048
8049pub fn ssh_workspace_position_from_db(
8050 host: String,
8051 port: Option<u16>,
8052 user: Option<String>,
8053 paths_to_open: &[PathBuf],
8054 cx: &App,
8055) -> Task<Result<WorkspacePosition>> {
8056 let paths = paths_to_open
8057 .iter()
8058 .map(|path| path.to_string_lossy().to_string())
8059 .collect::<Vec<_>>();
8060
8061 cx.background_spawn(async move {
8062 let serialized_ssh_project = persistence::DB
8063 .get_or_create_ssh_project(host, port, paths, user)
8064 .await
8065 .context("fetching serialized ssh project")?;
8066 let serialized_workspace =
8067 persistence::DB.workspace_for_ssh_project(&serialized_ssh_project);
8068
8069 let (window_bounds, display) = if let Some(bounds) = window_bounds_env_override() {
8070 (Some(WindowBounds::Windowed(bounds)), None)
8071 } else {
8072 let restorable_bounds = serialized_workspace
8073 .as_ref()
8074 .and_then(|workspace| Some((workspace.display?, workspace.window_bounds?)))
8075 .or_else(|| {
8076 let (display, window_bounds) = DB.last_window().log_err()?;
8077 Some((display?, window_bounds?))
8078 });
8079
8080 if let Some((serialized_display, serialized_status)) = restorable_bounds {
8081 (Some(serialized_status.0), Some(serialized_display))
8082 } else {
8083 (None, None)
8084 }
8085 };
8086
8087 let centered_layout = serialized_workspace
8088 .as_ref()
8089 .map(|w| w.centered_layout)
8090 .unwrap_or(false);
8091
8092 Ok(WorkspacePosition {
8093 window_bounds,
8094 display,
8095 centered_layout,
8096 })
8097 })
8098}
8099
8100pub fn with_active_or_new_workspace(
8101 cx: &mut App,
8102 f: impl FnOnce(&mut Workspace, &mut Window, &mut Context<Workspace>) + Send + 'static,
8103) {
8104 match cx.active_window().and_then(|w| w.downcast::<Workspace>()) {
8105 Some(workspace) => {
8106 cx.defer(move |cx| {
8107 workspace
8108 .update(cx, |workspace, window, cx| f(workspace, window, cx))
8109 .log_err();
8110 });
8111 }
8112 None => {
8113 let app_state = AppState::global(cx);
8114 if let Some(app_state) = app_state.upgrade() {
8115 open_new(
8116 OpenOptions::default(),
8117 app_state,
8118 cx,
8119 move |workspace, window, cx| f(workspace, window, cx),
8120 )
8121 .detach_and_log_err(cx);
8122 }
8123 }
8124 }
8125}
8126
8127#[cfg(test)]
8128mod tests {
8129 use std::{cell::RefCell, rc::Rc};
8130
8131 use super::*;
8132 use crate::{
8133 dock::{PanelEvent, test::TestPanel},
8134 item::{
8135 ItemEvent,
8136 test::{TestItem, TestProjectItem},
8137 },
8138 };
8139 use fs::FakeFs;
8140 use gpui::{
8141 DismissEvent, Empty, EventEmitter, FocusHandle, Focusable, Render, TestAppContext,
8142 UpdateGlobal, VisualTestContext, px,
8143 };
8144 use project::{Project, ProjectEntryId};
8145 use serde_json::json;
8146 use settings::SettingsStore;
8147
8148 #[gpui::test]
8149 async fn test_tab_disambiguation(cx: &mut TestAppContext) {
8150 init_test(cx);
8151
8152 let fs = FakeFs::new(cx.executor());
8153 let project = Project::test(fs, [], cx).await;
8154 let (workspace, cx) =
8155 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
8156
8157 // Adding an item with no ambiguity renders the tab without detail.
8158 let item1 = cx.new(|cx| {
8159 let mut item = TestItem::new(cx);
8160 item.tab_descriptions = Some(vec!["c", "b1/c", "a/b1/c"]);
8161 item
8162 });
8163 workspace.update_in(cx, |workspace, window, cx| {
8164 workspace.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx);
8165 });
8166 item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(0)));
8167
8168 // Adding an item that creates ambiguity increases the level of detail on
8169 // both tabs.
8170 let item2 = cx.new_window_entity(|_window, cx| {
8171 let mut item = TestItem::new(cx);
8172 item.tab_descriptions = Some(vec!["c", "b2/c", "a/b2/c"]);
8173 item
8174 });
8175 workspace.update_in(cx, |workspace, window, cx| {
8176 workspace.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx);
8177 });
8178 item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1)));
8179 item2.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1)));
8180
8181 // Adding an item that creates ambiguity increases the level of detail only
8182 // on the ambiguous tabs. In this case, the ambiguity can't be resolved so
8183 // we stop at the highest detail available.
8184 let item3 = cx.new(|cx| {
8185 let mut item = TestItem::new(cx);
8186 item.tab_descriptions = Some(vec!["c", "b2/c", "a/b2/c"]);
8187 item
8188 });
8189 workspace.update_in(cx, |workspace, window, cx| {
8190 workspace.add_item_to_active_pane(Box::new(item3.clone()), None, true, window, cx);
8191 });
8192 item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1)));
8193 item2.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(3)));
8194 item3.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(3)));
8195 }
8196
8197 #[gpui::test]
8198 async fn test_tracking_active_path(cx: &mut TestAppContext) {
8199 init_test(cx);
8200
8201 let fs = FakeFs::new(cx.executor());
8202 fs.insert_tree(
8203 "/root1",
8204 json!({
8205 "one.txt": "",
8206 "two.txt": "",
8207 }),
8208 )
8209 .await;
8210 fs.insert_tree(
8211 "/root2",
8212 json!({
8213 "three.txt": "",
8214 }),
8215 )
8216 .await;
8217
8218 let project = Project::test(fs, ["root1".as_ref()], cx).await;
8219 let (workspace, cx) =
8220 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
8221 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
8222 let worktree_id = project.update(cx, |project, cx| {
8223 project.worktrees(cx).next().unwrap().read(cx).id()
8224 });
8225
8226 let item1 = cx.new(|cx| {
8227 TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "one.txt", cx)])
8228 });
8229 let item2 = cx.new(|cx| {
8230 TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "two.txt", cx)])
8231 });
8232
8233 // Add an item to an empty pane
8234 workspace.update_in(cx, |workspace, window, cx| {
8235 workspace.add_item_to_active_pane(Box::new(item1), None, true, window, cx)
8236 });
8237 project.update(cx, |project, cx| {
8238 assert_eq!(
8239 project.active_entry(),
8240 project
8241 .entry_for_path(&(worktree_id, "one.txt").into(), cx)
8242 .map(|e| e.id)
8243 );
8244 });
8245 assert_eq!(cx.window_title().as_deref(), Some("root1 — one.txt"));
8246
8247 // Add a second item to a non-empty pane
8248 workspace.update_in(cx, |workspace, window, cx| {
8249 workspace.add_item_to_active_pane(Box::new(item2), None, true, window, cx)
8250 });
8251 assert_eq!(cx.window_title().as_deref(), Some("root1 — two.txt"));
8252 project.update(cx, |project, cx| {
8253 assert_eq!(
8254 project.active_entry(),
8255 project
8256 .entry_for_path(&(worktree_id, "two.txt").into(), cx)
8257 .map(|e| e.id)
8258 );
8259 });
8260
8261 // Close the active item
8262 pane.update_in(cx, |pane, window, cx| {
8263 pane.close_active_item(&Default::default(), window, cx)
8264 })
8265 .await
8266 .unwrap();
8267 assert_eq!(cx.window_title().as_deref(), Some("root1 — one.txt"));
8268 project.update(cx, |project, cx| {
8269 assert_eq!(
8270 project.active_entry(),
8271 project
8272 .entry_for_path(&(worktree_id, "one.txt").into(), cx)
8273 .map(|e| e.id)
8274 );
8275 });
8276
8277 // Add a project folder
8278 project
8279 .update(cx, |project, cx| {
8280 project.find_or_create_worktree("root2", true, cx)
8281 })
8282 .await
8283 .unwrap();
8284 assert_eq!(cx.window_title().as_deref(), Some("root1, root2 — one.txt"));
8285
8286 // Remove a project folder
8287 project.update(cx, |project, cx| project.remove_worktree(worktree_id, cx));
8288 assert_eq!(cx.window_title().as_deref(), Some("root2 — one.txt"));
8289 }
8290
8291 #[gpui::test]
8292 async fn test_close_window(cx: &mut TestAppContext) {
8293 init_test(cx);
8294
8295 let fs = FakeFs::new(cx.executor());
8296 fs.insert_tree("/root", json!({ "one": "" })).await;
8297
8298 let project = Project::test(fs, ["root".as_ref()], cx).await;
8299 let (workspace, cx) =
8300 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
8301
8302 // When there are no dirty items, there's nothing to do.
8303 let item1 = cx.new(TestItem::new);
8304 workspace.update_in(cx, |w, window, cx| {
8305 w.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx)
8306 });
8307 let task = workspace.update_in(cx, |w, window, cx| {
8308 w.prepare_to_close(CloseIntent::CloseWindow, window, cx)
8309 });
8310 assert!(task.await.unwrap());
8311
8312 // When there are dirty untitled items, prompt to save each one. If the user
8313 // cancels any prompt, then abort.
8314 let item2 = cx.new(|cx| TestItem::new(cx).with_dirty(true));
8315 let item3 = cx.new(|cx| {
8316 TestItem::new(cx)
8317 .with_dirty(true)
8318 .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
8319 });
8320 workspace.update_in(cx, |w, window, cx| {
8321 w.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx);
8322 w.add_item_to_active_pane(Box::new(item3.clone()), None, true, window, cx);
8323 });
8324 let task = workspace.update_in(cx, |w, window, cx| {
8325 w.prepare_to_close(CloseIntent::CloseWindow, window, cx)
8326 });
8327 cx.executor().run_until_parked();
8328 cx.simulate_prompt_answer("Cancel"); // cancel save all
8329 cx.executor().run_until_parked();
8330 assert!(!cx.has_pending_prompt());
8331 assert!(!task.await.unwrap());
8332 }
8333
8334 #[gpui::test]
8335 async fn test_close_window_with_serializable_items(cx: &mut TestAppContext) {
8336 init_test(cx);
8337
8338 // Register TestItem as a serializable item
8339 cx.update(|cx| {
8340 register_serializable_item::<TestItem>(cx);
8341 });
8342
8343 let fs = FakeFs::new(cx.executor());
8344 fs.insert_tree("/root", json!({ "one": "" })).await;
8345
8346 let project = Project::test(fs, ["root".as_ref()], cx).await;
8347 let (workspace, cx) =
8348 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
8349
8350 // When there are dirty untitled items, but they can serialize, then there is no prompt.
8351 let item1 = cx.new(|cx| {
8352 TestItem::new(cx)
8353 .with_dirty(true)
8354 .with_serialize(|| Some(Task::ready(Ok(()))))
8355 });
8356 let item2 = cx.new(|cx| {
8357 TestItem::new(cx)
8358 .with_dirty(true)
8359 .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
8360 .with_serialize(|| Some(Task::ready(Ok(()))))
8361 });
8362 workspace.update_in(cx, |w, window, cx| {
8363 w.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx);
8364 w.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx);
8365 });
8366 let task = workspace.update_in(cx, |w, window, cx| {
8367 w.prepare_to_close(CloseIntent::CloseWindow, window, cx)
8368 });
8369 assert!(task.await.unwrap());
8370 }
8371
8372 #[gpui::test]
8373 async fn test_close_pane_items(cx: &mut TestAppContext) {
8374 init_test(cx);
8375
8376 let fs = FakeFs::new(cx.executor());
8377
8378 let project = Project::test(fs, None, cx).await;
8379 let (workspace, cx) =
8380 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
8381
8382 let item1 = cx.new(|cx| {
8383 TestItem::new(cx)
8384 .with_dirty(true)
8385 .with_project_items(&[dirty_project_item(1, "1.txt", cx)])
8386 });
8387 let item2 = cx.new(|cx| {
8388 TestItem::new(cx)
8389 .with_dirty(true)
8390 .with_conflict(true)
8391 .with_project_items(&[dirty_project_item(2, "2.txt", cx)])
8392 });
8393 let item3 = cx.new(|cx| {
8394 TestItem::new(cx)
8395 .with_dirty(true)
8396 .with_conflict(true)
8397 .with_project_items(&[dirty_project_item(3, "3.txt", cx)])
8398 });
8399 let item4 = cx.new(|cx| {
8400 TestItem::new(cx).with_dirty(true).with_project_items(&[{
8401 let project_item = TestProjectItem::new_untitled(cx);
8402 project_item.update(cx, |project_item, _| project_item.is_dirty = true);
8403 project_item
8404 }])
8405 });
8406 let pane = workspace.update_in(cx, |workspace, window, cx| {
8407 workspace.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx);
8408 workspace.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx);
8409 workspace.add_item_to_active_pane(Box::new(item3.clone()), None, true, window, cx);
8410 workspace.add_item_to_active_pane(Box::new(item4.clone()), None, true, window, cx);
8411 workspace.active_pane().clone()
8412 });
8413
8414 let close_items = pane.update_in(cx, |pane, window, cx| {
8415 pane.activate_item(1, true, true, window, cx);
8416 assert_eq!(pane.active_item().unwrap().item_id(), item2.item_id());
8417 let item1_id = item1.item_id();
8418 let item3_id = item3.item_id();
8419 let item4_id = item4.item_id();
8420 pane.close_items(window, cx, SaveIntent::Close, move |id| {
8421 [item1_id, item3_id, item4_id].contains(&id)
8422 })
8423 });
8424 cx.executor().run_until_parked();
8425
8426 assert!(cx.has_pending_prompt());
8427 cx.simulate_prompt_answer("Save all");
8428
8429 cx.executor().run_until_parked();
8430
8431 // Item 1 is saved. There's a prompt to save item 3.
8432 pane.update(cx, |pane, cx| {
8433 assert_eq!(item1.read(cx).save_count, 1);
8434 assert_eq!(item1.read(cx).save_as_count, 0);
8435 assert_eq!(item1.read(cx).reload_count, 0);
8436 assert_eq!(pane.items_len(), 3);
8437 assert_eq!(pane.active_item().unwrap().item_id(), item3.item_id());
8438 });
8439 assert!(cx.has_pending_prompt());
8440
8441 // Cancel saving item 3.
8442 cx.simulate_prompt_answer("Discard");
8443 cx.executor().run_until_parked();
8444
8445 // Item 3 is reloaded. There's a prompt to save item 4.
8446 pane.update(cx, |pane, cx| {
8447 assert_eq!(item3.read(cx).save_count, 0);
8448 assert_eq!(item3.read(cx).save_as_count, 0);
8449 assert_eq!(item3.read(cx).reload_count, 1);
8450 assert_eq!(pane.items_len(), 2);
8451 assert_eq!(pane.active_item().unwrap().item_id(), item4.item_id());
8452 });
8453
8454 // There's a prompt for a path for item 4.
8455 cx.simulate_new_path_selection(|_| Some(Default::default()));
8456 close_items.await.unwrap();
8457
8458 // The requested items are closed.
8459 pane.update(cx, |pane, cx| {
8460 assert_eq!(item4.read(cx).save_count, 0);
8461 assert_eq!(item4.read(cx).save_as_count, 1);
8462 assert_eq!(item4.read(cx).reload_count, 0);
8463 assert_eq!(pane.items_len(), 1);
8464 assert_eq!(pane.active_item().unwrap().item_id(), item2.item_id());
8465 });
8466 }
8467
8468 #[gpui::test]
8469 async fn test_prompting_to_save_only_on_last_item_for_entry(cx: &mut TestAppContext) {
8470 init_test(cx);
8471
8472 let fs = FakeFs::new(cx.executor());
8473 let project = Project::test(fs, [], cx).await;
8474 let (workspace, cx) =
8475 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
8476
8477 // Create several workspace items with single project entries, and two
8478 // workspace items with multiple project entries.
8479 let single_entry_items = (0..=4)
8480 .map(|project_entry_id| {
8481 cx.new(|cx| {
8482 TestItem::new(cx)
8483 .with_dirty(true)
8484 .with_project_items(&[dirty_project_item(
8485 project_entry_id,
8486 &format!("{project_entry_id}.txt"),
8487 cx,
8488 )])
8489 })
8490 })
8491 .collect::<Vec<_>>();
8492 let item_2_3 = cx.new(|cx| {
8493 TestItem::new(cx)
8494 .with_dirty(true)
8495 .with_singleton(false)
8496 .with_project_items(&[
8497 single_entry_items[2].read(cx).project_items[0].clone(),
8498 single_entry_items[3].read(cx).project_items[0].clone(),
8499 ])
8500 });
8501 let item_3_4 = cx.new(|cx| {
8502 TestItem::new(cx)
8503 .with_dirty(true)
8504 .with_singleton(false)
8505 .with_project_items(&[
8506 single_entry_items[3].read(cx).project_items[0].clone(),
8507 single_entry_items[4].read(cx).project_items[0].clone(),
8508 ])
8509 });
8510
8511 // Create two panes that contain the following project entries:
8512 // left pane:
8513 // multi-entry items: (2, 3)
8514 // single-entry items: 0, 2, 3, 4
8515 // right pane:
8516 // single-entry items: 4, 1
8517 // multi-entry items: (3, 4)
8518 let (left_pane, right_pane) = workspace.update_in(cx, |workspace, window, cx| {
8519 let left_pane = workspace.active_pane().clone();
8520 workspace.add_item_to_active_pane(Box::new(item_2_3.clone()), None, true, window, cx);
8521 workspace.add_item_to_active_pane(
8522 single_entry_items[0].boxed_clone(),
8523 None,
8524 true,
8525 window,
8526 cx,
8527 );
8528 workspace.add_item_to_active_pane(
8529 single_entry_items[2].boxed_clone(),
8530 None,
8531 true,
8532 window,
8533 cx,
8534 );
8535 workspace.add_item_to_active_pane(
8536 single_entry_items[3].boxed_clone(),
8537 None,
8538 true,
8539 window,
8540 cx,
8541 );
8542 workspace.add_item_to_active_pane(
8543 single_entry_items[4].boxed_clone(),
8544 None,
8545 true,
8546 window,
8547 cx,
8548 );
8549
8550 let right_pane = workspace
8551 .split_and_clone(left_pane.clone(), SplitDirection::Right, window, cx)
8552 .unwrap();
8553
8554 right_pane.update(cx, |pane, cx| {
8555 pane.add_item(
8556 single_entry_items[1].boxed_clone(),
8557 true,
8558 true,
8559 None,
8560 window,
8561 cx,
8562 );
8563 pane.add_item(Box::new(item_3_4.clone()), true, true, None, window, cx);
8564 });
8565
8566 (left_pane, right_pane)
8567 });
8568
8569 cx.focus(&right_pane);
8570
8571 let mut close = right_pane.update_in(cx, |pane, window, cx| {
8572 pane.close_all_items(&CloseAllItems::default(), window, cx)
8573 .unwrap()
8574 });
8575 cx.executor().run_until_parked();
8576
8577 let msg = cx.pending_prompt().unwrap().0;
8578 assert!(msg.contains("1.txt"));
8579 assert!(!msg.contains("2.txt"));
8580 assert!(!msg.contains("3.txt"));
8581 assert!(!msg.contains("4.txt"));
8582
8583 cx.simulate_prompt_answer("Cancel");
8584 close.await;
8585
8586 left_pane
8587 .update_in(cx, |left_pane, window, cx| {
8588 left_pane.close_item_by_id(
8589 single_entry_items[3].entity_id(),
8590 SaveIntent::Skip,
8591 window,
8592 cx,
8593 )
8594 })
8595 .await
8596 .unwrap();
8597
8598 close = right_pane.update_in(cx, |pane, window, cx| {
8599 pane.close_all_items(&CloseAllItems::default(), window, cx)
8600 .unwrap()
8601 });
8602 cx.executor().run_until_parked();
8603
8604 let details = cx.pending_prompt().unwrap().1;
8605 assert!(details.contains("1.txt"));
8606 assert!(!details.contains("2.txt"));
8607 assert!(details.contains("3.txt"));
8608 // ideally this assertion could be made, but today we can only
8609 // save whole items not project items, so the orphaned item 3 causes
8610 // 4 to be saved too.
8611 // assert!(!details.contains("4.txt"));
8612
8613 cx.simulate_prompt_answer("Save all");
8614
8615 cx.executor().run_until_parked();
8616 close.await;
8617 right_pane.read_with(cx, |pane, _| {
8618 assert_eq!(pane.items_len(), 0);
8619 });
8620 }
8621
8622 #[gpui::test]
8623 async fn test_autosave(cx: &mut gpui::TestAppContext) {
8624 init_test(cx);
8625
8626 let fs = FakeFs::new(cx.executor());
8627 let project = Project::test(fs, [], cx).await;
8628 let (workspace, cx) =
8629 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
8630 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
8631
8632 let item = cx.new(|cx| {
8633 TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
8634 });
8635 let item_id = item.entity_id();
8636 workspace.update_in(cx, |workspace, window, cx| {
8637 workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx);
8638 });
8639
8640 // Autosave on window change.
8641 item.update(cx, |item, cx| {
8642 SettingsStore::update_global(cx, |settings, cx| {
8643 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
8644 settings.autosave = Some(AutosaveSetting::OnWindowChange);
8645 })
8646 });
8647 item.is_dirty = true;
8648 });
8649
8650 // Deactivating the window saves the file.
8651 cx.deactivate_window();
8652 item.read_with(cx, |item, _| assert_eq!(item.save_count, 1));
8653
8654 // Re-activating the window doesn't save the file.
8655 cx.update(|window, _| window.activate_window());
8656 cx.executor().run_until_parked();
8657 item.read_with(cx, |item, _| assert_eq!(item.save_count, 1));
8658
8659 // Autosave on focus change.
8660 item.update_in(cx, |item, window, cx| {
8661 cx.focus_self(window);
8662 SettingsStore::update_global(cx, |settings, cx| {
8663 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
8664 settings.autosave = Some(AutosaveSetting::OnFocusChange);
8665 })
8666 });
8667 item.is_dirty = true;
8668 });
8669
8670 // Blurring the item saves the file.
8671 item.update_in(cx, |_, window, _| window.blur());
8672 cx.executor().run_until_parked();
8673 item.read_with(cx, |item, _| assert_eq!(item.save_count, 2));
8674
8675 // Deactivating the window still saves the file.
8676 item.update_in(cx, |item, window, cx| {
8677 cx.focus_self(window);
8678 item.is_dirty = true;
8679 });
8680 cx.deactivate_window();
8681 item.update(cx, |item, _| assert_eq!(item.save_count, 3));
8682
8683 // Autosave after delay.
8684 item.update(cx, |item, cx| {
8685 SettingsStore::update_global(cx, |settings, cx| {
8686 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
8687 settings.autosave = Some(AutosaveSetting::AfterDelay { milliseconds: 500 });
8688 })
8689 });
8690 item.is_dirty = true;
8691 cx.emit(ItemEvent::Edit);
8692 });
8693
8694 // Delay hasn't fully expired, so the file is still dirty and unsaved.
8695 cx.executor().advance_clock(Duration::from_millis(250));
8696 item.read_with(cx, |item, _| assert_eq!(item.save_count, 3));
8697
8698 // After delay expires, the file is saved.
8699 cx.executor().advance_clock(Duration::from_millis(250));
8700 item.read_with(cx, |item, _| assert_eq!(item.save_count, 4));
8701
8702 // Autosave on focus change, ensuring closing the tab counts as such.
8703 item.update(cx, |item, cx| {
8704 SettingsStore::update_global(cx, |settings, cx| {
8705 settings.update_user_settings::<WorkspaceSettings>(cx, |settings| {
8706 settings.autosave = Some(AutosaveSetting::OnFocusChange);
8707 })
8708 });
8709 item.is_dirty = true;
8710 for project_item in &mut item.project_items {
8711 project_item.update(cx, |project_item, _| project_item.is_dirty = true);
8712 }
8713 });
8714
8715 pane.update_in(cx, |pane, window, cx| {
8716 pane.close_items(window, cx, SaveIntent::Close, move |id| id == item_id)
8717 })
8718 .await
8719 .unwrap();
8720 assert!(!cx.has_pending_prompt());
8721 item.read_with(cx, |item, _| assert_eq!(item.save_count, 5));
8722
8723 // Add the item again, ensuring autosave is prevented if the underlying file has been deleted.
8724 workspace.update_in(cx, |workspace, window, cx| {
8725 workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx);
8726 });
8727 item.update_in(cx, |item, window, cx| {
8728 item.project_items[0].update(cx, |item, _| {
8729 item.entry_id = None;
8730 });
8731 item.is_dirty = true;
8732 window.blur();
8733 });
8734 cx.run_until_parked();
8735 item.read_with(cx, |item, _| assert_eq!(item.save_count, 5));
8736
8737 // Ensure autosave is prevented for deleted files also when closing the buffer.
8738 let _close_items = pane.update_in(cx, |pane, window, cx| {
8739 pane.close_items(window, cx, SaveIntent::Close, move |id| id == item_id)
8740 });
8741 cx.run_until_parked();
8742 assert!(cx.has_pending_prompt());
8743 item.read_with(cx, |item, _| assert_eq!(item.save_count, 5));
8744 }
8745
8746 #[gpui::test]
8747 async fn test_pane_navigation(cx: &mut gpui::TestAppContext) {
8748 init_test(cx);
8749
8750 let fs = FakeFs::new(cx.executor());
8751
8752 let project = Project::test(fs, [], cx).await;
8753 let (workspace, cx) =
8754 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
8755
8756 let item = cx.new(|cx| {
8757 TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "1.txt", cx)])
8758 });
8759 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
8760 let toolbar = pane.read_with(cx, |pane, _| pane.toolbar().clone());
8761 let toolbar_notify_count = Rc::new(RefCell::new(0));
8762
8763 workspace.update_in(cx, |workspace, window, cx| {
8764 workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx);
8765 let toolbar_notification_count = toolbar_notify_count.clone();
8766 cx.observe_in(&toolbar, window, move |_, _, _, _| {
8767 *toolbar_notification_count.borrow_mut() += 1
8768 })
8769 .detach();
8770 });
8771
8772 pane.read_with(cx, |pane, _| {
8773 assert!(!pane.can_navigate_backward());
8774 assert!(!pane.can_navigate_forward());
8775 });
8776
8777 item.update_in(cx, |item, _, cx| {
8778 item.set_state("one".to_string(), cx);
8779 });
8780
8781 // Toolbar must be notified to re-render the navigation buttons
8782 assert_eq!(*toolbar_notify_count.borrow(), 1);
8783
8784 pane.read_with(cx, |pane, _| {
8785 assert!(pane.can_navigate_backward());
8786 assert!(!pane.can_navigate_forward());
8787 });
8788
8789 workspace
8790 .update_in(cx, |workspace, window, cx| {
8791 workspace.go_back(pane.downgrade(), window, cx)
8792 })
8793 .await
8794 .unwrap();
8795
8796 assert_eq!(*toolbar_notify_count.borrow(), 2);
8797 pane.read_with(cx, |pane, _| {
8798 assert!(!pane.can_navigate_backward());
8799 assert!(pane.can_navigate_forward());
8800 });
8801 }
8802
8803 #[gpui::test]
8804 async fn test_toggle_docks_and_panels(cx: &mut gpui::TestAppContext) {
8805 init_test(cx);
8806 let fs = FakeFs::new(cx.executor());
8807
8808 let project = Project::test(fs, [], cx).await;
8809 let (workspace, cx) =
8810 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
8811
8812 let panel = workspace.update_in(cx, |workspace, window, cx| {
8813 let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, cx));
8814 workspace.add_panel(panel.clone(), window, cx);
8815
8816 workspace
8817 .right_dock()
8818 .update(cx, |right_dock, cx| right_dock.set_open(true, window, cx));
8819
8820 panel
8821 });
8822
8823 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
8824 pane.update_in(cx, |pane, window, cx| {
8825 let item = cx.new(TestItem::new);
8826 pane.add_item(Box::new(item), true, true, None, window, cx);
8827 });
8828
8829 // Transfer focus from center to panel
8830 workspace.update_in(cx, |workspace, window, cx| {
8831 workspace.toggle_panel_focus::<TestPanel>(window, cx);
8832 });
8833
8834 workspace.update_in(cx, |workspace, window, cx| {
8835 assert!(workspace.right_dock().read(cx).is_open());
8836 assert!(!panel.is_zoomed(window, cx));
8837 assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8838 });
8839
8840 // Transfer focus from panel to center
8841 workspace.update_in(cx, |workspace, window, cx| {
8842 workspace.toggle_panel_focus::<TestPanel>(window, cx);
8843 });
8844
8845 workspace.update_in(cx, |workspace, window, cx| {
8846 assert!(workspace.right_dock().read(cx).is_open());
8847 assert!(!panel.is_zoomed(window, cx));
8848 assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8849 });
8850
8851 // Close the dock
8852 workspace.update_in(cx, |workspace, window, cx| {
8853 workspace.toggle_dock(DockPosition::Right, window, cx);
8854 });
8855
8856 workspace.update_in(cx, |workspace, window, cx| {
8857 assert!(!workspace.right_dock().read(cx).is_open());
8858 assert!(!panel.is_zoomed(window, cx));
8859 assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8860 });
8861
8862 // Open the dock
8863 workspace.update_in(cx, |workspace, window, cx| {
8864 workspace.toggle_dock(DockPosition::Right, window, cx);
8865 });
8866
8867 workspace.update_in(cx, |workspace, window, cx| {
8868 assert!(workspace.right_dock().read(cx).is_open());
8869 assert!(!panel.is_zoomed(window, cx));
8870 assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8871 });
8872
8873 // Focus and zoom panel
8874 panel.update_in(cx, |panel, window, cx| {
8875 cx.focus_self(window);
8876 panel.set_zoomed(true, window, cx)
8877 });
8878
8879 workspace.update_in(cx, |workspace, window, cx| {
8880 assert!(workspace.right_dock().read(cx).is_open());
8881 assert!(panel.is_zoomed(window, cx));
8882 assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8883 });
8884
8885 // Transfer focus to the center closes the dock
8886 workspace.update_in(cx, |workspace, window, cx| {
8887 workspace.toggle_panel_focus::<TestPanel>(window, cx);
8888 });
8889
8890 workspace.update_in(cx, |workspace, window, cx| {
8891 assert!(!workspace.right_dock().read(cx).is_open());
8892 assert!(panel.is_zoomed(window, cx));
8893 assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8894 });
8895
8896 // Transferring focus back to the panel keeps it zoomed
8897 workspace.update_in(cx, |workspace, window, cx| {
8898 workspace.toggle_panel_focus::<TestPanel>(window, cx);
8899 });
8900
8901 workspace.update_in(cx, |workspace, window, cx| {
8902 assert!(workspace.right_dock().read(cx).is_open());
8903 assert!(panel.is_zoomed(window, cx));
8904 assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8905 });
8906
8907 // Close the dock while it is zoomed
8908 workspace.update_in(cx, |workspace, window, cx| {
8909 workspace.toggle_dock(DockPosition::Right, window, cx)
8910 });
8911
8912 workspace.update_in(cx, |workspace, window, cx| {
8913 assert!(!workspace.right_dock().read(cx).is_open());
8914 assert!(panel.is_zoomed(window, cx));
8915 assert!(workspace.zoomed.is_none());
8916 assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8917 });
8918
8919 // Opening the dock, when it's zoomed, retains focus
8920 workspace.update_in(cx, |workspace, window, cx| {
8921 workspace.toggle_dock(DockPosition::Right, window, cx)
8922 });
8923
8924 workspace.update_in(cx, |workspace, window, cx| {
8925 assert!(workspace.right_dock().read(cx).is_open());
8926 assert!(panel.is_zoomed(window, cx));
8927 assert!(workspace.zoomed.is_some());
8928 assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx));
8929 });
8930
8931 // Unzoom and close the panel, zoom the active pane.
8932 panel.update_in(cx, |panel, window, cx| panel.set_zoomed(false, window, cx));
8933 workspace.update_in(cx, |workspace, window, cx| {
8934 workspace.toggle_dock(DockPosition::Right, window, cx)
8935 });
8936 pane.update_in(cx, |pane, window, cx| {
8937 pane.toggle_zoom(&Default::default(), window, cx)
8938 });
8939
8940 // Opening a dock unzooms the pane.
8941 workspace.update_in(cx, |workspace, window, cx| {
8942 workspace.toggle_dock(DockPosition::Right, window, cx)
8943 });
8944 workspace.update_in(cx, |workspace, window, cx| {
8945 let pane = pane.read(cx);
8946 assert!(!pane.is_zoomed());
8947 assert!(!pane.focus_handle(cx).is_focused(window));
8948 assert!(workspace.right_dock().read(cx).is_open());
8949 assert!(workspace.zoomed.is_none());
8950 });
8951 }
8952
8953 #[gpui::test]
8954 async fn test_join_pane_into_next(cx: &mut gpui::TestAppContext) {
8955 init_test(cx);
8956
8957 let fs = FakeFs::new(cx.executor());
8958
8959 let project = Project::test(fs, None, cx).await;
8960 let (workspace, cx) =
8961 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
8962
8963 // Let's arrange the panes like this:
8964 //
8965 // +-----------------------+
8966 // | top |
8967 // +------+--------+-------+
8968 // | left | center | right |
8969 // +------+--------+-------+
8970 // | bottom |
8971 // +-----------------------+
8972
8973 let top_item = cx.new(|cx| {
8974 TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "top.txt", cx)])
8975 });
8976 let bottom_item = cx.new(|cx| {
8977 TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "bottom.txt", cx)])
8978 });
8979 let left_item = cx.new(|cx| {
8980 TestItem::new(cx).with_project_items(&[TestProjectItem::new(3, "left.txt", cx)])
8981 });
8982 let right_item = cx.new(|cx| {
8983 TestItem::new(cx).with_project_items(&[TestProjectItem::new(4, "right.txt", cx)])
8984 });
8985 let center_item = cx.new(|cx| {
8986 TestItem::new(cx).with_project_items(&[TestProjectItem::new(5, "center.txt", cx)])
8987 });
8988
8989 let top_pane_id = workspace.update_in(cx, |workspace, window, cx| {
8990 let top_pane_id = workspace.active_pane().entity_id();
8991 workspace.add_item_to_active_pane(Box::new(top_item.clone()), None, false, window, cx);
8992 workspace.split_pane(
8993 workspace.active_pane().clone(),
8994 SplitDirection::Down,
8995 window,
8996 cx,
8997 );
8998 top_pane_id
8999 });
9000 let bottom_pane_id = workspace.update_in(cx, |workspace, window, cx| {
9001 let bottom_pane_id = workspace.active_pane().entity_id();
9002 workspace.add_item_to_active_pane(
9003 Box::new(bottom_item.clone()),
9004 None,
9005 false,
9006 window,
9007 cx,
9008 );
9009 workspace.split_pane(
9010 workspace.active_pane().clone(),
9011 SplitDirection::Up,
9012 window,
9013 cx,
9014 );
9015 bottom_pane_id
9016 });
9017 let left_pane_id = workspace.update_in(cx, |workspace, window, cx| {
9018 let left_pane_id = workspace.active_pane().entity_id();
9019 workspace.add_item_to_active_pane(Box::new(left_item.clone()), None, false, window, cx);
9020 workspace.split_pane(
9021 workspace.active_pane().clone(),
9022 SplitDirection::Right,
9023 window,
9024 cx,
9025 );
9026 left_pane_id
9027 });
9028 let right_pane_id = workspace.update_in(cx, |workspace, window, cx| {
9029 let right_pane_id = workspace.active_pane().entity_id();
9030 workspace.add_item_to_active_pane(
9031 Box::new(right_item.clone()),
9032 None,
9033 false,
9034 window,
9035 cx,
9036 );
9037 workspace.split_pane(
9038 workspace.active_pane().clone(),
9039 SplitDirection::Left,
9040 window,
9041 cx,
9042 );
9043 right_pane_id
9044 });
9045 let center_pane_id = workspace.update_in(cx, |workspace, window, cx| {
9046 let center_pane_id = workspace.active_pane().entity_id();
9047 workspace.add_item_to_active_pane(
9048 Box::new(center_item.clone()),
9049 None,
9050 false,
9051 window,
9052 cx,
9053 );
9054 center_pane_id
9055 });
9056 cx.executor().run_until_parked();
9057
9058 workspace.update_in(cx, |workspace, window, cx| {
9059 assert_eq!(center_pane_id, workspace.active_pane().entity_id());
9060
9061 // Join into next from center pane into right
9062 workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx);
9063 });
9064
9065 workspace.update_in(cx, |workspace, window, cx| {
9066 let active_pane = workspace.active_pane();
9067 assert_eq!(right_pane_id, active_pane.entity_id());
9068 assert_eq!(2, active_pane.read(cx).items_len());
9069 let item_ids_in_pane =
9070 HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id()));
9071 assert!(item_ids_in_pane.contains(¢er_item.item_id()));
9072 assert!(item_ids_in_pane.contains(&right_item.item_id()));
9073
9074 // Join into next from right pane into bottom
9075 workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx);
9076 });
9077
9078 workspace.update_in(cx, |workspace, window, cx| {
9079 let active_pane = workspace.active_pane();
9080 assert_eq!(bottom_pane_id, active_pane.entity_id());
9081 assert_eq!(3, active_pane.read(cx).items_len());
9082 let item_ids_in_pane =
9083 HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id()));
9084 assert!(item_ids_in_pane.contains(¢er_item.item_id()));
9085 assert!(item_ids_in_pane.contains(&right_item.item_id()));
9086 assert!(item_ids_in_pane.contains(&bottom_item.item_id()));
9087
9088 // Join into next from bottom pane into left
9089 workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx);
9090 });
9091
9092 workspace.update_in(cx, |workspace, window, cx| {
9093 let active_pane = workspace.active_pane();
9094 assert_eq!(left_pane_id, active_pane.entity_id());
9095 assert_eq!(4, active_pane.read(cx).items_len());
9096 let item_ids_in_pane =
9097 HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id()));
9098 assert!(item_ids_in_pane.contains(¢er_item.item_id()));
9099 assert!(item_ids_in_pane.contains(&right_item.item_id()));
9100 assert!(item_ids_in_pane.contains(&bottom_item.item_id()));
9101 assert!(item_ids_in_pane.contains(&left_item.item_id()));
9102
9103 // Join into next from left pane into top
9104 workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx);
9105 });
9106
9107 workspace.update_in(cx, |workspace, window, cx| {
9108 let active_pane = workspace.active_pane();
9109 assert_eq!(top_pane_id, active_pane.entity_id());
9110 assert_eq!(5, active_pane.read(cx).items_len());
9111 let item_ids_in_pane =
9112 HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id()));
9113 assert!(item_ids_in_pane.contains(¢er_item.item_id()));
9114 assert!(item_ids_in_pane.contains(&right_item.item_id()));
9115 assert!(item_ids_in_pane.contains(&bottom_item.item_id()));
9116 assert!(item_ids_in_pane.contains(&left_item.item_id()));
9117 assert!(item_ids_in_pane.contains(&top_item.item_id()));
9118
9119 // Single pane left: no-op
9120 workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx)
9121 });
9122
9123 workspace.update(cx, |workspace, _cx| {
9124 let active_pane = workspace.active_pane();
9125 assert_eq!(top_pane_id, active_pane.entity_id());
9126 });
9127 }
9128
9129 fn add_an_item_to_active_pane(
9130 cx: &mut VisualTestContext,
9131 workspace: &Entity<Workspace>,
9132 item_id: u64,
9133 ) -> Entity<TestItem> {
9134 let item = cx.new(|cx| {
9135 TestItem::new(cx).with_project_items(&[TestProjectItem::new(
9136 item_id,
9137 "item{item_id}.txt",
9138 cx,
9139 )])
9140 });
9141 workspace.update_in(cx, |workspace, window, cx| {
9142 workspace.add_item_to_active_pane(Box::new(item.clone()), None, false, window, cx);
9143 });
9144 item
9145 }
9146
9147 fn split_pane(cx: &mut VisualTestContext, workspace: &Entity<Workspace>) -> Entity<Pane> {
9148 workspace.update_in(cx, |workspace, window, cx| {
9149 workspace.split_pane(
9150 workspace.active_pane().clone(),
9151 SplitDirection::Right,
9152 window,
9153 cx,
9154 )
9155 })
9156 }
9157
9158 #[gpui::test]
9159 async fn test_join_all_panes(cx: &mut gpui::TestAppContext) {
9160 init_test(cx);
9161 let fs = FakeFs::new(cx.executor());
9162 let project = Project::test(fs, None, cx).await;
9163 let (workspace, cx) =
9164 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9165
9166 add_an_item_to_active_pane(cx, &workspace, 1);
9167 split_pane(cx, &workspace);
9168 add_an_item_to_active_pane(cx, &workspace, 2);
9169 split_pane(cx, &workspace); // empty pane
9170 split_pane(cx, &workspace);
9171 let last_item = add_an_item_to_active_pane(cx, &workspace, 3);
9172
9173 cx.executor().run_until_parked();
9174
9175 workspace.update(cx, |workspace, cx| {
9176 let num_panes = workspace.panes().len();
9177 let num_items_in_current_pane = workspace.active_pane().read(cx).items().count();
9178 let active_item = workspace
9179 .active_pane()
9180 .read(cx)
9181 .active_item()
9182 .expect("item is in focus");
9183
9184 assert_eq!(num_panes, 4);
9185 assert_eq!(num_items_in_current_pane, 1);
9186 assert_eq!(active_item.item_id(), last_item.item_id());
9187 });
9188
9189 workspace.update_in(cx, |workspace, window, cx| {
9190 workspace.join_all_panes(window, cx);
9191 });
9192
9193 workspace.update(cx, |workspace, cx| {
9194 let num_panes = workspace.panes().len();
9195 let num_items_in_current_pane = workspace.active_pane().read(cx).items().count();
9196 let active_item = workspace
9197 .active_pane()
9198 .read(cx)
9199 .active_item()
9200 .expect("item is in focus");
9201
9202 assert_eq!(num_panes, 1);
9203 assert_eq!(num_items_in_current_pane, 3);
9204 assert_eq!(active_item.item_id(), last_item.item_id());
9205 });
9206 }
9207 struct TestModal(FocusHandle);
9208
9209 impl TestModal {
9210 fn new(_: &mut Window, cx: &mut Context<Self>) -> Self {
9211 Self(cx.focus_handle())
9212 }
9213 }
9214
9215 impl EventEmitter<DismissEvent> for TestModal {}
9216
9217 impl Focusable for TestModal {
9218 fn focus_handle(&self, _cx: &App) -> FocusHandle {
9219 self.0.clone()
9220 }
9221 }
9222
9223 impl ModalView for TestModal {}
9224
9225 impl Render for TestModal {
9226 fn render(
9227 &mut self,
9228 _window: &mut Window,
9229 _cx: &mut Context<TestModal>,
9230 ) -> impl IntoElement {
9231 div().track_focus(&self.0)
9232 }
9233 }
9234
9235 #[gpui::test]
9236 async fn test_panels(cx: &mut gpui::TestAppContext) {
9237 init_test(cx);
9238 let fs = FakeFs::new(cx.executor());
9239
9240 let project = Project::test(fs, [], cx).await;
9241 let (workspace, cx) =
9242 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9243
9244 let (panel_1, panel_2) = workspace.update_in(cx, |workspace, window, cx| {
9245 let panel_1 = cx.new(|cx| TestPanel::new(DockPosition::Left, cx));
9246 workspace.add_panel(panel_1.clone(), window, cx);
9247 workspace.toggle_dock(DockPosition::Left, window, cx);
9248 let panel_2 = cx.new(|cx| TestPanel::new(DockPosition::Right, cx));
9249 workspace.add_panel(panel_2.clone(), window, cx);
9250 workspace.toggle_dock(DockPosition::Right, window, cx);
9251
9252 let left_dock = workspace.left_dock();
9253 assert_eq!(
9254 left_dock.read(cx).visible_panel().unwrap().panel_id(),
9255 panel_1.panel_id()
9256 );
9257 assert_eq!(
9258 left_dock.read(cx).active_panel_size(window, cx).unwrap(),
9259 panel_1.size(window, cx)
9260 );
9261
9262 left_dock.update(cx, |left_dock, cx| {
9263 left_dock.resize_active_panel(Some(px(1337.)), window, cx)
9264 });
9265 assert_eq!(
9266 workspace
9267 .right_dock()
9268 .read(cx)
9269 .visible_panel()
9270 .unwrap()
9271 .panel_id(),
9272 panel_2.panel_id(),
9273 );
9274
9275 (panel_1, panel_2)
9276 });
9277
9278 // Move panel_1 to the right
9279 panel_1.update_in(cx, |panel_1, window, cx| {
9280 panel_1.set_position(DockPosition::Right, window, cx)
9281 });
9282
9283 workspace.update_in(cx, |workspace, window, cx| {
9284 // Since panel_1 was visible on the left, it should now be visible now that it's been moved to the right.
9285 // Since it was the only panel on the left, the left dock should now be closed.
9286 assert!(!workspace.left_dock().read(cx).is_open());
9287 assert!(workspace.left_dock().read(cx).visible_panel().is_none());
9288 let right_dock = workspace.right_dock();
9289 assert_eq!(
9290 right_dock.read(cx).visible_panel().unwrap().panel_id(),
9291 panel_1.panel_id()
9292 );
9293 assert_eq!(
9294 right_dock.read(cx).active_panel_size(window, cx).unwrap(),
9295 px(1337.)
9296 );
9297
9298 // Now we move panel_2 to the left
9299 panel_2.set_position(DockPosition::Left, window, cx);
9300 });
9301
9302 workspace.update(cx, |workspace, cx| {
9303 // Since panel_2 was not visible on the right, we don't open the left dock.
9304 assert!(!workspace.left_dock().read(cx).is_open());
9305 // And the right dock is unaffected in its displaying of panel_1
9306 assert!(workspace.right_dock().read(cx).is_open());
9307 assert_eq!(
9308 workspace
9309 .right_dock()
9310 .read(cx)
9311 .visible_panel()
9312 .unwrap()
9313 .panel_id(),
9314 panel_1.panel_id(),
9315 );
9316 });
9317
9318 // Move panel_1 back to the left
9319 panel_1.update_in(cx, |panel_1, window, cx| {
9320 panel_1.set_position(DockPosition::Left, window, cx)
9321 });
9322
9323 workspace.update_in(cx, |workspace, window, cx| {
9324 // Since panel_1 was visible on the right, we open the left dock and make panel_1 active.
9325 let left_dock = workspace.left_dock();
9326 assert!(left_dock.read(cx).is_open());
9327 assert_eq!(
9328 left_dock.read(cx).visible_panel().unwrap().panel_id(),
9329 panel_1.panel_id()
9330 );
9331 assert_eq!(
9332 left_dock.read(cx).active_panel_size(window, cx).unwrap(),
9333 px(1337.)
9334 );
9335 // And the right dock should be closed as it no longer has any panels.
9336 assert!(!workspace.right_dock().read(cx).is_open());
9337
9338 // Now we move panel_1 to the bottom
9339 panel_1.set_position(DockPosition::Bottom, window, cx);
9340 });
9341
9342 workspace.update_in(cx, |workspace, window, cx| {
9343 // Since panel_1 was visible on the left, we close the left dock.
9344 assert!(!workspace.left_dock().read(cx).is_open());
9345 // The bottom dock is sized based on the panel's default size,
9346 // since the panel orientation changed from vertical to horizontal.
9347 let bottom_dock = workspace.bottom_dock();
9348 assert_eq!(
9349 bottom_dock.read(cx).active_panel_size(window, cx).unwrap(),
9350 panel_1.size(window, cx),
9351 );
9352 // Close bottom dock and move panel_1 back to the left.
9353 bottom_dock.update(cx, |bottom_dock, cx| {
9354 bottom_dock.set_open(false, window, cx)
9355 });
9356 panel_1.set_position(DockPosition::Left, window, cx);
9357 });
9358
9359 // Emit activated event on panel 1
9360 panel_1.update(cx, |_, cx| cx.emit(PanelEvent::Activate));
9361
9362 // Now the left dock is open and panel_1 is active and focused.
9363 workspace.update_in(cx, |workspace, window, cx| {
9364 let left_dock = workspace.left_dock();
9365 assert!(left_dock.read(cx).is_open());
9366 assert_eq!(
9367 left_dock.read(cx).visible_panel().unwrap().panel_id(),
9368 panel_1.panel_id(),
9369 );
9370 assert!(panel_1.focus_handle(cx).is_focused(window));
9371 });
9372
9373 // Emit closed event on panel 2, which is not active
9374 panel_2.update(cx, |_, cx| cx.emit(PanelEvent::Close));
9375
9376 // Wo don't close the left dock, because panel_2 wasn't the active panel
9377 workspace.update(cx, |workspace, cx| {
9378 let left_dock = workspace.left_dock();
9379 assert!(left_dock.read(cx).is_open());
9380 assert_eq!(
9381 left_dock.read(cx).visible_panel().unwrap().panel_id(),
9382 panel_1.panel_id(),
9383 );
9384 });
9385
9386 // Emitting a ZoomIn event shows the panel as zoomed.
9387 panel_1.update(cx, |_, cx| cx.emit(PanelEvent::ZoomIn));
9388 workspace.read_with(cx, |workspace, _| {
9389 assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade()));
9390 assert_eq!(workspace.zoomed_position, Some(DockPosition::Left));
9391 });
9392
9393 // Move panel to another dock while it is zoomed
9394 panel_1.update_in(cx, |panel, window, cx| {
9395 panel.set_position(DockPosition::Right, window, cx)
9396 });
9397 workspace.read_with(cx, |workspace, _| {
9398 assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade()));
9399
9400 assert_eq!(workspace.zoomed_position, Some(DockPosition::Right));
9401 });
9402
9403 // This is a helper for getting a:
9404 // - valid focus on an element,
9405 // - that isn't a part of the panes and panels system of the Workspace,
9406 // - and doesn't trigger the 'on_focus_lost' API.
9407 let focus_other_view = {
9408 let workspace = workspace.clone();
9409 move |cx: &mut VisualTestContext| {
9410 workspace.update_in(cx, |workspace, window, cx| {
9411 if workspace.active_modal::<TestModal>(cx).is_some() {
9412 workspace.toggle_modal(window, cx, TestModal::new);
9413 workspace.toggle_modal(window, cx, TestModal::new);
9414 } else {
9415 workspace.toggle_modal(window, cx, TestModal::new);
9416 }
9417 })
9418 }
9419 };
9420
9421 // If focus is transferred to another view that's not a panel or another pane, we still show
9422 // the panel as zoomed.
9423 focus_other_view(cx);
9424 workspace.read_with(cx, |workspace, _| {
9425 assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade()));
9426 assert_eq!(workspace.zoomed_position, Some(DockPosition::Right));
9427 });
9428
9429 // If focus is transferred elsewhere in the workspace, the panel is no longer zoomed.
9430 workspace.update_in(cx, |_workspace, window, cx| {
9431 cx.focus_self(window);
9432 });
9433 workspace.read_with(cx, |workspace, _| {
9434 assert_eq!(workspace.zoomed, None);
9435 assert_eq!(workspace.zoomed_position, None);
9436 });
9437
9438 // If focus is transferred again to another view that's not a panel or a pane, we won't
9439 // show the panel as zoomed because it wasn't zoomed before.
9440 focus_other_view(cx);
9441 workspace.read_with(cx, |workspace, _| {
9442 assert_eq!(workspace.zoomed, None);
9443 assert_eq!(workspace.zoomed_position, None);
9444 });
9445
9446 // When the panel is activated, it is zoomed again.
9447 cx.dispatch_action(ToggleRightDock);
9448 workspace.read_with(cx, |workspace, _| {
9449 assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade()));
9450 assert_eq!(workspace.zoomed_position, Some(DockPosition::Right));
9451 });
9452
9453 // Emitting a ZoomOut event unzooms the panel.
9454 panel_1.update(cx, |_, cx| cx.emit(PanelEvent::ZoomOut));
9455 workspace.read_with(cx, |workspace, _| {
9456 assert_eq!(workspace.zoomed, None);
9457 assert_eq!(workspace.zoomed_position, None);
9458 });
9459
9460 // Emit closed event on panel 1, which is active
9461 panel_1.update(cx, |_, cx| cx.emit(PanelEvent::Close));
9462
9463 // Now the left dock is closed, because panel_1 was the active panel
9464 workspace.update(cx, |workspace, cx| {
9465 let right_dock = workspace.right_dock();
9466 assert!(!right_dock.read(cx).is_open());
9467 });
9468 }
9469
9470 #[gpui::test]
9471 async fn test_no_save_prompt_when_multi_buffer_dirty_items_closed(cx: &mut TestAppContext) {
9472 init_test(cx);
9473
9474 let fs = FakeFs::new(cx.background_executor.clone());
9475 let project = Project::test(fs, [], cx).await;
9476 let (workspace, cx) =
9477 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9478 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
9479
9480 let dirty_regular_buffer = cx.new(|cx| {
9481 TestItem::new(cx)
9482 .with_dirty(true)
9483 .with_label("1.txt")
9484 .with_project_items(&[dirty_project_item(1, "1.txt", cx)])
9485 });
9486 let dirty_regular_buffer_2 = cx.new(|cx| {
9487 TestItem::new(cx)
9488 .with_dirty(true)
9489 .with_label("2.txt")
9490 .with_project_items(&[dirty_project_item(2, "2.txt", cx)])
9491 });
9492 let dirty_multi_buffer_with_both = cx.new(|cx| {
9493 TestItem::new(cx)
9494 .with_dirty(true)
9495 .with_singleton(false)
9496 .with_label("Fake Project Search")
9497 .with_project_items(&[
9498 dirty_regular_buffer.read(cx).project_items[0].clone(),
9499 dirty_regular_buffer_2.read(cx).project_items[0].clone(),
9500 ])
9501 });
9502 let multi_buffer_with_both_files_id = dirty_multi_buffer_with_both.item_id();
9503 workspace.update_in(cx, |workspace, window, cx| {
9504 workspace.add_item(
9505 pane.clone(),
9506 Box::new(dirty_regular_buffer.clone()),
9507 None,
9508 false,
9509 false,
9510 window,
9511 cx,
9512 );
9513 workspace.add_item(
9514 pane.clone(),
9515 Box::new(dirty_regular_buffer_2.clone()),
9516 None,
9517 false,
9518 false,
9519 window,
9520 cx,
9521 );
9522 workspace.add_item(
9523 pane.clone(),
9524 Box::new(dirty_multi_buffer_with_both.clone()),
9525 None,
9526 false,
9527 false,
9528 window,
9529 cx,
9530 );
9531 });
9532
9533 pane.update_in(cx, |pane, window, cx| {
9534 pane.activate_item(2, true, true, window, cx);
9535 assert_eq!(
9536 pane.active_item().unwrap().item_id(),
9537 multi_buffer_with_both_files_id,
9538 "Should select the multi buffer in the pane"
9539 );
9540 });
9541 let close_all_but_multi_buffer_task = pane.update_in(cx, |pane, window, cx| {
9542 pane.close_other_items(
9543 &CloseOtherItems {
9544 save_intent: Some(SaveIntent::Save),
9545 close_pinned: true,
9546 },
9547 None,
9548 window,
9549 cx,
9550 )
9551 });
9552 cx.background_executor.run_until_parked();
9553 assert!(!cx.has_pending_prompt());
9554 close_all_but_multi_buffer_task
9555 .await
9556 .expect("Closing all buffers but the multi buffer failed");
9557 pane.update(cx, |pane, cx| {
9558 assert_eq!(dirty_regular_buffer.read(cx).save_count, 1);
9559 assert_eq!(dirty_multi_buffer_with_both.read(cx).save_count, 0);
9560 assert_eq!(dirty_regular_buffer_2.read(cx).save_count, 1);
9561 assert_eq!(pane.items_len(), 1);
9562 assert_eq!(
9563 pane.active_item().unwrap().item_id(),
9564 multi_buffer_with_both_files_id,
9565 "Should have only the multi buffer left in the pane"
9566 );
9567 assert!(
9568 dirty_multi_buffer_with_both.read(cx).is_dirty,
9569 "The multi buffer containing the unsaved buffer should still be dirty"
9570 );
9571 });
9572
9573 dirty_regular_buffer.update(cx, |buffer, cx| {
9574 buffer.project_items[0].update(cx, |pi, _| pi.is_dirty = true)
9575 });
9576
9577 let close_multi_buffer_task = pane.update_in(cx, |pane, window, cx| {
9578 pane.close_active_item(
9579 &CloseActiveItem {
9580 save_intent: Some(SaveIntent::Close),
9581 close_pinned: false,
9582 },
9583 window,
9584 cx,
9585 )
9586 });
9587 cx.background_executor.run_until_parked();
9588 assert!(
9589 cx.has_pending_prompt(),
9590 "Dirty multi buffer should prompt a save dialog"
9591 );
9592 cx.simulate_prompt_answer("Save");
9593 cx.background_executor.run_until_parked();
9594 close_multi_buffer_task
9595 .await
9596 .expect("Closing the multi buffer failed");
9597 pane.update(cx, |pane, cx| {
9598 assert_eq!(
9599 dirty_multi_buffer_with_both.read(cx).save_count,
9600 1,
9601 "Multi buffer item should get be saved"
9602 );
9603 // Test impl does not save inner items, so we do not assert them
9604 assert_eq!(
9605 pane.items_len(),
9606 0,
9607 "No more items should be left in the pane"
9608 );
9609 assert!(pane.active_item().is_none());
9610 });
9611 }
9612
9613 #[gpui::test]
9614 async fn test_save_prompt_when_dirty_multi_buffer_closed_with_some_of_its_dirty_items_not_present_in_the_pane(
9615 cx: &mut TestAppContext,
9616 ) {
9617 init_test(cx);
9618
9619 let fs = FakeFs::new(cx.background_executor.clone());
9620 let project = Project::test(fs, [], cx).await;
9621 let (workspace, cx) =
9622 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9623 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
9624
9625 let dirty_regular_buffer = cx.new(|cx| {
9626 TestItem::new(cx)
9627 .with_dirty(true)
9628 .with_label("1.txt")
9629 .with_project_items(&[dirty_project_item(1, "1.txt", cx)])
9630 });
9631 let dirty_regular_buffer_2 = cx.new(|cx| {
9632 TestItem::new(cx)
9633 .with_dirty(true)
9634 .with_label("2.txt")
9635 .with_project_items(&[dirty_project_item(2, "2.txt", cx)])
9636 });
9637 let clear_regular_buffer = cx.new(|cx| {
9638 TestItem::new(cx)
9639 .with_label("3.txt")
9640 .with_project_items(&[TestProjectItem::new(3, "3.txt", cx)])
9641 });
9642
9643 let dirty_multi_buffer_with_both = cx.new(|cx| {
9644 TestItem::new(cx)
9645 .with_dirty(true)
9646 .with_singleton(false)
9647 .with_label("Fake Project Search")
9648 .with_project_items(&[
9649 dirty_regular_buffer.read(cx).project_items[0].clone(),
9650 dirty_regular_buffer_2.read(cx).project_items[0].clone(),
9651 clear_regular_buffer.read(cx).project_items[0].clone(),
9652 ])
9653 });
9654 let multi_buffer_with_both_files_id = dirty_multi_buffer_with_both.item_id();
9655 workspace.update_in(cx, |workspace, window, cx| {
9656 workspace.add_item(
9657 pane.clone(),
9658 Box::new(dirty_regular_buffer.clone()),
9659 None,
9660 false,
9661 false,
9662 window,
9663 cx,
9664 );
9665 workspace.add_item(
9666 pane.clone(),
9667 Box::new(dirty_multi_buffer_with_both.clone()),
9668 None,
9669 false,
9670 false,
9671 window,
9672 cx,
9673 );
9674 });
9675
9676 pane.update_in(cx, |pane, window, cx| {
9677 pane.activate_item(1, true, true, window, cx);
9678 assert_eq!(
9679 pane.active_item().unwrap().item_id(),
9680 multi_buffer_with_both_files_id,
9681 "Should select the multi buffer in the pane"
9682 );
9683 });
9684 let _close_multi_buffer_task = pane.update_in(cx, |pane, window, cx| {
9685 pane.close_active_item(
9686 &CloseActiveItem {
9687 save_intent: None,
9688 close_pinned: false,
9689 },
9690 window,
9691 cx,
9692 )
9693 });
9694 cx.background_executor.run_until_parked();
9695 assert!(
9696 cx.has_pending_prompt(),
9697 "With one dirty item from the multi buffer not being in the pane, a save prompt should be shown"
9698 );
9699 }
9700
9701 /// Tests that when `close_on_file_delete` is enabled, files are automatically
9702 /// closed when they are deleted from disk.
9703 #[gpui::test]
9704 async fn test_close_on_disk_deletion_enabled(cx: &mut TestAppContext) {
9705 init_test(cx);
9706
9707 // Enable the close_on_disk_deletion setting
9708 cx.update_global(|store: &mut SettingsStore, cx| {
9709 store.update_user_settings::<WorkspaceSettings>(cx, |settings| {
9710 settings.close_on_file_delete = Some(true);
9711 });
9712 });
9713
9714 let fs = FakeFs::new(cx.background_executor.clone());
9715 let project = Project::test(fs, [], cx).await;
9716 let (workspace, cx) =
9717 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9718 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
9719
9720 // Create a test item that simulates a file
9721 let item = cx.new(|cx| {
9722 TestItem::new(cx)
9723 .with_label("test.txt")
9724 .with_project_items(&[TestProjectItem::new(1, "test.txt", cx)])
9725 });
9726
9727 // Add item to workspace
9728 workspace.update_in(cx, |workspace, window, cx| {
9729 workspace.add_item(
9730 pane.clone(),
9731 Box::new(item.clone()),
9732 None,
9733 false,
9734 false,
9735 window,
9736 cx,
9737 );
9738 });
9739
9740 // Verify the item is in the pane
9741 pane.read_with(cx, |pane, _| {
9742 assert_eq!(pane.items().count(), 1);
9743 });
9744
9745 // Simulate file deletion by setting the item's deleted state
9746 item.update(cx, |item, _| {
9747 item.set_has_deleted_file(true);
9748 });
9749
9750 // Emit UpdateTab event to trigger the close behavior
9751 cx.run_until_parked();
9752 item.update(cx, |_, cx| {
9753 cx.emit(ItemEvent::UpdateTab);
9754 });
9755
9756 // Allow the close operation to complete
9757 cx.run_until_parked();
9758
9759 // Verify the item was automatically closed
9760 pane.read_with(cx, |pane, _| {
9761 assert_eq!(
9762 pane.items().count(),
9763 0,
9764 "Item should be automatically closed when file is deleted"
9765 );
9766 });
9767 }
9768
9769 /// Tests that when `close_on_file_delete` is disabled (default), files remain
9770 /// open with a strikethrough when they are deleted from disk.
9771 #[gpui::test]
9772 async fn test_close_on_disk_deletion_disabled(cx: &mut TestAppContext) {
9773 init_test(cx);
9774
9775 // Ensure close_on_disk_deletion is disabled (default)
9776 cx.update_global(|store: &mut SettingsStore, cx| {
9777 store.update_user_settings::<WorkspaceSettings>(cx, |settings| {
9778 settings.close_on_file_delete = Some(false);
9779 });
9780 });
9781
9782 let fs = FakeFs::new(cx.background_executor.clone());
9783 let project = Project::test(fs, [], cx).await;
9784 let (workspace, cx) =
9785 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9786 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
9787
9788 // Create a test item that simulates a file
9789 let item = cx.new(|cx| {
9790 TestItem::new(cx)
9791 .with_label("test.txt")
9792 .with_project_items(&[TestProjectItem::new(1, "test.txt", cx)])
9793 });
9794
9795 // Add item to workspace
9796 workspace.update_in(cx, |workspace, window, cx| {
9797 workspace.add_item(
9798 pane.clone(),
9799 Box::new(item.clone()),
9800 None,
9801 false,
9802 false,
9803 window,
9804 cx,
9805 );
9806 });
9807
9808 // Verify the item is in the pane
9809 pane.read_with(cx, |pane, _| {
9810 assert_eq!(pane.items().count(), 1);
9811 });
9812
9813 // Simulate file deletion
9814 item.update(cx, |item, _| {
9815 item.set_has_deleted_file(true);
9816 });
9817
9818 // Emit UpdateTab event
9819 cx.run_until_parked();
9820 item.update(cx, |_, cx| {
9821 cx.emit(ItemEvent::UpdateTab);
9822 });
9823
9824 // Allow any potential close operation to complete
9825 cx.run_until_parked();
9826
9827 // Verify the item remains open (with strikethrough)
9828 pane.read_with(cx, |pane, _| {
9829 assert_eq!(
9830 pane.items().count(),
9831 1,
9832 "Item should remain open when close_on_disk_deletion is disabled"
9833 );
9834 });
9835
9836 // Verify the item shows as deleted
9837 item.read_with(cx, |item, _| {
9838 assert!(
9839 item.has_deleted_file,
9840 "Item should be marked as having deleted file"
9841 );
9842 });
9843 }
9844
9845 /// Tests that dirty files are not automatically closed when deleted from disk,
9846 /// even when `close_on_file_delete` is enabled. This ensures users don't lose
9847 /// unsaved changes without being prompted.
9848 #[gpui::test]
9849 async fn test_close_on_disk_deletion_with_dirty_file(cx: &mut TestAppContext) {
9850 init_test(cx);
9851
9852 // Enable the close_on_file_delete setting
9853 cx.update_global(|store: &mut SettingsStore, cx| {
9854 store.update_user_settings::<WorkspaceSettings>(cx, |settings| {
9855 settings.close_on_file_delete = Some(true);
9856 });
9857 });
9858
9859 let fs = FakeFs::new(cx.background_executor.clone());
9860 let project = Project::test(fs, [], cx).await;
9861 let (workspace, cx) =
9862 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9863 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
9864
9865 // Create a dirty test item
9866 let item = cx.new(|cx| {
9867 TestItem::new(cx)
9868 .with_dirty(true)
9869 .with_label("test.txt")
9870 .with_project_items(&[TestProjectItem::new(1, "test.txt", cx)])
9871 });
9872
9873 // Add item to workspace
9874 workspace.update_in(cx, |workspace, window, cx| {
9875 workspace.add_item(
9876 pane.clone(),
9877 Box::new(item.clone()),
9878 None,
9879 false,
9880 false,
9881 window,
9882 cx,
9883 );
9884 });
9885
9886 // Simulate file deletion
9887 item.update(cx, |item, _| {
9888 item.set_has_deleted_file(true);
9889 });
9890
9891 // Emit UpdateTab event to trigger the close behavior
9892 cx.run_until_parked();
9893 item.update(cx, |_, cx| {
9894 cx.emit(ItemEvent::UpdateTab);
9895 });
9896
9897 // Allow any potential close operation to complete
9898 cx.run_until_parked();
9899
9900 // Verify the item remains open (dirty files are not auto-closed)
9901 pane.read_with(cx, |pane, _| {
9902 assert_eq!(
9903 pane.items().count(),
9904 1,
9905 "Dirty items should not be automatically closed even when file is deleted"
9906 );
9907 });
9908
9909 // Verify the item is marked as deleted and still dirty
9910 item.read_with(cx, |item, _| {
9911 assert!(
9912 item.has_deleted_file,
9913 "Item should be marked as having deleted file"
9914 );
9915 assert!(item.is_dirty, "Item should still be dirty");
9916 });
9917 }
9918
9919 /// Tests that navigation history is cleaned up when files are auto-closed
9920 /// due to deletion from disk.
9921 #[gpui::test]
9922 async fn test_close_on_disk_deletion_cleans_navigation_history(cx: &mut TestAppContext) {
9923 init_test(cx);
9924
9925 // Enable the close_on_file_delete setting
9926 cx.update_global(|store: &mut SettingsStore, cx| {
9927 store.update_user_settings::<WorkspaceSettings>(cx, |settings| {
9928 settings.close_on_file_delete = Some(true);
9929 });
9930 });
9931
9932 let fs = FakeFs::new(cx.background_executor.clone());
9933 let project = Project::test(fs, [], cx).await;
9934 let (workspace, cx) =
9935 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
9936 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
9937
9938 // Create test items
9939 let item1 = cx.new(|cx| {
9940 TestItem::new(cx)
9941 .with_label("test1.txt")
9942 .with_project_items(&[TestProjectItem::new(1, "test1.txt", cx)])
9943 });
9944 let item1_id = item1.item_id();
9945
9946 let item2 = cx.new(|cx| {
9947 TestItem::new(cx)
9948 .with_label("test2.txt")
9949 .with_project_items(&[TestProjectItem::new(2, "test2.txt", cx)])
9950 });
9951
9952 // Add items to workspace
9953 workspace.update_in(cx, |workspace, window, cx| {
9954 workspace.add_item(
9955 pane.clone(),
9956 Box::new(item1.clone()),
9957 None,
9958 false,
9959 false,
9960 window,
9961 cx,
9962 );
9963 workspace.add_item(
9964 pane.clone(),
9965 Box::new(item2.clone()),
9966 None,
9967 false,
9968 false,
9969 window,
9970 cx,
9971 );
9972 });
9973
9974 // Activate item1 to ensure it gets navigation entries
9975 pane.update_in(cx, |pane, window, cx| {
9976 pane.activate_item(0, true, true, window, cx);
9977 });
9978
9979 // Switch to item2 and back to create navigation history
9980 pane.update_in(cx, |pane, window, cx| {
9981 pane.activate_item(1, true, true, window, cx);
9982 });
9983 cx.run_until_parked();
9984
9985 pane.update_in(cx, |pane, window, cx| {
9986 pane.activate_item(0, true, true, window, cx);
9987 });
9988 cx.run_until_parked();
9989
9990 // Simulate file deletion for item1
9991 item1.update(cx, |item, _| {
9992 item.set_has_deleted_file(true);
9993 });
9994
9995 // Emit UpdateTab event to trigger the close behavior
9996 item1.update(cx, |_, cx| {
9997 cx.emit(ItemEvent::UpdateTab);
9998 });
9999 cx.run_until_parked();
10000
10001 // Verify item1 was closed
10002 pane.read_with(cx, |pane, _| {
10003 assert_eq!(
10004 pane.items().count(),
10005 1,
10006 "Should have 1 item remaining after auto-close"
10007 );
10008 });
10009
10010 // Check navigation history after close
10011 let has_item = pane.read_with(cx, |pane, cx| {
10012 let mut has_item = false;
10013 pane.nav_history().for_each_entry(cx, |entry, _| {
10014 if entry.item.id() == item1_id {
10015 has_item = true;
10016 }
10017 });
10018 has_item
10019 });
10020
10021 assert!(
10022 !has_item,
10023 "Navigation history should not contain closed item entries"
10024 );
10025 }
10026
10027 #[gpui::test]
10028 async fn test_no_save_prompt_when_dirty_multi_buffer_closed_with_all_of_its_dirty_items_present_in_the_pane(
10029 cx: &mut TestAppContext,
10030 ) {
10031 init_test(cx);
10032
10033 let fs = FakeFs::new(cx.background_executor.clone());
10034 let project = Project::test(fs, [], cx).await;
10035 let (workspace, cx) =
10036 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
10037 let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone());
10038
10039 let dirty_regular_buffer = cx.new(|cx| {
10040 TestItem::new(cx)
10041 .with_dirty(true)
10042 .with_label("1.txt")
10043 .with_project_items(&[dirty_project_item(1, "1.txt", cx)])
10044 });
10045 let dirty_regular_buffer_2 = cx.new(|cx| {
10046 TestItem::new(cx)
10047 .with_dirty(true)
10048 .with_label("2.txt")
10049 .with_project_items(&[dirty_project_item(2, "2.txt", cx)])
10050 });
10051 let clear_regular_buffer = cx.new(|cx| {
10052 TestItem::new(cx)
10053 .with_label("3.txt")
10054 .with_project_items(&[TestProjectItem::new(3, "3.txt", cx)])
10055 });
10056
10057 let dirty_multi_buffer = cx.new(|cx| {
10058 TestItem::new(cx)
10059 .with_dirty(true)
10060 .with_singleton(false)
10061 .with_label("Fake Project Search")
10062 .with_project_items(&[
10063 dirty_regular_buffer.read(cx).project_items[0].clone(),
10064 dirty_regular_buffer_2.read(cx).project_items[0].clone(),
10065 clear_regular_buffer.read(cx).project_items[0].clone(),
10066 ])
10067 });
10068 workspace.update_in(cx, |workspace, window, cx| {
10069 workspace.add_item(
10070 pane.clone(),
10071 Box::new(dirty_regular_buffer.clone()),
10072 None,
10073 false,
10074 false,
10075 window,
10076 cx,
10077 );
10078 workspace.add_item(
10079 pane.clone(),
10080 Box::new(dirty_regular_buffer_2.clone()),
10081 None,
10082 false,
10083 false,
10084 window,
10085 cx,
10086 );
10087 workspace.add_item(
10088 pane.clone(),
10089 Box::new(dirty_multi_buffer.clone()),
10090 None,
10091 false,
10092 false,
10093 window,
10094 cx,
10095 );
10096 });
10097
10098 pane.update_in(cx, |pane, window, cx| {
10099 pane.activate_item(2, true, true, window, cx);
10100 assert_eq!(
10101 pane.active_item().unwrap().item_id(),
10102 dirty_multi_buffer.item_id(),
10103 "Should select the multi buffer in the pane"
10104 );
10105 });
10106 let close_multi_buffer_task = pane.update_in(cx, |pane, window, cx| {
10107 pane.close_active_item(
10108 &CloseActiveItem {
10109 save_intent: None,
10110 close_pinned: false,
10111 },
10112 window,
10113 cx,
10114 )
10115 });
10116 cx.background_executor.run_until_parked();
10117 assert!(
10118 !cx.has_pending_prompt(),
10119 "All dirty items from the multi buffer are in the pane still, no save prompts should be shown"
10120 );
10121 close_multi_buffer_task
10122 .await
10123 .expect("Closing multi buffer failed");
10124 pane.update(cx, |pane, cx| {
10125 assert_eq!(dirty_regular_buffer.read(cx).save_count, 0);
10126 assert_eq!(dirty_multi_buffer.read(cx).save_count, 0);
10127 assert_eq!(dirty_regular_buffer_2.read(cx).save_count, 0);
10128 assert_eq!(
10129 pane.items()
10130 .map(|item| item.item_id())
10131 .sorted()
10132 .collect::<Vec<_>>(),
10133 vec![
10134 dirty_regular_buffer.item_id(),
10135 dirty_regular_buffer_2.item_id(),
10136 ],
10137 "Should have no multi buffer left in the pane"
10138 );
10139 assert!(dirty_regular_buffer.read(cx).is_dirty);
10140 assert!(dirty_regular_buffer_2.read(cx).is_dirty);
10141 });
10142 }
10143
10144 #[gpui::test]
10145 async fn test_move_focused_panel_to_next_position(cx: &mut gpui::TestAppContext) {
10146 init_test(cx);
10147 let fs = FakeFs::new(cx.executor());
10148 let project = Project::test(fs, [], cx).await;
10149 let (workspace, cx) =
10150 cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx));
10151
10152 // Add a new panel to the right dock, opening the dock and setting the
10153 // focus to the new panel.
10154 let panel = workspace.update_in(cx, |workspace, window, cx| {
10155 let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, cx));
10156 workspace.add_panel(panel.clone(), window, cx);
10157
10158 workspace
10159 .right_dock()
10160 .update(cx, |right_dock, cx| right_dock.set_open(true, window, cx));
10161
10162 workspace.toggle_panel_focus::<TestPanel>(window, cx);
10163
10164 panel
10165 });
10166
10167 // Dispatch the `MoveFocusedPanelToNextPosition` action, moving the
10168 // panel to the next valid position which, in this case, is the left
10169 // dock.
10170 cx.dispatch_action(MoveFocusedPanelToNextPosition);
10171 workspace.update(cx, |workspace, cx| {
10172 assert!(workspace.left_dock().read(cx).is_open());
10173 assert_eq!(panel.read(cx).position, DockPosition::Left);
10174 });
10175
10176 // Dispatch the `MoveFocusedPanelToNextPosition` action, moving the
10177 // panel to the next valid position which, in this case, is the bottom
10178 // dock.
10179 cx.dispatch_action(MoveFocusedPanelToNextPosition);
10180 workspace.update(cx, |workspace, cx| {
10181 assert!(workspace.bottom_dock().read(cx).is_open());
10182 assert_eq!(panel.read(cx).position, DockPosition::Bottom);
10183 });
10184
10185 // Dispatch the `MoveFocusedPanelToNextPosition` action again, this time
10186 // around moving the panel to its initial position, the right dock.
10187 cx.dispatch_action(MoveFocusedPanelToNextPosition);
10188 workspace.update(cx, |workspace, cx| {
10189 assert!(workspace.right_dock().read(cx).is_open());
10190 assert_eq!(panel.read(cx).position, DockPosition::Right);
10191 });
10192
10193 // Remove focus from the panel, ensuring that, if the panel is not
10194 // focused, the `MoveFocusedPanelToNextPosition` action does not update
10195 // the panel's position, so the panel is still in the right dock.
10196 workspace.update_in(cx, |workspace, window, cx| {
10197 workspace.toggle_panel_focus::<TestPanel>(window, cx);
10198 });
10199
10200 cx.dispatch_action(MoveFocusedPanelToNextPosition);
10201 workspace.update(cx, |workspace, cx| {
10202 assert!(workspace.right_dock().read(cx).is_open());
10203 assert_eq!(panel.read(cx).position, DockPosition::Right);
10204 });
10205 }
10206
10207 #[gpui::test]
10208 async fn test_moving_items_create_panes(cx: &mut TestAppContext) {
10209 init_test(cx);
10210
10211 let fs = FakeFs::new(cx.executor());
10212 let project = Project::test(fs, [], cx).await;
10213 let (workspace, cx) =
10214 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
10215
10216 let item_1 = cx.new(|cx| {
10217 TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "first.txt", cx)])
10218 });
10219 workspace.update_in(cx, |workspace, window, cx| {
10220 workspace.add_item_to_active_pane(Box::new(item_1), None, true, window, cx);
10221 workspace.move_item_to_pane_in_direction(
10222 &MoveItemToPaneInDirection {
10223 direction: SplitDirection::Right,
10224 focus: true,
10225 clone: false,
10226 },
10227 window,
10228 cx,
10229 );
10230 workspace.move_item_to_pane_at_index(
10231 &MoveItemToPane {
10232 destination: 3,
10233 focus: true,
10234 clone: false,
10235 },
10236 window,
10237 cx,
10238 );
10239
10240 assert_eq!(workspace.panes.len(), 1, "No new panes were created");
10241 assert_eq!(
10242 pane_items_paths(&workspace.active_pane, cx),
10243 vec!["first.txt".to_string()],
10244 "Single item was not moved anywhere"
10245 );
10246 });
10247
10248 let item_2 = cx.new(|cx| {
10249 TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "second.txt", cx)])
10250 });
10251 workspace.update_in(cx, |workspace, window, cx| {
10252 workspace.add_item_to_active_pane(Box::new(item_2), None, true, window, cx);
10253 assert_eq!(
10254 pane_items_paths(&workspace.panes[0], cx),
10255 vec!["first.txt".to_string(), "second.txt".to_string()],
10256 );
10257 workspace.move_item_to_pane_in_direction(
10258 &MoveItemToPaneInDirection {
10259 direction: SplitDirection::Right,
10260 focus: true,
10261 clone: false,
10262 },
10263 window,
10264 cx,
10265 );
10266
10267 assert_eq!(workspace.panes.len(), 2, "A new pane should be created");
10268 assert_eq!(
10269 pane_items_paths(&workspace.panes[0], cx),
10270 vec!["first.txt".to_string()],
10271 "After moving, one item should be left in the original pane"
10272 );
10273 assert_eq!(
10274 pane_items_paths(&workspace.panes[1], cx),
10275 vec!["second.txt".to_string()],
10276 "New item should have been moved to the new pane"
10277 );
10278 });
10279
10280 let item_3 = cx.new(|cx| {
10281 TestItem::new(cx).with_project_items(&[TestProjectItem::new(3, "third.txt", cx)])
10282 });
10283 workspace.update_in(cx, |workspace, window, cx| {
10284 let original_pane = workspace.panes[0].clone();
10285 workspace.set_active_pane(&original_pane, window, cx);
10286 workspace.add_item_to_active_pane(Box::new(item_3), None, true, window, cx);
10287 assert_eq!(workspace.panes.len(), 2, "No new panes were created");
10288 assert_eq!(
10289 pane_items_paths(&workspace.active_pane, cx),
10290 vec!["first.txt".to_string(), "third.txt".to_string()],
10291 "New pane should be ready to move one item out"
10292 );
10293
10294 workspace.move_item_to_pane_at_index(
10295 &MoveItemToPane {
10296 destination: 3,
10297 focus: true,
10298 clone: false,
10299 },
10300 window,
10301 cx,
10302 );
10303 assert_eq!(workspace.panes.len(), 3, "A new pane should be created");
10304 assert_eq!(
10305 pane_items_paths(&workspace.active_pane, cx),
10306 vec!["first.txt".to_string()],
10307 "After moving, one item should be left in the original pane"
10308 );
10309 assert_eq!(
10310 pane_items_paths(&workspace.panes[1], cx),
10311 vec!["second.txt".to_string()],
10312 "Previously created pane should be unchanged"
10313 );
10314 assert_eq!(
10315 pane_items_paths(&workspace.panes[2], cx),
10316 vec!["third.txt".to_string()],
10317 "New item should have been moved to the new pane"
10318 );
10319 });
10320 }
10321
10322 #[gpui::test]
10323 async fn test_moving_items_can_clone_panes(cx: &mut TestAppContext) {
10324 init_test(cx);
10325
10326 let fs = FakeFs::new(cx.executor());
10327 let project = Project::test(fs, [], cx).await;
10328 let (workspace, cx) =
10329 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
10330
10331 let item_1 = cx.new(|cx| {
10332 TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "first.txt", cx)])
10333 });
10334 workspace.update_in(cx, |workspace, window, cx| {
10335 workspace.add_item_to_active_pane(Box::new(item_1), None, true, window, cx);
10336 workspace.move_item_to_pane_in_direction(
10337 &MoveItemToPaneInDirection {
10338 direction: SplitDirection::Right,
10339 focus: true,
10340 clone: true,
10341 },
10342 window,
10343 cx,
10344 );
10345 workspace.move_item_to_pane_at_index(
10346 &MoveItemToPane {
10347 destination: 3,
10348 focus: true,
10349 clone: true,
10350 },
10351 window,
10352 cx,
10353 );
10354
10355 assert_eq!(workspace.panes.len(), 3, "Two new panes were created");
10356 for pane in workspace.panes() {
10357 assert_eq!(
10358 pane_items_paths(pane, cx),
10359 vec!["first.txt".to_string()],
10360 "Single item exists in all panes"
10361 );
10362 }
10363 });
10364
10365 // verify that the active pane has been updated after waiting for the
10366 // pane focus event to fire and resolve
10367 workspace.read_with(cx, |workspace, _app| {
10368 assert_eq!(
10369 workspace.active_pane(),
10370 &workspace.panes[2],
10371 "The third pane should be the active one: {:?}",
10372 workspace.panes
10373 );
10374 })
10375 }
10376
10377 mod register_project_item_tests {
10378
10379 use super::*;
10380
10381 // View
10382 struct TestPngItemView {
10383 focus_handle: FocusHandle,
10384 }
10385 // Model
10386 struct TestPngItem {}
10387
10388 impl project::ProjectItem for TestPngItem {
10389 fn try_open(
10390 _project: &Entity<Project>,
10391 path: &ProjectPath,
10392 cx: &mut App,
10393 ) -> Option<Task<anyhow::Result<Entity<Self>>>> {
10394 if path.path.extension().unwrap() == "png" {
10395 Some(cx.spawn(async move |cx| cx.new(|_| TestPngItem {})))
10396 } else {
10397 None
10398 }
10399 }
10400
10401 fn entry_id(&self, _: &App) -> Option<ProjectEntryId> {
10402 None
10403 }
10404
10405 fn project_path(&self, _: &App) -> Option<ProjectPath> {
10406 None
10407 }
10408
10409 fn is_dirty(&self) -> bool {
10410 false
10411 }
10412 }
10413
10414 impl Item for TestPngItemView {
10415 type Event = ();
10416 fn tab_content_text(&self, _detail: usize, _cx: &App) -> SharedString {
10417 "".into()
10418 }
10419 }
10420 impl EventEmitter<()> for TestPngItemView {}
10421 impl Focusable for TestPngItemView {
10422 fn focus_handle(&self, _cx: &App) -> FocusHandle {
10423 self.focus_handle.clone()
10424 }
10425 }
10426
10427 impl Render for TestPngItemView {
10428 fn render(
10429 &mut self,
10430 _window: &mut Window,
10431 _cx: &mut Context<Self>,
10432 ) -> impl IntoElement {
10433 Empty
10434 }
10435 }
10436
10437 impl ProjectItem for TestPngItemView {
10438 type Item = TestPngItem;
10439
10440 fn for_project_item(
10441 _project: Entity<Project>,
10442 _pane: Option<&Pane>,
10443 _item: Entity<Self::Item>,
10444 _: &mut Window,
10445 cx: &mut Context<Self>,
10446 ) -> Self
10447 where
10448 Self: Sized,
10449 {
10450 Self {
10451 focus_handle: cx.focus_handle(),
10452 }
10453 }
10454 }
10455
10456 // View
10457 struct TestIpynbItemView {
10458 focus_handle: FocusHandle,
10459 }
10460 // Model
10461 struct TestIpynbItem {}
10462
10463 impl project::ProjectItem for TestIpynbItem {
10464 fn try_open(
10465 _project: &Entity<Project>,
10466 path: &ProjectPath,
10467 cx: &mut App,
10468 ) -> Option<Task<anyhow::Result<Entity<Self>>>> {
10469 if path.path.extension().unwrap() == "ipynb" {
10470 Some(cx.spawn(async move |cx| cx.new(|_| TestIpynbItem {})))
10471 } else {
10472 None
10473 }
10474 }
10475
10476 fn entry_id(&self, _: &App) -> Option<ProjectEntryId> {
10477 None
10478 }
10479
10480 fn project_path(&self, _: &App) -> Option<ProjectPath> {
10481 None
10482 }
10483
10484 fn is_dirty(&self) -> bool {
10485 false
10486 }
10487 }
10488
10489 impl Item for TestIpynbItemView {
10490 type Event = ();
10491 fn tab_content_text(&self, _detail: usize, _cx: &App) -> SharedString {
10492 "".into()
10493 }
10494 }
10495 impl EventEmitter<()> for TestIpynbItemView {}
10496 impl Focusable for TestIpynbItemView {
10497 fn focus_handle(&self, _cx: &App) -> FocusHandle {
10498 self.focus_handle.clone()
10499 }
10500 }
10501
10502 impl Render for TestIpynbItemView {
10503 fn render(
10504 &mut self,
10505 _window: &mut Window,
10506 _cx: &mut Context<Self>,
10507 ) -> impl IntoElement {
10508 Empty
10509 }
10510 }
10511
10512 impl ProjectItem for TestIpynbItemView {
10513 type Item = TestIpynbItem;
10514
10515 fn for_project_item(
10516 _project: Entity<Project>,
10517 _pane: Option<&Pane>,
10518 _item: Entity<Self::Item>,
10519 _: &mut Window,
10520 cx: &mut Context<Self>,
10521 ) -> Self
10522 where
10523 Self: Sized,
10524 {
10525 Self {
10526 focus_handle: cx.focus_handle(),
10527 }
10528 }
10529 }
10530
10531 struct TestAlternatePngItemView {
10532 focus_handle: FocusHandle,
10533 }
10534
10535 impl Item for TestAlternatePngItemView {
10536 type Event = ();
10537 fn tab_content_text(&self, _detail: usize, _cx: &App) -> SharedString {
10538 "".into()
10539 }
10540 }
10541
10542 impl EventEmitter<()> for TestAlternatePngItemView {}
10543 impl Focusable for TestAlternatePngItemView {
10544 fn focus_handle(&self, _cx: &App) -> FocusHandle {
10545 self.focus_handle.clone()
10546 }
10547 }
10548
10549 impl Render for TestAlternatePngItemView {
10550 fn render(
10551 &mut self,
10552 _window: &mut Window,
10553 _cx: &mut Context<Self>,
10554 ) -> impl IntoElement {
10555 Empty
10556 }
10557 }
10558
10559 impl ProjectItem for TestAlternatePngItemView {
10560 type Item = TestPngItem;
10561
10562 fn for_project_item(
10563 _project: Entity<Project>,
10564 _pane: Option<&Pane>,
10565 _item: Entity<Self::Item>,
10566 _: &mut Window,
10567 cx: &mut Context<Self>,
10568 ) -> Self
10569 where
10570 Self: Sized,
10571 {
10572 Self {
10573 focus_handle: cx.focus_handle(),
10574 }
10575 }
10576 }
10577
10578 #[gpui::test]
10579 async fn test_register_project_item(cx: &mut TestAppContext) {
10580 init_test(cx);
10581
10582 cx.update(|cx| {
10583 register_project_item::<TestPngItemView>(cx);
10584 register_project_item::<TestIpynbItemView>(cx);
10585 });
10586
10587 let fs = FakeFs::new(cx.executor());
10588 fs.insert_tree(
10589 "/root1",
10590 json!({
10591 "one.png": "BINARYDATAHERE",
10592 "two.ipynb": "{ totally a notebook }",
10593 "three.txt": "editing text, sure why not?"
10594 }),
10595 )
10596 .await;
10597
10598 let project = Project::test(fs, ["root1".as_ref()], cx).await;
10599 let (workspace, cx) =
10600 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
10601
10602 let worktree_id = project.update(cx, |project, cx| {
10603 project.worktrees(cx).next().unwrap().read(cx).id()
10604 });
10605
10606 let handle = workspace
10607 .update_in(cx, |workspace, window, cx| {
10608 let project_path = (worktree_id, "one.png");
10609 workspace.open_path(project_path, None, true, window, cx)
10610 })
10611 .await
10612 .unwrap();
10613
10614 // Now we can check if the handle we got back errored or not
10615 assert_eq!(
10616 handle.to_any().entity_type(),
10617 TypeId::of::<TestPngItemView>()
10618 );
10619
10620 let handle = workspace
10621 .update_in(cx, |workspace, window, cx| {
10622 let project_path = (worktree_id, "two.ipynb");
10623 workspace.open_path(project_path, None, true, window, cx)
10624 })
10625 .await
10626 .unwrap();
10627
10628 assert_eq!(
10629 handle.to_any().entity_type(),
10630 TypeId::of::<TestIpynbItemView>()
10631 );
10632
10633 let handle = workspace
10634 .update_in(cx, |workspace, window, cx| {
10635 let project_path = (worktree_id, "three.txt");
10636 workspace.open_path(project_path, None, true, window, cx)
10637 })
10638 .await;
10639 assert!(handle.is_err());
10640 }
10641
10642 #[gpui::test]
10643 async fn test_register_project_item_two_enter_one_leaves(cx: &mut TestAppContext) {
10644 init_test(cx);
10645
10646 cx.update(|cx| {
10647 register_project_item::<TestPngItemView>(cx);
10648 register_project_item::<TestAlternatePngItemView>(cx);
10649 });
10650
10651 let fs = FakeFs::new(cx.executor());
10652 fs.insert_tree(
10653 "/root1",
10654 json!({
10655 "one.png": "BINARYDATAHERE",
10656 "two.ipynb": "{ totally a notebook }",
10657 "three.txt": "editing text, sure why not?"
10658 }),
10659 )
10660 .await;
10661 let project = Project::test(fs, ["root1".as_ref()], cx).await;
10662 let (workspace, cx) =
10663 cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx));
10664 let worktree_id = project.update(cx, |project, cx| {
10665 project.worktrees(cx).next().unwrap().read(cx).id()
10666 });
10667
10668 let handle = workspace
10669 .update_in(cx, |workspace, window, cx| {
10670 let project_path = (worktree_id, "one.png");
10671 workspace.open_path(project_path, None, true, window, cx)
10672 })
10673 .await
10674 .unwrap();
10675
10676 // This _must_ be the second item registered
10677 assert_eq!(
10678 handle.to_any().entity_type(),
10679 TypeId::of::<TestAlternatePngItemView>()
10680 );
10681
10682 let handle = workspace
10683 .update_in(cx, |workspace, window, cx| {
10684 let project_path = (worktree_id, "three.txt");
10685 workspace.open_path(project_path, None, true, window, cx)
10686 })
10687 .await;
10688 assert!(handle.is_err());
10689 }
10690 }
10691
10692 fn pane_items_paths(pane: &Entity<Pane>, cx: &App) -> Vec<String> {
10693 pane.read(cx)
10694 .items()
10695 .flat_map(|item| {
10696 item.project_paths(cx)
10697 .into_iter()
10698 .map(|path| path.path.to_string_lossy().to_string())
10699 })
10700 .collect()
10701 }
10702
10703 pub fn init_test(cx: &mut TestAppContext) {
10704 cx.update(|cx| {
10705 let settings_store = SettingsStore::test(cx);
10706 cx.set_global(settings_store);
10707 theme::init(theme::LoadThemes::JustBase, cx);
10708 language::init(cx);
10709 crate::init_settings(cx);
10710 Project::init_settings(cx);
10711 });
10712 }
10713
10714 fn dirty_project_item(id: u64, path: &str, cx: &mut App) -> Entity<TestProjectItem> {
10715 let item = TestProjectItem::new(id, path, cx);
10716 item.update(cx, |item, _| {
10717 item.is_dirty = true;
10718 });
10719 item
10720 }
10721}