mod persistence; pub mod terminal_element; pub mod terminal_panel; use collections::HashSet; use editor::{scroll::Autoscroll, Editor}; use futures::{stream::FuturesUnordered, StreamExt}; use gpui::{ anchored, deferred, div, impl_actions, AnyElement, AppContext, DismissEvent, EventEmitter, FocusHandle, FocusableView, KeyContext, KeyDownEvent, Keystroke, Model, MouseButton, MouseDownEvent, Pixels, Render, Styled, Subscription, Task, View, VisualContext, WeakView, }; use language::Bias; use persistence::TERMINAL_DB; use project::{search::SearchQuery, Fs, LocalWorktree, Metadata, Project}; use settings::SettingsStore; use terminal::{ alacritty_terminal::{ index::Point, term::{search::RegexSearch, TermMode}, }, terminal_settings::{TerminalBlink, TerminalSettings, WorkingDirectory}, Clear, Copy, Event, MaybeNavigationTarget, Paste, ShowCharacterPalette, TaskStatus, Terminal, }; use terminal_element::TerminalElement; use ui::{h_flex, prelude::*, ContextMenu, Icon, IconName, Label}; use util::{paths::PathLikeWithPosition, ResultExt}; use workspace::{ item::{BreadcrumbText, Item, ItemEvent, TabContentParams}, notifications::NotifyResultExt, register_deserializable_item, searchable::{SearchEvent, SearchOptions, SearchableItem, SearchableItemHandle}, CloseActiveItem, NewCenterTerminal, OpenVisible, Pane, ToolbarItemLocation, Workspace, WorkspaceId, }; use anyhow::Context; use dirs::home_dir; use serde::Deserialize; use settings::Settings; use smol::Timer; use std::{ ops::RangeInclusive, path::{Path, PathBuf}, sync::Arc, time::Duration, }; const REGEX_SPECIAL_CHARS: &[char] = &[ '\\', '.', '*', '+', '?', '|', '(', ')', '[', ']', '{', '}', '^', '$', ]; const CURSOR_BLINK_INTERVAL: Duration = Duration::from_millis(500); ///Event to transmit the scroll from the element to the view #[derive(Clone, Debug, PartialEq)] pub struct ScrollTerminal(pub i32); #[derive(Clone, Debug, Default, Deserialize, PartialEq)] pub struct SendText(String); #[derive(Clone, Debug, Default, Deserialize, PartialEq)] pub struct SendKeystroke(String); impl_actions!(terminal, [SendText, SendKeystroke]); pub fn init(cx: &mut AppContext) { terminal_panel::init(cx); terminal::init(cx); register_deserializable_item::(cx); cx.observe_new_views(|workspace: &mut Workspace, _| { workspace.register_action(TerminalView::deploy); }) .detach(); } ///A terminal view, maintains the PTY's file handles and communicates with the terminal pub struct TerminalView { terminal: Model, workspace: WeakView, focus_handle: FocusHandle, //Currently using iTerm bell, show bell emoji in tab until input is received has_bell: bool, context_menu: Option<(View, gpui::Point, Subscription)>, blink_state: bool, blinking_on: bool, blinking_paused: bool, blink_epoch: usize, can_navigate_to_selected_word: bool, workspace_id: WorkspaceId, show_title: bool, _subscriptions: Vec, _terminal_subscriptions: Vec, } impl EventEmitter for TerminalView {} impl EventEmitter for TerminalView {} impl EventEmitter for TerminalView {} impl FocusableView for TerminalView { fn focus_handle(&self, _cx: &AppContext) -> FocusHandle { self.focus_handle.clone() } } impl TerminalView { ///Create a new Terminal in the current working directory or the user's home directory pub fn deploy( workspace: &mut Workspace, _: &NewCenterTerminal, cx: &mut ViewContext, ) { let strategy = TerminalSettings::get_global(cx); let working_directory = get_working_directory(workspace, cx, strategy.working_directory.clone()); let window = cx.window_handle(); let terminal = workspace .project() .update(cx, |project, cx| { project.create_terminal(working_directory, None, window, cx) }) .notify_err(workspace, cx); if let Some(terminal) = terminal { let view = cx.new_view(|cx| { TerminalView::new( terminal, workspace.weak_handle(), workspace.database_id(), cx, ) }); workspace.add_item_to_active_pane(Box::new(view), cx) } } pub fn new( terminal: Model, workspace: WeakView, workspace_id: WorkspaceId, cx: &mut ViewContext, ) -> Self { let workspace_handle = workspace.clone(); let terminal_subscriptions = subscribe_for_terminal_events(&terminal, workspace, cx); let focus_handle = cx.focus_handle(); let focus_in = cx.on_focus_in(&focus_handle, |terminal_view, cx| { terminal_view.focus_in(cx); }); let focus_out = cx.on_focus_out(&focus_handle, |terminal_view, cx| { terminal_view.focus_out(cx); }); Self { terminal, workspace: workspace_handle, has_bell: false, focus_handle, context_menu: None, blink_state: true, blinking_on: false, blinking_paused: false, blink_epoch: 0, can_navigate_to_selected_word: false, workspace_id, show_title: TerminalSettings::get_global(cx).toolbar.title, _subscriptions: vec![ focus_in, focus_out, cx.observe_global::(Self::settings_changed), ], _terminal_subscriptions: terminal_subscriptions, } } pub fn model(&self) -> &Model { &self.terminal } pub fn has_bell(&self) -> bool { self.has_bell } pub fn clear_bel(&mut self, cx: &mut ViewContext) { self.has_bell = false; cx.emit(Event::Wakeup); } pub fn deploy_context_menu( &mut self, position: gpui::Point, cx: &mut ViewContext, ) { let context_menu = ContextMenu::build(cx, |menu, _| { menu.action("Clear", Box::new(Clear)) .action("Close", Box::new(CloseActiveItem { save_intent: None })) }); cx.focus_view(&context_menu); let subscription = cx.subscribe(&context_menu, |this, _, _: &DismissEvent, cx| { if this.context_menu.as_ref().is_some_and(|context_menu| { context_menu.0.focus_handle(cx).contains_focused(cx) }) { cx.focus_self(); } this.context_menu.take(); cx.notify(); }); self.context_menu = Some((context_menu, position, subscription)); } fn settings_changed(&mut self, cx: &mut ViewContext) { let settings = TerminalSettings::get_global(cx); self.show_title = settings.toolbar.title; cx.notify(); } fn show_character_palette(&mut self, _: &ShowCharacterPalette, cx: &mut ViewContext) { if self .terminal .read(cx) .last_content .mode .contains(TermMode::ALT_SCREEN) { self.terminal.update(cx, |term, cx| { term.try_keystroke( &Keystroke::parse("ctrl-cmd-space").unwrap(), TerminalSettings::get_global(cx).option_as_meta, ) }); } else { cx.show_character_palette(); } } fn select_all(&mut self, _: &editor::actions::SelectAll, cx: &mut ViewContext) { self.terminal.update(cx, |term, _| term.select_all()); cx.notify(); } fn clear(&mut self, _: &Clear, cx: &mut ViewContext) { self.terminal.update(cx, |term, _| term.clear()); cx.notify(); } pub fn should_show_cursor(&self, focused: bool, cx: &mut gpui::ViewContext) -> bool { //Don't blink the cursor when not focused, blinking is disabled, or paused if !focused || !self.blinking_on || self.blinking_paused || self .terminal .read(cx) .last_content .mode .contains(TermMode::ALT_SCREEN) { return true; } match TerminalSettings::get_global(cx).blinking { //If the user requested to never blink, don't blink it. TerminalBlink::Off => true, //If the terminal is controlling it, check terminal mode TerminalBlink::TerminalControlled | TerminalBlink::On => self.blink_state, } } fn blink_cursors(&mut self, epoch: usize, cx: &mut ViewContext) { if epoch == self.blink_epoch && !self.blinking_paused { self.blink_state = !self.blink_state; cx.notify(); let epoch = self.next_blink_epoch(); cx.spawn(|this, mut cx| async move { Timer::after(CURSOR_BLINK_INTERVAL).await; this.update(&mut cx, |this, cx| this.blink_cursors(epoch, cx)) .log_err(); }) .detach(); } } pub fn pause_cursor_blinking(&mut self, cx: &mut ViewContext) { self.blink_state = true; cx.notify(); let epoch = self.next_blink_epoch(); cx.spawn(|this, mut cx| async move { Timer::after(CURSOR_BLINK_INTERVAL).await; this.update(&mut cx, |this, cx| this.resume_cursor_blinking(epoch, cx)) .ok(); }) .detach(); } pub fn terminal(&self) -> &Model { &self.terminal } fn next_blink_epoch(&mut self) -> usize { self.blink_epoch += 1; self.blink_epoch } fn resume_cursor_blinking(&mut self, epoch: usize, cx: &mut ViewContext) { if epoch == self.blink_epoch { self.blinking_paused = false; self.blink_cursors(epoch, cx); } } ///Attempt to paste the clipboard into the terminal fn copy(&mut self, _: &Copy, cx: &mut ViewContext) { self.terminal.update(cx, |term, _| term.copy()); cx.notify(); } ///Attempt to paste the clipboard into the terminal fn paste(&mut self, _: &Paste, cx: &mut ViewContext) { if let Some(item) = cx.read_from_clipboard() { self.terminal .update(cx, |terminal, _cx| terminal.paste(item.text())); } } fn send_text(&mut self, text: &SendText, cx: &mut ViewContext) { self.clear_bel(cx); self.terminal.update(cx, |term, _| { term.input(text.0.to_string()); }); } fn send_keystroke(&mut self, text: &SendKeystroke, cx: &mut ViewContext) { if let Some(keystroke) = Keystroke::parse(&text.0).log_err() { self.clear_bel(cx); self.terminal.update(cx, |term, cx| { term.try_keystroke(&keystroke, TerminalSettings::get_global(cx).option_as_meta); }); } } fn dispatch_context(&self, cx: &AppContext) -> KeyContext { let mut dispatch_context = KeyContext::default(); dispatch_context.add("Terminal"); let mode = self.terminal.read(cx).last_content.mode; dispatch_context.set( "screen", if mode.contains(TermMode::ALT_SCREEN) { "alt" } else { "normal" }, ); if mode.contains(TermMode::APP_CURSOR) { dispatch_context.add("DECCKM"); } if mode.contains(TermMode::APP_KEYPAD) { dispatch_context.add("DECPAM"); } else { dispatch_context.add("DECPNM"); } if mode.contains(TermMode::SHOW_CURSOR) { dispatch_context.add("DECTCEM"); } if mode.contains(TermMode::LINE_WRAP) { dispatch_context.add("DECAWM"); } if mode.contains(TermMode::ORIGIN) { dispatch_context.add("DECOM"); } if mode.contains(TermMode::INSERT) { dispatch_context.add("IRM"); } //LNM is apparently the name for this. https://vt100.net/docs/vt510-rm/LNM.html if mode.contains(TermMode::LINE_FEED_NEW_LINE) { dispatch_context.add("LNM"); } if mode.contains(TermMode::FOCUS_IN_OUT) { dispatch_context.add("report_focus"); } if mode.contains(TermMode::ALTERNATE_SCROLL) { dispatch_context.add("alternate_scroll"); } if mode.contains(TermMode::BRACKETED_PASTE) { dispatch_context.add("bracketed_paste"); } if mode.intersects(TermMode::MOUSE_MODE) { dispatch_context.add("any_mouse_reporting"); } { let mouse_reporting = if mode.contains(TermMode::MOUSE_REPORT_CLICK) { "click" } else if mode.contains(TermMode::MOUSE_DRAG) { "drag" } else if mode.contains(TermMode::MOUSE_MOTION) { "motion" } else { "off" }; dispatch_context.set("mouse_reporting", mouse_reporting); } { let format = if mode.contains(TermMode::SGR_MOUSE) { "sgr" } else if mode.contains(TermMode::UTF8_MOUSE) { "utf8" } else { "normal" }; dispatch_context.set("mouse_format", format); }; dispatch_context } fn set_terminal(&mut self, terminal: Model, cx: &mut ViewContext<'_, TerminalView>) { self._terminal_subscriptions = subscribe_for_terminal_events(&terminal, self.workspace.clone(), cx); self.terminal = terminal; } } fn subscribe_for_terminal_events( terminal: &Model, workspace: WeakView, cx: &mut ViewContext<'_, TerminalView>, ) -> Vec { let terminal_subscription = cx.observe(terminal, |_, _, cx| cx.notify()); let terminal_events_subscription = cx.subscribe(terminal, move |this, _, event, cx| match event { Event::Wakeup => { cx.notify(); cx.emit(Event::Wakeup); cx.emit(ItemEvent::UpdateTab); cx.emit(SearchEvent::MatchesInvalidated); } Event::Bell => { this.has_bell = true; cx.emit(Event::Wakeup); } Event::BlinkChanged => this.blinking_on = !this.blinking_on, Event::TitleChanged => { cx.emit(ItemEvent::UpdateTab); let terminal = this.terminal().read(cx); if terminal.task().is_none() { if let Some(cwd) = terminal.get_cwd() { let item_id = cx.entity_id(); let workspace_id = this.workspace_id; cx.background_executor() .spawn(async move { TERMINAL_DB .save_working_directory(item_id.as_u64(), workspace_id, cwd) .await .log_err(); }) .detach(); } } } Event::NewNavigationTarget(maybe_navigation_target) => { this.can_navigate_to_selected_word = match maybe_navigation_target { Some(MaybeNavigationTarget::Url(_)) => true, Some(MaybeNavigationTarget::PathLike(path_like_target)) => { if let Ok(fs) = workspace.update(cx, |workspace, cx| { workspace.project().read(cx).fs().clone() }) { let valid_files_to_open_task = possible_open_targets( fs, &workspace, &path_like_target.terminal_dir, &path_like_target.maybe_path, cx, ); smol::block_on(valid_files_to_open_task).len() > 0 } else { false } } None => false, } } Event::Open(maybe_navigation_target) => match maybe_navigation_target { MaybeNavigationTarget::Url(url) => cx.open_url(url), MaybeNavigationTarget::PathLike(path_like_target) => { if !this.can_navigate_to_selected_word { return; } let task_workspace = workspace.clone(); let Some(fs) = workspace .update(cx, |workspace, cx| { workspace.project().read(cx).fs().clone() }) .ok() else { return; }; let path_like_target = path_like_target.clone(); cx.spawn(|terminal_view, mut cx| async move { let valid_files_to_open = terminal_view .update(&mut cx, |_, cx| { possible_open_targets( fs, &task_workspace, &path_like_target.terminal_dir, &path_like_target.maybe_path, cx, ) })? .await; let paths_to_open = valid_files_to_open .iter() .map(|(p, _)| p.path_like.clone()) .collect(); let opened_items = task_workspace .update(&mut cx, |workspace, cx| { workspace.open_paths( paths_to_open, OpenVisible::OnlyDirectories, None, cx, ) }) .context("workspace update")? .await; let mut has_dirs = false; for ((path, metadata), opened_item) in valid_files_to_open .into_iter() .zip(opened_items.into_iter()) { if metadata.is_dir { has_dirs = true; } else if let Some(Ok(opened_item)) = opened_item { if let Some(row) = path.row { let col = path.column.unwrap_or(0); if let Some(active_editor) = opened_item.downcast::() { active_editor .downgrade() .update(&mut cx, |editor, cx| { let snapshot = editor.snapshot(cx).display_snapshot; let point = snapshot.buffer_snapshot.clip_point( language::Point::new( row.saturating_sub(1), col.saturating_sub(1), ), Bias::Left, ); editor.change_selections( Some(Autoscroll::center()), cx, |s| s.select_ranges([point..point]), ); }) .log_err(); } } } } if has_dirs { task_workspace.update(&mut cx, |workspace, cx| { workspace.project().update(cx, |_, cx| { cx.emit(project::Event::ActivateProjectPanel); }) })?; } anyhow::Ok(()) }) .detach_and_log_err(cx) } }, Event::BreadcrumbsChanged => cx.emit(ItemEvent::UpdateBreadcrumbs), Event::CloseTerminal => cx.emit(ItemEvent::CloseItem), Event::SelectionsChanged => cx.emit(SearchEvent::ActiveMatchChanged), }); vec![terminal_subscription, terminal_events_subscription] } fn possible_open_paths_metadata( fs: Arc, row: Option, column: Option, potential_paths: HashSet, cx: &mut ViewContext, ) -> Task, Metadata)>> { cx.background_executor().spawn(async move { let mut paths_with_metadata = Vec::with_capacity(potential_paths.len()); let mut fetch_metadata_tasks = potential_paths .into_iter() .map(|potential_path| async { let metadata = fs.metadata(&potential_path).await.ok().flatten(); ( PathLikeWithPosition { path_like: potential_path, row, column, }, metadata, ) }) .collect::>(); while let Some((path, metadata)) = fetch_metadata_tasks.next().await { if let Some(metadata) = metadata { paths_with_metadata.push((path, metadata)); } } paths_with_metadata }) } fn possible_open_targets( fs: Arc, workspace: &WeakView, cwd: &Option, maybe_path: &String, cx: &mut ViewContext, ) -> Task, Metadata)>> { let path_like = PathLikeWithPosition::parse_str(maybe_path.as_str(), |path_str| { Ok::<_, std::convert::Infallible>(Path::new(path_str).to_path_buf()) }) .expect("infallible"); let row = path_like.row; let column = path_like.column; let maybe_path = path_like.path_like; let potential_abs_paths = if maybe_path.is_absolute() { HashSet::from_iter([maybe_path]) } else if maybe_path.starts_with("~") { if let Some(abs_path) = maybe_path .strip_prefix("~") .ok() .and_then(|maybe_path| Some(dirs::home_dir()?.join(maybe_path))) { HashSet::from_iter([abs_path]) } else { HashSet::default() } } else { // First check cwd and then workspace let mut potential_cwd_and_workspace_paths = HashSet::default(); if let Some(cwd) = cwd { potential_cwd_and_workspace_paths.insert(Path::join(cwd, &maybe_path)); } if let Some(workspace) = workspace.upgrade() { workspace.update(cx, |workspace, cx| { for potential_worktree_path in workspace .worktrees(cx) .map(|worktree| worktree.read(cx).abs_path().join(&maybe_path)) { potential_cwd_and_workspace_paths.insert(potential_worktree_path); } }); } potential_cwd_and_workspace_paths }; possible_open_paths_metadata(fs, row, column, potential_abs_paths, cx) } fn regex_to_literal(regex: &str) -> String { regex .chars() .flat_map(|c| { if REGEX_SPECIAL_CHARS.contains(&c) { vec!['\\', c] } else { vec![c] } }) .collect() } pub fn regex_search_for_query(query: &project::search::SearchQuery) -> Option { let query = query.as_str(); if query == "." { return None; } let searcher = RegexSearch::new(&query); searcher.ok() } impl TerminalView { fn key_down(&mut self, event: &KeyDownEvent, cx: &mut ViewContext) { self.clear_bel(cx); self.pause_cursor_blinking(cx); self.terminal.update(cx, |term, cx| { term.try_keystroke( &event.keystroke, TerminalSettings::get_global(cx).option_as_meta, ) }); } fn focus_in(&mut self, cx: &mut ViewContext) { self.terminal.read(cx).focus_in(); self.blink_cursors(self.blink_epoch, cx); cx.notify(); } fn focus_out(&mut self, cx: &mut ViewContext) { self.terminal.update(cx, |terminal, _| { terminal.focus_out(); }); cx.notify(); } } impl Render for TerminalView { fn render(&mut self, cx: &mut ViewContext) -> impl IntoElement { let terminal_handle = self.terminal.clone(); let focused = self.focus_handle.is_focused(cx); div() .size_full() .relative() .track_focus(&self.focus_handle) .key_context(self.dispatch_context(cx)) .on_action(cx.listener(TerminalView::send_text)) .on_action(cx.listener(TerminalView::send_keystroke)) .on_action(cx.listener(TerminalView::copy)) .on_action(cx.listener(TerminalView::paste)) .on_action(cx.listener(TerminalView::clear)) .on_action(cx.listener(TerminalView::show_character_palette)) .on_action(cx.listener(TerminalView::select_all)) .on_key_down(cx.listener(Self::key_down)) .on_mouse_down( MouseButton::Right, cx.listener(|this, event: &MouseDownEvent, cx| { if !this.terminal.read(cx).mouse_mode(event.modifiers.shift) { this.deploy_context_menu(event.position, cx); cx.notify(); } }), ) .child( // TODO: Oddly this wrapper div is needed for TerminalElement to not steal events from the context menu div().size_full().child(TerminalElement::new( terminal_handle, self.workspace.clone(), self.focus_handle.clone(), focused, self.should_show_cursor(focused, cx), self.can_navigate_to_selected_word, )), ) .children(self.context_menu.as_ref().map(|(menu, position, _)| { deferred( anchored() .position(*position) .anchor(gpui::AnchorCorner::TopLeft) .child(menu.clone()), ) .with_priority(1) })) } } impl Item for TerminalView { type Event = ItemEvent; fn tab_tooltip_text(&self, cx: &AppContext) -> Option { Some(self.terminal().read(cx).title(false).into()) } fn tab_content(&self, params: TabContentParams, cx: &WindowContext) -> AnyElement { let terminal = self.terminal().read(cx); let title = terminal.title(true); let icon = match terminal.task() { Some(terminal_task) => match &terminal_task.status { TaskStatus::Unknown => IconName::ExclamationTriangle, TaskStatus::Running => IconName::Play, TaskStatus::Completed { success } => { if *success { IconName::Check } else { IconName::XCircle } } }, None => IconName::Terminal, }; h_flex() .gap_2() .child(Icon::new(icon)) .child(Label::new(title).color(if params.selected { Color::Default } else { Color::Muted })) .into_any() } fn telemetry_event_text(&self) -> Option<&'static str> { None } fn clone_on_split( &self, _workspace_id: WorkspaceId, _cx: &mut ViewContext, ) -> Option> { //From what I can tell, there's no way to tell the current working //Directory of the terminal from outside the shell. There might be //solutions to this, but they are non-trivial and require more IPC // Some(TerminalContainer::new( // Err(anyhow::anyhow!("failed to instantiate terminal")), // workspace_id, // cx, // )) // TODO None } fn is_dirty(&self, cx: &gpui::AppContext) -> bool { match self.terminal.read(cx).task() { Some(task) => task.status == TaskStatus::Running, None => self.has_bell(), } } fn has_conflict(&self, _cx: &AppContext) -> bool { false } fn as_searchable(&self, handle: &View) -> Option> { Some(Box::new(handle.clone())) } fn breadcrumb_location(&self) -> ToolbarItemLocation { if self.show_title { ToolbarItemLocation::PrimaryLeft } else { ToolbarItemLocation::Hidden } } fn breadcrumbs(&self, _: &theme::Theme, cx: &AppContext) -> Option> { Some(vec![BreadcrumbText { text: self.terminal().read(cx).breadcrumb_text.clone(), highlights: None, font: None, }]) } fn serialized_item_kind() -> Option<&'static str> { Some("Terminal") } fn deserialize( project: Model, workspace: WeakView, workspace_id: workspace::WorkspaceId, item_id: workspace::ItemId, cx: &mut ViewContext, ) -> Task>> { let window = cx.window_handle(); cx.spawn(|pane, mut cx| async move { let cwd = TERMINAL_DB .get_working_directory(item_id, workspace_id) .log_err() .flatten() .or_else(|| { cx.update(|cx| { let strategy = TerminalSettings::get_global(cx).working_directory.clone(); workspace.upgrade().and_then(|workspace| { get_working_directory(workspace.read(cx), cx, strategy) }) }) .ok() .flatten() }) .filter(|cwd| !cwd.as_os_str().is_empty()); let terminal = project.update(&mut cx, |project, cx| { project.create_terminal(cwd, None, window, cx) })??; pane.update(&mut cx, |_, cx| { cx.new_view(|cx| TerminalView::new(terminal, workspace, workspace_id, cx)) }) }) } fn added_to_workspace(&mut self, workspace: &mut Workspace, cx: &mut ViewContext) { if self.terminal().read(cx).task().is_none() { cx.background_executor() .spawn(TERMINAL_DB.update_workspace_id( workspace.database_id(), self.workspace_id, cx.entity_id().as_u64(), )) .detach(); self.workspace_id = workspace.database_id(); } } fn to_item_events(event: &Self::Event, mut f: impl FnMut(ItemEvent)) { f(*event) } } impl SearchableItem for TerminalView { type Match = RangeInclusive; fn supported_options() -> SearchOptions { SearchOptions { case: false, word: false, regex: true, replacement: false, } } /// Clear stored matches fn clear_matches(&mut self, cx: &mut ViewContext) { self.terminal().update(cx, |term, _| term.matches.clear()) } /// Store matches returned from find_matches somewhere for rendering fn update_matches(&mut self, matches: &[Self::Match], cx: &mut ViewContext) { self.terminal() .update(cx, |term, _| term.matches = matches.to_vec()) } /// Returns the selection content to pre-load into this search fn query_suggestion(&mut self, cx: &mut ViewContext) -> String { self.terminal() .read(cx) .last_content .selection_text .clone() .unwrap_or_default() } /// Focus match at given index into the Vec of matches fn activate_match(&mut self, index: usize, _: &[Self::Match], cx: &mut ViewContext) { self.terminal() .update(cx, |term, _| term.activate_match(index)); cx.notify(); } /// Add selections for all matches given. fn select_matches(&mut self, matches: &[Self::Match], cx: &mut ViewContext) { self.terminal() .update(cx, |term, _| term.select_matches(matches)); cx.notify(); } /// Get all of the matches for this query, should be done on the background fn find_matches( &mut self, query: Arc, cx: &mut ViewContext, ) -> Task> { let searcher = match &*query { SearchQuery::Text { .. } => regex_search_for_query( &(SearchQuery::text( regex_to_literal(&query.as_str()), query.whole_word(), query.case_sensitive(), query.include_ignored(), query.files_to_include().to_vec(), query.files_to_exclude().to_vec(), ) .unwrap()), ), SearchQuery::Regex { .. } => regex_search_for_query(&query), }; if let Some(s) = searcher { self.terminal() .update(cx, |term, cx| term.find_matches(s, cx)) } else { Task::ready(vec![]) } } /// Reports back to the search toolbar what the active match should be (the selection) fn active_match_index( &mut self, matches: &[Self::Match], cx: &mut ViewContext, ) -> Option { // Selection head might have a value if there's a selection that isn't // associated with a match. Therefore, if there are no matches, we should // report None, no matter the state of the terminal let res = if matches.len() > 0 { if let Some(selection_head) = self.terminal().read(cx).selection_head { // If selection head is contained in a match. Return that match if let Some(ix) = matches .iter() .enumerate() .find(|(_, search_match)| { search_match.contains(&selection_head) || search_match.start() > &selection_head }) .map(|(ix, _)| ix) { Some(ix) } else { // If no selection after selection head, return the last match Some(matches.len().saturating_sub(1)) } } else { // Matches found but no active selection, return the first last one (closest to cursor) Some(matches.len().saturating_sub(1)) } } else { None }; res } fn replace(&mut self, _: &Self::Match, _: &SearchQuery, _: &mut ViewContext) { // Replacement is not supported in terminal view, so this is a no-op. } } ///Gets the working directory for the given workspace, respecting the user's settings. pub fn get_working_directory( workspace: &Workspace, cx: &AppContext, strategy: WorkingDirectory, ) -> Option { let res = match strategy { WorkingDirectory::CurrentProjectDirectory => current_project_directory(workspace, cx) .or_else(|| first_project_directory(workspace, cx)), WorkingDirectory::FirstProjectDirectory => first_project_directory(workspace, cx), WorkingDirectory::AlwaysHome => None, WorkingDirectory::Always { directory } => { shellexpand::full(&directory) //TODO handle this better .ok() .map(|dir| Path::new(&dir.to_string()).to_path_buf()) .filter(|dir| dir.is_dir()) } }; res.or_else(home_dir) } ///Gets the first project's home directory, or the home directory fn first_project_directory(workspace: &Workspace, cx: &AppContext) -> Option { workspace .worktrees(cx) .next() .and_then(|worktree_handle| worktree_handle.read(cx).as_local()) .and_then(get_path_from_wt) } ///Gets the intuitively correct working directory from the given workspace ///If there is an active entry for this project, returns that entry's worktree root. ///If there's no active entry but there is a worktree, returns that worktrees root. ///If either of these roots are files, or if there are any other query failures, /// returns the user's home directory fn current_project_directory(workspace: &Workspace, cx: &AppContext) -> Option { let project = workspace.project().read(cx); project .active_entry() .and_then(|entry_id| project.worktree_for_entry(entry_id, cx)) .or_else(|| workspace.worktrees(cx).next()) .and_then(|worktree_handle| worktree_handle.read(cx).as_local()) .and_then(get_path_from_wt) } fn get_path_from_wt(wt: &LocalWorktree) -> Option { wt.root_entry() .filter(|re| re.is_dir()) .map(|_| wt.abs_path().to_path_buf()) } #[cfg(test)] mod tests { use super::*; use gpui::TestAppContext; use project::{Entry, Project, ProjectPath, Worktree}; use std::path::Path; use workspace::AppState; // Working directory calculation tests // No Worktrees in project -> home_dir() #[gpui::test] async fn no_worktree(cx: &mut TestAppContext) { let (project, workspace) = init_test(cx).await; cx.read(|cx| { let workspace = workspace.read(cx); let active_entry = project.read(cx).active_entry(); //Make sure environment is as expected assert!(active_entry.is_none()); assert!(workspace.worktrees(cx).next().is_none()); let res = current_project_directory(workspace, cx); assert_eq!(res, None); let res = first_project_directory(workspace, cx); assert_eq!(res, None); }); } // No active entry, but a worktree, worktree is a file -> home_dir() #[gpui::test] async fn no_active_entry_worktree_is_file(cx: &mut TestAppContext) { let (project, workspace) = init_test(cx).await; create_file_wt(project.clone(), "/root.txt", cx).await; cx.read(|cx| { let workspace = workspace.read(cx); let active_entry = project.read(cx).active_entry(); //Make sure environment is as expected assert!(active_entry.is_none()); assert!(workspace.worktrees(cx).next().is_some()); let res = current_project_directory(workspace, cx); assert_eq!(res, None); let res = first_project_directory(workspace, cx); assert_eq!(res, None); }); } // No active entry, but a worktree, worktree is a folder -> worktree_folder #[gpui::test] async fn no_active_entry_worktree_is_dir(cx: &mut TestAppContext) { let (project, workspace) = init_test(cx).await; let (_wt, _entry) = create_folder_wt(project.clone(), "/root/", cx).await; cx.update(|cx| { let workspace = workspace.read(cx); let active_entry = project.read(cx).active_entry(); assert!(active_entry.is_none()); assert!(workspace.worktrees(cx).next().is_some()); let res = current_project_directory(workspace, cx); assert_eq!(res, Some((Path::new("/root/")).to_path_buf())); let res = first_project_directory(workspace, cx); assert_eq!(res, Some((Path::new("/root/")).to_path_buf())); }); } // Active entry with a work tree, worktree is a file -> home_dir() #[gpui::test] async fn active_entry_worktree_is_file(cx: &mut TestAppContext) { let (project, workspace) = init_test(cx).await; let (_wt, _entry) = create_folder_wt(project.clone(), "/root1/", cx).await; let (wt2, entry2) = create_file_wt(project.clone(), "/root2.txt", cx).await; insert_active_entry_for(wt2, entry2, project.clone(), cx); cx.update(|cx| { let workspace = workspace.read(cx); let active_entry = project.read(cx).active_entry(); assert!(active_entry.is_some()); let res = current_project_directory(workspace, cx); assert_eq!(res, None); let res = first_project_directory(workspace, cx); assert_eq!(res, Some((Path::new("/root1/")).to_path_buf())); }); } // Active entry, with a worktree, worktree is a folder -> worktree_folder #[gpui::test] async fn active_entry_worktree_is_dir(cx: &mut TestAppContext) { let (project, workspace) = init_test(cx).await; let (_wt, _entry) = create_folder_wt(project.clone(), "/root1/", cx).await; let (wt2, entry2) = create_folder_wt(project.clone(), "/root2/", cx).await; insert_active_entry_for(wt2, entry2, project.clone(), cx); cx.update(|cx| { let workspace = workspace.read(cx); let active_entry = project.read(cx).active_entry(); assert!(active_entry.is_some()); let res = current_project_directory(workspace, cx); assert_eq!(res, Some((Path::new("/root2/")).to_path_buf())); let res = first_project_directory(workspace, cx); assert_eq!(res, Some((Path::new("/root1/")).to_path_buf())); }); } /// Creates a worktree with 1 file: /root.txt pub async fn init_test(cx: &mut TestAppContext) -> (Model, View) { let params = cx.update(AppState::test); cx.update(|cx| { theme::init(theme::LoadThemes::JustBase, cx); Project::init_settings(cx); language::init(cx); }); let project = Project::test(params.fs.clone(), [], cx).await; let workspace = cx .add_window(|cx| Workspace::test_new(project.clone(), cx)) .root_view(cx) .unwrap(); (project, workspace) } /// Creates a worktree with 1 folder: /root{suffix}/ async fn create_folder_wt( project: Model, path: impl AsRef, cx: &mut TestAppContext, ) -> (Model, Entry) { create_wt(project, true, path, cx).await } /// Creates a worktree with 1 file: /root{suffix}.txt async fn create_file_wt( project: Model, path: impl AsRef, cx: &mut TestAppContext, ) -> (Model, Entry) { create_wt(project, false, path, cx).await } async fn create_wt( project: Model, is_dir: bool, path: impl AsRef, cx: &mut TestAppContext, ) -> (Model, Entry) { let (wt, _) = project .update(cx, |project, cx| { project.find_or_create_local_worktree(path, true, cx) }) .await .unwrap(); let entry = cx .update(|cx| { wt.update(cx, |wt, cx| { wt.as_local() .unwrap() .create_entry(Path::new(""), is_dir, cx) }) }) .await .unwrap() .unwrap(); (wt, entry) } pub fn insert_active_entry_for( wt: Model, entry: Entry, project: Model, cx: &mut TestAppContext, ) { cx.update(|cx| { let p = ProjectPath { worktree_id: wt.read(cx).id(), path: entry.path, }; project.update(cx, |project, cx| project.set_active_path(Some(p), cx)); }); } #[test] fn escapes_only_special_characters() { assert_eq!(regex_to_literal(r"test(\w)"), r"test\(\\w\)".to_string()); } #[test] fn empty_string_stays_empty() { assert_eq!(regex_to_literal(""), "".to_string()); } }