use anyhow::Result;use client::UserStore;use copilot::{Copilot, Status};use editor::{ actions::{ShowEditPrediction, ToggleEditPrediction}, scroll::Autoscroll, Editor,};use feature_flags::{ FeatureFlagAppExt, PredictEditsFeatureFlag, PredictEditsRateCompletionsFeatureFlag,};use fs::Fs;use gpui::{ actions, div, pulsating_between, Action, Animation, AnimationExt, App, AsyncWindowContext, Corner, Entity, FocusHandle, Focusable, IntoElement, ParentElement, Render, Subscription, WeakEntity,};use indoc::indoc;use language::{ language_settings::{self, all_language_settings, AllLanguageSettings, EditPredictionProvider}, File, Language,};use regex::Regex;use settings::{update_settings_file, Settings, SettingsStore};use std::{ sync::{Arc, LazyLock}, time::Duration,};use supermaven::{AccountStatus, Supermaven};use ui::{ prelude::*, Clickable, ContextMenu, ContextMenuEntry, IconButton, IconButtonShape, Indicator, PopoverMenu, PopoverMenuHandle, Tooltip,};use workspace::{ create_and_open_local_file, item::ItemHandle, notifications::NotificationId, StatusItemView, Toast, Workspace,};use zed_actions::OpenBrowser;use zeta::RateCompletions; actions!(edit_prediction, [ToggleMenu]); const COPILOT_SETTINGS_URL: &str = "https://github.com/settings/copilot"; struct CopilotErrorToast; pub struct InlineCompletionButton { editor_subscription: Option<(Subscription, usize)>, editor_enabled: Option<bool>, editor_show_predictions: bool, editor_focus_handle: Option<FocusHandle>, language: Option<Arc<Language>>, file: Option<Arc<dyn File>>, edit_prediction_provider: Option<Arc<dyn inline_completion::InlineCompletionProviderHandle>>, fs: Arc<dyn Fs>, user_store: Entity<UserStore>, popover_menu_handle: PopoverMenuHandle<ContextMenu>,} enum SupermavenButtonStatus { Ready, Errored(String), NeedsActivation(String), Initializing,} impl Render for InlineCompletionButton { fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement { let all_language_settings = all_language_settings(None, cx); match all_language_settings.edit_predictions.provider { EditPredictionProvider::None => div(), EditPredictionProvider::Copilot => { let Some(copilot) = Copilot::global(cx) else { return div(); }; let status = copilot.read(cx).status(); let enabled = self.editor_enabled.unwrap_or(false); let icon = match status { Status::Error(_) => IconName::CopilotError, Status::Authorized => { if enabled { IconName::Copilot } else { IconName::CopilotDisabled } } _ => IconName::CopilotInit, }; if let Status::Error(e) = status { return div().child( IconButton::new("copilot-error", icon) .icon_size(IconSize::Small) .on_click(cx.listener(move |_, _, window, cx| { if let Some(workspace) = window.root::<Workspace>().flatten() { workspace.update(cx, |workspace, cx| { workspace.show_toast( Toast::new( NotificationId::unique::<CopilotErrorToast>(), format!("Copilot can't be started: {}", e), ) .on_click( "Reinstall Copilot", |_, cx| { if let Some(copilot) = Copilot::global(cx) { copilot .update(cx, |copilot, cx| { copilot.reinstall(cx) }) .detach(); } }, ), cx, ); }); } })) .tooltip(|window, cx| { Tooltip::for_action("GitHub Copilot", &ToggleMenu, window, cx) }), ); } let this = cx.entity().clone(); div().child( PopoverMenu::new("copilot") .menu(move |window, cx| { Some(match status { Status::Authorized => this.update(cx, |this, cx| { this.build_copilot_context_menu(window, cx) }), _ => this.update(cx, |this, cx| { this.build_copilot_start_menu(window, cx) }), }) }) .anchor(Corner::BottomRight) .trigger_with_tooltip( IconButton::new("copilot-icon", icon), |window, cx| { Tooltip::for_action("GitHub Copilot", &ToggleMenu, window, cx) }, ) .with_handle(self.popover_menu_handle.clone()), ) } EditPredictionProvider::Supermaven => { let Some(supermaven) = Supermaven::global(cx) else { return div(); }; let supermaven = supermaven.read(cx); let status = match supermaven { Supermaven::Starting => SupermavenButtonStatus::Initializing, Supermaven::FailedDownload { error } => { SupermavenButtonStatus::Errored(error.to_string()) } Supermaven::Spawned(agent) => { let account_status = agent.account_status.clone(); match account_status { AccountStatus::NeedsActivation { activate_url } => { SupermavenButtonStatus::NeedsActivation(activate_url.clone()) } AccountStatus::Unknown => SupermavenButtonStatus::Initializing, AccountStatus::Ready => SupermavenButtonStatus::Ready, } } Supermaven::Error { error } => { SupermavenButtonStatus::Errored(error.to_string()) } }; let icon = status.to_icon(); let tooltip_text = status.to_tooltip(); let has_menu = status.has_menu(); let this = cx.entity().clone(); let fs = self.fs.clone(); return div().child( PopoverMenu::new("supermaven") .menu(move |window, cx| match &status { SupermavenButtonStatus::NeedsActivation(activate_url) => { Some(ContextMenu::build(window, cx, |menu, _, _| { let fs = fs.clone(); let activate_url = activate_url.clone(); menu.entry("Sign In", None, move |_, cx| { cx.open_url(activate_url.as_str()) }) .entry( "Use Copilot", None, move |_, cx| { set_completion_provider( fs.clone(), cx, EditPredictionProvider::Copilot, ) }, ) })) } SupermavenButtonStatus::Ready => Some(this.update(cx, |this, cx| { this.build_supermaven_context_menu(window, cx) })), _ => None, }) .anchor(Corner::BottomRight) .trigger_with_tooltip( IconButton::new("supermaven-icon", icon), move |window, cx| { if has_menu { Tooltip::for_action( tooltip_text.clone(), &ToggleMenu, window, cx, ) } else { Tooltip::text(tooltip_text.clone())(window, cx) } }, ) .with_handle(self.popover_menu_handle.clone()), ); } EditPredictionProvider::Zed => { if !cx.has_flag::<PredictEditsFeatureFlag>() { return div(); } let enabled = self.editor_enabled.unwrap_or(true); let zeta_icon = if enabled { IconName::ZedPredict } else { IconName::ZedPredictDisabled }; let current_user_terms_accepted = self.user_store.read(cx).current_user_has_accepted_terms(); if !current_user_terms_accepted.unwrap_or(false) { let signed_in = current_user_terms_accepted.is_some(); let tooltip_meta = if signed_in { "Read Terms of Service" } else { "Sign in to use" }; return div().child( IconButton::new("zed-predict-pending-button", zeta_icon) .shape(IconButtonShape::Square) .indicator(Indicator::dot().color(Color::Error)) .indicator_border_color(Some(cx.theme().colors().status_bar_background)) .tooltip(move |window, cx| { Tooltip::with_meta( "Edit Predictions", None, tooltip_meta, window, cx, ) }) .on_click(cx.listener(move |_, _, window, cx| { telemetry::event!( "Pending ToS Clicked", source = "Edit Prediction Status Button" ); window.dispatch_action( zed_actions::OpenZedPredictOnboarding.boxed_clone(), cx, ); })), ); } let show_editor_predictions = self.editor_show_predictions; let icon_button = IconButton::new("zed-predict-pending-button", zeta_icon) .shape(IconButtonShape::Square) .when(enabled && !show_editor_predictions, |this| { this.indicator(Indicator::dot().color(Color::Muted)) .indicator_border_color(Some(cx.theme().colors().status_bar_background)) }) .when(!self.popover_menu_handle.is_deployed(), |element| { element.tooltip(move |window, cx| { if enabled { if show_editor_predictions { Tooltip::for_action("Edit Prediction", &ToggleMenu, window, cx) } else { Tooltip::with_meta( "Edit Prediction", Some(&ToggleMenu), "Hidden For This File", window, cx, ) } } else { Tooltip::with_meta( "Edit Prediction", Some(&ToggleMenu), "Disabled For This File", window, cx, ) } }) }); let this = cx.entity().clone(); let mut popover_menu = PopoverMenu::new("zeta") .menu(move |window, cx| { Some(this.update(cx, |this, cx| this.build_zeta_context_menu(window, cx))) }) .anchor(Corner::BottomRight) .with_handle(self.popover_menu_handle.clone()); let is_refreshing = self .edit_prediction_provider .as_ref() .map_or(false, |provider| provider.is_refreshing(cx)); if is_refreshing { popover_menu = popover_menu.trigger( icon_button.with_animation( "pulsating-label", Animation::new(Duration::from_secs(2)) .repeat() .with_easing(pulsating_between(0.2, 1.0)), |icon_button, delta| icon_button.alpha(delta), ), ); } else { popover_menu = popover_menu.trigger(icon_button); } div().child(popover_menu.into_any_element()) } } }} impl InlineCompletionButton { pub fn new( fs: Arc<dyn Fs>, user_store: Entity<UserStore>, popover_menu_handle: PopoverMenuHandle<ContextMenu>, cx: &mut Context<Self>, ) -> Self { if let Some(copilot) = Copilot::global(cx) { cx.observe(&copilot, |_, _, cx| cx.notify()).detach() } cx.observe_global::<SettingsStore>(move |_, cx| cx.notify()) .detach(); Self { editor_subscription: None, editor_enabled: None, editor_show_predictions: true, editor_focus_handle: None, language: None, file: None, edit_prediction_provider: None, popover_menu_handle, fs, user_store, } } pub fn build_copilot_start_menu( &mut self, window: &mut Window, cx: &mut Context<Self>, ) -> Entity<ContextMenu> { let fs = self.fs.clone(); ContextMenu::build(window, cx, |menu, _, _| { menu.entry("Sign In", None, copilot::initiate_sign_in) .entry("Disable Copilot", None, { let fs = fs.clone(); move |_window, cx| hide_copilot(fs.clone(), cx) }) .entry("Use Supermaven", None, { let fs = fs.clone(); move |_window, cx| { set_completion_provider(fs.clone(), cx, EditPredictionProvider::Supermaven) } }) }) } pub fn build_language_settings_menu( &self, mut menu: ContextMenu, window: &Window, cx: &mut App, ) -> ContextMenu { let fs = self.fs.clone(); let line_height = window.line_height(); menu = menu.header("Show Edit Predictions For"); let language_state = self.language.as_ref().map(|language| { ( language.clone(), language_settings::language_settings(Some(language.name()), None, cx) .show_edit_predictions, ) }); if let Some(editor_focus_handle) = self.editor_focus_handle.clone() { let entry = ContextMenuEntry::new("This Buffer") .toggleable(IconPosition::Start, self.editor_show_predictions) .action(Box::new(ToggleEditPrediction)) .handler(move |window, cx| { editor_focus_handle.dispatch_action(&ToggleEditPrediction, window, cx); }); match language_state.clone() { Some((language, false)) => { menu = menu.item( entry .disabled(true) .documentation_aside(move |_cx| { Label::new(format!("Edit predictions cannot be toggled for this buffer because they are disabled for {}", language.name())) .into_any_element() }) ); } Some(_) | None => menu = menu.item(entry), } } if let Some((language, language_enabled)) = language_state { let fs = fs.clone(); menu = menu.toggleable_entry( language.name(), language_enabled, IconPosition::Start, None, move |_, cx| { toggle_show_inline_completions_for_language(language.clone(), fs.clone(), cx) }, ); } let settings = AllLanguageSettings::get_global(cx); let globally_enabled = settings.show_edit_predictions(None, cx); menu = menu.toggleable_entry("All Files", globally_enabled, IconPosition::Start, None, { let fs = fs.clone(); move |_, cx| toggle_inline_completions_globally(fs.clone(), cx) }); menu = menu.separator().header("Privacy Settings"); if let Some(provider) = &self.edit_prediction_provider { let data_collection = provider.data_collection_state(cx); if data_collection.is_supported() { let provider = provider.clone(); let enabled = data_collection.is_enabled(); let is_open_source = data_collection.is_project_open_source(); let is_collecting = data_collection.is_enabled(); let (icon_name, icon_color) = if is_open_source && is_collecting { (IconName::Check, Color::Success) } else { (IconName::Check, Color::Accent) }; menu = menu.item( ContextMenuEntry::new("Training Data Collection") .toggleable(IconPosition::Start, data_collection.is_enabled()) .icon(icon_name) .icon_color(icon_color) .documentation_aside(move |cx| { let (msg, label_color, icon_name, icon_color) = match (is_open_source, is_collecting) { (true, true) => ( "Project identified as open source, and you're sharing data.", Color::Default, IconName::Check, Color::Success, ), (true, false) => ( "Project identified as open source, but you're not sharing data.", Color::Muted, IconName::Close, Color::Muted, ), (false, true) => ( "Project not identified as open source. No data captured.", Color::Muted,Lines 1–500 of 863. Load more
use anyhow::Result; use client::UserStore; use copilot::{Copilot, Status}; use editor::{ actions::{ShowEditPrediction, ToggleEditPrediction}, scroll::Autoscroll, Editor, }; use feature_flags::{ FeatureFlagAppExt, PredictEditsFeatureFlag, PredictEditsRateCompletionsFeatureFlag, }; use fs::Fs; use gpui::{ actions, div, pulsating_between, Action, Animation, AnimationExt, App, AsyncWindowContext, Corner, Entity, FocusHandle, Focusable, IntoElement, ParentElement, Render, Subscription, WeakEntity, }; use indoc::indoc; use language::{ language_settings::{self, all_language_settings, AllLanguageSettings, EditPredictionProvider}, File, Language, }; use regex::Regex; use settings::{update_settings_file, Settings, SettingsStore}; use std::{ sync::{Arc, LazyLock}, time::Duration, }; use supermaven::{AccountStatus, Supermaven}; use ui::{ prelude::*, Clickable, ContextMenu, ContextMenuEntry, IconButton, IconButtonShape, Indicator, PopoverMenu, PopoverMenuHandle, Tooltip, }; use workspace::{ create_and_open_local_file, item::ItemHandle, notifications::NotificationId, StatusItemView, Toast, Workspace, }; use zed_actions::OpenBrowser; use zeta::RateCompletions; actions!(edit_prediction, [ToggleMenu]); const COPILOT_SETTINGS_URL: &str = "https://github.com/settings/copilot"; struct CopilotErrorToast; pub struct InlineCompletionButton { editor_subscription: Option<(Subscription, usize)>, editor_enabled: Option<bool>, editor_show_predictions: bool, editor_focus_handle: Option<FocusHandle>, language: Option<Arc<Language>>, file: Option<Arc<dyn File>>, edit_prediction_provider: Option<Arc<dyn inline_completion::InlineCompletionProviderHandle>>, fs: Arc<dyn Fs>, user_store: Entity<UserStore>, popover_menu_handle: PopoverMenuHandle<ContextMenu>, } enum SupermavenButtonStatus { Ready, Errored(String), NeedsActivation(String), Initializing, } impl Render for InlineCompletionButton { fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement { let all_language_settings = all_language_settings(None, cx); match all_language_settings.edit_predictions.provider { EditPredictionProvider::None => div(), EditPredictionProvider::Copilot => { let Some(copilot) = Copilot::global(cx) else { return div(); }; let status = copilot.read(cx).status(); let enabled = self.editor_enabled.unwrap_or(false); let icon = match status { Status::Error(_) => IconName::CopilotError, Status::Authorized => { if enabled { IconName::Copilot } else { IconName::CopilotDisabled } } _ => IconName::CopilotInit, }; if let Status::Error(e) = status { return div().child( IconButton::new("copilot-error", icon) .icon_size(IconSize::Small) .on_click(cx.listener(move |_, _, window, cx| { if let Some(workspace) = window.root::<Workspace>().flatten() { workspace.update(cx, |workspace, cx| { workspace.show_toast( Toast::new( NotificationId::unique::<CopilotErrorToast>(), format!("Copilot can't be started: {}", e), ) .on_click( "Reinstall Copilot", |_, cx| { if let Some(copilot) = Copilot::global(cx) { copilot .update(cx, |copilot, cx| { copilot.reinstall(cx) }) .detach(); } }, ), cx, ); }); } })) .tooltip(|window, cx| { Tooltip::for_action("GitHub Copilot", &ToggleMenu, window, cx) }), ); } let this = cx.entity().clone(); div().child( PopoverMenu::new("copilot") .menu(move |window, cx| { Some(match status { Status::Authorized => this.update(cx, |this, cx| { this.build_copilot_context_menu(window, cx) }), _ => this.update(cx, |this, cx| { this.build_copilot_start_menu(window, cx) }), }) }) .anchor(Corner::BottomRight) .trigger_with_tooltip( IconButton::new("copilot-icon", icon), |window, cx| { Tooltip::for_action("GitHub Copilot", &ToggleMenu, window, cx) }, ) .with_handle(self.popover_menu_handle.clone()), ) } EditPredictionProvider::Supermaven => { let Some(supermaven) = Supermaven::global(cx) else { return div(); }; let supermaven = supermaven.read(cx); let status = match supermaven { Supermaven::Starting => SupermavenButtonStatus::Initializing, Supermaven::FailedDownload { error } => { SupermavenButtonStatus::Errored(error.to_string()) } Supermaven::Spawned(agent) => { let account_status = agent.account_status.clone(); match account_status { AccountStatus::NeedsActivation { activate_url } => { SupermavenButtonStatus::NeedsActivation(activate_url.clone()) } AccountStatus::Unknown => SupermavenButtonStatus::Initializing, AccountStatus::Ready => SupermavenButtonStatus::Ready, } } Supermaven::Error { error } => { SupermavenButtonStatus::Errored(error.to_string()) } }; let icon = status.to_icon(); let tooltip_text = status.to_tooltip(); let has_menu = status.has_menu(); let this = cx.entity().clone(); let fs = self.fs.clone(); return div().child( PopoverMenu::new("supermaven") .menu(move |window, cx| match &status { SupermavenButtonStatus::NeedsActivation(activate_url) => { Some(ContextMenu::build(window, cx, |menu, _, _| { let fs = fs.clone(); let activate_url = activate_url.clone(); menu.entry("Sign In", None, move |_, cx| { cx.open_url(activate_url.as_str()) }) .entry( "Use Copilot", None, move |_, cx| { set_completion_provider( fs.clone(), cx, EditPredictionProvider::Copilot, ) }, ) })) } SupermavenButtonStatus::Ready => Some(this.update(cx, |this, cx| { this.build_supermaven_context_menu(window, cx) })), _ => None, }) .anchor(Corner::BottomRight) .trigger_with_tooltip( IconButton::new("supermaven-icon", icon), move |window, cx| { if has_menu { Tooltip::for_action( tooltip_text.clone(), &ToggleMenu, window, cx, ) } else { Tooltip::text(tooltip_text.clone())(window, cx) } }, ) .with_handle(self.popover_menu_handle.clone()), ); } EditPredictionProvider::Zed => { if !cx.has_flag::<PredictEditsFeatureFlag>() { return div(); } let enabled = self.editor_enabled.unwrap_or(true); let zeta_icon = if enabled { IconName::ZedPredict } else { IconName::ZedPredictDisabled }; let current_user_terms_accepted = self.user_store.read(cx).current_user_has_accepted_terms(); if !current_user_terms_accepted.unwrap_or(false) { let signed_in = current_user_terms_accepted.is_some(); let tooltip_meta = if signed_in { "Read Terms of Service" } else { "Sign in to use" }; return div().child( IconButton::new("zed-predict-pending-button", zeta_icon) .shape(IconButtonShape::Square) .indicator(Indicator::dot().color(Color::Error)) .indicator_border_color(Some(cx.theme().colors().status_bar_background)) .tooltip(move |window, cx| { Tooltip::with_meta( "Edit Predictions", None, tooltip_meta, window, cx, ) }) .on_click(cx.listener(move |_, _, window, cx| { telemetry::event!( "Pending ToS Clicked", source = "Edit Prediction Status Button" ); window.dispatch_action( zed_actions::OpenZedPredictOnboarding.boxed_clone(), cx, ); })), ); } let show_editor_predictions = self.editor_show_predictions; let icon_button = IconButton::new("zed-predict-pending-button", zeta_icon) .shape(IconButtonShape::Square) .when(enabled && !show_editor_predictions, |this| { this.indicator(Indicator::dot().color(Color::Muted)) .indicator_border_color(Some(cx.theme().colors().status_bar_background)) }) .when(!self.popover_menu_handle.is_deployed(), |element| { element.tooltip(move |window, cx| { if enabled { if show_editor_predictions { Tooltip::for_action("Edit Prediction", &ToggleMenu, window, cx) } else { Tooltip::with_meta( "Edit Prediction", Some(&ToggleMenu), "Hidden For This File", window, cx, ) } } else { Tooltip::with_meta( "Edit Prediction", Some(&ToggleMenu), "Disabled For This File", window, cx, ) } }) }); let this = cx.entity().clone(); let mut popover_menu = PopoverMenu::new("zeta") .menu(move |window, cx| { Some(this.update(cx, |this, cx| this.build_zeta_context_menu(window, cx))) }) .anchor(Corner::BottomRight) .with_handle(self.popover_menu_handle.clone()); let is_refreshing = self .edit_prediction_provider .as_ref() .map_or(false, |provider| provider.is_refreshing(cx)); if is_refreshing { popover_menu = popover_menu.trigger( icon_button.with_animation( "pulsating-label", Animation::new(Duration::from_secs(2)) .repeat() .with_easing(pulsating_between(0.2, 1.0)), |icon_button, delta| icon_button.alpha(delta), ), ); } else { popover_menu = popover_menu.trigger(icon_button); } div().child(popover_menu.into_any_element()) } } } } impl InlineCompletionButton { pub fn new( fs: Arc<dyn Fs>, user_store: Entity<UserStore>, popover_menu_handle: PopoverMenuHandle<ContextMenu>, cx: &mut Context<Self>, ) -> Self { if let Some(copilot) = Copilot::global(cx) { cx.observe(&copilot, |_, _, cx| cx.notify()).detach() } cx.observe_global::<SettingsStore>(move |_, cx| cx.notify()) .detach(); Self { editor_subscription: None, editor_enabled: None, editor_show_predictions: true, editor_focus_handle: None, language: None, file: None, edit_prediction_provider: None, popover_menu_handle, fs, user_store, } } pub fn build_copilot_start_menu( &mut self, window: &mut Window, cx: &mut Context<Self>, ) -> Entity<ContextMenu> { let fs = self.fs.clone(); ContextMenu::build(window, cx, |menu, _, _| { menu.entry("Sign In", None, copilot::initiate_sign_in) .entry("Disable Copilot", None, { let fs = fs.clone(); move |_window, cx| hide_copilot(fs.clone(), cx) }) .entry("Use Supermaven", None, { let fs = fs.clone(); move |_window, cx| { set_completion_provider(fs.clone(), cx, EditPredictionProvider::Supermaven) } }) }) } pub fn build_language_settings_menu( &self, mut menu: ContextMenu, window: &Window, cx: &mut App, ) -> ContextMenu { let fs = self.fs.clone(); let line_height = window.line_height(); menu = menu.header("Show Edit Predictions For"); let language_state = self.language.as_ref().map(|language| { ( language.clone(), language_settings::language_settings(Some(language.name()), None, cx) .show_edit_predictions, ) }); if let Some(editor_focus_handle) = self.editor_focus_handle.clone() { let entry = ContextMenuEntry::new("This Buffer") .toggleable(IconPosition::Start, self.editor_show_predictions) .action(Box::new(ToggleEditPrediction)) .handler(move |window, cx| { editor_focus_handle.dispatch_action(&ToggleEditPrediction, window, cx); }); match language_state.clone() { Some((language, false)) => { menu = menu.item( entry .disabled(true) .documentation_aside(move |_cx| { Label::new(format!("Edit predictions cannot be toggled for this buffer because they are disabled for {}", language.name())) .into_any_element() }) ); } Some(_) | None => menu = menu.item(entry), } } if let Some((language, language_enabled)) = language_state { let fs = fs.clone(); menu = menu.toggleable_entry( language.name(), language_enabled, IconPosition::Start, None, move |_, cx| { toggle_show_inline_completions_for_language(language.clone(), fs.clone(), cx) }, ); } let settings = AllLanguageSettings::get_global(cx); let globally_enabled = settings.show_edit_predictions(None, cx); menu = menu.toggleable_entry("All Files", globally_enabled, IconPosition::Start, None, { let fs = fs.clone(); move |_, cx| toggle_inline_completions_globally(fs.clone(), cx) }); menu = menu.separator().header("Privacy Settings"); if let Some(provider) = &self.edit_prediction_provider { let data_collection = provider.data_collection_state(cx); if data_collection.is_supported() { let provider = provider.clone(); let enabled = data_collection.is_enabled(); let is_open_source = data_collection.is_project_open_source(); let is_collecting = data_collection.is_enabled(); let (icon_name, icon_color) = if is_open_source && is_collecting { (IconName::Check, Color::Success) } else { (IconName::Check, Color::Accent) }; menu = menu.item( ContextMenuEntry::new("Training Data Collection") .toggleable(IconPosition::Start, data_collection.is_enabled()) .icon(icon_name) .icon_color(icon_color) .documentation_aside(move |cx| { let (msg, label_color, icon_name, icon_color) = match (is_open_source, is_collecting) { (true, true) => ( "Project identified as open source, and you're sharing data.", Color::Default, IconName::Check, Color::Success, ), (true, false) => ( "Project identified as open source, but you're not sharing data.", Color::Muted, IconName::Close, Color::Muted, ), (false, true) => ( "Project not identified as open source. No data captured.", Color::Muted,