1use gpui::{
2 AnyElement, App, DismissEvent, Entity, EventEmitter, FocusHandle, Focusable, Task, WeakEntity,
3};
4use itertools::Itertools;
5use picker::{Picker, PickerDelegate};
6use project::{git::Repository, Project};
7use std::sync::Arc;
8use ui::{prelude::*, ListItem, ListItemSpacing};
9
10pub struct RepositorySelector {
11 picker: Entity<Picker<RepositorySelectorDelegate>>,
12}
13
14impl RepositorySelector {
15 pub fn new(
16 project_handle: Entity<Project>,
17 window: &mut Window,
18 cx: &mut Context<Self>,
19 ) -> Self {
20 let project = project_handle.read(cx);
21 let git_store = project.git_store().clone();
22 let all_repositories = git_store.read(cx).all_repositories();
23 let filtered_repositories = all_repositories.clone();
24
25 let widest_item_ix = all_repositories.iter().position_max_by(|a, b| {
26 a.read(cx)
27 .display_name(project, cx)
28 .len()
29 .cmp(&b.read(cx).display_name(project, cx).len())
30 });
31
32 let delegate = RepositorySelectorDelegate {
33 project: project_handle.downgrade(),
34 repository_selector: cx.entity().downgrade(),
35 repository_entries: all_repositories.clone(),
36 filtered_repositories,
37 selected_index: 0,
38 };
39
40 let picker = cx.new(|cx| {
41 Picker::nonsearchable_uniform_list(delegate, window, cx)
42 .widest_item(widest_item_ix)
43 .max_height(Some(rems(20.).into()))
44 });
45
46 RepositorySelector { picker }
47 }
48}
49
50impl EventEmitter<DismissEvent> for RepositorySelector {}
51
52impl Focusable for RepositorySelector {
53 fn focus_handle(&self, cx: &App) -> FocusHandle {
54 self.picker.focus_handle(cx)
55 }
56}
57
58impl Render for RepositorySelector {
59 fn render(&mut self, _window: &mut Window, _cx: &mut Context<Self>) -> impl IntoElement {
60 self.picker.clone()
61 }
62}
63
64pub struct RepositorySelectorDelegate {
65 project: WeakEntity<Project>,
66 repository_selector: WeakEntity<RepositorySelector>,
67 repository_entries: Vec<Entity<Repository>>,
68 filtered_repositories: Vec<Entity<Repository>>,
69 selected_index: usize,
70}
71
72impl RepositorySelectorDelegate {
73 pub fn update_repository_entries(&mut self, all_repositories: Vec<Entity<Repository>>) {
74 self.repository_entries = all_repositories.clone();
75 self.filtered_repositories = all_repositories;
76 self.selected_index = 0;
77 }
78}
79
80impl PickerDelegate for RepositorySelectorDelegate {
81 type ListItem = ListItem;
82
83 fn match_count(&self) -> usize {
84 self.filtered_repositories.len()
85 }
86
87 fn selected_index(&self) -> usize {
88 self.selected_index
89 }
90
91 fn set_selected_index(
92 &mut self,
93 ix: usize,
94 _window: &mut Window,
95 cx: &mut Context<Picker<Self>>,
96 ) {
97 self.selected_index = ix.min(self.filtered_repositories.len().saturating_sub(1));
98 cx.notify();
99 }
100
101 fn placeholder_text(&self, _window: &mut Window, _cx: &mut App) -> Arc<str> {
102 "Select a repository...".into()
103 }
104
105 fn update_matches(
106 &mut self,
107 query: String,
108 window: &mut Window,
109 cx: &mut Context<Picker<Self>>,
110 ) -> Task<()> {
111 let all_repositories = self.repository_entries.clone();
112
113 cx.spawn_in(window, |this, mut cx| async move {
114 let filtered_repositories = cx
115 .background_spawn(async move {
116 if query.is_empty() {
117 all_repositories
118 } else {
119 all_repositories
120 .into_iter()
121 .filter(|_repo_info| {
122 // TODO: Implement repository filtering logic
123 true
124 })
125 .collect()
126 }
127 })
128 .await;
129
130 this.update_in(&mut cx, |this, window, cx| {
131 this.delegate.filtered_repositories = filtered_repositories;
132 this.delegate.set_selected_index(0, window, cx);
133 cx.notify();
134 })
135 .ok();
136 })
137 }
138
139 fn confirm(&mut self, _secondary: bool, window: &mut Window, cx: &mut Context<Picker<Self>>) {
140 let Some(selected_repo) = self.filtered_repositories.get(self.selected_index) else {
141 return;
142 };
143 selected_repo.update(cx, |selected_repo, cx| selected_repo.activate(cx));
144 self.dismissed(window, cx);
145 }
146
147 fn dismissed(&mut self, _window: &mut Window, cx: &mut Context<Picker<Self>>) {
148 self.repository_selector
149 .update(cx, |_this, cx| cx.emit(DismissEvent))
150 .ok();
151 }
152
153 fn render_header(
154 &self,
155 _window: &mut Window,
156 _cx: &mut Context<Picker<Self>>,
157 ) -> Option<AnyElement> {
158 // TODO: Implement header rendering if needed
159 None
160 }
161
162 fn render_match(
163 &self,
164 ix: usize,
165 selected: bool,
166 _window: &mut Window,
167 cx: &mut Context<Picker<Self>>,
168 ) -> Option<Self::ListItem> {
169 let project = self.project.upgrade()?;
170 let repo_info = self.filtered_repositories.get(ix)?;
171 let display_name = repo_info.read(cx).display_name(project.read(cx), cx);
172 Some(
173 ListItem::new(ix)
174 .inset(true)
175 .spacing(ListItemSpacing::Sparse)
176 .toggle_state(selected)
177 .child(Label::new(display_name)),
178 )
179 }
180}