// Allow binary to be called Zed for a nice application menu when running executable directly #![allow(non_snake_case)] use anyhow::{anyhow, Context, Result}; use backtrace::Backtrace; use channel::ChannelStore; use cli::{ ipc::{self, IpcSender}, CliRequest, CliResponse, IpcHandshake, FORCE_CLI_MODE_ENV_VAR_NAME, }; use client::{self, TelemetrySettings, UserStore, ZED_APP_VERSION, ZED_SECRET_CLIENT_TOKEN}; use db::kvp::KEY_VALUE_STORE; use editor::{scroll::autoscroll::Autoscroll, Editor}; use futures::{ channel::{mpsc, oneshot}, FutureExt, SinkExt, StreamExt, }; use gpui::{Action, App, AppContext, AssetSource, AsyncAppContext, Task}; use isahc::{config::Configurable, Request}; use language::{LanguageRegistry, Point}; use log::LevelFilter; use node_runtime::NodeRuntime; use parking_lot::Mutex; use project::Fs; use serde::{Deserialize, Serialize}; use settings::{default_settings, handle_settings_file_changes, watch_config_file, SettingsStore}; use simplelog::ConfigBuilder; use smol::process::Command; use std::{ collections::HashMap, env, ffi::OsStr, fs::OpenOptions, io::Write as _, os::unix::prelude::OsStrExt, panic, path::{Path, PathBuf}, str, sync::{ atomic::{AtomicBool, AtomicU32, Ordering}, Arc, Weak, }, thread, time::{Duration, SystemTime, UNIX_EPOCH}, }; use sum_tree::Bias; use util::{ channel::ReleaseChannel, http::{self, HttpClient}, paths::PathLikeWithPosition, }; use uuid::Uuid; use welcome::{show_welcome_experience, FIRST_OPEN}; use fs::RealFs; use util::{channel::RELEASE_CHANNEL, paths, ResultExt, TryFutureExt}; use workspace::AppState; use zed::{ assets::Assets, build_window_options, handle_keymap_file_changes, initialize_workspace, languages, menus, only_instance::{ensure_only_instance, IsOnlyInstance}, }; fn main() { let http = http::client(); init_paths(); init_logger(); if ensure_only_instance() != IsOnlyInstance::Yes { return; } log::info!("========== starting zed =========="); let mut app = gpui::App::new(Assets).unwrap(); let installation_id = app.background().block(installation_id()).ok(); init_panic_hook(&app, installation_id.clone()); load_embedded_fonts(&app); let fs = Arc::new(RealFs); let user_settings_file_rx = watch_config_file(app.background(), fs.clone(), paths::SETTINGS.clone()); let user_keymap_file_rx = watch_config_file(app.background(), fs.clone(), paths::KEYMAP.clone()); let login_shell_env_loaded = if stdout_is_a_pty() { Task::ready(()) } else { app.background().spawn(async { load_login_shell_environment().await.log_err(); }) }; let (cli_connections_tx, mut cli_connections_rx) = mpsc::unbounded(); let cli_connections_tx = Arc::new(cli_connections_tx); let (open_paths_tx, mut open_paths_rx) = mpsc::unbounded(); let open_paths_tx = Arc::new(open_paths_tx); let urls_callback_triggered = Arc::new(AtomicBool::new(false)); let callback_cli_connections_tx = Arc::clone(&cli_connections_tx); let callback_open_paths_tx = Arc::clone(&open_paths_tx); let callback_urls_callback_triggered = Arc::clone(&urls_callback_triggered); app.on_open_urls(move |urls, _| { callback_urls_callback_triggered.store(true, Ordering::Release); open_urls(urls, &callback_cli_connections_tx, &callback_open_paths_tx); }) .on_reopen(move |cx| { if cx.has_global::>() { if let Some(app_state) = cx.global::>().upgrade() { workspace::open_new(&app_state, cx, |workspace, cx| { Editor::new_file(workspace, &Default::default(), cx) }) .detach(); } } }); app.run(move |cx| { cx.set_global(*RELEASE_CHANNEL); #[cfg(debug_assertions)] { use feature_flags::FeatureFlagAppExt; cx.set_staff(true); } let mut store = SettingsStore::default(); store .set_default_settings(default_settings().as_ref(), cx) .unwrap(); cx.set_global(store); handle_settings_file_changes(user_settings_file_rx, cx); handle_keymap_file_changes(user_keymap_file_rx, cx); let client = client::Client::new(http.clone(), cx); let mut languages = LanguageRegistry::new(login_shell_env_loaded); languages.set_executor(cx.background().clone()); languages.set_language_server_download_dir(paths::LANGUAGES_DIR.clone()); let languages = Arc::new(languages); let node_runtime = NodeRuntime::instance(http.clone()); languages::init(languages.clone(), node_runtime.clone()); let user_store = cx.add_model(|cx| UserStore::new(client.clone(), http.clone(), cx)); let channel_store = cx.add_model(|cx| ChannelStore::new(client.clone(), user_store.clone(), cx)); cx.set_global(client.clone()); theme::init(Assets, cx); context_menu::init(cx); project::Project::init(&client, cx); client::init(&client, cx); command_palette::init(cx); language::init(cx); editor::init(cx); go_to_line::init(cx); file_finder::init(cx); outline::init(cx); project_symbols::init(cx); project_panel::init(Assets, cx); channel::init(&client); diagnostics::init(cx); search::init(cx); semantic_index::init(fs.clone(), http.clone(), languages.clone(), cx); vim::init(cx); terminal_view::init(cx); copilot::init(http.clone(), node_runtime, cx); ai::init(cx); component_test::init(cx); cx.spawn(|cx| watch_themes(fs.clone(), cx)).detach(); cx.spawn(|_| watch_languages(fs.clone(), languages.clone())) .detach(); watch_file_types(fs.clone(), cx); languages.set_theme(theme::current(cx).clone()); cx.observe_global::({ let languages = languages.clone(); move |cx| languages.set_theme(theme::current(cx).clone()) }) .detach(); client.telemetry().start(installation_id); let app_state = Arc::new(AppState { languages, client: client.clone(), user_store, channel_store, fs, build_window_options, initialize_workspace, background_actions, }); cx.set_global(Arc::downgrade(&app_state)); audio::init(Assets, cx); auto_update::init(http.clone(), client::ZED_SERVER_URL.clone(), cx); workspace::init(app_state.clone(), cx); recent_projects::init(cx); journal::init(app_state.clone(), cx); language_selector::init(cx); theme_selector::init(cx); activity_indicator::init(cx); language_tools::init(cx); call::init(app_state.client.clone(), app_state.user_store.clone(), cx); collab_ui::init(&app_state, cx); feedback::init(cx); welcome::init(cx); zed::init(&app_state, cx); cx.set_menus(menus::menus()); if stdout_is_a_pty() { cx.platform().activate(true); let paths = collect_path_args(); if paths.is_empty() { cx.spawn(|cx| async move { restore_or_create_workspace(&app_state, cx).await }) .detach() } else { workspace::open_paths(&paths, &app_state, None, cx).detach_and_log_err(cx); } } else { upload_previous_panics(http.clone(), cx); // TODO Development mode that forces the CLI mode usually runs Zed binary as is instead // of an *app, hence gets no specific callbacks run. Emulate them here, if needed. if std::env::var(FORCE_CLI_MODE_ENV_VAR_NAME).ok().is_some() && !urls_callback_triggered.load(Ordering::Acquire) { open_urls(collect_url_args(), &cli_connections_tx, &open_paths_tx) } if let Ok(Some(connection)) = cli_connections_rx.try_next() { cx.spawn(|cx| handle_cli_connection(connection, app_state.clone(), cx)) .detach(); } else if let Ok(Some(paths)) = open_paths_rx.try_next() { cx.update(|cx| workspace::open_paths(&paths, &app_state, None, cx)) .detach(); } else { cx.spawn({ let app_state = app_state.clone(); |cx| async move { restore_or_create_workspace(&app_state, cx).await } }) .detach() } cx.spawn(|cx| { let app_state = app_state.clone(); async move { while let Some(connection) = cli_connections_rx.next().await { handle_cli_connection(connection, app_state.clone(), cx.clone()).await; } } }) .detach(); cx.spawn(|mut cx| { let app_state = app_state.clone(); async move { while let Some(paths) = open_paths_rx.next().await { cx.update(|cx| workspace::open_paths(&paths, &app_state, None, cx)) .detach(); } } }) .detach(); } cx.spawn(|cx| async move { if stdout_is_a_pty() { if client::IMPERSONATE_LOGIN.is_some() { client.authenticate_and_connect(false, &cx).await?; } } else if client.has_keychain_credentials(&cx) { client.authenticate_and_connect(true, &cx).await?; } Ok::<_, anyhow::Error>(()) }) .detach_and_log_err(cx); }); } async fn installation_id() -> Result { let legacy_key_name = "device_id"; if let Ok(Some(installation_id)) = KEY_VALUE_STORE.read_kvp(legacy_key_name) { Ok(installation_id) } else { let installation_id = Uuid::new_v4().to_string(); KEY_VALUE_STORE .write_kvp(legacy_key_name.to_string(), installation_id.clone()) .await?; Ok(installation_id) } } fn open_urls( urls: Vec, cli_connections_tx: &mpsc::UnboundedSender<( mpsc::Receiver, IpcSender, )>, open_paths_tx: &mpsc::UnboundedSender>, ) { if let Some(server_name) = urls.first().and_then(|url| url.strip_prefix("zed-cli://")) { if let Some(cli_connection) = connect_to_cli(server_name).log_err() { cli_connections_tx .unbounded_send(cli_connection) .map_err(|_| anyhow!("no listener for cli connections")) .log_err(); }; } else { let paths: Vec<_> = urls .iter() .flat_map(|url| url.strip_prefix("file://")) .map(|url| { let decoded = urlencoding::decode_binary(url.as_bytes()); PathBuf::from(OsStr::from_bytes(decoded.as_ref())) }) .collect(); open_paths_tx .unbounded_send(paths) .map_err(|_| anyhow!("no listener for open urls requests")) .log_err(); } } async fn restore_or_create_workspace(app_state: &Arc, mut cx: AsyncAppContext) { if let Some(location) = workspace::last_opened_workspace_paths().await { cx.update(|cx| workspace::open_paths(location.paths().as_ref(), app_state, None, cx)) .await .log_err(); } else if matches!(KEY_VALUE_STORE.read_kvp(FIRST_OPEN), Ok(None)) { cx.update(|cx| show_welcome_experience(app_state, cx)); } else { cx.update(|cx| { workspace::open_new(app_state, cx, |workspace, cx| { Editor::new_file(workspace, &Default::default(), cx) }) .detach(); }); } } fn init_paths() { std::fs::create_dir_all(&*util::paths::CONFIG_DIR).expect("could not create config path"); std::fs::create_dir_all(&*util::paths::LANGUAGES_DIR).expect("could not create languages path"); std::fs::create_dir_all(&*util::paths::DB_DIR).expect("could not create database path"); std::fs::create_dir_all(&*util::paths::LOGS_DIR).expect("could not create logs path"); } fn init_logger() { if stdout_is_a_pty() { env_logger::init(); } else { let level = LevelFilter::Info; // Prevent log file from becoming too large. const KIB: u64 = 1024; const MIB: u64 = 1024 * KIB; const MAX_LOG_BYTES: u64 = MIB; if std::fs::metadata(&*paths::LOG).map_or(false, |metadata| metadata.len() > MAX_LOG_BYTES) { let _ = std::fs::rename(&*paths::LOG, &*paths::OLD_LOG); } let log_file = OpenOptions::new() .create(true) .append(true) .open(&*paths::LOG) .expect("could not open logfile"); let config = ConfigBuilder::new() .set_time_format_str("%Y-%m-%dT%T") //All timestamps are UTC .build(); simplelog::WriteLogger::init(level, config, log_file).expect("could not initialize logger"); } } #[derive(Serialize, Deserialize)] struct LocationData { file: String, line: u32, } #[derive(Serialize, Deserialize)] struct Panic { thread: String, payload: String, #[serde(skip_serializing_if = "Option::is_none")] location_data: Option, backtrace: Vec, app_version: String, release_channel: String, os_name: String, os_version: Option, architecture: String, panicked_on: u128, #[serde(skip_serializing_if = "Option::is_none")] installation_id: Option, } #[derive(Serialize)] struct PanicRequest { panic: Panic, token: String, } static PANIC_COUNT: AtomicU32 = AtomicU32::new(0); fn init_panic_hook(app: &App, installation_id: Option) { let is_pty = stdout_is_a_pty(); let platform = app.platform(); panic::set_hook(Box::new(move |info| { let prior_panic_count = PANIC_COUNT.fetch_add(1, Ordering::SeqCst); if prior_panic_count > 0 { // Give the panic-ing thread time to write the panic file loop { std::thread::yield_now(); } } let thread = thread::current(); let thread_name = thread.name().unwrap_or(""); let payload = info .payload() .downcast_ref::<&str>() .map(|s| s.to_string()) .or_else(|| info.payload().downcast_ref::().map(|s| s.clone())) .unwrap_or_else(|| "Box".to_string()); if *util::channel::RELEASE_CHANNEL == ReleaseChannel::Dev { let location = info.location().unwrap(); let backtrace = Backtrace::new(); eprintln!( "Thread {:?} panicked with {:?} at {}:{}:{}\n{:?}", thread_name, payload, location.file(), location.line(), location.column(), backtrace, ); std::process::exit(-1); } let app_version = ZED_APP_VERSION .or_else(|| platform.app_version().ok()) .map_or("dev".to_string(), |v| v.to_string()); let backtrace = Backtrace::new(); let mut backtrace = backtrace .frames() .iter() .filter_map(|frame| Some(format!("{:#}", frame.symbols().first()?.name()?))) .collect::>(); // Strip out leading stack frames for rust panic-handling. if let Some(ix) = backtrace .iter() .position(|name| name == "rust_begin_unwind") { backtrace.drain(0..=ix); } let panic_data = Panic { thread: thread_name.into(), payload: payload.into(), location_data: info.location().map(|location| LocationData { file: location.file().into(), line: location.line(), }), app_version: app_version.clone(), release_channel: RELEASE_CHANNEL.dev_name().into(), os_name: platform.os_name().into(), os_version: platform .os_version() .ok() .map(|os_version| os_version.to_string()), architecture: env::consts::ARCH.into(), panicked_on: SystemTime::now() .duration_since(UNIX_EPOCH) .unwrap() .as_millis(), backtrace, installation_id: installation_id.clone(), }; if is_pty { if let Some(panic_data_json) = serde_json::to_string_pretty(&panic_data).log_err() { eprintln!("{}", panic_data_json); } } else { if let Some(panic_data_json) = serde_json::to_string(&panic_data).log_err() { let timestamp = chrono::Utc::now().format("%Y_%m_%d %H_%M_%S").to_string(); let panic_file_path = paths::LOGS_DIR.join(format!("zed-{}.panic", timestamp)); let panic_file = std::fs::OpenOptions::new() .append(true) .create(true) .open(&panic_file_path) .log_err(); if let Some(mut panic_file) = panic_file { writeln!(&mut panic_file, "{}", panic_data_json).log_err(); panic_file.flush().log_err(); } } } std::process::abort(); })); } fn upload_previous_panics(http: Arc, cx: &mut AppContext) { let telemetry_settings = *settings::get::(cx); cx.background() .spawn({ async move { let panic_report_url = format!("{}/api/panic", &*client::ZED_SERVER_URL); let mut children = smol::fs::read_dir(&*paths::LOGS_DIR).await?; while let Some(child) = children.next().await { let child = child?; let child_path = child.path(); if child_path.extension() != Some(OsStr::new("panic")) { continue; } let filename = if let Some(filename) = child_path.file_name() { filename.to_string_lossy() } else { continue; }; if !filename.starts_with("zed") { continue; } if telemetry_settings.diagnostics { let panic_file_content = smol::fs::read_to_string(&child_path) .await .context("error reading panic file")?; let panic = serde_json::from_str(&panic_file_content) .ok() .or_else(|| { panic_file_content .lines() .next() .and_then(|line| serde_json::from_str(line).ok()) }) .unwrap_or_else(|| { log::error!( "failed to deserialize panic file {:?}", panic_file_content ); None }); if let Some(panic) = panic { let body = serde_json::to_string(&PanicRequest { panic, token: ZED_SECRET_CLIENT_TOKEN.into(), }) .unwrap(); let request = Request::post(&panic_report_url) .redirect_policy(isahc::config::RedirectPolicy::Follow) .header("Content-Type", "application/json") .body(body.into())?; let response = http.send(request).await.context("error sending panic")?; if !response.status().is_success() { log::error!( "Error uploading panic to server: {}", response.status() ); } } } // We've done what we can, delete the file std::fs::remove_file(child_path) .context("error removing panic") .log_err(); } Ok::<_, anyhow::Error>(()) } .log_err() }) .detach(); } async fn load_login_shell_environment() -> Result<()> { let marker = "ZED_LOGIN_SHELL_START"; let shell = env::var("SHELL").context( "SHELL environment variable is not assigned so we can't source login environment variables", )?; let output = Command::new(&shell) .args(["-lic", &format!("echo {marker} && /usr/bin/env -0")]) .output() .await .context("failed to spawn login shell to source login environment variables")?; if !output.status.success() { Err(anyhow!("login shell exited with error"))?; } let stdout = String::from_utf8_lossy(&output.stdout); if let Some(env_output_start) = stdout.find(marker) { let env_output = &stdout[env_output_start + marker.len()..]; for line in env_output.split_terminator('\0') { if let Some(separator_index) = line.find('=') { let key = &line[..separator_index]; let value = &line[separator_index + 1..]; env::set_var(key, value); } } log::info!( "set environment variables from shell:{}, path:{}", shell, env::var("PATH").unwrap_or_default(), ); } Ok(()) } fn stdout_is_a_pty() -> bool { std::env::var(FORCE_CLI_MODE_ENV_VAR_NAME).ok().is_none() && unsafe { libc::isatty(libc::STDOUT_FILENO as i32) != 0 } } fn collect_path_args() -> Vec { env::args() .skip(1) .filter_map(|arg| match std::fs::canonicalize(arg) { Ok(path) => Some(path), Err(error) => { log::error!("error parsing path argument: {}", error); None } }) .collect() } fn collect_url_args() -> Vec { env::args().skip(1).collect() } fn load_embedded_fonts(app: &App) { let font_paths = Assets.list("fonts"); let embedded_fonts = Mutex::new(Vec::new()); smol::block_on(app.background().scoped(|scope| { for font_path in &font_paths { if !font_path.ends_with(".ttf") { continue; } scope.spawn(async { let font_path = &*font_path; let font_bytes = Assets.load(font_path).unwrap().to_vec(); embedded_fonts.lock().push(Arc::from(font_bytes)); }); } })); app.platform() .fonts() .add_fonts(&embedded_fonts.into_inner()) .unwrap(); } #[cfg(debug_assertions)] async fn watch_themes(fs: Arc, mut cx: AsyncAppContext) -> Option<()> { let mut events = fs .watch("styles/src".as_ref(), Duration::from_millis(100)) .await; while (events.next().await).is_some() { let output = Command::new("npm") .current_dir("styles") .args(["run", "build"]) .output() .await .log_err()?; if output.status.success() { cx.update(|cx| theme_selector::reload(cx)) } else { eprintln!( "build script failed {}", String::from_utf8_lossy(&output.stderr) ); } } Some(()) } #[cfg(debug_assertions)] async fn watch_languages(fs: Arc, languages: Arc) -> Option<()> { let mut events = fs .watch( "crates/zed/src/languages".as_ref(), Duration::from_millis(100), ) .await; while (events.next().await).is_some() { languages.reload(); } Some(()) } #[cfg(debug_assertions)] fn watch_file_types(fs: Arc, cx: &mut AppContext) { cx.spawn(|mut cx| async move { let mut events = fs .watch( "assets/icons/file_icons/file_types.json".as_ref(), Duration::from_millis(100), ) .await; while (events.next().await).is_some() { cx.update(|cx| { cx.update_global(|file_types, _| { *file_types = project_panel::file_associations::FileAssociations::new(Assets); }); }) } }) .detach() } #[cfg(not(debug_assertions))] async fn watch_themes(_fs: Arc, _cx: AsyncAppContext) -> Option<()> { None } #[cfg(not(debug_assertions))] async fn watch_languages(_: Arc, _: Arc) -> Option<()> { None } #[cfg(not(debug_assertions))] fn watch_file_types(_fs: Arc, _cx: &mut AppContext) {} fn connect_to_cli( server_name: &str, ) -> Result<(mpsc::Receiver, IpcSender)> { let handshake_tx = cli::ipc::IpcSender::::connect(server_name.to_string()) .context("error connecting to cli")?; let (request_tx, request_rx) = ipc::channel::()?; let (response_tx, response_rx) = ipc::channel::()?; handshake_tx .send(IpcHandshake { requests: request_tx, responses: response_rx, }) .context("error sending ipc handshake")?; let (mut async_request_tx, async_request_rx) = futures::channel::mpsc::channel::(16); thread::spawn(move || { while let Ok(cli_request) = request_rx.recv() { if smol::block_on(async_request_tx.send(cli_request)).is_err() { break; } } Ok::<_, anyhow::Error>(()) }); Ok((async_request_rx, response_tx)) } async fn handle_cli_connection( (mut requests, responses): (mpsc::Receiver, IpcSender), app_state: Arc, mut cx: AsyncAppContext, ) { if let Some(request) = requests.next().await { match request { CliRequest::Open { paths, wait } => { let mut caret_positions = HashMap::new(); let paths = if paths.is_empty() { workspace::last_opened_workspace_paths() .await .map(|location| location.paths().to_vec()) .unwrap_or_default() } else { paths .into_iter() .filter_map(|path_with_position_string| { let path_with_position = PathLikeWithPosition::parse_str( &path_with_position_string, |path_str| { Ok::<_, std::convert::Infallible>( Path::new(path_str).to_path_buf(), ) }, ) .expect("Infallible"); let path = path_with_position.path_like; if let Some(row) = path_with_position.row { if path.is_file() { let row = row.saturating_sub(1); let col = path_with_position.column.unwrap_or(0).saturating_sub(1); caret_positions.insert(path.clone(), Point::new(row, col)); } } Some(path) }) .collect() }; let mut errored = false; match cx .update(|cx| workspace::open_paths(&paths, &app_state, None, cx)) .await { Ok((workspace, items)) => { let mut item_release_futures = Vec::new(); for (item, path) in items.into_iter().zip(&paths) { match item { Some(Ok(item)) => { if let Some(point) = caret_positions.remove(path) { if let Some(active_editor) = item.downcast::() { active_editor .downgrade() .update(&mut cx, |editor, cx| { let snapshot = editor.snapshot(cx).display_snapshot; let point = snapshot .buffer_snapshot .clip_point(point, Bias::Left); editor.change_selections( Some(Autoscroll::center()), cx, |s| s.select_ranges([point..point]), ); }) .log_err(); } } let released = oneshot::channel(); cx.update(|cx| { item.on_release( cx, Box::new(move |_| { let _ = released.0.send(()); }), ) .detach(); }); item_release_futures.push(released.1); } Some(Err(err)) => { responses .send(CliResponse::Stderr { message: format!("error opening {:?}: {}", path, err), }) .log_err(); errored = true; } None => {} } } if wait { let background = cx.background(); let wait = async move { if paths.is_empty() { let (done_tx, done_rx) = oneshot::channel(); if let Some(workspace) = workspace.upgrade(&cx) { let _subscription = cx.update(|cx| { cx.observe_release(&workspace, move |_, _| { let _ = done_tx.send(()); }) }); drop(workspace); let _ = done_rx.await; } } else { let _ = futures::future::try_join_all(item_release_futures).await; }; } .fuse(); futures::pin_mut!(wait); loop { // Repeatedly check if CLI is still open to avoid wasting resources // waiting for files or workspaces to close. let mut timer = background.timer(Duration::from_secs(1)).fuse(); futures::select_biased! { _ = wait => break, _ = timer => { if responses.send(CliResponse::Ping).is_err() { break; } } } } } } Err(error) => { errored = true; responses .send(CliResponse::Stderr { message: format!("error opening {:?}: {}", paths, error), }) .log_err(); } } responses .send(CliResponse::Exit { status: i32::from(errored), }) .log_err(); } } } } pub fn background_actions() -> &'static [(&'static str, &'static dyn Action)] { &[ ("Go to file", &file_finder::Toggle), ("Open command palette", &command_palette::Toggle), ("Open recent projects", &recent_projects::OpenRecent), ("Change your settings", &zed_actions::OpenSettings), ] }