884 lines · 30.7 KB · Rust
RawDownload
use std::{    env,    path::{Path, PathBuf},    process::Command,    rc::Rc,    sync::Arc,};#[cfg(any(feature = "wayland", feature = "x11"))]use std::{    ffi::OsString,    fs::File,    io::Read as _,    os::fd::{AsFd, AsRawFd, FromRawFd},    time::Duration,}; use anyhow::{Context as _, anyhow};use async_task::Runnable;use calloop::{LoopSignal, channel::Channel};use futures::channel::oneshot;use util::ResultExt as _;#[cfg(any(feature = "wayland", feature = "x11"))]use xkbcommon::xkb::{self, Keycode, Keysym, State}; use crate::{    Action, AnyWindowHandle, BackgroundExecutor, ClipboardItem, CursorStyle, DisplayId,    ForegroundExecutor, Keymap, LinuxDispatcher, Menu, MenuItem, OwnedMenu, PathPromptOptions,    Pixels, Platform, PlatformDisplay, PlatformTextSystem, PlatformWindow, Point, Result,    ScreenCaptureSource, Task, WindowAppearance, WindowParams, px,}; #[cfg(any(feature = "wayland", feature = "x11"))]pub(crate) const SCROLL_LINES: f32 = 3.0; // Values match the defaults on GTK.// Taken from https://github.com/GNOME/gtk/blob/main/gtk/gtksettings.c#L320#[cfg(any(feature = "wayland", feature = "x11"))]pub(crate) const DOUBLE_CLICK_INTERVAL: Duration = Duration::from_millis(400);pub(crate) const DOUBLE_CLICK_DISTANCE: Pixels = px(5.0);pub(crate) const KEYRING_LABEL: &str = "zed-github-account"; #[cfg(any(feature = "wayland", feature = "x11"))]const FILE_PICKER_PORTAL_MISSING: &str =    "Couldn't open file picker due to missing xdg-desktop-portal implementation."; pub trait LinuxClient {    fn compositor_name(&self) -> &'static str;    fn with_common<R>(&self, f: impl FnOnce(&mut LinuxCommon) -> R) -> R;    fn keyboard_layout(&self) -> String;    fn displays(&self) -> Vec<Rc<dyn PlatformDisplay>>;    #[allow(unused)]    fn display(&self, id: DisplayId) -> Option<Rc<dyn PlatformDisplay>>;    fn primary_display(&self) -> Option<Rc<dyn PlatformDisplay>>;    fn is_screen_capture_supported(&self) -> bool;    fn screen_capture_sources(        &self,    ) -> oneshot::Receiver<Result<Vec<Box<dyn ScreenCaptureSource>>>>;     fn open_window(        &self,        handle: AnyWindowHandle,        options: WindowParams,    ) -> anyhow::Result<Box<dyn PlatformWindow>>;    fn set_cursor_style(&self, style: CursorStyle);    fn open_uri(&self, uri: &str);    fn reveal_path(&self, path: PathBuf);    fn write_to_primary(&self, item: ClipboardItem);    fn write_to_clipboard(&self, item: ClipboardItem);    fn read_from_primary(&self) -> Option<ClipboardItem>;    fn read_from_clipboard(&self) -> Option<ClipboardItem>;    fn active_window(&self) -> Option<AnyWindowHandle>;    fn window_stack(&self) -> Option<Vec<AnyWindowHandle>>;    fn run(&self);} #[derive(Default)]pub(crate) struct PlatformHandlers {    pub(crate) open_urls: Option<Box<dyn FnMut(Vec<String>)>>,    pub(crate) quit: Option<Box<dyn FnMut()>>,    pub(crate) reopen: Option<Box<dyn FnMut()>>,    pub(crate) app_menu_action: Option<Box<dyn FnMut(&dyn Action)>>,    pub(crate) will_open_app_menu: Option<Box<dyn FnMut()>>,    pub(crate) validate_app_menu_command: Option<Box<dyn FnMut(&dyn Action) -> bool>>,    pub(crate) keyboard_layout_change: Option<Box<dyn FnMut()>>,} pub(crate) struct LinuxCommon {    pub(crate) background_executor: BackgroundExecutor,    pub(crate) foreground_executor: ForegroundExecutor,    pub(crate) text_system: Arc<dyn PlatformTextSystem>,    pub(crate) appearance: WindowAppearance,    pub(crate) auto_hide_scrollbars: bool,    pub(crate) callbacks: PlatformHandlers,    pub(crate) signal: LoopSignal,    pub(crate) menus: Vec<OwnedMenu>,} impl LinuxCommon {    pub fn new(signal: LoopSignal) -> (Self, Channel<Runnable>) {        let (main_sender, main_receiver) = calloop::channel::channel::<Runnable>();         #[cfg(any(feature = "wayland", feature = "x11"))]        let text_system = Arc::new(crate::CosmicTextSystem::new());        #[cfg(not(any(feature = "wayland", feature = "x11")))]        let text_system = Arc::new(crate::NoopTextSystem::new());         let callbacks = PlatformHandlers::default();         let dispatcher = Arc::new(LinuxDispatcher::new(main_sender.clone()));         let background_executor = BackgroundExecutor::new(dispatcher.clone());         let common = LinuxCommon {            background_executor,            foreground_executor: ForegroundExecutor::new(dispatcher.clone()),            text_system,            appearance: WindowAppearance::Light,            auto_hide_scrollbars: false,            callbacks,            signal,            menus: Vec::new(),        };         (common, main_receiver)    }} impl<P: LinuxClient + 'static> Platform for P {    fn background_executor(&self) -> BackgroundExecutor {        self.with_common(|common| common.background_executor.clone())    }     fn foreground_executor(&self) -> ForegroundExecutor {        self.with_common(|common| common.foreground_executor.clone())    }     fn text_system(&self) -> Arc<dyn PlatformTextSystem> {        self.with_common(|common| common.text_system.clone())    }     fn keyboard_layout(&self) -> String {        self.keyboard_layout()    }     fn on_keyboard_layout_change(&self, callback: Box<dyn FnMut()>) {        self.with_common(|common| common.callbacks.keyboard_layout_change = Some(callback));    }     fn run(&self, on_finish_launching: Box<dyn FnOnce()>) {        on_finish_launching();         LinuxClient::run(self);         let quit = self.with_common(|common| common.callbacks.quit.take());        if let Some(mut fun) = quit {            fun();        }    }     fn quit(&self) {        self.with_common(|common| common.signal.stop());    }     fn compositor_name(&self) -> &'static str {        self.compositor_name()    }     fn restart(&self, binary_path: Option<PathBuf>) {        use std::os::unix::process::CommandExt as _;         // get the process id of the current process        let app_pid = std::process::id().to_string();        // get the path to the executable        let app_path = if let Some(path) = binary_path {            path        } else {            match self.app_path() {                Ok(path) => path,                Err(err) => {                    log::error!("Failed to get app path: {:?}", err);                    return;                }            }        };         log::info!("Restarting process, using app path: {:?}", app_path);         // Script to wait for the current process to exit and then restart the app.        let script = format!(            r#"            while kill -0 {pid} 2>/dev/null; do                sleep 0.1            done             {app_path}            "#,            pid = app_pid,            app_path = app_path.display()        );         // execute the script using /bin/bash        let restart_process = Command::new("/bin/bash")            .arg("-c")            .arg(script)            .process_group(0)            .spawn();         match restart_process {            Ok(_) => self.quit(),            Err(e) => log::error!("failed to spawn restart script: {:?}", e),        }    }     fn activate(&self, _ignoring_other_apps: bool) {        log::info!("activate is not implemented on Linux, ignoring the call")    }     fn hide(&self) {        log::info!("hide is not implemented on Linux, ignoring the call")    }     fn hide_other_apps(&self) {        log::info!("hide_other_apps is not implemented on Linux, ignoring the call")    }     fn unhide_other_apps(&self) {        log::info!("unhide_other_apps is not implemented on Linux, ignoring the call")    }     fn primary_display(&self) -> Option<Rc<dyn PlatformDisplay>> {        self.primary_display()    }     fn displays(&self) -> Vec<Rc<dyn PlatformDisplay>> {        self.displays()    }     fn is_screen_capture_supported(&self) -> bool {        self.is_screen_capture_supported()    }     fn screen_capture_sources(        &self,    ) -> oneshot::Receiver<Result<Vec<Box<dyn ScreenCaptureSource>>>> {        self.screen_capture_sources()    }     fn active_window(&self) -> Option<AnyWindowHandle> {        self.active_window()    }     fn window_stack(&self) -> Option<Vec<AnyWindowHandle>> {        self.window_stack()    }     fn open_window(        &self,        handle: AnyWindowHandle,        options: WindowParams,    ) -> anyhow::Result<Box<dyn PlatformWindow>> {        self.open_window(handle, options)    }     fn open_url(&self, url: &str) {        self.open_uri(url);    }     fn on_open_urls(&self, callback: Box<dyn FnMut(Vec<String>)>) {        self.with_common(|common| common.callbacks.open_urls = Some(callback));    }     fn prompt_for_paths(        &self,        options: PathPromptOptions,    ) -> oneshot::Receiver<Result<Option<Vec<PathBuf>>>> {        let (done_tx, done_rx) = oneshot::channel();         #[cfg(not(any(feature = "wayland", feature = "x11")))]        let _ = (done_tx.send(Ok(None)), options);         #[cfg(any(feature = "wayland", feature = "x11"))]        self.foreground_executor()            .spawn(async move {                let title = if options.directories {                    "Open Folder"                } else {                    "Open File"                };                 let request = match ashpd::desktop::file_chooser::OpenFileRequest::default()                    .modal(true)                    .title(title)                    .multiple(options.multiple)                    .directory(options.directories)                    .send()                    .await                {                    Ok(request) => request,                    Err(err) => {                        let result = match err {                            ashpd::Error::PortalNotFound(_) => anyhow!(FILE_PICKER_PORTAL_MISSING),                            err => err.into(),                        };                        let _ = done_tx.send(Err(result));                        return;                    }                };                 let result = match request.response() {                    Ok(response) => Ok(Some(                        response                            .uris()                            .iter()                            .filter_map(|uri| uri.to_file_path().ok())                            .collect::<Vec<_>>(),                    )),                    Err(ashpd::Error::Response(_)) => Ok(None),                    Err(e) => Err(e.into()),                };                let _ = done_tx.send(result);            })            .detach();        done_rx    }     fn prompt_for_new_path(&self, directory: &Path) -> oneshot::Receiver<Result<Option<PathBuf>>> {        let (done_tx, done_rx) = oneshot::channel();         #[cfg(not(any(feature = "wayland", feature = "x11")))]        let _ = (done_tx.send(Ok(None)), directory);         #[cfg(any(feature = "wayland", feature = "x11"))]        self.foreground_executor()            .spawn({                let directory = directory.to_owned();                 async move {                    let request = match ashpd::desktop::file_chooser::SaveFileRequest::default()                        .modal(true)                        .title("Save File")                        .current_folder(directory)                        .expect("pathbuf should not be nul terminated")                        .send()                        .await                    {                        Ok(request) => request,                        Err(err) => {                            let result = match err {                                ashpd::Error::PortalNotFound(_) => {                                    anyhow!(FILE_PICKER_PORTAL_MISSING)                                }                                err => err.into(),                            };                            let _ = done_tx.send(Err(result));                            return;                        }                    };                     let result = match request.response() {                        Ok(response) => Ok(response                            .uris()                            .first()                            .and_then(|uri| uri.to_file_path().ok())),                        Err(ashpd::Error::Response(_)) => Ok(None),                        Err(e) => Err(e.into()),                    };                    let _ = done_tx.send(result);                }            })            .detach();         done_rx    }     fn can_select_mixed_files_and_dirs(&self) -> bool {        // org.freedesktop.portal.FileChooser only supports "pick files" and "pick directories".        false    }     fn reveal_path(&self, path: &Path) {        self.reveal_path(path.to_owned());    }     fn open_with_system(&self, path: &Path) {        let path = path.to_owned();        self.background_executor()            .spawn(async move {                let _ = std::process::Command::new("xdg-open")                    .arg(path)                    .spawn()                    .context("invoking xdg-open")                    .log_err();            })            .detach();    }     fn on_quit(&self, callback: Box<dyn FnMut()>) {        self.with_common(|common| {            common.callbacks.quit = Some(callback);        });    }     fn on_reopen(&self, callback: Box<dyn FnMut()>) {        self.with_common(|common| {            common.callbacks.reopen = Some(callback);        });    }     fn on_app_menu_action(&self, callback: Box<dyn FnMut(&dyn Action)>) {        self.with_common(|common| {            common.callbacks.app_menu_action = Some(callback);        });    }     fn on_will_open_app_menu(&self, callback: Box<dyn FnMut()>) {        self.with_common(|common| {            common.callbacks.will_open_app_menu = Some(callback);        });    }     fn on_validate_app_menu_command(&self, callback: Box<dyn FnMut(&dyn Action) -> bool>) {        self.with_common(|common| {            common.callbacks.validate_app_menu_command = Some(callback);        });    }     fn app_path(&self) -> Result<PathBuf> {        // get the path of the executable of the current process        let exe_path = env::current_exe()?;        Ok(exe_path)    }     fn set_menus(&self, menus: Vec<Menu>, _keymap: &Keymap) {        self.with_common(|common| {            common.menus = menus.into_iter().map(|menu| menu.owned()).collect();        })    }     fn get_menus(&self) -> Option<Vec<OwnedMenu>> {        self.with_common(|common| Some(common.menus.clone()))    }     fn set_dock_menu(&self, _menu: Vec<MenuItem>, _keymap: &Keymap) {}     fn path_for_auxiliary_executable(&self, _name: &str) -> Result<PathBuf> {        Err(anyhow::Error::msg(            "Platform<LinuxPlatform>::path_for_auxiliary_executable is not implemented yet",        ))    }     fn set_cursor_style(&self, style: CursorStyle) {        self.set_cursor_style(style)    }     fn should_auto_hide_scrollbars(&self) -> bool {        self.with_common(|common| common.auto_hide_scrollbars)    }     fn write_credentials(&self, url: &str, username: &str, password: &[u8]) -> Task<Result<()>> {        let url = url.to_string();        let username = username.to_string();        let password = password.to_vec();        self.background_executor().spawn(async move {            let keyring = oo7::Keyring::new().await?;            keyring.unlock().await?;            keyring                .create_item(                    KEYRING_LABEL,                    &vec![("url", &url), ("username", &username)],                    password,                    true,                )                .await?;            Ok(())        })    }     fn read_credentials(&self, url: &str) -> Task<Result<Option<(String, Vec<u8>)>>> {        let url = url.to_string();        self.background_executor().spawn(async move {            let keyring = oo7::Keyring::new().await?;            keyring.unlock().await?;             let items = keyring.search_items(&vec![("url", &url)]).await?;             for item in items.into_iter() {                if item.label().await.is_ok_and(|label| label == KEYRING_LABEL) {                    let attributes = item.attributes().await?;                    let username = attributes                        .get("username")                        .ok_or_else(|| anyhow!("Cannot find username in stored credentials"))?;                    let secret = item.secret().await?;                     // we lose the zeroizing capabilities at this boundary,                    // a current limitation GPUI's credentials api                    return Ok(Some((username.to_string(), secret.to_vec())));                } else {                    continue;                }            }

Lines 1–500 of 884.