Something went wrong. Try again.
fork of gpui
Something went wrong. Try again.
5.7 kB · 144 lines
Rust
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145use super::register_zed_scheme;use anyhow::Result;use gpui::{AppContext as _, AsyncApp, Context, PromptLevel, Window, actions};use release_channel::ReleaseChannel;use std::ops::Deref;use std::path::{Path, PathBuf};use util::ResultExt;use workspace::notifications::simple_message_notification::MessageNotification;use workspace::notifications::{DetachAndPromptErr, NotificationId};use workspace::{Toast, Workspace};
actions!( cli, [ /// Installs the Zed CLI tool to the system PATH. InstallCliBinary, ]);
const CANT_INSTALL_DOCS_URL: &str = "https://zed.dev/docs/macos#cant-install-cli";
/// Attempts to install the CLI symlink. Returns the installed path on success,/// or `None` if the user dismissed the macOS administrator authentication/// prompt. Returns an error if the install could not be completed, most/// commonly because the user is not an admin.async fn install_script(cx: &AsyncApp) -> Result<Option<PathBuf>> { let cli_path = cx.update(|cx| cx.path_for_auxiliary_executable("cli"))?; let link_path = Path::new("/usr/local/bin/zed"); let bin_dir_path = link_path.parent().unwrap();
// Don't re-create symlink if it points to the same CLI binary. if smol::fs::read_link(link_path).await.ok().as_ref() == Some(&cli_path) { return Ok(Some(link_path.into())); }
// If the symlink is not there or is outdated, first try replacing it // without escalating. smol::fs::remove_file(link_path).await.log_err(); if smol::fs::unix::symlink(&cli_path, link_path) .await .log_err() .is_some() { return Ok(Some(link_path.into())); }
// The symlink could not be created without escalating, so use osascript // with admin privileges to create it. let output = smol::process::Command::new("/usr/bin/osascript") .args([ "-e", &format!( "do shell script \" \ mkdir -p \'{}\' && \ ln -sf \'{}\' \'{}\' \ \" with administrator privileges", bin_dir_path.to_string_lossy(), cli_path.to_string_lossy(), link_path.to_string_lossy(), ), ]) .output() .await?;
if output.status.success() { return Ok(Some(link_path.into())); }
// osascript reports "User canceled." (error -128) when the administrator // prompt is dismissed. Treat that as a cancellation rather than a failure // so we don't show an error the user already chose to avoid. let stderr = String::from_utf8_lossy(&output.stderr); if stderr.contains("User canceled") || stderr.contains("-128") { return Ok(None); }
// The privileged write failed, most commonly because the user is not an // admin. anyhow::bail!("error running osascript: {}", stderr.trim());}
pub fn install_cli_binary(window: &mut Window, cx: &mut Context<Workspace>) { const LINUX_PROMPT_DETAIL: &str = "If you installed Zed from our official release add ~/.local/bin to your PATH.\n\nIf you installed Zed from a different source like your package manager, then you may need to create an alias/symlink manually.\n\nDepending on your package manager, the CLI might be named zeditor, zedit, zed-editor or something else.";
cx.spawn_in(window, async move |workspace, cx| { if cfg!(any(target_os = "linux", target_os = "freebsd")) { let prompt = cx.prompt( PromptLevel::Warning, "CLI should already be installed", Some(LINUX_PROMPT_DETAIL), &["OK"], ); cx.background_spawn(prompt).detach(); return Ok(()); } let path = match install_script(cx.deref()).await { Ok(Some(path)) => path, // The user dismissed the administrator prompt; nothing to do. Ok(None) => return Ok(()), Err(error) => { log::error!("failed to install zed CLI: {error:#}"); workspace.update(cx, |workspace, cx| { struct CliInstallFailed;
workspace.show_notification( NotificationId::unique::<CliInstallFailed>(), cx, |cx| { cx.new(|cx| { MessageNotification::new( "You can add `zed` to your PATH manually.", cx, ) .with_title("Couldn't install the Zed CLI") .more_info_message("Show me how") .more_info_url(CANT_INSTALL_DOCS_URL) }) }, ); })?; return Ok(()); } };
workspace.update_in(cx, |workspace, _, cx| { struct InstalledZedCli;
workspace.show_toast( Toast::new( NotificationId::unique::<InstalledZedCli>(), format!( "Installed `zed` to {}. You can launch {} from your terminal.", path.to_string_lossy(), ReleaseChannel::global(cx).display_name() ), ), cx, ) })?; register_zed_scheme(cx).await.log_err(); Ok(()) }) .detach_and_prompt_err("Cannot install the Zed CLI", window, cx, |_, _, _| None);}