use anyhow::Result;use client::UserStore;use copilot::{Copilot, Status};use editor::{scroll::Autoscroll, Editor};use feature_flags::{FeatureFlagAppExt, PredictEditsFeatureFlag};use fs::Fs;use gpui::{ actions, div, pulsating_between, Action, Animation, AnimationExt, AppContext, AsyncWindowContext, Corner, Entity, IntoElement, Model, ParentElement, Render, Subscription, View, ViewContext, WeakView, WindowContext,};use language::{ language_settings::{ self, all_language_settings, AllLanguageSettings, InlineCompletionProvider, }, File, Language,};use settings::{update_settings_file, Settings, SettingsStore};use std::{path::Path, sync::Arc, time::Duration};use supermaven::{AccountStatus, Supermaven};use ui::{ActiveTheme as _, ButtonLike, Color, Icon, IconWithIndicator, Indicator};use workspace::{ create_and_open_local_file, item::ItemHandle, notifications::NotificationId, ui::{ ButtonCommon, Clickable, ContextMenu, IconButton, IconName, IconSize, PopoverMenu, Tooltip, }, StatusItemView, Toast, Workspace,};use zed_actions::OpenBrowser;use zed_predict_tos::ZedPredictTos;use zeta::RateCompletionModal; actions!(zeta, [RateCompletions]); const COPILOT_SETTINGS_URL: &str = "https://github.com/settings/copilot"; struct CopilotErrorToast; pub struct InlineCompletionButton { editor_subscription: Option<(Subscription, usize)>, editor_enabled: Option<bool>, language: Option<Arc<Language>>, file: Option<Arc<dyn File>>, inline_completion_provider: Option<Arc<dyn inline_completion::InlineCompletionProviderHandle>>, fs: Arc<dyn Fs>, workspace: WeakView<Workspace>, user_store: Model<UserStore>,} enum SupermavenButtonStatus { Ready, Errored(String), NeedsActivation(String), Initializing,} impl Render for InlineCompletionButton { fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement { let all_language_settings = all_language_settings(None, cx); match all_language_settings.inline_completions.provider { InlineCompletionProvider::None => div(), InlineCompletionProvider::Copilot => { let Some(copilot) = Copilot::global(cx) else { return div(); }; let status = copilot.read(cx).status(); let enabled = self.editor_enabled.unwrap_or_else(|| { all_language_settings.inline_completions_enabled(None, None, cx) }); 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 |_, _, cx| { if let Some(workspace) = cx.window_handle().downcast::<Workspace>() { 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, ); }) .ok(); } })) .tooltip(|cx| Tooltip::text("GitHub Copilot", cx)), ); } let this = cx.view().clone(); div().child( PopoverMenu::new("copilot") .menu(move |cx| { Some(match status { Status::Authorized => { this.update(cx, |this, cx| this.build_copilot_context_menu(cx)) } _ => this.update(cx, |this, cx| this.build_copilot_start_menu(cx)), }) }) .anchor(Corner::BottomRight) .trigger( IconButton::new("copilot-icon", icon) .tooltip(|cx| Tooltip::text("GitHub Copilot", cx)), ), ) } InlineCompletionProvider::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 this = cx.view().clone(); let fs = self.fs.clone(); return div().child( PopoverMenu::new("supermaven") .menu(move |cx| match &status { SupermavenButtonStatus::NeedsActivation(activate_url) => { Some(ContextMenu::build(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, InlineCompletionProvider::Copilot, ) }, ) })) } SupermavenButtonStatus::Ready => Some( this.update(cx, |this, cx| this.build_supermaven_context_menu(cx)), ), _ => None, }) .anchor(Corner::BottomRight) .trigger( IconButton::new("supermaven-icon", icon) .tooltip(move |cx| Tooltip::text(tooltip_text.clone(), cx)), ), ); } InlineCompletionProvider::Zed => { if !cx.has_flag::<PredictEditsFeatureFlag>() { return div(); } if !self .user_store .read(cx) .current_user_has_accepted_terms() .unwrap_or(false) { let workspace = self.workspace.clone(); let user_store = self.user_store.clone(); return div().child( ButtonLike::new("zeta-pending-tos-icon") .child( IconWithIndicator::new( Icon::new(IconName::ZedPredict), Some(Indicator::dot().color(Color::Error)), ) .indicator_border_color(Some( cx.theme().colors().status_bar_background, )) .into_any_element(), ) .tooltip(|cx| { Tooltip::with_meta( "Edit Predictions", None, "Read Terms of Service", cx, ) }) .on_click(cx.listener(move |_, _, cx| { let user_store = user_store.clone(); if let Some(workspace) = workspace.upgrade() { ZedPredictTos::toggle(workspace, user_store, cx); } })), ); } let this = cx.view().clone(); let button = IconButton::new("zeta", IconName::ZedPredict) .tooltip(|cx| Tooltip::text("Edit Prediction", cx)); let is_refreshing = self .inline_completion_provider .as_ref() .map_or(false, |provider| provider.is_refreshing(cx)); let mut popover_menu = PopoverMenu::new("zeta") .menu(move |cx| { Some(this.update(cx, |this, cx| this.build_zeta_context_menu(cx))) }) .anchor(Corner::BottomRight); if is_refreshing { popover_menu = popover_menu.trigger( 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(button); } div().child(popover_menu.into_any_element()) } } }} impl InlineCompletionButton { pub fn new( workspace: WeakView<Workspace>, fs: Arc<dyn Fs>, user_store: Model<UserStore>, cx: &mut ViewContext<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, language: None, file: None, inline_completion_provider: None, workspace, fs, user_store, } } pub fn build_copilot_start_menu(&mut self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { let fs = self.fs.clone(); ContextMenu::build(cx, |menu, _| { menu.entry("Sign In", None, copilot::initiate_sign_in) .entry("Disable Copilot", None, { let fs = fs.clone(); move |cx| hide_copilot(fs.clone(), cx) }) .entry("Use Supermaven", None, { let fs = fs.clone(); move |cx| { set_completion_provider( fs.clone(), cx, InlineCompletionProvider::Supermaven, ) } }) }) } pub fn build_language_settings_menu( &self, mut menu: ContextMenu, cx: &mut WindowContext, ) -> ContextMenu { let fs = self.fs.clone(); if let Some(language) = self.language.clone() { let fs = fs.clone(); let language_enabled = language_settings::language_settings(Some(language.name()), None, cx) .show_inline_completions; menu = menu.entry( format!( "{} Inline Completions for {}", if language_enabled { "Hide" } else { "Show" }, language.name() ), None, move |cx| toggle_inline_completions_for_language(language.clone(), fs.clone(), cx), ); } let settings = AllLanguageSettings::get_global(cx); if let Some(file) = &self.file { let path = file.path().clone(); let path_enabled = settings.inline_completions_enabled_for_path(&path); menu = menu.entry( format!( "{} Inline Completions for This Path", if path_enabled { "Hide" } else { "Show" } ), None, move |cx| { if let Some(workspace) = cx.window_handle().downcast::<Workspace>() { if let Ok(workspace) = workspace.root_view(cx) { let workspace = workspace.downgrade(); cx.spawn(|cx| { configure_disabled_globs( workspace, path_enabled.then_some(path.clone()), cx, ) }) .detach_and_log_err(cx); } } }, ); } let globally_enabled = settings.inline_completions_enabled(None, None, cx); menu.entry( if globally_enabled { "Hide Inline Completions for All Files" } else { "Show Inline Completions for All Files" }, None, move |cx| toggle_inline_completions_globally(fs.clone(), cx), ) } fn build_copilot_context_menu(&self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { ContextMenu::build(cx, |menu, cx| { self.build_language_settings_menu(menu, cx) .separator() .link( "Go to Copilot Settings", OpenBrowser { url: COPILOT_SETTINGS_URL.to_string(), } .boxed_clone(), ) .action("Sign Out", copilot::SignOut.boxed_clone()) }) } fn build_supermaven_context_menu(&self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { ContextMenu::build(cx, |menu, cx| { self.build_language_settings_menu(menu, cx) .separator() .action("Sign Out", supermaven::SignOut.boxed_clone()) }) } fn build_zeta_context_menu(&self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { let workspace = self.workspace.clone(); ContextMenu::build(cx, |menu, cx| { self.build_language_settings_menu(menu, cx) .separator() .entry( "Rate Completions", Some(RateCompletions.boxed_clone()), move |cx| { workspace .update(cx, |workspace, cx| { RateCompletionModal::toggle(workspace, cx) }) .ok(); }, ) }) } pub fn update_enabled(&mut self, editor: View<Editor>, cx: &mut ViewContext<Self>) { let editor = editor.read(cx); let snapshot = editor.buffer().read(cx).snapshot(cx); let suggestion_anchor = editor.selections.newest_anchor().start; let language = snapshot.language_at(suggestion_anchor); let file = snapshot.file_at(suggestion_anchor).cloned(); self.editor_enabled = { let file = file.as_ref(); Some( file.map(|file| !file.is_private()).unwrap_or(true) && all_language_settings(file, cx).inline_completions_enabled( language, file.map(|file| file.path().as_ref()), cx, ), ) }; self.inline_completion_provider = editor.inline_completion_provider(); self.language = language.cloned(); self.file = file; cx.notify() }} impl StatusItemView for InlineCompletionButton { fn set_active_pane_item(&mut self, item: Option<&dyn ItemHandle>, cx: &mut ViewContext<Self>) { if let Some(editor) = item.and_then(|item| item.act_as::<Editor>(cx)) { self.editor_subscription = Some(( cx.observe(&editor, Self::update_enabled), editor.entity_id().as_u64() as usize, )); self.update_enabled(editor, cx); } else { self.language = None; self.editor_subscription = None; self.editor_enabled = None; } cx.notify(); }} impl SupermavenButtonStatus { fn to_icon(&self) -> IconName { match self { SupermavenButtonStatus::Ready => IconName::Supermaven, SupermavenButtonStatus::Errored(_) => IconName::SupermavenError, SupermavenButtonStatus::NeedsActivation(_) => IconName::SupermavenInit, SupermavenButtonStatus::Initializing => IconName::SupermavenInit, } } fn to_tooltip(&self) -> String { match self { SupermavenButtonStatus::Ready => "Supermaven is ready".to_string(), SupermavenButtonStatus::Errored(error) => format!("Supermaven error: {}", error), SupermavenButtonStatus::NeedsActivation(_) => "Supermaven needs activation".to_string(), SupermavenButtonStatus::Initializing => "Supermaven initializing".to_string(), } }Lines 1–500 of 597. Load more
use anyhow::Result; use client::UserStore; use copilot::{Copilot, Status}; use editor::{scroll::Autoscroll, Editor}; use feature_flags::{FeatureFlagAppExt, PredictEditsFeatureFlag}; use fs::Fs; use gpui::{ actions, div, pulsating_between, Action, Animation, AnimationExt, AppContext, AsyncWindowContext, Corner, Entity, IntoElement, Model, ParentElement, Render, Subscription, View, ViewContext, WeakView, WindowContext, }; use language::{ language_settings::{ self, all_language_settings, AllLanguageSettings, InlineCompletionProvider, }, File, Language, }; use settings::{update_settings_file, Settings, SettingsStore}; use std::{path::Path, sync::Arc, time::Duration}; use supermaven::{AccountStatus, Supermaven}; use ui::{ActiveTheme as _, ButtonLike, Color, Icon, IconWithIndicator, Indicator}; use workspace::{ create_and_open_local_file, item::ItemHandle, notifications::NotificationId, ui::{ ButtonCommon, Clickable, ContextMenu, IconButton, IconName, IconSize, PopoverMenu, Tooltip, }, StatusItemView, Toast, Workspace, }; use zed_actions::OpenBrowser; use zed_predict_tos::ZedPredictTos; use zeta::RateCompletionModal; actions!(zeta, [RateCompletions]); const COPILOT_SETTINGS_URL: &str = "https://github.com/settings/copilot"; struct CopilotErrorToast; pub struct InlineCompletionButton { editor_subscription: Option<(Subscription, usize)>, editor_enabled: Option<bool>, language: Option<Arc<Language>>, file: Option<Arc<dyn File>>, inline_completion_provider: Option<Arc<dyn inline_completion::InlineCompletionProviderHandle>>, fs: Arc<dyn Fs>, workspace: WeakView<Workspace>, user_store: Model<UserStore>, } enum SupermavenButtonStatus { Ready, Errored(String), NeedsActivation(String), Initializing, } impl Render for InlineCompletionButton { fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement { let all_language_settings = all_language_settings(None, cx); match all_language_settings.inline_completions.provider { InlineCompletionProvider::None => div(), InlineCompletionProvider::Copilot => { let Some(copilot) = Copilot::global(cx) else { return div(); }; let status = copilot.read(cx).status(); let enabled = self.editor_enabled.unwrap_or_else(|| { all_language_settings.inline_completions_enabled(None, None, cx) }); 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 |_, _, cx| { if let Some(workspace) = cx.window_handle().downcast::<Workspace>() { 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, ); }) .ok(); } })) .tooltip(|cx| Tooltip::text("GitHub Copilot", cx)), ); } let this = cx.view().clone(); div().child( PopoverMenu::new("copilot") .menu(move |cx| { Some(match status { Status::Authorized => { this.update(cx, |this, cx| this.build_copilot_context_menu(cx)) } _ => this.update(cx, |this, cx| this.build_copilot_start_menu(cx)), }) }) .anchor(Corner::BottomRight) .trigger( IconButton::new("copilot-icon", icon) .tooltip(|cx| Tooltip::text("GitHub Copilot", cx)), ), ) } InlineCompletionProvider::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 this = cx.view().clone(); let fs = self.fs.clone(); return div().child( PopoverMenu::new("supermaven") .menu(move |cx| match &status { SupermavenButtonStatus::NeedsActivation(activate_url) => { Some(ContextMenu::build(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, InlineCompletionProvider::Copilot, ) }, ) })) } SupermavenButtonStatus::Ready => Some( this.update(cx, |this, cx| this.build_supermaven_context_menu(cx)), ), _ => None, }) .anchor(Corner::BottomRight) .trigger( IconButton::new("supermaven-icon", icon) .tooltip(move |cx| Tooltip::text(tooltip_text.clone(), cx)), ), ); } InlineCompletionProvider::Zed => { if !cx.has_flag::<PredictEditsFeatureFlag>() { return div(); } if !self .user_store .read(cx) .current_user_has_accepted_terms() .unwrap_or(false) { let workspace = self.workspace.clone(); let user_store = self.user_store.clone(); return div().child( ButtonLike::new("zeta-pending-tos-icon") .child( IconWithIndicator::new( Icon::new(IconName::ZedPredict), Some(Indicator::dot().color(Color::Error)), ) .indicator_border_color(Some( cx.theme().colors().status_bar_background, )) .into_any_element(), ) .tooltip(|cx| { Tooltip::with_meta( "Edit Predictions", None, "Read Terms of Service", cx, ) }) .on_click(cx.listener(move |_, _, cx| { let user_store = user_store.clone(); if let Some(workspace) = workspace.upgrade() { ZedPredictTos::toggle(workspace, user_store, cx); } })), ); } let this = cx.view().clone(); let button = IconButton::new("zeta", IconName::ZedPredict) .tooltip(|cx| Tooltip::text("Edit Prediction", cx)); let is_refreshing = self .inline_completion_provider .as_ref() .map_or(false, |provider| provider.is_refreshing(cx)); let mut popover_menu = PopoverMenu::new("zeta") .menu(move |cx| { Some(this.update(cx, |this, cx| this.build_zeta_context_menu(cx))) }) .anchor(Corner::BottomRight); if is_refreshing { popover_menu = popover_menu.trigger( 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(button); } div().child(popover_menu.into_any_element()) } } } } impl InlineCompletionButton { pub fn new( workspace: WeakView<Workspace>, fs: Arc<dyn Fs>, user_store: Model<UserStore>, cx: &mut ViewContext<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, language: None, file: None, inline_completion_provider: None, workspace, fs, user_store, } } pub fn build_copilot_start_menu(&mut self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { let fs = self.fs.clone(); ContextMenu::build(cx, |menu, _| { menu.entry("Sign In", None, copilot::initiate_sign_in) .entry("Disable Copilot", None, { let fs = fs.clone(); move |cx| hide_copilot(fs.clone(), cx) }) .entry("Use Supermaven", None, { let fs = fs.clone(); move |cx| { set_completion_provider( fs.clone(), cx, InlineCompletionProvider::Supermaven, ) } }) }) } pub fn build_language_settings_menu( &self, mut menu: ContextMenu, cx: &mut WindowContext, ) -> ContextMenu { let fs = self.fs.clone(); if let Some(language) = self.language.clone() { let fs = fs.clone(); let language_enabled = language_settings::language_settings(Some(language.name()), None, cx) .show_inline_completions; menu = menu.entry( format!( "{} Inline Completions for {}", if language_enabled { "Hide" } else { "Show" }, language.name() ), None, move |cx| toggle_inline_completions_for_language(language.clone(), fs.clone(), cx), ); } let settings = AllLanguageSettings::get_global(cx); if let Some(file) = &self.file { let path = file.path().clone(); let path_enabled = settings.inline_completions_enabled_for_path(&path); menu = menu.entry( format!( "{} Inline Completions for This Path", if path_enabled { "Hide" } else { "Show" } ), None, move |cx| { if let Some(workspace) = cx.window_handle().downcast::<Workspace>() { if let Ok(workspace) = workspace.root_view(cx) { let workspace = workspace.downgrade(); cx.spawn(|cx| { configure_disabled_globs( workspace, path_enabled.then_some(path.clone()), cx, ) }) .detach_and_log_err(cx); } } }, ); } let globally_enabled = settings.inline_completions_enabled(None, None, cx); menu.entry( if globally_enabled { "Hide Inline Completions for All Files" } else { "Show Inline Completions for All Files" }, None, move |cx| toggle_inline_completions_globally(fs.clone(), cx), ) } fn build_copilot_context_menu(&self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { ContextMenu::build(cx, |menu, cx| { self.build_language_settings_menu(menu, cx) .separator() .link( "Go to Copilot Settings", OpenBrowser { url: COPILOT_SETTINGS_URL.to_string(), } .boxed_clone(), ) .action("Sign Out", copilot::SignOut.boxed_clone()) }) } fn build_supermaven_context_menu(&self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { ContextMenu::build(cx, |menu, cx| { self.build_language_settings_menu(menu, cx) .separator() .action("Sign Out", supermaven::SignOut.boxed_clone()) }) } fn build_zeta_context_menu(&self, cx: &mut ViewContext<Self>) -> View<ContextMenu> { let workspace = self.workspace.clone(); ContextMenu::build(cx, |menu, cx| { self.build_language_settings_menu(menu, cx) .separator() .entry( "Rate Completions", Some(RateCompletions.boxed_clone()), move |cx| { workspace .update(cx, |workspace, cx| { RateCompletionModal::toggle(workspace, cx) }) .ok(); }, ) }) } pub fn update_enabled(&mut self, editor: View<Editor>, cx: &mut ViewContext<Self>) { let editor = editor.read(cx); let snapshot = editor.buffer().read(cx).snapshot(cx); let suggestion_anchor = editor.selections.newest_anchor().start; let language = snapshot.language_at(suggestion_anchor); let file = snapshot.file_at(suggestion_anchor).cloned(); self.editor_enabled = { let file = file.as_ref(); Some( file.map(|file| !file.is_private()).unwrap_or(true) && all_language_settings(file, cx).inline_completions_enabled( language, file.map(|file| file.path().as_ref()), cx, ), ) }; self.inline_completion_provider = editor.inline_completion_provider(); self.language = language.cloned(); self.file = file; cx.notify() } } impl StatusItemView for InlineCompletionButton { fn set_active_pane_item(&mut self, item: Option<&dyn ItemHandle>, cx: &mut ViewContext<Self>) { if let Some(editor) = item.and_then(|item| item.act_as::<Editor>(cx)) { self.editor_subscription = Some(( cx.observe(&editor, Self::update_enabled), editor.entity_id().as_u64() as usize, )); self.update_enabled(editor, cx); } else { self.language = None; self.editor_subscription = None; self.editor_enabled = None; } cx.notify(); } } impl SupermavenButtonStatus { fn to_icon(&self) -> IconName { match self { SupermavenButtonStatus::Ready => IconName::Supermaven, SupermavenButtonStatus::Errored(_) => IconName::SupermavenError, SupermavenButtonStatus::NeedsActivation(_) => IconName::SupermavenInit, SupermavenButtonStatus::Initializing => IconName::SupermavenInit, } } fn to_tooltip(&self) -> String { match self { SupermavenButtonStatus::Ready => "Supermaven is ready".to_string(), SupermavenButtonStatus::Errored(error) => format!("Supermaven error: {}", error), SupermavenButtonStatus::NeedsActivation(_) => "Supermaven needs activation".to_string(), SupermavenButtonStatus::Initializing => "Supermaven initializing".to_string(), } }