ZIm/crates/terminal_view/src/terminal_panel.rs

use crate::TerminalView;
use gpui::{
    elements::*, AppContext, Entity, ModelHandle, Subscription, View, ViewContext, ViewHandle,
    WeakViewHandle,
};
use project::Project;
use settings::{Settings, WorkingDirectory};
use util::ResultExt;
use workspace::{dock::{Panel, DockPosition}, pane, DraggedItem, Pane, Workspace};

pub fn init(cx: &mut AppContext) {
    cx.add_action(TerminalPanel::add_terminal);
}

pub enum Event {
    Close,
}

pub struct TerminalPanel {
    project: ModelHandle<Project>,
    pane: ViewHandle<Pane>,
    workspace: WeakViewHandle<Workspace>,
    _subscriptions: Vec<Subscription>,
}

impl TerminalPanel {
    pub fn new(workspace: &Workspace, cx: &mut ViewContext<Self>) -> Self {
        let this = cx.weak_handle();
        let pane = cx.add_view(|cx| {
            let window_id = cx.window_id();
            let mut pane = Pane::new(
                workspace.weak_handle(),
                workspace.app_state().background_actions,
                cx,
            );
            pane.set_can_split(false, cx);
            pane.on_can_drop(move |drag_and_drop, cx| {
                drag_and_drop
                    .currently_dragged::<DraggedItem>(window_id)
                    .map_or(false, |(_, item)| {
                        item.handle.act_as::<TerminalView>(cx).is_some()
                    })
            });
            pane.set_render_tab_bar_buttons(cx, move |_, cx| {
                let this = this.clone();
                Pane::render_tab_bar_button(
                    0,
                    "icons/plus_12.svg",
                    cx,
                    move |_, cx| {
                        let this = this.clone();
                        cx.window_context().defer(move |cx| {
                            if let Some(this) = this.upgrade(cx) {
                                this.update(cx, |this, cx| {
                                    this.add_terminal(&Default::default(), cx);
                                });
                            }
                        })
                    },
                    None,
                )
            });
            pane
        });
        let subscriptions = vec![
            cx.observe(&pane, |_, _, cx| cx.notify()),
            cx.subscribe(&pane, Self::handle_pane_event),
        ];
        Self {
            project: workspace.project().clone(),
            pane,
            workspace: workspace.weak_handle(),
            _subscriptions: subscriptions,
        }
    }

    fn handle_pane_event(
        &mut self,
        _pane: ViewHandle<Pane>,
        event: &pane::Event,
        cx: &mut ViewContext<Self>,
    ) {
        match event {
            pane::Event::Remove => cx.emit(Event::Close),
            _ => {}
        }
    }

    fn add_terminal(&mut self, _: &workspace::NewTerminal, cx: &mut ViewContext<Self>) {
        if let Some(workspace) = self.workspace.upgrade(cx) {
            let working_directory_strategy = cx
                .global::<Settings>()
                .terminal_overrides
                .working_directory
                .clone()
                .unwrap_or(WorkingDirectory::CurrentProjectDirectory);
            let working_directory =
                crate::get_working_directory(workspace.read(cx), cx, working_directory_strategy);
            let window_id = cx.window_id();
            if let Some(terminal) = self.project.update(cx, |project, cx| {
                project
                    .create_terminal(working_directory, window_id, cx)
                    .log_err()
            }) {
                workspace.update(cx, |workspace, cx| {
                    let terminal =
                        Box::new(cx.add_view(|cx| {
                            TerminalView::new(terminal, workspace.database_id(), cx)
                        }));
                    Pane::add_item(workspace, &self.pane, terminal, true, true, None, cx);
                });
            }
        }
    }
}

impl Entity for TerminalPanel {
    type Event = Event;
}

impl View for TerminalPanel {
    fn ui_name() -> &'static str {
        "TerminalPanel"
    }

    fn render(&mut self, cx: &mut ViewContext<Self>) -> gpui::AnyElement<Self> {
        ChildView::new(&self.pane, cx).into_any()
    }

    fn focus_in(&mut self, _: gpui::AnyViewHandle, cx: &mut ViewContext<Self>) {
        if self.pane.read(cx).items_len() == 0 {
            self.add_terminal(&Default::default(), cx)
        }
    }
}

impl Panel for TerminalPanel {
    fn position(&self, cx: &gpui::WindowContext) -> DockPosition {
        cx.global::<Settings>().terminal_overrides.dock.into()
    }

    fn position_is_valid(&self, _: DockPosition) -> bool {
        true
    }

    fn icon_path(&self) -> &'static str {
        "icons/terminal_12.svg"
    }

    fn icon_tooltip(&self) -> String {
        "Terminals".to_string()
    }

    fn icon_label(&self, cx: &AppContext) -> Option<String> {
        let count = self.pane.read(cx).items_len();
        if count == 0 {
            None
        } else {
            Some(count.to_string())
        }
    }

    fn should_change_position_on_event(&self, _: &Self::Event, _: &AppContext) -> bool {
        todo!()
    }

    fn should_activate_on_event(&self, _: &Self::Event, _: &AppContext) -> bool {
        todo!()
    }

    fn should_close_on_event(&self, event: &Event, _: &AppContext) -> bool {
        matches!(event, Event::Close)
    }
}