use crate::{CompletionDiffElement, InlineCompletion, InlineCompletionRating, Zeta};use editor::Editor;use gpui::{actions, prelude::*, App, DismissEvent, Entity, EventEmitter, FocusHandle, Focusable};use language::language_settings;use std::time::Duration;use ui::{prelude::*, KeyBinding, List, ListItem, ListItemSpacing, Tooltip};use workspace::{ModalView, Workspace}; actions!( zeta, [ ThumbsUpActiveCompletion, ThumbsDownActiveCompletion, NextEdit, PreviousEdit, FocusCompletions, PreviewCompletion, ]); pub struct RateCompletionModal { zeta: Entity<Zeta>, active_completion: Option<ActiveCompletion>, selected_index: usize, focus_handle: FocusHandle, _subscription: gpui::Subscription, current_view: RateCompletionView,} struct ActiveCompletion { completion: InlineCompletion, feedback_editor: Entity<Editor>,} #[derive(Debug, Clone, Copy, PartialEq, PartialOrd)]enum RateCompletionView { SuggestedEdits, RawInput,} impl RateCompletionView { pub fn name(&self) -> &'static str { match self { Self::SuggestedEdits => "Suggested Edits", Self::RawInput => "Recorded Events & Input", } }} impl RateCompletionModal { pub fn toggle(workspace: &mut Workspace, window: &mut Window, cx: &mut Context<Workspace>) { if let Some(zeta) = Zeta::global(cx) { workspace.toggle_modal(window, cx, |_window, cx| RateCompletionModal::new(zeta, cx)); telemetry::event!("Rate Completion Modal Open", source = "Edit Prediction"); } } pub fn new(zeta: Entity<Zeta>, cx: &mut Context<Self>) -> Self { let subscription = cx.observe(&zeta, |_, _, cx| cx.notify()); Self { zeta, selected_index: 0, focus_handle: cx.focus_handle(), active_completion: None, _subscription: subscription, current_view: RateCompletionView::SuggestedEdits, } } fn dismiss(&mut self, _: &menu::Cancel, _: &mut Window, cx: &mut Context<Self>) { cx.emit(DismissEvent); } fn select_next(&mut self, _: &menu::SelectNext, _: &mut Window, cx: &mut Context<Self>) { self.selected_index += 1; self.selected_index = usize::min( self.selected_index, self.zeta.read(cx).shown_completions().count(), ); cx.notify(); } fn select_prev(&mut self, _: &menu::SelectPrev, _: &mut Window, cx: &mut Context<Self>) { self.selected_index = self.selected_index.saturating_sub(1); cx.notify(); } fn select_next_edit(&mut self, _: &NextEdit, _: &mut Window, cx: &mut Context<Self>) { let next_index = self .zeta .read(cx) .shown_completions() .skip(self.selected_index) .enumerate() .skip(1) // Skip straight to the next item .find(|(_, completion)| !completion.edits.is_empty()) .map(|(ix, _)| ix + self.selected_index); if let Some(next_index) = next_index { self.selected_index = next_index; cx.notify(); } } fn select_prev_edit(&mut self, _: &PreviousEdit, _: &mut Window, cx: &mut Context<Self>) { let zeta = self.zeta.read(cx); let completions_len = zeta.shown_completions_len(); let prev_index = self .zeta .read(cx) .shown_completions() .rev() .skip((completions_len - 1) - self.selected_index) .enumerate() .skip(1) // Skip straight to the previous item .find(|(_, completion)| !completion.edits.is_empty()) .map(|(ix, _)| self.selected_index - ix); if let Some(prev_index) = prev_index { self.selected_index = prev_index; cx.notify(); } cx.notify(); } fn select_first(&mut self, _: &menu::SelectFirst, _: &mut Window, cx: &mut Context<Self>) { self.selected_index = 0; cx.notify(); } fn select_last(&mut self, _: &menu::SelectLast, _window: &mut Window, cx: &mut Context<Self>) { self.selected_index = self.zeta.read(cx).shown_completions_len() - 1; cx.notify(); } pub fn thumbs_up_active( &mut self, _: &ThumbsUpActiveCompletion, window: &mut Window, cx: &mut Context<Self>, ) { self.zeta.update(cx, |zeta, cx| { if let Some(active) = &self.active_completion { zeta.rate_completion( &active.completion, InlineCompletionRating::Positive, active.feedback_editor.read(cx).text(cx), cx, ); } }); let current_completion = self .active_completion .as_ref() .map(|completion| completion.completion.clone()); self.select_completion(current_completion, false, window, cx); self.select_next_edit(&Default::default(), window, cx); self.confirm(&Default::default(), window, cx); cx.notify(); } pub fn thumbs_down_active( &mut self, _: &ThumbsDownActiveCompletion, window: &mut Window, cx: &mut Context<Self>, ) { if let Some(active) = &self.active_completion { if active.feedback_editor.read(cx).text(cx).is_empty() { return; } self.zeta.update(cx, |zeta, cx| { zeta.rate_completion( &active.completion, InlineCompletionRating::Negative, active.feedback_editor.read(cx).text(cx), cx, ); }); } let current_completion = self .active_completion .as_ref() .map(|completion| completion.completion.clone()); self.select_completion(current_completion, false, window, cx); self.select_next_edit(&Default::default(), window, cx); self.confirm(&Default::default(), window, cx); cx.notify(); } fn focus_completions( &mut self, _: &FocusCompletions, window: &mut Window, cx: &mut Context<Self>, ) { cx.focus_self(window); cx.notify(); } fn preview_completion( &mut self, _: &PreviewCompletion, window: &mut Window, cx: &mut Context<Self>, ) { let completion = self .zeta .read(cx) .shown_completions() .skip(self.selected_index) .take(1) .next() .cloned(); self.select_completion(completion, false, window, cx); } fn confirm(&mut self, _: &menu::Confirm, window: &mut Window, cx: &mut Context<Self>) { let completion = self .zeta .read(cx) .shown_completions() .skip(self.selected_index) .take(1) .next() .cloned(); self.select_completion(completion, true, window, cx); } pub fn select_completion( &mut self, completion: Option<InlineCompletion>, focus: bool, window: &mut Window, cx: &mut Context<Self>, ) { // Avoid resetting completion rating if it's already selected. if let Some(completion) = completion.as_ref() { self.selected_index = self .zeta .read(cx) .shown_completions() .enumerate() .find(|(_, completion_b)| completion.id == completion_b.id) .map(|(ix, _)| ix) .unwrap_or(self.selected_index); cx.notify(); if let Some(prev_completion) = self.active_completion.as_ref() { if completion.id == prev_completion.completion.id { if focus { window.focus(&prev_completion.feedback_editor.focus_handle(cx)); } return; } } } self.active_completion = completion.map(|completion| ActiveCompletion { completion, feedback_editor: cx.new(|cx| { let mut editor = Editor::multi_line(window, cx); editor.set_soft_wrap_mode(language_settings::SoftWrap::EditorWidth, cx); editor.set_show_line_numbers(false, cx); editor.set_show_scrollbars(false, cx); editor.set_show_git_diff_gutter(false, cx); editor.set_show_code_actions(false, cx); editor.set_show_runnables(false, cx); editor.set_show_wrap_guides(false, cx); editor.set_show_indent_guides(false, cx); editor.set_show_edit_predictions(Some(false), window, cx); editor.set_placeholder_text("Add your feedback…", cx); if focus { cx.focus_self(window); } editor }), }); cx.notify(); } fn render_view_nav(&self, cx: &Context<Self>) -> impl IntoElement { h_flex() .h_8() .px_1() .border_b_1() .border_color(cx.theme().colors().border) .bg(cx.theme().colors().elevated_surface_background) .gap_1() .child( Button::new( ElementId::Name("suggested-edits".into()), RateCompletionView::SuggestedEdits.name(), ) .label_size(LabelSize::Small) .on_click(cx.listener(move |this, _, _window, cx| { this.current_view = RateCompletionView::SuggestedEdits; cx.notify(); })) .toggle_state(self.current_view == RateCompletionView::SuggestedEdits), ) .child( Button::new( ElementId::Name("raw-input".into()), RateCompletionView::RawInput.name(), ) .label_size(LabelSize::Small) .on_click(cx.listener(move |this, _, _window, cx| { this.current_view = RateCompletionView::RawInput; cx.notify(); })) .toggle_state(self.current_view == RateCompletionView::RawInput), ) } fn render_suggested_edits(&self, cx: &mut Context<Self>) -> Option<gpui::Stateful<Div>> { let active_completion = self.active_completion.as_ref()?; let bg_color = cx.theme().colors().editor_background; Some( div() .id("diff") .p_4() .size_full() .bg(bg_color) .overflow_scroll() .whitespace_nowrap() .child(CompletionDiffElement::new( &active_completion.completion, cx, )), ) } fn render_raw_input(&self, cx: &mut Context<Self>) -> Option<gpui::Stateful<Div>> { Some( v_flex() .size_full() .overflow_hidden() .relative() .child( div() .id("raw-input") .py_4() .px_6() .size_full() .bg(cx.theme().colors().editor_background) .overflow_scroll() .child(if let Some(active_completion) = &self.active_completion { format!( "{}\n{}", active_completion.completion.input_events, active_completion.completion.input_excerpt ) } else { "No active completion".to_string() }), ) .id("raw-input-view"), ) } fn render_active_completion( &mut self, window: &mut Window, cx: &mut Context<Self>, ) -> Option<impl IntoElement> { let active_completion = self.active_completion.as_ref()?; let completion_id = active_completion.completion.id; let focus_handle = &self.focus_handle(cx); let border_color = cx.theme().colors().border; let bg_color = cx.theme().colors().editor_background; let rated = self.zeta.read(cx).is_completion_rated(completion_id); let feedback_empty = active_completion .feedback_editor .read(cx) .text(cx) .is_empty(); let label_container = h_flex().pl_1().gap_1p5(); Some( v_flex() .size_full() .overflow_hidden() .relative() .child( v_flex() .size_full() .overflow_hidden() .relative() .child(self.render_view_nav(cx)) .when_some(match self.current_view { RateCompletionView::SuggestedEdits => self.render_suggested_edits(cx), RateCompletionView::RawInput => self.render_raw_input(cx), }, |this, element| this.child(element)) ) .when(!rated, |this| { this.child( h_flex() .p_2() .gap_2() .border_y_1() .border_color(border_color) .child( Icon::new(IconName::Info) .size(IconSize::XSmall) .color(Color::Muted) ) .child( div() .w_full() .pr_2() .flex_wrap() .child( Label::new("Explain why this completion is good or bad. If it's negative, describe what you expected instead.") .size(LabelSize::Small) .color(Color::Muted) ) ) ) }) .when(!rated, |this| { this.child( div() .h_40() .pt_1() .bg(bg_color) .child(active_completion.feedback_editor.clone()) ) }) .child( h_flex() .p_1() .h_8() .max_h_8() .border_t_1() .border_color(border_color) .max_w_full() .justify_between() .children(if rated { Some( label_container .child( Icon::new(IconName::Check) .size(IconSize::Small) .color(Color::Success), ) .child(Label::new("Rated completion.").color(Color::Muted)), ) } else if active_completion.completion.edits.is_empty() { Some( label_container .child( Icon::new(IconName::Warning) .size(IconSize::Small) .color(Color::Warning), ) .child(Label::new("No edits produced.").color(Color::Muted)), ) } else { Some(label_container) }) .child( h_flex() .gap_1() .child( Button::new("bad", "Bad Completion") .icon(IconName::ThumbsDown) .icon_size(IconSize::Small) .icon_position(IconPosition::Start) .disabled(rated || feedback_empty) .when(feedback_empty, |this| { this.tooltip(Tooltip::text("Explain what's bad about it before reporting it")) }) .key_binding(KeyBinding::for_action_in( &ThumbsDownActiveCompletion, focus_handle, window, cx )) .on_click(cx.listener(move |this, _, window, cx| { this.thumbs_down_active( &ThumbsDownActiveCompletion, window, cx, ); })), )Lines 1–500 of 675. Load more
use crate::{CompletionDiffElement, InlineCompletion, InlineCompletionRating, Zeta}; use editor::Editor; use gpui::{actions, prelude::*, App, DismissEvent, Entity, EventEmitter, FocusHandle, Focusable}; use language::language_settings; use std::time::Duration; use ui::{prelude::*, KeyBinding, List, ListItem, ListItemSpacing, Tooltip}; use workspace::{ModalView, Workspace}; actions!( zeta, [ ThumbsUpActiveCompletion, ThumbsDownActiveCompletion, NextEdit, PreviousEdit, FocusCompletions, PreviewCompletion, ] ); pub struct RateCompletionModal { zeta: Entity<Zeta>, active_completion: Option<ActiveCompletion>, selected_index: usize, focus_handle: FocusHandle, _subscription: gpui::Subscription, current_view: RateCompletionView, } struct ActiveCompletion { completion: InlineCompletion, feedback_editor: Entity<Editor>, } #[derive(Debug, Clone, Copy, PartialEq, PartialOrd)] enum RateCompletionView { SuggestedEdits, RawInput, } impl RateCompletionView { pub fn name(&self) -> &'static str { match self { Self::SuggestedEdits => "Suggested Edits", Self::RawInput => "Recorded Events & Input", } } } impl RateCompletionModal { pub fn toggle(workspace: &mut Workspace, window: &mut Window, cx: &mut Context<Workspace>) { if let Some(zeta) = Zeta::global(cx) { workspace.toggle_modal(window, cx, |_window, cx| RateCompletionModal::new(zeta, cx)); telemetry::event!("Rate Completion Modal Open", source = "Edit Prediction"); } } pub fn new(zeta: Entity<Zeta>, cx: &mut Context<Self>) -> Self { let subscription = cx.observe(&zeta, |_, _, cx| cx.notify()); Self { zeta, selected_index: 0, focus_handle: cx.focus_handle(), active_completion: None, _subscription: subscription, current_view: RateCompletionView::SuggestedEdits, } } fn dismiss(&mut self, _: &menu::Cancel, _: &mut Window, cx: &mut Context<Self>) { cx.emit(DismissEvent); } fn select_next(&mut self, _: &menu::SelectNext, _: &mut Window, cx: &mut Context<Self>) { self.selected_index += 1; self.selected_index = usize::min( self.selected_index, self.zeta.read(cx).shown_completions().count(), ); cx.notify(); } fn select_prev(&mut self, _: &menu::SelectPrev, _: &mut Window, cx: &mut Context<Self>) { self.selected_index = self.selected_index.saturating_sub(1); cx.notify(); } fn select_next_edit(&mut self, _: &NextEdit, _: &mut Window, cx: &mut Context<Self>) { let next_index = self .zeta .read(cx) .shown_completions() .skip(self.selected_index) .enumerate() .skip(1) // Skip straight to the next item .find(|(_, completion)| !completion.edits.is_empty()) .map(|(ix, _)| ix + self.selected_index); if let Some(next_index) = next_index { self.selected_index = next_index; cx.notify(); } } fn select_prev_edit(&mut self, _: &PreviousEdit, _: &mut Window, cx: &mut Context<Self>) { let zeta = self.zeta.read(cx); let completions_len = zeta.shown_completions_len(); let prev_index = self .zeta .read(cx) .shown_completions() .rev() .skip((completions_len - 1) - self.selected_index) .enumerate() .skip(1) // Skip straight to the previous item .find(|(_, completion)| !completion.edits.is_empty()) .map(|(ix, _)| self.selected_index - ix); if let Some(prev_index) = prev_index { self.selected_index = prev_index; cx.notify(); } cx.notify(); } fn select_first(&mut self, _: &menu::SelectFirst, _: &mut Window, cx: &mut Context<Self>) { self.selected_index = 0; cx.notify(); } fn select_last(&mut self, _: &menu::SelectLast, _window: &mut Window, cx: &mut Context<Self>) { self.selected_index = self.zeta.read(cx).shown_completions_len() - 1; cx.notify(); } pub fn thumbs_up_active( &mut self, _: &ThumbsUpActiveCompletion, window: &mut Window, cx: &mut Context<Self>, ) { self.zeta.update(cx, |zeta, cx| { if let Some(active) = &self.active_completion { zeta.rate_completion( &active.completion, InlineCompletionRating::Positive, active.feedback_editor.read(cx).text(cx), cx, ); } }); let current_completion = self .active_completion .as_ref() .map(|completion| completion.completion.clone()); self.select_completion(current_completion, false, window, cx); self.select_next_edit(&Default::default(), window, cx); self.confirm(&Default::default(), window, cx); cx.notify(); } pub fn thumbs_down_active( &mut self, _: &ThumbsDownActiveCompletion, window: &mut Window, cx: &mut Context<Self>, ) { if let Some(active) = &self.active_completion { if active.feedback_editor.read(cx).text(cx).is_empty() { return; } self.zeta.update(cx, |zeta, cx| { zeta.rate_completion( &active.completion, InlineCompletionRating::Negative, active.feedback_editor.read(cx).text(cx), cx, ); }); } let current_completion = self .active_completion .as_ref() .map(|completion| completion.completion.clone()); self.select_completion(current_completion, false, window, cx); self.select_next_edit(&Default::default(), window, cx); self.confirm(&Default::default(), window, cx); cx.notify(); } fn focus_completions( &mut self, _: &FocusCompletions, window: &mut Window, cx: &mut Context<Self>, ) { cx.focus_self(window); cx.notify(); } fn preview_completion( &mut self, _: &PreviewCompletion, window: &mut Window, cx: &mut Context<Self>, ) { let completion = self .zeta .read(cx) .shown_completions() .skip(self.selected_index) .take(1) .next() .cloned(); self.select_completion(completion, false, window, cx); } fn confirm(&mut self, _: &menu::Confirm, window: &mut Window, cx: &mut Context<Self>) { let completion = self .zeta .read(cx) .shown_completions() .skip(self.selected_index) .take(1) .next() .cloned(); self.select_completion(completion, true, window, cx); } pub fn select_completion( &mut self, completion: Option<InlineCompletion>, focus: bool, window: &mut Window, cx: &mut Context<Self>, ) { // Avoid resetting completion rating if it's already selected. if let Some(completion) = completion.as_ref() { self.selected_index = self .zeta .read(cx) .shown_completions() .enumerate() .find(|(_, completion_b)| completion.id == completion_b.id) .map(|(ix, _)| ix) .unwrap_or(self.selected_index); cx.notify(); if let Some(prev_completion) = self.active_completion.as_ref() { if completion.id == prev_completion.completion.id { if focus { window.focus(&prev_completion.feedback_editor.focus_handle(cx)); } return; } } } self.active_completion = completion.map(|completion| ActiveCompletion { completion, feedback_editor: cx.new(|cx| { let mut editor = Editor::multi_line(window, cx); editor.set_soft_wrap_mode(language_settings::SoftWrap::EditorWidth, cx); editor.set_show_line_numbers(false, cx); editor.set_show_scrollbars(false, cx); editor.set_show_git_diff_gutter(false, cx); editor.set_show_code_actions(false, cx); editor.set_show_runnables(false, cx); editor.set_show_wrap_guides(false, cx); editor.set_show_indent_guides(false, cx); editor.set_show_edit_predictions(Some(false), window, cx); editor.set_placeholder_text("Add your feedback…", cx); if focus { cx.focus_self(window); } editor }), }); cx.notify(); } fn render_view_nav(&self, cx: &Context<Self>) -> impl IntoElement { h_flex() .h_8() .px_1() .border_b_1() .border_color(cx.theme().colors().border) .bg(cx.theme().colors().elevated_surface_background) .gap_1() .child( Button::new( ElementId::Name("suggested-edits".into()), RateCompletionView::SuggestedEdits.name(), ) .label_size(LabelSize::Small) .on_click(cx.listener(move |this, _, _window, cx| { this.current_view = RateCompletionView::SuggestedEdits; cx.notify(); })) .toggle_state(self.current_view == RateCompletionView::SuggestedEdits), ) .child( Button::new( ElementId::Name("raw-input".into()), RateCompletionView::RawInput.name(), ) .label_size(LabelSize::Small) .on_click(cx.listener(move |this, _, _window, cx| { this.current_view = RateCompletionView::RawInput; cx.notify(); })) .toggle_state(self.current_view == RateCompletionView::RawInput), ) } fn render_suggested_edits(&self, cx: &mut Context<Self>) -> Option<gpui::Stateful<Div>> { let active_completion = self.active_completion.as_ref()?; let bg_color = cx.theme().colors().editor_background; Some( div() .id("diff") .p_4() .size_full() .bg(bg_color) .overflow_scroll() .whitespace_nowrap() .child(CompletionDiffElement::new( &active_completion.completion, cx, )), ) } fn render_raw_input(&self, cx: &mut Context<Self>) -> Option<gpui::Stateful<Div>> { Some( v_flex() .size_full() .overflow_hidden() .relative() .child( div() .id("raw-input") .py_4() .px_6() .size_full() .bg(cx.theme().colors().editor_background) .overflow_scroll() .child(if let Some(active_completion) = &self.active_completion { format!( "{}\n{}", active_completion.completion.input_events, active_completion.completion.input_excerpt ) } else { "No active completion".to_string() }), ) .id("raw-input-view"), ) } fn render_active_completion( &mut self, window: &mut Window, cx: &mut Context<Self>, ) -> Option<impl IntoElement> { let active_completion = self.active_completion.as_ref()?; let completion_id = active_completion.completion.id; let focus_handle = &self.focus_handle(cx); let border_color = cx.theme().colors().border; let bg_color = cx.theme().colors().editor_background; let rated = self.zeta.read(cx).is_completion_rated(completion_id); let feedback_empty = active_completion .feedback_editor .read(cx) .text(cx) .is_empty(); let label_container = h_flex().pl_1().gap_1p5(); Some( v_flex() .size_full() .overflow_hidden() .relative() .child( v_flex() .size_full() .overflow_hidden() .relative() .child(self.render_view_nav(cx)) .when_some(match self.current_view { RateCompletionView::SuggestedEdits => self.render_suggested_edits(cx), RateCompletionView::RawInput => self.render_raw_input(cx), }, |this, element| this.child(element)) ) .when(!rated, |this| { this.child( h_flex() .p_2() .gap_2() .border_y_1() .border_color(border_color) .child( Icon::new(IconName::Info) .size(IconSize::XSmall) .color(Color::Muted) ) .child( div() .w_full() .pr_2() .flex_wrap() .child( Label::new("Explain why this completion is good or bad. If it's negative, describe what you expected instead.") .size(LabelSize::Small) .color(Color::Muted) ) ) ) }) .when(!rated, |this| { this.child( div() .h_40() .pt_1() .bg(bg_color) .child(active_completion.feedback_editor.clone()) ) }) .child( h_flex() .p_1() .h_8() .max_h_8() .border_t_1() .border_color(border_color) .max_w_full() .justify_between() .children(if rated { Some( label_container .child( Icon::new(IconName::Check) .size(IconSize::Small) .color(Color::Success), ) .child(Label::new("Rated completion.").color(Color::Muted)), ) } else if active_completion.completion.edits.is_empty() { Some( label_container .child( Icon::new(IconName::Warning) .size(IconSize::Small) .color(Color::Warning), ) .child(Label::new("No edits produced.").color(Color::Muted)), ) } else { Some(label_container) }) .child( h_flex() .gap_1() .child( Button::new("bad", "Bad Completion") .icon(IconName::ThumbsDown) .icon_size(IconSize::Small) .icon_position(IconPosition::Start) .disabled(rated || feedback_empty) .when(feedback_empty, |this| { this.tooltip(Tooltip::text("Explain what's bad about it before reporting it")) }) .key_binding(KeyBinding::for_action_in( &ThumbsDownActiveCompletion, focus_handle, window, cx )) .on_click(cx.listener(move |this, _, window, cx| { this.thumbs_down_active( &ThumbsDownActiveCompletion, window, cx, ); })), )