209 lines · 7.1 KB · Rust
RawDownload
use std::path::PathBuf; use gpui::{ClickEvent, DismissEvent, EventEmitter, FocusHandle, Focusable, Render, WeakEntity};use project::project_settings::ProjectSettings;use remote::SshConnectionOptions;use settings::Settings;use ui::{    div, h_flex, rems, Button, ButtonCommon, ButtonStyle, Clickable, Context, ElevationIndex,    FluentBuilder, Headline, HeadlineSize, IconName, IconPosition, InteractiveElement, IntoElement,    Label, Modal, ModalFooter, ModalHeader, ParentElement, Section, Styled, StyledExt, Window,};use workspace::{notifications::DetachAndPromptErr, ModalView, OpenOptions, Workspace}; use crate::open_ssh_project; enum Host {    RemoteProject,    SshRemoteProject(SshConnectionOptions),} pub struct DisconnectedOverlay {    workspace: WeakEntity<Workspace>,    host: Host,    focus_handle: FocusHandle,    finished: bool,} impl EventEmitter<DismissEvent> for DisconnectedOverlay {}impl Focusable for DisconnectedOverlay {    fn focus_handle(&self, _cx: &gpui::App) -> gpui::FocusHandle {        self.focus_handle.clone()    }}impl ModalView for DisconnectedOverlay {    fn on_before_dismiss(        &mut self,        _window: &mut Window,        _: &mut Context<Self>,    ) -> workspace::DismissDecision {        return workspace::DismissDecision::Dismiss(self.finished);    }    fn fade_out_background(&self) -> bool {        true    }} impl DisconnectedOverlay {    pub fn register(        workspace: &mut Workspace,        window: Option<&mut Window>,        cx: &mut Context<Workspace>,    ) {        let Some(window) = window else {            return;        };        cx.subscribe_in(            workspace.project(),            window,            |workspace, project, event, window, cx| {                if !matches!(                    event,                    project::Event::DisconnectedFromHost                        | project::Event::DisconnectedFromSshRemote                ) {                    return;                }                let handle = cx.entity().downgrade();                 let ssh_connection_options = project.read(cx).ssh_connection_options(cx);                let host = if let Some(ssh_connection_options) = ssh_connection_options {                    Host::SshRemoteProject(ssh_connection_options)                } else {                    Host::RemoteProject                };                 workspace.toggle_modal(window, cx, |_, cx| DisconnectedOverlay {                    finished: false,                    workspace: handle,                    host,                    focus_handle: cx.focus_handle(),                });            },        )        .detach();    }     fn handle_reconnect(&mut self, _: &ClickEvent, window: &mut Window, cx: &mut Context<Self>) {        self.finished = true;        cx.emit(DismissEvent);         match &self.host {            Host::SshRemoteProject(ssh_connection_options) => {                self.reconnect_to_ssh_remote(ssh_connection_options.clone(), window, cx);            }            _ => {}        }    }     fn reconnect_to_ssh_remote(        &self,        connection_options: SshConnectionOptions,        window: &mut Window,        cx: &mut Context<Self>,    ) {        let Some(workspace) = self.workspace.upgrade() else {            return;        };         let Some(ssh_project) = workspace.read(cx).serialized_ssh_project() else {            return;        };         let Some(window_handle) = window.window_handle().downcast::<Workspace>() else {            return;        };         let app_state = workspace.read(cx).app_state().clone();         let paths = ssh_project.paths.iter().map(PathBuf::from).collect();         cx.spawn_in(window, move |_, mut cx| async move {            open_ssh_project(                connection_options,                paths,                app_state,                OpenOptions {                    replace_window: Some(window_handle),                    ..Default::default()                },                &mut cx,            )            .await?;            Ok(())        })        .detach_and_prompt_err("Failed to reconnect", window, cx, |_, _, _| None);    }     fn cancel(&mut self, _: &menu::Cancel, _: &mut Window, cx: &mut Context<Self>) {        self.finished = true;        cx.emit(DismissEvent)    }} impl Render for DisconnectedOverlay {    fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {        let can_reconnect = matches!(self.host, Host::SshRemoteProject(_));         let message = match &self.host {            Host::RemoteProject => {                "Your connection to the remote project has been lost.".to_string()            }            Host::SshRemoteProject(options) => {                let autosave = if ProjectSettings::get_global(cx)                    .session                    .restore_unsaved_buffers                {                    "\nUnsaved changes are stored locally."                } else {                    ""                };                format!(                    "Your connection to {} has been lost.{}",                    options.host, autosave                )            }        };         div()            .track_focus(&self.focus_handle(cx))            .elevation_3(cx)            .on_action(cx.listener(Self::cancel))            .occlude()            .w(rems(24.))            .max_h(rems(40.))            .child(                Modal::new("disconnected", None)                    .header(                        ModalHeader::new()                            .show_dismiss_button(true)                            .child(Headline::new("Disconnected").size(HeadlineSize::Small)),                    )                    .section(Section::new().child(Label::new(message)))                    .footer(                        ModalFooter::new().end_slot(                            h_flex()                                .gap_2()                                .child(                                    Button::new("close-window", "Close Window")                                        .style(ButtonStyle::Filled)                                        .layer(ElevationIndex::ModalSurface)                                        .on_click(cx.listener(move |_, _, window, _| {                                            window.remove_window();                                        })),                                )                                .when(can_reconnect, |el| {                                    el.child(                                        Button::new("reconnect", "Reconnect")                                            .style(ButtonStyle::Filled)                                            .layer(ElevationIndex::ModalSurface)                                            .icon(IconName::ArrowCircle)                                            .icon_position(IconPosition::Start)                                            .on_click(cx.listener(Self::handle_reconnect)),                                    )                                }),                        ),                    ),            )    }}