use crate::{ Anchor, Autoscroll, Editor, EditorEvent, EditorSettings, ExcerptId, ExcerptRange, FormatTarget, MultiBuffer, MultiBufferSnapshot, NavigationData, ReportEditorEvent, SearchWithinRange, SelectionEffects, ToPoint as _, display_map::HighlightKey, editor_settings::SeedQuerySetting, persistence::{DB, SerializedEditor}, scroll::ScrollAnchor, }; use anyhow::{Context as _, Result, anyhow}; use collections::{HashMap, HashSet}; use file_icons::FileIcons; use futures::future::try_join_all; use git::status::GitSummary; use gpui::{ AnyElement, App, AsyncWindowContext, Context, Entity, EntityId, EventEmitter, IntoElement, ParentElement, Pixels, SharedString, Styled, Task, WeakEntity, Window, point, }; use language::{ Bias, Buffer, BufferRow, CharKind, DiskState, LocalFile, Point, SelectionGoal, proto::serialize_anchor as serialize_text_anchor, }; use lsp::DiagnosticSeverity; use project::{ Project, ProjectItem as _, ProjectPath, lsp_store::FormatTrigger, project_settings::ProjectSettings, search::SearchQuery, }; use rpc::proto::{self, update_view}; use settings::Settings; use std::{ any::TypeId, borrow::Cow, cmp::{self, Ordering}, iter, ops::Range, path::{Path, PathBuf}, sync::Arc, }; use text::{BufferId, BufferSnapshot, Selection}; use theme::{Theme, ThemeSettings}; use ui::{IconDecorationKind, prelude::*}; use util::{ResultExt, TryFutureExt, paths::PathExt}; use workspace::{ CollaboratorId, ItemId, ItemNavHistory, ToolbarItemLocation, ViewId, Workspace, WorkspaceId, item::{FollowableItem, Item, ItemEvent, ProjectItem, SaveOptions}, searchable::{Direction, SearchEvent, SearchableItem, SearchableItemHandle}, }; use workspace::{ OpenOptions, item::{Dedup, ItemSettings, SerializableItem, TabContentParams}, }; use workspace::{ OpenVisible, Pane, WorkspaceSettings, item::{BreadcrumbText, FollowEvent, ProjectItemKind}, searchable::SearchOptions, }; pub const MAX_TAB_TITLE_LEN: usize = 24; impl FollowableItem for Editor { fn remote_id(&self) -> Option { self.remote_id } fn from_state_proto( workspace: Entity, remote_id: ViewId, state: &mut Option, window: &mut Window, cx: &mut App, ) -> Option>>> { let project = workspace.read(cx).project().to_owned(); let Some(proto::view::Variant::Editor(_)) = state else { return None; }; let Some(proto::view::Variant::Editor(state)) = state.take() else { unreachable!() }; let buffer_ids = state .excerpts .iter() .map(|excerpt| excerpt.buffer_id) .collect::>(); let buffers = project.update(cx, |project, cx| { buffer_ids .iter() .map(|id| BufferId::new(*id).map(|id| project.open_buffer_by_id(id, cx))) .collect::>>() }); Some(window.spawn(cx, async move |cx| { let mut buffers = futures::future::try_join_all(buffers?) .await .debug_assert_ok("leaders don't share views for unshared buffers")?; let editor = cx.update(|window, cx| { let multibuffer = cx.new(|cx| { let mut multibuffer; if state.singleton && buffers.len() == 1 { multibuffer = MultiBuffer::singleton(buffers.pop().unwrap(), cx) } else { multibuffer = MultiBuffer::new(project.read(cx).capability()); let mut sorted_excerpts = state.excerpts.clone(); sorted_excerpts.sort_by_key(|e| e.id); let mut sorted_excerpts = sorted_excerpts.into_iter().peekable(); while let Some(excerpt) = sorted_excerpts.next() { let Ok(buffer_id) = BufferId::new(excerpt.buffer_id) else { continue; }; let mut insert_position = ExcerptId::min(); for e in &state.excerpts { if e.id == excerpt.id { break; } if e.id < excerpt.id { insert_position = ExcerptId::from_proto(e.id); } } let buffer = buffers.iter().find(|b| b.read(cx).remote_id() == buffer_id); let Some(excerpt) = deserialize_excerpt_range(excerpt) else { continue; }; let Some(buffer) = buffer else { continue }; multibuffer.insert_excerpts_with_ids_after( insert_position, buffer.clone(), [excerpt], cx, ); } }; if let Some(title) = &state.title { multibuffer = multibuffer.with_title(title.clone()) } multibuffer }); cx.new(|cx| { let mut editor = Editor::for_multibuffer(multibuffer, Some(project.clone()), window, cx); editor.remote_id = Some(remote_id); editor }) })?; update_editor_from_message( editor.downgrade(), project, proto::update_view::Editor { selections: state.selections, pending_selection: state.pending_selection, scroll_top_anchor: state.scroll_top_anchor, scroll_x: state.scroll_x, scroll_y: state.scroll_y, ..Default::default() }, cx, ) .await?; Ok(editor) })) } fn set_leader_id( &mut self, leader_id: Option, window: &mut Window, cx: &mut Context, ) { self.leader_id = leader_id; if self.leader_id.is_some() { self.buffer.update(cx, |buffer, cx| { buffer.remove_active_selections(cx); }); } else if self.focus_handle.is_focused(window) { self.buffer.update(cx, |buffer, cx| { buffer.set_active_selections( &self.selections.disjoint_anchors(), self.selections.line_mode, self.cursor_shape, cx, ); }); } cx.notify(); } fn to_state_proto(&self, _: &Window, cx: &App) -> Option { let buffer = self.buffer.read(cx); if buffer .as_singleton() .and_then(|buffer| buffer.read(cx).file()) .is_some_and(|file| file.is_private()) { return None; } let scroll_anchor = self.scroll_manager.anchor(); let excerpts = buffer .read(cx) .excerpts() .map(|(id, buffer, range)| proto::Excerpt { id: id.to_proto(), buffer_id: buffer.remote_id().into(), context_start: Some(serialize_text_anchor(&range.context.start)), context_end: Some(serialize_text_anchor(&range.context.end)), primary_start: Some(serialize_text_anchor(&range.primary.start)), primary_end: Some(serialize_text_anchor(&range.primary.end)), }) .collect(); let snapshot = buffer.snapshot(cx); Some(proto::view::Variant::Editor(proto::view::Editor { singleton: buffer.is_singleton(), title: (!buffer.is_singleton()).then(|| buffer.title(cx).into()), excerpts, scroll_top_anchor: Some(serialize_anchor(&scroll_anchor.anchor, &snapshot)), scroll_x: scroll_anchor.offset.x, scroll_y: scroll_anchor.offset.y, selections: self .selections .disjoint_anchors() .iter() .map(|s| serialize_selection(s, &snapshot)) .collect(), pending_selection: self .selections .pending_anchor() .as_ref() .map(|s| serialize_selection(s, &snapshot)), })) } fn to_follow_event(event: &EditorEvent) -> Option { match event { EditorEvent::Edited { .. } => Some(FollowEvent::Unfollow), EditorEvent::SelectionsChanged { local } | EditorEvent::ScrollPositionChanged { local, .. } => { if *local { Some(FollowEvent::Unfollow) } else { None } } _ => None, } } fn add_event_to_update_proto( &self, event: &EditorEvent, update: &mut Option, _: &Window, cx: &App, ) -> bool { let update = update.get_or_insert_with(|| proto::update_view::Variant::Editor(Default::default())); match update { proto::update_view::Variant::Editor(update) => match event { EditorEvent::ExcerptsAdded { buffer, predecessor, excerpts, } => { let buffer_id = buffer.read(cx).remote_id(); let mut excerpts = excerpts.iter(); if let Some((id, range)) = excerpts.next() { update.inserted_excerpts.push(proto::ExcerptInsertion { previous_excerpt_id: Some(predecessor.to_proto()), excerpt: serialize_excerpt(buffer_id, id, range), }); update.inserted_excerpts.extend(excerpts.map(|(id, range)| { proto::ExcerptInsertion { previous_excerpt_id: None, excerpt: serialize_excerpt(buffer_id, id, range), } })) } true } EditorEvent::ExcerptsRemoved { ids, .. } => { update .deleted_excerpts .extend(ids.iter().map(ExcerptId::to_proto)); true } EditorEvent::ScrollPositionChanged { autoscroll, .. } if !autoscroll => { let snapshot = self.buffer.read(cx).snapshot(cx); let scroll_anchor = self.scroll_manager.anchor(); update.scroll_top_anchor = Some(serialize_anchor(&scroll_anchor.anchor, &snapshot)); update.scroll_x = scroll_anchor.offset.x; update.scroll_y = scroll_anchor.offset.y; true } EditorEvent::SelectionsChanged { .. } => { let snapshot = self.buffer.read(cx).snapshot(cx); update.selections = self .selections .disjoint_anchors() .iter() .map(|s| serialize_selection(s, &snapshot)) .collect(); update.pending_selection = self .selections .pending_anchor() .as_ref() .map(|s| serialize_selection(s, &snapshot)); true } _ => false, }, } } fn apply_update_proto( &mut self, project: &Entity, message: update_view::Variant, window: &mut Window, cx: &mut Context, ) -> Task> { let update_view::Variant::Editor(message) = message; let project = project.clone(); cx.spawn_in(window, async move |this, cx| { update_editor_from_message(this, project, message, cx).await }) } fn is_project_item(&self, _window: &Window, _cx: &App) -> bool { true } fn dedup(&self, existing: &Self, _: &Window, cx: &App) -> Option { let self_singleton = self.buffer.read(cx).as_singleton()?; let other_singleton = existing.buffer.read(cx).as_singleton()?; if self_singleton == other_singleton { Some(Dedup::KeepExisting) } else { None } } fn update_agent_location( &mut self, location: language::Anchor, window: &mut Window, cx: &mut Context, ) { let buffer = self.buffer.read(cx); let buffer = buffer.read(cx); let Some((excerpt_id, _, _)) = buffer.as_singleton() else { return; }; let position = buffer.anchor_in_excerpt(*excerpt_id, location).unwrap(); let selection = Selection { id: 0, reversed: false, start: position, end: position, goal: SelectionGoal::None, }; drop(buffer); self.set_selections_from_remote(vec![selection], None, window, cx); self.request_autoscroll_remotely(Autoscroll::fit(), cx); } } async fn update_editor_from_message( this: WeakEntity, project: Entity, message: proto::update_view::Editor, cx: &mut AsyncWindowContext, ) -> Result<()> { // Open all of the buffers of which excerpts were added to the editor. let inserted_excerpt_buffer_ids = message .inserted_excerpts .iter() .filter_map(|insertion| Some(insertion.excerpt.as_ref()?.buffer_id)) .collect::>(); let inserted_excerpt_buffers = project.update(cx, |project, cx| { inserted_excerpt_buffer_ids .into_iter() .map(|id| BufferId::new(id).map(|id| project.open_buffer_by_id(id, cx))) .collect::>>() })??; let _inserted_excerpt_buffers = try_join_all(inserted_excerpt_buffers).await?; // Update the editor's excerpts. this.update(cx, |editor, cx| { editor.buffer.update(cx, |multibuffer, cx| { let mut removed_excerpt_ids = message .deleted_excerpts .into_iter() .map(ExcerptId::from_proto) .collect::>(); removed_excerpt_ids.sort_by({ let multibuffer = multibuffer.read(cx); move |a, b| a.cmp(b, &multibuffer) }); let mut insertions = message.inserted_excerpts.into_iter().peekable(); while let Some(insertion) = insertions.next() { let Some(excerpt) = insertion.excerpt else { continue; }; let Some(previous_excerpt_id) = insertion.previous_excerpt_id else { continue; }; let buffer_id = BufferId::new(excerpt.buffer_id)?; let Some(buffer) = project.read(cx).buffer_for_id(buffer_id, cx) else { continue; }; let adjacent_excerpts = iter::from_fn(|| { let insertion = insertions.peek()?; if insertion.previous_excerpt_id.is_none() && insertion.excerpt.as_ref()?.buffer_id == u64::from(buffer_id) { insertions.next()?.excerpt } else { None } }); multibuffer.insert_excerpts_with_ids_after( ExcerptId::from_proto(previous_excerpt_id), buffer, [excerpt] .into_iter() .chain(adjacent_excerpts) .filter_map(deserialize_excerpt_range), cx, ); } multibuffer.remove_excerpts(removed_excerpt_ids, cx); anyhow::Ok(()) }) })??; // Deserialize the editor state. let (selections, pending_selection, scroll_top_anchor) = this.update(cx, |editor, cx| { let buffer = editor.buffer.read(cx).read(cx); let selections = message .selections .into_iter() .filter_map(|selection| deserialize_selection(&buffer, selection)) .collect::>(); let pending_selection = message .pending_selection .and_then(|selection| deserialize_selection(&buffer, selection)); let scroll_top_anchor = message .scroll_top_anchor .and_then(|anchor| deserialize_anchor(&buffer, anchor)); anyhow::Ok((selections, pending_selection, scroll_top_anchor)) })??; // Wait until the buffer has received all of the operations referenced by // the editor's new state. this.update(cx, |editor, cx| { editor.buffer.update(cx, |buffer, cx| { buffer.wait_for_anchors( selections .iter() .chain(pending_selection.as_ref()) .flat_map(|selection| [selection.start, selection.end]) .chain(scroll_top_anchor), cx, ) }) })? .await?; // Update the editor's state. this.update_in(cx, |editor, window, cx| { if !selections.is_empty() || pending_selection.is_some() { editor.set_selections_from_remote(selections, pending_selection, window, cx); editor.request_autoscroll_remotely(Autoscroll::newest(), cx); } else if let Some(scroll_top_anchor) = scroll_top_anchor { editor.set_scroll_anchor_remote( ScrollAnchor { anchor: scroll_top_anchor, offset: point(message.scroll_x, message.scroll_y), }, window, cx, ); } })?; Ok(()) } fn serialize_excerpt( buffer_id: BufferId, id: &ExcerptId, range: &ExcerptRange, ) -> Option { Some(proto::Excerpt { id: id.to_proto(), buffer_id: buffer_id.into(), context_start: Some(serialize_text_anchor(&range.context.start)), context_end: Some(serialize_text_anchor(&range.context.end)), primary_start: Some(serialize_text_anchor(&range.primary.start)), primary_end: Some(serialize_text_anchor(&range.primary.end)), }) } fn serialize_selection( selection: &Selection, buffer: &MultiBufferSnapshot, ) -> proto::Selection { proto::Selection { id: selection.id as u64, start: Some(serialize_anchor(&selection.start, buffer)), end: Some(serialize_anchor(&selection.end, buffer)), reversed: selection.reversed, } } fn serialize_anchor(anchor: &Anchor, buffer: &MultiBufferSnapshot) -> proto::EditorAnchor { proto::EditorAnchor { excerpt_id: buffer.latest_excerpt_id(anchor.excerpt_id).to_proto(), anchor: Some(serialize_text_anchor(&anchor.text_anchor)), } } fn deserialize_excerpt_range( excerpt: proto::Excerpt, ) -> Option<(ExcerptId, ExcerptRange)> { let context = { let start = language::proto::deserialize_anchor(excerpt.context_start?)?; let end = language::proto::deserialize_anchor(excerpt.context_end?)?; start..end }; let primary = excerpt .primary_start .zip(excerpt.primary_end) .and_then(|(start, end)| { let start = language::proto::deserialize_anchor(start)?; let end = language::proto::deserialize_anchor(end)?; Some(start..end) }) .unwrap_or_else(|| context.clone()); Some(( ExcerptId::from_proto(excerpt.id), ExcerptRange { context, primary }, )) } fn deserialize_selection( buffer: &MultiBufferSnapshot, selection: proto::Selection, ) -> Option> { Some(Selection { id: selection.id as usize, start: deserialize_anchor(buffer, selection.start?)?, end: deserialize_anchor(buffer, selection.end?)?, reversed: selection.reversed, goal: SelectionGoal::None, }) } fn deserialize_anchor(buffer: &MultiBufferSnapshot, anchor: proto::EditorAnchor) -> Option { let excerpt_id = ExcerptId::from_proto(anchor.excerpt_id); Some(Anchor { excerpt_id, text_anchor: language::proto::deserialize_anchor(anchor.anchor?)?, buffer_id: buffer.buffer_id_for_excerpt(excerpt_id), diff_base_anchor: None, }) } impl Item for Editor { type Event = EditorEvent; fn navigate( &mut self, data: Box, window: &mut Window, cx: &mut Context, ) -> bool { if let Ok(data) = data.downcast::() { let newest_selection = self.selections.newest::(cx); let buffer = self.buffer.read(cx).read(cx); let offset = if buffer.can_resolve(&data.cursor_anchor) { data.cursor_anchor.to_point(&buffer) } else { buffer.clip_point(data.cursor_position, Bias::Left) }; let mut scroll_anchor = data.scroll_anchor; if !buffer.can_resolve(&scroll_anchor.anchor) { scroll_anchor.anchor = buffer.anchor_before( buffer.clip_point(Point::new(data.scroll_top_row, 0), Bias::Left), ); } drop(buffer); if newest_selection.head() == offset { false } else { self.set_scroll_anchor(scroll_anchor, window, cx); self.change_selections( SelectionEffects::default().nav_history(false), window, cx, |s| s.select_ranges([offset..offset]), ); true } } else { false } } fn tab_tooltip_text(&self, cx: &App) -> Option { let file_path = self .buffer() .read(cx) .as_singleton()? .read(cx) .file() .and_then(|f| f.as_local())? .abs_path(cx); let file_path = file_path.compact().to_string_lossy().to_string(); Some(file_path.into()) } fn telemetry_event_text(&self) -> Option<&'static str> { None } fn tab_content_text(&self, detail: usize, cx: &App) -> SharedString { if let Some(path) = path_for_buffer(&self.buffer, detail, true, cx) { path.to_string_lossy().to_string().into() } else { "untitled".into() } } fn suggested_filename(&self, cx: &App) -> SharedString { self.buffer.read(cx).title(cx).to_string().into() } fn tab_icon(&self, _: &Window, cx: &App) -> Option { ItemSettings::get_global(cx) .file_icons .then(|| { path_for_buffer(&self.buffer, 0, true, cx) .and_then(|path| FileIcons::get_icon(path.as_ref(), cx)) }) .flatten() .map(Icon::from_path) } fn tab_content(&self, params: TabContentParams, _: &Window, cx: &App) -> AnyElement { let label_color = if ItemSettings::get_global(cx).git_status { self.buffer() .read(cx) .as_singleton() .and_then(|buffer| { let buffer = buffer.read(cx); let path = buffer.project_path(cx)?; let buffer_id = buffer.remote_id(); let project = self.project()?.read(cx); let entry = project.entry_for_path(&path, cx)?; let (repo, repo_path) = project .git_store() .read(cx) .repository_and_path_for_buffer_id(buffer_id, cx)?; let status = repo.read(cx).status_for_path(&repo_path)?.status; Some(entry_git_aware_label_color( status.summary(), entry.is_ignored, params.selected, )) }) .unwrap_or_else(|| entry_label_color(params.selected)) } else { entry_label_color(params.selected) }; let description = params.detail.and_then(|detail| { let path = path_for_buffer(&self.buffer, detail, false, cx)?; let description = path.to_string_lossy(); let description = description.trim(); if description.is_empty() { return None; } Some(util::truncate_and_trailoff(description, MAX_TAB_TITLE_LEN)) }); // Whether the file was saved in the past but is now deleted. let was_deleted: bool = self .buffer() .read(cx) .as_singleton() .and_then(|buffer| buffer.read(cx).file()) .is_some_and(|file| file.disk_state() == DiskState::Deleted); h_flex() .gap_2() .child( Label::new(self.title(cx).to_string()) .color(label_color) .when(params.preview, |this| this.italic()) .when(was_deleted, |this| this.strikethrough()), ) .when_some(description, |this, description| { this.child( Label::new(description) .size(LabelSize::XSmall) .color(Color::Muted), ) }) .into_any_element() } fn for_each_project_item( &self, cx: &App, f: &mut dyn FnMut(EntityId, &dyn project::ProjectItem), ) { self.buffer .read(cx) .for_each_buffer(|buffer| f(buffer.entity_id(), buffer.read(cx))); } fn is_singleton(&self, cx: &App) -> bool { self.buffer.read(cx).is_singleton() } fn can_save_as(&self, cx: &App) -> bool { self.buffer.read(cx).is_singleton() } fn clone_on_split( &self, _workspace_id: Option, window: &mut Window, cx: &mut Context, ) -> Option> where Self: Sized, { Some(cx.new(|cx| self.clone(window, cx))) } fn set_nav_history( &mut self, history: ItemNavHistory, _window: &mut Window, _: &mut Context, ) { self.nav_history = Some(history); } fn discarded(&self, _project: Entity, _: &mut Window, cx: &mut Context) { for buffer in self.buffer().clone().read(cx).all_buffers() { buffer.update(cx, |buffer, cx| buffer.discarded(cx)) } } fn on_removed(&self, cx: &App) { self.report_editor_event(ReportEditorEvent::Closed, None, cx); } fn deactivated(&mut self, _: &mut Window, cx: &mut Context) { let selection = self.selections.newest_anchor(); self.push_to_nav_history(selection.head(), None, true, false, cx); } fn workspace_deactivated(&mut self, _: &mut Window, cx: &mut Context) { self.hide_hovered_link(cx); } fn is_dirty(&self, cx: &App) -> bool { self.buffer().read(cx).read(cx).is_dirty() } fn has_deleted_file(&self, cx: &App) -> bool { self.buffer().read(cx).read(cx).has_deleted_file() } fn has_conflict(&self, cx: &App) -> bool { self.buffer().read(cx).read(cx).has_conflict() } fn can_save(&self, cx: &App) -> bool { let buffer = &self.buffer().read(cx); if let Some(buffer) = buffer.as_singleton() { buffer.read(cx).project_path(cx).is_some() } else { true } } fn save( &mut self, options: SaveOptions, project: Entity, window: &mut Window, cx: &mut Context, ) -> Task> { // Add meta data tracking # of auto saves if options.autosave { self.report_editor_event(ReportEditorEvent::Saved { auto_saved: true }, None, cx); } else { self.report_editor_event(ReportEditorEvent::Saved { auto_saved: false }, None, cx); } let buffers = self.buffer().clone().read(cx).all_buffers(); let buffers = buffers .into_iter() .map(|handle| handle.read(cx).base_buffer().unwrap_or(handle.clone())) .collect::>(); // let mut buffers_to_save = let buffers_to_save = if self.buffer.read(cx).is_singleton() && !options.autosave { buffers.clone() } else { buffers .iter() .filter(|buffer| buffer.read(cx).is_dirty()) .cloned() .collect() }; cx.spawn_in(window, async move |this, cx| { if options.format { this.update_in(cx, |editor, window, cx| { editor.perform_format( project.clone(), FormatTrigger::Save, FormatTarget::Buffers(buffers_to_save.clone()), window, cx, ) })? .await?; } if !buffers_to_save.is_empty() { project .update(cx, |project, cx| { project.save_buffers(buffers_to_save.clone(), cx) })? .await?; } // Notify about clean buffers for language server events let buffers_that_were_not_saved: Vec<_> = buffers .into_iter() .filter(|b| !buffers_to_save.contains(b)) .collect(); for buffer in buffers_that_were_not_saved { buffer .update(cx, |buffer, cx| { let version = buffer.saved_version().clone(); let mtime = buffer.saved_mtime(); buffer.did_save(version, mtime, cx); }) .ok(); } Ok(()) }) } fn save_as( &mut self, project: Entity, path: ProjectPath, _: &mut Window, cx: &mut Context, ) -> Task> { let buffer = self .buffer() .read(cx) .as_singleton() .expect("cannot call save_as on an excerpt list"); let file_extension = path .path .extension() .map(|a| a.to_string_lossy().to_string()); self.report_editor_event( ReportEditorEvent::Saved { auto_saved: false }, file_extension, cx, ); project.update(cx, |project, cx| project.save_buffer_as(buffer, path, cx)) } fn reload( &mut self, project: Entity, window: &mut Window, cx: &mut Context, ) -> Task> { let buffer = self.buffer().clone(); let buffers = self.buffer.read(cx).all_buffers(); let reload_buffers = project.update(cx, |project, cx| project.reload_buffers(buffers, true, cx)); cx.spawn_in(window, async move |this, cx| { let transaction = reload_buffers.log_err().await; this.update(cx, |editor, cx| { editor.request_autoscroll(Autoscroll::fit(), cx) })?; buffer .update(cx, |buffer, cx| { if let Some(transaction) = transaction && !buffer.is_singleton() { buffer.push_transaction(&transaction.0, cx); } }) .ok(); Ok(()) }) } fn as_searchable(&self, handle: &Entity) -> Option> { Some(Box::new(handle.clone())) } fn pixel_position_of_cursor(&self, _: &App) -> Option> { self.pixel_position_of_newest_cursor } fn breadcrumb_location(&self, _: &App) -> ToolbarItemLocation { if self.show_breadcrumbs { ToolbarItemLocation::PrimaryLeft } else { ToolbarItemLocation::Hidden } } fn breadcrumbs(&self, variant: &Theme, cx: &App) -> Option> { let cursor = self.selections.newest_anchor().head(); let multibuffer = &self.buffer().read(cx); let (buffer_id, symbols) = multibuffer.symbols_containing(cursor, Some(variant.syntax()), cx)?; let buffer = multibuffer.buffer(buffer_id)?; let buffer = buffer.read(cx); let text = self.breadcrumb_header.clone().unwrap_or_else(|| { buffer .snapshot() .resolve_file_path( cx, self.project .as_ref() .map(|project| project.read(cx).visible_worktrees(cx).count() > 1) .unwrap_or_default(), ) .map(|path| path.to_string_lossy().to_string()) .unwrap_or_else(|| { if multibuffer.is_singleton() { multibuffer.title(cx).to_string() } else { "untitled".to_string() } }) }); let settings = ThemeSettings::get_global(cx); let mut breadcrumbs = vec![BreadcrumbText { text, highlights: None, font: Some(settings.buffer_font.clone()), }]; breadcrumbs.extend(symbols.into_iter().map(|symbol| BreadcrumbText { text: symbol.text, highlights: Some(symbol.highlight_ranges), font: Some(settings.buffer_font.clone()), })); Some(breadcrumbs) } fn added_to_workspace( &mut self, workspace: &mut Workspace, _window: &mut Window, cx: &mut Context, ) { self.workspace = Some((workspace.weak_handle(), workspace.database_id())); if let Some(workspace) = &workspace.weak_handle().upgrade() { cx.subscribe( workspace, |editor, _, event: &workspace::Event, _cx| match event { workspace::Event::ModalOpened => { editor.mouse_context_menu.take(); editor.inline_blame_popover.take(); } _ => {} }, ) .detach(); } } fn to_item_events(event: &EditorEvent, mut f: impl FnMut(ItemEvent)) { match event { EditorEvent::Closed => f(ItemEvent::CloseItem), EditorEvent::Saved | EditorEvent::TitleChanged => { f(ItemEvent::UpdateTab); f(ItemEvent::UpdateBreadcrumbs); } EditorEvent::Reparsed(_) => { f(ItemEvent::UpdateBreadcrumbs); } EditorEvent::SelectionsChanged { local } if *local => { f(ItemEvent::UpdateBreadcrumbs); } EditorEvent::BreadcrumbsChanged => { f(ItemEvent::UpdateBreadcrumbs); } EditorEvent::DirtyChanged => { f(ItemEvent::UpdateTab); } EditorEvent::BufferEdited => { f(ItemEvent::Edit); f(ItemEvent::UpdateBreadcrumbs); } EditorEvent::ExcerptsAdded { .. } | EditorEvent::ExcerptsRemoved { .. } => { f(ItemEvent::Edit); } _ => {} } } fn preserve_preview(&self, cx: &App) -> bool { self.buffer.read(cx).preserve_preview(cx) } } impl SerializableItem for Editor { fn serialized_item_kind() -> &'static str { "Editor" } fn cleanup( workspace_id: WorkspaceId, alive_items: Vec, _window: &mut Window, cx: &mut App, ) -> Task> { workspace::delete_unloaded_items(alive_items, workspace_id, "editors", &DB, cx) } fn deserialize( project: Entity, workspace: WeakEntity, workspace_id: workspace::WorkspaceId, item_id: ItemId, window: &mut Window, cx: &mut App, ) -> Task>> { let serialized_editor = match DB .get_serialized_editor(item_id, workspace_id) .context("Failed to query editor state") { Ok(Some(serialized_editor)) => { if ProjectSettings::get_global(cx) .session .restore_unsaved_buffers { serialized_editor } else { SerializedEditor { abs_path: serialized_editor.abs_path, contents: None, language: None, mtime: None, } } } Ok(None) => { return Task::ready(Err(anyhow!("No path or contents found for buffer"))); } Err(error) => { return Task::ready(Err(error)); } }; match serialized_editor { SerializedEditor { abs_path: None, contents: Some(contents), language, .. } => window.spawn(cx, { let project = project.clone(); async move |cx| { let language_registry = project.read_with(cx, |project, _| project.languages().clone())?; let language = if let Some(language_name) = language { // We don't fail here, because we'd rather not set the language if the name changed // than fail to restore the buffer. language_registry .language_for_name(&language_name) .await .ok() } else { None }; // First create the empty buffer let buffer = project .update(cx, |project, cx| project.create_buffer(cx))? .await?; // Then set the text so that the dirty bit is set correctly buffer.update(cx, |buffer, cx| { buffer.set_language_registry(language_registry); if let Some(language) = language { buffer.set_language(Some(language), cx); } buffer.set_text(contents, cx); if let Some(entry) = buffer.peek_undo_stack() { buffer.forget_transaction(entry.transaction_id()); } })?; cx.update(|window, cx| { cx.new(|cx| { let mut editor = Editor::for_buffer(buffer, Some(project), window, cx); editor.read_metadata_from_db(item_id, workspace_id, window, cx); editor }) }) } }), SerializedEditor { abs_path: Some(abs_path), contents, mtime, .. } => { let opened_buffer = project.update(cx, |project, cx| { let (worktree, path) = project.find_worktree(&abs_path, cx)?; let project_path = ProjectPath { worktree_id: worktree.read(cx).id(), path: path.into(), }; Some(project.open_path(project_path, cx)) }); match opened_buffer { Some(opened_buffer) => { window.spawn(cx, async move |cx| { let (_, buffer) = opened_buffer.await?; // This is a bit wasteful: we're loading the whole buffer from // disk and then overwrite the content. // But for now, it keeps the implementation of the content serialization // simple, because we don't have to persist all of the metadata that we get // by loading the file (git diff base, ...). if let Some(buffer_text) = contents { buffer.update(cx, |buffer, cx| { // If we did restore an mtime, we want to store it on the buffer // so that the next edit will mark the buffer as dirty/conflicted. if mtime.is_some() { buffer.did_reload( buffer.version(), buffer.line_ending(), mtime, cx, ); } buffer.set_text(buffer_text, cx); if let Some(entry) = buffer.peek_undo_stack() { buffer.forget_transaction(entry.transaction_id()); } })?; } cx.update(|window, cx| { cx.new(|cx| { let mut editor = Editor::for_buffer(buffer, Some(project), window, cx); editor.read_metadata_from_db(item_id, workspace_id, window, cx); editor }) }) }) } None => { let open_by_abs_path = workspace.update(cx, |workspace, cx| { workspace.open_abs_path( abs_path.clone(), OpenOptions { visible: Some(OpenVisible::None), ..Default::default() }, window, cx, ) }); window.spawn(cx, async move |cx| { let editor = open_by_abs_path?.await?.downcast::().with_context(|| format!("Failed to downcast to Editor after opening abs path {abs_path:?}"))?; editor.update_in(cx, |editor, window, cx| { editor.read_metadata_from_db(item_id, workspace_id, window, cx); })?; Ok(editor) }) } } } SerializedEditor { abs_path: None, contents: None, .. } => window.spawn(cx, async move |cx| { let buffer = project .update(cx, |project, cx| project.create_buffer(cx))? .await?; cx.update(|window, cx| { cx.new(|cx| { let mut editor = Editor::for_buffer(buffer, Some(project), window, cx); editor.read_metadata_from_db(item_id, workspace_id, window, cx); editor }) }) }), } } fn serialize( &mut self, workspace: &mut Workspace, item_id: ItemId, closing: bool, window: &mut Window, cx: &mut Context, ) -> Option>> { if self.mode.is_minimap() { return None; } let mut serialize_dirty_buffers = self.serialize_dirty_buffers; let project = self.project.clone()?; if project.read(cx).visible_worktrees(cx).next().is_none() { // If we don't have a worktree, we don't serialize, because // projects without worktrees aren't deserialized. serialize_dirty_buffers = false; } if closing && !serialize_dirty_buffers { return None; } let workspace_id = workspace.database_id()?; let buffer = self.buffer().read(cx).as_singleton()?; let abs_path = buffer.read(cx).file().and_then(|file| { let worktree_id = file.worktree_id(cx); project .read(cx) .worktree_for_id(worktree_id, cx) .and_then(|worktree| worktree.read(cx).absolutize(file.path()).ok()) .or_else(|| { let full_path = file.full_path(cx); let project_path = project.read(cx).find_project_path(&full_path, cx)?; project.read(cx).absolute_path(&project_path, cx) }) }); let is_dirty = buffer.read(cx).is_dirty(); let mtime = buffer.read(cx).saved_mtime(); let snapshot = buffer.read(cx).snapshot(); Some(cx.spawn_in(window, async move |_this, cx| { cx.background_spawn(async move { let (contents, language) = if serialize_dirty_buffers && is_dirty { let contents = snapshot.text(); let language = snapshot.language().map(|lang| lang.name().to_string()); (Some(contents), language) } else { (None, None) }; let editor = SerializedEditor { abs_path, contents, language, mtime, }; log::debug!("Serializing editor {item_id:?} in workspace {workspace_id:?}"); DB.save_serialized_editor(item_id, workspace_id, editor) .await .context("failed to save serialized editor") }) .await .context("failed to save contents of buffer")?; Ok(()) })) } fn should_serialize(&self, event: &Self::Event) -> bool { matches!( event, EditorEvent::Saved | EditorEvent::DirtyChanged | EditorEvent::BufferEdited ) } } #[derive(Debug, Default)] struct EditorRestorationData { entries: HashMap, } #[derive(Default, Debug)] pub struct RestorationData { pub scroll_position: (BufferRow, gpui::Point), pub folds: Vec>, pub selections: Vec>, } impl ProjectItem for Editor { type Item = Buffer; fn project_item_kind() -> Option { Some(ProjectItemKind("Editor")) } fn for_project_item( project: Entity, pane: Option<&Pane>, buffer: Entity, window: &mut Window, cx: &mut Context, ) -> Self { let mut editor = Self::for_buffer(buffer.clone(), Some(project), window, cx); if let Some((excerpt_id, buffer_id, snapshot)) = editor.buffer().read(cx).snapshot(cx).as_singleton() && WorkspaceSettings::get(None, cx).restore_on_file_reopen && let Some(restoration_data) = Self::project_item_kind() .and_then(|kind| pane.as_ref()?.project_item_restoration_data.get(&kind)) .and_then(|data| data.downcast_ref::()) .and_then(|data| { let file = project::File::from_dyn(buffer.read(cx).file())?; data.entries.get(&file.abs_path(cx)) }) { editor.fold_ranges( clip_ranges(&restoration_data.folds, snapshot), false, window, cx, ); if !restoration_data.selections.is_empty() { editor.change_selections(SelectionEffects::no_scroll(), window, cx, |s| { s.select_ranges(clip_ranges(&restoration_data.selections, snapshot)); }); } let (top_row, offset) = restoration_data.scroll_position; let anchor = Anchor::in_buffer( *excerpt_id, buffer_id, snapshot.anchor_before(Point::new(top_row, 0)), ); editor.set_scroll_anchor(ScrollAnchor { anchor, offset }, window, cx); } editor } } fn clip_ranges<'a>( original: impl IntoIterator> + 'a, snapshot: &'a BufferSnapshot, ) -> Vec> { original .into_iter() .map(|range| { snapshot.clip_point(range.start, Bias::Left) ..snapshot.clip_point(range.end, Bias::Right) }) .collect() } impl EventEmitter for Editor {} impl Editor { pub fn update_restoration_data( &self, cx: &mut Context, write: impl for<'a> FnOnce(&'a mut RestorationData) + 'static, ) { if self.mode.is_minimap() || !WorkspaceSettings::get(None, cx).restore_on_file_reopen { return; } let editor = cx.entity(); cx.defer(move |cx| { editor.update(cx, |editor, cx| { let kind = Editor::project_item_kind()?; let pane = editor.workspace()?.read(cx).pane_for(&cx.entity())?; let buffer = editor.buffer().read(cx).as_singleton()?; let file_abs_path = project::File::from_dyn(buffer.read(cx).file())?.abs_path(cx); pane.update(cx, |pane, _| { let data = pane .project_item_restoration_data .entry(kind) .or_insert_with(|| Box::new(EditorRestorationData::default()) as Box<_>); let data = match data.downcast_mut::() { Some(data) => data, None => { *data = Box::new(EditorRestorationData::default()); data.downcast_mut::() .expect("just written the type downcasted to") } }; let data = data.entries.entry(file_abs_path).or_default(); write(data); Some(()) }) }); }); } } pub(crate) enum BufferSearchHighlights {} impl SearchableItem for Editor { type Match = Range; fn get_matches(&self, _window: &mut Window, _: &mut App) -> Vec> { self.background_highlights .get(&HighlightKey::Type(TypeId::of::())) .map_or(Vec::new(), |(_color, ranges)| { ranges.iter().cloned().collect() }) } fn clear_matches(&mut self, _: &mut Window, cx: &mut Context) { if self .clear_background_highlights::(cx) .is_some() { cx.emit(SearchEvent::MatchesInvalidated); } } fn update_matches( &mut self, matches: &[Range], _: &mut Window, cx: &mut Context, ) { let existing_range = self .background_highlights .get(&HighlightKey::Type(TypeId::of::())) .map(|(_, range)| range.as_ref()); let updated = existing_range != Some(matches); self.highlight_background::( matches, |theme| theme.colors().search_match_background, cx, ); if updated { cx.emit(SearchEvent::MatchesInvalidated); } } fn has_filtered_search_ranges(&mut self) -> bool { self.has_background_highlights::() } fn toggle_filtered_search_ranges( &mut self, enabled: bool, _: &mut Window, cx: &mut Context, ) { if self.has_filtered_search_ranges() { self.previous_search_ranges = self .clear_background_highlights::(cx) .map(|(_, ranges)| ranges) } if !enabled { return; } let ranges = self.selections.disjoint_anchor_ranges().collect::>(); if ranges.iter().any(|s| s.start != s.end) { self.set_search_within_ranges(&ranges, cx); } else if let Some(previous_search_ranges) = self.previous_search_ranges.take() { self.set_search_within_ranges(&previous_search_ranges, cx) } } fn supported_options(&self) -> SearchOptions { if self.in_project_search { SearchOptions { case: true, word: true, regex: true, replacement: false, selection: false, find_in_results: true, } } else { SearchOptions { case: true, word: true, regex: true, replacement: true, selection: true, find_in_results: false, } } } fn query_suggestion(&mut self, window: &mut Window, cx: &mut Context) -> String { let setting = EditorSettings::get_global(cx).seed_search_query_from_cursor; let snapshot = &self.snapshot(window, cx).buffer_snapshot; let selection = self.selections.newest_adjusted(cx); match setting { SeedQuerySetting::Never => String::new(), SeedQuerySetting::Selection | SeedQuerySetting::Always if !selection.is_empty() => { let text: String = snapshot .text_for_range(selection.start..selection.end) .collect(); if text.contains('\n') { String::new() } else { text } } SeedQuerySetting::Selection => String::new(), SeedQuerySetting::Always => { let (range, kind) = snapshot.surrounding_word(selection.start, true); if kind == Some(CharKind::Word) { let text: String = snapshot.text_for_range(range).collect(); if !text.trim().is_empty() { return text; } } String::new() } } } fn activate_match( &mut self, index: usize, matches: &[Range], window: &mut Window, cx: &mut Context, ) { self.unfold_ranges(&[matches[index].clone()], false, true, cx); let range = self.range_for_match(&matches[index]); self.change_selections(Default::default(), window, cx, |s| { s.select_ranges([range]); }) } fn select_matches( &mut self, matches: &[Self::Match], window: &mut Window, cx: &mut Context, ) { self.unfold_ranges(matches, false, false, cx); self.change_selections(SelectionEffects::no_scroll(), window, cx, |s| { s.select_ranges(matches.iter().cloned()) }); } fn replace( &mut self, identifier: &Self::Match, query: &SearchQuery, window: &mut Window, cx: &mut Context, ) { let text = self.buffer.read(cx); let text = text.snapshot(cx); let text = text.text_for_range(identifier.clone()).collect::>(); let text: Cow<_> = if text.len() == 1 { text.first().cloned().unwrap().into() } else { let joined_chunks = text.join(""); joined_chunks.into() }; if let Some(replacement) = query.replacement_for(&text) { self.transact(window, cx, |this, _, cx| { this.edit([(identifier.clone(), Arc::from(&*replacement))], cx); }); } } fn replace_all( &mut self, matches: &mut dyn Iterator, query: &SearchQuery, window: &mut Window, cx: &mut Context, ) { let text = self.buffer.read(cx); let text = text.snapshot(cx); let mut edits = vec![]; for m in matches { let text = text.text_for_range(m.clone()).collect::>(); let text: Cow<_> = if text.len() == 1 { text.first().cloned().unwrap().into() } else { let joined_chunks = text.join(""); joined_chunks.into() }; if let Some(replacement) = query.replacement_for(&text) { edits.push((m.clone(), Arc::from(&*replacement))); } } if !edits.is_empty() { self.transact(window, cx, |this, _, cx| { this.edit(edits, cx); }); } } fn match_index_for_direction( &mut self, matches: &[Range], current_index: usize, direction: Direction, count: usize, _: &mut Window, cx: &mut Context, ) -> usize { let buffer = self.buffer().read(cx).snapshot(cx); let current_index_position = if self.selections.disjoint_anchors().len() == 1 { self.selections.newest_anchor().head() } else { matches[current_index].start }; let mut count = count % matches.len(); if count == 0 { return current_index; } match direction { Direction::Next => { if matches[current_index] .start .cmp(¤t_index_position, &buffer) .is_gt() { count -= 1 } (current_index + count) % matches.len() } Direction::Prev => { if matches[current_index] .end .cmp(¤t_index_position, &buffer) .is_lt() { count -= 1; } if current_index >= count { current_index - count } else { matches.len() - (count - current_index) } } } } fn find_matches( &mut self, query: Arc, _: &mut Window, cx: &mut Context, ) -> Task>> { let buffer = self.buffer().read(cx).snapshot(cx); let search_within_ranges = self .background_highlights .get(&HighlightKey::Type(TypeId::of::())) .map_or(vec![], |(_color, ranges)| { ranges.iter().cloned().collect::>() }); cx.background_spawn(async move { let mut ranges = Vec::new(); let search_within_ranges = if search_within_ranges.is_empty() { vec![buffer.anchor_before(0)..buffer.anchor_after(buffer.len())] } else { search_within_ranges }; for range in search_within_ranges { for (search_buffer, search_range, excerpt_id, deleted_hunk_anchor) in buffer.range_to_buffer_ranges_with_deleted_hunks(range) { ranges.extend( query .search(search_buffer, Some(search_range.clone())) .await .into_iter() .map(|match_range| { if let Some(deleted_hunk_anchor) = deleted_hunk_anchor { let start = search_buffer .anchor_after(search_range.start + match_range.start); let end = search_buffer .anchor_before(search_range.start + match_range.end); Anchor { diff_base_anchor: Some(start), ..deleted_hunk_anchor }..Anchor { diff_base_anchor: Some(end), ..deleted_hunk_anchor } } else { let start = search_buffer .anchor_after(search_range.start + match_range.start); let end = search_buffer .anchor_before(search_range.start + match_range.end); Anchor::range_in_buffer( excerpt_id, search_buffer.remote_id(), start..end, ) } }), ); } } ranges }) } fn active_match_index( &mut self, direction: Direction, matches: &[Range], _: &mut Window, cx: &mut Context, ) -> Option { active_match_index( direction, matches, &self.selections.newest_anchor().head(), &self.buffer().read(cx).snapshot(cx), ) } fn search_bar_visibility_changed(&mut self, _: bool, _: &mut Window, _: &mut Context) { self.expect_bounds_change = self.last_bounds; } } pub fn active_match_index( direction: Direction, ranges: &[Range], cursor: &Anchor, buffer: &MultiBufferSnapshot, ) -> Option { if ranges.is_empty() { None } else { let r = ranges.binary_search_by(|probe| { if probe.end.cmp(cursor, buffer).is_lt() { Ordering::Less } else if probe.start.cmp(cursor, buffer).is_gt() { Ordering::Greater } else { Ordering::Equal } }); match direction { Direction::Prev => match r { Ok(i) => Some(i), Err(i) => Some(i.saturating_sub(1)), }, Direction::Next => match r { Ok(i) | Err(i) => Some(cmp::min(i, ranges.len() - 1)), }, } } } pub fn entry_label_color(selected: bool) -> Color { if selected { Color::Default } else { Color::Muted } } pub fn entry_diagnostic_aware_icon_name_and_color( diagnostic_severity: Option, ) -> Option<(IconName, Color)> { match diagnostic_severity { Some(DiagnosticSeverity::ERROR) => Some((IconName::Close, Color::Error)), Some(DiagnosticSeverity::WARNING) => Some((IconName::Triangle, Color::Warning)), _ => None, } } pub fn entry_diagnostic_aware_icon_decoration_and_color( diagnostic_severity: Option, ) -> Option<(IconDecorationKind, Color)> { match diagnostic_severity { Some(DiagnosticSeverity::ERROR) => Some((IconDecorationKind::X, Color::Error)), Some(DiagnosticSeverity::WARNING) => Some((IconDecorationKind::Triangle, Color::Warning)), _ => None, } } pub fn entry_git_aware_label_color(git_status: GitSummary, ignored: bool, selected: bool) -> Color { let tracked = git_status.index + git_status.worktree; if ignored { Color::Ignored } else if git_status.conflict > 0 { Color::Conflict } else if tracked.modified > 0 { Color::Modified } else if tracked.added > 0 || git_status.untracked > 0 { Color::Created } else { entry_label_color(selected) } } fn path_for_buffer<'a>( buffer: &Entity, height: usize, include_filename: bool, cx: &'a App, ) -> Option> { let file = buffer.read(cx).as_singleton()?.read(cx).file()?; path_for_file(file.as_ref(), height, include_filename, cx) } fn path_for_file<'a>( file: &'a dyn language::File, mut height: usize, include_filename: bool, cx: &'a App, ) -> Option> { // Ensure we always render at least the filename. height += 1; let mut prefix = file.path().as_ref(); while height > 0 { if let Some(parent) = prefix.parent() { prefix = parent; height -= 1; } else { break; } } // Here we could have just always used `full_path`, but that is very // allocation-heavy and so we try to use a `Cow` if we haven't // traversed all the way up to the worktree's root. if height > 0 { let full_path = file.full_path(cx); if include_filename { Some(full_path.into()) } else { Some(full_path.parent()?.to_path_buf().into()) } } else { let mut path = file.path().strip_prefix(prefix).ok()?; if !include_filename { path = path.parent()?; } Some(path.into()) } } #[cfg(test)] mod tests { use crate::editor_tests::init_test; use fs::Fs; use super::*; use fs::MTime; use gpui::{App, VisualTestContext}; use language::{LanguageMatcher, TestFile}; use project::FakeFs; use std::path::{Path, PathBuf}; use util::path; #[gpui::test] fn test_path_for_file(cx: &mut App) { let file = TestFile { path: Path::new("").into(), root_name: String::new(), local_root: None, }; assert_eq!(path_for_file(&file, 0, false, cx), None); } async fn deserialize_editor( item_id: ItemId, workspace_id: WorkspaceId, workspace: Entity, project: Entity, cx: &mut VisualTestContext, ) -> Entity { workspace .update_in(cx, |workspace, window, cx| { let pane = workspace.active_pane(); pane.update(cx, |_, cx| { Editor::deserialize( project.clone(), workspace.weak_handle(), workspace_id, item_id, window, cx, ) }) }) .await .unwrap() } fn rust_language() -> Arc { Arc::new(language::Language::new( language::LanguageConfig { name: "Rust".into(), matcher: LanguageMatcher { path_suffixes: vec!["rs".to_string()], ..Default::default() }, ..Default::default() }, Some(tree_sitter_rust::LANGUAGE.into()), )) } #[gpui::test] async fn test_deserialize(cx: &mut gpui::TestAppContext) { init_test(cx, |_| {}); let fs = FakeFs::new(cx.executor()); fs.insert_file(path!("/file.rs"), Default::default()).await; // Test case 1: Deserialize with path and contents { let project = Project::test(fs.clone(), [path!("/file.rs").as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let workspace_id = workspace::WORKSPACE_DB.next_id().await.unwrap(); let item_id = 1234 as ItemId; let mtime = fs .metadata(Path::new(path!("/file.rs"))) .await .unwrap() .unwrap() .mtime; let serialized_editor = SerializedEditor { abs_path: Some(PathBuf::from(path!("/file.rs"))), contents: Some("fn main() {}".to_string()), language: Some("Rust".to_string()), mtime: Some(mtime), }; DB.save_serialized_editor(item_id, workspace_id, serialized_editor.clone()) .await .unwrap(); let deserialized = deserialize_editor(item_id, workspace_id, workspace, project, cx).await; deserialized.update(cx, |editor, cx| { assert_eq!(editor.text(cx), "fn main() {}"); assert!(editor.is_dirty(cx)); assert!(!editor.has_conflict(cx)); let buffer = editor.buffer().read(cx).as_singleton().unwrap().read(cx); assert!(buffer.file().is_some()); }); } // Test case 2: Deserialize with only path { let project = Project::test(fs.clone(), [path!("/file.rs").as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let workspace_id = workspace::WORKSPACE_DB.next_id().await.unwrap(); let item_id = 5678 as ItemId; let serialized_editor = SerializedEditor { abs_path: Some(PathBuf::from(path!("/file.rs"))), contents: None, language: None, mtime: None, }; DB.save_serialized_editor(item_id, workspace_id, serialized_editor) .await .unwrap(); let deserialized = deserialize_editor(item_id, workspace_id, workspace, project, cx).await; deserialized.update(cx, |editor, cx| { assert_eq!(editor.text(cx), ""); // The file should be empty as per our initial setup assert!(!editor.is_dirty(cx)); assert!(!editor.has_conflict(cx)); let buffer = editor.buffer().read(cx).as_singleton().unwrap().read(cx); assert!(buffer.file().is_some()); }); } // Test case 3: Deserialize with no path (untitled buffer, with content and language) { let project = Project::test(fs.clone(), [path!("/file.rs").as_ref()], cx).await; // Add Rust to the language, so that we can restore the language of the buffer project.read_with(cx, |project, _| project.languages().add(rust_language())); let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let workspace_id = workspace::WORKSPACE_DB.next_id().await.unwrap(); let item_id = 9012 as ItemId; let serialized_editor = SerializedEditor { abs_path: None, contents: Some("hello".to_string()), language: Some("Rust".to_string()), mtime: None, }; DB.save_serialized_editor(item_id, workspace_id, serialized_editor) .await .unwrap(); let deserialized = deserialize_editor(item_id, workspace_id, workspace, project, cx).await; deserialized.update(cx, |editor, cx| { assert_eq!(editor.text(cx), "hello"); assert!(editor.is_dirty(cx)); // The editor should be dirty for an untitled buffer let buffer = editor.buffer().read(cx).as_singleton().unwrap().read(cx); assert_eq!( buffer.language().map(|lang| lang.name()), Some("Rust".into()) ); // Language should be set to Rust assert!(buffer.file().is_none()); // The buffer should not have an associated file }); } // Test case 4: Deserialize with path, content, and old mtime { let project = Project::test(fs.clone(), [path!("/file.rs").as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let workspace_id = workspace::WORKSPACE_DB.next_id().await.unwrap(); let item_id = 9345 as ItemId; let old_mtime = MTime::from_seconds_and_nanos(0, 50); let serialized_editor = SerializedEditor { abs_path: Some(PathBuf::from(path!("/file.rs"))), contents: Some("fn main() {}".to_string()), language: Some("Rust".to_string()), mtime: Some(old_mtime), }; DB.save_serialized_editor(item_id, workspace_id, serialized_editor) .await .unwrap(); let deserialized = deserialize_editor(item_id, workspace_id, workspace, project, cx).await; deserialized.update(cx, |editor, cx| { assert_eq!(editor.text(cx), "fn main() {}"); assert!(editor.has_conflict(cx)); // The editor should have a conflict }); } // Test case 5: Deserialize with no path, no content, no language, and no old mtime (new, empty, unsaved buffer) { let project = Project::test(fs.clone(), [path!("/file.rs").as_ref()], cx).await; let (workspace, cx) = cx.add_window_view(|window, cx| Workspace::test_new(project.clone(), window, cx)); let workspace_id = workspace::WORKSPACE_DB.next_id().await.unwrap(); let item_id = 10000 as ItemId; let serialized_editor = SerializedEditor { abs_path: None, contents: None, language: None, mtime: None, }; DB.save_serialized_editor(item_id, workspace_id, serialized_editor) .await .unwrap(); let deserialized = deserialize_editor(item_id, workspace_id, workspace, project, cx).await; deserialized.update(cx, |editor, cx| { assert_eq!(editor.text(cx), ""); assert!(!editor.is_dirty(cx)); assert!(!editor.has_conflict(cx)); let buffer = editor.buffer().read(cx).as_singleton().unwrap().read(cx); assert!(buffer.file().is_none()); }); } } }