1use std::rc::Rc;
 2
 3use acp_thread::AgentModelSelector;
 4use agent_client_protocol as acp;
 5use gpui::{Entity, FocusHandle};
 6use picker::popover_menu::PickerPopoverMenu;
 7use ui::{
 8    ButtonLike, Context, IntoElement, PopoverMenuHandle, SharedString, Tooltip, Window, prelude::*,
 9};
10use zed_actions::agent::ToggleModelSelector;
11
12use crate::acp::{AcpModelSelector, model_selector::acp_model_selector};
13
14pub struct AcpModelSelectorPopover {
15    selector: Entity<AcpModelSelector>,
16    menu_handle: PopoverMenuHandle<AcpModelSelector>,
17    focus_handle: FocusHandle,
18}
19
20impl AcpModelSelectorPopover {
21    pub(crate) fn new(
22        session_id: acp::SessionId,
23        selector: Rc<dyn AgentModelSelector>,
24        menu_handle: PopoverMenuHandle<AcpModelSelector>,
25        focus_handle: FocusHandle,
26        window: &mut Window,
27        cx: &mut Context<Self>,
28    ) -> Self {
29        Self {
30            selector: cx.new(move |cx| acp_model_selector(session_id, selector, window, cx)),
31            menu_handle,
32            focus_handle,
33        }
34    }
35
36    pub fn toggle(&self, window: &mut Window, cx: &mut Context<Self>) {
37        self.menu_handle.toggle(window, cx);
38    }
39}
40
41impl Render for AcpModelSelectorPopover {
42    fn render(&mut self, window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
43        let model = self.selector.read(cx).delegate.active_model();
44        let model_name = model
45            .as_ref()
46            .map(|model| model.name.clone())
47            .unwrap_or_else(|| SharedString::from("Select a Model"));
48
49        let model_icon = model.as_ref().and_then(|model| model.icon);
50
51        let focus_handle = self.focus_handle.clone();
52
53        PickerPopoverMenu::new(
54            self.selector.clone(),
55            ButtonLike::new("active-model")
56                .when_some(model_icon, |this, icon| {
57                    this.child(Icon::new(icon).color(Color::Muted).size(IconSize::XSmall))
58                })
59                .child(
60                    Label::new(model_name)
61                        .color(Color::Muted)
62                        .size(LabelSize::Small)
63                        .ml_0p5(),
64                )
65                .child(
66                    Icon::new(IconName::ChevronDown)
67                        .color(Color::Muted)
68                        .size(IconSize::XSmall),
69                ),
70            move |window, cx| {
71                Tooltip::for_action_in(
72                    "Change Model",
73                    &ToggleModelSelector,
74                    &focus_handle,
75                    window,
76                    cx,
77                )
78            },
79            gpui::Corner::BottomRight,
80            cx,
81        )
82        .with_handle(self.menu_handle.clone())
83        .render(window, cx)
84    }
85}