use std::{path::PathBuf, sync::Arc}; use crate::TerminalView; use db::kvp::KEY_VALUE_STORE; use gpui::{ actions, div, serde_json, AppContext, AsyncWindowContext, Div, Entity, EventEmitter, FocusHandle, FocusableView, ParentComponent, Render, Subscription, Task, View, ViewContext, VisualContext, WeakView, WindowContext, }; use project::Fs; use serde::{Deserialize, Serialize}; use settings::{Settings, SettingsStore}; use terminal::terminal_settings::{TerminalDockPosition, TerminalSettings}; use util::{ResultExt, TryFutureExt}; use workspace::{ dock::{DockPosition, Panel, PanelEvent}, item::Item, pane, ui::Icon, Pane, Workspace, }; use anyhow::Result; const TERMINAL_PANEL_KEY: &'static str = "TerminalPanel"; actions!(ToggleFocus); pub fn init(cx: &mut AppContext) { cx.observe_new_views( |workspace: &mut Workspace, _: &mut ViewContext| { workspace.register_action(TerminalPanel::new_terminal); workspace.register_action(TerminalPanel::open_terminal); workspace.register_action(|workspace, _: &ToggleFocus, cx| { workspace.toggle_panel_focus::(cx); }); }, ) .detach(); } pub struct TerminalPanel { pane: View, fs: Arc, workspace: WeakView, width: Option, height: Option, pending_serialization: Task>, _subscriptions: Vec, } impl TerminalPanel { fn new(workspace: &Workspace, cx: &mut ViewContext) -> Self { let _weak_self = cx.view().downgrade(); let pane = cx.build_view(|cx| { let _window = cx.window_handle(); let mut pane = Pane::new( workspace.weak_handle(), workspace.project().clone(), Default::default(), cx, ); pane.set_can_split(false, cx); pane.set_can_navigate(false, cx); // todo!() // pane.on_can_drop(move |drag_and_drop, cx| { // drag_and_drop // .currently_dragged::(window) // .map_or(false, |(_, item)| { // item.handle.act_as::(cx).is_some() // }) // }); // pane.set_render_tab_bar_buttons(cx, move |pane, cx| { // let this = weak_self.clone(); // Flex::row() // .with_child(Pane::render_tab_bar_button( // 0, // "icons/plus.svg", // false, // Some(("New Terminal", Some(Box::new(workspace::NewTerminal)))), // cx, // move |_, cx| { // let this = this.clone(); // cx.window_context().defer(move |cx| { // if let Some(this) = this.upgrade() { // this.update(cx, |this, cx| { // this.add_terminal(None, cx); // }); // } // }) // }, // |_, _| {}, // None, // )) // .with_child(Pane::render_tab_bar_button( // 1, // if pane.is_zoomed() { // "icons/minimize.svg" // } else { // "icons/maximize.svg" // }, // pane.is_zoomed(), // Some(("Toggle Zoom".into(), Some(Box::new(workspace::ToggleZoom)))), // cx, // move |pane, cx| pane.toggle_zoom(&Default::default(), cx), // |_, _| {}, // None, // )) // .into_any() // }); // let buffer_search_bar = cx.build_view(search::BufferSearchBar::new); // pane.toolbar() // .update(cx, |toolbar, cx| toolbar.add_item(buffer_search_bar, cx)); pane }); let subscriptions = vec![ cx.observe(&pane, |_, _, cx| cx.notify()), cx.subscribe(&pane, Self::handle_pane_event), ]; let this = Self { pane, fs: workspace.app_state().fs.clone(), workspace: workspace.weak_handle(), pending_serialization: Task::ready(None), width: None, height: None, _subscriptions: subscriptions, }; let mut old_dock_position = this.position(cx); cx.observe_global::(move |this, cx| { let new_dock_position = this.position(cx); if new_dock_position != old_dock_position { old_dock_position = new_dock_position; cx.emit(PanelEvent::ChangePosition); } }) .detach(); this } pub async fn load( workspace: WeakView, mut cx: AsyncWindowContext, ) -> Result> { let serialized_panel = cx .background_executor() .spawn(async move { KEY_VALUE_STORE.read_kvp(TERMINAL_PANEL_KEY) }) .await .log_err() .flatten() .map(|panel| serde_json::from_str::(&panel)) .transpose() .log_err() .flatten(); let (panel, pane, items) = workspace.update(&mut cx, |workspace, cx| { let panel = cx.build_view(|cx| TerminalPanel::new(workspace, cx)); let items = if let Some(serialized_panel) = serialized_panel.as_ref() { panel.update(cx, |panel, cx| { cx.notify(); panel.height = serialized_panel.height; panel.width = serialized_panel.width; panel.pane.update(cx, |_, cx| { serialized_panel .items .iter() .map(|item_id| { TerminalView::deserialize( workspace.project().clone(), workspace.weak_handle(), workspace.database_id(), *item_id, cx, ) }) .collect::>() }) }) } else { Default::default() }; let pane = panel.read(cx).pane.clone(); (panel, pane, items) })?; let pane = pane.downgrade(); let items = futures::future::join_all(items).await; pane.update(&mut cx, |pane, cx| { let active_item_id = serialized_panel .as_ref() .and_then(|panel| panel.active_item_id); let mut active_ix = None; for item in items { if let Some(item) = item.log_err() { let item_id = item.entity_id().as_u64(); pane.add_item(Box::new(item), false, false, None, cx); if Some(item_id) == active_item_id { active_ix = Some(pane.items_len() - 1); } } } if let Some(active_ix) = active_ix { pane.activate_item(active_ix, false, false, cx) } })?; Ok(panel) } fn handle_pane_event( &mut self, _pane: View, event: &pane::Event, cx: &mut ViewContext, ) { match event { pane::Event::ActivateItem { .. } => self.serialize(cx), pane::Event::RemoveItem { .. } => self.serialize(cx), pane::Event::Remove => cx.emit(PanelEvent::Close), pane::Event::ZoomIn => cx.emit(PanelEvent::ZoomIn), pane::Event::ZoomOut => cx.emit(PanelEvent::ZoomOut), pane::Event::Focus => cx.emit(PanelEvent::Focus), pane::Event::AddItem { item } => { if let Some(workspace) = self.workspace.upgrade() { let pane = self.pane.clone(); workspace.update(cx, |workspace, cx| item.added_to_pane(workspace, pane, cx)) } } _ => {} } } pub fn open_terminal( workspace: &mut Workspace, action: &workspace::OpenTerminal, cx: &mut ViewContext, ) { let Some(this) = workspace.focus_panel::(cx) else { return; }; this.update(cx, |this, cx| { this.add_terminal(Some(action.working_directory.clone()), cx) }) } ///Create a new Terminal in the current working directory or the user's home directory fn new_terminal( workspace: &mut Workspace, _: &workspace::NewTerminal, cx: &mut ViewContext, ) { let Some(this) = workspace.focus_panel::(cx) else { return; }; this.update(cx, |this, cx| this.add_terminal(None, cx)) } fn add_terminal(&mut self, working_directory: Option, cx: &mut ViewContext) { let workspace = self.workspace.clone(); cx.spawn(|this, mut cx| async move { let pane = this.update(&mut cx, |this, _| this.pane.clone())?; workspace.update(&mut cx, |workspace, cx| { let working_directory = if let Some(working_directory) = working_directory { Some(working_directory) } else { let working_directory_strategy = TerminalSettings::get_global(cx).working_directory.clone(); crate::get_working_directory(workspace, cx, working_directory_strategy) }; let window = cx.window_handle(); if let Some(terminal) = workspace.project().update(cx, |project, cx| { project .create_terminal(working_directory, window, cx) .log_err() }) { let terminal = Box::new(cx.build_view(|cx| { TerminalView::new( terminal, workspace.weak_handle(), workspace.database_id(), cx, ) })); pane.update(cx, |pane, cx| { let focus = pane.has_focus(cx); pane.add_item(terminal, true, focus, None, cx); }); } })?; this.update(&mut cx, |this, cx| this.serialize(cx))?; anyhow::Ok(()) }) .detach_and_log_err(cx); } fn serialize(&mut self, cx: &mut ViewContext) { let items = self .pane .read(cx) .items() .map(|item| item.id().as_u64()) .collect::>(); let active_item_id = self .pane .read(cx) .active_item() .map(|item| item.id().as_u64()); let height = self.height; let width = self.width; self.pending_serialization = cx.background_executor().spawn( async move { KEY_VALUE_STORE .write_kvp( TERMINAL_PANEL_KEY.into(), serde_json::to_string(&SerializedTerminalPanel { items, active_item_id, height, width, })?, ) .await?; anyhow::Ok(()) } .log_err(), ); } } impl EventEmitter for TerminalPanel {} impl Render for TerminalPanel { type Element = Div; fn render(&mut self, _cx: &mut ViewContext) -> Self::Element { div().child(self.pane.clone()) } } impl FocusableView for TerminalPanel { fn focus_handle(&self, cx: &AppContext) -> FocusHandle { self.pane.focus_handle(cx) } } impl Panel for TerminalPanel { fn position(&self, cx: &WindowContext) -> DockPosition { match TerminalSettings::get_global(cx).dock { TerminalDockPosition::Left => DockPosition::Left, TerminalDockPosition::Bottom => DockPosition::Bottom, TerminalDockPosition::Right => DockPosition::Right, } } fn position_is_valid(&self, _: DockPosition) -> bool { true } fn set_position(&mut self, position: DockPosition, cx: &mut ViewContext) { settings::update_settings_file::(self.fs.clone(), cx, move |settings| { let dock = match position { DockPosition::Left => TerminalDockPosition::Left, DockPosition::Bottom => TerminalDockPosition::Bottom, DockPosition::Right => TerminalDockPosition::Right, }; settings.dock = Some(dock); }); } fn size(&self, cx: &WindowContext) -> f32 { let settings = TerminalSettings::get_global(cx); match self.position(cx) { DockPosition::Left | DockPosition::Right => { self.width.unwrap_or_else(|| settings.default_width) } DockPosition::Bottom => self.height.unwrap_or_else(|| settings.default_height), } } fn set_size(&mut self, size: Option, cx: &mut ViewContext) { match self.position(cx) { DockPosition::Left | DockPosition::Right => self.width = size, DockPosition::Bottom => self.height = size, } self.serialize(cx); cx.notify(); } fn is_zoomed(&self, cx: &WindowContext) -> bool { self.pane.read(cx).is_zoomed() } fn set_zoomed(&mut self, zoomed: bool, cx: &mut ViewContext) { self.pane.update(cx, |pane, cx| pane.set_zoomed(zoomed, cx)); } fn set_active(&mut self, active: bool, cx: &mut ViewContext) { if active && self.pane.read(cx).items_len() == 0 { self.add_terminal(None, cx) } } fn icon_label(&self, cx: &WindowContext) -> Option { let count = self.pane.read(cx).items_len(); if count == 0 { None } else { Some(count.to_string()) } } fn has_focus(&self, cx: &WindowContext) -> bool { self.pane.read(cx).has_focus(cx) } fn persistent_name() -> &'static str { "TerminalPanel" } // todo!() // fn icon_tooltip(&self) -> (String, Option>) { // ("Terminal Panel".into(), Some(Box::new(ToggleFocus))) // } fn icon(&self, _cx: &WindowContext) -> Option { Some(Icon::Terminal) } fn toggle_action(&self) -> Box { Box::new(ToggleFocus) } } #[derive(Serialize, Deserialize)] struct SerializedTerminalPanel { items: Vec, active_item_id: Option, width: Option, height: Option, }