1039 lines · 36.5 KB · Rust
RawDownload
use std::{    env,    path::{Path, PathBuf},    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 calloop::{LoopSignal, channel::Channel};use futures::channel::oneshot;use util::ResultExt as _;use util::command::{new_smol_command, new_std_command};#[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, PlatformKeyboardLayout, PlatformKeyboardMapper,    PlatformTextSystem, PlatformWindow, Point, Result, RunnableVariant, 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) -> Box<dyn PlatformKeyboardLayout>;    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>>;    #[cfg(feature = "screen-capture")]    fn is_screen_capture_supported(&self) -> bool;    #[cfg(feature = "screen-capture")]    fn screen_capture_sources(        &self,    ) -> oneshot::Receiver<Result<Vec<Rc<dyn crate::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);     #[cfg(any(feature = "wayland", feature = "x11"))]    fn window_identifier(        &self,    ) -> impl Future<Output = Option<ashpd::WindowIdentifier>> + Send + 'static {        std::future::ready::<Option<ashpd::WindowIdentifier>>(None)    }} #[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<RunnableVariant>) {        let (main_sender, main_receiver) = calloop::channel::channel::<RunnableVariant>();         #[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));         let background_executor = BackgroundExecutor::new(dispatcher.clone());         let common = LinuxCommon {            background_executor,            foreground_executor: ForegroundExecutor::new(dispatcher),            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) -> Box<dyn PlatformKeyboardLayout> {        self.keyboard_layout()    }     fn keyboard_mapper(&self) -> Rc<dyn PlatformKeyboardMapper> {        Rc::new(crate::DummyKeyboardMapper)    }     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()        );         #[allow(            clippy::disallowed_methods,            reason = "We are restarting ourselves, using std command thus is fine"        )]        let restart_process = new_std_command("/usr/bin/env")            .arg("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()    }     #[cfg(feature = "screen-capture")]    fn is_screen_capture_supported(&self) -> bool {        self.is_screen_capture_supported()    }     #[cfg(feature = "screen-capture")]    fn screen_capture_sources(        &self,    ) -> oneshot::Receiver<Result<Vec<Rc<dyn crate::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"))]        let identifier = self.window_identifier();         #[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()                    .identifier(identifier.await)                    .modal(true)                    .title(title)                    .accept_label(options.prompt.as_ref().map(crate::SharedString::as_str))                    .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,        suggested_name: Option<&str>,    ) -> 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, suggested_name);         #[cfg(any(feature = "wayland", feature = "x11"))]        let identifier = self.window_identifier();         #[cfg(any(feature = "wayland", feature = "x11"))]        self.foreground_executor()            .spawn({                let directory = directory.to_owned();                let suggested_name = suggested_name.map(|s| s.to_owned());                 async move {                    let mut request_builder =                        ashpd::desktop::file_chooser::SaveFileRequest::default()                            .identifier(identifier.await)                            .modal(true)                            .title("Save File")                            .current_folder(directory)                            .expect("pathbuf should not be nul terminated");                     if let Some(suggested_name) = suggested_name {                        request_builder = request_builder.current_name(suggested_name.as_str());                    }                     let request = match request_builder.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 _ = new_smol_command("xdg-open")                    .arg(path)                    .spawn()                    .context("invoking xdg-open")                    .log_err()?                    .status()                    .await                    .log_err()?;                Some(())            })            .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 app_path = env::current_exe()?;        Ok(app_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) {        // todo(linux)    }     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)    }

Lines 1–500 of 1039.