1use anyhow::Result;
2use client::UserStore;
3use copilot::{Copilot, Status};
4use editor::{
5 actions::{ShowEditPrediction, ToggleEditPrediction},
6 scroll::Autoscroll,
7 Editor,
8};
9use feature_flags::{
10 FeatureFlagAppExt, PredictEditsFeatureFlag, PredictEditsRateCompletionsFeatureFlag,
11};
12use fs::Fs;
13use gpui::{
14 actions, div, pulsating_between, Action, Animation, AnimationExt, App, AsyncWindowContext,
15 Corner, Entity, FocusHandle, Focusable, IntoElement, ParentElement, Render, Subscription,
16 WeakEntity,
17};
18use indoc::indoc;
19use language::{
20 language_settings::{self, all_language_settings, AllLanguageSettings, EditPredictionProvider},
21 File, Language,
22};
23use regex::Regex;
24use settings::{update_settings_file, Settings, SettingsStore};
25use std::{
26 sync::{Arc, LazyLock},
27 time::Duration,
28};
29use supermaven::{AccountStatus, Supermaven};
30use ui::{
31 prelude::*, Clickable, ContextMenu, ContextMenuEntry, IconButton, IconButtonShape, Indicator,
32 PopoverMenu, PopoverMenuHandle, Tooltip,
33};
34use workspace::{
35 create_and_open_local_file, item::ItemHandle, notifications::NotificationId, StatusItemView,
36 Toast, Workspace,
37};
38use zed_actions::OpenBrowser;
39use zeta::RateCompletions;
40
41actions!(edit_prediction, [ToggleMenu]);
42
43const COPILOT_SETTINGS_URL: &str = "https://github.com/settings/copilot";
44
45struct CopilotErrorToast;
46
47pub struct InlineCompletionButton {
48 editor_subscription: Option<(Subscription, usize)>,
49 editor_enabled: Option<bool>,
50 editor_show_predictions: bool,
51 editor_focus_handle: Option<FocusHandle>,
52 language: Option<Arc<Language>>,
53 file: Option<Arc<dyn File>>,
54 edit_prediction_provider: Option<Arc<dyn inline_completion::InlineCompletionProviderHandle>>,
55 fs: Arc<dyn Fs>,
56 user_store: Entity<UserStore>,
57 popover_menu_handle: PopoverMenuHandle<ContextMenu>,
58}
59
60enum SupermavenButtonStatus {
61 Ready,
62 Errored(String),
63 NeedsActivation(String),
64 Initializing,
65}
66
67impl Render for InlineCompletionButton {
68 fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
69 let all_language_settings = all_language_settings(None, cx);
70
71 match all_language_settings.edit_predictions.provider {
72 EditPredictionProvider::None => div(),
73
74 EditPredictionProvider::Copilot => {
75 let Some(copilot) = Copilot::global(cx) else {
76 return div();
77 };
78 let status = copilot.read(cx).status();
79
80 let enabled = self.editor_enabled.unwrap_or(false);
81
82 let icon = match status {
83 Status::Error(_) => IconName::CopilotError,
84 Status::Authorized => {
85 if enabled {
86 IconName::Copilot
87 } else {
88 IconName::CopilotDisabled
89 }
90 }
91 _ => IconName::CopilotInit,
92 };
93
94 if let Status::Error(e) = status {
95 return div().child(
96 IconButton::new("copilot-error", icon)
97 .icon_size(IconSize::Small)
98 .on_click(cx.listener(move |_, _, window, cx| {
99 if let Some(workspace) = window.root::<Workspace>().flatten() {
100 workspace.update(cx, |workspace, cx| {
101 workspace.show_toast(
102 Toast::new(
103 NotificationId::unique::<CopilotErrorToast>(),
104 format!("Copilot can't be started: {}", e),
105 )
106 .on_click(
107 "Reinstall Copilot",
108 |_, cx| {
109 if let Some(copilot) = Copilot::global(cx) {
110 copilot
111 .update(cx, |copilot, cx| {
112 copilot.reinstall(cx)
113 })
114 .detach();
115 }
116 },
117 ),
118 cx,
119 );
120 });
121 }
122 }))
123 .tooltip(|window, cx| {
124 Tooltip::for_action("GitHub Copilot", &ToggleMenu, window, cx)
125 }),
126 );
127 }
128 let this = cx.entity().clone();
129
130 div().child(
131 PopoverMenu::new("copilot")
132 .menu(move |window, cx| {
133 Some(match status {
134 Status::Authorized => this.update(cx, |this, cx| {
135 this.build_copilot_context_menu(window, cx)
136 }),
137 _ => this.update(cx, |this, cx| {
138 this.build_copilot_start_menu(window, cx)
139 }),
140 })
141 })
142 .anchor(Corner::BottomRight)
143 .trigger_with_tooltip(
144 IconButton::new("copilot-icon", icon),
145 |window, cx| {
146 Tooltip::for_action("GitHub Copilot", &ToggleMenu, window, cx)
147 },
148 )
149 .with_handle(self.popover_menu_handle.clone()),
150 )
151 }
152
153 EditPredictionProvider::Supermaven => {
154 let Some(supermaven) = Supermaven::global(cx) else {
155 return div();
156 };
157
158 let supermaven = supermaven.read(cx);
159
160 let status = match supermaven {
161 Supermaven::Starting => SupermavenButtonStatus::Initializing,
162 Supermaven::FailedDownload { error } => {
163 SupermavenButtonStatus::Errored(error.to_string())
164 }
165 Supermaven::Spawned(agent) => {
166 let account_status = agent.account_status.clone();
167 match account_status {
168 AccountStatus::NeedsActivation { activate_url } => {
169 SupermavenButtonStatus::NeedsActivation(activate_url.clone())
170 }
171 AccountStatus::Unknown => SupermavenButtonStatus::Initializing,
172 AccountStatus::Ready => SupermavenButtonStatus::Ready,
173 }
174 }
175 Supermaven::Error { error } => {
176 SupermavenButtonStatus::Errored(error.to_string())
177 }
178 };
179
180 let icon = status.to_icon();
181 let tooltip_text = status.to_tooltip();
182 let has_menu = status.has_menu();
183 let this = cx.entity().clone();
184 let fs = self.fs.clone();
185
186 return div().child(
187 PopoverMenu::new("supermaven")
188 .menu(move |window, cx| match &status {
189 SupermavenButtonStatus::NeedsActivation(activate_url) => {
190 Some(ContextMenu::build(window, cx, |menu, _, _| {
191 let fs = fs.clone();
192 let activate_url = activate_url.clone();
193 menu.entry("Sign In", None, move |_, cx| {
194 cx.open_url(activate_url.as_str())
195 })
196 .entry(
197 "Use Copilot",
198 None,
199 move |_, cx| {
200 set_completion_provider(
201 fs.clone(),
202 cx,
203 EditPredictionProvider::Copilot,
204 )
205 },
206 )
207 }))
208 }
209 SupermavenButtonStatus::Ready => Some(this.update(cx, |this, cx| {
210 this.build_supermaven_context_menu(window, cx)
211 })),
212 _ => None,
213 })
214 .anchor(Corner::BottomRight)
215 .trigger_with_tooltip(
216 IconButton::new("supermaven-icon", icon),
217 move |window, cx| {
218 if has_menu {
219 Tooltip::for_action(
220 tooltip_text.clone(),
221 &ToggleMenu,
222 window,
223 cx,
224 )
225 } else {
226 Tooltip::text(tooltip_text.clone())(window, cx)
227 }
228 },
229 )
230 .with_handle(self.popover_menu_handle.clone()),
231 );
232 }
233
234 EditPredictionProvider::Zed => {
235 if !cx.has_flag::<PredictEditsFeatureFlag>() {
236 return div();
237 }
238
239 let enabled = self.editor_enabled.unwrap_or(true);
240
241 let zeta_icon = if enabled {
242 IconName::ZedPredict
243 } else {
244 IconName::ZedPredictDisabled
245 };
246
247 let current_user_terms_accepted =
248 self.user_store.read(cx).current_user_has_accepted_terms();
249
250 if !current_user_terms_accepted.unwrap_or(false) {
251 let signed_in = current_user_terms_accepted.is_some();
252 let tooltip_meta = if signed_in {
253 "Read Terms of Service"
254 } else {
255 "Sign in to use"
256 };
257
258 return div().child(
259 IconButton::new("zed-predict-pending-button", zeta_icon)
260 .shape(IconButtonShape::Square)
261 .indicator(Indicator::dot().color(Color::Error))
262 .indicator_border_color(Some(cx.theme().colors().status_bar_background))
263 .tooltip(move |window, cx| {
264 Tooltip::with_meta(
265 "Edit Predictions",
266 None,
267 tooltip_meta,
268 window,
269 cx,
270 )
271 })
272 .on_click(cx.listener(move |_, _, window, cx| {
273 telemetry::event!(
274 "Pending ToS Clicked",
275 source = "Edit Prediction Status Button"
276 );
277 window.dispatch_action(
278 zed_actions::OpenZedPredictOnboarding.boxed_clone(),
279 cx,
280 );
281 })),
282 );
283 }
284
285 let show_editor_predictions = self.editor_show_predictions;
286
287 let icon_button = IconButton::new("zed-predict-pending-button", zeta_icon)
288 .shape(IconButtonShape::Square)
289 .when(enabled && !show_editor_predictions, |this| {
290 this.indicator(Indicator::dot().color(Color::Muted))
291 .indicator_border_color(Some(cx.theme().colors().status_bar_background))
292 })
293 .when(!self.popover_menu_handle.is_deployed(), |element| {
294 element.tooltip(move |window, cx| {
295 if enabled {
296 if show_editor_predictions {
297 Tooltip::for_action("Edit Prediction", &ToggleMenu, window, cx)
298 } else {
299 Tooltip::with_meta(
300 "Edit Prediction",
301 Some(&ToggleMenu),
302 "Hidden For This File",
303 window,
304 cx,
305 )
306 }
307 } else {
308 Tooltip::with_meta(
309 "Edit Prediction",
310 Some(&ToggleMenu),
311 "Disabled For This File",
312 window,
313 cx,
314 )
315 }
316 })
317 });
318
319 let this = cx.entity().clone();
320
321 let mut popover_menu = PopoverMenu::new("zeta")
322 .menu(move |window, cx| {
323 Some(this.update(cx, |this, cx| this.build_zeta_context_menu(window, cx)))
324 })
325 .anchor(Corner::BottomRight)
326 .with_handle(self.popover_menu_handle.clone());
327
328 let is_refreshing = self
329 .edit_prediction_provider
330 .as_ref()
331 .map_or(false, |provider| provider.is_refreshing(cx));
332
333 if is_refreshing {
334 popover_menu = popover_menu.trigger(
335 icon_button.with_animation(
336 "pulsating-label",
337 Animation::new(Duration::from_secs(2))
338 .repeat()
339 .with_easing(pulsating_between(0.2, 1.0)),
340 |icon_button, delta| icon_button.alpha(delta),
341 ),
342 );
343 } else {
344 popover_menu = popover_menu.trigger(icon_button);
345 }
346
347 div().child(popover_menu.into_any_element())
348 }
349 }
350 }
351}
352
353impl InlineCompletionButton {
354 pub fn new(
355 fs: Arc<dyn Fs>,
356 user_store: Entity<UserStore>,
357 popover_menu_handle: PopoverMenuHandle<ContextMenu>,
358 cx: &mut Context<Self>,
359 ) -> Self {
360 if let Some(copilot) = Copilot::global(cx) {
361 cx.observe(&copilot, |_, _, cx| cx.notify()).detach()
362 }
363
364 cx.observe_global::<SettingsStore>(move |_, cx| cx.notify())
365 .detach();
366
367 Self {
368 editor_subscription: None,
369 editor_enabled: None,
370 editor_show_predictions: true,
371 editor_focus_handle: None,
372 language: None,
373 file: None,
374 edit_prediction_provider: None,
375 popover_menu_handle,
376 fs,
377 user_store,
378 }
379 }
380
381 pub fn build_copilot_start_menu(
382 &mut self,
383 window: &mut Window,
384 cx: &mut Context<Self>,
385 ) -> Entity<ContextMenu> {
386 let fs = self.fs.clone();
387 ContextMenu::build(window, cx, |menu, _, _| {
388 menu.entry("Sign In", None, copilot::initiate_sign_in)
389 .entry("Disable Copilot", None, {
390 let fs = fs.clone();
391 move |_window, cx| hide_copilot(fs.clone(), cx)
392 })
393 .entry("Use Supermaven", None, {
394 let fs = fs.clone();
395 move |_window, cx| {
396 set_completion_provider(fs.clone(), cx, EditPredictionProvider::Supermaven)
397 }
398 })
399 })
400 }
401
402 pub fn build_language_settings_menu(
403 &self,
404 mut menu: ContextMenu,
405 window: &Window,
406 cx: &mut App,
407 ) -> ContextMenu {
408 let fs = self.fs.clone();
409 let line_height = window.line_height();
410
411 menu = menu.header("Show Edit Predictions For");
412
413 if let Some(editor_focus_handle) = self.editor_focus_handle.clone() {
414 menu = menu.toggleable_entry(
415 "This Buffer",
416 self.editor_show_predictions,
417 IconPosition::Start,
418 Some(Box::new(ToggleEditPrediction)),
419 {
420 let editor_focus_handle = editor_focus_handle.clone();
421 move |window, cx| {
422 editor_focus_handle.dispatch_action(&ToggleEditPrediction, window, cx);
423 }
424 },
425 );
426 }
427
428 if let Some(language) = self.language.clone() {
429 let fs = fs.clone();
430 let language_enabled =
431 language_settings::language_settings(Some(language.name()), None, cx)
432 .show_edit_predictions;
433
434 menu = menu.toggleable_entry(
435 language.name(),
436 language_enabled,
437 IconPosition::Start,
438 None,
439 move |_, cx| {
440 toggle_show_inline_completions_for_language(language.clone(), fs.clone(), cx)
441 },
442 );
443 }
444
445 let settings = AllLanguageSettings::get_global(cx);
446 let globally_enabled = settings.show_inline_completions(None, cx);
447 menu = menu.toggleable_entry("All Files", globally_enabled, IconPosition::Start, None, {
448 let fs = fs.clone();
449 move |_, cx| toggle_inline_completions_globally(fs.clone(), cx)
450 });
451 menu = menu.separator().header("Privacy Settings");
452
453 if let Some(provider) = &self.edit_prediction_provider {
454 let data_collection = provider.data_collection_state(cx);
455 if data_collection.is_supported() {
456 let provider = provider.clone();
457 let enabled = data_collection.is_enabled();
458 let is_open_source = data_collection.is_project_open_source();
459 let is_collecting = data_collection.is_enabled();
460 let (icon_name, icon_color) = if is_open_source && is_collecting {
461 (IconName::Check, Color::Success)
462 } else {
463 (IconName::Check, Color::Accent)
464 };
465
466 menu = menu.item(
467 ContextMenuEntry::new("Training Data Collection")
468 .toggleable(IconPosition::Start, data_collection.is_enabled())
469 .icon(icon_name)
470 .icon_color(icon_color)
471 .documentation_aside(move |cx| {
472 let (msg, label_color, icon_name, icon_color) = match (is_open_source, is_collecting) {
473 (true, true) => (
474 "Project identified as open source, and you're sharing data.",
475 Color::Default,
476 IconName::Check,
477 Color::Success,
478 ),
479 (true, false) => (
480 "Project identified as open source, but you're not sharing data.",
481 Color::Muted,
482 IconName::Close,
483 Color::Muted,
484 ),
485 (false, true) => (
486 "Project not identified as open source. No data captured.",
487 Color::Muted,
488 IconName::Close,
489 Color::Muted,
490 ),
491 (false, false) => (
492 "Project not identified as open source, and setting turned off.",
493 Color::Muted,
494 IconName::Close,
495 Color::Muted,
496 ),
497 };
498 v_flex()
499 .gap_2()
500 .child(
501 Label::new(indoc!{
502 "Help us improve our open dataset model by sharing data from open source repositories. \
503 Zed must detect a license file in your repo for this setting to take effect."
504 })
505 )
506 .child(
507 h_flex()
508 .items_start()
509 .pt_2()
510 .flex_1()
511 .gap_1p5()
512 .border_t_1()
513 .border_color(cx.theme().colors().border_variant)
514 .child(h_flex().flex_shrink_0().h(line_height).child(Icon::new(icon_name).size(IconSize::XSmall).color(icon_color)))
515 .child(div().child(msg).w_full().text_sm().text_color(label_color.color(cx)))
516 )
517 .into_any_element()
518 })
519 .handler(move |_, cx| {
520 provider.toggle_data_collection(cx);
521
522 if !enabled {
523 telemetry::event!(
524 "Data Collection Enabled",
525 source = "Edit Prediction Status Menu"
526 );
527 } else {
528 telemetry::event!(
529 "Data Collection Disabled",
530 source = "Edit Prediction Status Menu"
531 );
532 }
533 })
534 );
535
536 if is_collecting && !is_open_source {
537 menu = menu.item(
538 ContextMenuEntry::new("No data captured.")
539 .disabled(true)
540 .icon(IconName::Close)
541 .icon_color(Color::Error)
542 .icon_size(IconSize::Small),
543 );
544 }
545 }
546 }
547
548 menu = menu.item(
549 ContextMenuEntry::new("Configure Excluded Files")
550 .icon(IconName::LockOutlined)
551 .icon_color(Color::Muted)
552 .documentation_aside(|_| {
553 Label::new(indoc!{"
554 Open your settings to add sensitive paths for which Zed will never predict edits."}).into_any_element()
555 })
556 .handler(move |window, cx| {
557 if let Some(workspace) = window.root().flatten() {
558 let workspace = workspace.downgrade();
559 window
560 .spawn(cx, |cx| {
561 open_disabled_globs_setting_in_editor(
562 workspace,
563 cx,
564 )
565 })
566 .detach_and_log_err(cx);
567 }
568 }),
569 );
570
571 if !self.editor_enabled.unwrap_or(true) {
572 menu = menu.item(
573 ContextMenuEntry::new("This file is excluded.")
574 .disabled(true)
575 .icon(IconName::ZedPredictDisabled)
576 .icon_size(IconSize::Small),
577 );
578 }
579
580 if cx.has_flag::<feature_flags::PredictEditsNonEagerModeFeatureFlag>() {
581 let is_eager_preview_enabled = match settings.edit_predictions_mode() {
582 language::EditPredictionsMode::Auto => false,
583 language::EditPredictionsMode::EagerPreview => true,
584 };
585 menu = menu.separator().toggleable_entry(
586 "Eager Preview Mode",
587 is_eager_preview_enabled,
588 IconPosition::Start,
589 None,
590 {
591 let fs = fs.clone();
592 move |_window, cx| {
593 update_settings_file::<AllLanguageSettings>(
594 fs.clone(),
595 cx,
596 move |settings, _cx| {
597 let new_mode = match is_eager_preview_enabled {
598 true => language::EditPredictionsMode::Auto,
599 false => language::EditPredictionsMode::EagerPreview,
600 };
601
602 if let Some(edit_predictions) = settings.edit_predictions.as_mut() {
603 edit_predictions.mode = new_mode;
604 } else {
605 settings.edit_predictions =
606 Some(language_settings::EditPredictionSettingsContent {
607 mode: new_mode,
608 ..Default::default()
609 });
610 }
611 },
612 );
613 }
614 },
615 );
616 }
617
618 if let Some(editor_focus_handle) = self.editor_focus_handle.clone() {
619 menu = menu
620 .separator()
621 .entry(
622 "Predict Edit at Cursor",
623 Some(Box::new(ShowEditPrediction)),
624 {
625 let editor_focus_handle = editor_focus_handle.clone();
626 move |window, cx| {
627 editor_focus_handle.dispatch_action(&ShowEditPrediction, window, cx);
628 }
629 },
630 )
631 .context(editor_focus_handle);
632 }
633
634 menu
635 }
636
637 fn build_copilot_context_menu(
638 &self,
639 window: &mut Window,
640 cx: &mut Context<Self>,
641 ) -> Entity<ContextMenu> {
642 ContextMenu::build(window, cx, |menu, window, cx| {
643 self.build_language_settings_menu(menu, window, cx)
644 .separator()
645 .link(
646 "Go to Copilot Settings",
647 OpenBrowser {
648 url: COPILOT_SETTINGS_URL.to_string(),
649 }
650 .boxed_clone(),
651 )
652 .action("Sign Out", copilot::SignOut.boxed_clone())
653 })
654 }
655
656 fn build_supermaven_context_menu(
657 &self,
658 window: &mut Window,
659 cx: &mut Context<Self>,
660 ) -> Entity<ContextMenu> {
661 ContextMenu::build(window, cx, |menu, window, cx| {
662 self.build_language_settings_menu(menu, window, cx)
663 .separator()
664 .action("Sign Out", supermaven::SignOut.boxed_clone())
665 })
666 }
667
668 fn build_zeta_context_menu(
669 &self,
670 window: &mut Window,
671 cx: &mut Context<Self>,
672 ) -> Entity<ContextMenu> {
673 ContextMenu::build(window, cx, |menu, window, cx| {
674 self.build_language_settings_menu(menu, window, cx).when(
675 cx.has_flag::<PredictEditsRateCompletionsFeatureFlag>(),
676 |this| this.action("Rate Completions", RateCompletions.boxed_clone()),
677 )
678 })
679 }
680
681 pub fn update_enabled(&mut self, editor: Entity<Editor>, cx: &mut Context<Self>) {
682 let editor = editor.read(cx);
683 let snapshot = editor.buffer().read(cx).snapshot(cx);
684 let suggestion_anchor = editor.selections.newest_anchor().start;
685 let language = snapshot.language_at(suggestion_anchor);
686 let file = snapshot.file_at(suggestion_anchor).cloned();
687 self.editor_enabled = {
688 let file = file.as_ref();
689 Some(
690 file.map(|file| {
691 all_language_settings(Some(file), cx)
692 .inline_completions_enabled_for_path(file.path())
693 })
694 .unwrap_or(true),
695 )
696 };
697 self.editor_show_predictions = editor.edit_predictions_enabled();
698 self.edit_prediction_provider = editor.edit_prediction_provider();
699 self.language = language.cloned();
700 self.file = file;
701 self.editor_focus_handle = Some(editor.focus_handle(cx));
702
703 cx.notify();
704 }
705
706 pub fn toggle_menu(&mut self, window: &mut Window, cx: &mut Context<Self>) {
707 self.popover_menu_handle.toggle(window, cx);
708 }
709}
710
711impl StatusItemView for InlineCompletionButton {
712 fn set_active_pane_item(
713 &mut self,
714 item: Option<&dyn ItemHandle>,
715 _: &mut Window,
716 cx: &mut Context<Self>,
717 ) {
718 if let Some(editor) = item.and_then(|item| item.act_as::<Editor>(cx)) {
719 self.editor_subscription = Some((
720 cx.observe(&editor, Self::update_enabled),
721 editor.entity_id().as_u64() as usize,
722 ));
723 self.update_enabled(editor, cx);
724 } else {
725 self.language = None;
726 self.editor_subscription = None;
727 self.editor_enabled = None;
728 }
729 cx.notify();
730 }
731}
732
733impl SupermavenButtonStatus {
734 fn to_icon(&self) -> IconName {
735 match self {
736 SupermavenButtonStatus::Ready => IconName::Supermaven,
737 SupermavenButtonStatus::Errored(_) => IconName::SupermavenError,
738 SupermavenButtonStatus::NeedsActivation(_) => IconName::SupermavenInit,
739 SupermavenButtonStatus::Initializing => IconName::SupermavenInit,
740 }
741 }
742
743 fn to_tooltip(&self) -> String {
744 match self {
745 SupermavenButtonStatus::Ready => "Supermaven is ready".to_string(),
746 SupermavenButtonStatus::Errored(error) => format!("Supermaven error: {}", error),
747 SupermavenButtonStatus::NeedsActivation(_) => "Supermaven needs activation".to_string(),
748 SupermavenButtonStatus::Initializing => "Supermaven initializing".to_string(),
749 }
750 }
751
752 fn has_menu(&self) -> bool {
753 match self {
754 SupermavenButtonStatus::Ready | SupermavenButtonStatus::NeedsActivation(_) => true,
755 SupermavenButtonStatus::Errored(_) | SupermavenButtonStatus::Initializing => false,
756 }
757 }
758}
759
760async fn open_disabled_globs_setting_in_editor(
761 workspace: WeakEntity<Workspace>,
762 mut cx: AsyncWindowContext,
763) -> Result<()> {
764 let settings_editor = workspace
765 .update_in(&mut cx, |_, window, cx| {
766 create_and_open_local_file(paths::settings_file(), window, cx, || {
767 settings::initial_user_settings_content().as_ref().into()
768 })
769 })?
770 .await?
771 .downcast::<Editor>()
772 .unwrap();
773
774 settings_editor
775 .downgrade()
776 .update_in(&mut cx, |item, window, cx| {
777 let text = item.buffer().read(cx).snapshot(cx).text();
778
779 let settings = cx.global::<SettingsStore>();
780
781 // Ensure that we always have "inline_completions { "disabled_globs": [] }"
782 let edits = settings.edits_for_update::<AllLanguageSettings>(&text, |file| {
783 file.edit_predictions
784 .get_or_insert_with(Default::default)
785 .disabled_globs
786 .get_or_insert_with(Vec::new);
787 });
788
789 if !edits.is_empty() {
790 item.edit(edits.iter().cloned(), cx);
791 }
792
793 let text = item.buffer().read(cx).snapshot(cx).text();
794
795 static DISABLED_GLOBS_REGEX: LazyLock<Regex> = LazyLock::new(|| {
796 Regex::new(r#""disabled_globs":\s*\[\s*(?P<content>(?:.|\n)*?)\s*\]"#).unwrap()
797 });
798 // Only capture [...]
799 let range = DISABLED_GLOBS_REGEX.captures(&text).and_then(|captures| {
800 captures
801 .name("content")
802 .map(|inner_match| inner_match.start()..inner_match.end())
803 });
804 if let Some(range) = range {
805 item.change_selections(Some(Autoscroll::newest()), window, cx, |selections| {
806 selections.select_ranges(vec![range]);
807 });
808 }
809 })?;
810
811 anyhow::Ok(())
812}
813
814fn toggle_inline_completions_globally(fs: Arc<dyn Fs>, cx: &mut App) {
815 let show_edit_predictions = all_language_settings(None, cx).show_inline_completions(None, cx);
816 update_settings_file::<AllLanguageSettings>(fs, cx, move |file, _| {
817 file.defaults.show_edit_predictions = Some(!show_edit_predictions)
818 });
819}
820
821fn set_completion_provider(fs: Arc<dyn Fs>, cx: &mut App, provider: EditPredictionProvider) {
822 update_settings_file::<AllLanguageSettings>(fs, cx, move |file, _| {
823 file.features
824 .get_or_insert(Default::default())
825 .edit_prediction_provider = Some(provider);
826 });
827}
828
829fn toggle_show_inline_completions_for_language(
830 language: Arc<Language>,
831 fs: Arc<dyn Fs>,
832 cx: &mut App,
833) {
834 let show_edit_predictions =
835 all_language_settings(None, cx).show_inline_completions(Some(&language), cx);
836 update_settings_file::<AllLanguageSettings>(fs, cx, move |file, _| {
837 file.languages
838 .entry(language.name())
839 .or_default()
840 .show_edit_predictions = Some(!show_edit_predictions);
841 });
842}
843
844fn hide_copilot(fs: Arc<dyn Fs>, cx: &mut App) {
845 update_settings_file::<AllLanguageSettings>(fs, cx, move |file, _| {
846 file.features
847 .get_or_insert(Default::default())
848 .edit_prediction_provider = Some(EditPredictionProvider::None);
849 });
850}