pub mod active_file_name; pub mod dock; pub mod history_manager; pub mod invalid_item_view; pub mod item; mod modal_layer; mod multi_workspace; #[cfg(test)] mod multi_workspace_tests; pub mod notifications; pub mod pane; pub mod pane_group; pub mod path_list { pub use util::path_list::{PathList, SerializedPathList}; } mod persistence; pub mod searchable; mod security_modal; pub mod shared_screen; use db::smol::future::yield_now; pub use shared_screen::SharedScreen; pub mod focus_follows_mouse; mod status_bar; pub mod tasks; mod theme_preview; mod toast_layer; mod toolbar; pub mod welcome; mod workspace_settings; pub use crate::notifications::NotificationFrame; pub use dock::Panel; pub use multi_workspace::{ CloseWorkspaceSidebar, DraggedSidebar, FocusWorkspaceSidebar, MultiWorkspace, MultiWorkspaceEvent, NewThread, NextProject, NextThread, PreviousProject, PreviousThread, ShowFewerThreads, ShowMoreThreads, Sidebar, SidebarEvent, SidebarHandle, SidebarRenderState, SidebarSide, ToggleWorkspaceSidebar, sidebar_side_context_menu, }; pub use path_list::{PathList, SerializedPathList}; pub use toast_layer::{ToastAction, ToastLayer, ToastView}; use anyhow::{Context as _, Result, anyhow}; use client::{ ChannelId, Client, ErrorExt, ParticipantIndex, Status, TypedEnvelope, User, UserStore, proto::{self, ErrorCode, PanelId, PeerId}, }; use collections::{HashMap, HashSet, hash_map}; use dock::{Dock, DockPosition, PanelButtons, PanelHandle, RESIZE_HANDLE_SIZE}; use fs::Fs; use futures::{ Future, FutureExt, StreamExt, channel::{ mpsc::{self, UnboundedReceiver, UnboundedSender}, oneshot, }, future::{Shared, try_join_all}, }; use gpui::{ Action, AnyEntity, AnyView, AnyWeakView, App, AsyncApp, AsyncWindowContext, Axis, Bounds, Context, CursorStyle, Decorations, DragMoveEvent, Entity, EntityId, EventEmitter, FocusHandle, Focusable, Global, HitboxBehavior, Hsla, KeyContext, Keystroke, ManagedView, MouseButton, PathPromptOptions, Point, PromptLevel, Render, ResizeEdge, Size, Stateful, Subscription, SystemWindowTabController, Task, Tiling, WeakEntity, WindowBounds, WindowHandle, WindowId, WindowOptions, actions, canvas, point, relative, size, transparent_black, }; pub use history_manager::*; pub use item::{ FollowableItem, FollowableItemHandle, Item, ItemHandle, ItemSettings, PreviewTabsSettings, ProjectItem, SerializableItem, SerializableItemHandle, WeakItemHandle, }; use itertools::Itertools; use language::{Buffer, LanguageRegistry, Rope, language_settings::all_language_settings}; pub use modal_layer::*; use node_runtime::NodeRuntime; use notifications::{ DetachAndPromptErr, Notifications, dismiss_app_notification, simple_message_notification::MessageNotification, }; pub use pane::*; pub use pane_group::{ ActivePaneDecorator, HANDLE_HITBOX_SIZE, Member, PaneAxis, PaneGroup, PaneRenderContext, SplitDirection, }; use persistence::{SerializedWindowBounds, model::SerializedWorkspace}; pub use persistence::{ WorkspaceDb, delete_unloaded_items, model::{ DockStructure, ItemId, MultiWorkspaceState, SerializedMultiWorkspace, SerializedWorkspaceLocation, SessionWorkspace, }, read_serialized_multi_workspaces, resolve_worktree_workspaces, }; use postage::stream::Stream; use project::{ DirectoryLister, Project, ProjectEntryId, ProjectGroupKey, ProjectPath, ResolvedPath, Worktree, WorktreeId, WorktreeSettings, debugger::{breakpoint_store::BreakpointStoreEvent, session::ThreadStatus}, project_settings::ProjectSettings, toolchain_store::ToolchainStoreEvent, trusted_worktrees::{RemoteHostLocation, TrustedWorktrees, TrustedWorktreesEvent}, }; use remote::{ RemoteClientDelegate, RemoteConnection, RemoteConnectionOptions, remote_client::ConnectionIdentifier, }; use schemars::JsonSchema; use serde::Deserialize; use session::AppSession; use settings::{ CenteredPaddingSettings, Settings, SettingsLocation, SettingsStore, update_settings_file, }; use sqlez::{ bindable::{Bind, Column, StaticColumnCount}, statement::Statement, }; use status_bar::StatusBar; pub use status_bar::StatusItemView; use std::{ any::TypeId, borrow::Cow, cell::RefCell, cmp, collections::VecDeque, env, hash::Hash, path::{Path, PathBuf}, process::ExitStatus, rc::Rc, sync::{ Arc, LazyLock, atomic::{AtomicBool, AtomicUsize}, }, time::Duration, }; use task::{DebugScenario, SharedTaskContext, SpawnInTerminal}; use theme::{ActiveTheme, SystemAppearance}; use theme_settings::ThemeSettings; pub use toolbar::{ PaneSearchBarCallbacks, Toolbar, ToolbarItemEvent, ToolbarItemLocation, ToolbarItemView, }; pub use ui; use ui::{Window, prelude::*}; use util::{ ResultExt, TryFutureExt, paths::{PathStyle, SanitizedPath}, rel_path::RelPath, serde::default_true, }; use uuid::Uuid; pub use workspace_settings::{ AutosaveSetting, BottomDockLayout, FocusFollowsMouse, RestoreOnStartupBehavior, StatusBarSettings, TabBarSettings, WorkspaceSettings, }; use zed_actions::{Spawn, feedback::FileBugReport, theme::ToggleMode}; use crate::{dock::PanelSizeState, item::ItemBufferKind, notifications::NotificationId}; use crate::{ persistence::{ SerializedAxis, model::{DockData, SerializedItem, SerializedPane, SerializedPaneGroup}, }, security_modal::SecurityModal, }; pub const SERIALIZATION_THROTTLE_TIME: Duration = Duration::from_millis(200); static ZED_WINDOW_SIZE: LazyLock>> = LazyLock::new(|| { env::var("ZED_WINDOW_SIZE") .ok() .as_deref() .and_then(parse_pixel_size_env_var) }); static ZED_WINDOW_POSITION: LazyLock>> = LazyLock::new(|| { env::var("ZED_WINDOW_POSITION") .ok() .as_deref() .and_then(parse_pixel_position_env_var) }); pub trait TerminalProvider { fn spawn( &self, task: SpawnInTerminal, window: &mut Window, cx: &mut App, ) -> Task>>; } pub trait DebuggerProvider { // `active_buffer` is used to resolve build task's name against language-specific tasks. fn start_session( &self, definition: DebugScenario, task_context: SharedTaskContext, active_buffer: Option>, worktree_id: Option, window: &mut Window, cx: &mut App, ); fn spawn_task_or_modal( &self, workspace: &mut Workspace, action: &Spawn, window: &mut Window, cx: &mut Context, ); fn task_scheduled(&self, cx: &mut App); fn debug_scenario_scheduled(&self, cx: &mut App); fn debug_scenario_scheduled_last(&self, cx: &App) -> bool; fn active_thread_state(&self, cx: &App) -> Option; } /// Opens a file or directory. #[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] pub struct Open { /// When true, opens in a new window. When false, adds to the current /// window as a new workspace (multi-workspace). #[serde(default = "Open::default_create_new_window")] pub create_new_window: bool, } impl Open { pub const DEFAULT: Self = Self { create_new_window: true, }; /// Used by `#[serde(default)]` on the `create_new_window` field so that /// the serde default and `Open::DEFAULT` stay in sync. fn default_create_new_window() -> bool { Self::DEFAULT.create_new_window } } impl Default for Open { fn default() -> Self { Self::DEFAULT } } actions!( workspace, [ /// Activates the next pane in the workspace. ActivateNextPane, /// Activates the previous pane in the workspace. ActivatePreviousPane, /// Activates the last pane in the workspace. ActivateLastPane, /// Switches to the next window. ActivateNextWindow, /// Switches to the previous window. ActivatePreviousWindow, /// Adds a folder to the current project. AddFolderToProject, /// Clears all notifications. ClearAllNotifications, /// Clears all navigation history, including forward/backward navigation, recently opened files, and recently closed tabs. **This action is irreversible**. ClearNavigationHistory, /// Closes the active dock. CloseActiveDock, /// Closes all docks. CloseAllDocks, /// Toggles all docks. ToggleAllDocks, /// Closes the current window. CloseWindow, /// Closes the current project. CloseProject, /// Opens the feedback dialog. Feedback, /// Follows the next collaborator in the session. FollowNextCollaborator, /// Moves the focused panel to the next position. MoveFocusedPanelToNextPosition, /// Creates a new file. NewFile, /// Creates a new file in a vertical split. NewFileSplitVertical, /// Creates a new file in a horizontal split. NewFileSplitHorizontal, /// Opens a new search. NewSearch, /// Opens a new window. NewWindow, /// Opens multiple files. OpenFiles, /// Opens the current location in terminal. OpenInTerminal, /// Opens the component preview. OpenComponentPreview, /// Reloads the active item. ReloadActiveItem, /// Resets the active dock to its default size. ResetActiveDockSize, /// Resets all open docks to their default sizes. ResetOpenDocksSize, /// Reloads the application Reload, /// Saves the current file with a new name. SaveAs, /// Saves without formatting. SaveWithoutFormat, /// Shuts down all debug adapters. ShutdownDebugAdapters, /// Suppresses the current notification. SuppressNotification, /// Toggles the bottom dock. ToggleBottomDock, /// Toggles centered layout mode. ToggleCenteredLayout, /// Toggles edit prediction feature globally for all files. ToggleEditPrediction, /// Toggles the left dock. ToggleLeftDock, /// Toggles the right dock. ToggleRightDock, /// Toggles zoom on the active pane. ToggleZoom, /// Toggles read-only mode for the active item (if supported by that item). ToggleReadOnlyFile, /// Zooms in on the active pane. ZoomIn, /// Zooms out of the active pane. ZoomOut, /// If any worktrees are in restricted mode, shows a modal with possible actions. /// If the modal is shown already, closes it without trusting any worktree. ToggleWorktreeSecurity, /// Clears all trusted worktrees, placing them in restricted mode on next open. /// Requires restart to take effect on already opened projects. ClearTrustedWorktrees, /// Stops following a collaborator. Unfollow, /// Restores the banner. RestoreBanner, /// Toggles expansion of the selected item. ToggleExpandItem, ] ); /// Activates a specific pane by its index. #[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)] #[action(namespace = workspace)] pub struct ActivatePane(pub usize); /// Moves an item to a specific pane by index. #[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct MoveItemToPane { #[serde(default = "default_1")] pub destination: usize, #[serde(default = "default_true")] pub focus: bool, #[serde(default)] pub clone: bool, } fn default_1() -> usize { 1 } /// Moves an item to a pane in the specified direction. #[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct MoveItemToPaneInDirection { #[serde(default = "default_right")] pub direction: SplitDirection, #[serde(default = "default_true")] pub focus: bool, #[serde(default)] pub clone: bool, } /// Creates a new file in a split of the desired direction. #[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct NewFileSplit(pub SplitDirection); fn default_right() -> SplitDirection { SplitDirection::Right } /// Saves all open files in the workspace. #[derive(Clone, PartialEq, Debug, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct SaveAll { #[serde(default)] pub save_intent: Option, } /// Saves the current file with the specified options. #[derive(Clone, PartialEq, Debug, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct Save { #[serde(default)] pub save_intent: Option, } /// Moves Focus to the central panes in the workspace. #[derive(Clone, Debug, PartialEq, Eq, Action)] #[action(namespace = workspace)] pub struct FocusCenterPane; /// Closes all items and panes in the workspace. #[derive(Clone, PartialEq, Debug, Deserialize, Default, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct CloseAllItemsAndPanes { #[serde(default)] pub save_intent: Option, } /// Closes all inactive tabs and panes in the workspace. #[derive(Clone, PartialEq, Debug, Deserialize, Default, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct CloseInactiveTabsAndPanes { #[serde(default)] pub save_intent: Option, } /// Closes the active item across all panes. #[derive(Clone, PartialEq, Debug, Deserialize, Default, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct CloseItemInAllPanes { #[serde(default)] pub save_intent: Option, #[serde(default)] pub close_pinned: bool, } /// Sends a sequence of keystrokes to the active element. #[derive(Clone, Deserialize, PartialEq, JsonSchema, Action)] #[action(namespace = workspace)] pub struct SendKeystrokes(pub String); actions!( project_symbols, [ /// Toggles the project symbols search. #[action(name = "Toggle")] ToggleProjectSymbols ] ); /// Toggles the file finder interface. #[derive(Default, PartialEq, Eq, Clone, Deserialize, JsonSchema, Action)] #[action(namespace = file_finder, name = "Toggle")] #[serde(deny_unknown_fields)] pub struct ToggleFileFinder { #[serde(default)] pub separate_history: bool, } /// Opens a new terminal in the center. #[derive(Default, PartialEq, Eq, Clone, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct NewCenterTerminal { /// If true, creates a local terminal even in remote projects. #[serde(default)] pub local: bool, } /// Opens a new terminal. #[derive(Default, PartialEq, Eq, Clone, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct NewTerminal { /// If true, creates a local terminal even in remote projects. #[serde(default)] pub local: bool, } /// Increases size of a currently focused dock by a given amount of pixels. #[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct IncreaseActiveDockSize { /// For 0px parameter, uses UI font size value. #[serde(default)] pub px: u32, } /// Decreases size of a currently focused dock by a given amount of pixels. #[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct DecreaseActiveDockSize { /// For 0px parameter, uses UI font size value. #[serde(default)] pub px: u32, } /// Increases size of all currently visible docks uniformly, by a given amount of pixels. #[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct IncreaseOpenDocksSize { /// For 0px parameter, uses UI font size value. #[serde(default)] pub px: u32, } /// Decreases size of all currently visible docks uniformly, by a given amount of pixels. #[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct DecreaseOpenDocksSize { /// For 0px parameter, uses UI font size value. #[serde(default)] pub px: u32, } actions!( workspace, [ /// Activates the pane to the left. ActivatePaneLeft, /// Activates the pane to the right. ActivatePaneRight, /// Activates the pane above. ActivatePaneUp, /// Activates the pane below. ActivatePaneDown, /// Swaps the current pane with the one to the left. SwapPaneLeft, /// Swaps the current pane with the one to the right. SwapPaneRight, /// Swaps the current pane with the one above. SwapPaneUp, /// Swaps the current pane with the one below. SwapPaneDown, // Swaps the current pane with the first available adjacent pane (searching in order: below, above, right, left) and activates that pane. SwapPaneAdjacent, /// Move the current pane to be at the far left. MovePaneLeft, /// Move the current pane to be at the far right. MovePaneRight, /// Move the current pane to be at the very top. MovePaneUp, /// Move the current pane to be at the very bottom. MovePaneDown, ] ); #[derive(PartialEq, Eq, Debug)] pub enum CloseIntent { /// Quit the program entirely. Quit, /// Close a window. CloseWindow, /// Replace the workspace in an existing window. ReplaceWindow, } #[derive(Clone)] pub struct Toast { id: NotificationId, msg: Cow<'static, str>, autohide: bool, on_click: Option<(Cow<'static, str>, Arc)>, } impl Toast { pub fn new>>(id: NotificationId, msg: I) -> Self { Toast { id, msg: msg.into(), on_click: None, autohide: false, } } pub fn on_click(mut self, message: M, on_click: F) -> Self where M: Into>, F: Fn(&mut Window, &mut App) + 'static, { self.on_click = Some((message.into(), Arc::new(on_click))); self } pub fn autohide(mut self) -> Self { self.autohide = true; self } } impl PartialEq for Toast { fn eq(&self, other: &Self) -> bool { self.id == other.id && self.msg == other.msg && self.on_click.is_some() == other.on_click.is_some() } } /// Opens a new terminal with the specified working directory. #[derive(Debug, Default, Clone, Deserialize, PartialEq, JsonSchema, Action)] #[action(namespace = workspace)] #[serde(deny_unknown_fields)] pub struct OpenTerminal { pub working_directory: PathBuf, /// If true, creates a local terminal even in remote projects. #[serde(default)] pub local: bool, } #[derive( Clone, Copy, Debug, Default, Hash, PartialEq, Eq, PartialOrd, Ord, serde::Serialize, serde::Deserialize, )] pub struct WorkspaceId(i64); impl WorkspaceId { pub fn from_i64(value: i64) -> Self { Self(value) } } impl StaticColumnCount for WorkspaceId {} impl Bind for WorkspaceId { fn bind(&self, statement: &Statement, start_index: i32) -> Result { self.0.bind(statement, start_index) } } impl Column for WorkspaceId { fn column(statement: &mut Statement, start_index: i32) -> Result<(Self, i32)> { i64::column(statement, start_index) .map(|(i, next_index)| (Self(i), next_index)) .with_context(|| format!("Failed to read WorkspaceId at index {start_index}")) } } impl From for i64 { fn from(val: WorkspaceId) -> Self { val.0 } } fn prompt_and_open_paths( app_state: Arc, options: PathPromptOptions, create_new_window: bool, cx: &mut App, ) { if let Some(workspace_window) = local_workspace_windows(cx).into_iter().next() { workspace_window .update(cx, |multi_workspace, window, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { prompt_for_open_path_and_open( workspace, app_state, options, create_new_window, window, cx, ); }); }) .ok(); } else { let task = Workspace::new_local( Vec::new(), app_state.clone(), None, None, None, OpenMode::Activate, cx, ); cx.spawn(async move |cx| { let OpenResult { window, .. } = task.await?; window.update(cx, |multi_workspace, window, cx| { window.activate_window(); let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { prompt_for_open_path_and_open( workspace, app_state, options, create_new_window, window, cx, ); }); })?; anyhow::Ok(()) }) .detach_and_log_err(cx); } } pub fn prompt_for_open_path_and_open( workspace: &mut Workspace, app_state: Arc, options: PathPromptOptions, create_new_window: bool, window: &mut Window, cx: &mut Context, ) { let paths = workspace.prompt_for_open_path( options, DirectoryLister::Local(workspace.project().clone(), app_state.fs.clone()), window, cx, ); let multi_workspace_handle = window.window_handle().downcast::(); cx.spawn_in(window, async move |this, cx| { let Some(paths) = paths.await.log_err().flatten() else { return; }; if !create_new_window { if let Some(handle) = multi_workspace_handle { if let Some(task) = handle .update(cx, |multi_workspace, window, cx| { multi_workspace.open_project(paths, OpenMode::Activate, window, cx) }) .log_err() { task.await.log_err(); } return; } } if let Some(task) = this .update_in(cx, |this, window, cx| { this.open_workspace_for_paths(OpenMode::NewWindow, paths, window, cx) }) .log_err() { task.await.log_err(); } }) .detach(); } pub fn init(app_state: Arc, cx: &mut App) { component::init(); theme_preview::init(cx); toast_layer::init(cx); history_manager::init(app_state.fs.clone(), cx); cx.on_action(|_: &CloseWindow, cx| Workspace::close_global(cx)) .on_action(|_: &Reload, cx| reload(cx)) .on_action(|action: &Open, cx: &mut App| { let app_state = AppState::global(cx); prompt_and_open_paths( app_state, PathPromptOptions { files: true, directories: true, multiple: true, prompt: None, }, action.create_new_window, cx, ); }) .on_action(|_: &OpenFiles, cx: &mut App| { let directories = cx.can_select_mixed_files_and_dirs(); let app_state = AppState::global(cx); prompt_and_open_paths( app_state, PathPromptOptions { files: true, directories, multiple: true, prompt: None, }, true, cx, ); }); } type BuildProjectItemFn = fn(AnyEntity, Entity, Option<&Pane>, &mut Window, &mut App) -> Box; type BuildProjectItemForPathFn = fn( &Entity, &ProjectPath, &mut Window, &mut App, ) -> Option, WorkspaceItemBuilder)>>>; #[derive(Clone, Default)] struct ProjectItemRegistry { build_project_item_fns_by_type: HashMap, build_project_item_for_path_fns: Vec, } impl ProjectItemRegistry { fn register(&mut self) { self.build_project_item_fns_by_type.insert( TypeId::of::(), |item, project, pane, window, cx| { let item = item.downcast().unwrap(); Box::new(cx.new(|cx| T::for_project_item(project, pane, item, window, cx))) as Box }, ); self.build_project_item_for_path_fns .push(|project, project_path, window, cx| { let project_path = project_path.clone(); let is_file = project .read(cx) .entry_for_path(&project_path, cx) .is_some_and(|entry| entry.is_file()); let entry_abs_path = project.read(cx).absolute_path(&project_path, cx); let is_local = project.read(cx).is_local(); let project_item = ::try_open(project, &project_path, cx)?; let project = project.clone(); Some(window.spawn(cx, async move |cx| { match project_item.await.with_context(|| { format!( "opening project path {:?}", entry_abs_path.as_deref().unwrap_or(&project_path.path.as_std_path()) ) }) { Ok(project_item) => { let project_item = project_item; let project_entry_id: Option = project_item.read_with(cx, project::ProjectItem::entry_id); let build_workspace_item = Box::new( |pane: &mut Pane, window: &mut Window, cx: &mut Context| { Box::new(cx.new(|cx| { T::for_project_item( project, Some(pane), project_item, window, cx, ) })) as Box }, ) as Box<_>; Ok((project_entry_id, build_workspace_item)) } Err(e) => { log::warn!("Failed to open a project item: {e:#}"); if e.error_code() == ErrorCode::Internal { if let Some(abs_path) = entry_abs_path.as_deref().filter(|_| is_file) { if let Some(broken_project_item_view) = cx.update(|window, cx| { T::for_broken_project_item( abs_path, is_local, &e, window, cx, ) })? { let build_workspace_item = Box::new( move |_: &mut Pane, _: &mut Window, cx: &mut Context| { cx.new(|_| broken_project_item_view).boxed_clone() }, ) as Box<_>; return Ok((None, build_workspace_item)); } } } Err(e) } } })) }); } fn open_path( &self, project: &Entity, path: &ProjectPath, window: &mut Window, cx: &mut App, ) -> Task, WorkspaceItemBuilder)>> { let Some(open_project_item) = self .build_project_item_for_path_fns .iter() .rev() .find_map(|open_project_item| open_project_item(project, path, window, cx)) else { return Task::ready(Err(anyhow!("cannot open file {:?}", path.path))); }; open_project_item } fn build_item( &self, item: Entity, project: Entity, pane: Option<&Pane>, window: &mut Window, cx: &mut App, ) -> Option> { let build = self .build_project_item_fns_by_type .get(&TypeId::of::())?; Some(build(item.into_any(), project, pane, window, cx)) } } type WorkspaceItemBuilder = Box) -> Box>; impl Global for ProjectItemRegistry {} /// Registers a [ProjectItem] for the app. When opening a file, all the registered /// items will get a chance to open the file, starting from the project item that /// was added last. pub fn register_project_item(cx: &mut App) { cx.default_global::().register::(); } #[derive(Default)] pub struct FollowableViewRegistry(HashMap); struct FollowableViewDescriptor { from_state_proto: fn( Entity, ViewId, &mut Option, &mut Window, &mut App, ) -> Option>>>, to_followable_view: fn(&AnyView) -> Box, } impl Global for FollowableViewRegistry {} impl FollowableViewRegistry { pub fn register(cx: &mut App) { cx.default_global::().0.insert( TypeId::of::(), FollowableViewDescriptor { from_state_proto: |workspace, id, state, window, cx| { I::from_state_proto(workspace, id, state, window, cx).map(|task| { cx.foreground_executor() .spawn(async move { Ok(Box::new(task.await?) as Box<_>) }) }) }, to_followable_view: |view| Box::new(view.clone().downcast::().unwrap()), }, ); } pub fn from_state_proto( workspace: Entity, view_id: ViewId, mut state: Option, window: &mut Window, cx: &mut App, ) -> Option>>> { cx.update_default_global(|this: &mut Self, cx| { this.0.values().find_map(|descriptor| { (descriptor.from_state_proto)(workspace.clone(), view_id, &mut state, window, cx) }) }) } pub fn to_followable_view( view: impl Into, cx: &App, ) -> Option> { let this = cx.try_global::()?; let view = view.into(); let descriptor = this.0.get(&view.entity_type())?; Some((descriptor.to_followable_view)(&view)) } } #[derive(Copy, Clone)] struct SerializableItemDescriptor { deserialize: fn( Entity, WeakEntity, WorkspaceId, ItemId, &mut Window, &mut Context, ) -> Task>>, cleanup: fn(WorkspaceId, Vec, &mut Window, &mut App) -> Task>, view_to_serializable_item: fn(AnyView) -> Box, } #[derive(Default)] struct SerializableItemRegistry { descriptors_by_kind: HashMap, SerializableItemDescriptor>, descriptors_by_type: HashMap, } impl Global for SerializableItemRegistry {} impl SerializableItemRegistry { fn deserialize( item_kind: &str, project: Entity, workspace: WeakEntity, workspace_id: WorkspaceId, item_item: ItemId, window: &mut Window, cx: &mut Context, ) -> Task>> { let Some(descriptor) = Self::descriptor(item_kind, cx) else { return Task::ready(Err(anyhow!( "cannot deserialize {}, descriptor not found", item_kind ))); }; (descriptor.deserialize)(project, workspace, workspace_id, item_item, window, cx) } fn cleanup( item_kind: &str, workspace_id: WorkspaceId, loaded_items: Vec, window: &mut Window, cx: &mut App, ) -> Task> { let Some(descriptor) = Self::descriptor(item_kind, cx) else { return Task::ready(Err(anyhow!( "cannot cleanup {}, descriptor not found", item_kind ))); }; (descriptor.cleanup)(workspace_id, loaded_items, window, cx) } fn view_to_serializable_item_handle( view: AnyView, cx: &App, ) -> Option> { let this = cx.try_global::()?; let descriptor = this.descriptors_by_type.get(&view.entity_type())?; Some((descriptor.view_to_serializable_item)(view)) } fn descriptor(item_kind: &str, cx: &App) -> Option { let this = cx.try_global::()?; this.descriptors_by_kind.get(item_kind).copied() } } pub fn register_serializable_item(cx: &mut App) { let serialized_item_kind = I::serialized_item_kind(); let registry = cx.default_global::(); let descriptor = SerializableItemDescriptor { deserialize: |project, workspace, workspace_id, item_id, window, cx| { let task = I::deserialize(project, workspace, workspace_id, item_id, window, cx); cx.foreground_executor() .spawn(async { Ok(Box::new(task.await?) as Box<_>) }) }, cleanup: |workspace_id, loaded_items, window, cx| { I::cleanup(workspace_id, loaded_items, window, cx) }, view_to_serializable_item: |view| Box::new(view.downcast::().unwrap()), }; registry .descriptors_by_kind .insert(Arc::from(serialized_item_kind), descriptor); registry .descriptors_by_type .insert(TypeId::of::(), descriptor); } pub struct AppState { pub languages: Arc, pub client: Arc, pub user_store: Entity, pub workspace_store: Entity, pub fs: Arc, pub build_window_options: fn(Option, &mut App) -> WindowOptions, pub node_runtime: NodeRuntime, pub session: Entity, } struct GlobalAppState(Arc); impl Global for GlobalAppState {} pub struct WorkspaceStore { workspaces: HashSet<(gpui::AnyWindowHandle, WeakEntity)>, client: Arc, _subscriptions: Vec, } #[derive(Copy, Clone, Debug, Hash, Eq, PartialEq, PartialOrd, Ord)] pub enum CollaboratorId { PeerId(PeerId), Agent, } impl From for CollaboratorId { fn from(peer_id: PeerId) -> Self { CollaboratorId::PeerId(peer_id) } } impl From<&PeerId> for CollaboratorId { fn from(peer_id: &PeerId) -> Self { CollaboratorId::PeerId(*peer_id) } } #[derive(PartialEq, Eq, PartialOrd, Ord, Debug)] struct Follower { project_id: Option, peer_id: PeerId, } impl AppState { #[track_caller] pub fn global(cx: &App) -> Arc { cx.global::().0.clone() } pub fn try_global(cx: &App) -> Option> { cx.try_global::() .map(|state| state.0.clone()) } pub fn set_global(state: Arc, cx: &mut App) { cx.set_global(GlobalAppState(state)); } #[cfg(any(test, feature = "test-support"))] pub fn test(cx: &mut App) -> Arc { use fs::Fs; use node_runtime::NodeRuntime; use session::Session; use settings::SettingsStore; if !cx.has_global::() { let settings_store = SettingsStore::test(cx); cx.set_global(settings_store); } let fs = fs::FakeFs::new(cx.background_executor().clone()); ::set_global(fs.clone(), cx); let languages = Arc::new(LanguageRegistry::test(cx.background_executor().clone())); let clock = Arc::new(clock::FakeSystemClock::new()); let http_client = http_client::FakeHttpClient::with_404_response(); let client = Client::new(clock, http_client, cx); let session = cx.new(|cx| AppSession::new(Session::test(), cx)); let user_store = cx.new(|cx| UserStore::new(client.clone(), cx)); let workspace_store = cx.new(|cx| WorkspaceStore::new(client.clone(), cx)); theme_settings::init(theme::LoadThemes::JustBase, cx); client::init(&client, cx); Arc::new(Self { client, fs, languages, user_store, workspace_store, node_runtime: NodeRuntime::unavailable(), build_window_options: |_, _| Default::default(), session, }) } } struct DelayedDebouncedEditAction { task: Option>, cancel_channel: Option>, } impl DelayedDebouncedEditAction { fn new() -> DelayedDebouncedEditAction { DelayedDebouncedEditAction { task: None, cancel_channel: None, } } fn fire_new( &mut self, delay: Duration, window: &mut Window, cx: &mut Context, func: F, ) where F: 'static + Send + FnOnce(&mut Workspace, &mut Window, &mut Context) -> Task>, { if let Some(channel) = self.cancel_channel.take() { _ = channel.send(()); } let (sender, mut receiver) = oneshot::channel::<()>(); self.cancel_channel = Some(sender); let previous_task = self.task.take(); self.task = Some(cx.spawn_in(window, async move |workspace, cx| { let mut timer = cx.background_executor().timer(delay).fuse(); if let Some(previous_task) = previous_task { previous_task.await; } futures::select_biased! { _ = receiver => return, _ = timer => {} } if let Some(result) = workspace .update_in(cx, |workspace, window, cx| (func)(workspace, window, cx)) .log_err() { result.await.log_err(); } })); } } pub enum Event { PaneAdded(Entity), PaneRemoved, ItemAdded { item: Box, }, ActiveItemChanged, ItemRemoved { item_id: EntityId, }, UserSavedItem { pane: WeakEntity, item: Box, save_intent: SaveIntent, }, ContactRequestedJoin(u64), WorkspaceCreated(WeakEntity), OpenBundledFile { text: Cow<'static, str>, title: &'static str, language: &'static str, }, ZoomChanged, ModalOpened, Activate, PanelAdded(AnyView), } #[derive(Debug, Clone)] pub enum OpenVisible { All, None, OnlyFiles, OnlyDirectories, } enum WorkspaceLocation { // Valid local paths or SSH project to serialize Location(SerializedWorkspaceLocation, PathList), // No valid location found hence clear session id DetachFromSession, // No valid location found to serialize None, } type PromptForNewPath = Box< dyn Fn( &mut Workspace, DirectoryLister, Option, &mut Window, &mut Context, ) -> oneshot::Receiver>>, >; type PromptForOpenPath = Box< dyn Fn( &mut Workspace, DirectoryLister, &mut Window, &mut Context, ) -> oneshot::Receiver>>, >; #[derive(Default)] struct DispatchingKeystrokes { dispatched: HashSet>, queue: VecDeque, task: Option>>, } /// Collects everything project-related for a certain window opened. /// In some way, is a counterpart of a window, as the [`WindowHandle`] could be downcast into `Workspace`. /// /// A `Workspace` usually consists of 1 or more projects, a central pane group, 3 docks and a status bar. /// The `Workspace` owns everybody's state and serves as a default, "global context", /// that can be used to register a global action to be triggered from any place in the window. pub struct Workspace { weak_self: WeakEntity, workspace_actions: Vec) -> Div>>, zoomed: Option, previous_dock_drag_coordinates: Option>, zoomed_position: Option, center: PaneGroup, left_dock: Entity, bottom_dock: Entity, right_dock: Entity, panes: Vec>, active_worktree_override: Option, panes_by_item: HashMap>, active_pane: Entity, last_active_center_pane: Option>, last_active_view_id: Option, status_bar: Entity, pub(crate) modal_layer: Entity, toast_layer: Entity, titlebar_item: Option, notifications: Notifications, suppressed_notifications: HashSet, project: Entity, follower_states: HashMap, last_leaders_by_pane: HashMap, CollaboratorId>, window_edited: bool, last_window_title: Option, dirty_items: HashMap, active_call: Option<(GlobalAnyActiveCall, Vec)>, leader_updates_tx: mpsc::UnboundedSender<(PeerId, proto::UpdateFollowers)>, database_id: Option, app_state: Arc, dispatching_keystrokes: Rc>, _subscriptions: Vec, _apply_leader_updates: Task>, _observe_current_user: Task>, _schedule_serialize_workspace: Option>, _serialize_workspace_task: Option>, _schedule_serialize_ssh_paths: Option>, pane_history_timestamp: Arc, bounds: Bounds, pub centered_layout: bool, bounds_save_task_queued: Option>, on_prompt_for_new_path: Option, on_prompt_for_open_path: Option, terminal_provider: Option>, debugger_provider: Option>, serializable_items_tx: UnboundedSender>, _items_serializer: Task>, session_id: Option, scheduled_tasks: Vec>, last_open_dock_positions: Vec, removing: bool, open_in_dev_container: bool, _dev_container_task: Option>>, _panels_task: Option>>, sidebar_focus_handle: Option, multi_workspace: Option>, } impl EventEmitter for Workspace {} #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)] pub struct ViewId { pub creator: CollaboratorId, pub id: u64, } pub struct FollowerState { center_pane: Entity, dock_pane: Option>, active_view_id: Option, items_by_leader_view_id: HashMap, } struct FollowerView { view: Box, location: Option, } #[derive(Debug, Clone, Copy, PartialEq, Eq, Default)] pub enum OpenMode { /// Open the workspace in a new window. NewWindow, /// Add to the window's multi workspace without activating it (used during deserialization). Add, /// Add to the window's multi workspace and activate it. #[default] Activate, } impl Workspace { pub fn new( workspace_id: Option, project: Entity, app_state: Arc, window: &mut Window, cx: &mut Context, ) -> Self { if let Some(trusted_worktrees) = TrustedWorktrees::try_get_global(cx) { cx.subscribe(&trusted_worktrees, |_, worktrees_store, e, cx| { if let TrustedWorktreesEvent::Trusted(..) = e { // Do not persist auto trusted worktrees if !ProjectSettings::get_global(cx).session.trust_all_worktrees { worktrees_store.update(cx, |worktrees_store, cx| { worktrees_store.schedule_serialization( cx, |new_trusted_worktrees, cx| { let timeout = cx.background_executor().timer(SERIALIZATION_THROTTLE_TIME); let db = WorkspaceDb::global(cx); cx.background_spawn(async move { timeout.await; db.save_trusted_worktrees(new_trusted_worktrees) .await .log_err(); }) }, ) }); } } }) .detach(); cx.observe_global::(|_, cx| { if ProjectSettings::get_global(cx).session.trust_all_worktrees { if let Some(trusted_worktrees) = TrustedWorktrees::try_get_global(cx) { trusted_worktrees.update(cx, |trusted_worktrees, cx| { trusted_worktrees.auto_trust_all(cx); }) } } }) .detach(); } cx.subscribe_in(&project, window, move |this, _, event, window, cx| { match event { project::Event::RemoteIdChanged(_) => { this.update_window_title(window, cx); } project::Event::CollaboratorLeft(peer_id) => { this.collaborator_left(*peer_id, window, cx); } &project::Event::WorktreeRemoved(_) => { this.update_window_title(window, cx); this.serialize_workspace(window, cx); this.update_history(cx); } &project::Event::WorktreeAdded(id) => { this.update_window_title(window, cx); if this .project() .read(cx) .worktree_for_id(id, cx) .is_some_and(|wt| wt.read(cx).is_visible()) { this.serialize_workspace(window, cx); this.update_history(cx); } } project::Event::WorktreeUpdatedEntries(..) => { this.update_window_title(window, cx); this.serialize_workspace(window, cx); } project::Event::DisconnectedFromHost => { this.update_window_edited(window, cx); let leaders_to_unfollow = this.follower_states.keys().copied().collect::>(); for leader_id in leaders_to_unfollow { this.unfollow(leader_id, window, cx); } } project::Event::DisconnectedFromRemote { server_not_running: _, } => { this.update_window_edited(window, cx); } project::Event::Closed => { window.remove_window(); } project::Event::DeletedEntry(_, entry_id) => { for pane in this.panes.iter() { pane.update(cx, |pane, cx| { pane.handle_deleted_project_item(*entry_id, window, cx) }); } } project::Event::Toast { notification_id, message, link, } => this.show_notification( NotificationId::named(notification_id.clone()), cx, |cx| { let mut notification = MessageNotification::new(message.clone(), cx); if let Some(link) = link { notification = notification .more_info_message(link.label) .more_info_url(link.url); } cx.new(|_| notification) }, ), project::Event::HideToast { notification_id } => { this.dismiss_notification(&NotificationId::named(notification_id.clone()), cx) } project::Event::LanguageServerPrompt(request) => { struct LanguageServerPrompt; this.show_notification( NotificationId::composite::(request.id), cx, |cx| { cx.new(|cx| { notifications::LanguageServerPrompt::new(request.clone(), cx) }) }, ); } project::Event::AgentLocationChanged => { this.handle_agent_location_changed(window, cx) } _ => {} } cx.notify() }) .detach(); cx.subscribe_in( &project.read(cx).breakpoint_store(), window, |workspace, _, event, window, cx| match event { BreakpointStoreEvent::BreakpointsUpdated(_, _) | BreakpointStoreEvent::BreakpointsCleared(_) => { workspace.serialize_workspace(window, cx); } BreakpointStoreEvent::SetDebugLine | BreakpointStoreEvent::ClearDebugLines => {} }, ) .detach(); if let Some(toolchain_store) = project.read(cx).toolchain_store() { cx.subscribe_in( &toolchain_store, window, |workspace, _, event, window, cx| match event { ToolchainStoreEvent::CustomToolchainsModified => { workspace.serialize_workspace(window, cx); } _ => {} }, ) .detach(); } cx.on_focus_lost(window, |this, window, cx| { let focus_handle = this.focus_handle(cx); window.focus(&focus_handle, cx); }) .detach(); let weak_handle = cx.entity().downgrade(); let pane_history_timestamp = Arc::new(AtomicUsize::new(0)); let center_pane = cx.new(|cx| { let mut center_pane = Pane::new( weak_handle.clone(), project.clone(), pane_history_timestamp.clone(), None, NewFile.boxed_clone(), true, window, cx, ); center_pane.set_can_split(Some(Arc::new(|_, _, _, _| true))); center_pane.set_should_display_welcome_page(true); center_pane }); cx.subscribe_in(¢er_pane, window, Self::handle_pane_event) .detach(); window.focus(¢er_pane.focus_handle(cx), cx); cx.emit(Event::PaneAdded(center_pane.clone())); let any_window_handle = window.window_handle(); app_state.workspace_store.update(cx, |store, _| { store .workspaces .insert((any_window_handle, weak_handle.clone())); }); let mut current_user = app_state.user_store.read(cx).watch_current_user(); let mut connection_status = app_state.client.status(); let _observe_current_user = cx.spawn_in(window, async move |this, cx| { current_user.next().await; connection_status.next().await; let mut stream = Stream::map(current_user, drop).merge(Stream::map(connection_status, drop)); while stream.recv().await.is_some() { this.update(cx, |_, cx| cx.notify())?; } anyhow::Ok(()) }); // All leader updates are enqueued and then processed in a single task, so // that each asynchronous operation can be run in order. let (leader_updates_tx, mut leader_updates_rx) = mpsc::unbounded::<(PeerId, proto::UpdateFollowers)>(); let _apply_leader_updates = cx.spawn_in(window, async move |this, cx| { while let Some((leader_id, update)) = leader_updates_rx.next().await { Self::process_leader_update(&this, leader_id, update, cx) .await .log_err(); } Ok(()) }); cx.emit(Event::WorkspaceCreated(weak_handle.clone())); let modal_layer = cx.new(|_| ModalLayer::new()); let toast_layer = cx.new(|_| ToastLayer::new()); cx.subscribe( &modal_layer, |_, _, _: &modal_layer::ModalOpenedEvent, cx| { cx.emit(Event::ModalOpened); }, ) .detach(); let left_dock = Dock::new(DockPosition::Left, modal_layer.clone(), window, cx); let bottom_dock = Dock::new(DockPosition::Bottom, modal_layer.clone(), window, cx); let right_dock = Dock::new(DockPosition::Right, modal_layer.clone(), window, cx); let left_dock_buttons = cx.new(|cx| PanelButtons::new(left_dock.clone(), cx)); let bottom_dock_buttons = cx.new(|cx| PanelButtons::new(bottom_dock.clone(), cx)); let right_dock_buttons = cx.new(|cx| PanelButtons::new(right_dock.clone(), cx)); let multi_workspace = window .root::() .flatten() .map(|mw| mw.downgrade()); let status_bar = cx.new(|cx| { let mut status_bar = StatusBar::new(¢er_pane.clone(), multi_workspace.clone(), window, cx); status_bar.add_left_item(left_dock_buttons, window, cx); status_bar.add_right_item(right_dock_buttons, window, cx); status_bar.add_right_item(bottom_dock_buttons, window, cx); status_bar }); let session_id = app_state.session.read(cx).id().to_owned(); let mut active_call = None; if let Some(call) = GlobalAnyActiveCall::try_global(cx).cloned() { let subscriptions = vec![ call.0 .subscribe(window, cx, Box::new(Self::on_active_call_event)), ]; active_call = Some((call, subscriptions)); } let (serializable_items_tx, serializable_items_rx) = mpsc::unbounded::>(); let _items_serializer = cx.spawn_in(window, async move |this, cx| { Self::serialize_items(&this, serializable_items_rx, cx).await }); let subscriptions = vec![ cx.observe_window_activation(window, Self::on_window_activation_changed), cx.observe_window_bounds(window, move |this, window, cx| { if this.bounds_save_task_queued.is_some() { return; } this.bounds_save_task_queued = Some(cx.spawn_in(window, async move |this, cx| { cx.background_executor() .timer(Duration::from_millis(100)) .await; this.update_in(cx, |this, window, cx| { this.save_window_bounds(window, cx).detach(); this.bounds_save_task_queued.take(); }) .ok(); })); cx.notify(); }), cx.observe_window_appearance(window, |_, window, cx| { let window_appearance = window.appearance(); *SystemAppearance::global_mut(cx) = SystemAppearance(window_appearance.into()); theme_settings::reload_theme(cx); theme_settings::reload_icon_theme(cx); }), cx.on_release({ let weak_handle = weak_handle.clone(); move |this, cx| { this.app_state.workspace_store.update(cx, move |store, _| { store.workspaces.retain(|(_, weak)| weak != &weak_handle); }) } }), ]; cx.defer_in(window, move |this, window, cx| { this.update_window_title(window, cx); this.show_initial_notifications(cx); }); let mut center = PaneGroup::new(center_pane.clone()); center.set_is_center(true); center.mark_positions(cx); Workspace { weak_self: weak_handle.clone(), zoomed: None, zoomed_position: None, previous_dock_drag_coordinates: None, center, panes: vec![center_pane.clone()], panes_by_item: Default::default(), active_pane: center_pane.clone(), last_active_center_pane: Some(center_pane.downgrade()), last_active_view_id: None, status_bar, modal_layer, toast_layer, titlebar_item: None, active_worktree_override: None, notifications: Notifications::default(), suppressed_notifications: HashSet::default(), left_dock, bottom_dock, right_dock, _panels_task: None, project: project.clone(), follower_states: Default::default(), last_leaders_by_pane: Default::default(), dispatching_keystrokes: Default::default(), window_edited: false, last_window_title: None, dirty_items: Default::default(), active_call, database_id: workspace_id, app_state, _observe_current_user, _apply_leader_updates, _schedule_serialize_workspace: None, _serialize_workspace_task: None, _schedule_serialize_ssh_paths: None, leader_updates_tx, _subscriptions: subscriptions, pane_history_timestamp, workspace_actions: Default::default(), // This data will be incorrect, but it will be overwritten by the time it needs to be used. bounds: Default::default(), centered_layout: false, bounds_save_task_queued: None, on_prompt_for_new_path: None, on_prompt_for_open_path: None, terminal_provider: None, debugger_provider: None, serializable_items_tx, _items_serializer, session_id: Some(session_id), scheduled_tasks: Vec::new(), last_open_dock_positions: Vec::new(), removing: false, sidebar_focus_handle: None, multi_workspace, open_in_dev_container: false, _dev_container_task: None, } } pub fn new_local( abs_paths: Vec, app_state: Arc, requesting_window: Option>, env: Option>, init: Option) + Send>>, open_mode: OpenMode, cx: &mut App, ) -> Task> { let project_handle = Project::local( app_state.client.clone(), app_state.node_runtime.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), env, Default::default(), cx, ); let db = WorkspaceDb::global(cx); let kvp = db::kvp::KeyValueStore::global(cx); cx.spawn(async move |cx| { let mut paths_to_open = Vec::with_capacity(abs_paths.len()); for path in abs_paths.into_iter() { if let Some(canonical) = app_state.fs.canonicalize(&path).await.ok() { paths_to_open.push(canonical) } else { paths_to_open.push(path) } } let serialized_workspace = db.workspace_for_roots(paths_to_open.as_slice()); if let Some(paths) = serialized_workspace.as_ref().map(|ws| &ws.paths) { paths_to_open = paths.ordered_paths().cloned().collect(); if !paths.is_lexicographically_ordered() { project_handle.update(cx, |project, cx| { project.set_worktrees_reordered(true, cx); }); } } // Get project paths for all of the abs_paths let mut project_paths: Vec<(PathBuf, Option)> = Vec::with_capacity(paths_to_open.len()); for path in paths_to_open.into_iter() { if let Some((_, project_entry)) = cx .update(|cx| { Workspace::project_path_for_path(project_handle.clone(), &path, true, cx) }) .await .log_err() { project_paths.push((path, Some(project_entry))); } else { project_paths.push((path, None)); } } let workspace_id = if let Some(serialized_workspace) = serialized_workspace.as_ref() { serialized_workspace.id } else { db.next_id().await.unwrap_or_else(|_| Default::default()) }; let toolchains = db.toolchains(workspace_id).await?; for (toolchain, worktree_path, path) in toolchains { let toolchain_path = PathBuf::from(toolchain.path.clone().to_string()); let Some(worktree_id) = project_handle.read_with(cx, |this, cx| { this.find_worktree(&worktree_path, cx) .and_then(|(worktree, rel_path)| { if rel_path.is_empty() { Some(worktree.read(cx).id()) } else { None } }) }) else { // We did not find a worktree with a given path, but that's whatever. continue; }; if !app_state.fs.is_file(toolchain_path.as_path()).await { continue; } project_handle .update(cx, |this, cx| { this.activate_toolchain(ProjectPath { worktree_id, path }, toolchain, cx) }) .await; } if let Some(workspace) = serialized_workspace.as_ref() { project_handle.update(cx, |this, cx| { for (scope, toolchains) in &workspace.user_toolchains { for toolchain in toolchains { this.add_toolchain(toolchain.clone(), scope.clone(), cx); } } }); } let window_to_replace = match open_mode { OpenMode::NewWindow => None, _ => requesting_window, }; let (window, workspace): (WindowHandle, Entity) = if let Some(window) = window_to_replace { let centered_layout = serialized_workspace .as_ref() .map(|w| w.centered_layout) .unwrap_or(false); let workspace = window.update(cx, |multi_workspace, window, cx| { let workspace = cx.new(|cx| { let mut workspace = Workspace::new( Some(workspace_id), project_handle.clone(), app_state.clone(), window, cx, ); workspace.centered_layout = centered_layout; // Call init callback to add items before window renders if let Some(init) = init { init(&mut workspace, window, cx); } workspace }); match open_mode { OpenMode::Activate => { multi_workspace.activate(workspace.clone(), window, cx); } OpenMode::Add => { multi_workspace.add(workspace.clone(), &*window, cx); } OpenMode::NewWindow => { unreachable!() } } workspace })?; (window, workspace) } else { let window_bounds_override = window_bounds_env_override(); let (window_bounds, display) = if let Some(bounds) = window_bounds_override { (Some(WindowBounds::Windowed(bounds)), None) } else if let Some(workspace) = serialized_workspace.as_ref() && let Some(display) = workspace.display && let Some(bounds) = workspace.window_bounds.as_ref() { // Reopening an existing workspace - restore its saved bounds (Some(bounds.0), Some(display)) } else if let Some((display, bounds)) = persistence::read_default_window_bounds(&kvp) { // New or empty workspace - use the last known window bounds (Some(bounds), Some(display)) } else { // New window - let GPUI's default_bounds() handle cascading (None, None) }; // Use the serialized workspace to construct the new window let mut options = cx.update(|cx| (app_state.build_window_options)(display, cx)); options.window_bounds = window_bounds; let centered_layout = serialized_workspace .as_ref() .map(|w| w.centered_layout) .unwrap_or(false); let window = cx.open_window(options, { let app_state = app_state.clone(); let project_handle = project_handle.clone(); move |window, cx| { let workspace = cx.new(|cx| { let mut workspace = Workspace::new( Some(workspace_id), project_handle, app_state, window, cx, ); workspace.centered_layout = centered_layout; // Call init callback to add items before window renders if let Some(init) = init { init(&mut workspace, window, cx); } workspace }); cx.new(|cx| MultiWorkspace::new(workspace, window, cx)) } })?; let workspace = window.update(cx, |multi_workspace: &mut MultiWorkspace, _, _cx| { multi_workspace.workspace().clone() })?; (window, workspace) }; notify_if_database_failed(window, cx); // Check if this is an empty workspace (no paths to open) // An empty workspace is one where project_paths is empty let is_empty_workspace = project_paths.is_empty(); // Check if serialized workspace has paths before it's moved let serialized_workspace_has_paths = serialized_workspace .as_ref() .map(|ws| !ws.paths.is_empty()) .unwrap_or(false); let opened_items = window .update(cx, |_, window, cx| { workspace.update(cx, |_workspace: &mut Workspace, cx| { open_items(serialized_workspace, project_paths, window, cx) }) })? .await .unwrap_or_default(); // Restore default dock state for empty workspaces // Only restore if: // 1. This is an empty workspace (no paths), AND // 2. The serialized workspace either doesn't exist or has no paths if is_empty_workspace && !serialized_workspace_has_paths { if let Some(default_docks) = persistence::read_default_dock_state(&kvp) { window .update(cx, |_, window, cx| { workspace.update(cx, |workspace, cx| { for (dock, serialized_dock) in [ (&workspace.right_dock, &default_docks.right), (&workspace.left_dock, &default_docks.left), (&workspace.bottom_dock, &default_docks.bottom), ] { dock.update(cx, |dock, cx| { dock.serialized_dock = Some(serialized_dock.clone()); dock.restore_state(window, cx); }); } cx.notify(); }); }) .log_err(); } } window .update(cx, |_, _window, cx| { workspace.update(cx, |this: &mut Workspace, cx| { this.update_history(cx); }); }) .log_err(); Ok(OpenResult { window, workspace, opened_items, }) }) } pub fn project_group_key(&self, cx: &App) -> ProjectGroupKey { self.project.read(cx).project_group_key(cx) } pub fn weak_handle(&self) -> WeakEntity { self.weak_self.clone() } pub fn left_dock(&self) -> &Entity { &self.left_dock } pub fn bottom_dock(&self) -> &Entity { &self.bottom_dock } pub fn set_bottom_dock_layout( &mut self, layout: BottomDockLayout, window: &mut Window, cx: &mut Context, ) { let fs = self.project().read(cx).fs(); settings::update_settings_file(fs.clone(), cx, move |content, _cx| { content.workspace.bottom_dock_layout = Some(layout); }); cx.notify(); self.serialize_workspace(window, cx); } pub fn right_dock(&self) -> &Entity { &self.right_dock } pub fn all_docks(&self) -> [&Entity; 3] { [&self.left_dock, &self.bottom_dock, &self.right_dock] } pub fn capture_dock_state(&self, _window: &Window, cx: &App) -> DockStructure { let left_dock = self.left_dock.read(cx); let left_visible = left_dock.is_open(); let left_active_panel = left_dock .active_panel() .map(|panel| panel.persistent_name().to_string()); // `zoomed_position` is kept in sync with individual panel zoom state // by the dock code in `Dock::new` and `Dock::add_panel`. let left_dock_zoom = self.zoomed_position == Some(DockPosition::Left); let right_dock = self.right_dock.read(cx); let right_visible = right_dock.is_open(); let right_active_panel = right_dock .active_panel() .map(|panel| panel.persistent_name().to_string()); let right_dock_zoom = self.zoomed_position == Some(DockPosition::Right); let bottom_dock = self.bottom_dock.read(cx); let bottom_visible = bottom_dock.is_open(); let bottom_active_panel = bottom_dock .active_panel() .map(|panel| panel.persistent_name().to_string()); let bottom_dock_zoom = self.zoomed_position == Some(DockPosition::Bottom); DockStructure { left: DockData { visible: left_visible, active_panel: left_active_panel, zoom: left_dock_zoom, }, right: DockData { visible: right_visible, active_panel: right_active_panel, zoom: right_dock_zoom, }, bottom: DockData { visible: bottom_visible, active_panel: bottom_active_panel, zoom: bottom_dock_zoom, }, } } pub fn set_dock_structure( &self, docks: DockStructure, window: &mut Window, cx: &mut Context, ) { for (dock, data) in [ (&self.left_dock, docks.left), (&self.bottom_dock, docks.bottom), (&self.right_dock, docks.right), ] { dock.update(cx, |dock, cx| { dock.serialized_dock = Some(data); dock.restore_state(window, cx); }); } } pub fn open_item_abs_paths(&self, cx: &App) -> Vec { self.items(cx) .filter_map(|item| { let project_path = item.project_path(cx)?; self.project.read(cx).absolute_path(&project_path, cx) }) .collect() } pub fn dock_at_position(&self, position: DockPosition) -> &Entity { match position { DockPosition::Left => &self.left_dock, DockPosition::Bottom => &self.bottom_dock, DockPosition::Right => &self.right_dock, } } pub fn agent_panel_position(&self, cx: &App) -> Option { self.all_docks().into_iter().find_map(|dock| { let dock = dock.read(cx); dock.has_agent_panel(cx).then_some(dock.position()) }) } pub fn panel_size_state(&self, cx: &App) -> Option { self.all_docks().into_iter().find_map(|dock| { let dock = dock.read(cx); let panel = dock.panel::()?; dock.stored_panel_size_state(&panel) }) } pub fn persisted_panel_size_state( &self, panel_key: &'static str, cx: &App, ) -> Option { dock::Dock::load_persisted_size_state(self, panel_key, cx) } pub fn persist_panel_size_state( &self, panel_key: &str, size_state: dock::PanelSizeState, cx: &mut App, ) { let Some(workspace_id) = self .database_id() .map(|id| i64::from(id).to_string()) .or(self.session_id()) else { return; }; let kvp = db::kvp::KeyValueStore::global(cx); let panel_key = panel_key.to_string(); cx.background_spawn(async move { let scope = kvp.scoped(dock::PANEL_SIZE_STATE_KEY); scope .write( format!("{workspace_id}:{panel_key}"), serde_json::to_string(&size_state)?, ) .await }) .detach_and_log_err(cx); } pub fn set_panel_size_state( &mut self, size_state: dock::PanelSizeState, window: &mut Window, cx: &mut Context, ) -> bool { let Some(panel) = self.panel::(cx) else { return false; }; let dock = self.dock_at_position(panel.position(window, cx)); let did_set = dock.update(cx, |dock, cx| { dock.set_panel_size_state(&panel, size_state, cx) }); if did_set { self.persist_panel_size_state(T::panel_key(), size_state, cx); } did_set } pub fn toggle_dock_panel_flexible_size( &self, dock: &Entity, panel: &dyn PanelHandle, window: &mut Window, cx: &mut App, ) { let position = dock.read(cx).position(); let current_size = self.dock_size(&dock.read(cx), window, cx); let current_flex = current_size.and_then(|size| self.dock_flex_for_size(position, size, window, cx)); dock.update(cx, |dock, cx| { dock.toggle_panel_flexible_size(panel, current_size, current_flex, window, cx); }); } fn dock_size(&self, dock: &Dock, window: &Window, cx: &App) -> Option { let panel = dock.active_panel()?; let size_state = dock .stored_panel_size_state(panel.as_ref()) .unwrap_or_default(); let position = dock.position(); let use_flex = panel.has_flexible_size(window, cx); if position.axis() == Axis::Horizontal && use_flex && let Some(flex) = size_state.flex.or_else(|| self.default_dock_flex(position)) { let workspace_width = self.bounds.size.width; if workspace_width <= Pixels::ZERO { return None; } let flex = flex.max(0.001); let opposite = self.opposite_dock_panel_and_size_state(position, window, cx); if let Some(opposite_flex) = opposite.as_ref().and_then(|(_, s)| s.flex) { // Both docks are flex items sharing the full workspace width. let total_flex = flex + 1.0 + opposite_flex; return Some((flex / total_flex * workspace_width).max(RESIZE_HANDLE_SIZE)); } else { // Opposite dock is fixed-width; flex items share (W - fixed). let opposite_fixed = opposite .map(|(panel, s)| s.size.unwrap_or_else(|| panel.default_size(window, cx))) .unwrap_or_default(); let available = (workspace_width - opposite_fixed).max(RESIZE_HANDLE_SIZE); return Some((flex / (flex + 1.0) * available).max(RESIZE_HANDLE_SIZE)); } } Some( size_state .size .unwrap_or_else(|| panel.default_size(window, cx)), ) } pub fn dock_flex_for_size( &self, position: DockPosition, size: Pixels, window: &Window, cx: &App, ) -> Option { if position.axis() != Axis::Horizontal { return None; } let workspace_width = self.bounds.size.width; if workspace_width <= Pixels::ZERO { return None; } let opposite = self.opposite_dock_panel_and_size_state(position, window, cx); if let Some(opposite_flex) = opposite.as_ref().and_then(|(_, s)| s.flex) { let size = size.clamp(px(0.), workspace_width - px(1.)); Some((size * (1.0 + opposite_flex) / (workspace_width - size)).max(0.0)) } else { let opposite_width = opposite .map(|(panel, s)| s.size.unwrap_or_else(|| panel.default_size(window, cx))) .unwrap_or_default(); let available = (workspace_width - opposite_width).max(RESIZE_HANDLE_SIZE); let remaining = (available - size).max(px(1.)); Some((size / remaining).max(0.0)) } } fn opposite_dock_panel_and_size_state( &self, position: DockPosition, window: &Window, cx: &App, ) -> Option<(Arc, PanelSizeState)> { let opposite_position = match position { DockPosition::Left => DockPosition::Right, DockPosition::Right => DockPosition::Left, DockPosition::Bottom => return None, }; let opposite_dock = self.dock_at_position(opposite_position).read(cx); let panel = opposite_dock.visible_panel()?; let mut size_state = opposite_dock .stored_panel_size_state(panel.as_ref()) .unwrap_or_default(); if size_state.flex.is_none() && panel.has_flexible_size(window, cx) { size_state.flex = self.default_dock_flex(opposite_position); } Some((panel.clone(), size_state)) } pub fn default_dock_flex(&self, position: DockPosition) -> Option { if position.axis() != Axis::Horizontal { return None; } let pane = self.last_active_center_pane.clone()?.upgrade()?; Some(self.center.width_fraction_for_pane(&pane).unwrap_or(1.0)) } pub fn is_edited(&self) -> bool { self.window_edited } pub fn add_panel( &mut self, panel: Entity, window: &mut Window, cx: &mut Context, ) { let focus_handle = panel.panel_focus_handle(cx); cx.on_focus_in(&focus_handle, window, Self::handle_panel_focused) .detach(); let dock_position = panel.position(window, cx); let dock = self.dock_at_position(dock_position); let any_panel = panel.to_any(); let persisted_size_state = self.persisted_panel_size_state(T::panel_key(), cx) .or_else(|| { load_legacy_panel_size(T::panel_key(), dock_position, self, cx).map(|size| { let state = dock::PanelSizeState { size: Some(size), flex: None, }; self.persist_panel_size_state(T::panel_key(), state, cx); state }) }); dock.update(cx, |dock, cx| { let index = dock.add_panel(panel.clone(), self.weak_self.clone(), window, cx); if let Some(size_state) = persisted_size_state { dock.set_panel_size_state(&panel, size_state, cx); } index }); cx.emit(Event::PanelAdded(any_panel)); } pub fn remove_panel( &mut self, panel: &Entity, window: &mut Window, cx: &mut Context, ) { for dock in [&self.left_dock, &self.bottom_dock, &self.right_dock] { dock.update(cx, |dock, cx| dock.remove_panel(panel, window, cx)); } } pub fn status_bar(&self) -> &Entity { &self.status_bar } pub fn set_sidebar_focus_handle(&mut self, handle: Option) { self.sidebar_focus_handle = handle; } pub fn status_bar_visible(&self, cx: &App) -> bool { StatusBarSettings::get_global(cx).show } pub fn multi_workspace(&self) -> Option<&WeakEntity> { self.multi_workspace.as_ref() } pub fn set_multi_workspace( &mut self, multi_workspace: WeakEntity, cx: &mut App, ) { self.status_bar.update(cx, |status_bar, cx| { status_bar.set_multi_workspace(multi_workspace.clone(), cx); }); self.multi_workspace = Some(multi_workspace); } pub fn app_state(&self) -> &Arc { &self.app_state } pub fn set_panels_task(&mut self, task: Task>) { self._panels_task = Some(task); } pub fn take_panels_task(&mut self) -> Option>> { self._panels_task.take() } pub fn user_store(&self) -> &Entity { &self.app_state.user_store } pub fn project(&self) -> &Entity { &self.project } pub fn path_style(&self, cx: &App) -> PathStyle { self.project.read(cx).path_style(cx) } pub fn recently_activated_items(&self, cx: &App) -> HashMap { let mut history: HashMap = HashMap::default(); for pane_handle in &self.panes { let pane = pane_handle.read(cx); for entry in pane.activation_history() { history.insert( entry.entity_id, history .get(&entry.entity_id) .cloned() .unwrap_or(0) .max(entry.timestamp), ); } } history } pub fn recent_active_item_by_type(&self, cx: &App) -> Option> { let mut recent_item: Option> = None; let mut recent_timestamp = 0; for pane_handle in &self.panes { let pane = pane_handle.read(cx); let item_map: HashMap> = pane.items().map(|item| (item.item_id(), item)).collect(); for entry in pane.activation_history() { if entry.timestamp > recent_timestamp && let Some(&item) = item_map.get(&entry.entity_id) && let Some(typed_item) = item.act_as::(cx) { recent_timestamp = entry.timestamp; recent_item = Some(typed_item); } } } recent_item } pub fn recent_navigation_history_iter( &self, cx: &App, ) -> impl Iterator)> + use<> { let mut abs_paths_opened: HashMap> = HashMap::default(); let mut history: HashMap, usize)> = HashMap::default(); for pane in &self.panes { let pane = pane.read(cx); pane.nav_history() .for_each_entry(cx, &mut |entry, (project_path, fs_path)| { if let Some(fs_path) = &fs_path { abs_paths_opened .entry(fs_path.clone()) .or_default() .insert(project_path.clone()); } let timestamp = entry.timestamp; match history.entry(project_path) { hash_map::Entry::Occupied(mut entry) => { let (_, old_timestamp) = entry.get(); if ×tamp > old_timestamp { entry.insert((fs_path, timestamp)); } } hash_map::Entry::Vacant(entry) => { entry.insert((fs_path, timestamp)); } } }); if let Some(item) = pane.active_item() && let Some(project_path) = item.project_path(cx) { let fs_path = self.project.read(cx).absolute_path(&project_path, cx); if let Some(fs_path) = &fs_path { abs_paths_opened .entry(fs_path.clone()) .or_default() .insert(project_path.clone()); } history.insert(project_path, (fs_path, std::usize::MAX)); } } history .into_iter() .sorted_by_key(|(_, (_, order))| *order) .map(|(project_path, (fs_path, _))| (project_path, fs_path)) .rev() .filter(move |(history_path, abs_path)| { let latest_project_path_opened = abs_path .as_ref() .and_then(|abs_path| abs_paths_opened.get(abs_path)) .and_then(|project_paths| { project_paths .iter() .max_by(|b1, b2| b1.worktree_id.cmp(&b2.worktree_id)) }); latest_project_path_opened.is_none_or(|path| path == history_path) }) } pub fn recent_navigation_history( &self, limit: Option, cx: &App, ) -> Vec<(ProjectPath, Option)> { self.recent_navigation_history_iter(cx) .take(limit.unwrap_or(usize::MAX)) .collect() } pub fn clear_navigation_history(&mut self, _window: &mut Window, cx: &mut Context) { for pane in &self.panes { pane.update(cx, |pane, cx| pane.nav_history_mut().clear(cx)); } } fn navigate_history( &mut self, pane: WeakEntity, mode: NavigationMode, window: &mut Window, cx: &mut Context, ) -> Task> { self.navigate_history_impl( pane, mode, window, &mut |history, cx| history.pop(mode, cx), cx, ) } fn navigate_tag_history( &mut self, pane: WeakEntity, mode: TagNavigationMode, window: &mut Window, cx: &mut Context, ) -> Task> { self.navigate_history_impl( pane, NavigationMode::Normal, window, &mut |history, _cx| history.pop_tag(mode), cx, ) } fn navigate_history_impl( &mut self, pane: WeakEntity, mode: NavigationMode, window: &mut Window, cb: &mut dyn FnMut(&mut NavHistory, &mut App) -> Option, cx: &mut Context, ) -> Task> { let to_load = if let Some(pane) = pane.upgrade() { pane.update(cx, |pane, cx| { window.focus(&pane.focus_handle(cx), cx); loop { // Retrieve the weak item handle from the history. let entry = cb(pane.nav_history_mut(), cx)?; // If the item is still present in this pane, then activate it. if let Some(index) = entry .item .upgrade() .and_then(|v| pane.index_for_item(v.as_ref())) { let prev_active_item_index = pane.active_item_index(); pane.nav_history_mut().set_mode(mode); pane.activate_item(index, true, true, window, cx); pane.nav_history_mut().set_mode(NavigationMode::Normal); let mut navigated = prev_active_item_index != pane.active_item_index(); if let Some(data) = entry.data { navigated |= pane.active_item()?.navigate(data, window, cx); } if navigated { break None; } } else { // If the item is no longer present in this pane, then retrieve its // path info in order to reopen it. break pane .nav_history() .path_for_item(entry.item.id()) .map(|(project_path, abs_path)| (project_path, abs_path, entry)); } } }) } else { None }; if let Some((project_path, abs_path, entry)) = to_load { // If the item was no longer present, then load it again from its previous path, first try the local path let open_by_project_path = self.load_path(project_path.clone(), window, cx); cx.spawn_in(window, async move |workspace, cx| { let open_by_project_path = open_by_project_path.await; let mut navigated = false; match open_by_project_path .with_context(|| format!("Navigating to {project_path:?}")) { Ok((project_entry_id, build_item)) => { let prev_active_item_id = pane.update(cx, |pane, _| { pane.nav_history_mut().set_mode(mode); pane.active_item().map(|p| p.item_id()) })?; pane.update_in(cx, |pane, window, cx| { let item = pane.open_item( project_entry_id, project_path, true, entry.is_preview, true, None, window, cx, build_item, ); navigated |= Some(item.item_id()) != prev_active_item_id; pane.nav_history_mut().set_mode(NavigationMode::Normal); if let Some(data) = entry.data { navigated |= item.navigate(data, window, cx); } })?; } Err(open_by_project_path_e) => { // Fall back to opening by abs path, in case an external file was opened and closed, // and its worktree is now dropped if let Some(abs_path) = abs_path { let prev_active_item_id = pane.update(cx, |pane, _| { pane.nav_history_mut().set_mode(mode); pane.active_item().map(|p| p.item_id()) })?; let open_by_abs_path = workspace.update_in(cx, |workspace, window, cx| { workspace.open_abs_path(abs_path.clone(), OpenOptions { visible: Some(OpenVisible::None), ..Default::default() }, window, cx) })?; match open_by_abs_path .await .with_context(|| format!("Navigating to {abs_path:?}")) { Ok(item) => { pane.update_in(cx, |pane, window, cx| { navigated |= Some(item.item_id()) != prev_active_item_id; pane.nav_history_mut().set_mode(NavigationMode::Normal); if let Some(data) = entry.data { navigated |= item.navigate(data, window, cx); } })?; } Err(open_by_abs_path_e) => { log::error!("Failed to navigate history: {open_by_project_path_e:#} and {open_by_abs_path_e:#}"); } } } } } if !navigated { workspace .update_in(cx, |workspace, window, cx| { Self::navigate_history(workspace, pane, mode, window, cx) })? .await?; } Ok(()) }) } else { Task::ready(Ok(())) } } pub fn go_back( &mut self, pane: WeakEntity, window: &mut Window, cx: &mut Context, ) -> Task> { self.navigate_history(pane, NavigationMode::GoingBack, window, cx) } pub fn go_forward( &mut self, pane: WeakEntity, window: &mut Window, cx: &mut Context, ) -> Task> { self.navigate_history(pane, NavigationMode::GoingForward, window, cx) } pub fn reopen_closed_item( &mut self, window: &mut Window, cx: &mut Context, ) -> Task> { self.navigate_history( self.active_pane().downgrade(), NavigationMode::ReopeningClosedItem, window, cx, ) } pub fn client(&self) -> &Arc { &self.app_state.client } pub fn set_titlebar_item(&mut self, item: AnyView, _: &mut Window, cx: &mut Context) { self.titlebar_item = Some(item); cx.notify(); } pub fn set_prompt_for_new_path(&mut self, prompt: PromptForNewPath) { self.on_prompt_for_new_path = Some(prompt) } pub fn set_prompt_for_open_path(&mut self, prompt: PromptForOpenPath) { self.on_prompt_for_open_path = Some(prompt) } pub fn set_terminal_provider(&mut self, provider: impl TerminalProvider + 'static) { self.terminal_provider = Some(Box::new(provider)); } pub fn set_debugger_provider(&mut self, provider: impl DebuggerProvider + 'static) { self.debugger_provider = Some(Arc::new(provider)); } pub fn set_open_in_dev_container(&mut self, value: bool) { self.open_in_dev_container = value; } pub fn open_in_dev_container(&self) -> bool { self.open_in_dev_container } pub fn set_dev_container_task(&mut self, task: Task>) { self._dev_container_task = Some(task); } pub fn debugger_provider(&self) -> Option> { self.debugger_provider.clone() } pub fn prompt_for_open_path( &mut self, path_prompt_options: PathPromptOptions, lister: DirectoryLister, window: &mut Window, cx: &mut Context, ) -> oneshot::Receiver>> { if !lister.is_local(cx) || !WorkspaceSettings::get_global(cx).use_system_path_prompts { let prompt = self.on_prompt_for_open_path.take().unwrap(); let rx = prompt(self, lister, window, cx); self.on_prompt_for_open_path = Some(prompt); rx } else { let (tx, rx) = oneshot::channel(); let abs_path = cx.prompt_for_paths(path_prompt_options); cx.spawn_in(window, async move |workspace, cx| { let Ok(result) = abs_path.await else { return Ok(()); }; match result { Ok(result) => { tx.send(result).ok(); } Err(err) => { let rx = workspace.update_in(cx, |workspace, window, cx| { workspace.show_portal_error(err.to_string(), cx); let prompt = workspace.on_prompt_for_open_path.take().unwrap(); let rx = prompt(workspace, lister, window, cx); workspace.on_prompt_for_open_path = Some(prompt); rx })?; if let Ok(path) = rx.await { tx.send(path).ok(); } } }; anyhow::Ok(()) }) .detach(); rx } } pub fn prompt_for_new_path( &mut self, lister: DirectoryLister, suggested_name: Option, window: &mut Window, cx: &mut Context, ) -> oneshot::Receiver>> { if self.project.read(cx).is_via_collab() || self.project.read(cx).is_via_remote_server() || !WorkspaceSettings::get_global(cx).use_system_path_prompts { let prompt = self.on_prompt_for_new_path.take().unwrap(); let rx = prompt(self, lister, suggested_name, window, cx); self.on_prompt_for_new_path = Some(prompt); return rx; } let (tx, rx) = oneshot::channel(); cx.spawn_in(window, async move |workspace, cx| { let abs_path = workspace.update(cx, |workspace, cx| { let relative_to = workspace .most_recent_active_path(cx) .and_then(|p| p.parent().map(|p| p.to_path_buf())) .or_else(|| { let project = workspace.project.read(cx); project.visible_worktrees(cx).find_map(|worktree| { Some(worktree.read(cx).as_local()?.abs_path().to_path_buf()) }) }) .or_else(std::env::home_dir) .unwrap_or_else(|| PathBuf::from("")); cx.prompt_for_new_path(&relative_to, suggested_name.as_deref()) })?; let abs_path = match abs_path.await? { Ok(path) => path, Err(err) => { let rx = workspace.update_in(cx, |workspace, window, cx| { workspace.show_portal_error(err.to_string(), cx); let prompt = workspace.on_prompt_for_new_path.take().unwrap(); let rx = prompt(workspace, lister, suggested_name, window, cx); workspace.on_prompt_for_new_path = Some(prompt); rx })?; if let Ok(path) = rx.await { tx.send(path).ok(); } return anyhow::Ok(()); } }; tx.send(abs_path.map(|path| vec![path])).ok(); anyhow::Ok(()) }) .detach(); rx } pub fn titlebar_item(&self) -> Option { self.titlebar_item.clone() } /// Returns the worktree override set by the user (e.g., via the project dropdown). /// When set, git-related operations should use this worktree instead of deriving /// the active worktree from the focused file. pub fn active_worktree_override(&self) -> Option { self.active_worktree_override } pub fn set_active_worktree_override( &mut self, worktree_id: Option, cx: &mut Context, ) { self.active_worktree_override = worktree_id; cx.notify(); } pub fn clear_active_worktree_override(&mut self, cx: &mut Context) { self.active_worktree_override = None; cx.notify(); } /// Call the given callback with a workspace whose project is local or remote via WSL (allowing host access). /// /// If the given workspace has a local project, then it will be passed /// to the callback. Otherwise, a new empty window will be created. pub fn with_local_workspace( &mut self, window: &mut Window, cx: &mut Context, callback: F, ) -> Task> where T: 'static, F: 'static + FnOnce(&mut Workspace, &mut Window, &mut Context) -> T, { if self.project.read(cx).is_local() { Task::ready(Ok(callback(self, window, cx))) } else { let env = self.project.read(cx).cli_environment(cx); let task = Self::new_local( Vec::new(), self.app_state.clone(), None, env, None, OpenMode::Activate, cx, ); cx.spawn_in(window, async move |_vh, cx| { let OpenResult { window: multi_workspace_window, .. } = task.await?; multi_workspace_window.update(cx, |multi_workspace, window, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| callback(workspace, window, cx)) }) }) } } /// Call the given callback with a workspace whose project is local or remote via WSL (allowing host access). /// /// If the given workspace has a local project, then it will be passed /// to the callback. Otherwise, a new empty window will be created. pub fn with_local_or_wsl_workspace( &mut self, window: &mut Window, cx: &mut Context, callback: F, ) -> Task> where T: 'static, F: 'static + FnOnce(&mut Workspace, &mut Window, &mut Context) -> T, { let project = self.project.read(cx); if project.is_local() || project.is_via_wsl_with_host_interop(cx) { Task::ready(Ok(callback(self, window, cx))) } else { let env = self.project.read(cx).cli_environment(cx); let task = Self::new_local( Vec::new(), self.app_state.clone(), None, env, None, OpenMode::Activate, cx, ); cx.spawn_in(window, async move |_vh, cx| { let OpenResult { window: multi_workspace_window, .. } = task.await?; multi_workspace_window.update(cx, |multi_workspace, window, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| callback(workspace, window, cx)) }) }) } } pub fn worktrees<'a>(&self, cx: &'a App) -> impl 'a + Iterator> { self.project.read(cx).worktrees(cx) } pub fn visible_worktrees<'a>( &self, cx: &'a App, ) -> impl 'a + Iterator> { self.project.read(cx).visible_worktrees(cx) } pub fn worktree_scans_complete(&self, cx: &App) -> impl Future + 'static + use<> { let futures = self .worktrees(cx) .filter_map(|worktree| worktree.read(cx).as_local()) .map(|worktree| worktree.scan_complete()) .collect::>(); async move { for future in futures { future.await; } } } pub fn close_global(cx: &mut App) { cx.defer(|cx| { cx.windows().iter().find(|window| { window .update(cx, |_, window, _| { if window.is_window_active() { //This can only get called when the window's project connection has been lost //so we don't need to prompt the user for anything and instead just close the window window.remove_window(); true } else { false } }) .unwrap_or(false) }); }); } pub fn move_focused_panel_to_next_position( &mut self, _: &MoveFocusedPanelToNextPosition, window: &mut Window, cx: &mut Context, ) { let docks = self.all_docks(); let active_dock = docks .into_iter() .find(|dock| dock.focus_handle(cx).contains_focused(window, cx)); if let Some(dock) = active_dock { dock.update(cx, |dock, cx| { let active_panel = dock .active_panel() .filter(|panel| panel.panel_focus_handle(cx).contains_focused(window, cx)); if let Some(panel) = active_panel { panel.move_to_next_position(window, cx); } }) } } pub fn prepare_to_close( &mut self, close_intent: CloseIntent, window: &mut Window, cx: &mut Context, ) -> Task> { let active_call = self.active_global_call(); cx.spawn_in(window, async move |this, cx| { this.update(cx, |this, _| { if close_intent == CloseIntent::CloseWindow { this.removing = true; } })?; let workspace_count = cx.update(|_window, cx| { cx.windows() .iter() .filter(|window| window.downcast::().is_some()) .count() })?; #[cfg(target_os = "macos")] let save_last_workspace = false; // On Linux and Windows, closing the last window should restore the last workspace. #[cfg(not(target_os = "macos"))] let save_last_workspace = { let remaining_workspaces = cx.update(|_window, cx| { cx.windows() .iter() .filter_map(|window| window.downcast::()) .filter_map(|multi_workspace| { multi_workspace .update(cx, |multi_workspace, _, cx| { multi_workspace.workspace().read(cx).removing }) .ok() }) .filter(|removing| !removing) .count() })?; close_intent != CloseIntent::ReplaceWindow && remaining_workspaces == 0 }; if let Some(active_call) = active_call && workspace_count == 1 && cx .update(|_window, cx| active_call.0.is_in_room(cx)) .unwrap_or(false) { if close_intent == CloseIntent::CloseWindow { this.update(cx, |_, cx| cx.emit(Event::Activate))?; let answer = cx.update(|window, cx| { window.prompt( PromptLevel::Warning, "Do you want to leave the current call?", None, &["Close window and hang up", "Cancel"], cx, ) })?; if answer.await.log_err() == Some(1) { return anyhow::Ok(false); } else { if let Ok(task) = cx.update(|_window, cx| active_call.0.hang_up(cx)) { task.await.log_err(); } } } if close_intent == CloseIntent::ReplaceWindow { _ = cx.update(|_window, cx| { let multi_workspace = cx .windows() .iter() .filter_map(|window| window.downcast::()) .next() .unwrap(); let project = multi_workspace .read(cx)? .workspace() .read(cx) .project .clone(); if project.read(cx).is_shared() { active_call.0.unshare_project(project, cx)?; } Ok::<_, anyhow::Error>(()) }); } } let save_result = this .update_in(cx, |this, window, cx| { this.save_all_internal(SaveIntent::Close, window, cx) })? .await; // If we're not quitting, but closing, we remove the workspace from // the current session. if close_intent != CloseIntent::Quit && !save_last_workspace && save_result.as_ref().is_ok_and(|&res| res) { this.update_in(cx, |this, window, cx| this.remove_from_session(window, cx))? .await; } save_result }) } fn save_all(&mut self, action: &SaveAll, window: &mut Window, cx: &mut Context) { self.save_all_internal( action.save_intent.unwrap_or(SaveIntent::SaveAll), window, cx, ) .detach_and_log_err(cx); } fn send_keystrokes( &mut self, action: &SendKeystrokes, window: &mut Window, cx: &mut Context, ) { let keystrokes: Vec = action .0 .split(' ') .flat_map(|k| Keystroke::parse(k).log_err()) .map(|k| { cx.keyboard_mapper() .map_key_equivalent(k, false) .inner() .clone() }) .collect(); let _ = self.send_keystrokes_impl(keystrokes, window, cx); } pub fn send_keystrokes_impl( &mut self, keystrokes: Vec, window: &mut Window, cx: &mut Context, ) -> Shared> { let mut state = self.dispatching_keystrokes.borrow_mut(); if !state.dispatched.insert(keystrokes.clone()) { cx.propagate(); return state.task.clone().unwrap(); } state.queue.extend(keystrokes); let keystrokes = self.dispatching_keystrokes.clone(); if state.task.is_none() { state.task = Some( window .spawn(cx, async move |cx| { // limit to 100 keystrokes to avoid infinite recursion. for _ in 0..100 { let keystroke = { let mut state = keystrokes.borrow_mut(); let Some(keystroke) = state.queue.pop_front() else { state.dispatched.clear(); state.task.take(); return; }; keystroke }; cx.update(|window, cx| { let focused = window.focused(cx); window.dispatch_keystroke(keystroke.clone(), cx); if window.focused(cx) != focused { // dispatch_keystroke may cause the focus to change. // draw's side effect is to schedule the FocusChanged events in the current flush effect cycle // And we need that to happen before the next keystroke to keep vim mode happy... // (Note that the tests always do this implicitly, so you must manually test with something like: // "bindings": { "g z": ["workspace::SendKeystrokes", ": j u"]} // ) window.draw(cx).clear(); } }) .ok(); // Yield between synthetic keystrokes so deferred focus and // other effects can settle before dispatching the next key. yield_now().await; } *keystrokes.borrow_mut() = Default::default(); log::error!("over 100 keystrokes passed to send_keystrokes"); }) .shared(), ); } state.task.clone().unwrap() } /// Prompts the user to save or discard each dirty item, returning /// `true` if they confirmed (saved/discarded everything) or `false` /// if they cancelled. Used before removing worktree roots during /// thread archival. pub fn prompt_to_save_or_discard_dirty_items( &mut self, window: &mut Window, cx: &mut Context, ) -> Task> { self.save_all_internal(SaveIntent::Close, window, cx) } fn save_all_internal( &mut self, mut save_intent: SaveIntent, window: &mut Window, cx: &mut Context, ) -> Task> { if self.project.read(cx).is_disconnected(cx) { return Task::ready(Ok(true)); } let dirty_items = self .panes .iter() .flat_map(|pane| { pane.read(cx).items().filter_map(|item| { if item.is_dirty(cx) { item.tab_content_text(0, cx); Some((pane.downgrade(), item.boxed_clone())) } else { None } }) }) .collect::>(); let project = self.project.clone(); cx.spawn_in(window, async move |workspace, cx| { let dirty_items = if save_intent == SaveIntent::Close && !dirty_items.is_empty() { let (serialize_tasks, remaining_dirty_items) = workspace.update_in(cx, |workspace, window, cx| { let mut remaining_dirty_items = Vec::new(); let mut serialize_tasks = Vec::new(); for (pane, item) in dirty_items { if let Some(task) = item .to_serializable_item_handle(cx) .and_then(|handle| handle.serialize(workspace, true, window, cx)) { serialize_tasks.push(task); } else { remaining_dirty_items.push((pane, item)); } } (serialize_tasks, remaining_dirty_items) })?; futures::future::try_join_all(serialize_tasks).await?; if !remaining_dirty_items.is_empty() { workspace.update(cx, |_, cx| cx.emit(Event::Activate))?; } if remaining_dirty_items.len() > 1 { let answer = workspace.update_in(cx, |_, window, cx| { cx.emit(Event::Activate); let detail = Pane::file_names_for_prompt( &mut remaining_dirty_items.iter().map(|(_, handle)| handle), cx, ); window.prompt( PromptLevel::Warning, "Do you want to save all changes in the following files?", Some(&detail), &["Save all", "Discard all", "Cancel"], cx, ) })?; match answer.await.log_err() { Some(0) => save_intent = SaveIntent::SaveAll, Some(1) => save_intent = SaveIntent::Skip, Some(2) => return Ok(false), _ => {} } } remaining_dirty_items } else { dirty_items }; for (pane, item) in dirty_items { let (singleton, project_entry_ids) = cx.update(|_, cx| { ( item.buffer_kind(cx) == ItemBufferKind::Singleton, item.project_entry_ids(cx), ) })?; if (singleton || !project_entry_ids.is_empty()) && !Pane::save_item(project.clone(), &pane, &*item, save_intent, cx).await? { return Ok(false); } } Ok(true) }) } pub fn open_workspace_for_paths( &mut self, // replace_current_window: bool, mut open_mode: OpenMode, paths: Vec, window: &mut Window, cx: &mut Context, ) -> Task>> { let requesting_window = window.window_handle().downcast::(); let is_remote = self.project.read(cx).is_via_collab(); let has_worktree = self.project.read(cx).worktrees(cx).next().is_some(); let has_dirty_items = self.items(cx).any(|item| item.is_dirty(cx)); let workspace_is_empty = !is_remote && !has_worktree && !has_dirty_items; if workspace_is_empty { open_mode = OpenMode::Activate; } let app_state = self.app_state.clone(); cx.spawn(async move |_, cx| { let OpenResult { workspace, .. } = cx .update(|cx| { open_paths( &paths, app_state, OpenOptions { requesting_window, open_mode, ..Default::default() }, cx, ) }) .await?; Ok(workspace) }) } #[allow(clippy::type_complexity)] pub fn open_paths( &mut self, mut abs_paths: Vec, options: OpenOptions, pane: Option>, window: &mut Window, cx: &mut Context, ) -> Task>>>> { let fs = self.app_state.fs.clone(); let caller_ordered_abs_paths = abs_paths.clone(); // Sort the paths to ensure we add worktrees for parents before their children. abs_paths.sort_unstable(); cx.spawn_in(window, async move |this, cx| { let mut tasks = Vec::with_capacity(abs_paths.len()); for abs_path in &abs_paths { let visible = match options.visible.as_ref().unwrap_or(&OpenVisible::None) { OpenVisible::All => Some(true), OpenVisible::None => Some(false), OpenVisible::OnlyFiles => match fs.metadata(abs_path).await.log_err() { Some(Some(metadata)) => Some(!metadata.is_dir), Some(None) => Some(true), None => None, }, OpenVisible::OnlyDirectories => match fs.metadata(abs_path).await.log_err() { Some(Some(metadata)) => Some(metadata.is_dir), Some(None) => Some(false), None => None, }, }; let project_path = match visible { Some(visible) => match this .update(cx, |this, cx| { Workspace::project_path_for_path( this.project.clone(), abs_path, visible, cx, ) }) .log_err() { Some(project_path) => project_path.await.log_err(), None => None, }, None => None, }; let this = this.clone(); let abs_path: Arc = SanitizedPath::new(&abs_path).as_path().into(); let fs = fs.clone(); let pane = pane.clone(); let task = cx.spawn(async move |cx| { let (_worktree, project_path) = project_path?; if fs.is_dir(&abs_path).await { // Opening a directory should not race to update the active entry. // We'll select/reveal a deterministic final entry after all paths finish opening. None } else { Some( this.update_in(cx, |this, window, cx| { this.open_path( project_path, pane, options.focus.unwrap_or(true), window, cx, ) }) .ok()? .await, ) } }); tasks.push(task); } let results = futures::future::join_all(tasks).await; // Determine the winner using the fake/abstract FS metadata, not `Path::is_dir`. let mut winner: Option<(PathBuf, bool)> = None; for abs_path in caller_ordered_abs_paths.into_iter().rev() { if let Some(Some(metadata)) = fs.metadata(&abs_path).await.log_err() { if !metadata.is_dir { winner = Some((abs_path, false)); break; } if winner.is_none() { winner = Some((abs_path, true)); } } else if winner.is_none() { winner = Some((abs_path, false)); } } // Compute the winner entry id on the foreground thread and emit once, after all // paths finish opening. This avoids races between concurrently-opening paths // (directories in particular) and makes the resulting project panel selection // deterministic. if let Some((winner_abs_path, winner_is_dir)) = winner { 'emit_winner: { let winner_abs_path: Arc = SanitizedPath::new(&winner_abs_path).as_path().into(); let visible = match options.visible.as_ref().unwrap_or(&OpenVisible::None) { OpenVisible::All => true, OpenVisible::None => false, OpenVisible::OnlyFiles => !winner_is_dir, OpenVisible::OnlyDirectories => winner_is_dir, }; let Some(worktree_task) = this .update(cx, |workspace, cx| { workspace.project.update(cx, |project, cx| { project.find_or_create_worktree( winner_abs_path.as_ref(), visible, cx, ) }) }) .ok() else { break 'emit_winner; }; let Ok((worktree, _)) = worktree_task.await else { break 'emit_winner; }; let Ok(Some(entry_id)) = this.update(cx, |_, cx| { let worktree = worktree.read(cx); let worktree_abs_path = worktree.abs_path(); let entry = if winner_abs_path.as_ref() == worktree_abs_path.as_ref() { worktree.root_entry() } else { winner_abs_path .strip_prefix(worktree_abs_path.as_ref()) .ok() .and_then(|relative_path| { let relative_path = RelPath::new(relative_path, PathStyle::local()) .log_err()?; worktree.entry_for_path(&relative_path) }) }?; Some(entry.id) }) else { break 'emit_winner; }; this.update(cx, |workspace, cx| { workspace.project.update(cx, |_, cx| { cx.emit(project::Event::ActiveEntryChanged(Some(entry_id))); }); }) .ok(); } } results }) } pub fn open_resolved_path( &mut self, path: ResolvedPath, window: &mut Window, cx: &mut Context, ) -> Task>> { match path { ResolvedPath::ProjectPath { project_path, .. } => { self.open_path(project_path, None, true, window, cx) } ResolvedPath::AbsPath { path, .. } => self.open_abs_path( PathBuf::from(path), OpenOptions { visible: Some(OpenVisible::None), ..Default::default() }, window, cx, ), } } pub fn absolute_path_of_worktree( &self, worktree_id: WorktreeId, cx: &mut Context, ) -> Option { self.project .read(cx) .worktree_for_id(worktree_id, cx) // TODO: use `abs_path` or `root_dir` .map(|wt| wt.read(cx).abs_path().as_ref().to_path_buf()) } pub fn add_folder_to_project( &mut self, _: &AddFolderToProject, window: &mut Window, cx: &mut Context, ) { let project = self.project.read(cx); if project.is_via_collab() { self.show_error( &anyhow!("You cannot add folders to someone else's project"), cx, ); return; } let paths = self.prompt_for_open_path( PathPromptOptions { files: false, directories: true, multiple: true, prompt: None, }, DirectoryLister::Project(self.project.clone()), window, cx, ); cx.spawn_in(window, async move |this, cx| { if let Some(paths) = paths.await.log_err().flatten() { let results = this .update_in(cx, |this, window, cx| { this.open_paths( paths, OpenOptions { visible: Some(OpenVisible::All), ..Default::default() }, None, window, cx, ) })? .await; for result in results.into_iter().flatten() { result.log_err(); } } anyhow::Ok(()) }) .detach_and_log_err(cx); } pub fn project_path_for_path( project: Entity, abs_path: &Path, visible: bool, cx: &mut App, ) -> Task, ProjectPath)>> { let entry = project.update(cx, |project, cx| { project.find_or_create_worktree(abs_path, visible, cx) }); cx.spawn(async move |cx| { let (worktree, path) = entry.await?; let worktree_id = worktree.read_with(cx, |t, _| t.id()); Ok((worktree, ProjectPath { worktree_id, path })) }) } pub fn items<'a>(&'a self, cx: &'a App) -> impl 'a + Iterator> { self.panes.iter().flat_map(|pane| pane.read(cx).items()) } pub fn item_of_type(&self, cx: &App) -> Option> { self.items_of_type(cx).max_by_key(|item| item.item_id()) } pub fn items_of_type<'a, T: Item>( &'a self, cx: &'a App, ) -> impl 'a + Iterator> { self.panes .iter() .flat_map(|pane| pane.read(cx).items_of_type()) } pub fn active_item(&self, cx: &App) -> Option> { self.active_pane().read(cx).active_item() } pub fn active_item_as(&self, cx: &App) -> Option> { let item = self.active_item(cx)?; item.to_any_view().downcast::().ok() } fn active_project_path(&self, cx: &App) -> Option { self.active_item(cx).and_then(|item| item.project_path(cx)) } pub fn most_recent_active_path(&self, cx: &App) -> Option { self.recent_navigation_history_iter(cx) .filter_map(|(path, abs_path)| { let worktree = self .project .read(cx) .worktree_for_id(path.worktree_id, cx)?; if worktree.read(cx).is_visible() { abs_path } else { None } }) .next() } pub fn save_active_item( &mut self, save_intent: SaveIntent, window: &mut Window, cx: &mut App, ) -> Task> { let project = self.project.clone(); let pane = self.active_pane(); let item = pane.read(cx).active_item(); let pane = pane.downgrade(); window.spawn(cx, async move |cx| { if let Some(item) = item { Pane::save_item(project, &pane, item.as_ref(), save_intent, cx) .await .map(|_| ()) } else { Ok(()) } }) } pub fn close_inactive_items_and_panes( &mut self, action: &CloseInactiveTabsAndPanes, window: &mut Window, cx: &mut Context, ) { if let Some(task) = self.close_all_internal( true, action.save_intent.unwrap_or(SaveIntent::Close), window, cx, ) { task.detach_and_log_err(cx) } } pub fn close_all_items_and_panes( &mut self, action: &CloseAllItemsAndPanes, window: &mut Window, cx: &mut Context, ) { if let Some(task) = self.close_all_internal( false, action.save_intent.unwrap_or(SaveIntent::Close), window, cx, ) { task.detach_and_log_err(cx) } } /// Closes the active item across all panes. pub fn close_item_in_all_panes( &mut self, action: &CloseItemInAllPanes, window: &mut Window, cx: &mut Context, ) { let Some(active_item) = self.active_pane().read(cx).active_item() else { return; }; let save_intent = action.save_intent.unwrap_or(SaveIntent::Close); let close_pinned = action.close_pinned; if let Some(project_path) = active_item.project_path(cx) { self.close_items_with_project_path( &project_path, save_intent, close_pinned, window, cx, ); } else if close_pinned || !self.active_pane().read(cx).is_active_item_pinned() { let item_id = active_item.item_id(); self.active_pane().update(cx, |pane, cx| { pane.close_item_by_id(item_id, save_intent, window, cx) .detach_and_log_err(cx); }); } } /// Closes all items with the given project path across all panes. pub fn close_items_with_project_path( &mut self, project_path: &ProjectPath, save_intent: SaveIntent, close_pinned: bool, window: &mut Window, cx: &mut Context, ) { let panes = self.panes().to_vec(); for pane in panes { pane.update(cx, |pane, cx| { pane.close_items_for_project_path( project_path, save_intent, close_pinned, window, cx, ) .detach_and_log_err(cx); }); } } fn close_all_internal( &mut self, retain_active_pane: bool, save_intent: SaveIntent, window: &mut Window, cx: &mut Context, ) -> Option>> { let current_pane = self.active_pane(); let mut tasks = Vec::new(); if retain_active_pane { let current_pane_close = current_pane.update(cx, |pane, cx| { pane.close_other_items( &CloseOtherItems { save_intent: None, close_pinned: false, }, None, window, cx, ) }); tasks.push(current_pane_close); } for pane in self.panes() { if retain_active_pane && pane.entity_id() == current_pane.entity_id() { continue; } let close_pane_items = pane.update(cx, |pane: &mut Pane, cx| { pane.close_all_items( &CloseAllItems { save_intent: Some(save_intent), close_pinned: false, }, window, cx, ) }); tasks.push(close_pane_items) } if tasks.is_empty() { None } else { Some(cx.spawn_in(window, async move |_, _| { for task in tasks { task.await? } Ok(()) })) } } pub fn is_dock_at_position_open(&self, position: DockPosition, cx: &mut Context) -> bool { self.dock_at_position(position).read(cx).is_open() } pub fn toggle_dock( &mut self, dock_side: DockPosition, window: &mut Window, cx: &mut Context, ) { let mut focus_center = false; let mut reveal_dock = false; let other_is_zoomed = self.zoomed.is_some() && self.zoomed_position != Some(dock_side); let was_visible = self.is_dock_at_position_open(dock_side, cx) && !other_is_zoomed; if let Some(panel) = self.dock_at_position(dock_side).read(cx).active_panel() { telemetry::event!( "Panel Button Clicked", name = panel.persistent_name(), toggle_state = !was_visible ); } if was_visible { self.save_open_dock_positions(cx); } let dock = self.dock_at_position(dock_side); dock.update(cx, |dock, cx| { dock.set_open(!was_visible, window, cx); if dock.active_panel().is_none() { let Some(panel_ix) = dock .first_enabled_panel_idx(cx) .log_with_level(log::Level::Info) else { return; }; dock.activate_panel(panel_ix, window, cx); } if let Some(active_panel) = dock.active_panel() { if was_visible { if active_panel .panel_focus_handle(cx) .contains_focused(window, cx) { focus_center = true; } } else { let focus_handle = &active_panel.panel_focus_handle(cx); window.focus(focus_handle, cx); reveal_dock = true; } } }); if reveal_dock { self.dismiss_zoomed_items_to_reveal(Some(dock_side), window, cx); } if focus_center { self.active_pane .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx), cx)) } cx.notify(); self.serialize_workspace(window, cx); } fn active_dock(&self, window: &Window, cx: &Context) -> Option<&Entity> { self.all_docks().into_iter().find(|&dock| { dock.read(cx).is_open() && dock.focus_handle(cx).contains_focused(window, cx) }) } fn close_active_dock(&mut self, window: &mut Window, cx: &mut Context) -> bool { if let Some(dock) = self.active_dock(window, cx).cloned() { self.save_open_dock_positions(cx); dock.update(cx, |dock, cx| { dock.set_open(false, window, cx); }); return true; } false } pub fn close_all_docks(&mut self, window: &mut Window, cx: &mut Context) { self.save_open_dock_positions(cx); for dock in self.all_docks() { dock.update(cx, |dock, cx| { dock.set_open(false, window, cx); }); } cx.focus_self(window); cx.notify(); self.serialize_workspace(window, cx); } fn get_open_dock_positions(&self, cx: &Context) -> Vec { self.all_docks() .into_iter() .filter_map(|dock| { let dock_ref = dock.read(cx); if dock_ref.is_open() { Some(dock_ref.position()) } else { None } }) .collect() } /// Saves the positions of currently open docks. /// /// Updates `last_open_dock_positions` with positions of all currently open /// docks, to later be restored by the 'Toggle All Docks' action. fn save_open_dock_positions(&mut self, cx: &mut Context) { let open_dock_positions = self.get_open_dock_positions(cx); if !open_dock_positions.is_empty() { self.last_open_dock_positions = open_dock_positions; } } /// Toggles all docks between open and closed states. /// /// If any docks are open, closes all and remembers their positions. If all /// docks are closed, restores the last remembered dock configuration. fn toggle_all_docks( &mut self, _: &ToggleAllDocks, window: &mut Window, cx: &mut Context, ) { let open_dock_positions = self.get_open_dock_positions(cx); if !open_dock_positions.is_empty() { self.close_all_docks(window, cx); } else if !self.last_open_dock_positions.is_empty() { self.restore_last_open_docks(window, cx); } } /// Reopens docks from the most recently remembered configuration. /// /// Opens all docks whose positions are stored in `last_open_dock_positions` /// and clears the stored positions. fn restore_last_open_docks(&mut self, window: &mut Window, cx: &mut Context) { let positions_to_open = std::mem::take(&mut self.last_open_dock_positions); for position in positions_to_open { let dock = self.dock_at_position(position); dock.update(cx, |dock, cx| dock.set_open(true, window, cx)); } cx.focus_self(window); cx.notify(); self.serialize_workspace(window, cx); } /// Transfer focus to the panel of the given type. pub fn focus_panel( &mut self, window: &mut Window, cx: &mut Context, ) -> Option> { let panel = self.focus_or_unfocus_panel::(window, cx, &mut |_, _, _| true)?; panel.to_any().downcast().ok() } /// Focus the panel of the given type if it isn't already focused. If it is /// already focused, then transfer focus back to the workspace center. /// When the `close_panel_on_toggle` setting is enabled, also closes the /// panel when transferring focus back to the center. pub fn toggle_panel_focus( &mut self, window: &mut Window, cx: &mut Context, ) -> bool { let mut did_focus_panel = false; self.focus_or_unfocus_panel::(window, cx, &mut |panel, window, cx| { did_focus_panel = !panel.panel_focus_handle(cx).contains_focused(window, cx); did_focus_panel }); if !did_focus_panel && WorkspaceSettings::get_global(cx).close_panel_on_toggle { self.close_panel::(window, cx); } telemetry::event!( "Panel Button Clicked", name = T::persistent_name(), toggle_state = did_focus_panel ); did_focus_panel } pub fn focus_center_pane(&mut self, window: &mut Window, cx: &mut Context) { if let Some(item) = self.active_item(cx) { item.item_focus_handle(cx).focus(window, cx); } else { log::error!("Could not find a focus target when switching focus to the center panes",); } } pub fn activate_panel_for_proto_id( &mut self, panel_id: PanelId, window: &mut Window, cx: &mut Context, ) -> Option> { let mut panel = None; for dock in self.all_docks() { if let Some(panel_index) = dock.read(cx).panel_index_for_proto_id(panel_id) { panel = dock.update(cx, |dock, cx| { dock.activate_panel(panel_index, window, cx); dock.set_open(true, window, cx); dock.active_panel().cloned() }); break; } } if panel.is_some() { cx.notify(); self.serialize_workspace(window, cx); } panel } /// Focus or unfocus the given panel type, depending on the given callback. fn focus_or_unfocus_panel( &mut self, window: &mut Window, cx: &mut Context, should_focus: &mut dyn FnMut(&dyn PanelHandle, &mut Window, &mut Context) -> bool, ) -> Option> { let mut result_panel = None; let mut serialize = false; for dock in self.all_docks() { if let Some(panel_index) = dock.read(cx).panel_index_for_type::() { let mut focus_center = false; let panel = dock.update(cx, |dock, cx| { dock.activate_panel(panel_index, window, cx); let panel = dock.active_panel().cloned(); if let Some(panel) = panel.as_ref() { if should_focus(&**panel, window, cx) { dock.set_open(true, window, cx); panel.panel_focus_handle(cx).focus(window, cx); } else { focus_center = true; } } panel }); if focus_center { self.active_pane .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx), cx)) } result_panel = panel; serialize = true; break; } } if serialize { self.serialize_workspace(window, cx); } cx.notify(); result_panel } /// Open the panel of the given type pub fn open_panel(&mut self, window: &mut Window, cx: &mut Context) { for dock in self.all_docks() { if let Some(panel_index) = dock.read(cx).panel_index_for_type::() { dock.update(cx, |dock, cx| { dock.activate_panel(panel_index, window, cx); dock.set_open(true, window, cx); }); } } } /// Open the panel of the given type, dismissing any zoomed items that /// would obscure it (e.g. a zoomed terminal). pub fn reveal_panel(&mut self, window: &mut Window, cx: &mut Context) { let dock_position = self.all_docks().iter().find_map(|dock| { let dock = dock.read(cx); dock.panel_index_for_type::().map(|_| dock.position()) }); self.dismiss_zoomed_items_to_reveal(dock_position, window, cx); self.open_panel::(window, cx); } pub fn close_panel(&self, window: &mut Window, cx: &mut Context) { for dock in self.all_docks().iter() { dock.update(cx, |dock, cx| { if dock.panel::().is_some() { dock.set_open(false, window, cx) } }) } } pub fn panel(&self, cx: &App) -> Option> { self.all_docks() .iter() .find_map(|dock| dock.read(cx).panel::()) } fn dismiss_zoomed_items_to_reveal( &mut self, dock_to_reveal: Option, window: &mut Window, cx: &mut Context, ) { // If a center pane is zoomed, unzoom it. for pane in &self.panes { if pane != &self.active_pane || dock_to_reveal.is_some() { pane.update(cx, |pane, cx| pane.set_zoomed(false, cx)); } } // If another dock is zoomed, hide it. let mut focus_center = false; for dock in self.all_docks() { dock.update(cx, |dock, cx| { if Some(dock.position()) != dock_to_reveal && let Some(panel) = dock.active_panel() && panel.is_zoomed(window, cx) { focus_center |= panel.panel_focus_handle(cx).contains_focused(window, cx); dock.set_open(false, window, cx); } }); } if focus_center { self.active_pane .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx), cx)) } if self.zoomed_position != dock_to_reveal { self.zoomed = None; self.zoomed_position = None; cx.emit(Event::ZoomChanged); } cx.notify(); } fn add_pane(&mut self, window: &mut Window, cx: &mut Context) -> Entity { let pane = cx.new(|cx| { let mut pane = Pane::new( self.weak_handle(), self.project.clone(), self.pane_history_timestamp.clone(), None, NewFile.boxed_clone(), true, window, cx, ); pane.set_can_split(Some(Arc::new(|_, _, _, _| true))); pane }); cx.subscribe_in(&pane, window, Self::handle_pane_event) .detach(); self.panes.push(pane.clone()); window.focus(&pane.focus_handle(cx), cx); cx.emit(Event::PaneAdded(pane.clone())); pane } pub fn add_item_to_center( &mut self, item: Box, window: &mut Window, cx: &mut Context, ) -> bool { if let Some(center_pane) = self.last_active_center_pane.clone() { if let Some(center_pane) = center_pane.upgrade() { center_pane.update(cx, |pane, cx| { pane.add_item(item, true, true, None, window, cx) }); true } else { false } } else { false } } pub fn add_item_to_active_pane( &mut self, item: Box, destination_index: Option, focus_item: bool, window: &mut Window, cx: &mut App, ) { self.add_item( self.active_pane.clone(), item, destination_index, false, focus_item, window, cx, ) } pub fn add_item( &mut self, pane: Entity, item: Box, destination_index: Option, activate_pane: bool, focus_item: bool, window: &mut Window, cx: &mut App, ) { pane.update(cx, |pane, cx| { pane.add_item( item, activate_pane, focus_item, destination_index, window, cx, ) }); } pub fn split_item( &mut self, split_direction: SplitDirection, item: Box, window: &mut Window, cx: &mut Context, ) { let new_pane = self.split_pane(self.active_pane.clone(), split_direction, window, cx); self.add_item(new_pane, item, None, true, true, window, cx); } pub fn open_abs_path( &mut self, abs_path: PathBuf, options: OpenOptions, window: &mut Window, cx: &mut Context, ) -> Task>> { cx.spawn_in(window, async move |workspace, cx| { let open_paths_task_result = workspace .update_in(cx, |workspace, window, cx| { workspace.open_paths(vec![abs_path.clone()], options, None, window, cx) }) .with_context(|| format!("open abs path {abs_path:?} task spawn"))? .await; anyhow::ensure!( open_paths_task_result.len() == 1, "open abs path {abs_path:?} task returned incorrect number of results" ); match open_paths_task_result .into_iter() .next() .expect("ensured single task result") { Some(open_result) => { open_result.with_context(|| format!("open abs path {abs_path:?} task join")) } None => anyhow::bail!("open abs path {abs_path:?} task returned None"), } }) } pub fn split_abs_path( &mut self, abs_path: PathBuf, visible: bool, window: &mut Window, cx: &mut Context, ) -> Task>> { let project_path_task = Workspace::project_path_for_path(self.project.clone(), &abs_path, visible, cx); cx.spawn_in(window, async move |this, cx| { let (_, path) = project_path_task.await?; this.update_in(cx, |this, window, cx| this.split_path(path, window, cx))? .await }) } pub fn open_path( &mut self, path: impl Into, pane: Option>, focus_item: bool, window: &mut Window, cx: &mut App, ) -> Task>> { self.open_path_preview(path, pane, focus_item, false, true, window, cx) } pub fn open_path_preview( &mut self, path: impl Into, pane: Option>, focus_item: bool, allow_preview: bool, activate: bool, window: &mut Window, cx: &mut App, ) -> Task>> { let pane = pane.unwrap_or_else(|| { self.last_active_center_pane.clone().unwrap_or_else(|| { self.panes .first() .expect("There must be an active pane") .downgrade() }) }); let project_path = path.into(); let task = self.load_path(project_path.clone(), window, cx); window.spawn(cx, async move |cx| { let (project_entry_id, build_item) = task.await?; pane.update_in(cx, |pane, window, cx| { pane.open_item( project_entry_id, project_path, focus_item, allow_preview, activate, None, window, cx, build_item, ) }) }) } pub fn split_path( &mut self, path: impl Into, window: &mut Window, cx: &mut Context, ) -> Task>> { self.split_path_preview(path, false, None, window, cx) } pub fn split_path_preview( &mut self, path: impl Into, allow_preview: bool, split_direction: Option, window: &mut Window, cx: &mut Context, ) -> Task>> { let pane = self.last_active_center_pane.clone().unwrap_or_else(|| { self.panes .first() .expect("There must be an active pane") .downgrade() }); if let Member::Pane(center_pane) = &self.center.root && center_pane.read(cx).items_len() == 0 { return self.open_path(path, Some(pane), true, window, cx); } let project_path = path.into(); let task = self.load_path(project_path.clone(), window, cx); cx.spawn_in(window, async move |this, cx| { let (project_entry_id, build_item) = task.await?; this.update_in(cx, move |this, window, cx| -> Option<_> { let pane = pane.upgrade()?; let new_pane = this.split_pane( pane, split_direction.unwrap_or(SplitDirection::Right), window, cx, ); new_pane.update(cx, |new_pane, cx| { Some(new_pane.open_item( project_entry_id, project_path, true, allow_preview, true, None, window, cx, build_item, )) }) }) .map(|option| option.context("pane was dropped"))? }) } fn load_path( &mut self, path: ProjectPath, window: &mut Window, cx: &mut App, ) -> Task, WorkspaceItemBuilder)>> { let registry = cx.default_global::().clone(); registry.open_path(self.project(), &path, window, cx) } pub fn find_project_item( &self, pane: &Entity, project_item: &Entity, cx: &App, ) -> Option> where T: ProjectItem, { use project::ProjectItem as _; let project_item = project_item.read(cx); let entry_id = project_item.entry_id(cx); let project_path = project_item.project_path(cx); let mut item = None; if let Some(entry_id) = entry_id { item = pane.read(cx).item_for_entry(entry_id, cx); } if item.is_none() && let Some(project_path) = project_path { item = pane.read(cx).item_for_path(project_path, cx); } item.and_then(|item| item.downcast::()) } pub fn is_project_item_open( &self, pane: &Entity, project_item: &Entity, cx: &App, ) -> bool where T: ProjectItem, { self.find_project_item::(pane, project_item, cx) .is_some() } pub fn open_project_item( &mut self, pane: Entity, project_item: Entity, activate_pane: bool, focus_item: bool, keep_old_preview: bool, allow_new_preview: bool, window: &mut Window, cx: &mut Context, ) -> Entity where T: ProjectItem, { let old_item_id = pane.read(cx).active_item().map(|item| item.item_id()); if let Some(item) = self.find_project_item(&pane, &project_item, cx) { if !keep_old_preview && let Some(old_id) = old_item_id && old_id != item.item_id() { // switching to a different item, so unpreview old active item pane.update(cx, |pane, _| { pane.unpreview_item_if_preview(old_id); }); } self.activate_item(&item, activate_pane, focus_item, window, cx); if !allow_new_preview { pane.update(cx, |pane, _| { pane.unpreview_item_if_preview(item.item_id()); }); } return item; } let item = pane.update(cx, |pane, cx| { cx.new(|cx| { T::for_project_item(self.project().clone(), Some(pane), project_item, window, cx) }) }); let mut destination_index = None; pane.update(cx, |pane, cx| { if !keep_old_preview && let Some(old_id) = old_item_id { pane.unpreview_item_if_preview(old_id); } if allow_new_preview { destination_index = pane.replace_preview_item_id(item.item_id(), window, cx); } }); self.add_item( pane, Box::new(item.clone()), destination_index, activate_pane, focus_item, window, cx, ); item } pub fn open_shared_screen( &mut self, peer_id: PeerId, window: &mut Window, cx: &mut Context, ) { if let Some(shared_screen) = self.shared_screen_for_peer(peer_id, &self.active_pane, window, cx) { self.active_pane.update(cx, |pane, cx| { pane.add_item(Box::new(shared_screen), false, true, None, window, cx) }); } } pub fn activate_item( &mut self, item: &dyn ItemHandle, activate_pane: bool, focus_item: bool, window: &mut Window, cx: &mut App, ) -> bool { let result = self.panes.iter().find_map(|pane| { pane.read(cx) .index_for_item(item) .map(|ix| (pane.clone(), ix)) }); if let Some((pane, ix)) = result { pane.update(cx, |pane, cx| { pane.activate_item(ix, activate_pane, focus_item, window, cx) }); true } else { false } } fn activate_pane_at_index( &mut self, action: &ActivatePane, window: &mut Window, cx: &mut Context, ) { let panes = self.center.panes(); if let Some(pane) = panes.get(action.0).map(|p| (*p).clone()) { window.focus(&pane.focus_handle(cx), cx); } else { self.split_and_clone(self.active_pane.clone(), SplitDirection::Right, window, cx) .detach(); } } fn move_item_to_pane_at_index( &mut self, action: &MoveItemToPane, window: &mut Window, cx: &mut Context, ) { let panes = self.center.panes(); let destination = match panes.get(action.destination) { Some(&destination) => destination.clone(), None => { if !action.clone && self.active_pane.read(cx).items_len() < 2 { return; } let direction = SplitDirection::Right; let split_off_pane = self .find_pane_in_direction(direction, cx) .unwrap_or_else(|| self.active_pane.clone()); let new_pane = self.add_pane(window, cx); self.center.split(&split_off_pane, &new_pane, direction, cx); new_pane } }; if action.clone { if self .active_pane .read(cx) .active_item() .is_some_and(|item| item.can_split(cx)) { clone_active_item( self.database_id(), &self.active_pane, &destination, action.focus, window, cx, ); return; } } move_active_item( &self.active_pane, &destination, action.focus, true, window, cx, ) } pub fn activate_next_pane(&mut self, window: &mut Window, cx: &mut App) { let panes = self.center.panes(); if let Some(ix) = panes.iter().position(|pane| **pane == self.active_pane) { let next_ix = (ix + 1) % panes.len(); let next_pane = panes[next_ix].clone(); window.focus(&next_pane.focus_handle(cx), cx); } } pub fn activate_previous_pane(&mut self, window: &mut Window, cx: &mut App) { let panes = self.center.panes(); if let Some(ix) = panes.iter().position(|pane| **pane == self.active_pane) { let prev_ix = cmp::min(ix.wrapping_sub(1), panes.len() - 1); let prev_pane = panes[prev_ix].clone(); window.focus(&prev_pane.focus_handle(cx), cx); } } pub fn activate_last_pane(&mut self, window: &mut Window, cx: &mut App) { let last_pane = self.center.last_pane(); window.focus(&last_pane.focus_handle(cx), cx); } pub fn activate_pane_in_direction( &mut self, direction: SplitDirection, window: &mut Window, cx: &mut App, ) { use ActivateInDirectionTarget as Target; enum Origin { Sidebar, LeftDock, RightDock, BottomDock, Center, } let origin: Origin = if self .sidebar_focus_handle .as_ref() .is_some_and(|h| h.contains_focused(window, cx)) { Origin::Sidebar } else { [ (&self.left_dock, Origin::LeftDock), (&self.right_dock, Origin::RightDock), (&self.bottom_dock, Origin::BottomDock), ] .into_iter() .find_map(|(dock, origin)| { if dock.focus_handle(cx).contains_focused(window, cx) && dock.read(cx).is_open() { Some(origin) } else { None } }) .unwrap_or(Origin::Center) }; let get_last_active_pane = || { let pane = self .last_active_center_pane .clone() .unwrap_or_else(|| { self.panes .first() .expect("There must be an active pane") .downgrade() }) .upgrade()?; (pane.read(cx).items_len() != 0).then_some(pane) }; let try_dock = |dock: &Entity| dock.read(cx).is_open().then(|| Target::Dock(dock.clone())); let sidebar_target = self .sidebar_focus_handle .as_ref() .map(|h| Target::Sidebar(h.clone())); let sidebar_on_right = self .multi_workspace .as_ref() .and_then(|mw| mw.upgrade()) .map_or(false, |mw| { mw.read(cx).sidebar_side(cx) == SidebarSide::Right }); let away_from_sidebar = if sidebar_on_right { SplitDirection::Left } else { SplitDirection::Right }; let (near_dock, far_dock) = if sidebar_on_right { (&self.right_dock, &self.left_dock) } else { (&self.left_dock, &self.right_dock) }; let target = match (origin, direction) { (Origin::Sidebar, dir) if dir == away_from_sidebar => try_dock(near_dock) .or_else(|| get_last_active_pane().map(Target::Pane)) .or_else(|| try_dock(&self.bottom_dock)) .or_else(|| try_dock(far_dock)), (Origin::Sidebar, _) => None, // We're in the center, so we first try to go to a different pane, // otherwise try to go to a dock. (Origin::Center, direction) => { if let Some(pane) = self.find_pane_in_direction(direction, cx) { Some(Target::Pane(pane)) } else { match direction { SplitDirection::Up => None, SplitDirection::Down => try_dock(&self.bottom_dock), SplitDirection::Left => { let dock_target = try_dock(&self.left_dock); if sidebar_on_right { dock_target } else { dock_target.or(sidebar_target) } } SplitDirection::Right => { let dock_target = try_dock(&self.right_dock); if sidebar_on_right { dock_target.or(sidebar_target) } else { dock_target } } } } } (Origin::LeftDock, SplitDirection::Right) => { if let Some(last_active_pane) = get_last_active_pane() { Some(Target::Pane(last_active_pane)) } else { try_dock(&self.bottom_dock).or_else(|| try_dock(&self.right_dock)) } } (Origin::LeftDock, SplitDirection::Left) => { if sidebar_on_right { None } else { sidebar_target } } (Origin::LeftDock, SplitDirection::Down) | (Origin::RightDock, SplitDirection::Down) => try_dock(&self.bottom_dock), (Origin::BottomDock, SplitDirection::Up) => get_last_active_pane().map(Target::Pane), (Origin::BottomDock, SplitDirection::Left) => { let dock_target = try_dock(&self.left_dock); if sidebar_on_right { dock_target } else { dock_target.or(sidebar_target) } } (Origin::BottomDock, SplitDirection::Right) => { let dock_target = try_dock(&self.right_dock); if sidebar_on_right { dock_target.or(sidebar_target) } else { dock_target } } (Origin::RightDock, SplitDirection::Left) => { if let Some(last_active_pane) = get_last_active_pane() { Some(Target::Pane(last_active_pane)) } else { try_dock(&self.bottom_dock).or_else(|| try_dock(&self.left_dock)) } } (Origin::RightDock, SplitDirection::Right) => { if sidebar_on_right { sidebar_target } else { None } } _ => None, }; match target { Some(ActivateInDirectionTarget::Pane(pane)) => { let pane = pane.read(cx); if let Some(item) = pane.active_item() { item.item_focus_handle(cx).focus(window, cx); } else { log::error!( "Could not find a focus target when in switching focus in {direction} direction for a pane", ); } } Some(ActivateInDirectionTarget::Dock(dock)) => { // Defer this to avoid a panic when the dock's active panel is already on the stack. window.defer(cx, move |window, cx| { let dock = dock.read(cx); if let Some(panel) = dock.active_panel() { panel.panel_focus_handle(cx).focus(window, cx); } else { log::error!("Could not find a focus target when in switching focus in {direction} direction for a {:?} dock", dock.position()); } }) } Some(ActivateInDirectionTarget::Sidebar(focus_handle)) => { focus_handle.focus(window, cx); } None => {} } } pub fn move_item_to_pane_in_direction( &mut self, action: &MoveItemToPaneInDirection, window: &mut Window, cx: &mut Context, ) { let destination = match self.find_pane_in_direction(action.direction, cx) { Some(destination) => destination, None => { if !action.clone && self.active_pane.read(cx).items_len() < 2 { return; } let new_pane = self.add_pane(window, cx); self.center .split(&self.active_pane, &new_pane, action.direction, cx); new_pane } }; if action.clone { if self .active_pane .read(cx) .active_item() .is_some_and(|item| item.can_split(cx)) { clone_active_item( self.database_id(), &self.active_pane, &destination, action.focus, window, cx, ); return; } } move_active_item( &self.active_pane, &destination, action.focus, true, window, cx, ); } pub fn bounding_box_for_pane(&self, pane: &Entity) -> Option> { self.center.bounding_box_for_pane(pane) } pub fn find_pane_in_direction( &mut self, direction: SplitDirection, cx: &App, ) -> Option> { self.center .find_pane_in_direction(&self.active_pane, direction, cx) .cloned() } pub fn swap_pane_in_direction(&mut self, direction: SplitDirection, cx: &mut Context) { if let Some(to) = self.find_pane_in_direction(direction, cx) { self.center.swap(&self.active_pane, &to, cx); cx.notify(); } } pub fn move_pane_to_border(&mut self, direction: SplitDirection, cx: &mut Context) { if self .center .move_to_border(&self.active_pane, direction, cx) .unwrap() { cx.notify(); } } pub fn resize_pane( &mut self, axis: gpui::Axis, amount: Pixels, window: &mut Window, cx: &mut Context, ) { let docks = self.all_docks(); let active_dock = docks .into_iter() .find(|dock| dock.focus_handle(cx).contains_focused(window, cx)); if let Some(dock_entity) = active_dock { let dock = dock_entity.read(cx); let Some(panel_size) = self.dock_size(&dock, window, cx) else { return; }; match dock.position() { DockPosition::Left => self.resize_left_dock(panel_size + amount, window, cx), DockPosition::Bottom => self.resize_bottom_dock(panel_size + amount, window, cx), DockPosition::Right => self.resize_right_dock(panel_size + amount, window, cx), } } else { self.center .resize(&self.active_pane, axis, amount, &self.bounds, cx); } cx.notify(); } pub fn reset_pane_sizes(&mut self, cx: &mut Context) { self.center.reset_pane_sizes(cx); cx.notify(); } fn handle_pane_focused( &mut self, pane: Entity, window: &mut Window, cx: &mut Context, ) { // This is explicitly hoisted out of the following check for pane identity as // terminal panel panes are not registered as a center panes. self.status_bar.update(cx, |status_bar, cx| { status_bar.set_active_pane(&pane, window, cx); }); if self.active_pane != pane { self.set_active_pane(&pane, window, cx); } if self.last_active_center_pane.is_none() { self.last_active_center_pane = Some(pane.downgrade()); } // If this pane is in a dock, preserve that dock when dismissing zoomed items. // This prevents the dock from closing when focus events fire during window activation. // We also preserve any dock whose active panel itself has focus — this covers // panels like AgentPanel that don't implement `pane()` but can still be zoomed. let dock_to_preserve = self.all_docks().iter().find_map(|dock| { let dock_read = dock.read(cx); if let Some(panel) = dock_read.active_panel() { if panel.pane(cx).is_some_and(|dock_pane| dock_pane == pane) || panel.panel_focus_handle(cx).contains_focused(window, cx) { return Some(dock_read.position()); } } None }); self.dismiss_zoomed_items_to_reveal(dock_to_preserve, window, cx); if pane.read(cx).is_zoomed() { self.zoomed = Some(pane.downgrade().into()); } else { self.zoomed = None; } self.zoomed_position = None; cx.emit(Event::ZoomChanged); self.update_active_view_for_followers(window, cx); pane.update(cx, |pane, _| { pane.track_alternate_file_items(); }); cx.notify(); } fn set_active_pane( &mut self, pane: &Entity, window: &mut Window, cx: &mut Context, ) { self.active_pane = pane.clone(); self.active_item_path_changed(true, window, cx); self.last_active_center_pane = Some(pane.downgrade()); } fn handle_panel_focused(&mut self, window: &mut Window, cx: &mut Context) { self.update_active_view_for_followers(window, cx); } fn handle_pane_event( &mut self, pane: &Entity, event: &pane::Event, window: &mut Window, cx: &mut Context, ) { let mut serialize_workspace = true; match event { pane::Event::AddItem { item } => { item.added_to_pane(self, pane.clone(), window, cx); cx.emit(Event::ItemAdded { item: item.boxed_clone(), }); } pane::Event::Split { direction, mode } => { match mode { SplitMode::ClonePane => { self.split_and_clone(pane.clone(), *direction, window, cx) .detach(); } SplitMode::EmptyPane => { self.split_pane(pane.clone(), *direction, window, cx); } SplitMode::MovePane => { self.split_and_move(pane.clone(), *direction, window, cx); } }; } pane::Event::JoinIntoNext => { self.join_pane_into_next(pane.clone(), window, cx); } pane::Event::JoinAll => { self.join_all_panes(window, cx); } pane::Event::Remove { focus_on_pane } => { self.remove_pane(pane.clone(), focus_on_pane.clone(), window, cx); } pane::Event::ActivateItem { local, focus_changed, } => { window.invalidate_character_coordinates(); pane.update(cx, |pane, _| { pane.track_alternate_file_items(); }); if *local { self.unfollow_in_pane(pane, window, cx); } serialize_workspace = *focus_changed || pane != self.active_pane(); if pane == self.active_pane() { self.active_item_path_changed(*focus_changed, window, cx); self.update_active_view_for_followers(window, cx); } else if *local { self.set_active_pane(pane, window, cx); } } pane::Event::UserSavedItem { item, save_intent } => { cx.emit(Event::UserSavedItem { pane: pane.downgrade(), item: item.boxed_clone(), save_intent: *save_intent, }); serialize_workspace = false; } pane::Event::ChangeItemTitle => { if *pane == self.active_pane { self.active_item_path_changed(false, window, cx); } serialize_workspace = false; } pane::Event::RemovedItem { item } => { cx.emit(Event::ActiveItemChanged); self.update_window_edited(window, cx); if let hash_map::Entry::Occupied(entry) = self.panes_by_item.entry(item.item_id()) && entry.get().entity_id() == pane.entity_id() { entry.remove(); } cx.emit(Event::ItemRemoved { item_id: item.item_id(), }); } pane::Event::Focus => { window.invalidate_character_coordinates(); self.handle_pane_focused(pane.clone(), window, cx); } pane::Event::ZoomIn => { if *pane == self.active_pane { pane.update(cx, |pane, cx| pane.set_zoomed(true, cx)); if pane.read(cx).has_focus(window, cx) { self.zoomed = Some(pane.downgrade().into()); self.zoomed_position = None; cx.emit(Event::ZoomChanged); } cx.notify(); } } pane::Event::ZoomOut => { pane.update(cx, |pane, cx| pane.set_zoomed(false, cx)); if self.zoomed_position.is_none() { self.zoomed = None; cx.emit(Event::ZoomChanged); } cx.notify(); } pane::Event::ItemPinned | pane::Event::ItemUnpinned => {} } if serialize_workspace { self.serialize_workspace(window, cx); } } pub fn unfollow_in_pane( &mut self, pane: &Entity, window: &mut Window, cx: &mut Context, ) -> Option { let leader_id = self.leader_for_pane(pane)?; self.unfollow(leader_id, window, cx); Some(leader_id) } pub fn split_pane( &mut self, pane_to_split: Entity, split_direction: SplitDirection, window: &mut Window, cx: &mut Context, ) -> Entity { let new_pane = self.add_pane(window, cx); self.center .split(&pane_to_split, &new_pane, split_direction, cx); cx.notify(); new_pane } pub fn split_and_move( &mut self, pane: Entity, direction: SplitDirection, window: &mut Window, cx: &mut Context, ) { let Some(item) = pane.update(cx, |pane, cx| pane.take_active_item(window, cx)) else { return; }; let new_pane = self.add_pane(window, cx); new_pane.update(cx, |pane, cx| { pane.add_item(item, true, true, None, window, cx) }); self.center.split(&pane, &new_pane, direction, cx); cx.notify(); } pub fn split_and_clone( &mut self, pane: Entity, direction: SplitDirection, window: &mut Window, cx: &mut Context, ) -> Task>> { let Some(item) = pane.read(cx).active_item() else { return Task::ready(None); }; if !item.can_split(cx) { return Task::ready(None); } let task = item.clone_on_split(self.database_id(), window, cx); cx.spawn_in(window, async move |this, cx| { if let Some(clone) = task.await { this.update_in(cx, |this, window, cx| { let new_pane = this.add_pane(window, cx); let nav_history = pane.read(cx).fork_nav_history(); new_pane.update(cx, |pane, cx| { pane.set_nav_history(nav_history, cx); pane.add_item(clone, true, true, None, window, cx) }); this.center.split(&pane, &new_pane, direction, cx); cx.notify(); new_pane }) .ok() } else { None } }) } pub fn join_all_panes(&mut self, window: &mut Window, cx: &mut Context) { let active_item = self.active_pane.read(cx).active_item(); for pane in &self.panes { join_pane_into_active(&self.active_pane, pane, window, cx); } if let Some(active_item) = active_item { self.activate_item(active_item.as_ref(), true, true, window, cx); } cx.notify(); } pub fn join_pane_into_next( &mut self, pane: Entity, window: &mut Window, cx: &mut Context, ) { let next_pane = self .find_pane_in_direction(SplitDirection::Right, cx) .or_else(|| self.find_pane_in_direction(SplitDirection::Down, cx)) .or_else(|| self.find_pane_in_direction(SplitDirection::Left, cx)) .or_else(|| self.find_pane_in_direction(SplitDirection::Up, cx)); let Some(next_pane) = next_pane else { return; }; move_all_items(&pane, &next_pane, window, cx); cx.notify(); } fn remove_pane( &mut self, pane: Entity, focus_on: Option>, window: &mut Window, cx: &mut Context, ) { if self.center.remove(&pane, cx).unwrap() { self.force_remove_pane(&pane, &focus_on, window, cx); self.unfollow_in_pane(&pane, window, cx); self.last_leaders_by_pane.remove(&pane.downgrade()); for removed_item in pane.read(cx).items() { self.panes_by_item.remove(&removed_item.item_id()); } cx.notify(); } else { self.active_item_path_changed(true, window, cx); } cx.emit(Event::PaneRemoved); } pub fn panes_mut(&mut self) -> &mut [Entity] { &mut self.panes } pub fn panes(&self) -> &[Entity] { &self.panes } pub fn active_pane(&self) -> &Entity { &self.active_pane } pub fn focused_pane(&self, window: &Window, cx: &App) -> Entity { for dock in self.all_docks() { if dock.focus_handle(cx).contains_focused(window, cx) && let Some(pane) = dock .read(cx) .active_panel() .and_then(|panel| panel.pane(cx)) { return pane; } } self.active_pane().clone() } pub fn adjacent_pane(&mut self, window: &mut Window, cx: &mut Context) -> Entity { self.find_pane_in_direction(SplitDirection::Right, cx) .unwrap_or_else(|| { self.split_pane(self.active_pane.clone(), SplitDirection::Right, window, cx) }) } pub fn pane_for(&self, handle: &dyn ItemHandle) -> Option> { self.pane_for_item_id(handle.item_id()) } pub fn pane_for_item_id(&self, item_id: EntityId) -> Option> { let weak_pane = self.panes_by_item.get(&item_id)?; weak_pane.upgrade() } pub fn pane_for_entity_id(&self, entity_id: EntityId) -> Option> { self.panes .iter() .find(|pane| pane.entity_id() == entity_id) .cloned() } fn collaborator_left(&mut self, peer_id: PeerId, window: &mut Window, cx: &mut Context) { self.follower_states.retain(|leader_id, state| { if *leader_id == CollaboratorId::PeerId(peer_id) { for item in state.items_by_leader_view_id.values() { item.view.set_leader_id(None, window, cx); } false } else { true } }); cx.notify(); } pub fn start_following( &mut self, leader_id: impl Into, window: &mut Window, cx: &mut Context, ) -> Option>> { let leader_id = leader_id.into(); let pane = self.active_pane().clone(); self.last_leaders_by_pane .insert(pane.downgrade(), leader_id); self.unfollow(leader_id, window, cx); self.unfollow_in_pane(&pane, window, cx); self.follower_states.insert( leader_id, FollowerState { center_pane: pane.clone(), dock_pane: None, active_view_id: None, items_by_leader_view_id: Default::default(), }, ); cx.notify(); match leader_id { CollaboratorId::PeerId(leader_peer_id) => { let room_id = self.active_call()?.room_id(cx)?; let project_id = self.project.read(cx).remote_id(); let request = self.app_state.client.request(proto::Follow { room_id, project_id, leader_id: Some(leader_peer_id), }); Some(cx.spawn_in(window, async move |this, cx| { let response = request.await?; this.update(cx, |this, _| { let state = this .follower_states .get_mut(&leader_id) .context("following interrupted")?; state.active_view_id = response .active_view .as_ref() .and_then(|view| ViewId::from_proto(view.id.clone()?).ok()); anyhow::Ok(()) })??; if let Some(view) = response.active_view { Self::add_view_from_leader(this.clone(), leader_peer_id, &view, cx).await?; } this.update_in(cx, |this, window, cx| { this.leader_updated(leader_id, window, cx) })?; Ok(()) })) } CollaboratorId::Agent => { self.leader_updated(leader_id, window, cx)?; Some(Task::ready(Ok(()))) } } } pub fn follow_next_collaborator( &mut self, _: &FollowNextCollaborator, window: &mut Window, cx: &mut Context, ) { let collaborators = self.project.read(cx).collaborators(); let next_leader_id = if let Some(leader_id) = self.leader_for_pane(&self.active_pane) { let mut collaborators = collaborators.keys().copied(); for peer_id in collaborators.by_ref() { if CollaboratorId::PeerId(peer_id) == leader_id { break; } } collaborators.next().map(CollaboratorId::PeerId) } else if let Some(last_leader_id) = self.last_leaders_by_pane.get(&self.active_pane.downgrade()) { match last_leader_id { CollaboratorId::PeerId(peer_id) => { if collaborators.contains_key(peer_id) { Some(*last_leader_id) } else { None } } CollaboratorId::Agent => Some(CollaboratorId::Agent), } } else { None }; let pane = self.active_pane.clone(); let Some(leader_id) = next_leader_id.or_else(|| { Some(CollaboratorId::PeerId( collaborators.keys().copied().next()?, )) }) else { return; }; if self.unfollow_in_pane(&pane, window, cx) == Some(leader_id) { return; } if let Some(task) = self.start_following(leader_id, window, cx) { task.detach_and_log_err(cx) } } pub fn follow( &mut self, leader_id: impl Into, window: &mut Window, cx: &mut Context, ) { let leader_id = leader_id.into(); if let CollaboratorId::PeerId(peer_id) = leader_id { let Some(active_call) = GlobalAnyActiveCall::try_global(cx) else { return; }; let Some(remote_participant) = active_call.0.remote_participant_for_peer_id(peer_id, cx) else { return; }; let project = self.project.read(cx); let other_project_id = match remote_participant.location { ParticipantLocation::External => None, ParticipantLocation::UnsharedProject => None, ParticipantLocation::SharedProject { project_id } => { if Some(project_id) == project.remote_id() { None } else { Some(project_id) } } }; // if they are active in another project, follow there. if let Some(project_id) = other_project_id { let app_state = self.app_state.clone(); crate::join_in_room_project(project_id, remote_participant.user.id, app_state, cx) .detach_and_prompt_err("Failed to join project", window, cx, |error, _, _| { Some(format!("{error:#}")) }); } } // if you're already following, find the right pane and focus it. if let Some(follower_state) = self.follower_states.get(&leader_id) { window.focus(&follower_state.pane().focus_handle(cx), cx); return; } // Otherwise, follow. if let Some(task) = self.start_following(leader_id, window, cx) { task.detach_and_log_err(cx) } } pub fn unfollow( &mut self, leader_id: impl Into, window: &mut Window, cx: &mut Context, ) -> Option<()> { cx.notify(); let leader_id = leader_id.into(); let state = self.follower_states.remove(&leader_id)?; for (_, item) in state.items_by_leader_view_id { item.view.set_leader_id(None, window, cx); } if let CollaboratorId::PeerId(leader_peer_id) = leader_id { let project_id = self.project.read(cx).remote_id(); let room_id = self.active_call()?.room_id(cx)?; self.app_state .client .send(proto::Unfollow { room_id, project_id, leader_id: Some(leader_peer_id), }) .log_err(); } Some(()) } pub fn is_being_followed(&self, id: impl Into) -> bool { self.follower_states.contains_key(&id.into()) } fn active_item_path_changed( &mut self, focus_changed: bool, window: &mut Window, cx: &mut Context, ) { cx.emit(Event::ActiveItemChanged); let active_entry = self.active_project_path(cx); self.project.update(cx, |project, cx| { project.set_active_path(active_entry.clone(), cx) }); if focus_changed && let Some(project_path) = &active_entry { let git_store_entity = self.project.read(cx).git_store().clone(); git_store_entity.update(cx, |git_store, cx| { git_store.set_active_repo_for_path(project_path, cx); }); } self.update_window_title(window, cx); } fn update_window_title(&mut self, window: &mut Window, cx: &mut App) { let project = self.project().read(cx); let mut title = String::new(); for (i, worktree) in project.visible_worktrees(cx).enumerate() { let name = { let settings_location = SettingsLocation { worktree_id: worktree.read(cx).id(), path: RelPath::empty(), }; let settings = WorktreeSettings::get(Some(settings_location), cx); match &settings.project_name { Some(name) => name.as_str(), None => worktree.read(cx).root_name_str(), } }; if i > 0 { title.push_str(", "); } title.push_str(name); } if title.is_empty() { title = "empty project".to_string(); } if let Some(path) = self.active_item(cx).and_then(|item| item.project_path(cx)) { let filename = path.path.file_name().or_else(|| { Some( project .worktree_for_id(path.worktree_id, cx)? .read(cx) .root_name_str(), ) }); if let Some(filename) = filename { title.push_str(" — "); title.push_str(filename.as_ref()); } } if project.is_via_collab() { title.push_str(" ↙"); } else if project.is_shared() { title.push_str(" ↗"); } if let Some(last_title) = self.last_window_title.as_ref() && &title == last_title { return; } window.set_window_title(&title); SystemWindowTabController::update_tab_title( cx, window.window_handle().window_id(), SharedString::from(&title), ); self.last_window_title = Some(title); } fn update_window_edited(&mut self, window: &mut Window, cx: &mut App) { let is_edited = !self.project.read(cx).is_disconnected(cx) && !self.dirty_items.is_empty(); if is_edited != self.window_edited { self.window_edited = is_edited; window.set_window_edited(self.window_edited) } } fn update_item_dirty_state( &mut self, item: &dyn ItemHandle, window: &mut Window, cx: &mut App, ) { let is_dirty = item.is_dirty(cx); let item_id = item.item_id(); let was_dirty = self.dirty_items.contains_key(&item_id); if is_dirty == was_dirty { return; } if was_dirty { self.dirty_items.remove(&item_id); self.update_window_edited(window, cx); return; } let workspace = self.weak_handle(); let Some(window_handle) = window.window_handle().downcast::() else { return; }; let on_release_callback = Box::new(move |cx: &mut App| { window_handle .update(cx, |_, window, cx| { workspace .update(cx, |workspace, cx| { workspace.dirty_items.remove(&item_id); workspace.update_window_edited(window, cx) }) .ok(); }) .ok(); }); let s = item.on_release(cx, on_release_callback); self.dirty_items.insert(item_id, s); self.update_window_edited(window, cx); } fn render_notifications(&self, _window: &mut Window, _cx: &mut Context) -> Option
{ if self.notifications.is_empty() { None } else { Some( div() .absolute() .right_3() .bottom_3() .w_112() .h_full() .flex() .flex_col() .justify_end() .gap_2() .children( self.notifications .iter() .map(|(_, notification)| notification.clone().into_any()), ), ) } } // RPC handlers fn active_view_for_follower( &self, follower_project_id: Option, window: &mut Window, cx: &mut Context, ) -> Option { let (item, panel_id) = self.active_item_for_followers(window, cx); let item = item?; let leader_id = self .pane_for(&*item) .and_then(|pane| self.leader_for_pane(&pane)); let leader_peer_id = match leader_id { Some(CollaboratorId::PeerId(peer_id)) => Some(peer_id), Some(CollaboratorId::Agent) | None => None, }; let item_handle = item.to_followable_item_handle(cx)?; let id = item_handle.remote_id(&self.app_state.client, window, cx)?; let variant = item_handle.to_state_proto(window, cx)?; if item_handle.is_project_item(window, cx) && (follower_project_id.is_none() || follower_project_id != self.project.read(cx).remote_id()) { return None; } Some(proto::View { id: id.to_proto(), leader_id: leader_peer_id, variant: Some(variant), panel_id: panel_id.map(|id| id as i32), }) } fn handle_follow( &mut self, follower_project_id: Option, window: &mut Window, cx: &mut Context, ) -> proto::FollowResponse { let active_view = self.active_view_for_follower(follower_project_id, window, cx); cx.notify(); proto::FollowResponse { views: active_view.iter().cloned().collect(), active_view, } } fn handle_update_followers( &mut self, leader_id: PeerId, message: proto::UpdateFollowers, _window: &mut Window, _cx: &mut Context, ) { self.leader_updates_tx .unbounded_send((leader_id, message)) .ok(); } async fn process_leader_update( this: &WeakEntity, leader_id: PeerId, update: proto::UpdateFollowers, cx: &mut AsyncWindowContext, ) -> Result<()> { match update.variant.context("invalid update")? { proto::update_followers::Variant::CreateView(view) => { let view_id = ViewId::from_proto(view.id.clone().context("invalid view id")?)?; let should_add_view = this.update(cx, |this, _| { if let Some(state) = this.follower_states.get_mut(&leader_id.into()) { anyhow::Ok(!state.items_by_leader_view_id.contains_key(&view_id)) } else { anyhow::Ok(false) } })??; if should_add_view { Self::add_view_from_leader(this.clone(), leader_id, &view, cx).await? } } proto::update_followers::Variant::UpdateActiveView(update_active_view) => { let should_add_view = this.update(cx, |this, _| { if let Some(state) = this.follower_states.get_mut(&leader_id.into()) { state.active_view_id = update_active_view .view .as_ref() .and_then(|view| ViewId::from_proto(view.id.clone()?).ok()); if state.active_view_id.is_some_and(|view_id| { !state.items_by_leader_view_id.contains_key(&view_id) }) { anyhow::Ok(true) } else { anyhow::Ok(false) } } else { anyhow::Ok(false) } })??; if should_add_view && let Some(view) = update_active_view.view { Self::add_view_from_leader(this.clone(), leader_id, &view, cx).await? } } proto::update_followers::Variant::UpdateView(update_view) => { let variant = update_view.variant.context("missing update view variant")?; let id = update_view.id.context("missing update view id")?; let mut tasks = Vec::new(); this.update_in(cx, |this, window, cx| { let project = this.project.clone(); if let Some(state) = this.follower_states.get(&leader_id.into()) { let view_id = ViewId::from_proto(id.clone())?; if let Some(item) = state.items_by_leader_view_id.get(&view_id) { tasks.push(item.view.apply_update_proto( &project, variant.clone(), window, cx, )); } } anyhow::Ok(()) })??; try_join_all(tasks).await.log_err(); } } this.update_in(cx, |this, window, cx| { this.leader_updated(leader_id, window, cx) })?; Ok(()) } async fn add_view_from_leader( this: WeakEntity, leader_id: PeerId, view: &proto::View, cx: &mut AsyncWindowContext, ) -> Result<()> { let this = this.upgrade().context("workspace dropped")?; let Some(id) = view.id.clone() else { anyhow::bail!("no id for view"); }; let id = ViewId::from_proto(id)?; let panel_id = view.panel_id.and_then(proto::PanelId::from_i32); let pane = this.update(cx, |this, _cx| { let state = this .follower_states .get(&leader_id.into()) .context("stopped following")?; anyhow::Ok(state.pane().clone()) })?; let existing_item = pane.update_in(cx, |pane, window, cx| { let client = this.read(cx).client().clone(); pane.items().find_map(|item| { let item = item.to_followable_item_handle(cx)?; if item.remote_id(&client, window, cx) == Some(id) { Some(item) } else { None } }) })?; let item = if let Some(existing_item) = existing_item { existing_item } else { let variant = view.variant.clone(); anyhow::ensure!(variant.is_some(), "missing view variant"); let task = cx.update(|window, cx| { FollowableViewRegistry::from_state_proto(this.clone(), id, variant, window, cx) })?; let Some(task) = task else { anyhow::bail!( "failed to construct view from leader (maybe from a different version of zed?)" ); }; let mut new_item = task.await?; pane.update_in(cx, |pane, window, cx| { let mut item_to_remove = None; for (ix, item) in pane.items().enumerate() { if let Some(item) = item.to_followable_item_handle(cx) { match new_item.dedup(item.as_ref(), window, cx) { Some(item::Dedup::KeepExisting) => { new_item = item.boxed_clone().to_followable_item_handle(cx).unwrap(); break; } Some(item::Dedup::ReplaceExisting) => { item_to_remove = Some((ix, item.item_id())); break; } None => {} } } } if let Some((ix, id)) = item_to_remove { pane.remove_item(id, false, false, window, cx); pane.add_item(new_item.boxed_clone(), false, false, Some(ix), window, cx); } })?; new_item }; this.update_in(cx, |this, window, cx| { let state = this.follower_states.get_mut(&leader_id.into())?; item.set_leader_id(Some(leader_id.into()), window, cx); state.items_by_leader_view_id.insert( id, FollowerView { view: item, location: panel_id, }, ); Some(()) }) .context("no follower state")?; Ok(()) } fn handle_agent_location_changed(&mut self, window: &mut Window, cx: &mut Context) { let Some(follower_state) = self.follower_states.get_mut(&CollaboratorId::Agent) else { return; }; if let Some(agent_location) = self.project.read(cx).agent_location() { let buffer_entity_id = agent_location.buffer.entity_id(); let view_id = ViewId { creator: CollaboratorId::Agent, id: buffer_entity_id.as_u64(), }; follower_state.active_view_id = Some(view_id); let item = match follower_state.items_by_leader_view_id.entry(view_id) { hash_map::Entry::Occupied(entry) => Some(entry.into_mut()), hash_map::Entry::Vacant(entry) => { let existing_view = follower_state .center_pane .read(cx) .items() .find_map(|item| { let item = item.to_followable_item_handle(cx)?; if item.buffer_kind(cx) == ItemBufferKind::Singleton && item.project_item_model_ids(cx).as_slice() == [buffer_entity_id] { Some(item) } else { None } }); let view = existing_view.or_else(|| { agent_location.buffer.upgrade().and_then(|buffer| { cx.update_default_global(|registry: &mut ProjectItemRegistry, cx| { registry.build_item(buffer, self.project.clone(), None, window, cx) })? .to_followable_item_handle(cx) }) }); view.map(|view| { entry.insert(FollowerView { view, location: None, }) }) } }; if let Some(item) = item { item.view .set_leader_id(Some(CollaboratorId::Agent), window, cx); item.view .update_agent_location(agent_location.position, window, cx); } } else { follower_state.active_view_id = None; } self.leader_updated(CollaboratorId::Agent, window, cx); } pub fn update_active_view_for_followers(&mut self, window: &mut Window, cx: &mut App) { let mut is_project_item = true; let mut update = proto::UpdateActiveView::default(); if window.is_window_active() { let (active_item, panel_id) = self.active_item_for_followers(window, cx); if let Some(item) = active_item && item.item_focus_handle(cx).contains_focused(window, cx) { let leader_id = self .pane_for(&*item) .and_then(|pane| self.leader_for_pane(&pane)); let leader_peer_id = match leader_id { Some(CollaboratorId::PeerId(peer_id)) => Some(peer_id), Some(CollaboratorId::Agent) | None => None, }; if let Some(item) = item.to_followable_item_handle(cx) { let id = item .remote_id(&self.app_state.client, window, cx) .map(|id| id.to_proto()); if let Some(id) = id && let Some(variant) = item.to_state_proto(window, cx) { let view = Some(proto::View { id, leader_id: leader_peer_id, variant: Some(variant), panel_id: panel_id.map(|id| id as i32), }); is_project_item = item.is_project_item(window, cx); update = proto::UpdateActiveView { view }; }; } } } let active_view_id = update.view.as_ref().and_then(|view| view.id.as_ref()); if active_view_id != self.last_active_view_id.as_ref() { self.last_active_view_id = active_view_id.cloned(); self.update_followers( is_project_item, proto::update_followers::Variant::UpdateActiveView(update), window, cx, ); } } fn active_item_for_followers( &self, window: &mut Window, cx: &mut App, ) -> (Option>, Option) { let mut active_item = None; let mut panel_id = None; for dock in self.all_docks() { if dock.focus_handle(cx).contains_focused(window, cx) && let Some(panel) = dock.read(cx).active_panel() && let Some(pane) = panel.pane(cx) && let Some(item) = pane.read(cx).active_item() { active_item = Some(item); panel_id = panel.remote_id(); break; } } if active_item.is_none() { active_item = self.active_pane().read(cx).active_item(); } (active_item, panel_id) } fn update_followers( &self, project_only: bool, update: proto::update_followers::Variant, _: &mut Window, cx: &mut App, ) -> Option<()> { // If this update only applies to for followers in the current project, // then skip it unless this project is shared. If it applies to all // followers, regardless of project, then set `project_id` to none, // indicating that it goes to all followers. let project_id = if project_only { Some(self.project.read(cx).remote_id()?) } else { None }; self.app_state().workspace_store.update(cx, |store, cx| { store.update_followers(project_id, update, cx) }) } pub fn leader_for_pane(&self, pane: &Entity) -> Option { self.follower_states.iter().find_map(|(leader_id, state)| { if state.center_pane == *pane || state.dock_pane.as_ref() == Some(pane) { Some(*leader_id) } else { None } }) } fn leader_updated( &mut self, leader_id: impl Into, window: &mut Window, cx: &mut Context, ) -> Option> { cx.notify(); let leader_id = leader_id.into(); let (panel_id, item) = match leader_id { CollaboratorId::PeerId(peer_id) => self.active_item_for_peer(peer_id, window, cx)?, CollaboratorId::Agent => (None, self.active_item_for_agent()?), }; let state = self.follower_states.get(&leader_id)?; let mut transfer_focus = state.center_pane.read(cx).has_focus(window, cx); let pane; if let Some(panel_id) = panel_id { pane = self .activate_panel_for_proto_id(panel_id, window, cx)? .pane(cx)?; let state = self.follower_states.get_mut(&leader_id)?; state.dock_pane = Some(pane.clone()); } else { pane = state.center_pane.clone(); let state = self.follower_states.get_mut(&leader_id)?; if let Some(dock_pane) = state.dock_pane.take() { transfer_focus |= dock_pane.focus_handle(cx).contains_focused(window, cx); } } pane.update(cx, |pane, cx| { let focus_active_item = pane.has_focus(window, cx) || transfer_focus; if let Some(index) = pane.index_for_item(item.as_ref()) { pane.activate_item(index, false, false, window, cx); } else { pane.add_item(item.boxed_clone(), false, false, None, window, cx) } if focus_active_item { pane.focus_active_item(window, cx) } }); Some(item) } fn active_item_for_agent(&self) -> Option> { let state = self.follower_states.get(&CollaboratorId::Agent)?; let active_view_id = state.active_view_id?; Some( state .items_by_leader_view_id .get(&active_view_id)? .view .boxed_clone(), ) } fn active_item_for_peer( &self, peer_id: PeerId, window: &mut Window, cx: &mut Context, ) -> Option<(Option, Box)> { let call = self.active_call()?; let participant = call.remote_participant_for_peer_id(peer_id, cx)?; let leader_in_this_app; let leader_in_this_project; match participant.location { ParticipantLocation::SharedProject { project_id } => { leader_in_this_app = true; leader_in_this_project = Some(project_id) == self.project.read(cx).remote_id(); } ParticipantLocation::UnsharedProject => { leader_in_this_app = true; leader_in_this_project = false; } ParticipantLocation::External => { leader_in_this_app = false; leader_in_this_project = false; } }; let state = self.follower_states.get(&peer_id.into())?; let mut item_to_activate = None; if let (Some(active_view_id), true) = (state.active_view_id, leader_in_this_app) { if let Some(item) = state.items_by_leader_view_id.get(&active_view_id) && (leader_in_this_project || !item.view.is_project_item(window, cx)) { item_to_activate = Some((item.location, item.view.boxed_clone())); } } else if let Some(shared_screen) = self.shared_screen_for_peer(peer_id, &state.center_pane, window, cx) { item_to_activate = Some((None, Box::new(shared_screen))); } item_to_activate } fn shared_screen_for_peer( &self, peer_id: PeerId, pane: &Entity, window: &mut Window, cx: &mut App, ) -> Option> { self.active_call()? .create_shared_screen(peer_id, pane, window, cx) } pub fn on_window_activation_changed(&mut self, window: &mut Window, cx: &mut Context) { if window.is_window_active() { self.update_active_view_for_followers(window, cx); if let Some(database_id) = self.database_id { let db = WorkspaceDb::global(cx); cx.background_spawn(async move { db.update_timestamp(database_id).await }) .detach(); } } else { for pane in &self.panes { pane.update(cx, |pane, cx| { if let Some(item) = pane.active_item() { item.workspace_deactivated(window, cx); } for item in pane.items() { if matches!( item.workspace_settings(cx).autosave, AutosaveSetting::OnWindowChange | AutosaveSetting::OnFocusChange ) { Pane::autosave_item(item.as_ref(), self.project.clone(), window, cx) .detach_and_log_err(cx); } } }); } } } pub fn active_call(&self) -> Option<&dyn AnyActiveCall> { self.active_call.as_ref().map(|(call, _)| &*call.0) } pub fn active_global_call(&self) -> Option { self.active_call.as_ref().map(|(call, _)| call.clone()) } fn on_active_call_event( &mut self, event: &ActiveCallEvent, window: &mut Window, cx: &mut Context, ) { match event { ActiveCallEvent::ParticipantLocationChanged { participant_id } | ActiveCallEvent::RemoteVideoTracksChanged { participant_id } => { self.leader_updated(participant_id, window, cx); } } } pub fn database_id(&self) -> Option { self.database_id } #[cfg(any(test, feature = "test-support"))] pub(crate) fn set_database_id(&mut self, id: WorkspaceId) { self.database_id = Some(id); } pub fn session_id(&self) -> Option { self.session_id.clone() } fn save_window_bounds(&self, window: &mut Window, cx: &mut App) -> Task<()> { let Some(display) = window.display(cx) else { return Task::ready(()); }; let Ok(display_uuid) = display.uuid() else { return Task::ready(()); }; let window_bounds = window.inner_window_bounds(); let database_id = self.database_id; let has_paths = !self.root_paths(cx).is_empty(); let db = WorkspaceDb::global(cx); let kvp = db::kvp::KeyValueStore::global(cx); cx.background_executor().spawn(async move { if !has_paths { persistence::write_default_window_bounds(&kvp, window_bounds, display_uuid) .await .log_err(); } if let Some(database_id) = database_id { db.set_window_open_status( database_id, SerializedWindowBounds(window_bounds), display_uuid, ) .await .log_err(); } else { persistence::write_default_window_bounds(&kvp, window_bounds, display_uuid) .await .log_err(); } }) } /// Bypass the 200ms serialization throttle and write workspace state to /// the DB immediately. Returns a task the caller can await to ensure the /// write completes. Used by the quit handler so the most recent state /// isn't lost to a pending throttle timer when the process exits. pub fn flush_serialization(&mut self, window: &mut Window, cx: &mut App) -> Task<()> { self._schedule_serialize_workspace.take(); self._serialize_workspace_task.take(); self.bounds_save_task_queued.take(); let bounds_task = self.save_window_bounds(window, cx); let serialize_task = self.serialize_workspace_internal(window, cx); cx.spawn(async move |_| { bounds_task.await; serialize_task.await; }) } pub fn root_paths(&self, cx: &App) -> Vec> { let project = self.project().read(cx); project .visible_worktrees(cx) .map(|worktree| worktree.read(cx).abs_path()) .collect::>() } fn remove_panes(&mut self, member: Member, window: &mut Window, cx: &mut Context) { match member { Member::Axis(PaneAxis { members, .. }) => { for child in members.iter() { self.remove_panes(child.clone(), window, cx) } } Member::Pane(pane) => { self.force_remove_pane(&pane, &None, window, cx); } } } fn remove_from_session(&mut self, window: &mut Window, cx: &mut App) -> Task<()> { self.session_id.take(); self.serialize_workspace_internal(window, cx) } fn force_remove_pane( &mut self, pane: &Entity, focus_on: &Option>, window: &mut Window, cx: &mut Context, ) { self.panes.retain(|p| p != pane); if let Some(focus_on) = focus_on { focus_on.update(cx, |pane, cx| window.focus(&pane.focus_handle(cx), cx)); } else if self.active_pane() == pane { self.panes .last() .unwrap() .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx), cx)); } if self.last_active_center_pane == Some(pane.downgrade()) { self.last_active_center_pane = None; } cx.notify(); } fn serialize_workspace(&mut self, window: &mut Window, cx: &mut Context) { if self._schedule_serialize_workspace.is_none() { self._schedule_serialize_workspace = Some(cx.spawn_in(window, async move |this, cx| { cx.background_executor() .timer(SERIALIZATION_THROTTLE_TIME) .await; this.update_in(cx, |this, window, cx| { this._serialize_workspace_task = Some(this.serialize_workspace_internal(window, cx)); this._schedule_serialize_workspace.take(); }) .log_err(); })); } } fn serialize_workspace_internal(&self, window: &mut Window, cx: &mut App) -> Task<()> { let Some(database_id) = self.database_id() else { return Task::ready(()); }; fn serialize_pane_handle( pane_handle: &Entity, window: &mut Window, cx: &mut App, ) -> SerializedPane { let (items, active, pinned_count) = { let pane = pane_handle.read(cx); let active_item_id = pane.active_item().map(|item| item.item_id()); ( pane.items() .filter_map(|handle| { let handle = handle.to_serializable_item_handle(cx)?; Some(SerializedItem { kind: Arc::from(handle.serialized_item_kind()), item_id: handle.item_id().as_u64(), active: Some(handle.item_id()) == active_item_id, preview: pane.is_active_preview_item(handle.item_id()), }) }) .collect::>(), pane.has_focus(window, cx), pane.pinned_count(), ) }; SerializedPane::new(items, active, pinned_count) } fn build_serialized_pane_group( pane_group: &Member, window: &mut Window, cx: &mut App, ) -> SerializedPaneGroup { match pane_group { Member::Axis(PaneAxis { axis, members, flexes, bounding_boxes: _, }) => SerializedPaneGroup::Group { axis: SerializedAxis(*axis), children: members .iter() .map(|member| build_serialized_pane_group(member, window, cx)) .collect::>(), flexes: Some(flexes.lock().clone()), }, Member::Pane(pane_handle) => { SerializedPaneGroup::Pane(serialize_pane_handle(pane_handle, window, cx)) } } } fn build_serialized_docks( this: &Workspace, window: &mut Window, cx: &mut App, ) -> DockStructure { this.capture_dock_state(window, cx) } match self.workspace_location(cx) { WorkspaceLocation::Location(location, paths) => { let breakpoints = self.project.update(cx, |project, cx| { project .breakpoint_store() .read(cx) .all_source_breakpoints(cx) }); let user_toolchains = self .project .read(cx) .user_toolchains(cx) .unwrap_or_default(); let center_group = build_serialized_pane_group(&self.center.root, window, cx); let docks = build_serialized_docks(self, window, cx); let window_bounds = Some(SerializedWindowBounds(window.window_bounds())); let serialized_workspace = SerializedWorkspace { id: database_id, location, paths, center_group, window_bounds, display: Default::default(), docks, centered_layout: self.centered_layout, session_id: self.session_id.clone(), breakpoints, window_id: Some(window.window_handle().window_id().as_u64()), user_toolchains, }; let db = WorkspaceDb::global(cx); window.spawn(cx, async move |_| { db.save_workspace(serialized_workspace).await; }) } WorkspaceLocation::DetachFromSession => { let window_bounds = SerializedWindowBounds(window.window_bounds()); let display = window.display(cx).and_then(|d| d.uuid().ok()); // Save dock state for empty local workspaces let docks = build_serialized_docks(self, window, cx); let db = WorkspaceDb::global(cx); let kvp = db::kvp::KeyValueStore::global(cx); window.spawn(cx, async move |_| { db.set_window_open_status( database_id, window_bounds, display.unwrap_or_default(), ) .await .log_err(); db.set_session_id(database_id, None).await.log_err(); persistence::write_default_dock_state(&kvp, docks) .await .log_err(); }) } WorkspaceLocation::None => { // Save dock state for empty non-local workspaces let docks = build_serialized_docks(self, window, cx); let kvp = db::kvp::KeyValueStore::global(cx); window.spawn(cx, async move |_| { persistence::write_default_dock_state(&kvp, docks) .await .log_err(); }) } } } fn has_any_items_open(&self, cx: &App) -> bool { self.panes.iter().any(|pane| pane.read(cx).items_len() > 0) } fn workspace_location(&self, cx: &App) -> WorkspaceLocation { let paths = PathList::new(&self.root_paths(cx)); if let Some(connection) = self.project.read(cx).remote_connection_options(cx) { WorkspaceLocation::Location(SerializedWorkspaceLocation::Remote(connection), paths) } else if self.project.read(cx).is_local() { if !paths.is_empty() || self.has_any_items_open(cx) { WorkspaceLocation::Location(SerializedWorkspaceLocation::Local, paths) } else { WorkspaceLocation::DetachFromSession } } else { WorkspaceLocation::None } } fn update_history(&self, cx: &mut App) { let Some(id) = self.database_id() else { return; }; if !self.project.read(cx).is_local() { return; } if let Some(manager) = HistoryManager::global(cx) { let paths = PathList::new(&self.root_paths(cx)); manager.update(cx, |this, cx| { this.update_history(id, HistoryManagerEntry::new(id, &paths), cx); }); } } async fn serialize_items( this: &WeakEntity, items_rx: UnboundedReceiver>, cx: &mut AsyncWindowContext, ) -> Result<()> { const CHUNK_SIZE: usize = 200; let mut serializable_items = items_rx.ready_chunks(CHUNK_SIZE); while let Some(items_received) = serializable_items.next().await { let unique_items = items_received .into_iter() .fold(HashMap::default(), |mut acc, item| { acc.entry(item.item_id()).or_insert(item); acc }); // We use into_iter() here so that the references to the items are moved into // the tasks and not kept alive while we're sleeping. for (_, item) in unique_items.into_iter() { if let Ok(Some(task)) = this.update_in(cx, |workspace, window, cx| { item.serialize(workspace, false, window, cx) }) { cx.background_spawn(async move { task.await.log_err() }) .detach(); } } cx.background_executor() .timer(SERIALIZATION_THROTTLE_TIME) .await; } Ok(()) } pub(crate) fn enqueue_item_serialization( &mut self, item: Box, ) -> Result<()> { self.serializable_items_tx .unbounded_send(item) .map_err(|err| anyhow!("failed to send serializable item over channel: {err}")) } pub(crate) fn load_workspace( serialized_workspace: SerializedWorkspace, paths_to_open: Vec>, window: &mut Window, cx: &mut Context, ) -> Task>>>> { cx.spawn_in(window, async move |workspace, cx| { let project = workspace.read_with(cx, |workspace, _| workspace.project().clone())?; let mut center_group = None; let mut center_items = None; // Traverse the splits tree and add to things if let Some((group, active_pane, items)) = serialized_workspace .center_group .deserialize(&project, serialized_workspace.id, workspace.clone(), cx) .await { center_items = Some(items); center_group = Some((group, active_pane)) } let mut items_by_project_path = HashMap::default(); let mut item_ids_by_kind = HashMap::default(); let mut all_deserialized_items = Vec::default(); cx.update(|_, cx| { for item in center_items.unwrap_or_default().into_iter().flatten() { if let Some(serializable_item_handle) = item.to_serializable_item_handle(cx) { item_ids_by_kind .entry(serializable_item_handle.serialized_item_kind()) .or_insert(Vec::new()) .push(item.item_id().as_u64() as ItemId); } if let Some(project_path) = item.project_path(cx) { items_by_project_path.insert(project_path, item.clone()); } all_deserialized_items.push(item); } })?; let opened_items = paths_to_open .into_iter() .map(|path_to_open| { path_to_open .and_then(|path_to_open| items_by_project_path.remove(&path_to_open)) }) .collect::>(); // Remove old panes from workspace panes list workspace.update_in(cx, |workspace, window, cx| { if let Some((center_group, active_pane)) = center_group { workspace.remove_panes(workspace.center.root.clone(), window, cx); // Swap workspace center group workspace.center = PaneGroup::with_root(center_group); workspace.center.set_is_center(true); workspace.center.mark_positions(cx); if let Some(active_pane) = active_pane { workspace.set_active_pane(&active_pane, window, cx); cx.focus_self(window); } else { workspace.set_active_pane(&workspace.center.first_pane(), window, cx); } } let docks = serialized_workspace.docks; for (dock, serialized_dock) in [ (&mut workspace.right_dock, docks.right), (&mut workspace.left_dock, docks.left), (&mut workspace.bottom_dock, docks.bottom), ] .iter_mut() { dock.update(cx, |dock, cx| { dock.serialized_dock = Some(serialized_dock.clone()); dock.restore_state(window, cx); }); } cx.notify(); })?; let _ = project .update(cx, |project, cx| { project .breakpoint_store() .update(cx, |breakpoint_store, cx| { breakpoint_store .with_serialized_breakpoints(serialized_workspace.breakpoints, cx) }) }) .await; // Clean up all the items that have _not_ been loaded. Our ItemIds aren't stable. That means // after loading the items, we might have different items and in order to avoid // the database filling up, we delete items that haven't been loaded now. // // The items that have been loaded, have been saved after they've been added to the workspace. let clean_up_tasks = workspace.update_in(cx, |_, window, cx| { item_ids_by_kind .into_iter() .map(|(item_kind, loaded_items)| { SerializableItemRegistry::cleanup( item_kind, serialized_workspace.id, loaded_items, window, cx, ) .log_err() }) .collect::>() })?; futures::future::join_all(clean_up_tasks).await; workspace .update_in(cx, |workspace, window, cx| { // Serialize ourself to make sure our timestamps and any pane / item changes are replicated workspace.serialize_workspace_internal(window, cx).detach(); // Ensure that we mark the window as edited if we did load dirty items workspace.update_window_edited(window, cx); }) .ok(); Ok(opened_items) }) } pub fn key_context(&self, cx: &App) -> KeyContext { let mut context = KeyContext::new_with_defaults(); context.add("Workspace"); context.set("keyboard_layout", cx.keyboard_layout().name().to_string()); if let Some(status) = self .debugger_provider .as_ref() .and_then(|provider| provider.active_thread_state(cx)) { match status { ThreadStatus::Running | ThreadStatus::Stepping => { context.add("debugger_running"); } ThreadStatus::Stopped => context.add("debugger_stopped"), ThreadStatus::Exited | ThreadStatus::Ended => {} } } if self.left_dock.read(cx).is_open() { if let Some(active_panel) = self.left_dock.read(cx).active_panel() { context.set("left_dock", active_panel.panel_key()); } } if self.right_dock.read(cx).is_open() { if let Some(active_panel) = self.right_dock.read(cx).active_panel() { context.set("right_dock", active_panel.panel_key()); } } if self.bottom_dock.read(cx).is_open() { if let Some(active_panel) = self.bottom_dock.read(cx).active_panel() { context.set("bottom_dock", active_panel.panel_key()); } } context } /// Multiworkspace uses this to add workspace action handling to itself pub fn actions(&self, div: Div, window: &mut Window, cx: &mut Context) -> Div { self.add_workspace_actions_listeners(div, window, cx) .on_action(cx.listener( |_workspace, action_sequence: &settings::ActionSequence, window, cx| { for action in &action_sequence.0 { window.dispatch_action(action.boxed_clone(), cx); } }, )) .on_action(cx.listener(Self::close_inactive_items_and_panes)) .on_action(cx.listener(Self::close_all_items_and_panes)) .on_action(cx.listener(Self::close_item_in_all_panes)) .on_action(cx.listener(Self::save_all)) .on_action(cx.listener(Self::send_keystrokes)) .on_action(cx.listener(Self::add_folder_to_project)) .on_action(cx.listener(Self::follow_next_collaborator)) .on_action(cx.listener(Self::activate_pane_at_index)) .on_action(cx.listener(Self::move_item_to_pane_at_index)) .on_action(cx.listener(Self::move_focused_panel_to_next_position)) .on_action(cx.listener(Self::toggle_edit_predictions_all_files)) .on_action(cx.listener(Self::toggle_theme_mode)) .on_action(cx.listener(|workspace, _: &Unfollow, window, cx| { let pane = workspace.active_pane().clone(); workspace.unfollow_in_pane(&pane, window, cx); })) .on_action(cx.listener(|workspace, action: &Save, window, cx| { workspace .save_active_item(action.save_intent.unwrap_or(SaveIntent::Save), window, cx) .detach_and_prompt_err("Failed to save", window, cx, |_, _, _| None); })) .on_action(cx.listener(|workspace, _: &SaveWithoutFormat, window, cx| { workspace .save_active_item(SaveIntent::SaveWithoutFormat, window, cx) .detach_and_prompt_err("Failed to save", window, cx, |_, _, _| None); })) .on_action(cx.listener(|workspace, _: &SaveAs, window, cx| { workspace .save_active_item(SaveIntent::SaveAs, window, cx) .detach_and_prompt_err("Failed to save", window, cx, |_, _, _| None); })) .on_action( cx.listener(|workspace, _: &ActivatePreviousPane, window, cx| { workspace.activate_previous_pane(window, cx) }), ) .on_action(cx.listener(|workspace, _: &ActivateNextPane, window, cx| { workspace.activate_next_pane(window, cx) })) .on_action(cx.listener(|workspace, _: &ActivateLastPane, window, cx| { workspace.activate_last_pane(window, cx) })) .on_action( cx.listener(|workspace, _: &ActivateNextWindow, _window, cx| { workspace.activate_next_window(cx) }), ) .on_action( cx.listener(|workspace, _: &ActivatePreviousWindow, _window, cx| { workspace.activate_previous_window(cx) }), ) .on_action(cx.listener(|workspace, _: &ActivatePaneLeft, window, cx| { workspace.activate_pane_in_direction(SplitDirection::Left, window, cx) })) .on_action(cx.listener(|workspace, _: &ActivatePaneRight, window, cx| { workspace.activate_pane_in_direction(SplitDirection::Right, window, cx) })) .on_action(cx.listener(|workspace, _: &ActivatePaneUp, window, cx| { workspace.activate_pane_in_direction(SplitDirection::Up, window, cx) })) .on_action(cx.listener(|workspace, _: &ActivatePaneDown, window, cx| { workspace.activate_pane_in_direction(SplitDirection::Down, window, cx) })) .on_action(cx.listener( |workspace, action: &MoveItemToPaneInDirection, window, cx| { workspace.move_item_to_pane_in_direction(action, window, cx) }, )) .on_action(cx.listener(|workspace, _: &SwapPaneLeft, _, cx| { workspace.swap_pane_in_direction(SplitDirection::Left, cx) })) .on_action(cx.listener(|workspace, _: &SwapPaneRight, _, cx| { workspace.swap_pane_in_direction(SplitDirection::Right, cx) })) .on_action(cx.listener(|workspace, _: &SwapPaneUp, _, cx| { workspace.swap_pane_in_direction(SplitDirection::Up, cx) })) .on_action(cx.listener(|workspace, _: &SwapPaneDown, _, cx| { workspace.swap_pane_in_direction(SplitDirection::Down, cx) })) .on_action(cx.listener(|workspace, _: &SwapPaneAdjacent, window, cx| { const DIRECTION_PRIORITY: [SplitDirection; 4] = [ SplitDirection::Down, SplitDirection::Up, SplitDirection::Right, SplitDirection::Left, ]; for dir in DIRECTION_PRIORITY { if workspace.find_pane_in_direction(dir, cx).is_some() { workspace.swap_pane_in_direction(dir, cx); workspace.activate_pane_in_direction(dir.opposite(), window, cx); break; } } })) .on_action(cx.listener(|workspace, _: &MovePaneLeft, _, cx| { workspace.move_pane_to_border(SplitDirection::Left, cx) })) .on_action(cx.listener(|workspace, _: &MovePaneRight, _, cx| { workspace.move_pane_to_border(SplitDirection::Right, cx) })) .on_action(cx.listener(|workspace, _: &MovePaneUp, _, cx| { workspace.move_pane_to_border(SplitDirection::Up, cx) })) .on_action(cx.listener(|workspace, _: &MovePaneDown, _, cx| { workspace.move_pane_to_border(SplitDirection::Down, cx) })) .on_action(cx.listener(|this, _: &ToggleLeftDock, window, cx| { this.toggle_dock(DockPosition::Left, window, cx); })) .on_action(cx.listener( |workspace: &mut Workspace, _: &ToggleRightDock, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx); }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &ToggleBottomDock, window, cx| { workspace.toggle_dock(DockPosition::Bottom, window, cx); }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &CloseActiveDock, window, cx| { if !workspace.close_active_dock(window, cx) { cx.propagate(); } }, )) .on_action( cx.listener(|workspace: &mut Workspace, _: &CloseAllDocks, window, cx| { workspace.close_all_docks(window, cx); }), ) .on_action(cx.listener(Self::toggle_all_docks)) .on_action(cx.listener( |workspace: &mut Workspace, _: &ClearAllNotifications, _, cx| { workspace.clear_all_notifications(cx); }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &ClearNavigationHistory, window, cx| { workspace.clear_navigation_history(window, cx); }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &SuppressNotification, _, cx| { if let Some((notification_id, _)) = workspace.notifications.pop() { workspace.suppress_notification(¬ification_id, cx); } }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &ToggleWorktreeSecurity, window, cx| { workspace.show_worktree_trust_security_modal(true, window, cx); }, )) .on_action( cx.listener(|_: &mut Workspace, _: &ClearTrustedWorktrees, _, cx| { if let Some(trusted_worktrees) = TrustedWorktrees::try_get_global(cx) { trusted_worktrees.update(cx, |trusted_worktrees, _| { trusted_worktrees.clear_trusted_paths() }); let db = WorkspaceDb::global(cx); cx.spawn(async move |_, cx| { if db.clear_trusted_worktrees().await.log_err().is_some() { cx.update(|cx| reload(cx)); } }) .detach(); } }), ) .on_action(cx.listener( |workspace: &mut Workspace, _: &ReopenClosedItem, window, cx| { workspace.reopen_closed_item(window, cx).detach(); }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &ResetActiveDockSize, window, cx| { for dock in workspace.all_docks() { if dock.focus_handle(cx).contains_focused(window, cx) { let panel = dock.read(cx).active_panel().cloned(); if let Some(panel) = panel { dock.update(cx, |dock, cx| { dock.set_panel_size_state( panel.as_ref(), dock::PanelSizeState::default(), cx, ); }); } return; } } }, )) .on_action(cx.listener( |workspace: &mut Workspace, _: &ResetOpenDocksSize, _window, cx| { for dock in workspace.all_docks() { let panel = dock.read(cx).visible_panel().cloned(); if let Some(panel) = panel { dock.update(cx, |dock, cx| { dock.set_panel_size_state( panel.as_ref(), dock::PanelSizeState::default(), cx, ); }); } } }, )) .on_action(cx.listener( |workspace: &mut Workspace, act: &IncreaseActiveDockSize, window, cx| { adjust_active_dock_size_by_px( px_with_ui_font_fallback(act.px, cx), workspace, window, cx, ); }, )) .on_action(cx.listener( |workspace: &mut Workspace, act: &DecreaseActiveDockSize, window, cx| { adjust_active_dock_size_by_px( px_with_ui_font_fallback(act.px, cx) * -1., workspace, window, cx, ); }, )) .on_action(cx.listener( |workspace: &mut Workspace, act: &IncreaseOpenDocksSize, window, cx| { adjust_open_docks_size_by_px( px_with_ui_font_fallback(act.px, cx), workspace, window, cx, ); }, )) .on_action(cx.listener( |workspace: &mut Workspace, act: &DecreaseOpenDocksSize, window, cx| { adjust_open_docks_size_by_px( px_with_ui_font_fallback(act.px, cx) * -1., workspace, window, cx, ); }, )) .on_action(cx.listener(Workspace::toggle_centered_layout)) .on_action(cx.listener( |workspace: &mut Workspace, action: &pane::ActivateNextItem, window, cx| { if let Some(active_dock) = workspace.active_dock(window, cx) { let dock = active_dock.read(cx); if let Some(active_panel) = dock.active_panel() { if active_panel.pane(cx).is_none() { let mut recent_pane: Option> = None; let mut recent_timestamp = 0; for pane_handle in workspace.panes() { let pane = pane_handle.read(cx); for entry in pane.activation_history() { if entry.timestamp > recent_timestamp { recent_timestamp = entry.timestamp; recent_pane = Some(pane_handle.clone()); } } } if let Some(pane) = recent_pane { let wrap_around = action.wrap_around; pane.update(cx, |pane, cx| { let current_index = pane.active_item_index(); let items_len = pane.items_len(); if items_len > 0 { let next_index = if current_index + 1 < items_len { current_index + 1 } else if wrap_around { 0 } else { return; }; pane.activate_item( next_index, false, false, window, cx, ); } }); return; } } } } cx.propagate(); }, )) .on_action(cx.listener( |workspace: &mut Workspace, action: &pane::ActivatePreviousItem, window, cx| { if let Some(active_dock) = workspace.active_dock(window, cx) { let dock = active_dock.read(cx); if let Some(active_panel) = dock.active_panel() { if active_panel.pane(cx).is_none() { let mut recent_pane: Option> = None; let mut recent_timestamp = 0; for pane_handle in workspace.panes() { let pane = pane_handle.read(cx); for entry in pane.activation_history() { if entry.timestamp > recent_timestamp { recent_timestamp = entry.timestamp; recent_pane = Some(pane_handle.clone()); } } } if let Some(pane) = recent_pane { let wrap_around = action.wrap_around; pane.update(cx, |pane, cx| { let current_index = pane.active_item_index(); let items_len = pane.items_len(); if items_len > 0 { let prev_index = if current_index > 0 { current_index - 1 } else if wrap_around { items_len.saturating_sub(1) } else { return; }; pane.activate_item( prev_index, false, false, window, cx, ); } }); return; } } } } cx.propagate(); }, )) .on_action(cx.listener( |workspace: &mut Workspace, action: &pane::CloseActiveItem, window, cx| { if let Some(active_dock) = workspace.active_dock(window, cx) { let dock = active_dock.read(cx); if let Some(active_panel) = dock.active_panel() { if active_panel.pane(cx).is_none() { let active_pane = workspace.active_pane().clone(); active_pane.update(cx, |pane, cx| { pane.close_active_item(action, window, cx) .detach_and_log_err(cx); }); return; } } } cx.propagate(); }, )) .on_action( cx.listener(|workspace, _: &ToggleReadOnlyFile, window, cx| { let pane = workspace.active_pane().clone(); if let Some(item) = pane.read(cx).active_item() { item.toggle_read_only(window, cx); } }), ) .on_action(cx.listener(|workspace, _: &FocusCenterPane, window, cx| { workspace.focus_center_pane(window, cx); })) .on_action(cx.listener(Workspace::cancel)) } #[cfg(any(test, feature = "test-support"))] pub fn set_random_database_id(&mut self) { self.database_id = Some(WorkspaceId(Uuid::new_v4().as_u64_pair().0 as i64)); } #[cfg(any(test, feature = "test-support"))] pub(crate) fn test_new( project: Entity, window: &mut Window, cx: &mut Context, ) -> Self { use node_runtime::NodeRuntime; use session::Session; let client = project.read(cx).client(); let user_store = project.read(cx).user_store(); let workspace_store = cx.new(|cx| WorkspaceStore::new(client.clone(), cx)); let session = cx.new(|cx| AppSession::new(Session::test(), cx)); window.activate_window(); let app_state = Arc::new(AppState { languages: project.read(cx).languages().clone(), workspace_store, client, user_store, fs: project.read(cx).fs().clone(), build_window_options: |_, _| Default::default(), node_runtime: NodeRuntime::unavailable(), session, }); let workspace = Self::new(Default::default(), project, app_state, window, cx); workspace .active_pane .update(cx, |pane, cx| window.focus(&pane.focus_handle(cx), cx)); workspace } pub fn register_action( &mut self, callback: impl Fn(&mut Self, &A, &mut Window, &mut Context) + 'static, ) -> &mut Self { let callback = Arc::new(callback); self.workspace_actions.push(Box::new(move |div, _, _, cx| { let callback = callback.clone(); div.on_action(cx.listener(move |workspace, event, window, cx| { (callback)(workspace, event, window, cx) })) })); self } pub fn register_action_renderer( &mut self, callback: impl Fn(Div, &Workspace, &mut Window, &mut Context) -> Div + 'static, ) -> &mut Self { self.workspace_actions.push(Box::new(callback)); self } fn add_workspace_actions_listeners( &self, mut div: Div, window: &mut Window, cx: &mut Context, ) -> Div { for action in self.workspace_actions.iter() { div = (action)(div, self, window, cx) } div } pub fn has_active_modal(&self, _: &mut Window, cx: &mut App) -> bool { self.modal_layer.read(cx).has_active_modal() } pub fn is_active_modal_command_palette(&self, cx: &mut App) -> bool { self.modal_layer .read(cx) .is_active_modal_command_palette(cx) } pub fn active_modal(&self, cx: &App) -> Option> { self.modal_layer.read(cx).active_modal() } /// Toggles a modal of type `V`. If a modal of the same type is currently active, /// it will be hidden. If a different modal is active, it will be replaced with the new one. /// If no modal is active, the new modal will be shown. /// /// If closing the current modal fails (e.g., due to `on_before_dismiss` returning /// `DismissDecision::Dismiss(false)` or `DismissDecision::Pending`), the new modal /// will not be shown. pub fn toggle_modal(&mut self, window: &mut Window, cx: &mut App, build: B) where B: FnOnce(&mut Window, &mut Context) -> V, { self.modal_layer.update(cx, |modal_layer, cx| { modal_layer.toggle_modal(window, cx, build) }) } pub fn hide_modal(&mut self, window: &mut Window, cx: &mut App) -> bool { self.modal_layer .update(cx, |modal_layer, cx| modal_layer.hide_modal(window, cx)) } pub fn toggle_status_toast(&mut self, entity: Entity, cx: &mut App) { self.toast_layer .update(cx, |toast_layer, cx| toast_layer.toggle_toast(cx, entity)) } pub fn toggle_centered_layout( &mut self, _: &ToggleCenteredLayout, _: &mut Window, cx: &mut Context, ) { self.centered_layout = !self.centered_layout; if let Some(database_id) = self.database_id() { let db = WorkspaceDb::global(cx); let centered_layout = self.centered_layout; cx.background_spawn(async move { db.set_centered_layout(database_id, centered_layout).await }) .detach_and_log_err(cx); } cx.notify(); } fn adjust_padding(padding: Option) -> f32 { padding .unwrap_or(CenteredPaddingSettings::default().0) .clamp( CenteredPaddingSettings::MIN_PADDING, CenteredPaddingSettings::MAX_PADDING, ) } fn render_dock( &self, position: DockPosition, dock: &Entity, window: &mut Window, cx: &mut App, ) -> Option
{ if self.zoomed_position == Some(position) { return None; } let leader_border = dock.read(cx).active_panel().and_then(|panel| { let pane = panel.pane(cx)?; let follower_states = &self.follower_states; leader_border_for_pane(follower_states, &pane, window, cx) }); let mut container = div() .flex() .overflow_hidden() .flex_none() .child(dock.clone()) .children(leader_border); // Apply sizing only when the dock is open. When closed the dock is still // included in the element tree so its focus handle remains mounted — without // this, toggle_panel_focus cannot focus the panel when the dock is closed. let dock = dock.read(cx); if let Some(panel) = dock.visible_panel() { let size_state = dock.stored_panel_size_state(panel.as_ref()); if position.axis() == Axis::Horizontal { let use_flexible = panel.has_flexible_size(window, cx); let flex_grow = if use_flexible { size_state .and_then(|state| state.flex) .or_else(|| self.default_dock_flex(position)) } else { None }; if let Some(grow) = flex_grow { let grow = grow.max(0.001); let style = container.style(); style.flex_grow = Some(grow); style.flex_shrink = Some(1.0); style.flex_basis = Some(relative(0.).into()); } else { let size = size_state .and_then(|state| state.size) .unwrap_or_else(|| panel.default_size(window, cx)); container = container.w(size); } } else { let size = size_state .and_then(|state| state.size) .unwrap_or_else(|| panel.default_size(window, cx)); container = container.h(size); } } Some(container) } pub fn for_window(window: &Window, cx: &App) -> Option> { window .root::() .flatten() .map(|multi_workspace| multi_workspace.read(cx).workspace().clone()) } pub fn zoomed_item(&self) -> Option<&AnyWeakView> { self.zoomed.as_ref() } pub fn activate_next_window(&mut self, cx: &mut Context) { let Some(current_window_id) = cx.active_window().map(|a| a.window_id()) else { return; }; let windows = cx.windows(); let next_window = SystemWindowTabController::get_next_tab_group_window(cx, current_window_id).or_else( || { windows .iter() .cycle() .skip_while(|window| window.window_id() != current_window_id) .nth(1) }, ); if let Some(window) = next_window { window .update(cx, |_, window, _| window.activate_window()) .ok(); } } pub fn activate_previous_window(&mut self, cx: &mut Context) { let Some(current_window_id) = cx.active_window().map(|a| a.window_id()) else { return; }; let windows = cx.windows(); let prev_window = SystemWindowTabController::get_prev_tab_group_window(cx, current_window_id).or_else( || { windows .iter() .rev() .cycle() .skip_while(|window| window.window_id() != current_window_id) .nth(1) }, ); if let Some(window) = prev_window { window .update(cx, |_, window, _| window.activate_window()) .ok(); } } pub fn cancel(&mut self, _: &menu::Cancel, window: &mut Window, cx: &mut Context) { if cx.stop_active_drag(window) { } else if let Some((notification_id, _)) = self.notifications.pop() { dismiss_app_notification(¬ification_id, cx); } else { cx.propagate(); } } fn resize_dock( &mut self, dock_pos: DockPosition, new_size: Pixels, window: &mut Window, cx: &mut Context, ) { match dock_pos { DockPosition::Left => self.resize_left_dock(new_size, window, cx), DockPosition::Right => self.resize_right_dock(new_size, window, cx), DockPosition::Bottom => self.resize_bottom_dock(new_size, window, cx), } } fn resize_left_dock(&mut self, new_size: Pixels, window: &mut Window, cx: &mut App) { let workspace_width = self.bounds.size.width; let mut size = new_size.min(workspace_width - RESIZE_HANDLE_SIZE); self.right_dock.read_with(cx, |right_dock, cx| { let right_dock_size = right_dock .stored_active_panel_size(window, cx) .unwrap_or(Pixels::ZERO); if right_dock_size + size > workspace_width { size = workspace_width - right_dock_size } }); let flex_grow = self.dock_flex_for_size(DockPosition::Left, size, window, cx); self.left_dock.update(cx, |left_dock, cx| { if WorkspaceSettings::get_global(cx) .resize_all_panels_in_dock .contains(&DockPosition::Left) { left_dock.resize_all_panels(Some(size), flex_grow, window, cx); } else { left_dock.resize_active_panel(Some(size), flex_grow, window, cx); } }); } fn resize_right_dock(&mut self, new_size: Pixels, window: &mut Window, cx: &mut App) { let workspace_width = self.bounds.size.width; let mut size = new_size.min(workspace_width - RESIZE_HANDLE_SIZE); self.left_dock.read_with(cx, |left_dock, cx| { let left_dock_size = left_dock .stored_active_panel_size(window, cx) .unwrap_or(Pixels::ZERO); if left_dock_size + size > workspace_width { size = workspace_width - left_dock_size } }); let flex_grow = self.dock_flex_for_size(DockPosition::Right, size, window, cx); self.right_dock.update(cx, |right_dock, cx| { if WorkspaceSettings::get_global(cx) .resize_all_panels_in_dock .contains(&DockPosition::Right) { right_dock.resize_all_panels(Some(size), flex_grow, window, cx); } else { right_dock.resize_active_panel(Some(size), flex_grow, window, cx); } }); } fn resize_bottom_dock(&mut self, new_size: Pixels, window: &mut Window, cx: &mut App) { let size = new_size.min(self.bounds.bottom() - RESIZE_HANDLE_SIZE - self.bounds.top()); self.bottom_dock.update(cx, |bottom_dock, cx| { if WorkspaceSettings::get_global(cx) .resize_all_panels_in_dock .contains(&DockPosition::Bottom) { bottom_dock.resize_all_panels(Some(size), None, window, cx); } else { bottom_dock.resize_active_panel(Some(size), None, window, cx); } }); } fn toggle_edit_predictions_all_files( &mut self, _: &ToggleEditPrediction, _window: &mut Window, cx: &mut Context, ) { let fs = self.project().read(cx).fs().clone(); let show_edit_predictions = all_language_settings(None, cx).show_edit_predictions(None, cx); update_settings_file(fs, cx, move |file, _| { file.project.all_languages.defaults.show_edit_predictions = Some(!show_edit_predictions) }); } fn toggle_theme_mode(&mut self, _: &ToggleMode, _window: &mut Window, cx: &mut Context) { let current_mode = ThemeSettings::get_global(cx).theme.mode(); let next_mode = match current_mode { Some(theme_settings::ThemeAppearanceMode::Light) => { theme_settings::ThemeAppearanceMode::Dark } Some(theme_settings::ThemeAppearanceMode::Dark) => { theme_settings::ThemeAppearanceMode::Light } Some(theme_settings::ThemeAppearanceMode::System) | None => { match cx.theme().appearance() { theme::Appearance::Light => theme_settings::ThemeAppearanceMode::Dark, theme::Appearance::Dark => theme_settings::ThemeAppearanceMode::Light, } } }; let fs = self.project().read(cx).fs().clone(); settings::update_settings_file(fs, cx, move |settings, _cx| { theme_settings::set_mode(settings, next_mode); }); } pub fn show_worktree_trust_security_modal( &mut self, toggle: bool, window: &mut Window, cx: &mut Context, ) { if let Some(security_modal) = self.active_modal::(cx) { if toggle { security_modal.update(cx, |security_modal, cx| { security_modal.dismiss(cx); }) } else { security_modal.update(cx, |security_modal, cx| { security_modal.refresh_restricted_paths(cx); }); } } else { let has_restricted_worktrees = TrustedWorktrees::try_get_global(cx) .map(|trusted_worktrees| { trusted_worktrees .read(cx) .has_restricted_worktrees(&self.project().read(cx).worktree_store(), cx) }) .unwrap_or(false); if has_restricted_worktrees { let project = self.project().read(cx); let remote_host = project .remote_connection_options(cx) .map(RemoteHostLocation::from); let worktree_store = project.worktree_store().downgrade(); self.toggle_modal(window, cx, |_, cx| { SecurityModal::new(worktree_store, remote_host, cx) }); } } } } pub trait AnyActiveCall { fn entity(&self) -> AnyEntity; fn is_in_room(&self, _: &App) -> bool; fn room_id(&self, _: &App) -> Option; fn channel_id(&self, _: &App) -> Option; fn hang_up(&self, _: &mut App) -> Task>; fn unshare_project(&self, _: Entity, _: &mut App) -> Result<()>; fn remote_participant_for_peer_id(&self, _: PeerId, _: &App) -> Option; fn is_sharing_project(&self, _: &App) -> bool; fn has_remote_participants(&self, _: &App) -> bool; fn local_participant_is_guest(&self, _: &App) -> bool; fn client(&self, _: &App) -> Arc; fn share_on_join(&self, _: &App) -> bool; fn join_channel(&self, _: ChannelId, _: &mut App) -> Task>; fn room_update_completed(&self, _: &mut App) -> Task<()>; fn most_active_project(&self, _: &App) -> Option<(u64, u64)>; fn share_project(&self, _: Entity, _: &mut App) -> Task>; fn join_project( &self, _: u64, _: Arc, _: Arc, _: &mut App, ) -> Task>>; fn peer_id_for_user_in_room(&self, _: u64, _: &App) -> Option; fn subscribe( &self, _: &mut Window, _: &mut Context, _: Box)>, ) -> Subscription; fn create_shared_screen( &self, _: PeerId, _: &Entity, _: &mut Window, _: &mut App, ) -> Option>; } #[derive(Clone)] pub struct GlobalAnyActiveCall(pub Arc); impl Global for GlobalAnyActiveCall {} impl GlobalAnyActiveCall { pub(crate) fn try_global(cx: &App) -> Option<&Self> { cx.try_global() } pub(crate) fn global(cx: &App) -> &Self { cx.global() } } /// Workspace-local view of a remote participant's location. #[derive(Clone, Copy, Debug, PartialEq, Eq)] pub enum ParticipantLocation { SharedProject { project_id: u64 }, UnsharedProject, External, } impl ParticipantLocation { pub fn from_proto(location: Option) -> Result { match location .and_then(|l| l.variant) .context("participant location was not provided")? { proto::participant_location::Variant::SharedProject(project) => { Ok(Self::SharedProject { project_id: project.id, }) } proto::participant_location::Variant::UnsharedProject(_) => Ok(Self::UnsharedProject), proto::participant_location::Variant::External(_) => Ok(Self::External), } } } /// Workspace-local view of a remote collaborator's state. /// This is the subset of `call::RemoteParticipant` that workspace needs. #[derive(Clone)] pub struct RemoteCollaborator { pub user: Arc, pub peer_id: PeerId, pub location: ParticipantLocation, pub participant_index: ParticipantIndex, } pub enum ActiveCallEvent { ParticipantLocationChanged { participant_id: PeerId }, RemoteVideoTracksChanged { participant_id: PeerId }, } fn leader_border_for_pane( follower_states: &HashMap, pane: &Entity, _: &Window, cx: &App, ) -> Option
{ let (leader_id, _follower_state) = follower_states.iter().find_map(|(leader_id, state)| { if state.pane() == pane { Some((*leader_id, state)) } else { None } })?; let mut leader_color = match leader_id { CollaboratorId::PeerId(leader_peer_id) => { let leader = GlobalAnyActiveCall::try_global(cx)? .0 .remote_participant_for_peer_id(leader_peer_id, cx)?; cx.theme() .players() .color_for_participant(leader.participant_index.0) .cursor } CollaboratorId::Agent => cx.theme().players().agent().cursor, }; leader_color.fade_out(0.3); Some( div() .absolute() .size_full() .left_0() .top_0() .border_2() .border_color(leader_color), ) } fn window_bounds_env_override() -> Option> { ZED_WINDOW_POSITION .zip(*ZED_WINDOW_SIZE) .map(|(position, size)| Bounds { origin: position, size, }) } fn open_items( serialized_workspace: Option, mut project_paths_to_open: Vec<(PathBuf, Option)>, window: &mut Window, cx: &mut Context, ) -> impl 'static + Future>>>>> + use<> { let restored_items = serialized_workspace.map(|serialized_workspace| { Workspace::load_workspace( serialized_workspace, project_paths_to_open .iter() .map(|(_, project_path)| project_path) .cloned() .collect(), window, cx, ) }); cx.spawn_in(window, async move |workspace, cx| { let mut opened_items = Vec::with_capacity(project_paths_to_open.len()); if let Some(restored_items) = restored_items { let restored_items = restored_items.await?; let restored_project_paths = restored_items .iter() .filter_map(|item| { cx.update(|_, cx| item.as_ref()?.project_path(cx)) .ok() .flatten() }) .collect::>(); for restored_item in restored_items { opened_items.push(restored_item.map(Ok)); } project_paths_to_open .iter_mut() .for_each(|(_, project_path)| { if let Some(project_path_to_open) = project_path && restored_project_paths.contains(project_path_to_open) { *project_path = None; } }); } else { for _ in 0..project_paths_to_open.len() { opened_items.push(None); } } assert!(opened_items.len() == project_paths_to_open.len()); let tasks = project_paths_to_open .into_iter() .enumerate() .map(|(ix, (abs_path, project_path))| { let workspace = workspace.clone(); cx.spawn(async move |cx| { let file_project_path = project_path?; let abs_path_task = workspace.update(cx, |workspace, cx| { workspace.project().update(cx, |project, cx| { project.resolve_abs_path(abs_path.to_string_lossy().as_ref(), cx) }) }); // We only want to open file paths here. If one of the items // here is a directory, it was already opened further above // with a `find_or_create_worktree`. if let Ok(task) = abs_path_task && task.await.is_none_or(|p| p.is_file()) { return Some(( ix, workspace .update_in(cx, |workspace, window, cx| { workspace.open_path( file_project_path, None, true, window, cx, ) }) .log_err()? .await, )); } None }) }); let tasks = tasks.collect::>(); let tasks = futures::future::join_all(tasks); for (ix, path_open_result) in tasks.await.into_iter().flatten() { opened_items[ix] = Some(path_open_result); } Ok(opened_items) }) } #[derive(Clone)] enum ActivateInDirectionTarget { Pane(Entity), Dock(Entity), Sidebar(FocusHandle), } fn notify_if_database_failed(window: WindowHandle, cx: &mut AsyncApp) { window .update(cx, |multi_workspace, _, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { if (*db::ALL_FILE_DB_FAILED).load(std::sync::atomic::Ordering::Acquire) { struct DatabaseFailedNotification; workspace.show_notification( NotificationId::unique::(), cx, |cx| { cx.new(|cx| { MessageNotification::new("Failed to load the database file.", cx) .primary_message("File an Issue") .primary_icon(IconName::Plus) .primary_on_click(|window, cx| { window.dispatch_action(Box::new(FileBugReport), cx) }) }) }, ); } }); }) .log_err(); } fn px_with_ui_font_fallback(val: u32, cx: &Context) -> Pixels { if val == 0 { ThemeSettings::get_global(cx).ui_font_size(cx) } else { px(val as f32) } } fn adjust_active_dock_size_by_px( px: Pixels, workspace: &mut Workspace, window: &mut Window, cx: &mut Context, ) { let Some(active_dock) = workspace .all_docks() .into_iter() .find(|dock| dock.focus_handle(cx).contains_focused(window, cx)) else { return; }; let dock = active_dock.read(cx); let Some(panel_size) = workspace.dock_size(&dock, window, cx) else { return; }; workspace.resize_dock(dock.position(), panel_size + px, window, cx); } fn adjust_open_docks_size_by_px( px: Pixels, workspace: &mut Workspace, window: &mut Window, cx: &mut Context, ) { let docks = workspace .all_docks() .into_iter() .filter_map(|dock_entity| { let dock = dock_entity.read(cx); if dock.is_open() { let dock_pos = dock.position(); let panel_size = workspace.dock_size(&dock, window, cx)?; Some((dock_pos, panel_size + px)) } else { None } }) .collect::>(); for (position, new_size) in docks { workspace.resize_dock(position, new_size, window, cx); } } impl Focusable for Workspace { fn focus_handle(&self, cx: &App) -> FocusHandle { self.active_pane.focus_handle(cx) } } #[derive(Clone)] struct DraggedDock(DockPosition); impl Render for DraggedDock { fn render(&mut self, _window: &mut Window, _cx: &mut Context) -> impl IntoElement { gpui::Empty } } impl Render for Workspace { fn render(&mut self, window: &mut Window, cx: &mut Context) -> impl IntoElement { static FIRST_PAINT: AtomicBool = AtomicBool::new(true); if FIRST_PAINT.swap(false, std::sync::atomic::Ordering::Relaxed) { log::info!("Rendered first frame"); } let centered_layout = self.centered_layout && self.center.panes().len() == 1 && self.active_item(cx).is_some(); let render_padding = |size| { (size > 0.0).then(|| { div() .h_full() .w(relative(size)) .bg(cx.theme().colors().editor_background) .border_color(cx.theme().colors().pane_group_border) }) }; let paddings = if centered_layout { let settings = WorkspaceSettings::get_global(cx).centered_layout; ( render_padding(Self::adjust_padding( settings.left_padding.map(|padding| padding.0), )), render_padding(Self::adjust_padding( settings.right_padding.map(|padding| padding.0), )), ) } else { (None, None) }; let ui_font = theme_settings::setup_ui_font(window, cx); let theme = cx.theme().clone(); let colors = theme.colors(); let notification_entities = self .notifications .iter() .map(|(_, notification)| notification.entity_id()) .collect::>(); let bottom_dock_layout = WorkspaceSettings::get_global(cx).bottom_dock_layout; div() .relative() .size_full() .flex() .flex_col() .font(ui_font) .gap_0() .justify_start() .items_start() .text_color(colors.text) .overflow_hidden() .children(self.titlebar_item.clone()) .on_modifiers_changed(move |_, _, cx| { for &id in ¬ification_entities { cx.notify(id); } }) .child( div() .size_full() .relative() .flex_1() .flex() .flex_col() .child( div() .id("workspace") .bg(colors.background) .relative() .flex_1() .w_full() .flex() .flex_col() .overflow_hidden() .border_t_1() .border_b_1() .border_color(colors.border) .child({ let this = cx.entity(); canvas( move |bounds, window, cx| { this.update(cx, |this, cx| { let bounds_changed = this.bounds != bounds; this.bounds = bounds; if bounds_changed { this.left_dock.update(cx, |dock, cx| { dock.clamp_panel_size( bounds.size.width, window, cx, ) }); this.right_dock.update(cx, |dock, cx| { dock.clamp_panel_size( bounds.size.width, window, cx, ) }); this.bottom_dock.update(cx, |dock, cx| { dock.clamp_panel_size( bounds.size.height, window, cx, ) }); } }) }, |_, _, _, _| {}, ) .absolute() .size_full() }) .when(self.zoomed.is_none(), |this| { this.on_drag_move(cx.listener( move |workspace, e: &DragMoveEvent, window, cx| { if workspace.previous_dock_drag_coordinates != Some(e.event.position) { workspace.previous_dock_drag_coordinates = Some(e.event.position); match e.drag(cx).0 { DockPosition::Left => { workspace.resize_left_dock( e.event.position.x - workspace.bounds.left(), window, cx, ); } DockPosition::Right => { workspace.resize_right_dock( workspace.bounds.right() - e.event.position.x, window, cx, ); } DockPosition::Bottom => { workspace.resize_bottom_dock( workspace.bounds.bottom() - e.event.position.y, window, cx, ); } }; workspace.serialize_workspace(window, cx); } }, )) }) .child({ match bottom_dock_layout { BottomDockLayout::Full => div() .flex() .flex_col() .h_full() .child( div() .flex() .flex_row() .flex_1() .overflow_hidden() .children(self.render_dock( DockPosition::Left, &self.left_dock, window, cx, )) .child( div() .flex() .flex_col() .flex_1() .overflow_hidden() .child( h_flex() .flex_1() .when_some( paddings.0, |this, p| { this.child( p.border_r_1(), ) }, ) .child(self.center.render( self.zoomed.as_ref(), &PaneRenderContext { follower_states: &self.follower_states, active_call: self.active_call(), active_pane: &self.active_pane, app_state: &self.app_state, project: &self.project, workspace: &self.weak_self, }, window, cx, )) .when_some( paddings.1, |this, p| { this.child( p.border_l_1(), ) }, ), ), ) .children(self.render_dock( DockPosition::Right, &self.right_dock, window, cx, )), ) .child(div().w_full().children(self.render_dock( DockPosition::Bottom, &self.bottom_dock, window, cx ))), BottomDockLayout::LeftAligned => div() .flex() .flex_row() .h_full() .child( div() .flex() .flex_col() .flex_1() .h_full() .child( div() .flex() .flex_row() .flex_1() .children(self.render_dock(DockPosition::Left, &self.left_dock, window, cx)) .child( div() .flex() .flex_col() .flex_1() .overflow_hidden() .child( h_flex() .flex_1() .when_some(paddings.0, |this, p| this.child(p.border_r_1())) .child(self.center.render( self.zoomed.as_ref(), &PaneRenderContext { follower_states: &self.follower_states, active_call: self.active_call(), active_pane: &self.active_pane, app_state: &self.app_state, project: &self.project, workspace: &self.weak_self, }, window, cx, )) .when_some(paddings.1, |this, p| this.child(p.border_l_1())), ) ) ) .child( div() .w_full() .children(self.render_dock(DockPosition::Bottom, &self.bottom_dock, window, cx)) ), ) .children(self.render_dock( DockPosition::Right, &self.right_dock, window, cx, )), BottomDockLayout::RightAligned => div() .flex() .flex_row() .h_full() .children(self.render_dock( DockPosition::Left, &self.left_dock, window, cx, )) .child( div() .flex() .flex_col() .flex_1() .h_full() .child( div() .flex() .flex_row() .flex_1() .child( div() .flex() .flex_col() .flex_1() .overflow_hidden() .child( h_flex() .flex_1() .when_some(paddings.0, |this, p| this.child(p.border_r_1())) .child(self.center.render( self.zoomed.as_ref(), &PaneRenderContext { follower_states: &self.follower_states, active_call: self.active_call(), active_pane: &self.active_pane, app_state: &self.app_state, project: &self.project, workspace: &self.weak_self, }, window, cx, )) .when_some(paddings.1, |this, p| this.child(p.border_l_1())), ) ) .children(self.render_dock(DockPosition::Right, &self.right_dock, window, cx)) ) .child( div() .w_full() .children(self.render_dock(DockPosition::Bottom, &self.bottom_dock, window, cx)) ), ), BottomDockLayout::Contained => div() .flex() .flex_row() .h_full() .children(self.render_dock( DockPosition::Left, &self.left_dock, window, cx, )) .child( div() .flex() .flex_col() .flex_1() .overflow_hidden() .child( h_flex() .flex_1() .when_some(paddings.0, |this, p| { this.child(p.border_r_1()) }) .child(self.center.render( self.zoomed.as_ref(), &PaneRenderContext { follower_states: &self.follower_states, active_call: self.active_call(), active_pane: &self.active_pane, app_state: &self.app_state, project: &self.project, workspace: &self.weak_self, }, window, cx, )) .when_some(paddings.1, |this, p| { this.child(p.border_l_1()) }), ) .children(self.render_dock( DockPosition::Bottom, &self.bottom_dock, window, cx, )), ) .children(self.render_dock( DockPosition::Right, &self.right_dock, window, cx, )), } }) .children(self.zoomed.as_ref().and_then(|view| { let zoomed_view = view.upgrade()?; let div = div() .occlude() .absolute() .overflow_hidden() .border_color(colors.border) .bg(colors.background) .child(zoomed_view) .inset_0() .shadow_lg(); if !WorkspaceSettings::get_global(cx).zoomed_padding { return Some(div); } Some(match self.zoomed_position { Some(DockPosition::Left) => div.right_2().border_r_1(), Some(DockPosition::Right) => div.left_2().border_l_1(), Some(DockPosition::Bottom) => div.top_2().border_t_1(), None => { div.top_2().bottom_2().left_2().right_2().border_1() } }) })) .children(self.render_notifications(window, cx)), ) .when(self.status_bar_visible(cx), |parent| { parent.child(self.status_bar.clone()) }) .child(self.toast_layer.clone()), ) } } impl WorkspaceStore { pub fn new(client: Arc, cx: &mut Context) -> Self { Self { workspaces: Default::default(), _subscriptions: vec![ client.add_request_handler(cx.weak_entity(), Self::handle_follow), client.add_message_handler(cx.weak_entity(), Self::handle_update_followers), ], client, } } pub fn update_followers( &self, project_id: Option, update: proto::update_followers::Variant, cx: &App, ) -> Option<()> { let active_call = GlobalAnyActiveCall::try_global(cx)?; let room_id = active_call.0.room_id(cx)?; self.client .send(proto::UpdateFollowers { room_id, project_id, variant: Some(update), }) .log_err() } pub async fn handle_follow( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result { this.update(&mut cx, |this, cx| { let follower = Follower { project_id: envelope.payload.project_id, peer_id: envelope.original_sender_id()?, }; let mut response = proto::FollowResponse::default(); this.workspaces.retain(|(window_handle, weak_workspace)| { let Some(workspace) = weak_workspace.upgrade() else { return false; }; window_handle .update(cx, |_, window, cx| { workspace.update(cx, |workspace, cx| { let handler_response = workspace.handle_follow(follower.project_id, window, cx); if let Some(active_view) = handler_response.active_view && workspace.project.read(cx).remote_id() == follower.project_id { response.active_view = Some(active_view) } }); }) .is_ok() }); Ok(response) }) } async fn handle_update_followers( this: Entity, envelope: TypedEnvelope, mut cx: AsyncApp, ) -> Result<()> { let leader_id = envelope.original_sender_id()?; let update = envelope.payload; this.update(&mut cx, |this, cx| { this.workspaces.retain(|(window_handle, weak_workspace)| { let Some(workspace) = weak_workspace.upgrade() else { return false; }; window_handle .update(cx, |_, window, cx| { workspace.update(cx, |workspace, cx| { let project_id = workspace.project.read(cx).remote_id(); if update.project_id != project_id && update.project_id.is_some() { return; } workspace.handle_update_followers( leader_id, update.clone(), window, cx, ); }); }) .is_ok() }); Ok(()) }) } pub fn workspaces(&self) -> impl Iterator> { self.workspaces.iter().map(|(_, weak)| weak) } pub fn workspaces_with_windows( &self, ) -> impl Iterator)> { self.workspaces.iter().map(|(window, weak)| (*window, weak)) } } impl ViewId { pub(crate) fn from_proto(message: proto::ViewId) -> Result { Ok(Self { creator: message .creator .map(CollaboratorId::PeerId) .context("creator is missing")?, id: message.id, }) } pub(crate) fn to_proto(self) -> Option { if let CollaboratorId::PeerId(peer_id) = self.creator { Some(proto::ViewId { creator: Some(peer_id), id: self.id, }) } else { None } } } impl FollowerState { fn pane(&self) -> &Entity { self.dock_pane.as_ref().unwrap_or(&self.center_pane) } } pub trait WorkspaceHandle { fn file_project_paths(&self, cx: &App) -> Vec; } impl WorkspaceHandle for Entity { fn file_project_paths(&self, cx: &App) -> Vec { self.read(cx) .worktrees(cx) .flat_map(|worktree| { let worktree_id = worktree.read(cx).id(); worktree.read(cx).files(true, 0).map(move |f| ProjectPath { worktree_id, path: f.path.clone(), }) }) .collect::>() } } pub async fn last_opened_workspace_location( db: &WorkspaceDb, fs: &dyn fs::Fs, ) -> Option<(WorkspaceId, SerializedWorkspaceLocation, PathList)> { db.last_workspace(fs) .await .log_err() .flatten() .map(|(id, location, paths, _timestamp)| (id, location, paths)) } pub async fn last_session_workspace_locations( db: &WorkspaceDb, last_session_id: &str, last_session_window_stack: Option>, fs: &dyn fs::Fs, ) -> Option> { db.last_session_workspace_locations(last_session_id, last_session_window_stack, fs) .await .log_err() } pub async fn restore_multiworkspace( multi_workspace: SerializedMultiWorkspace, app_state: Arc, cx: &mut AsyncApp, ) -> anyhow::Result> { let SerializedMultiWorkspace { active_workspace, state, } = multi_workspace; let MultiWorkspaceState { sidebar_open, project_group_keys, sidebar_state, .. } = state; let workspace_result = if active_workspace.paths.is_empty() { cx.update(|cx| { open_workspace_by_id(active_workspace.workspace_id, app_state.clone(), None, cx) }) .await } else { cx.update(|cx| { Workspace::new_local( active_workspace.paths.paths().to_vec(), app_state.clone(), None, None, None, OpenMode::Activate, cx, ) }) .await .map(|result| result.window) }; let window_handle = match workspace_result { Ok(handle) => handle, Err(err) => { log::error!("Failed to restore active workspace: {err:#}"); // Try each project group's paths as a fallback. let mut fallback_handle = None; for key in &project_group_keys { let key: ProjectGroupKey = key.clone().into(); let paths = key.path_list().paths().to_vec(); match cx .update(|cx| { Workspace::new_local( paths, app_state.clone(), None, None, None, OpenMode::Activate, cx, ) }) .await { Ok(OpenResult { window, .. }) => { fallback_handle = Some(window); break; } Err(fallback_err) => { log::error!("Fallback project group also failed: {fallback_err:#}"); } } } fallback_handle.ok_or(err)? } }; if !project_group_keys.is_empty() { let fs = app_state.fs.clone(); // Resolve linked worktree paths to their main repo paths so // stale keys from previous sessions get normalized and deduped. let mut resolved_keys: Vec = Vec::new(); for key in project_group_keys.into_iter().map(ProjectGroupKey::from) { if key.path_list().paths().is_empty() { continue; } let mut resolved_paths = Vec::new(); for path in key.path_list().paths() { if let Some(common_dir) = project::discover_root_repo_common_dir(path, fs.as_ref()).await { let main_path = common_dir.parent().unwrap_or(&common_dir); resolved_paths.push(main_path.to_path_buf()); } else { resolved_paths.push(path.to_path_buf()); } } let resolved = ProjectGroupKey::new(key.host(), PathList::new(&resolved_paths)); if !resolved_keys.contains(&resolved) { resolved_keys.push(resolved); } } window_handle .update(cx, |multi_workspace, _window, _cx| { multi_workspace.restore_project_group_keys(resolved_keys); }) .ok(); } if sidebar_open { window_handle .update(cx, |multi_workspace, _, cx| { multi_workspace.open_sidebar(cx); }) .ok(); } if let Some(sidebar_state) = sidebar_state { window_handle .update(cx, |multi_workspace, window, cx| { if let Some(sidebar) = multi_workspace.sidebar() { sidebar.restore_serialized_state(&sidebar_state, window, cx); } multi_workspace.serialize(cx); }) .ok(); } window_handle .update(cx, |_, window, _cx| { window.activate_window(); }) .ok(); Ok(window_handle) } actions!( collab, [ /// Opens the channel notes for the current call. /// /// Use `collab_panel::OpenSelectedChannelNotes` to open the channel notes for the selected /// channel in the collab panel. /// /// If you want to open a specific channel, use `zed::OpenZedUrl` with a channel notes URL - /// can be copied via "Copy link to section" in the context menu of the channel notes /// buffer. These URLs look like `https://zed.dev/channel/channel-name-CHANNEL_ID/notes`. OpenChannelNotes, /// Mutes your microphone. Mute, /// Deafens yourself (mute both microphone and speakers). Deafen, /// Leaves the current call. LeaveCall, /// Shares the current project with collaborators. ShareProject, /// Shares your screen with collaborators. ScreenShare, /// Copies the current room name and session id for debugging purposes. CopyRoomId, ] ); /// Opens the channel notes for a specific channel by its ID. #[derive(Clone, PartialEq, Deserialize, JsonSchema, Action)] #[action(namespace = collab)] #[serde(deny_unknown_fields)] pub struct OpenChannelNotesById { pub channel_id: u64, } actions!( zed, [ /// Opens the Zed log file. OpenLog, /// Reveals the Zed log file in the system file manager. RevealLogInFileManager ] ); async fn join_channel_internal( channel_id: ChannelId, app_state: &Arc, requesting_window: Option>, requesting_workspace: Option>, active_call: &dyn AnyActiveCall, cx: &mut AsyncApp, ) -> Result { let (should_prompt, already_in_channel) = cx.update(|cx| { if !active_call.is_in_room(cx) { return (false, false); } let already_in_channel = active_call.channel_id(cx) == Some(channel_id); let should_prompt = active_call.is_sharing_project(cx) && active_call.has_remote_participants(cx) && !already_in_channel; (should_prompt, already_in_channel) }); if already_in_channel { let task = cx.update(|cx| { if let Some((project, host)) = active_call.most_active_project(cx) { Some(join_in_room_project(project, host, app_state.clone(), cx)) } else { None } }); if let Some(task) = task { task.await?; } return anyhow::Ok(true); } if should_prompt { if let Some(multi_workspace) = requesting_window { let answer = multi_workspace .update(cx, |_, window, cx| { window.prompt( PromptLevel::Warning, "Do you want to switch channels?", Some("Leaving this call will unshare your current project."), &["Yes, Join Channel", "Cancel"], cx, ) })? .await; if answer == Ok(1) { return Ok(false); } } else { return Ok(false); } } let client = cx.update(|cx| active_call.client(cx)); let mut client_status = client.status(); // this loop will terminate within client::CONNECTION_TIMEOUT seconds. 'outer: loop { let Some(status) = client_status.recv().await else { anyhow::bail!("error connecting"); }; match status { Status::Connecting | Status::Authenticating | Status::Authenticated | Status::Reconnecting | Status::Reauthenticating | Status::Reauthenticated => continue, Status::Connected { .. } => break 'outer, Status::SignedOut | Status::AuthenticationError => { return Err(ErrorCode::SignedOut.into()); } Status::UpgradeRequired => return Err(ErrorCode::UpgradeRequired.into()), Status::ConnectionError | Status::ConnectionLost | Status::ReconnectionError { .. } => { return Err(ErrorCode::Disconnected.into()); } } } let joined = cx .update(|cx| active_call.join_channel(channel_id, cx)) .await?; if !joined { return anyhow::Ok(true); } cx.update(|cx| active_call.room_update_completed(cx)).await; let task = cx.update(|cx| { if let Some((project, host)) = active_call.most_active_project(cx) { return Some(join_in_room_project(project, host, app_state.clone(), cx)); } // If you are the first to join a channel, see if you should share your project. if !active_call.has_remote_participants(cx) && !active_call.local_participant_is_guest(cx) && let Some(workspace) = requesting_workspace.as_ref().and_then(|w| w.upgrade()) { let project = workspace.update(cx, |workspace, cx| { let project = workspace.project.read(cx); if !active_call.share_on_join(cx) { return None; } if (project.is_local() || project.is_via_remote_server()) && project.visible_worktrees(cx).any(|tree| { tree.read(cx) .root_entry() .is_some_and(|entry| entry.is_dir()) }) { Some(workspace.project.clone()) } else { None } }); if let Some(project) = project { let share_task = active_call.share_project(project, cx); return Some(cx.spawn(async move |_cx| -> Result<()> { share_task.await?; Ok(()) })); } } None }); if let Some(task) = task { task.await?; return anyhow::Ok(true); } anyhow::Ok(false) } pub fn join_channel( channel_id: ChannelId, app_state: Arc, requesting_window: Option>, requesting_workspace: Option>, cx: &mut App, ) -> Task> { let active_call = GlobalAnyActiveCall::global(cx).clone(); cx.spawn(async move |cx| { let result = join_channel_internal( channel_id, &app_state, requesting_window, requesting_workspace, &*active_call.0, cx, ) .await; // join channel succeeded, and opened a window if matches!(result, Ok(true)) { return anyhow::Ok(()); } // find an existing workspace to focus and show call controls let mut active_window = requesting_window.or_else(|| activate_any_workspace_window(cx)); if active_window.is_none() { // no open workspaces, make one to show the error in (blergh) let OpenResult { window: window_handle, .. } = cx .update(|cx| { Workspace::new_local( vec![], app_state.clone(), requesting_window, None, None, OpenMode::Activate, cx, ) }) .await?; window_handle .update(cx, |_, window, _cx| { window.activate_window(); }) .ok(); if result.is_ok() { cx.update(|cx| { cx.dispatch_action(&OpenChannelNotes); }); } active_window = Some(window_handle); } if let Err(err) = result { log::error!("failed to join channel: {}", err); if let Some(active_window) = active_window { active_window .update(cx, |_, window, cx| { let detail: SharedString = match err.error_code() { ErrorCode::SignedOut => "Please sign in to continue.".into(), ErrorCode::UpgradeRequired => concat!( "Your are running an unsupported version of Zed. ", "Please update to continue." ) .into(), ErrorCode::NoSuchChannel => concat!( "No matching channel was found. ", "Please check the link and try again." ) .into(), ErrorCode::Forbidden => concat!( "This channel is private, and you do not have access. ", "Please ask someone to add you and try again." ) .into(), ErrorCode::Disconnected => { "Please check your internet connection and try again.".into() } _ => format!("{}\n\nPlease try again.", err).into(), }; window.prompt( PromptLevel::Critical, "Failed to join channel", Some(&detail), &["Ok"], cx, ) })? .await .ok(); } } // return ok, we showed the error to the user. anyhow::Ok(()) }) } pub async fn get_any_active_multi_workspace( app_state: Arc, mut cx: AsyncApp, ) -> anyhow::Result> { // find an existing workspace to focus and show call controls let active_window = activate_any_workspace_window(&mut cx); if active_window.is_none() { cx.update(|cx| { Workspace::new_local( vec![], app_state.clone(), None, None, None, OpenMode::Activate, cx, ) }) .await?; } activate_any_workspace_window(&mut cx).context("could not open zed") } fn activate_any_workspace_window(cx: &mut AsyncApp) -> Option> { cx.update(|cx| { if let Some(workspace_window) = cx .active_window() .and_then(|window| window.downcast::()) { return Some(workspace_window); } for window in cx.windows() { if let Some(workspace_window) = window.downcast::() { workspace_window .update(cx, |_, window, _| window.activate_window()) .ok(); return Some(workspace_window); } } None }) } pub fn local_workspace_windows(cx: &App) -> Vec> { workspace_windows_for_location(&SerializedWorkspaceLocation::Local, cx) } pub fn workspace_windows_for_location( serialized_location: &SerializedWorkspaceLocation, cx: &App, ) -> Vec> { cx.windows() .into_iter() .filter_map(|window| window.downcast::()) .filter(|multi_workspace| { let same_host = |left: &RemoteConnectionOptions, right: &RemoteConnectionOptions| match (left, right) { (RemoteConnectionOptions::Ssh(a), RemoteConnectionOptions::Ssh(b)) => { (&a.host, &a.username, &a.port) == (&b.host, &b.username, &b.port) } (RemoteConnectionOptions::Wsl(a), RemoteConnectionOptions::Wsl(b)) => { // The WSL username is not consistently populated in the workspace location, so ignore it for now. a.distro_name == b.distro_name } (RemoteConnectionOptions::Docker(a), RemoteConnectionOptions::Docker(b)) => { a.container_id == b.container_id } #[cfg(any(test, feature = "test-support"))] (RemoteConnectionOptions::Mock(a), RemoteConnectionOptions::Mock(b)) => { a.id == b.id } _ => false, }; multi_workspace.read(cx).is_ok_and(|multi_workspace| { multi_workspace.workspaces().any(|workspace| { match workspace.read(cx).workspace_location(cx) { WorkspaceLocation::Location(location, _) => { match (&location, serialized_location) { ( SerializedWorkspaceLocation::Local, SerializedWorkspaceLocation::Local, ) => true, ( SerializedWorkspaceLocation::Remote(a), SerializedWorkspaceLocation::Remote(b), ) => same_host(a, b), _ => false, } } _ => false, } }) }) }) .collect() } pub async fn find_existing_workspace( abs_paths: &[PathBuf], open_options: &OpenOptions, location: &SerializedWorkspaceLocation, cx: &mut AsyncApp, ) -> ( Option<(WindowHandle, Entity)>, OpenVisible, ) { let mut existing: Option<(WindowHandle, Entity)> = None; let mut open_visible = OpenVisible::All; let mut best_match = None; cx.update(|cx| { for window in workspace_windows_for_location(location, cx) { if let Ok(multi_workspace) = window.read(cx) { for workspace in multi_workspace.workspaces() { let project = workspace.read(cx).project.read(cx); let m = project.visibility_for_paths( abs_paths, open_options.open_new_workspace == None, cx, ); if m > best_match { existing = Some((window, workspace.clone())); best_match = m; } else if best_match.is_none() && open_options.open_new_workspace == Some(false) { existing = Some((window, workspace.clone())) } } } } }); // With -n, only reuse a window if the path is genuinely contained // within an existing worktree (don't fall back to any arbitrary window). if open_options.open_new_workspace == Some(true) && best_match.is_none() { existing = None; } if open_options.open_new_workspace != Some(true) { let all_paths_are_files = existing .as_ref() .and_then(|(_, target_workspace)| { cx.update(|cx| { let workspace = target_workspace.read(cx); let project = workspace.project.read(cx); let path_style = workspace.path_style(cx); Some(!abs_paths.iter().any(|path| { let path = util::paths::SanitizedPath::new(path); project.worktrees(cx).any(|worktree| { let worktree = worktree.read(cx); let abs_path = worktree.abs_path(); path_style .strip_prefix(path.as_ref(), abs_path.as_ref()) .and_then(|rel| worktree.entry_for_path(&rel)) .is_some_and(|e| e.is_dir()) }) })) }) }) .unwrap_or(false); if open_options.open_new_workspace.is_none() && existing.is_some() && open_options.wait && all_paths_are_files { cx.update(|cx| { let windows = workspace_windows_for_location(location, cx); let window = cx .active_window() .and_then(|window| window.downcast::()) .filter(|window| windows.contains(window)) .or_else(|| windows.into_iter().next()); if let Some(window) = window { if let Ok(multi_workspace) = window.read(cx) { let active_workspace = multi_workspace.workspace().clone(); existing = Some((window, active_workspace)); open_visible = OpenVisible::None; } } }); } } (existing, open_visible) } #[derive(Default, Clone)] pub struct OpenOptions { pub visible: Option, pub focus: Option, pub open_new_workspace: Option, pub wait: bool, pub requesting_window: Option>, pub open_mode: OpenMode, pub env: Option>, pub open_in_dev_container: bool, } /// The result of opening a workspace via [`open_paths`], [`Workspace::new_local`], /// or [`Workspace::open_workspace_for_paths`]. pub struct OpenResult { pub window: WindowHandle, pub workspace: Entity, pub opened_items: Vec>>>, } /// Opens a workspace by its database ID, used for restoring empty workspaces with unsaved content. pub fn open_workspace_by_id( workspace_id: WorkspaceId, app_state: Arc, requesting_window: Option>, cx: &mut App, ) -> Task>> { let project_handle = Project::local( app_state.client.clone(), app_state.node_runtime.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), None, project::LocalProjectFlags { init_worktree_trust: true, ..project::LocalProjectFlags::default() }, cx, ); let db = WorkspaceDb::global(cx); let kvp = db::kvp::KeyValueStore::global(cx); cx.spawn(async move |cx| { let serialized_workspace = db .workspace_for_id(workspace_id) .with_context(|| format!("Workspace {workspace_id:?} not found"))?; let centered_layout = serialized_workspace.centered_layout; let (window, workspace) = if let Some(window) = requesting_window { let workspace = window.update(cx, |multi_workspace, window, cx| { let workspace = cx.new(|cx| { let mut workspace = Workspace::new( Some(workspace_id), project_handle.clone(), app_state.clone(), window, cx, ); workspace.centered_layout = centered_layout; workspace }); multi_workspace.add(workspace.clone(), &*window, cx); workspace })?; (window, workspace) } else { let window_bounds_override = window_bounds_env_override(); let (window_bounds, display) = if let Some(bounds) = window_bounds_override { (Some(WindowBounds::Windowed(bounds)), None) } else if let Some(display) = serialized_workspace.display && let Some(bounds) = serialized_workspace.window_bounds.as_ref() { (Some(bounds.0), Some(display)) } else if let Some((display, bounds)) = persistence::read_default_window_bounds(&kvp) { (Some(bounds), Some(display)) } else { (None, None) }; let options = cx.update(|cx| { let mut options = (app_state.build_window_options)(display, cx); options.window_bounds = window_bounds; options }); let window = cx.open_window(options, { let app_state = app_state.clone(); let project_handle = project_handle.clone(); move |window, cx| { let workspace = cx.new(|cx| { let mut workspace = Workspace::new( Some(workspace_id), project_handle, app_state, window, cx, ); workspace.centered_layout = centered_layout; workspace }); cx.new(|cx| MultiWorkspace::new(workspace, window, cx)) } })?; let workspace = window.update(cx, |multi_workspace: &mut MultiWorkspace, _, _cx| { multi_workspace.workspace().clone() })?; (window, workspace) }; notify_if_database_failed(window, cx); // Restore items from the serialized workspace window .update(cx, |_, window, cx| { workspace.update(cx, |_workspace, cx| { open_items(Some(serialized_workspace), vec![], window, cx) }) })? .await?; window.update(cx, |_, window, cx| { workspace.update(cx, |workspace, cx| { workspace.serialize_workspace(window, cx); }); })?; Ok(window) }) } #[allow(clippy::type_complexity)] pub fn open_paths( abs_paths: &[PathBuf], app_state: Arc, mut open_options: OpenOptions, cx: &mut App, ) -> Task> { let abs_paths = abs_paths.to_vec(); #[cfg(target_os = "windows")] let wsl_path = abs_paths .iter() .find_map(|p| util::paths::WslPath::from_path(p)); cx.spawn(async move |cx| { let (mut existing, mut open_visible) = find_existing_workspace( &abs_paths, &open_options, &SerializedWorkspaceLocation::Local, cx, ) .await; // Fallback: if no workspace contains the paths and all paths are files, // prefer an existing local workspace window (active window first). if open_options.open_new_workspace.is_none() && existing.is_none() { let all_paths = abs_paths.iter().map(|path| app_state.fs.metadata(path)); let all_metadatas = futures::future::join_all(all_paths) .await .into_iter() .filter_map(|result| result.ok().flatten()); if all_metadatas.into_iter().all(|file| !file.is_dir) { cx.update(|cx| { let windows = workspace_windows_for_location( &SerializedWorkspaceLocation::Local, cx, ); let window = cx .active_window() .and_then(|window| window.downcast::()) .filter(|window| windows.contains(window)) .or_else(|| windows.into_iter().next()); if let Some(window) = window { if let Ok(multi_workspace) = window.read(cx) { let active_workspace = multi_workspace.workspace().clone(); existing = Some((window, active_workspace)); open_visible = OpenVisible::None; } } }); } } // Fallback for directories: when no flag is specified and no existing // workspace matched, add the directory as a new workspace in the // active window's MultiWorkspace (instead of opening a new window). if open_options.open_new_workspace.is_none() && existing.is_none() { let target_window = cx.update(|cx| { let windows = workspace_windows_for_location( &SerializedWorkspaceLocation::Local, cx, ); let window = cx .active_window() .and_then(|window| window.downcast::()) .filter(|window| windows.contains(window)) .or_else(|| windows.into_iter().next()); window.filter(|window| { window.read(cx).is_ok_and(|mw| mw.multi_workspace_enabled(cx)) }) }); if let Some(window) = target_window { open_options.requesting_window = Some(window); window .update(cx, |multi_workspace, _, cx| { multi_workspace.open_sidebar(cx); }) .log_err(); } } let open_in_dev_container = open_options.open_in_dev_container; let result = if let Some((existing, target_workspace)) = existing { let open_task = existing .update(cx, |multi_workspace, window, cx| { window.activate_window(); multi_workspace.activate(target_workspace.clone(), window, cx); target_workspace.update(cx, |workspace, cx| { if open_in_dev_container { workspace.set_open_in_dev_container(true); } workspace.open_paths( abs_paths, OpenOptions { visible: Some(open_visible), ..Default::default() }, None, window, cx, ) }) })? .await; _ = existing.update(cx, |multi_workspace, _, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { for item in open_task.iter().flatten() { if let Err(e) = item { workspace.show_error(&e, cx); } } }); }); Ok(OpenResult { window: existing, workspace: target_workspace, opened_items: open_task }) } else { let init = if open_in_dev_container { Some(Box::new(|workspace: &mut Workspace, _window: &mut Window, _cx: &mut Context| { workspace.set_open_in_dev_container(true); }) as Box) + Send>) } else { None }; let result = cx .update(move |cx| { Workspace::new_local( abs_paths, app_state.clone(), open_options.requesting_window, open_options.env, init, open_options.open_mode, cx, ) }) .await; if let Ok(ref result) = result { result.window .update(cx, |_, window, _cx| { window.activate_window(); }) .log_err(); } result }; #[cfg(target_os = "windows")] if let Some(util::paths::WslPath{distro, path}) = wsl_path && let Ok(ref result) = result { result.window .update(cx, move |multi_workspace, _window, cx| { struct OpenInWsl; let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { workspace.show_notification(NotificationId::unique::(), cx, move |cx| { let display_path = util::markdown::MarkdownInlineCode(&path.to_string_lossy()); let msg = format!("{display_path} is inside a WSL filesystem, some features may not work unless you open it with WSL remote"); cx.new(move |cx| { MessageNotification::new(msg, cx) .primary_message("Open in WSL") .primary_icon(IconName::FolderOpen) .primary_on_click(move |window, cx| { window.dispatch_action(Box::new(remote::OpenWslPath { distro: remote::WslConnectionOptions { distro_name: distro.clone(), user: None, }, paths: vec![path.clone().into()], }), cx) }) }) }); }); }) .unwrap(); }; result }) } pub fn open_new( open_options: OpenOptions, app_state: Arc, cx: &mut App, init: impl FnOnce(&mut Workspace, &mut Window, &mut Context) + 'static + Send, ) -> Task> { let addition = open_options.open_mode; let task = Workspace::new_local( Vec::new(), app_state, open_options.requesting_window, open_options.env, Some(Box::new(init)), addition, cx, ); cx.spawn(async move |cx| { let OpenResult { window, .. } = task.await?; window .update(cx, |_, window, _cx| { window.activate_window(); }) .ok(); Ok(()) }) } pub fn create_and_open_local_file( path: &'static Path, window: &mut Window, cx: &mut Context, default_content: impl 'static + Send + FnOnce() -> Rope, ) -> Task>> { cx.spawn_in(window, async move |workspace, cx| { let fs = workspace.read_with(cx, |workspace, _| workspace.app_state().fs.clone())?; if !fs.is_file(path).await { fs.create_file(path, Default::default()).await?; fs.save(path, &default_content(), Default::default()) .await?; } workspace .update_in(cx, |workspace, window, cx| { workspace.with_local_or_wsl_workspace(window, cx, |workspace, window, cx| { let path = workspace .project .read_with(cx, |project, cx| project.try_windows_path_to_wsl(path, cx)); cx.spawn_in(window, async move |workspace, cx| { let path = path.await?; let path = fs.canonicalize(&path).await.unwrap_or(path); let mut items = workspace .update_in(cx, |workspace, window, cx| { workspace.open_paths( vec![path.to_path_buf()], OpenOptions { visible: Some(OpenVisible::None), ..Default::default() }, None, window, cx, ) })? .await; let item = items.pop().flatten(); item.with_context(|| format!("path {path:?} is not a file"))? }) }) })? .await? .await }) } pub fn open_remote_project_with_new_connection( window: WindowHandle, remote_connection: Arc, cancel_rx: oneshot::Receiver<()>, delegate: Arc, app_state: Arc, paths: Vec, cx: &mut App, ) -> Task>>>> { cx.spawn(async move |cx| { let (workspace_id, serialized_workspace) = deserialize_remote_project(remote_connection.connection_options(), paths.clone(), cx) .await?; let session = match cx .update(|cx| { remote::RemoteClient::new( ConnectionIdentifier::Workspace(workspace_id.0), remote_connection, cancel_rx, delegate, cx, ) }) .await? { Some(result) => result, None => return Ok(Vec::new()), }; let project = cx.update(|cx| { project::Project::remote( session, app_state.client.clone(), app_state.node_runtime.clone(), app_state.user_store.clone(), app_state.languages.clone(), app_state.fs.clone(), true, cx, ) }); open_remote_project_inner( project, paths, workspace_id, serialized_workspace, app_state, window, cx, ) .await }) } pub fn open_remote_project_with_existing_connection( connection_options: RemoteConnectionOptions, project: Entity, paths: Vec, app_state: Arc, window: WindowHandle, cx: &mut AsyncApp, ) -> Task>>>> { cx.spawn(async move |cx| { let (workspace_id, serialized_workspace) = deserialize_remote_project(connection_options.clone(), paths.clone(), cx).await?; open_remote_project_inner( project, paths, workspace_id, serialized_workspace, app_state, window, cx, ) .await }) } async fn open_remote_project_inner( project: Entity, paths: Vec, workspace_id: WorkspaceId, serialized_workspace: Option, app_state: Arc, window: WindowHandle, cx: &mut AsyncApp, ) -> Result>>> { let db = cx.update(|cx| WorkspaceDb::global(cx)); let toolchains = db.toolchains(workspace_id).await?; for (toolchain, worktree_path, path) in toolchains { project .update(cx, |this, cx| { let Some(worktree_id) = this.find_worktree(&worktree_path, cx) .and_then(|(worktree, rel_path)| { if rel_path.is_empty() { Some(worktree.read(cx).id()) } else { None } }) else { return Task::ready(None); }; this.activate_toolchain(ProjectPath { worktree_id, path }, toolchain, cx) }) .await; } let mut project_paths_to_open = vec![]; let mut project_path_errors = vec![]; for path in paths { let result = cx .update(|cx| Workspace::project_path_for_path(project.clone(), &path, true, cx)) .await; match result { Ok((_, project_path)) => { project_paths_to_open.push((path.clone(), Some(project_path))); } Err(error) => { project_path_errors.push(error); } }; } if project_paths_to_open.is_empty() { return Err(project_path_errors.pop().context("no paths given")?); } let workspace = window.update(cx, |multi_workspace, window, cx| { telemetry::event!("SSH Project Opened"); let new_workspace = cx.new(|cx| { let mut workspace = Workspace::new(Some(workspace_id), project, app_state.clone(), window, cx); workspace.update_history(cx); if let Some(ref serialized) = serialized_workspace { workspace.centered_layout = serialized.centered_layout; } workspace }); multi_workspace.activate(new_workspace.clone(), window, cx); new_workspace })?; let items = window .update(cx, |_, window, cx| { window.activate_window(); workspace.update(cx, |_workspace, cx| { open_items(serialized_workspace, project_paths_to_open, window, cx) }) })? .await?; workspace.update(cx, |workspace, cx| { for error in project_path_errors { if error.error_code() == proto::ErrorCode::DevServerProjectPathDoesNotExist { if let Some(path) = error.error_tag("path") { workspace.show_error(&anyhow!("'{path}' does not exist"), cx) } } else { workspace.show_error(&error, cx) } } }); Ok(items.into_iter().map(|item| item?.ok()).collect()) } fn deserialize_remote_project( connection_options: RemoteConnectionOptions, paths: Vec, cx: &AsyncApp, ) -> Task)>> { let db = cx.update(|cx| WorkspaceDb::global(cx)); cx.background_spawn(async move { let remote_connection_id = db .get_or_create_remote_connection(connection_options) .await?; let serialized_workspace = db.remote_workspace_for_roots(&paths, remote_connection_id); let workspace_id = if let Some(workspace_id) = serialized_workspace.as_ref().map(|workspace| workspace.id) { workspace_id } else { db.next_id().await? }; Ok((workspace_id, serialized_workspace)) }) } pub fn join_in_room_project( project_id: u64, follow_user_id: u64, app_state: Arc, cx: &mut App, ) -> Task> { let windows = cx.windows(); cx.spawn(async move |cx| { let existing_window_and_workspace: Option<( WindowHandle, Entity, )> = windows.into_iter().find_map(|window_handle| { window_handle .downcast::() .and_then(|window_handle| { window_handle .update(cx, |multi_workspace, _window, cx| { for workspace in multi_workspace.workspaces() { if workspace.read(cx).project().read(cx).remote_id() == Some(project_id) { return Some((window_handle, workspace.clone())); } } None }) .unwrap_or(None) }) }); let multi_workspace_window = if let Some((existing_window, target_workspace)) = existing_window_and_workspace { existing_window .update(cx, |multi_workspace, window, cx| { multi_workspace.activate(target_workspace, window, cx); }) .ok(); existing_window } else { let active_call = cx.update(|cx| GlobalAnyActiveCall::global(cx).clone()); let project = cx .update(|cx| { active_call.0.join_project( project_id, app_state.languages.clone(), app_state.fs.clone(), cx, ) }) .await?; let window_bounds_override = window_bounds_env_override(); cx.update(|cx| { let mut options = (app_state.build_window_options)(None, cx); options.window_bounds = window_bounds_override.map(WindowBounds::Windowed); cx.open_window(options, |window, cx| { let workspace = cx.new(|cx| { Workspace::new(Default::default(), project, app_state.clone(), window, cx) }); cx.new(|cx| MultiWorkspace::new(workspace, window, cx)) }) })? }; multi_workspace_window.update(cx, |multi_workspace, window, cx| { cx.activate(true); window.activate_window(); // We set the active workspace above, so this is the correct workspace. let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { let follow_peer_id = GlobalAnyActiveCall::try_global(cx) .and_then(|call| call.0.peer_id_for_user_in_room(follow_user_id, cx)) .or_else(|| { // If we couldn't follow the given user, follow the host instead. let collaborator = workspace .project() .read(cx) .collaborators() .values() .find(|collaborator| collaborator.is_host)?; Some(collaborator.peer_id) }); if let Some(follow_peer_id) = follow_peer_id { workspace.follow(follow_peer_id, window, cx); } }); })?; anyhow::Ok(()) }) } pub fn reload(cx: &mut App) { let should_confirm = WorkspaceSettings::get_global(cx).confirm_quit; let mut workspace_windows = cx .windows() .into_iter() .filter_map(|window| window.downcast::()) .collect::>(); // If multiple windows have unsaved changes, and need a save prompt, // prompt in the active window before switching to a different window. workspace_windows.sort_by_key(|window| window.is_active(cx) == Some(false)); let mut prompt = None; if let (true, Some(window)) = (should_confirm, workspace_windows.first()) { prompt = window .update(cx, |_, window, cx| { window.prompt( PromptLevel::Info, "Are you sure you want to restart?", None, &["Restart", "Cancel"], cx, ) }) .ok(); } cx.spawn(async move |cx| { if let Some(prompt) = prompt { let answer = prompt.await?; if answer != 0 { return anyhow::Ok(()); } } // If the user cancels any save prompt, then keep the app open. for window in workspace_windows { if let Ok(should_close) = window.update(cx, |multi_workspace, window, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| { workspace.prepare_to_close(CloseIntent::Quit, window, cx) }) }) && !should_close.await? { return anyhow::Ok(()); } } cx.update(|cx| cx.restart()); anyhow::Ok(()) }) .detach_and_log_err(cx); } fn parse_pixel_position_env_var(value: &str) -> Option> { let mut parts = value.split(','); let x: usize = parts.next()?.parse().ok()?; let y: usize = parts.next()?.parse().ok()?; Some(point(px(x as f32), px(y as f32))) } fn parse_pixel_size_env_var(value: &str) -> Option> { let mut parts = value.split(','); let width: usize = parts.next()?.parse().ok()?; let height: usize = parts.next()?.parse().ok()?; Some(size(px(width as f32), px(height as f32))) } /// Add client-side decorations (rounded corners, shadows, resize handling) when /// appropriate. /// /// The `border_radius_tiling` parameter allows overriding which corners get /// rounded, independently of the actual window tiling state. This is used /// specifically for the workspace switcher sidebar: when the sidebar is open, /// we want square corners on the left (so the sidebar appears flush with the /// window edge) but we still need the shadow padding for proper visual /// appearance. Unlike actual window tiling, this only affects border radius - /// not padding or shadows. pub fn client_side_decorations( element: impl IntoElement, window: &mut Window, cx: &mut App, border_radius_tiling: Tiling, ) -> Stateful
{ const BORDER_SIZE: Pixels = px(1.0); let decorations = window.window_decorations(); let tiling = match decorations { Decorations::Server => Tiling::default(), Decorations::Client { tiling } => tiling, }; match decorations { Decorations::Client { .. } => window.set_client_inset(theme::CLIENT_SIDE_DECORATION_SHADOW), Decorations::Server => window.set_client_inset(px(0.0)), } struct GlobalResizeEdge(ResizeEdge); impl Global for GlobalResizeEdge {} div() .id("window-backdrop") .bg(transparent_black()) .map(|div| match decorations { Decorations::Server => div, Decorations::Client { .. } => div .when( !(tiling.top || tiling.right || border_radius_tiling.top || border_radius_tiling.right), |div| div.rounded_tr(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when( !(tiling.top || tiling.left || border_radius_tiling.top || border_radius_tiling.left), |div| div.rounded_tl(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when( !(tiling.bottom || tiling.right || border_radius_tiling.bottom || border_radius_tiling.right), |div| div.rounded_br(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when( !(tiling.bottom || tiling.left || border_radius_tiling.bottom || border_radius_tiling.left), |div| div.rounded_bl(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when(!tiling.top, |div| { div.pt(theme::CLIENT_SIDE_DECORATION_SHADOW) }) .when(!tiling.bottom, |div| { div.pb(theme::CLIENT_SIDE_DECORATION_SHADOW) }) .when(!tiling.left, |div| { div.pl(theme::CLIENT_SIDE_DECORATION_SHADOW) }) .when(!tiling.right, |div| { div.pr(theme::CLIENT_SIDE_DECORATION_SHADOW) }) .on_mouse_move(move |e, window, cx| { let size = window.window_bounds().get_bounds().size; let pos = e.position; let new_edge = resize_edge(pos, theme::CLIENT_SIDE_DECORATION_SHADOW, size, tiling); let edge = cx.try_global::(); if new_edge != edge.map(|edge| edge.0) { window .window_handle() .update(cx, |workspace, _, cx| { cx.notify(workspace.entity_id()); }) .ok(); } }) .on_mouse_down(MouseButton::Left, move |e, window, _| { let size = window.window_bounds().get_bounds().size; let pos = e.position; let edge = match resize_edge( pos, theme::CLIENT_SIDE_DECORATION_SHADOW, size, tiling, ) { Some(value) => value, None => return, }; window.start_window_resize(edge); }), }) .size_full() .child( div() .cursor(CursorStyle::Arrow) .map(|div| match decorations { Decorations::Server => div, Decorations::Client { .. } => div .border_color(cx.theme().colors().border) .when( !(tiling.top || tiling.right || border_radius_tiling.top || border_radius_tiling.right), |div| div.rounded_tr(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when( !(tiling.top || tiling.left || border_radius_tiling.top || border_radius_tiling.left), |div| div.rounded_tl(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when( !(tiling.bottom || tiling.right || border_radius_tiling.bottom || border_radius_tiling.right), |div| div.rounded_br(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when( !(tiling.bottom || tiling.left || border_radius_tiling.bottom || border_radius_tiling.left), |div| div.rounded_bl(theme::CLIENT_SIDE_DECORATION_ROUNDING), ) .when(!tiling.top, |div| div.border_t(BORDER_SIZE)) .when(!tiling.bottom, |div| div.border_b(BORDER_SIZE)) .when(!tiling.left, |div| div.border_l(BORDER_SIZE)) .when(!tiling.right, |div| div.border_r(BORDER_SIZE)) .when(!tiling.is_tiled(), |div| { div.shadow(vec![gpui::BoxShadow { color: Hsla { h: 0., s: 0., l: 0., a: 0.4, }, blur_radius: theme::CLIENT_SIDE_DECORATION_SHADOW / 2., spread_radius: px(0.), offset: point(px(0.0), px(0.0)), }]) }), }) .on_mouse_move(|_e, _, cx| { cx.stop_propagation(); }) .size_full() .child(element), ) .map(|div| match decorations { Decorations::Server => div, Decorations::Client { tiling, .. } => div.child( canvas( |_bounds, window, _| { window.insert_hitbox( Bounds::new( point(px(0.0), px(0.0)), window.window_bounds().get_bounds().size, ), HitboxBehavior::Normal, ) }, move |_bounds, hitbox, window, cx| { let mouse = window.mouse_position(); let size = window.window_bounds().get_bounds().size; let Some(edge) = resize_edge(mouse, theme::CLIENT_SIDE_DECORATION_SHADOW, size, tiling) else { return; }; cx.set_global(GlobalResizeEdge(edge)); window.set_cursor_style( match edge { ResizeEdge::Top | ResizeEdge::Bottom => CursorStyle::ResizeUpDown, ResizeEdge::Left | ResizeEdge::Right => { CursorStyle::ResizeLeftRight } ResizeEdge::TopLeft | ResizeEdge::BottomRight => { CursorStyle::ResizeUpLeftDownRight } ResizeEdge::TopRight | ResizeEdge::BottomLeft => { CursorStyle::ResizeUpRightDownLeft } }, &hitbox, ); }, ) .size_full() .absolute(), ), }) } fn resize_edge( pos: Point, shadow_size: Pixels, window_size: Size, tiling: Tiling, ) -> Option { let bounds = Bounds::new(Point::default(), window_size).inset(shadow_size * 1.5); if bounds.contains(&pos) { return None; } let corner_size = size(shadow_size * 1.5, shadow_size * 1.5); let top_left_bounds = Bounds::new(Point::new(px(0.), px(0.)), corner_size); if !tiling.top && top_left_bounds.contains(&pos) { return Some(ResizeEdge::TopLeft); } let top_right_bounds = Bounds::new( Point::new(window_size.width - corner_size.width, px(0.)), corner_size, ); if !tiling.top && top_right_bounds.contains(&pos) { return Some(ResizeEdge::TopRight); } let bottom_left_bounds = Bounds::new( Point::new(px(0.), window_size.height - corner_size.height), corner_size, ); if !tiling.bottom && bottom_left_bounds.contains(&pos) { return Some(ResizeEdge::BottomLeft); } let bottom_right_bounds = Bounds::new( Point::new( window_size.width - corner_size.width, window_size.height - corner_size.height, ), corner_size, ); if !tiling.bottom && bottom_right_bounds.contains(&pos) { return Some(ResizeEdge::BottomRight); } if !tiling.top && pos.y < shadow_size { Some(ResizeEdge::Top) } else if !tiling.bottom && pos.y > window_size.height - shadow_size { Some(ResizeEdge::Bottom) } else if !tiling.left && pos.x < shadow_size { Some(ResizeEdge::Left) } else if !tiling.right && pos.x > window_size.width - shadow_size { Some(ResizeEdge::Right) } else { None } } fn join_pane_into_active( active_pane: &Entity, pane: &Entity, window: &mut Window, cx: &mut App, ) { if pane == active_pane { } else if pane.read(cx).items_len() == 0 { pane.update(cx, |_, cx| { cx.emit(pane::Event::Remove { focus_on_pane: None, }); }) } else { move_all_items(pane, active_pane, window, cx); } } fn move_all_items( from_pane: &Entity, to_pane: &Entity, window: &mut Window, cx: &mut App, ) { let destination_is_different = from_pane != to_pane; let mut moved_items = 0; for (item_ix, item_handle) in from_pane .read(cx) .items() .enumerate() .map(|(ix, item)| (ix, item.clone())) .collect::>() { let ix = item_ix - moved_items; if destination_is_different { // Close item from previous pane from_pane.update(cx, |source, cx| { source.remove_item_and_focus_on_pane(ix, false, to_pane.clone(), window, cx); }); moved_items += 1; } // This automatically removes duplicate items in the pane to_pane.update(cx, |destination, cx| { destination.add_item(item_handle, true, true, None, window, cx); window.focus(&destination.focus_handle(cx), cx) }); } } pub fn move_item( source: &Entity, destination: &Entity, item_id_to_move: EntityId, destination_index: usize, activate: bool, window: &mut Window, cx: &mut App, ) { let Some((item_ix, item_handle)) = source .read(cx) .items() .enumerate() .find(|(_, item_handle)| item_handle.item_id() == item_id_to_move) .map(|(ix, item)| (ix, item.clone())) else { // Tab was closed during drag return; }; if source != destination { // Close item from previous pane source.update(cx, |source, cx| { source.remove_item_and_focus_on_pane(item_ix, false, destination.clone(), window, cx); }); } // This automatically removes duplicate items in the pane destination.update(cx, |destination, cx| { destination.add_item_inner( item_handle, activate, activate, activate, Some(destination_index), window, cx, ); if activate { window.focus(&destination.focus_handle(cx), cx) } }); } pub fn move_active_item( source: &Entity, destination: &Entity, focus_destination: bool, close_if_empty: bool, window: &mut Window, cx: &mut App, ) { if source == destination { return; } let Some(active_item) = source.read(cx).active_item() else { return; }; source.update(cx, |source_pane, cx| { let item_id = active_item.item_id(); source_pane.remove_item(item_id, false, close_if_empty, window, cx); destination.update(cx, |target_pane, cx| { target_pane.add_item( active_item, focus_destination, focus_destination, Some(target_pane.items_len()), window, cx, ); }); }); } pub fn clone_active_item( workspace_id: Option, source: &Entity, destination: &Entity, focus_destination: bool, window: &mut Window, cx: &mut App, ) { if source == destination { return; } let Some(active_item) = source.read(cx).active_item() else { return; }; if !active_item.can_split(cx) { return; } let destination = destination.downgrade(); let task = active_item.clone_on_split(workspace_id, window, cx); window .spawn(cx, async move |cx| { let Some(clone) = task.await else { return; }; destination .update_in(cx, |target_pane, window, cx| { target_pane.add_item( clone, focus_destination, focus_destination, Some(target_pane.items_len()), window, cx, ); }) .log_err(); }) .detach(); } #[derive(Debug)] pub struct WorkspacePosition { pub window_bounds: Option, pub display: Option, pub centered_layout: bool, } pub fn remote_workspace_position_from_db( connection_options: RemoteConnectionOptions, paths_to_open: &[PathBuf], cx: &App, ) -> Task> { let paths = paths_to_open.to_vec(); let db = WorkspaceDb::global(cx); let kvp = db::kvp::KeyValueStore::global(cx); cx.background_spawn(async move { let remote_connection_id = db .get_or_create_remote_connection(connection_options) .await .context("fetching serialized ssh project")?; let serialized_workspace = db.remote_workspace_for_roots(&paths, remote_connection_id); let (window_bounds, display) = if let Some(bounds) = window_bounds_env_override() { (Some(WindowBounds::Windowed(bounds)), None) } else { let restorable_bounds = serialized_workspace .as_ref() .and_then(|workspace| { Some((workspace.display?, workspace.window_bounds.map(|b| b.0)?)) }) .or_else(|| persistence::read_default_window_bounds(&kvp)); if let Some((serialized_display, serialized_bounds)) = restorable_bounds { (Some(serialized_bounds), Some(serialized_display)) } else { (None, None) } }; let centered_layout = serialized_workspace .as_ref() .map(|w| w.centered_layout) .unwrap_or(false); Ok(WorkspacePosition { window_bounds, display, centered_layout, }) }) } pub fn with_active_or_new_workspace( cx: &mut App, f: impl FnOnce(&mut Workspace, &mut Window, &mut Context) + Send + 'static, ) { match cx .active_window() .and_then(|w| w.downcast::()) { Some(multi_workspace) => { cx.defer(move |cx| { multi_workspace .update(cx, |multi_workspace, window, cx| { let workspace = multi_workspace.workspace().clone(); workspace.update(cx, |workspace, cx| f(workspace, window, cx)); }) .log_err(); }); } None => { let app_state = AppState::global(cx); open_new( OpenOptions::default(), app_state, cx, move |workspace, window, cx| f(workspace, window, cx), ) .detach_and_log_err(cx); } } } /// Reads a panel's pixel size from its legacy KVP format and deletes the legacy /// key. This migration path only runs once per panel per workspace. fn load_legacy_panel_size( panel_key: &str, dock_position: DockPosition, workspace: &Workspace, cx: &mut App, ) -> Option { #[derive(Deserialize)] struct LegacyPanelState { #[serde(default)] width: Option, #[serde(default)] height: Option, } let workspace_id = workspace .database_id() .map(|id| i64::from(id).to_string()) .or_else(|| workspace.session_id())?; let legacy_key = match panel_key { "ProjectPanel" => { format!("{}-{:?}", "ProjectPanel", workspace_id) } "OutlinePanel" => { format!("{}-{:?}", "OutlinePanel", workspace_id) } "GitPanel" => { format!("{}-{:?}", "GitPanel", workspace_id) } "TerminalPanel" => { format!("{:?}-{:?}", "TerminalPanel", workspace_id) } _ => return None, }; let kvp = db::kvp::KeyValueStore::global(cx); let json = kvp.read_kvp(&legacy_key).log_err().flatten()?; let state = serde_json::from_str::(&json).log_err()?; let size = match dock_position { DockPosition::Bottom => state.height, DockPosition::Left | DockPosition::Right => state.width, }?; cx.background_spawn(async move { kvp.delete_kvp(legacy_key).await }) .detach_and_log_err(cx); Some(size) } #[cfg(test)] mod tests { use std::{cell::RefCell, rc::Rc, sync::Arc, time::Duration}; use super::*; use crate::{ dock::{PanelEvent, test::TestPanel}, item::{ ItemBufferKind, ItemEvent, test::{TestItem, TestProjectItem}, }, }; use fs::FakeFs; use gpui::{ DismissEvent, Empty, EventEmitter, FocusHandle, Focusable, Render, TestAppContext, UpdateGlobal, VisualTestContext, px, }; use project::{Project, ProjectEntryId}; use serde_json::json; use settings::SettingsStore; use util::path; use util::rel_path::rel_path; #[gpui::test] async fn test_tab_disambiguation(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); // Adding an item with no ambiguity renders the tab without detail. let item1 = cx.new(|cx| { let mut item = TestItem::new(cx); item.tab_descriptions = Some(vec!["c", "b1/c", "a/b1/c"]); item }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx); }); item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(0))); // Adding an item that creates ambiguity increases the level of detail on // both tabs. let item2 = cx.new_window_entity(|_window, cx| { let mut item = TestItem::new(cx); item.tab_descriptions = Some(vec!["c", "b2/c", "a/b2/c"]); item }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx); }); item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1))); item2.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1))); // Adding an item that creates ambiguity increases the level of detail only // on the ambiguous tabs. In this case, the ambiguity can't be resolved so // we stop at the highest detail available. let item3 = cx.new(|cx| { let mut item = TestItem::new(cx); item.tab_descriptions = Some(vec!["c", "b2/c", "a/b2/c"]); item }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item3.clone()), None, true, window, cx); }); item1.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(1))); item2.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(3))); item3.read_with(cx, |item, _| assert_eq!(item.tab_detail.get(), Some(3))); } #[gpui::test] async fn test_tracking_active_path(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); fs.insert_tree( "/root1", json!({ "one.txt": "", "two.txt": "", }), ) .await; fs.insert_tree( "/root2", json!({ "three.txt": "", }), ) .await; let project = Project::test(fs, ["root1".as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let worktree_id = project.update(cx, |project, cx| { project.worktrees(cx).next().unwrap().read(cx).id() }); let item1 = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "one.txt", cx)]) }); let item2 = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "two.txt", cx)]) }); // Add an item to an empty pane workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item1), None, true, window, cx) }); project.update(cx, |project, cx| { assert_eq!( project.active_entry(), project .entry_for_path(&(worktree_id, rel_path("one.txt")).into(), cx) .map(|e| e.id) ); }); assert_eq!(cx.window_title().as_deref(), Some("root1 — one.txt")); // Add a second item to a non-empty pane workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item2), None, true, window, cx) }); assert_eq!(cx.window_title().as_deref(), Some("root1 — two.txt")); project.update(cx, |project, cx| { assert_eq!( project.active_entry(), project .entry_for_path(&(worktree_id, rel_path("two.txt")).into(), cx) .map(|e| e.id) ); }); // Close the active item pane.update_in(cx, |pane, window, cx| { pane.close_active_item(&Default::default(), window, cx) }) .await .unwrap(); assert_eq!(cx.window_title().as_deref(), Some("root1 — one.txt")); project.update(cx, |project, cx| { assert_eq!( project.active_entry(), project .entry_for_path(&(worktree_id, rel_path("one.txt")).into(), cx) .map(|e| e.id) ); }); // Add a project folder project .update(cx, |project, cx| { project.find_or_create_worktree("root2", true, cx) }) .await .unwrap(); assert_eq!(cx.window_title().as_deref(), Some("root1, root2 — one.txt")); // Remove a project folder project.update(cx, |project, cx| project.remove_worktree(worktree_id, cx)); assert_eq!(cx.window_title().as_deref(), Some("root2 — one.txt")); } #[gpui::test] async fn test_close_window(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); fs.insert_tree("/root", json!({ "one": "" })).await; let project = Project::test(fs, ["root".as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); // When there are no dirty items, there's nothing to do. let item1 = cx.new(TestItem::new); workspace.update_in(cx, |w, window, cx| { w.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx) }); let task = workspace.update_in(cx, |w, window, cx| { w.prepare_to_close(CloseIntent::CloseWindow, window, cx) }); assert!(task.await.unwrap()); // When there are dirty untitled items, prompt to save each one. If the user // cancels any prompt, then abort. let item2 = cx.new(|cx| TestItem::new(cx).with_dirty(true)); let item3 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)]) }); workspace.update_in(cx, |w, window, cx| { w.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx); w.add_item_to_active_pane(Box::new(item3.clone()), None, true, window, cx); }); let task = workspace.update_in(cx, |w, window, cx| { w.prepare_to_close(CloseIntent::CloseWindow, window, cx) }); cx.executor().run_until_parked(); cx.simulate_prompt_answer("Cancel"); // cancel save all cx.executor().run_until_parked(); assert!(!cx.has_pending_prompt()); assert!(!task.await.unwrap()); } #[gpui::test] async fn test_multi_workspace_close_window_multiple_workspaces_cancel(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); fs.insert_tree("/root", json!({ "one": "" })).await; let project_a = Project::test(fs.clone(), ["root".as_ref()], cx).await; let project_b = Project::test(fs, ["root".as_ref()], cx).await; let multi_workspace_handle = cx.add_window(|window, cx| MultiWorkspace::test_new(project_a.clone(), window, cx)); cx.run_until_parked(); multi_workspace_handle .update(cx, |mw, _window, cx| { mw.open_sidebar(cx); }) .unwrap(); let workspace_a = multi_workspace_handle .read_with(cx, |mw, _| mw.workspace().clone()) .unwrap(); let workspace_b = multi_workspace_handle .update(cx, |mw, window, cx| { mw.test_add_workspace(project_b, window, cx) }) .unwrap(); // Activate workspace A multi_workspace_handle .update(cx, |mw, window, cx| { let workspace = mw.workspaces().next().unwrap().clone(); mw.activate(workspace, window, cx); }) .unwrap(); let cx = &mut VisualTestContext::from_window(multi_workspace_handle.into(), cx); // Workspace A has a clean item let item_a = cx.new(TestItem::new); workspace_a.update_in(cx, |w, window, cx| { w.add_item_to_active_pane(Box::new(item_a.clone()), None, true, window, cx) }); // Workspace B has a dirty item let item_b = cx.new(|cx| TestItem::new(cx).with_dirty(true)); workspace_b.update_in(cx, |w, window, cx| { w.add_item_to_active_pane(Box::new(item_b.clone()), None, true, window, cx) }); // Verify workspace A is active multi_workspace_handle .read_with(cx, |mw, _| { assert_eq!(mw.workspace(), &workspace_a); }) .unwrap(); // Dispatch CloseWindow — workspace A will pass, workspace B will prompt multi_workspace_handle .update(cx, |mw, window, cx| { mw.close_window(&CloseWindow, window, cx); }) .unwrap(); cx.run_until_parked(); // Workspace B should now be active since it has dirty items that need attention multi_workspace_handle .read_with(cx, |mw, _| { assert_eq!( mw.workspace(), &workspace_b, "workspace B should be activated when it prompts" ); }) .unwrap(); // User cancels the save prompt from workspace B cx.simulate_prompt_answer("Cancel"); cx.run_until_parked(); // Window should still exist because workspace B's close was cancelled assert!( multi_workspace_handle.update(cx, |_, _, _| ()).is_ok(), "window should still exist after cancelling one workspace's close" ); } #[gpui::test] async fn test_remove_workspace_prompts_for_unsaved_changes(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); fs.insert_tree("/root", json!({ "one": "" })).await; let project_a = Project::test(fs.clone(), ["root".as_ref()], cx).await; let project_b = Project::test(fs.clone(), ["root".as_ref()], cx).await; let multi_workspace_handle = cx.add_window(|window, cx| MultiWorkspace::test_new(project_a.clone(), window, cx)); cx.run_until_parked(); multi_workspace_handle .update(cx, |mw, _window, cx| mw.open_sidebar(cx)) .unwrap(); let workspace_a = multi_workspace_handle .read_with(cx, |mw, _| mw.workspace().clone()) .unwrap(); let workspace_b = multi_workspace_handle .update(cx, |mw, window, cx| { mw.test_add_workspace(project_b, window, cx) }) .unwrap(); // Activate workspace A. multi_workspace_handle .update(cx, |mw, window, cx| { mw.activate(workspace_a.clone(), window, cx); }) .unwrap(); let cx = &mut VisualTestContext::from_window(multi_workspace_handle.into(), cx); // Workspace B has a dirty item. let item_b = cx.new(|cx| TestItem::new(cx).with_dirty(true)); workspace_b.update_in(cx, |w, window, cx| { w.add_item_to_active_pane(Box::new(item_b.clone()), None, true, window, cx) }); // Try to remove workspace B. It should prompt because of the dirty item. let remove_task = multi_workspace_handle .update(cx, |mw, window, cx| { mw.remove([workspace_b.clone()], |_, _, _| unreachable!(), window, cx) }) .unwrap(); cx.run_until_parked(); // The prompt should have activated workspace B. multi_workspace_handle .read_with(cx, |mw, _| { assert_eq!( mw.workspace(), &workspace_b, "workspace B should be active while prompting" ); }) .unwrap(); // Cancel the prompt — user stays on workspace B. cx.simulate_prompt_answer("Cancel"); cx.run_until_parked(); let removed = remove_task.await.unwrap(); assert!(!removed, "removal should have been cancelled"); multi_workspace_handle .read_with(cx, |mw, _| { assert_eq!( mw.workspace(), &workspace_b, "user should stay on workspace B after cancelling" ); assert_eq!(mw.workspaces().count(), 2, "both workspaces should remain"); }) .unwrap(); // Try again. This time accept the prompt. let remove_task = multi_workspace_handle .update(cx, |mw, window, cx| { // First switch back to A. mw.activate(workspace_a.clone(), window, cx); mw.remove([workspace_b.clone()], |_, _, _| unreachable!(), window, cx) }) .unwrap(); cx.run_until_parked(); // Accept the save prompt. cx.simulate_prompt_answer("Don't Save"); cx.run_until_parked(); let removed = remove_task.await.unwrap(); assert!(removed, "removal should have succeeded"); // Should be back on workspace A, and B should be gone. multi_workspace_handle .read_with(cx, |mw, _| { assert_eq!( mw.workspace(), &workspace_a, "should be back on workspace A after removing B" ); assert_eq!(mw.workspaces().count(), 1, "only workspace A should remain"); }) .unwrap(); } #[gpui::test] async fn test_close_window_with_serializable_items(cx: &mut TestAppContext) { init_test(cx); // Register TestItem as a serializable item cx.update(|cx| { register_serializable_item::(cx); }); let fs = FakeFs::new(cx.executor()); fs.insert_tree("/root", json!({ "one": "" })).await; let project = Project::test(fs, ["root".as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); // When there are dirty untitled items, but they can serialize, then there is no prompt. let item1 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_serialize(|| Some(Task::ready(Ok(())))) }); let item2 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)]) .with_serialize(|| Some(Task::ready(Ok(())))) }); workspace.update_in(cx, |w, window, cx| { w.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx); w.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx); }); let task = workspace.update_in(cx, |w, window, cx| { w.prepare_to_close(CloseIntent::CloseWindow, window, cx) }); assert!(task.await.unwrap()); } #[gpui::test] async fn test_close_pane_items(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, None, cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let item1 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_project_items(&[dirty_project_item(1, "1.txt", cx)]) }); let item2 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_conflict(true) .with_project_items(&[dirty_project_item(2, "2.txt", cx)]) }); let item3 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_conflict(true) .with_project_items(&[dirty_project_item(3, "3.txt", cx)]) }); let item4 = cx.new(|cx| { TestItem::new(cx).with_dirty(true).with_project_items(&[{ let project_item = TestProjectItem::new_untitled(cx); project_item.update(cx, |project_item, _| project_item.is_dirty = true); project_item }]) }); let pane = workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item1.clone()), None, true, window, cx); workspace.add_item_to_active_pane(Box::new(item2.clone()), None, true, window, cx); workspace.add_item_to_active_pane(Box::new(item3.clone()), None, true, window, cx); workspace.add_item_to_active_pane(Box::new(item4.clone()), None, true, window, cx); workspace.active_pane().clone() }); let close_items = pane.update_in(cx, |pane, window, cx| { pane.activate_item(1, true, true, window, cx); assert_eq!(pane.active_item().unwrap().item_id(), item2.item_id()); let item1_id = item1.item_id(); let item3_id = item3.item_id(); let item4_id = item4.item_id(); pane.close_items(window, cx, SaveIntent::Close, &move |id| { [item1_id, item3_id, item4_id].contains(&id) }) }); cx.executor().run_until_parked(); assert!(cx.has_pending_prompt()); cx.simulate_prompt_answer("Save all"); cx.executor().run_until_parked(); // Item 1 is saved. There's a prompt to save item 3. pane.update(cx, |pane, cx| { assert_eq!(item1.read(cx).save_count, 1); assert_eq!(item1.read(cx).save_as_count, 0); assert_eq!(item1.read(cx).reload_count, 0); assert_eq!(pane.items_len(), 3); assert_eq!(pane.active_item().unwrap().item_id(), item3.item_id()); }); assert!(cx.has_pending_prompt()); // Cancel saving item 3. cx.simulate_prompt_answer("Discard"); cx.executor().run_until_parked(); // Item 3 is reloaded. There's a prompt to save item 4. pane.update(cx, |pane, cx| { assert_eq!(item3.read(cx).save_count, 0); assert_eq!(item3.read(cx).save_as_count, 0); assert_eq!(item3.read(cx).reload_count, 1); assert_eq!(pane.items_len(), 2); assert_eq!(pane.active_item().unwrap().item_id(), item4.item_id()); }); // There's a prompt for a path for item 4. cx.simulate_new_path_selection(|_| Some(Default::default())); close_items.await.unwrap(); // The requested items are closed. pane.update(cx, |pane, cx| { assert_eq!(item4.read(cx).save_count, 0); assert_eq!(item4.read(cx).save_as_count, 1); assert_eq!(item4.read(cx).reload_count, 0); assert_eq!(pane.items_len(), 1); assert_eq!(pane.active_item().unwrap().item_id(), item2.item_id()); }); } #[gpui::test] async fn test_prompting_to_save_only_on_last_item_for_entry(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); // Create several workspace items with single project entries, and two // workspace items with multiple project entries. let single_entry_items = (0..=4) .map(|project_entry_id| { cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_project_items(&[dirty_project_item( project_entry_id, &format!("{project_entry_id}.txt"), cx, )]) }) }) .collect::>(); let item_2_3 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_buffer_kind(ItemBufferKind::Multibuffer) .with_project_items(&[ single_entry_items[2].read(cx).project_items[0].clone(), single_entry_items[3].read(cx).project_items[0].clone(), ]) }); let item_3_4 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_buffer_kind(ItemBufferKind::Multibuffer) .with_project_items(&[ single_entry_items[3].read(cx).project_items[0].clone(), single_entry_items[4].read(cx).project_items[0].clone(), ]) }); // Create two panes that contain the following project entries: // left pane: // multi-entry items: (2, 3) // single-entry items: 0, 2, 3, 4 // right pane: // single-entry items: 4, 1 // multi-entry items: (3, 4) let (left_pane, right_pane) = workspace.update_in(cx, |workspace, window, cx| { let left_pane = workspace.active_pane().clone(); workspace.add_item_to_active_pane(Box::new(item_2_3.clone()), None, true, window, cx); workspace.add_item_to_active_pane( single_entry_items[0].boxed_clone(), None, true, window, cx, ); workspace.add_item_to_active_pane( single_entry_items[2].boxed_clone(), None, true, window, cx, ); workspace.add_item_to_active_pane( single_entry_items[3].boxed_clone(), None, true, window, cx, ); workspace.add_item_to_active_pane( single_entry_items[4].boxed_clone(), None, true, window, cx, ); let right_pane = workspace.split_and_clone(left_pane.clone(), SplitDirection::Right, window, cx); let boxed_clone = single_entry_items[1].boxed_clone(); let right_pane = window.spawn(cx, async move |cx| { right_pane.await.inspect(|right_pane| { right_pane .update_in(cx, |pane, window, cx| { pane.add_item(boxed_clone, true, true, None, window, cx); pane.add_item(Box::new(item_3_4.clone()), true, true, None, window, cx); }) .unwrap(); }) }); (left_pane, right_pane) }); let right_pane = right_pane.await.unwrap(); cx.focus(&right_pane); let close = right_pane.update_in(cx, |pane, window, cx| { pane.close_all_items(&CloseAllItems::default(), window, cx) .unwrap() }); cx.executor().run_until_parked(); let msg = cx.pending_prompt().unwrap().0; assert!(msg.contains("1.txt")); assert!(!msg.contains("2.txt")); assert!(!msg.contains("3.txt")); assert!(!msg.contains("4.txt")); // With best-effort close, cancelling item 1 keeps it open but items 4 // and (3,4) still close since their entries exist in left pane. cx.simulate_prompt_answer("Cancel"); close.await; right_pane.read_with(cx, |pane, _| { assert_eq!(pane.items_len(), 1); }); // Remove item 3 from left pane, making (2,3) the only item with entry 3. left_pane .update_in(cx, |left_pane, window, cx| { left_pane.close_item_by_id( single_entry_items[3].entity_id(), SaveIntent::Skip, window, cx, ) }) .await .unwrap(); let close = left_pane.update_in(cx, |pane, window, cx| { pane.close_all_items(&CloseAllItems::default(), window, cx) .unwrap() }); cx.executor().run_until_parked(); let details = cx.pending_prompt().unwrap().1; assert!(details.contains("0.txt")); assert!(details.contains("3.txt")); assert!(details.contains("4.txt")); // Ideally 2.txt wouldn't appear since entry 2 still exists in item 2. // But we can only save whole items, so saving (2,3) for entry 3 includes 2. // assert!(!details.contains("2.txt")); cx.simulate_prompt_answer("Save all"); cx.executor().run_until_parked(); close.await; left_pane.read_with(cx, |pane, _| { assert_eq!(pane.items_len(), 0); }); } #[gpui::test] async fn test_autosave(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "1.txt", cx)]) }); let item_id = item.entity_id(); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx); }); // Autosave on window change. item.update(cx, |item, cx| { SettingsStore::update_global(cx, |settings, cx| { settings.update_user_settings(cx, |settings| { settings.workspace.autosave = Some(AutosaveSetting::OnWindowChange); }) }); item.is_dirty = true; }); // Deactivating the window saves the file. cx.deactivate_window(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 1)); // Re-activating the window doesn't save the file. cx.update(|window, _| window.activate_window()); cx.executor().run_until_parked(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 1)); // Autosave on focus change. item.update_in(cx, |item, window, cx| { cx.focus_self(window); SettingsStore::update_global(cx, |settings, cx| { settings.update_user_settings(cx, |settings| { settings.workspace.autosave = Some(AutosaveSetting::OnFocusChange); }) }); item.is_dirty = true; }); // Blurring the item saves the file. item.update_in(cx, |_, window, _| window.blur()); cx.executor().run_until_parked(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 2)); // Deactivating the window still saves the file. item.update_in(cx, |item, window, cx| { cx.focus_self(window); item.is_dirty = true; }); cx.deactivate_window(); item.update(cx, |item, _| assert_eq!(item.save_count, 3)); // Autosave after delay. item.update(cx, |item, cx| { SettingsStore::update_global(cx, |settings, cx| { settings.update_user_settings(cx, |settings| { settings.workspace.autosave = Some(AutosaveSetting::AfterDelay { milliseconds: 500.into(), }); }) }); item.is_dirty = true; cx.emit(ItemEvent::Edit); }); // Delay hasn't fully expired, so the file is still dirty and unsaved. cx.executor().advance_clock(Duration::from_millis(250)); item.read_with(cx, |item, _| assert_eq!(item.save_count, 3)); // After delay expires, the file is saved. cx.executor().advance_clock(Duration::from_millis(250)); item.read_with(cx, |item, _| assert_eq!(item.save_count, 4)); // Autosave after delay, should save earlier than delay if tab is closed item.update(cx, |item, cx| { item.is_dirty = true; cx.emit(ItemEvent::Edit); }); cx.executor().advance_clock(Duration::from_millis(250)); item.read_with(cx, |item, _| assert_eq!(item.save_count, 4)); // // Ensure auto save with delay saves the item on close, even if the timer hasn't yet run out. pane.update_in(cx, |pane, window, cx| { pane.close_items(window, cx, SaveIntent::Close, &move |id| id == item_id) }) .await .unwrap(); assert!(!cx.has_pending_prompt()); item.read_with(cx, |item, _| assert_eq!(item.save_count, 5)); // Add the item again, ensuring autosave is prevented if the underlying file has been deleted. workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx); }); item.update_in(cx, |item, _window, cx| { item.is_dirty = true; for project_item in &mut item.project_items { project_item.update(cx, |project_item, _| project_item.is_dirty = true); } }); cx.run_until_parked(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 5)); // Autosave on focus change, ensuring closing the tab counts as such. item.update(cx, |item, cx| { SettingsStore::update_global(cx, |settings, cx| { settings.update_user_settings(cx, |settings| { settings.workspace.autosave = Some(AutosaveSetting::OnFocusChange); }) }); item.is_dirty = true; for project_item in &mut item.project_items { project_item.update(cx, |project_item, _| project_item.is_dirty = true); } }); pane.update_in(cx, |pane, window, cx| { pane.close_items(window, cx, SaveIntent::Close, &move |id| id == item_id) }) .await .unwrap(); assert!(!cx.has_pending_prompt()); item.read_with(cx, |item, _| assert_eq!(item.save_count, 6)); // Add the item again, ensuring autosave is prevented if the underlying file has been deleted. workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx); }); item.update_in(cx, |item, window, cx| { item.project_items[0].update(cx, |item, _| { item.entry_id = None; }); item.is_dirty = true; window.blur(); }); cx.run_until_parked(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 6)); // Ensure autosave is prevented for deleted files also when closing the buffer. let _close_items = pane.update_in(cx, |pane, window, cx| { pane.close_items(window, cx, SaveIntent::Close, &move |id| id == item_id) }); cx.run_until_parked(); assert!(cx.has_pending_prompt()); item.read_with(cx, |item, _| assert_eq!(item.save_count, 6)); } #[gpui::test] async fn test_autosave_on_focus_change_in_multibuffer(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); // Create a multibuffer-like item with two child focus handles, // simulating individual buffer editors within a multibuffer. let item = cx.new(|cx| { TestItem::new(cx) .with_project_items(&[TestProjectItem::new(1, "1.txt", cx)]) .with_child_focus_handles(2, cx) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx); }); // Set autosave to OnFocusChange and focus the first child handle, // simulating the user's cursor being inside one of the multibuffer's excerpts. item.update_in(cx, |item, window, cx| { SettingsStore::update_global(cx, |settings, cx| { settings.update_user_settings(cx, |settings| { settings.workspace.autosave = Some(AutosaveSetting::OnFocusChange); }) }); item.is_dirty = true; window.focus(&item.child_focus_handles[0], cx); }); cx.executor().run_until_parked(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 0)); // Moving focus from one child to another within the same item should // NOT trigger autosave — focus is still within the item's focus hierarchy. item.update_in(cx, |item, window, cx| { window.focus(&item.child_focus_handles[1], cx); }); cx.executor().run_until_parked(); item.read_with(cx, |item, _| { assert_eq!( item.save_count, 0, "Switching focus between children within the same item should not autosave" ); }); // Blurring the item saves the file. This is the core regression scenario: // with `on_blur`, this would NOT trigger because `on_blur` only fires when // the item's own focus handle is the leaf that lost focus. In a multibuffer, // the leaf is always a child focus handle, so `on_blur` never detected // focus leaving the item. item.update_in(cx, |_, window, _| window.blur()); cx.executor().run_until_parked(); item.read_with(cx, |item, _| { assert_eq!( item.save_count, 1, "Blurring should trigger autosave when focus was on a child of the item" ); }); // Deactivating the window should also trigger autosave when a child of // the multibuffer item currently owns focus. item.update_in(cx, |item, window, cx| { item.is_dirty = true; window.focus(&item.child_focus_handles[0], cx); }); cx.executor().run_until_parked(); item.read_with(cx, |item, _| assert_eq!(item.save_count, 1)); cx.deactivate_window(); item.read_with(cx, |item, _| { assert_eq!( item.save_count, 2, "Deactivating window should trigger autosave when focus was on a child" ); }); } #[gpui::test] async fn test_pane_navigation(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "1.txt", cx)]) }); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let toolbar = pane.read_with(cx, |pane, _| pane.toolbar().clone()); let toolbar_notify_count = Rc::new(RefCell::new(0)); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item.clone()), None, true, window, cx); let toolbar_notification_count = toolbar_notify_count.clone(); cx.observe_in(&toolbar, window, move |_, _, _, _| { *toolbar_notification_count.borrow_mut() += 1 }) .detach(); }); pane.read_with(cx, |pane, _| { assert!(!pane.can_navigate_backward()); assert!(!pane.can_navigate_forward()); }); item.update_in(cx, |item, _, cx| { item.set_state("one".to_string(), cx); }); // Toolbar must be notified to re-render the navigation buttons assert_eq!(*toolbar_notify_count.borrow(), 1); pane.read_with(cx, |pane, _| { assert!(pane.can_navigate_backward()); assert!(!pane.can_navigate_forward()); }); workspace .update_in(cx, |workspace, window, cx| { workspace.go_back(pane.downgrade(), window, cx) }) .await .unwrap(); assert_eq!(*toolbar_notify_count.borrow(), 2); pane.read_with(cx, |pane, _| { assert!(!pane.can_navigate_backward()); assert!(pane.can_navigate_forward()); }); } /// Tests that the navigation history deduplicates entries for the same item. /// /// When navigating back and forth between items (e.g., A -> B -> A -> B -> A -> B -> C), /// the navigation history deduplicates by keeping only the most recent visit to each item, /// resulting in [A, B, C] instead of [A, B, A, B, A, B, C]. This ensures that Go Back (Ctrl-O) /// navigates through unique items efficiently: C -> B -> A, rather than bouncing between /// repeated entries: C -> B -> A -> B -> A -> B -> A. /// /// This behavior prevents the navigation history from growing unnecessarily large and provides /// a better user experience by eliminating redundant navigation steps when jumping between files. #[gpui::test] async fn test_navigation_history_deduplication(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let item_a = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "a.txt", cx)]) }); let item_b = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "b.txt", cx)]) }); let item_c = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(3, "c.txt", cx)]) }); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item_a.clone()), None, true, window, cx); workspace.add_item_to_active_pane(Box::new(item_b.clone()), None, true, window, cx); workspace.add_item_to_active_pane(Box::new(item_c.clone()), None, true, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_a, false, false, window, cx); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_b, false, false, window, cx); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_a, false, false, window, cx); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_b, false, false, window, cx); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_a, false, false, window, cx); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_b, false, false, window, cx); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_item(&item_c, false, false, window, cx); }); cx.run_until_parked(); let backward_count = pane.read_with(cx, |pane, cx| { let mut count = 0; pane.nav_history().for_each_entry(cx, &mut |_, _| { count += 1; }); count }); assert!( backward_count <= 4, "Should have at most 4 entries, got {}", backward_count ); workspace .update_in(cx, |workspace, window, cx| { workspace.go_back(pane.downgrade(), window, cx) }) .await .unwrap(); let active_item = workspace.read_with(cx, |workspace, cx| { workspace.active_item(cx).unwrap().item_id() }); assert_eq!( active_item, item_b.entity_id(), "After first go_back, should be at item B" ); workspace .update_in(cx, |workspace, window, cx| { workspace.go_back(pane.downgrade(), window, cx) }) .await .unwrap(); let active_item = workspace.read_with(cx, |workspace, cx| { workspace.active_item(cx).unwrap().item_id() }); assert_eq!( active_item, item_a.entity_id(), "After second go_back, should be at item A" ); pane.read_with(cx, |pane, _| { assert!(pane.can_navigate_forward(), "Should be able to go forward"); }); } #[gpui::test] async fn test_activate_last_pane(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); workspace.update_in(cx, |workspace, window, cx| { let first_item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "1.txt", cx)]) }); workspace.add_item_to_active_pane(Box::new(first_item), None, true, window, cx); workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Right, window, cx, ); workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Right, window, cx, ); }); let (first_pane_id, target_last_pane_id) = workspace.update(cx, |workspace, _cx| { let panes = workspace.center.panes(); assert!(panes.len() >= 2); ( panes.first().expect("at least one pane").entity_id(), panes.last().expect("at least one pane").entity_id(), ) }); workspace.update_in(cx, |workspace, window, cx| { workspace.activate_pane_at_index(&ActivatePane(0), window, cx); }); workspace.update(cx, |workspace, _| { assert_eq!(workspace.active_pane().entity_id(), first_pane_id); assert_ne!(workspace.active_pane().entity_id(), target_last_pane_id); }); cx.dispatch_action(ActivateLastPane); workspace.update(cx, |workspace, _| { assert_eq!(workspace.active_pane().entity_id(), target_last_pane_id); }); } #[gpui::test] async fn test_toggle_docks_and_panels(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let panel = workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace .right_dock() .update(cx, |right_dock, cx| right_dock.set_open(true, window, cx)); panel }); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); pane.update_in(cx, |pane, window, cx| { let item = cx.new(TestItem::new); pane.add_item(Box::new(item), true, true, None, window, cx); }); // Transfer focus from center to panel workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(!panel.is_zoomed(window, cx)); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Transfer focus from panel to center workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(!panel.is_zoomed(window, cx)); assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx)); assert!(pane.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Close the dock workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(!workspace.right_dock().read(cx).is_open()); assert!(!panel.is_zoomed(window, cx)); assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx)); assert!(pane.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Open the dock workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(!panel.is_zoomed(window, cx)); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Focus and zoom panel panel.update_in(cx, |panel, window, cx| { cx.focus_self(window); panel.set_zoomed(true, window, cx) }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(panel.is_zoomed(window, cx)); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Transfer focus to the center closes the dock workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(!workspace.right_dock().read(cx).is_open()); assert!(panel.is_zoomed(window, cx)); assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Transferring focus back to the panel keeps it zoomed workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(panel.is_zoomed(window, cx)); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Close the dock while it is zoomed workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx) }); workspace.update_in(cx, |workspace, window, cx| { assert!(!workspace.right_dock().read(cx).is_open()); assert!(panel.is_zoomed(window, cx)); assert!(workspace.zoomed.is_none()); assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Opening the dock, when it's zoomed, retains focus workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx) }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(panel.is_zoomed(window, cx)); assert!(workspace.zoomed.is_some()); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Unzoom and close the panel, zoom the active pane. panel.update_in(cx, |panel, window, cx| panel.set_zoomed(false, window, cx)); workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx) }); pane.update_in(cx, |pane, window, cx| { pane.toggle_zoom(&Default::default(), window, cx) }); // Opening a dock unzooms the pane. workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx) }); workspace.update_in(cx, |workspace, window, cx| { let pane = pane.read(cx); assert!(!pane.is_zoomed()); assert!(!pane.focus_handle(cx).is_focused(window)); assert!(workspace.right_dock().read(cx).is_open()); assert!(workspace.zoomed.is_none()); }); } #[gpui::test] async fn test_close_panel_on_toggle(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let panel = workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); panel }); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); pane.update_in(cx, |pane, window, cx| { let item = cx.new(TestItem::new); pane.add_item(Box::new(item), true, true, None, window, cx); }); // Enable close_panel_on_toggle cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.workspace.close_panel_on_toggle = Some(true); }); }); // Panel starts closed. Toggling should open and focus it. workspace.update_in(cx, |workspace, window, cx| { assert!(!workspace.right_dock().read(cx).is_open()); workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!( workspace.right_dock().read(cx).is_open(), "Dock should be open after toggling from center" ); assert!( panel.read(cx).focus_handle(cx).contains_focused(window, cx), "Panel should be focused after toggling from center" ); }); // Panel is open and focused. Toggling should close the panel and // return focus to the center. workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!( !workspace.right_dock().read(cx).is_open(), "Dock should be closed after toggling from focused panel" ); assert!( !panel.read(cx).focus_handle(cx).contains_focused(window, cx), "Panel should not be focused after toggling from focused panel" ); }); // Open the dock and focus something else so the panel is open but not // focused. Toggling should focus the panel (not close it). workspace.update_in(cx, |workspace, window, cx| { workspace .right_dock() .update(cx, |dock, cx| dock.set_open(true, window, cx)); window.focus(&pane.read(cx).focus_handle(cx), cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(!panel.read(cx).focus_handle(cx).contains_focused(window, cx)); workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!( workspace.right_dock().read(cx).is_open(), "Dock should remain open when toggling focuses an open-but-unfocused panel" ); assert!( panel.read(cx).focus_handle(cx).contains_focused(window, cx), "Panel should be focused after toggling an open-but-unfocused panel" ); }); // Now disable the setting and verify the original behavior: toggling // from a focused panel moves focus to center but leaves the dock open. cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.workspace.close_panel_on_toggle = Some(false); }); }); workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!( workspace.right_dock().read(cx).is_open(), "Dock should remain open when setting is disabled" ); assert!( !panel.read(cx).focus_handle(cx).contains_focused(window, cx), "Panel should not be focused after toggling with setting disabled" ); }); } #[gpui::test] async fn test_pane_zoom_in_out(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.update_in(cx, |workspace, _window, _cx| { workspace.active_pane().clone() }); // Add an item to the pane so it can be zoomed workspace.update_in(cx, |workspace, window, cx| { let item = cx.new(TestItem::new); workspace.add_item(pane.clone(), Box::new(item), None, true, true, window, cx); }); // Initially not zoomed workspace.update_in(cx, |workspace, _window, cx| { assert!(!pane.read(cx).is_zoomed(), "Pane starts unzoomed"); assert!( workspace.zoomed.is_none(), "Workspace should track no zoomed pane" ); assert!(pane.read(cx).items_len() > 0, "Pane should have items"); }); // Zoom In pane.update_in(cx, |pane, window, cx| { pane.zoom_in(&crate::ZoomIn, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!( pane.read(cx).is_zoomed(), "Pane should be zoomed after ZoomIn" ); assert!( workspace.zoomed.is_some(), "Workspace should track the zoomed pane" ); assert!( pane.read(cx).focus_handle(cx).contains_focused(window, cx), "ZoomIn should focus the pane" ); }); // Zoom In again is a no-op pane.update_in(cx, |pane, window, cx| { pane.zoom_in(&crate::ZoomIn, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(pane.read(cx).is_zoomed(), "Second ZoomIn keeps pane zoomed"); assert!( workspace.zoomed.is_some(), "Workspace still tracks zoomed pane" ); assert!( pane.read(cx).focus_handle(cx).contains_focused(window, cx), "Pane remains focused after repeated ZoomIn" ); }); // Zoom Out pane.update_in(cx, |pane, window, cx| { pane.zoom_out(&crate::ZoomOut, window, cx); }); workspace.update_in(cx, |workspace, _window, cx| { assert!( !pane.read(cx).is_zoomed(), "Pane should unzoom after ZoomOut" ); assert!( workspace.zoomed.is_none(), "Workspace clears zoom tracking after ZoomOut" ); }); // Zoom Out again is a no-op pane.update_in(cx, |pane, window, cx| { pane.zoom_out(&crate::ZoomOut, window, cx); }); workspace.update_in(cx, |workspace, _window, cx| { assert!( !pane.read(cx).is_zoomed(), "Second ZoomOut keeps pane unzoomed" ); assert!( workspace.zoomed.is_none(), "Workspace remains without zoomed pane" ); }); } #[gpui::test] async fn test_toggle_all_docks(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); workspace.update_in(cx, |workspace, window, cx| { // Open two docks let left_dock = workspace.dock_at_position(DockPosition::Left); let right_dock = workspace.dock_at_position(DockPosition::Right); left_dock.update(cx, |dock, cx| dock.set_open(true, window, cx)); right_dock.update(cx, |dock, cx| dock.set_open(true, window, cx)); assert!(left_dock.read(cx).is_open()); assert!(right_dock.read(cx).is_open()); }); workspace.update_in(cx, |workspace, window, cx| { // Toggle all docks - should close both workspace.toggle_all_docks(&ToggleAllDocks, window, cx); let left_dock = workspace.dock_at_position(DockPosition::Left); let right_dock = workspace.dock_at_position(DockPosition::Right); assert!(!left_dock.read(cx).is_open()); assert!(!right_dock.read(cx).is_open()); }); workspace.update_in(cx, |workspace, window, cx| { // Toggle again - should reopen both workspace.toggle_all_docks(&ToggleAllDocks, window, cx); let left_dock = workspace.dock_at_position(DockPosition::Left); let right_dock = workspace.dock_at_position(DockPosition::Right); assert!(left_dock.read(cx).is_open()); assert!(right_dock.read(cx).is_open()); }); } #[gpui::test] async fn test_toggle_all_with_manual_close(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); workspace.update_in(cx, |workspace, window, cx| { // Open two docks let left_dock = workspace.dock_at_position(DockPosition::Left); let right_dock = workspace.dock_at_position(DockPosition::Right); left_dock.update(cx, |dock, cx| dock.set_open(true, window, cx)); right_dock.update(cx, |dock, cx| dock.set_open(true, window, cx)); assert!(left_dock.read(cx).is_open()); assert!(right_dock.read(cx).is_open()); }); workspace.update_in(cx, |workspace, window, cx| { // Close them manually workspace.toggle_dock(DockPosition::Left, window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); let left_dock = workspace.dock_at_position(DockPosition::Left); let right_dock = workspace.dock_at_position(DockPosition::Right); assert!(!left_dock.read(cx).is_open()); assert!(!right_dock.read(cx).is_open()); }); workspace.update_in(cx, |workspace, window, cx| { // Toggle all docks - only last closed (right dock) should reopen workspace.toggle_all_docks(&ToggleAllDocks, window, cx); let left_dock = workspace.dock_at_position(DockPosition::Left); let right_dock = workspace.dock_at_position(DockPosition::Right); assert!(!left_dock.read(cx).is_open()); assert!(right_dock.read(cx).is_open()); }); } #[gpui::test] async fn test_toggle_all_docks_after_dock_move(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); // Open two docks (left and right) with one panel each let (left_panel, right_panel) = workspace.update_in(cx, |workspace, window, cx| { let left_panel = cx.new(|cx| TestPanel::new(DockPosition::Left, 100, cx)); workspace.add_panel(left_panel.clone(), window, cx); let right_panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 101, cx)); workspace.add_panel(right_panel.clone(), window, cx); workspace.toggle_dock(DockPosition::Left, window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); // Verify initial state assert!( workspace.left_dock().read(cx).is_open(), "Left dock should be open" ); assert_eq!( workspace .left_dock() .read(cx) .visible_panel() .unwrap() .panel_id(), left_panel.panel_id(), "Left panel should be visible in left dock" ); assert!( workspace.right_dock().read(cx).is_open(), "Right dock should be open" ); assert_eq!( workspace .right_dock() .read(cx) .visible_panel() .unwrap() .panel_id(), right_panel.panel_id(), "Right panel should be visible in right dock" ); assert!( !workspace.bottom_dock().read(cx).is_open(), "Bottom dock should be closed" ); (left_panel, right_panel) }); // Focus the left panel and move it to the next position (bottom dock) workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); // Focus left panel assert!( left_panel.read(cx).focus_handle(cx).is_focused(window), "Left panel should be focused" ); }); cx.dispatch_action(MoveFocusedPanelToNextPosition); // Verify the left panel has moved to the bottom dock, and the bottom dock is now open workspace.update(cx, |workspace, cx| { assert!( !workspace.left_dock().read(cx).is_open(), "Left dock should be closed" ); assert!( workspace.bottom_dock().read(cx).is_open(), "Bottom dock should now be open" ); assert_eq!( left_panel.read(cx).position, DockPosition::Bottom, "Left panel should now be in the bottom dock" ); assert_eq!( workspace .bottom_dock() .read(cx) .visible_panel() .unwrap() .panel_id(), left_panel.panel_id(), "Left panel should be the visible panel in the bottom dock" ); }); // Toggle all docks off workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_all_docks(&ToggleAllDocks, window, cx); assert!( !workspace.left_dock().read(cx).is_open(), "Left dock should be closed" ); assert!( !workspace.right_dock().read(cx).is_open(), "Right dock should be closed" ); assert!( !workspace.bottom_dock().read(cx).is_open(), "Bottom dock should be closed" ); }); // Toggle all docks back on and verify positions are restored workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_all_docks(&ToggleAllDocks, window, cx); assert!( !workspace.left_dock().read(cx).is_open(), "Left dock should remain closed" ); assert!( workspace.right_dock().read(cx).is_open(), "Right dock should remain open" ); assert!( workspace.bottom_dock().read(cx).is_open(), "Bottom dock should remain open" ); assert_eq!( left_panel.read(cx).position, DockPosition::Bottom, "Left panel should remain in the bottom dock" ); assert_eq!( right_panel.read(cx).position, DockPosition::Right, "Right panel should remain in the right dock" ); assert_eq!( workspace .bottom_dock() .read(cx) .visible_panel() .unwrap() .panel_id(), left_panel.panel_id(), "Left panel should be the visible panel in the right dock" ); }); } #[gpui::test] async fn test_join_pane_into_next(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, None, cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); // Let's arrange the panes like this: // // +-----------------------+ // | top | // +------+--------+-------+ // | left | center | right | // +------+--------+-------+ // | bottom | // +-----------------------+ let top_item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "top.txt", cx)]) }); let bottom_item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "bottom.txt", cx)]) }); let left_item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(3, "left.txt", cx)]) }); let right_item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(4, "right.txt", cx)]) }); let center_item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(5, "center.txt", cx)]) }); let top_pane_id = workspace.update_in(cx, |workspace, window, cx| { let top_pane_id = workspace.active_pane().entity_id(); workspace.add_item_to_active_pane(Box::new(top_item.clone()), None, false, window, cx); workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Down, window, cx, ); top_pane_id }); let bottom_pane_id = workspace.update_in(cx, |workspace, window, cx| { let bottom_pane_id = workspace.active_pane().entity_id(); workspace.add_item_to_active_pane( Box::new(bottom_item.clone()), None, false, window, cx, ); workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Up, window, cx, ); bottom_pane_id }); let left_pane_id = workspace.update_in(cx, |workspace, window, cx| { let left_pane_id = workspace.active_pane().entity_id(); workspace.add_item_to_active_pane(Box::new(left_item.clone()), None, false, window, cx); workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Right, window, cx, ); left_pane_id }); let right_pane_id = workspace.update_in(cx, |workspace, window, cx| { let right_pane_id = workspace.active_pane().entity_id(); workspace.add_item_to_active_pane( Box::new(right_item.clone()), None, false, window, cx, ); workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Left, window, cx, ); right_pane_id }); let center_pane_id = workspace.update_in(cx, |workspace, window, cx| { let center_pane_id = workspace.active_pane().entity_id(); workspace.add_item_to_active_pane( Box::new(center_item.clone()), None, false, window, cx, ); center_pane_id }); cx.executor().run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { assert_eq!(center_pane_id, workspace.active_pane().entity_id()); // Join into next from center pane into right workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx); }); workspace.update_in(cx, |workspace, window, cx| { let active_pane = workspace.active_pane(); assert_eq!(right_pane_id, active_pane.entity_id()); assert_eq!(2, active_pane.read(cx).items_len()); let item_ids_in_pane = HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id())); assert!(item_ids_in_pane.contains(¢er_item.item_id())); assert!(item_ids_in_pane.contains(&right_item.item_id())); // Join into next from right pane into bottom workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx); }); workspace.update_in(cx, |workspace, window, cx| { let active_pane = workspace.active_pane(); assert_eq!(bottom_pane_id, active_pane.entity_id()); assert_eq!(3, active_pane.read(cx).items_len()); let item_ids_in_pane = HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id())); assert!(item_ids_in_pane.contains(¢er_item.item_id())); assert!(item_ids_in_pane.contains(&right_item.item_id())); assert!(item_ids_in_pane.contains(&bottom_item.item_id())); // Join into next from bottom pane into left workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx); }); workspace.update_in(cx, |workspace, window, cx| { let active_pane = workspace.active_pane(); assert_eq!(left_pane_id, active_pane.entity_id()); assert_eq!(4, active_pane.read(cx).items_len()); let item_ids_in_pane = HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id())); assert!(item_ids_in_pane.contains(¢er_item.item_id())); assert!(item_ids_in_pane.contains(&right_item.item_id())); assert!(item_ids_in_pane.contains(&bottom_item.item_id())); assert!(item_ids_in_pane.contains(&left_item.item_id())); // Join into next from left pane into top workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx); }); workspace.update_in(cx, |workspace, window, cx| { let active_pane = workspace.active_pane(); assert_eq!(top_pane_id, active_pane.entity_id()); assert_eq!(5, active_pane.read(cx).items_len()); let item_ids_in_pane = HashSet::from_iter(active_pane.read(cx).items().map(|item| item.item_id())); assert!(item_ids_in_pane.contains(¢er_item.item_id())); assert!(item_ids_in_pane.contains(&right_item.item_id())); assert!(item_ids_in_pane.contains(&bottom_item.item_id())); assert!(item_ids_in_pane.contains(&left_item.item_id())); assert!(item_ids_in_pane.contains(&top_item.item_id())); // Single pane left: no-op workspace.join_pane_into_next(workspace.active_pane().clone(), window, cx) }); workspace.update(cx, |workspace, _cx| { let active_pane = workspace.active_pane(); assert_eq!(top_pane_id, active_pane.entity_id()); }); } fn add_an_item_to_active_pane( cx: &mut VisualTestContext, workspace: &Entity, item_id: u64, ) -> Entity { let item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new( item_id, "item{item_id}.txt", cx, )]) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item.clone()), None, false, window, cx); }); item } fn split_pane(cx: &mut VisualTestContext, workspace: &Entity) -> Entity { workspace.update_in(cx, |workspace, window, cx| { workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Right, window, cx, ) }) } #[gpui::test] async fn test_join_all_panes(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, None, cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); add_an_item_to_active_pane(cx, &workspace, 1); split_pane(cx, &workspace); add_an_item_to_active_pane(cx, &workspace, 2); split_pane(cx, &workspace); // empty pane split_pane(cx, &workspace); let last_item = add_an_item_to_active_pane(cx, &workspace, 3); cx.executor().run_until_parked(); workspace.update(cx, |workspace, cx| { let num_panes = workspace.panes().len(); let num_items_in_current_pane = workspace.active_pane().read(cx).items().count(); let active_item = workspace .active_pane() .read(cx) .active_item() .expect("item is in focus"); assert_eq!(num_panes, 4); assert_eq!(num_items_in_current_pane, 1); assert_eq!(active_item.item_id(), last_item.item_id()); }); workspace.update_in(cx, |workspace, window, cx| { workspace.join_all_panes(window, cx); }); workspace.update(cx, |workspace, cx| { let num_panes = workspace.panes().len(); let num_items_in_current_pane = workspace.active_pane().read(cx).items().count(); let active_item = workspace .active_pane() .read(cx) .active_item() .expect("item is in focus"); assert_eq!(num_panes, 1); assert_eq!(num_items_in_current_pane, 3); assert_eq!(active_item.item_id(), last_item.item_id()); }); } #[gpui::test] async fn test_flexible_dock_sizing(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); workspace.update(cx, |workspace, _cx| { workspace.bounds.size.width = px(800.); }); workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new_flexible(DockPosition::Right, 100, cx)); workspace.add_panel(panel, window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); }); let (panel, resized_width, ratio_basis_width) = workspace.update_in(cx, |workspace, window, cx| { let item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "one.txt", cx)]) }); workspace.add_item_to_active_pane(Box::new(item), None, true, window, cx); let dock = workspace.right_dock().read(cx); let workspace_width = workspace.bounds.size.width; let initial_width = workspace .dock_size(&dock, window, cx) .expect("flexible dock should have an initial width"); assert_eq!(initial_width, workspace_width / 2.); workspace.resize_right_dock(px(300.), window, cx); let dock = workspace.right_dock().read(cx); let resized_width = workspace .dock_size(&dock, window, cx) .expect("flexible dock should keep its resized width"); assert_eq!(resized_width, px(300.)); let panel = workspace .right_dock() .read(cx) .visible_panel() .expect("flexible dock should have a visible panel") .panel_id(); (panel, resized_width, workspace_width) }); workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_dock(DockPosition::Right, window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); let dock = workspace.right_dock().read(cx); let reopened_width = workspace .dock_size(&dock, window, cx) .expect("flexible dock should restore when reopened"); assert_eq!(reopened_width, resized_width); let right_dock = workspace.right_dock().read(cx); let flexible_panel = right_dock .visible_panel() .expect("flexible dock should still have a visible panel"); assert_eq!(flexible_panel.panel_id(), panel); assert_eq!( right_dock .stored_panel_size_state(flexible_panel.as_ref()) .and_then(|size_state| size_state.flex), Some( resized_width.to_f64() as f32 / (workspace.bounds.size.width - resized_width).to_f64() as f32 ) ); }); workspace.update_in(cx, |workspace, window, cx| { workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Right, window, cx, ); let dock = workspace.right_dock().read(cx); let split_width = workspace .dock_size(&dock, window, cx) .expect("flexible dock should keep its user-resized proportion"); assert_eq!(split_width, px(300.)); workspace.bounds.size.width = px(1600.); let dock = workspace.right_dock().read(cx); let resized_window_width = workspace .dock_size(&dock, window, cx) .expect("flexible dock should preserve proportional size on window resize"); assert_eq!( resized_window_width, workspace.bounds.size.width * (resized_width.to_f64() as f32 / ratio_basis_width.to_f64() as f32) ); }); } #[gpui::test] async fn test_panel_size_state_persistence(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); // Fixed-width panel: pixel size is persisted to KVP and restored on re-add. { let project = Project::test(fs.clone(), [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); workspace.update(cx, |workspace, _cx| { workspace.set_random_database_id(); workspace.bounds.size.width = px(800.); }); let panel = workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Left, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace.toggle_dock(DockPosition::Left, window, cx); panel }); workspace.update_in(cx, |workspace, window, cx| { workspace.resize_left_dock(px(350.), window, cx); }); cx.run_until_parked(); let persisted = workspace.read_with(cx, |workspace, cx| { workspace.persisted_panel_size_state(TestPanel::panel_key(), cx) }); assert_eq!( persisted.and_then(|s| s.size), Some(px(350.)), "fixed-width panel size should be persisted to KVP" ); // Remove the panel and re-add a fresh instance with the same key. // The new instance should have its size state restored from KVP. workspace.update_in(cx, |workspace, window, cx| { workspace.remove_panel(&panel, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { let new_panel = cx.new(|cx| TestPanel::new(DockPosition::Left, 100, cx)); workspace.add_panel(new_panel, window, cx); let left_dock = workspace.left_dock().read(cx); let size_state = left_dock .panel::() .and_then(|p| left_dock.stored_panel_size_state(&p)); assert_eq!( size_state.and_then(|s| s.size), Some(px(350.)), "re-added fixed-width panel should restore persisted size from KVP" ); }); } // Flexible panel: both pixel size and ratio are persisted and restored. { let project = Project::test(fs.clone(), [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); workspace.update(cx, |workspace, _cx| { workspace.set_random_database_id(); workspace.bounds.size.width = px(800.); }); let panel = workspace.update_in(cx, |workspace, window, cx| { let item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "one.txt", cx)]) }); workspace.add_item_to_active_pane(Box::new(item), None, true, window, cx); let panel = cx.new(|cx| TestPanel::new_flexible(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); panel }); workspace.update_in(cx, |workspace, window, cx| { workspace.resize_right_dock(px(300.), window, cx); }); cx.run_until_parked(); let persisted = workspace .read_with(cx, |workspace, cx| { workspace.persisted_panel_size_state(TestPanel::panel_key(), cx) }) .expect("flexible panel state should be persisted to KVP"); assert_eq!( persisted.size, None, "flexible panel should not persist a redundant pixel size" ); let original_ratio = persisted.flex.expect("panel's flex should be persisted"); // Remove the panel and re-add: both size and ratio should be restored. workspace.update_in(cx, |workspace, window, cx| { workspace.remove_panel(&panel, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { let new_panel = cx.new(|cx| TestPanel::new_flexible(DockPosition::Right, 100, cx)); workspace.add_panel(new_panel, window, cx); let right_dock = workspace.right_dock().read(cx); let size_state = right_dock .panel::() .and_then(|p| right_dock.stored_panel_size_state(&p)) .expect("re-added flexible panel should have restored size state from KVP"); assert_eq!( size_state.size, None, "re-added flexible panel should not have a persisted pixel size" ); assert_eq!( size_state.flex, Some(original_ratio), "re-added flexible panel should restore persisted flex" ); }); } } #[gpui::test] async fn test_flexible_panel_left_dock_sizing(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); workspace.update(cx, |workspace, _cx| { workspace.bounds.size.width = px(900.); }); // Step 1: Add a tab to the center pane then open a flexible panel in the left // dock. With one full-width center pane the default ratio is 0.5, so the panel // and the center pane each take half the workspace width. workspace.update_in(cx, |workspace, window, cx| { let item = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "one.txt", cx)]) }); workspace.add_item_to_active_pane(Box::new(item), None, true, window, cx); let panel = cx.new(|cx| TestPanel::new_flexible(DockPosition::Left, 100, cx)); workspace.add_panel(panel, window, cx); workspace.toggle_dock(DockPosition::Left, window, cx); let left_dock = workspace.left_dock().read(cx); let left_width = workspace .dock_size(&left_dock, window, cx) .expect("left dock should have an active panel"); assert_eq!( left_width, workspace.bounds.size.width / 2., "flexible left panel should split evenly with the center pane" ); }); // Step 2: Split the center pane vertically (top/bottom). Vertical splits do not // change horizontal width fractions, so the flexible panel stays at the same // width as each half of the split. workspace.update_in(cx, |workspace, window, cx| { workspace.split_pane( workspace.active_pane().clone(), SplitDirection::Down, window, cx, ); let left_dock = workspace.left_dock().read(cx); let left_width = workspace .dock_size(&left_dock, window, cx) .expect("left dock should still have an active panel after vertical split"); assert_eq!( left_width, workspace.bounds.size.width / 2., "flexible left panel width should match each vertically-split pane" ); }); // Step 3: Open a fixed-width panel in the right dock. The right dock's default // size reduces the available width, so the flexible left panel and the center // panes all shrink proportionally to accommodate it. workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 200, cx)); workspace.add_panel(panel, window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); let right_dock = workspace.right_dock().read(cx); let right_width = workspace .dock_size(&right_dock, window, cx) .expect("right dock should have an active panel"); let left_dock = workspace.left_dock().read(cx); let left_width = workspace .dock_size(&left_dock, window, cx) .expect("left dock should still have an active panel"); let available_width = workspace.bounds.size.width - right_width; assert_eq!( left_width, available_width / 2., "flexible left panel should shrink proportionally as the right dock takes space" ); }); // Step 4: Toggle the right dock's panel to flexible. Now both docks use // flex sizing and the workspace width is divided among left-flex, center // (implicit flex 1.0), and right-flex. workspace.update_in(cx, |workspace, window, cx| { let right_dock = workspace.right_dock().clone(); let right_panel = right_dock .read(cx) .visible_panel() .expect("right dock should have a visible panel") .clone(); workspace.toggle_dock_panel_flexible_size( &right_dock, right_panel.as_ref(), window, cx, ); let right_dock = right_dock.read(cx); let right_panel = right_dock .visible_panel() .expect("right dock should still have a visible panel"); assert!( right_panel.has_flexible_size(window, cx), "right panel should now be flexible" ); let right_size_state = right_dock .stored_panel_size_state(right_panel.as_ref()) .expect("right panel should have a stored size state after toggling"); let right_flex = right_size_state .flex .expect("right panel should have a flex value after toggling"); let left_dock = workspace.left_dock().read(cx); let left_width = workspace .dock_size(&left_dock, window, cx) .expect("left dock should still have an active panel"); let right_width = workspace .dock_size(&right_dock, window, cx) .expect("right dock should still have an active panel"); let left_flex = workspace .default_dock_flex(DockPosition::Left) .expect("left dock should have a default flex"); let total_flex = left_flex + 1.0 + right_flex; let expected_left = left_flex / total_flex * workspace.bounds.size.width; let expected_right = right_flex / total_flex * workspace.bounds.size.width; assert_eq!( left_width, expected_left, "flexible left panel should share workspace width via flex ratios" ); assert_eq!( right_width, expected_right, "flexible right panel should share workspace width via flex ratios" ); }); } struct TestModal(FocusHandle); impl TestModal { fn new(_: &mut Window, cx: &mut Context) -> Self { Self(cx.focus_handle()) } } impl EventEmitter for TestModal {} impl Focusable for TestModal { fn focus_handle(&self, _cx: &App) -> FocusHandle { self.0.clone() } } impl ModalView for TestModal {} impl Render for TestModal { fn render( &mut self, _window: &mut Window, _cx: &mut Context, ) -> impl IntoElement { div().track_focus(&self.0) } } #[gpui::test] async fn test_panels(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); let (panel_1, panel_2) = workspace.update_in(cx, |workspace, window, cx| { let panel_1 = cx.new(|cx| TestPanel::new(DockPosition::Left, 100, cx)); workspace.add_panel(panel_1.clone(), window, cx); workspace.toggle_dock(DockPosition::Left, window, cx); let panel_2 = cx.new(|cx| TestPanel::new(DockPosition::Right, 101, cx)); workspace.add_panel(panel_2.clone(), window, cx); workspace.toggle_dock(DockPosition::Right, window, cx); let left_dock = workspace.left_dock(); assert_eq!( left_dock.read(cx).visible_panel().unwrap().panel_id(), panel_1.panel_id() ); assert_eq!( workspace.dock_size(&left_dock.read(cx), window, cx), Some(px(300.)) ); workspace.resize_left_dock(px(1337.), window, cx); assert_eq!( workspace .right_dock() .read(cx) .visible_panel() .unwrap() .panel_id(), panel_2.panel_id(), ); (panel_1, panel_2) }); // Move panel_1 to the right panel_1.update_in(cx, |panel_1, window, cx| { panel_1.set_position(DockPosition::Right, window, cx) }); workspace.update_in(cx, |workspace, window, cx| { // Since panel_1 was visible on the left, it should now be visible now that it's been moved to the right. // Since it was the only panel on the left, the left dock should now be closed. assert!(!workspace.left_dock().read(cx).is_open()); assert!(workspace.left_dock().read(cx).visible_panel().is_none()); let right_dock = workspace.right_dock(); assert_eq!( right_dock.read(cx).visible_panel().unwrap().panel_id(), panel_1.panel_id() ); assert_eq!( right_dock .read(cx) .active_panel_size() .unwrap() .size .unwrap(), px(1337.) ); // Now we move panel_2 to the left panel_2.set_position(DockPosition::Left, window, cx); }); workspace.update(cx, |workspace, cx| { // Since panel_2 was not visible on the right, we don't open the left dock. assert!(!workspace.left_dock().read(cx).is_open()); // And the right dock is unaffected in its displaying of panel_1 assert!(workspace.right_dock().read(cx).is_open()); assert_eq!( workspace .right_dock() .read(cx) .visible_panel() .unwrap() .panel_id(), panel_1.panel_id(), ); }); // Move panel_1 back to the left panel_1.update_in(cx, |panel_1, window, cx| { panel_1.set_position(DockPosition::Left, window, cx) }); workspace.update_in(cx, |workspace, window, cx| { // Since panel_1 was visible on the right, we open the left dock and make panel_1 active. let left_dock = workspace.left_dock(); assert!(left_dock.read(cx).is_open()); assert_eq!( left_dock.read(cx).visible_panel().unwrap().panel_id(), panel_1.panel_id() ); assert_eq!( workspace.dock_size(&left_dock.read(cx), window, cx), Some(px(1337.)) ); // And the right dock should be closed as it no longer has any panels. assert!(!workspace.right_dock().read(cx).is_open()); // Now we move panel_1 to the bottom panel_1.set_position(DockPosition::Bottom, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { // Since panel_1 was visible on the left, we close the left dock. assert!(!workspace.left_dock().read(cx).is_open()); // The bottom dock is sized based on the panel's default size, // since the panel orientation changed from vertical to horizontal. let bottom_dock = workspace.bottom_dock(); assert_eq!( workspace.dock_size(&bottom_dock.read(cx), window, cx), Some(px(300.)) ); // Close bottom dock and move panel_1 back to the left. bottom_dock.update(cx, |bottom_dock, cx| { bottom_dock.set_open(false, window, cx) }); panel_1.set_position(DockPosition::Left, window, cx); }); // Emit activated event on panel 1 panel_1.update(cx, |_, cx| cx.emit(PanelEvent::Activate)); // Now the left dock is open and panel_1 is active and focused. workspace.update_in(cx, |workspace, window, cx| { let left_dock = workspace.left_dock(); assert!(left_dock.read(cx).is_open()); assert_eq!( left_dock.read(cx).visible_panel().unwrap().panel_id(), panel_1.panel_id(), ); assert!(panel_1.focus_handle(cx).is_focused(window)); }); // Emit closed event on panel 2, which is not active panel_2.update(cx, |_, cx| cx.emit(PanelEvent::Close)); // Wo don't close the left dock, because panel_2 wasn't the active panel workspace.update(cx, |workspace, cx| { let left_dock = workspace.left_dock(); assert!(left_dock.read(cx).is_open()); assert_eq!( left_dock.read(cx).visible_panel().unwrap().panel_id(), panel_1.panel_id(), ); }); // Emitting a ZoomIn event shows the panel as zoomed. panel_1.update(cx, |_, cx| cx.emit(PanelEvent::ZoomIn)); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade())); assert_eq!(workspace.zoomed_position, Some(DockPosition::Left)); }); // Move panel to another dock while it is zoomed panel_1.update_in(cx, |panel, window, cx| { panel.set_position(DockPosition::Right, window, cx) }); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade())); assert_eq!(workspace.zoomed_position, Some(DockPosition::Right)); }); // This is a helper for getting a: // - valid focus on an element, // - that isn't a part of the panes and panels system of the Workspace, // - and doesn't trigger the 'on_focus_lost' API. let focus_other_view = { let workspace = workspace.clone(); move |cx: &mut VisualTestContext| { workspace.update_in(cx, |workspace, window, cx| { if workspace.active_modal::(cx).is_some() { workspace.toggle_modal(window, cx, TestModal::new); workspace.toggle_modal(window, cx, TestModal::new); } else { workspace.toggle_modal(window, cx, TestModal::new); } }) } }; // If focus is transferred to another view that's not a panel or another pane, we still show // the panel as zoomed. focus_other_view(cx); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade())); assert_eq!(workspace.zoomed_position, Some(DockPosition::Right)); }); // If focus is transferred elsewhere in the workspace, the panel is no longer zoomed. workspace.update_in(cx, |_workspace, window, cx| { cx.focus_self(window); }); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, None); assert_eq!(workspace.zoomed_position, None); }); // If focus is transferred again to another view that's not a panel or a pane, we won't // show the panel as zoomed because it wasn't zoomed before. focus_other_view(cx); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, None); assert_eq!(workspace.zoomed_position, None); }); // When the panel is activated, it is zoomed again. cx.dispatch_action(ToggleRightDock); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, Some(panel_1.to_any().downgrade())); assert_eq!(workspace.zoomed_position, Some(DockPosition::Right)); }); // Emitting a ZoomOut event unzooms the panel. panel_1.update(cx, |_, cx| cx.emit(PanelEvent::ZoomOut)); workspace.read_with(cx, |workspace, _| { assert_eq!(workspace.zoomed, None); assert_eq!(workspace.zoomed_position, None); }); // Emit closed event on panel 1, which is active panel_1.update(cx, |_, cx| cx.emit(PanelEvent::Close)); // Now the left dock is closed, because panel_1 was the active panel workspace.update(cx, |workspace, cx| { let right_dock = workspace.right_dock(); assert!(!right_dock.read(cx).is_open()); }); } #[gpui::test] async fn test_no_save_prompt_when_multi_buffer_dirty_items_closed(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let dirty_regular_buffer = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("1.txt") .with_project_items(&[dirty_project_item(1, "1.txt", cx)]) }); let dirty_regular_buffer_2 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("2.txt") .with_project_items(&[dirty_project_item(2, "2.txt", cx)]) }); let dirty_multi_buffer_with_both = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_buffer_kind(ItemBufferKind::Multibuffer) .with_label("Fake Project Search") .with_project_items(&[ dirty_regular_buffer.read(cx).project_items[0].clone(), dirty_regular_buffer_2.read(cx).project_items[0].clone(), ]) }); let multi_buffer_with_both_files_id = dirty_multi_buffer_with_both.item_id(); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(dirty_regular_buffer.clone()), None, false, false, window, cx, ); workspace.add_item( pane.clone(), Box::new(dirty_regular_buffer_2.clone()), None, false, false, window, cx, ); workspace.add_item( pane.clone(), Box::new(dirty_multi_buffer_with_both.clone()), None, false, false, window, cx, ); }); pane.update_in(cx, |pane, window, cx| { pane.activate_item(2, true, true, window, cx); assert_eq!( pane.active_item().unwrap().item_id(), multi_buffer_with_both_files_id, "Should select the multi buffer in the pane" ); }); let close_all_but_multi_buffer_task = pane.update_in(cx, |pane, window, cx| { pane.close_other_items( &CloseOtherItems { save_intent: Some(SaveIntent::Save), close_pinned: true, }, None, window, cx, ) }); cx.background_executor.run_until_parked(); assert!(!cx.has_pending_prompt()); close_all_but_multi_buffer_task .await .expect("Closing all buffers but the multi buffer failed"); pane.update(cx, |pane, cx| { assert_eq!(dirty_regular_buffer.read(cx).save_count, 1); assert_eq!(dirty_multi_buffer_with_both.read(cx).save_count, 0); assert_eq!(dirty_regular_buffer_2.read(cx).save_count, 1); assert_eq!(pane.items_len(), 1); assert_eq!( pane.active_item().unwrap().item_id(), multi_buffer_with_both_files_id, "Should have only the multi buffer left in the pane" ); assert!( dirty_multi_buffer_with_both.read(cx).is_dirty, "The multi buffer containing the unsaved buffer should still be dirty" ); }); dirty_regular_buffer.update(cx, |buffer, cx| { buffer.project_items[0].update(cx, |pi, _| pi.is_dirty = true) }); let close_multi_buffer_task = pane.update_in(cx, |pane, window, cx| { pane.close_active_item( &CloseActiveItem { save_intent: Some(SaveIntent::Close), close_pinned: false, }, window, cx, ) }); cx.background_executor.run_until_parked(); assert!( cx.has_pending_prompt(), "Dirty multi buffer should prompt a save dialog" ); cx.simulate_prompt_answer("Save"); cx.background_executor.run_until_parked(); close_multi_buffer_task .await .expect("Closing the multi buffer failed"); pane.update(cx, |pane, cx| { assert_eq!( dirty_multi_buffer_with_both.read(cx).save_count, 1, "Multi buffer item should get be saved" ); // Test impl does not save inner items, so we do not assert them assert_eq!( pane.items_len(), 0, "No more items should be left in the pane" ); assert!(pane.active_item().is_none()); }); } #[gpui::test] async fn test_save_prompt_when_dirty_multi_buffer_closed_with_some_of_its_dirty_items_not_present_in_the_pane( cx: &mut TestAppContext, ) { init_test(cx); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let dirty_regular_buffer = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("1.txt") .with_project_items(&[dirty_project_item(1, "1.txt", cx)]) }); let dirty_regular_buffer_2 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("2.txt") .with_project_items(&[dirty_project_item(2, "2.txt", cx)]) }); let clear_regular_buffer = cx.new(|cx| { TestItem::new(cx) .with_label("3.txt") .with_project_items(&[TestProjectItem::new(3, "3.txt", cx)]) }); let dirty_multi_buffer_with_both = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_buffer_kind(ItemBufferKind::Multibuffer) .with_label("Fake Project Search") .with_project_items(&[ dirty_regular_buffer.read(cx).project_items[0].clone(), dirty_regular_buffer_2.read(cx).project_items[0].clone(), clear_regular_buffer.read(cx).project_items[0].clone(), ]) }); let multi_buffer_with_both_files_id = dirty_multi_buffer_with_both.item_id(); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(dirty_regular_buffer.clone()), None, false, false, window, cx, ); workspace.add_item( pane.clone(), Box::new(dirty_multi_buffer_with_both.clone()), None, false, false, window, cx, ); }); pane.update_in(cx, |pane, window, cx| { pane.activate_item(1, true, true, window, cx); assert_eq!( pane.active_item().unwrap().item_id(), multi_buffer_with_both_files_id, "Should select the multi buffer in the pane" ); }); let _close_multi_buffer_task = pane.update_in(cx, |pane, window, cx| { pane.close_active_item( &CloseActiveItem { save_intent: None, close_pinned: false, }, window, cx, ) }); cx.background_executor.run_until_parked(); assert!( cx.has_pending_prompt(), "With one dirty item from the multi buffer not being in the pane, a save prompt should be shown" ); } /// Tests that when `close_on_file_delete` is enabled, files are automatically /// closed when they are deleted from disk. #[gpui::test] async fn test_close_on_disk_deletion_enabled(cx: &mut TestAppContext) { init_test(cx); // Enable the close_on_disk_deletion setting cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.workspace.close_on_file_delete = Some(true); }); }); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); // Create a test item that simulates a file let item = cx.new(|cx| { TestItem::new(cx) .with_label("test.txt") .with_project_items(&[TestProjectItem::new(1, "test.txt", cx)]) }); // Add item to workspace workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(item.clone()), None, false, false, window, cx, ); }); // Verify the item is in the pane pane.read_with(cx, |pane, _| { assert_eq!(pane.items().count(), 1); }); // Simulate file deletion by setting the item's deleted state item.update(cx, |item, _| { item.set_has_deleted_file(true); }); // Emit UpdateTab event to trigger the close behavior cx.run_until_parked(); item.update(cx, |_, cx| { cx.emit(ItemEvent::UpdateTab); }); // Allow the close operation to complete cx.run_until_parked(); // Verify the item was automatically closed pane.read_with(cx, |pane, _| { assert_eq!( pane.items().count(), 0, "Item should be automatically closed when file is deleted" ); }); } /// Tests that when `close_on_file_delete` is disabled (default), files remain /// open with a strikethrough when they are deleted from disk. #[gpui::test] async fn test_close_on_disk_deletion_disabled(cx: &mut TestAppContext) { init_test(cx); // Ensure close_on_disk_deletion is disabled (default) cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.workspace.close_on_file_delete = Some(false); }); }); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); // Create a test item that simulates a file let item = cx.new(|cx| { TestItem::new(cx) .with_label("test.txt") .with_project_items(&[TestProjectItem::new(1, "test.txt", cx)]) }); // Add item to workspace workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(item.clone()), None, false, false, window, cx, ); }); // Verify the item is in the pane pane.read_with(cx, |pane, _| { assert_eq!(pane.items().count(), 1); }); // Simulate file deletion item.update(cx, |item, _| { item.set_has_deleted_file(true); }); // Emit UpdateTab event cx.run_until_parked(); item.update(cx, |_, cx| { cx.emit(ItemEvent::UpdateTab); }); // Allow any potential close operation to complete cx.run_until_parked(); // Verify the item remains open (with strikethrough) pane.read_with(cx, |pane, _| { assert_eq!( pane.items().count(), 1, "Item should remain open when close_on_disk_deletion is disabled" ); }); // Verify the item shows as deleted item.read_with(cx, |item, _| { assert!( item.has_deleted_file, "Item should be marked as having deleted file" ); }); } /// Tests that dirty files are not automatically closed when deleted from disk, /// even when `close_on_file_delete` is enabled. This ensures users don't lose /// unsaved changes without being prompted. #[gpui::test] async fn test_close_on_disk_deletion_with_dirty_file(cx: &mut TestAppContext) { init_test(cx); // Enable the close_on_file_delete setting cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.workspace.close_on_file_delete = Some(true); }); }); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); // Create a dirty test item let item = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("test.txt") .with_project_items(&[TestProjectItem::new(1, "test.txt", cx)]) }); // Add item to workspace workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(item.clone()), None, false, false, window, cx, ); }); // Simulate file deletion item.update(cx, |item, _| { item.set_has_deleted_file(true); }); // Emit UpdateTab event to trigger the close behavior cx.run_until_parked(); item.update(cx, |_, cx| { cx.emit(ItemEvent::UpdateTab); }); // Allow any potential close operation to complete cx.run_until_parked(); // Verify the item remains open (dirty files are not auto-closed) pane.read_with(cx, |pane, _| { assert_eq!( pane.items().count(), 1, "Dirty items should not be automatically closed even when file is deleted" ); }); // Verify the item is marked as deleted and still dirty item.read_with(cx, |item, _| { assert!( item.has_deleted_file, "Item should be marked as having deleted file" ); assert!(item.is_dirty, "Item should still be dirty"); }); } /// Tests that navigation history is cleaned up when files are auto-closed /// due to deletion from disk. #[gpui::test] async fn test_close_on_disk_deletion_cleans_navigation_history(cx: &mut TestAppContext) { init_test(cx); // Enable the close_on_file_delete setting cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.workspace.close_on_file_delete = Some(true); }); }); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); // Create test items let item1 = cx.new(|cx| { TestItem::new(cx) .with_label("test1.txt") .with_project_items(&[TestProjectItem::new(1, "test1.txt", cx)]) }); let item1_id = item1.item_id(); let item2 = cx.new(|cx| { TestItem::new(cx) .with_label("test2.txt") .with_project_items(&[TestProjectItem::new(2, "test2.txt", cx)]) }); // Add items to workspace workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(item1.clone()), None, false, false, window, cx, ); workspace.add_item( pane.clone(), Box::new(item2.clone()), None, false, false, window, cx, ); }); // Activate item1 to ensure it gets navigation entries pane.update_in(cx, |pane, window, cx| { pane.activate_item(0, true, true, window, cx); }); // Switch to item2 and back to create navigation history pane.update_in(cx, |pane, window, cx| { pane.activate_item(1, true, true, window, cx); }); cx.run_until_parked(); pane.update_in(cx, |pane, window, cx| { pane.activate_item(0, true, true, window, cx); }); cx.run_until_parked(); // Simulate file deletion for item1 item1.update(cx, |item, _| { item.set_has_deleted_file(true); }); // Emit UpdateTab event to trigger the close behavior item1.update(cx, |_, cx| { cx.emit(ItemEvent::UpdateTab); }); cx.run_until_parked(); // Verify item1 was closed pane.read_with(cx, |pane, _| { assert_eq!( pane.items().count(), 1, "Should have 1 item remaining after auto-close" ); }); // Check navigation history after close let has_item = pane.read_with(cx, |pane, cx| { let mut has_item = false; pane.nav_history().for_each_entry(cx, &mut |entry, _| { if entry.item.id() == item1_id { has_item = true; } }); has_item }); assert!( !has_item, "Navigation history should not contain closed item entries" ); } #[gpui::test] async fn test_no_save_prompt_when_dirty_multi_buffer_closed_with_all_of_its_dirty_items_present_in_the_pane( cx: &mut TestAppContext, ) { init_test(cx); let fs = FakeFs::new(cx.background_executor.clone()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let dirty_regular_buffer = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("1.txt") .with_project_items(&[dirty_project_item(1, "1.txt", cx)]) }); let dirty_regular_buffer_2 = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_label("2.txt") .with_project_items(&[dirty_project_item(2, "2.txt", cx)]) }); let clear_regular_buffer = cx.new(|cx| { TestItem::new(cx) .with_label("3.txt") .with_project_items(&[TestProjectItem::new(3, "3.txt", cx)]) }); let dirty_multi_buffer = cx.new(|cx| { TestItem::new(cx) .with_dirty(true) .with_buffer_kind(ItemBufferKind::Multibuffer) .with_label("Fake Project Search") .with_project_items(&[ dirty_regular_buffer.read(cx).project_items[0].clone(), dirty_regular_buffer_2.read(cx).project_items[0].clone(), clear_regular_buffer.read(cx).project_items[0].clone(), ]) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item( pane.clone(), Box::new(dirty_regular_buffer.clone()), None, false, false, window, cx, ); workspace.add_item( pane.clone(), Box::new(dirty_regular_buffer_2.clone()), None, false, false, window, cx, ); workspace.add_item( pane.clone(), Box::new(dirty_multi_buffer.clone()), None, false, false, window, cx, ); }); pane.update_in(cx, |pane, window, cx| { pane.activate_item(2, true, true, window, cx); assert_eq!( pane.active_item().unwrap().item_id(), dirty_multi_buffer.item_id(), "Should select the multi buffer in the pane" ); }); let close_multi_buffer_task = pane.update_in(cx, |pane, window, cx| { pane.close_active_item( &CloseActiveItem { save_intent: None, close_pinned: false, }, window, cx, ) }); cx.background_executor.run_until_parked(); assert!( !cx.has_pending_prompt(), "All dirty items from the multi buffer are in the pane still, no save prompts should be shown" ); close_multi_buffer_task .await .expect("Closing multi buffer failed"); pane.update(cx, |pane, cx| { assert_eq!(dirty_regular_buffer.read(cx).save_count, 0); assert_eq!(dirty_multi_buffer.read(cx).save_count, 0); assert_eq!(dirty_regular_buffer_2.read(cx).save_count, 0); assert_eq!( pane.items() .map(|item| item.item_id()) .sorted() .collect::>(), vec![ dirty_regular_buffer.item_id(), dirty_regular_buffer_2.item_id(), ], "Should have no multi buffer left in the pane" ); assert!(dirty_regular_buffer.read(cx).is_dirty); assert!(dirty_regular_buffer_2.read(cx).is_dirty); }); } #[gpui::test] async fn test_move_focused_panel_to_next_position(cx: &mut gpui::TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); // Add a new panel to the right dock, opening the dock and setting the // focus to the new panel. let panel = workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace .right_dock() .update(cx, |right_dock, cx| right_dock.set_open(true, window, cx)); workspace.toggle_panel_focus::(window, cx); panel }); // Dispatch the `MoveFocusedPanelToNextPosition` action, moving the // panel to the next valid position which, in this case, is the left // dock. cx.dispatch_action(MoveFocusedPanelToNextPosition); workspace.update(cx, |workspace, cx| { assert!(workspace.left_dock().read(cx).is_open()); assert_eq!(panel.read(cx).position, DockPosition::Left); }); // Dispatch the `MoveFocusedPanelToNextPosition` action, moving the // panel to the next valid position which, in this case, is the bottom // dock. cx.dispatch_action(MoveFocusedPanelToNextPosition); workspace.update(cx, |workspace, cx| { assert!(workspace.bottom_dock().read(cx).is_open()); assert_eq!(panel.read(cx).position, DockPosition::Bottom); }); // Dispatch the `MoveFocusedPanelToNextPosition` action again, this time // around moving the panel to its initial position, the right dock. cx.dispatch_action(MoveFocusedPanelToNextPosition); workspace.update(cx, |workspace, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert_eq!(panel.read(cx).position, DockPosition::Right); }); // Remove focus from the panel, ensuring that, if the panel is not // focused, the `MoveFocusedPanelToNextPosition` action does not update // the panel's position, so the panel is still in the right dock. workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); cx.dispatch_action(MoveFocusedPanelToNextPosition); workspace.update(cx, |workspace, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert_eq!(panel.read(cx).position, DockPosition::Right); }); } #[gpui::test] async fn test_moving_items_create_panes(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let item_1 = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "first.txt", cx)]) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item_1), None, true, window, cx); workspace.move_item_to_pane_in_direction( &MoveItemToPaneInDirection { direction: SplitDirection::Right, focus: true, clone: false, }, window, cx, ); workspace.move_item_to_pane_at_index( &MoveItemToPane { destination: 3, focus: true, clone: false, }, window, cx, ); assert_eq!(workspace.panes.len(), 1, "No new panes were created"); assert_eq!( pane_items_paths(&workspace.active_pane, cx), vec!["first.txt".to_string()], "Single item was not moved anywhere" ); }); let item_2 = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(2, "second.txt", cx)]) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item_2), None, true, window, cx); assert_eq!( pane_items_paths(&workspace.panes[0], cx), vec!["first.txt".to_string(), "second.txt".to_string()], ); workspace.move_item_to_pane_in_direction( &MoveItemToPaneInDirection { direction: SplitDirection::Right, focus: true, clone: false, }, window, cx, ); assert_eq!(workspace.panes.len(), 2, "A new pane should be created"); assert_eq!( pane_items_paths(&workspace.panes[0], cx), vec!["first.txt".to_string()], "After moving, one item should be left in the original pane" ); assert_eq!( pane_items_paths(&workspace.panes[1], cx), vec!["second.txt".to_string()], "New item should have been moved to the new pane" ); }); let item_3 = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(3, "third.txt", cx)]) }); workspace.update_in(cx, |workspace, window, cx| { let original_pane = workspace.panes[0].clone(); workspace.set_active_pane(&original_pane, window, cx); workspace.add_item_to_active_pane(Box::new(item_3), None, true, window, cx); assert_eq!(workspace.panes.len(), 2, "No new panes were created"); assert_eq!( pane_items_paths(&workspace.active_pane, cx), vec!["first.txt".to_string(), "third.txt".to_string()], "New pane should be ready to move one item out" ); workspace.move_item_to_pane_at_index( &MoveItemToPane { destination: 3, focus: true, clone: false, }, window, cx, ); assert_eq!(workspace.panes.len(), 3, "A new pane should be created"); assert_eq!( pane_items_paths(&workspace.active_pane, cx), vec!["first.txt".to_string()], "After moving, one item should be left in the original pane" ); assert_eq!( pane_items_paths(&workspace.panes[1], cx), vec!["second.txt".to_string()], "Previously created pane should be unchanged" ); assert_eq!( pane_items_paths(&workspace.panes[2], cx), vec!["third.txt".to_string()], "New item should have been moved to the new pane" ); }); } #[gpui::test] async fn test_moving_items_can_clone_panes(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let item_1 = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "first.txt", cx)]) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item_1), None, true, window, cx); workspace.move_item_to_pane_in_direction( &MoveItemToPaneInDirection { direction: SplitDirection::Right, focus: true, clone: true, }, window, cx, ); }); cx.run_until_parked(); workspace.update_in(cx, |workspace, window, cx| { workspace.move_item_to_pane_at_index( &MoveItemToPane { destination: 3, focus: true, clone: true, }, window, cx, ); }); cx.run_until_parked(); workspace.update(cx, |workspace, cx| { assert_eq!(workspace.panes.len(), 3, "Two new panes were created"); for pane in workspace.panes() { assert_eq!( pane_items_paths(pane, cx), vec!["first.txt".to_string()], "Single item exists in all panes" ); } }); // verify that the active pane has been updated after waiting for the // pane focus event to fire and resolve workspace.read_with(cx, |workspace, _app| { assert_eq!( workspace.active_pane(), &workspace.panes[2], "The third pane should be the active one: {:?}", workspace.panes ); }) } #[gpui::test] async fn test_close_item_in_all_panes(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); fs.insert_tree("/root", json!({ "test.txt": "" })).await; let project = Project::test(fs, ["root".as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let pane_a = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); // Add item to pane A with project path let item_a = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "test.txt", cx)]) }); workspace.update_in(cx, |workspace, window, cx| { workspace.add_item_to_active_pane(Box::new(item_a.clone()), None, true, window, cx) }); // Split to create pane B let pane_b = workspace.update_in(cx, |workspace, window, cx| { workspace.split_pane(pane_a.clone(), SplitDirection::Right, window, cx) }); // Add item with SAME project path to pane B, and pin it let item_b = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "test.txt", cx)]) }); pane_b.update_in(cx, |pane, window, cx| { pane.add_item(Box::new(item_b.clone()), true, true, None, window, cx); pane.set_pinned_count(1); }); assert_eq!(pane_a.read_with(cx, |pane, _| pane.items_len()), 1); assert_eq!(pane_b.read_with(cx, |pane, _| pane.items_len()), 1); // close_pinned: false should only close the unpinned copy workspace.update_in(cx, |workspace, window, cx| { workspace.close_item_in_all_panes( &CloseItemInAllPanes { save_intent: Some(SaveIntent::Close), close_pinned: false, }, window, cx, ) }); cx.executor().run_until_parked(); let item_count_a = pane_a.read_with(cx, |pane, _| pane.items_len()); let item_count_b = pane_b.read_with(cx, |pane, _| pane.items_len()); assert_eq!(item_count_a, 0, "Unpinned item in pane A should be closed"); assert_eq!(item_count_b, 1, "Pinned item in pane B should remain"); // Split again, seeing as closing the previous item also closed its // pane, so only pane remains, which does not allow us to properly test // that both items close when `close_pinned: true`. let pane_c = workspace.update_in(cx, |workspace, window, cx| { workspace.split_pane(pane_b.clone(), SplitDirection::Right, window, cx) }); // Add an item with the same project path to pane C so that // close_item_in_all_panes can determine what to close across all panes // (it reads the active item from the active pane, and split_pane // creates an empty pane). let item_c = cx.new(|cx| { TestItem::new(cx).with_project_items(&[TestProjectItem::new(1, "test.txt", cx)]) }); pane_c.update_in(cx, |pane, window, cx| { pane.add_item(Box::new(item_c.clone()), true, true, None, window, cx); }); // close_pinned: true should close the pinned copy too workspace.update_in(cx, |workspace, window, cx| { let panes_count = workspace.panes().len(); assert_eq!(panes_count, 2, "Workspace should have two panes (B and C)"); workspace.close_item_in_all_panes( &CloseItemInAllPanes { save_intent: Some(SaveIntent::Close), close_pinned: true, }, window, cx, ) }); cx.executor().run_until_parked(); let item_count_b = pane_b.read_with(cx, |pane, _| pane.items_len()); let item_count_c = pane_c.read_with(cx, |pane, _| pane.items_len()); assert_eq!(item_count_b, 0, "Pinned item in pane B should be closed"); assert_eq!(item_count_c, 0, "Unpinned item in pane C should be closed"); } mod register_project_item_tests { use super::*; // View struct TestPngItemView { focus_handle: FocusHandle, } // Model struct TestPngItem {} impl project::ProjectItem for TestPngItem { fn try_open( _project: &Entity, path: &ProjectPath, cx: &mut App, ) -> Option>>> { if path.path.extension().unwrap() == "png" { Some(cx.spawn(async move |cx| Ok(cx.new(|_| TestPngItem {})))) } else { None } } fn entry_id(&self, _: &App) -> Option { None } fn project_path(&self, _: &App) -> Option { None } fn is_dirty(&self) -> bool { false } } impl Item for TestPngItemView { type Event = (); fn tab_content_text(&self, _detail: usize, _cx: &App) -> SharedString { "".into() } } impl EventEmitter<()> for TestPngItemView {} impl Focusable for TestPngItemView { fn focus_handle(&self, _cx: &App) -> FocusHandle { self.focus_handle.clone() } } impl Render for TestPngItemView { fn render( &mut self, _window: &mut Window, _cx: &mut Context, ) -> impl IntoElement { Empty } } impl ProjectItem for TestPngItemView { type Item = TestPngItem; fn for_project_item( _project: Entity, _pane: Option<&Pane>, _item: Entity, _: &mut Window, cx: &mut Context, ) -> Self where Self: Sized, { Self { focus_handle: cx.focus_handle(), } } } // View struct TestIpynbItemView { focus_handle: FocusHandle, } // Model struct TestIpynbItem {} impl project::ProjectItem for TestIpynbItem { fn try_open( _project: &Entity, path: &ProjectPath, cx: &mut App, ) -> Option>>> { if path.path.extension().unwrap() == "ipynb" { Some(cx.spawn(async move |cx| Ok(cx.new(|_| TestIpynbItem {})))) } else { None } } fn entry_id(&self, _: &App) -> Option { None } fn project_path(&self, _: &App) -> Option { None } fn is_dirty(&self) -> bool { false } } impl Item for TestIpynbItemView { type Event = (); fn tab_content_text(&self, _detail: usize, _cx: &App) -> SharedString { "".into() } } impl EventEmitter<()> for TestIpynbItemView {} impl Focusable for TestIpynbItemView { fn focus_handle(&self, _cx: &App) -> FocusHandle { self.focus_handle.clone() } } impl Render for TestIpynbItemView { fn render( &mut self, _window: &mut Window, _cx: &mut Context, ) -> impl IntoElement { Empty } } impl ProjectItem for TestIpynbItemView { type Item = TestIpynbItem; fn for_project_item( _project: Entity, _pane: Option<&Pane>, _item: Entity, _: &mut Window, cx: &mut Context, ) -> Self where Self: Sized, { Self { focus_handle: cx.focus_handle(), } } } struct TestAlternatePngItemView { focus_handle: FocusHandle, } impl Item for TestAlternatePngItemView { type Event = (); fn tab_content_text(&self, _detail: usize, _cx: &App) -> SharedString { "".into() } } impl EventEmitter<()> for TestAlternatePngItemView {} impl Focusable for TestAlternatePngItemView { fn focus_handle(&self, _cx: &App) -> FocusHandle { self.focus_handle.clone() } } impl Render for TestAlternatePngItemView { fn render( &mut self, _window: &mut Window, _cx: &mut Context, ) -> impl IntoElement { Empty } } impl ProjectItem for TestAlternatePngItemView { type Item = TestPngItem; fn for_project_item( _project: Entity, _pane: Option<&Pane>, _item: Entity, _: &mut Window, cx: &mut Context, ) -> Self where Self: Sized, { Self { focus_handle: cx.focus_handle(), } } } #[gpui::test] async fn test_register_project_item(cx: &mut TestAppContext) { init_test(cx); cx.update(|cx| { register_project_item::(cx); register_project_item::(cx); }); let fs = FakeFs::new(cx.executor()); fs.insert_tree( "/root1", json!({ "one.png": "BINARYDATAHERE", "two.ipynb": "{ totally a notebook }", "three.txt": "editing text, sure why not?" }), ) .await; let project = Project::test(fs, ["root1".as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let worktree_id = project.update(cx, |project, cx| { project.worktrees(cx).next().unwrap().read(cx).id() }); let handle = workspace .update_in(cx, |workspace, window, cx| { let project_path = (worktree_id, rel_path("one.png")); workspace.open_path(project_path, None, true, window, cx) }) .await .unwrap(); // Now we can check if the handle we got back errored or not assert_eq!( handle.to_any_view().entity_type(), TypeId::of::() ); let handle = workspace .update_in(cx, |workspace, window, cx| { let project_path = (worktree_id, rel_path("two.ipynb")); workspace.open_path(project_path, None, true, window, cx) }) .await .unwrap(); assert_eq!( handle.to_any_view().entity_type(), TypeId::of::() ); let handle = workspace .update_in(cx, |workspace, window, cx| { let project_path = (worktree_id, rel_path("three.txt")); workspace.open_path(project_path, None, true, window, cx) }) .await; assert!(handle.is_err()); } #[gpui::test] async fn test_register_project_item_two_enter_one_leaves(cx: &mut TestAppContext) { init_test(cx); cx.update(|cx| { register_project_item::(cx); register_project_item::(cx); }); let fs = FakeFs::new(cx.executor()); fs.insert_tree( "/root1", json!({ "one.png": "BINARYDATAHERE", "two.ipynb": "{ totally a notebook }", "three.txt": "editing text, sure why not?" }), ) .await; let project = Project::test(fs, ["root1".as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let worktree_id = project.update(cx, |project, cx| { project.worktrees(cx).next().unwrap().read(cx).id() }); let handle = workspace .update_in(cx, |workspace, window, cx| { let project_path = (worktree_id, rel_path("one.png")); workspace.open_path(project_path, None, true, window, cx) }) .await .unwrap(); // This _must_ be the second item registered assert_eq!( handle.to_any_view().entity_type(), TypeId::of::() ); let handle = workspace .update_in(cx, |workspace, window, cx| { let project_path = (worktree_id, rel_path("three.txt")); workspace.open_path(project_path, None, true, window, cx) }) .await; assert!(handle.is_err()); } } #[gpui::test] async fn test_status_bar_visibility(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, _cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); // Test with status bar shown (default) workspace.read_with(cx, |workspace, cx| { let visible = workspace.status_bar_visible(cx); assert!(visible, "Status bar should be visible by default"); }); // Test with status bar hidden cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.status_bar.get_or_insert_default().show = Some(false); }); }); workspace.read_with(cx, |workspace, cx| { let visible = workspace.status_bar_visible(cx); assert!(!visible, "Status bar should be hidden when show is false"); }); // Test with status bar shown explicitly cx.update_global(|store: &mut SettingsStore, cx| { store.update_user_settings(cx, |settings| { settings.status_bar.get_or_insert_default().show = Some(true); }); }); workspace.read_with(cx, |workspace, cx| { let visible = workspace.status_bar_visible(cx); assert!(visible, "Status bar should be visible when show is true"); }); } #[gpui::test] async fn test_pane_close_active_item(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project, window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); let panel = workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace .right_dock() .update(cx, |right_dock, cx| right_dock.set_open(true, window, cx)); panel }); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); let item_a = cx.new(TestItem::new); let item_b = cx.new(TestItem::new); let item_a_id = item_a.entity_id(); let item_b_id = item_b.entity_id(); pane.update_in(cx, |pane, window, cx| { pane.add_item(Box::new(item_a.clone()), true, true, None, window, cx); pane.add_item(Box::new(item_b.clone()), true, true, None, window, cx); }); pane.read_with(cx, |pane, _| { assert_eq!(pane.items_len(), 2); assert_eq!(pane.active_item().unwrap().item_id(), item_b_id); }); workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); workspace.update_in(cx, |_, window, cx| { assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Assert that the `pane::CloseActiveItem` action is handled at the // workspace level when one of the dock panels is focused and, in that // case, the center pane's active item is closed but the focus is not // moved. cx.dispatch_action(pane::CloseActiveItem::default()); cx.run_until_parked(); pane.read_with(cx, |pane, _| { assert_eq!(pane.items_len(), 1); assert_eq!(pane.active_item().unwrap().item_id(), item_a_id); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); } #[gpui::test] async fn test_panel_zoom_preserved_across_workspace_switch(cx: &mut TestAppContext) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project_a = Project::test(fs.clone(), [], cx).await; let project_b = Project::test(fs, [], cx).await; let multi_workspace_handle = cx.add_window(|window, cx| MultiWorkspace::test_new(project_a.clone(), window, cx)); cx.run_until_parked(); multi_workspace_handle .update(cx, |mw, _window, cx| { mw.open_sidebar(cx); }) .unwrap(); let workspace_a = multi_workspace_handle .read_with(cx, |mw, _| mw.workspace().clone()) .unwrap(); let _workspace_b = multi_workspace_handle .update(cx, |mw, window, cx| { mw.test_add_workspace(project_b, window, cx) }) .unwrap(); // Switch to workspace A multi_workspace_handle .update(cx, |mw, window, cx| { let workspace = mw.workspaces().next().unwrap().clone(); mw.activate(workspace, window, cx); }) .unwrap(); let cx = &mut VisualTestContext::from_window(multi_workspace_handle.into(), cx); // Add a panel to workspace A's right dock and open the dock let panel = workspace_a.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace .right_dock() .update(cx, |dock, cx| dock.set_open(true, window, cx)); panel }); // Focus the panel through the workspace (matching existing test pattern) workspace_a.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); // Zoom the panel panel.update_in(cx, |panel, window, cx| { panel.set_zoomed(true, window, cx); }); // Verify the panel is zoomed and the dock is open workspace_a.update_in(cx, |workspace, window, cx| { assert!( workspace.right_dock().read(cx).is_open(), "dock should be open before switch" ); assert!( panel.is_zoomed(window, cx), "panel should be zoomed before switch" ); assert!( panel.read(cx).focus_handle(cx).contains_focused(window, cx), "panel should be focused before switch" ); }); // Switch to workspace B multi_workspace_handle .update(cx, |mw, window, cx| { let workspace = mw.workspaces().nth(1).unwrap().clone(); mw.activate(workspace, window, cx); }) .unwrap(); cx.run_until_parked(); // Switch back to workspace A multi_workspace_handle .update(cx, |mw, window, cx| { let workspace = mw.workspaces().next().unwrap().clone(); mw.activate(workspace, window, cx); }) .unwrap(); cx.run_until_parked(); // Verify the panel is still zoomed and the dock is still open workspace_a.update_in(cx, |workspace, window, cx| { assert!( workspace.right_dock().read(cx).is_open(), "dock should still be open after switching back" ); assert!( panel.is_zoomed(window, cx), "panel should still be zoomed after switching back" ); }); } fn pane_items_paths(pane: &Entity, cx: &App) -> Vec { pane.read(cx) .items() .flat_map(|item| { item.project_paths(cx) .into_iter() .map(|path| path.path.display(PathStyle::local()).into_owned()) }) .collect() } pub fn init_test(cx: &mut TestAppContext) { cx.update(|cx| { let settings_store = SettingsStore::test(cx); cx.set_global(settings_store); cx.set_global(db::AppDatabase::test_new()); theme_settings::init(theme::LoadThemes::JustBase, cx); }); } #[gpui::test] async fn test_toggle_theme_mode_persists_and_updates_active_theme(cx: &mut TestAppContext) { use settings::{ThemeName, ThemeSelection}; use theme::SystemAppearance; use zed_actions::theme::ToggleMode; init_test(cx); let fs = FakeFs::new(cx.executor()); let settings_fs: Arc = fs.clone(); fs.insert_tree(path!("/root"), json!({ "file.rs": "fn main() {}\n" })) .await; // Build a test project and workspace view so the test can invoke // the workspace action handler the same way the UI would. let project = Project::test(fs.clone(), [path!("/root").as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); // Seed the settings file with a plain static light theme so the // first toggle always starts from a known persisted state. workspace.update_in(cx, |_workspace, _window, cx| { *SystemAppearance::global_mut(cx) = SystemAppearance(theme::Appearance::Light); settings::update_settings_file(settings_fs.clone(), cx, |settings, _cx| { settings.theme.theme = Some(ThemeSelection::Static(ThemeName("One Light".into()))); }); }); cx.executor().advance_clock(Duration::from_millis(200)); cx.run_until_parked(); // Confirm the initial persisted settings contain the static theme // we just wrote before any toggling happens. let settings_text = SettingsStore::load_settings(&settings_fs).await.unwrap(); assert!(settings_text.contains(r#""theme": "One Light""#)); // Toggle once. This should migrate the persisted theme settings // into light/dark slots and enable system mode. workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_theme_mode(&ToggleMode, window, cx); }); cx.executor().advance_clock(Duration::from_millis(200)); cx.run_until_parked(); // 1. Static -> Dynamic // this assertion checks theme changed from static to dynamic. let settings_text = SettingsStore::load_settings(&settings_fs).await.unwrap(); let parsed: serde_json::Value = settings::parse_json_with_comments(&settings_text).unwrap(); assert_eq!( parsed["theme"], serde_json::json!({ "mode": "system", "light": "One Light", "dark": "One Dark" }) ); // 2. Toggle again, suppose it will change the mode to light workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_theme_mode(&ToggleMode, window, cx); }); cx.executor().advance_clock(Duration::from_millis(200)); cx.run_until_parked(); let settings_text = SettingsStore::load_settings(&settings_fs).await.unwrap(); assert!(settings_text.contains(r#""mode": "light""#)); } fn dirty_project_item(id: u64, path: &str, cx: &mut App) -> Entity { let item = TestProjectItem::new(id, path, cx); item.update(cx, |item, _| { item.is_dirty = true; }); item } #[gpui::test] async fn test_zoomed_panel_without_pane_preserved_on_center_focus( cx: &mut gpui::TestAppContext, ) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); let panel = workspace.update_in(cx, |workspace, window, cx| { let panel = cx.new(|cx| TestPanel::new(DockPosition::Right, 100, cx)); workspace.add_panel(panel.clone(), window, cx); workspace .right_dock() .update(cx, |dock, cx| dock.set_open(true, window, cx)); panel }); let pane = workspace.read_with(cx, |workspace, _| workspace.active_pane().clone()); pane.update_in(cx, |pane, window, cx| { let item = cx.new(TestItem::new); pane.add_item(Box::new(item), true, true, None, window, cx); }); // Transfer focus to the panel, then zoom it. Using toggle_panel_focus // mirrors the real-world flow and avoids side effects from directly // focusing the panel while the center pane is active. workspace.update_in(cx, |workspace, window, cx| { workspace.toggle_panel_focus::(window, cx); }); panel.update_in(cx, |panel, window, cx| { panel.set_zoomed(true, window, cx); }); workspace.update_in(cx, |workspace, window, cx| { assert!(workspace.right_dock().read(cx).is_open()); assert!(panel.is_zoomed(window, cx)); assert!(panel.read(cx).focus_handle(cx).contains_focused(window, cx)); }); // Simulate a spurious pane::Event::Focus on the center pane while the // panel still has focus. This mirrors what happens during macOS window // activation: the center pane fires a focus event even though actual // focus remains on the dock panel. pane.update_in(cx, |_, _, cx| { cx.emit(pane::Event::Focus); }); // The dock must remain open because the panel had focus at the time the // event was processed. Before the fix, dock_to_preserve was None for // panels that don't implement pane(), causing the dock to close. workspace.update_in(cx, |workspace, window, cx| { assert!( workspace.right_dock().read(cx).is_open(), "Dock should stay open when its zoomed panel (without pane()) still has focus" ); assert!(panel.is_zoomed(window, cx)); }); } #[gpui::test] async fn test_panels_stay_open_after_position_change_and_settings_update( cx: &mut gpui::TestAppContext, ) { init_test(cx); let fs = FakeFs::new(cx.executor()); let project = Project::test(fs, [], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project, window, cx)); // Add two panels to the left dock and open it. let (panel_a, panel_b) = workspace.update_in(cx, |workspace, window, cx| { let panel_a = cx.new(|cx| TestPanel::new(DockPosition::Left, 100, cx)); let panel_b = cx.new(|cx| TestPanel::new(DockPosition::Left, 101, cx)); workspace.add_panel(panel_a.clone(), window, cx); workspace.add_panel(panel_b.clone(), window, cx); workspace.left_dock().update(cx, |dock, cx| { dock.set_open(true, window, cx); dock.activate_panel(0, window, cx); }); (panel_a, panel_b) }); workspace.update_in(cx, |workspace, _, cx| { assert!(workspace.left_dock().read(cx).is_open()); }); // Simulate a feature flag changing default dock positions: both panels // move from Left to Right. workspace.update_in(cx, |_workspace, _window, cx| { panel_a.update(cx, |p, _cx| p.position = DockPosition::Right); panel_b.update(cx, |p, _cx| p.position = DockPosition::Right); cx.update_global::(|_, _| {}); }); // Both panels should now be in the right dock. workspace.update_in(cx, |workspace, _, cx| { let right_dock = workspace.right_dock().read(cx); assert_eq!(right_dock.panels_len(), 2); }); // Open the right dock and activate panel_b (simulating the user // opening the panel after it moved). workspace.update_in(cx, |workspace, window, cx| { workspace.right_dock().update(cx, |dock, cx| { dock.set_open(true, window, cx); dock.activate_panel(1, window, cx); }); }); // Now trigger another SettingsStore change workspace.update_in(cx, |_workspace, _window, cx| { cx.update_global::(|_, _| {}); }); workspace.update_in(cx, |workspace, _, cx| { assert!( workspace.right_dock().read(cx).is_open(), "Right dock should still be open after a settings change" ); assert_eq!( workspace.right_dock().read(cx).panels_len(), 2, "Both panels should still be in the right dock" ); }); } }