use crate::feedback_editor::{FeedbackEditor, SubmitFeedback}; use anyhow::Result; use gpui::{ elements::{Label, MouseEventHandler}, platform::{CursorStyle, MouseButton}, AnyElement, AppContext, Element, Entity, Task, View, ViewContext, ViewHandle, }; use workspace::{item::ItemHandle, ToolbarItemLocation, ToolbarItemView}; pub fn init(cx: &mut AppContext) { cx.add_async_action(SubmitFeedbackButton::submit); } pub struct SubmitFeedbackButton { pub(crate) active_item: Option>, } impl SubmitFeedbackButton { pub fn new() -> Self { Self { active_item: Default::default(), } } pub fn submit( &mut self, _: &SubmitFeedback, cx: &mut ViewContext, ) -> Option>> { if let Some(active_item) = self.active_item.as_ref() { Some(active_item.update(cx, |feedback_editor, cx| feedback_editor.submit(cx))) } else { None } } } impl Entity for SubmitFeedbackButton { type Event = (); } impl View for SubmitFeedbackButton { fn ui_name() -> &'static str { "SubmitFeedbackButton" } fn render(&mut self, cx: &mut ViewContext) -> AnyElement { let theme = theme::current(cx).clone(); let allow_submission = self .active_item .as_ref() .map_or(true, |i| i.read(cx).allow_submission); enum SubmitFeedbackButton {} MouseEventHandler::::new(0, cx, |state, _| { let text; let style = if allow_submission { text = "Submit as Markdown"; theme.feedback.submit_button.style_for(state) } else { text = "Submitting..."; theme .feedback .submit_button .disabled .as_ref() .unwrap_or(&theme.feedback.submit_button.default) }; Label::new(text, style.text.clone()) .contained() .with_style(style.container) }) .with_cursor_style(CursorStyle::PointingHand) .on_click(MouseButton::Left, |_, this, cx| { this.submit(&Default::default(), cx); }) .aligned() .contained() .with_margin_left(theme.feedback.button_margin) .with_tooltip::( 0, "cmd-s", Some(Box::new(SubmitFeedback)), theme.tooltip.clone(), cx, ) .into_any() } } impl ToolbarItemView for SubmitFeedbackButton { fn set_active_pane_item( &mut self, active_pane_item: Option<&dyn ItemHandle>, cx: &mut ViewContext, ) -> workspace::ToolbarItemLocation { cx.notify(); if let Some(feedback_editor) = active_pane_item.and_then(|i| i.downcast::()) { self.active_item = Some(feedback_editor); ToolbarItemLocation::PrimaryRight { flex: None } } else { self.active_item = None; ToolbarItemLocation::Hidden } } }