217 lines · 7.6 KB · Rust
RawDownload
use std::sync::Arc; use agent_settings::{    AgentDockPosition, AgentProfile, AgentProfileId, AgentSettings, GroupedAgentProfiles,    builtin_profiles,};use fs::Fs;use gpui::{Action, Empty, Entity, FocusHandle, Subscription, WeakEntity, prelude::*};use language_model::LanguageModelRegistry;use settings::{Settings as _, SettingsStore, update_settings_file};use ui::{    ContextMenu, ContextMenuEntry, DocumentationSide, PopoverMenu, PopoverMenuHandle, Tooltip,    prelude::*,};use util::ResultExt as _; use crate::{ManageProfiles, Thread, ThreadStore, ToggleProfileSelector}; pub struct ProfileSelector {    profiles: GroupedAgentProfiles,    fs: Arc<dyn Fs>,    thread: Entity<Thread>,    thread_store: WeakEntity<ThreadStore>,    menu_handle: PopoverMenuHandle<ContextMenu>,    focus_handle: FocusHandle,    _subscriptions: Vec<Subscription>,} impl ProfileSelector {    pub fn new(        fs: Arc<dyn Fs>,        thread: Entity<Thread>,        thread_store: WeakEntity<ThreadStore>,        focus_handle: FocusHandle,        cx: &mut Context<Self>,    ) -> Self {        let settings_subscription = cx.observe_global::<SettingsStore>(move |this, cx| {            this.refresh_profiles(cx);        });         Self {            profiles: GroupedAgentProfiles::from_settings(AgentSettings::get_global(cx)),            fs,            thread,            thread_store,            menu_handle: PopoverMenuHandle::default(),            focus_handle,            _subscriptions: vec![settings_subscription],        }    }     pub fn menu_handle(&self) -> PopoverMenuHandle<ContextMenu> {        self.menu_handle.clone()    }     fn refresh_profiles(&mut self, cx: &mut Context<Self>) {        self.profiles = GroupedAgentProfiles::from_settings(AgentSettings::get_global(cx));    }     fn build_context_menu(        &self,        window: &mut Window,        cx: &mut Context<Self>,    ) -> Entity<ContextMenu> {        ContextMenu::build(window, cx, |mut menu, _window, cx| {            let settings = AgentSettings::get_global(cx);            for (profile_id, profile) in self.profiles.builtin.iter() {                menu = menu.item(self.menu_entry_for_profile(                    profile_id.clone(),                    profile,                    settings,                    cx,                ));            }             if !self.profiles.custom.is_empty() {                menu = menu.separator().header("Custom Profiles");                for (profile_id, profile) in self.profiles.custom.iter() {                    menu = menu.item(self.menu_entry_for_profile(                        profile_id.clone(),                        profile,                        settings,                        cx,                    ));                }            }             menu = menu.separator();            menu = menu.item(ContextMenuEntry::new("Configure Profiles…").handler(                move |window, cx| {                    window.dispatch_action(ManageProfiles::default().boxed_clone(), cx);                },            ));             menu        })    }     fn menu_entry_for_profile(        &self,        profile_id: AgentProfileId,        profile: &AgentProfile,        settings: &AgentSettings,        _cx: &App,    ) -> ContextMenuEntry {        let documentation = match profile.name.to_lowercase().as_str() {            builtin_profiles::WRITE => Some("Get help to write anything."),            builtin_profiles::ASK => Some("Chat about your codebase."),            builtin_profiles::MINIMAL => Some("Chat about anything with no tools."),            _ => None,        };         let entry = ContextMenuEntry::new(profile.name.clone())            .toggleable(IconPosition::End, profile_id == settings.default_profile);         let entry = if let Some(doc_text) = documentation {            entry.documentation_aside(documentation_side(settings.dock), move |_| {                Label::new(doc_text).into_any_element()            })        } else {            entry        };         entry.handler({            let fs = self.fs.clone();            let thread_store = self.thread_store.clone();            let profile_id = profile_id.clone();            move |_window, cx| {                update_settings_file::<AgentSettings>(fs.clone(), cx, {                    let profile_id = profile_id.clone();                    move |settings, _cx| {                        settings.set_profile(profile_id.clone());                    }                });                 thread_store                    .update(cx, |this, cx| {                        this.load_profile_by_id(profile_id.clone(), cx);                    })                    .log_err();            }        })    }} impl Render for ProfileSelector {    fn render(&mut self, _window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {        let settings = AgentSettings::get_global(cx);        let profile_id = &settings.default_profile;        let profile = settings.profiles.get(profile_id);         let selected_profile = profile            .map(|profile| profile.name.clone())            .unwrap_or_else(|| "Unknown".into());         let configured_model = self.thread.read(cx).configured_model().or_else(|| {            let model_registry = LanguageModelRegistry::read_global(cx);            model_registry.default_model()        });        let Some(configured_model) = configured_model else {            return Empty.into_any_element();        };         if configured_model.model.supports_tools() {            let this = cx.entity().clone();            let focus_handle = self.focus_handle.clone();            let trigger_button = Button::new("profile-selector-model", selected_profile)                .label_size(LabelSize::Small)                .color(Color::Muted)                .icon(IconName::ChevronDown)                .icon_size(IconSize::XSmall)                .icon_position(IconPosition::End)                .icon_color(Color::Muted);             PopoverMenu::new("profile-selector")                .trigger_with_tooltip(trigger_button, {                    let focus_handle = focus_handle.clone();                    move |window, cx| {                        Tooltip::for_action_in(                            "Toggle Profile Menu",                            &ToggleProfileSelector,                            &focus_handle,                            window,                            cx,                        )                    }                })                .anchor(                    if documentation_side(settings.dock) == DocumentationSide::Left {                        gpui::Corner::BottomRight                    } else {                        gpui::Corner::BottomLeft                    },                )                .with_handle(self.menu_handle.clone())                .menu(move |window, cx| {                    Some(this.update(cx, |this, cx| this.build_context_menu(window, cx)))                })                .into_any_element()        } else {            Button::new("tools-not-supported-button", "Tools Unsupported")                .disabled(true)                .label_size(LabelSize::Small)                .color(Color::Muted)                .tooltip(Tooltip::text("This model does not support tools."))                .into_any_element()        }    }} fn documentation_side(position: AgentDockPosition) -> DocumentationSide {    match position {        AgentDockPosition::Left => DocumentationSide::Right,        AgentDockPosition::Bottom => DocumentationSide::Left,        AgentDockPosition::Right => DocumentationSide::Left,    }}