917 lines · 37.5 KB · Rust
RawDownload
use anyhow::Result;use client::{UserStore, zed_urls};use copilot::{Copilot, Status};use editor::{    Editor,    actions::{ShowEditPrediction, ToggleEditPrediction},    scroll::Autoscroll,};use feature_flags::{FeatureFlagAppExt, PredictEditsRateCompletionsFeatureFlag};use fs::Fs;use gpui::{    Action, Animation, AnimationExt, App, AsyncWindowContext, Corner, Entity, FocusHandle,    Focusable, IntoElement, ParentElement, Render, Subscription, WeakEntity, actions, div,    pulsating_between,};use indoc::indoc;use language::{    EditPredictionsMode, File, Language,    language_settings::{self, AllLanguageSettings, EditPredictionProvider, all_language_settings},};use regex::Regex;use settings::{Settings, SettingsStore, update_settings_file};use std::{    sync::{Arc, LazyLock},    time::Duration,};use supermaven::{AccountStatus, Supermaven};use ui::{    Clickable, ContextMenu, ContextMenuEntry, DocumentationSide, IconButton, IconButtonShape,    Indicator, PopoverMenu, PopoverMenuHandle, ProgressBar, Tooltip, prelude::*,};use workspace::{    StatusItemView, Toast, Workspace, create_and_open_local_file, item::ItemHandle,    notifications::NotificationId,};use zed_actions::OpenBrowser;use zed_llm_client::UsageLimit;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",                                                |window, cx| {                                                    copilot::reinstall_and_sign_in(window, cx)                                                },                                            ),                                            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 => {                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();                let has_subscription = self.user_store.read(cx).current_plan().is_some()                    && self.user_store.read(cx).subscription_period().is_some();                 if !has_subscription || !current_user_terms_accepted.unwrap_or(false) {                    let signed_in = current_user_terms_accepted.is_some();                    let tooltip_meta = if signed_in {                        if has_subscription {                            "Read Terms of Service"                        } else {                            "Choose a Plan"                        }                    } else {                        "Sign in to use"                    };                     return div().child(                        IconButton::new("zed-predict-pending-button", zeta_icon)                            .shape(IconButtonShape::Square)                            .indicator(Indicator::dot().color(Color::Muted))                            .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();         if let Some(usage) = self            .edit_prediction_provider            .as_ref()            .and_then(|provider| provider.usage(cx))        {            menu = menu.header("Usage");            menu = menu                .custom_entry(                    move |_window, cx| {                        let used_percentage = match usage.limit {                            UsageLimit::Limited(limit) => {                                Some((usage.amount as f32 / limit as f32) * 100.)                            }                            UsageLimit::Unlimited => None,                        };                         h_flex()                            .flex_1()                            .gap_1p5()                            .children(                                used_percentage                                    .map(|percent| ProgressBar::new("usage", percent, 100., cx)),                            )                            .child(                                Label::new(match usage.limit {                                    UsageLimit::Limited(limit) => {                                        format!("{} / {limit}", usage.amount)                                    }                                    UsageLimit::Unlimited => format!("{} / ∞", usage.amount),                                })                                .size(LabelSize::Small)                                .color(Color::Muted),                            )                            .into_any_element()                    },                    move |_, cx| cx.open_url(&zed_urls::account_url(cx)),                )                .separator();        }         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(DocumentationSide::Left, 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)        }); 

Lines 1–500 of 917.