1use collections::{HashMap, VecDeque};
2use editor::{Editor, EditorEvent, MoveToEnd};
3use futures::{channel::mpsc, StreamExt};
4use gpui::{
5 actions, div, AnchorCorner, AnyElement, AppContext, Context, EventEmitter, FocusHandle,
6 FocusableView, IntoElement, Model, ModelContext, ParentElement, Render, Styled, Subscription,
7 View, ViewContext, VisualContext, WeakModel, WindowContext,
8};
9use language::{LanguageServerId, LanguageServerName};
10use lsp::IoKind;
11use project::{search::SearchQuery, Project};
12use std::{borrow::Cow, sync::Arc};
13use ui::{popover_menu, prelude::*, Button, Checkbox, ContextMenu, Label, Selection};
14use workspace::{
15 item::{Item, ItemHandle},
16 searchable::{SearchEvent, SearchableItem, SearchableItemHandle},
17 ToolbarItemEvent, ToolbarItemLocation, ToolbarItemView, Workspace,
18};
19
20const SEND_LINE: &str = "// Send:";
21const RECEIVE_LINE: &str = "// Receive:";
22const MAX_STORED_LOG_ENTRIES: usize = 2000;
23
24pub struct LogStore {
25 projects: HashMap<WeakModel<Project>, ProjectState>,
26 io_tx: mpsc::UnboundedSender<(WeakModel<Project>, LanguageServerId, IoKind, String)>,
27}
28
29struct ProjectState {
30 servers: HashMap<LanguageServerId, LanguageServerState>,
31 _subscriptions: [gpui::Subscription; 2],
32}
33
34struct LanguageServerState {
35 log_messages: VecDeque<String>,
36 rpc_state: Option<LanguageServerRpcState>,
37 _io_logs_subscription: Option<lsp::Subscription>,
38 _lsp_logs_subscription: Option<lsp::Subscription>,
39}
40
41struct LanguageServerRpcState {
42 rpc_messages: VecDeque<String>,
43 last_message_kind: Option<MessageKind>,
44}
45
46pub struct LspLogView {
47 pub(crate) editor: View<Editor>,
48 editor_subscription: Subscription,
49 log_store: Model<LogStore>,
50 current_server_id: Option<LanguageServerId>,
51 is_showing_rpc_trace: bool,
52 project: Model<Project>,
53 focus_handle: FocusHandle,
54 _log_store_subscriptions: Vec<Subscription>,
55}
56
57pub struct LspLogToolbarItemView {
58 log_view: Option<View<LspLogView>>,
59 _log_view_subscription: Option<Subscription>,
60}
61
62#[derive(Copy, Clone, PartialEq, Eq)]
63enum MessageKind {
64 Send,
65 Receive,
66}
67
68#[derive(Clone, Debug, PartialEq)]
69pub(crate) struct LogMenuItem {
70 pub server_id: LanguageServerId,
71 pub server_name: LanguageServerName,
72 pub worktree_root_name: String,
73 pub rpc_trace_enabled: bool,
74 pub rpc_trace_selected: bool,
75 pub logs_selected: bool,
76}
77
78actions!(debug, [OpenLanguageServerLogs]);
79
80pub fn init(cx: &mut AppContext) {
81 let log_store = cx.new_model(|cx| LogStore::new(cx));
82
83 cx.observe_new_views(move |workspace: &mut Workspace, cx| {
84 let project = workspace.project();
85 if project.read(cx).is_local() {
86 log_store.update(cx, |store, cx| {
87 store.add_project(&project, cx);
88 });
89 }
90
91 let log_store = log_store.clone();
92 workspace.register_action(move |workspace, _: &OpenLanguageServerLogs, cx| {
93 let project = workspace.project().read(cx);
94 if project.is_local() {
95 workspace.add_item(
96 Box::new(cx.new_view(|cx| {
97 LspLogView::new(workspace.project().clone(), log_store.clone(), cx)
98 })),
99 cx,
100 );
101 }
102 });
103 })
104 .detach();
105}
106
107impl LogStore {
108 pub fn new(cx: &mut ModelContext<Self>) -> Self {
109 let (io_tx, mut io_rx) = mpsc::unbounded();
110 let this = Self {
111 projects: HashMap::default(),
112 io_tx,
113 };
114 cx.spawn(|this, mut cx| async move {
115 while let Some((project, server_id, io_kind, message)) = io_rx.next().await {
116 if let Some(this) = this.upgrade() {
117 this.update(&mut cx, |this, cx| {
118 this.on_io(project, server_id, io_kind, &message, cx);
119 })?;
120 }
121 }
122 anyhow::Ok(())
123 })
124 .detach_and_log_err(cx);
125 this
126 }
127
128 pub fn add_project(&mut self, project: &Model<Project>, cx: &mut ModelContext<Self>) {
129 let weak_project = project.downgrade();
130 self.projects.insert(
131 project.downgrade(),
132 ProjectState {
133 servers: HashMap::default(),
134 _subscriptions: [
135 cx.observe_release(project, move |this, _, _| {
136 this.projects.remove(&weak_project);
137 }),
138 cx.subscribe(project, |this, project, event, cx| match event {
139 project::Event::LanguageServerAdded(id) => {
140 this.add_language_server(&project, *id, cx);
141 }
142 project::Event::LanguageServerRemoved(id) => {
143 this.remove_language_server(&project, *id, cx);
144 }
145 project::Event::LanguageServerLog(id, message) => {
146 this.add_language_server_log(&project, *id, message, cx);
147 }
148 _ => {}
149 }),
150 ],
151 },
152 );
153 }
154
155 fn add_language_server(
156 &mut self,
157 project: &Model<Project>,
158 id: LanguageServerId,
159 cx: &mut ModelContext<Self>,
160 ) -> Option<&mut LanguageServerState> {
161 let project_state = self.projects.get_mut(&project.downgrade())?;
162 let server_state = project_state.servers.entry(id).or_insert_with(|| {
163 cx.notify();
164 LanguageServerState {
165 rpc_state: None,
166 log_messages: VecDeque::with_capacity(MAX_STORED_LOG_ENTRIES),
167 _io_logs_subscription: None,
168 _lsp_logs_subscription: None,
169 }
170 });
171
172 let server = project.read(cx).language_server_for_id(id);
173 if let Some(server) = server.as_deref() {
174 if server.has_notification_handler::<lsp::notification::LogMessage>() {
175 // Another event wants to re-add the server that was already added and subscribed to, avoid doing it again.
176 return Some(server_state);
177 }
178 }
179
180 let weak_project = project.downgrade();
181 let io_tx = self.io_tx.clone();
182 server_state._io_logs_subscription = server.as_ref().map(|server| {
183 server.on_io(move |io_kind, message| {
184 io_tx
185 .unbounded_send((weak_project.clone(), id, io_kind, message.to_string()))
186 .ok();
187 })
188 });
189 let this = cx.handle().downgrade();
190 let weak_project = project.downgrade();
191 server_state._lsp_logs_subscription = server.map(|server| {
192 let server_id = server.server_id();
193 server.on_notification::<lsp::notification::LogMessage, _>({
194 move |params, mut cx| {
195 if let Some((project, this)) = weak_project.upgrade().zip(this.upgrade()) {
196 this.update(&mut cx, |this, cx| {
197 this.add_language_server_log(&project, server_id, ¶ms.message, cx);
198 })
199 .ok();
200 }
201 }
202 })
203 });
204 Some(server_state)
205 }
206
207 fn add_language_server_log(
208 &mut self,
209 project: &Model<Project>,
210 id: LanguageServerId,
211 message: &str,
212 cx: &mut ModelContext<Self>,
213 ) -> Option<()> {
214 let language_server_state = match self
215 .projects
216 .get_mut(&project.downgrade())?
217 .servers
218 .get_mut(&id)
219 {
220 Some(existing_state) => existing_state,
221 None => self.add_language_server(&project, id, cx)?,
222 };
223
224 let log_lines = &mut language_server_state.log_messages;
225 while log_lines.len() >= MAX_STORED_LOG_ENTRIES {
226 log_lines.pop_front();
227 }
228 let message = message.trim();
229 log_lines.push_back(message.to_string());
230 cx.emit(Event::NewServerLogEntry {
231 id,
232 entry: message.to_string(),
233 is_rpc: false,
234 });
235 cx.notify();
236 Some(())
237 }
238
239 fn remove_language_server(
240 &mut self,
241 project: &Model<Project>,
242 id: LanguageServerId,
243 cx: &mut ModelContext<Self>,
244 ) -> Option<()> {
245 let project_state = self.projects.get_mut(&project.downgrade())?;
246 project_state.servers.remove(&id);
247 cx.notify();
248 Some(())
249 }
250
251 fn server_logs(
252 &self,
253 project: &Model<Project>,
254 server_id: LanguageServerId,
255 ) -> Option<&VecDeque<String>> {
256 let weak_project = project.downgrade();
257 let project_state = self.projects.get(&weak_project)?;
258 let server_state = project_state.servers.get(&server_id)?;
259 Some(&server_state.log_messages)
260 }
261
262 fn enable_rpc_trace_for_language_server(
263 &mut self,
264 project: &Model<Project>,
265 server_id: LanguageServerId,
266 ) -> Option<&mut LanguageServerRpcState> {
267 let weak_project = project.downgrade();
268 let project_state = self.projects.get_mut(&weak_project)?;
269 let server_state = project_state.servers.get_mut(&server_id)?;
270 let rpc_state = server_state
271 .rpc_state
272 .get_or_insert_with(|| LanguageServerRpcState {
273 rpc_messages: VecDeque::with_capacity(MAX_STORED_LOG_ENTRIES),
274 last_message_kind: None,
275 });
276 Some(rpc_state)
277 }
278
279 pub fn disable_rpc_trace_for_language_server(
280 &mut self,
281 project: &Model<Project>,
282 server_id: LanguageServerId,
283 _: &mut ModelContext<Self>,
284 ) -> Option<()> {
285 let project = project.downgrade();
286 let project_state = self.projects.get_mut(&project)?;
287 let server_state = project_state.servers.get_mut(&server_id)?;
288 server_state.rpc_state.take();
289 Some(())
290 }
291
292 fn on_io(
293 &mut self,
294 project: WeakModel<Project>,
295 language_server_id: LanguageServerId,
296 io_kind: IoKind,
297 message: &str,
298 cx: &mut ModelContext<Self>,
299 ) -> Option<()> {
300 let is_received = match io_kind {
301 IoKind::StdOut => true,
302 IoKind::StdIn => false,
303 IoKind::StdErr => {
304 let project = project.upgrade()?;
305 let message = format!("stderr: {}", message.trim());
306 self.add_language_server_log(&project, language_server_id, &message, cx);
307 return Some(());
308 }
309 };
310
311 let state = self
312 .projects
313 .get_mut(&project)?
314 .servers
315 .get_mut(&language_server_id)?
316 .rpc_state
317 .as_mut()?;
318 let kind = if is_received {
319 MessageKind::Receive
320 } else {
321 MessageKind::Send
322 };
323
324 let rpc_log_lines = &mut state.rpc_messages;
325 if state.last_message_kind != Some(kind) {
326 let line_before_message = match kind {
327 MessageKind::Send => SEND_LINE,
328 MessageKind::Receive => RECEIVE_LINE,
329 };
330 rpc_log_lines.push_back(line_before_message.to_string());
331 cx.emit(Event::NewServerLogEntry {
332 id: language_server_id,
333 entry: line_before_message.to_string(),
334 is_rpc: true,
335 });
336 }
337
338 while rpc_log_lines.len() >= MAX_STORED_LOG_ENTRIES {
339 rpc_log_lines.pop_front();
340 }
341 let message = message.trim();
342 rpc_log_lines.push_back(message.to_string());
343 cx.emit(Event::NewServerLogEntry {
344 id: language_server_id,
345 entry: message.to_string(),
346 is_rpc: true,
347 });
348 cx.notify();
349 Some(())
350 }
351}
352
353impl LspLogView {
354 pub fn new(
355 project: Model<Project>,
356 log_store: Model<LogStore>,
357 cx: &mut ViewContext<Self>,
358 ) -> Self {
359 let server_id = log_store
360 .read(cx)
361 .projects
362 .get(&project.downgrade())
363 .and_then(|project| project.servers.keys().copied().next());
364 let model_changes_subscription = cx.observe(&log_store, |this, store, cx| {
365 (|| -> Option<()> {
366 let project_state = store.read(cx).projects.get(&this.project.downgrade())?;
367 if let Some(current_lsp) = this.current_server_id {
368 if !project_state.servers.contains_key(¤t_lsp) {
369 if let Some(server) = project_state.servers.iter().next() {
370 if this.is_showing_rpc_trace {
371 this.show_rpc_trace_for_server(*server.0, cx)
372 } else {
373 this.show_logs_for_server(*server.0, cx)
374 }
375 } else {
376 this.current_server_id = None;
377 this.editor.update(cx, |editor, cx| {
378 editor.set_read_only(false);
379 editor.clear(cx);
380 editor.set_read_only(true);
381 });
382 cx.notify();
383 }
384 }
385 } else {
386 if let Some(server) = project_state.servers.iter().next() {
387 if this.is_showing_rpc_trace {
388 this.show_rpc_trace_for_server(*server.0, cx)
389 } else {
390 this.show_logs_for_server(*server.0, cx)
391 }
392 }
393 }
394
395 Some(())
396 })();
397
398 cx.notify();
399 });
400 let events_subscriptions = cx.subscribe(&log_store, |log_view, _, e, cx| match e {
401 Event::NewServerLogEntry { id, entry, is_rpc } => {
402 if log_view.current_server_id == Some(*id) {
403 if (*is_rpc && log_view.is_showing_rpc_trace)
404 || (!*is_rpc && !log_view.is_showing_rpc_trace)
405 {
406 log_view.editor.update(cx, |editor, cx| {
407 editor.set_read_only(false);
408 editor.handle_input(entry.trim(), cx);
409 editor.handle_input("\n", cx);
410 editor.set_read_only(true);
411 });
412 }
413 }
414 }
415 });
416 let (editor, editor_subscription) = Self::editor_for_logs(String::new(), cx);
417
418 let focus_handle = cx.focus_handle();
419 let focus_subscription = cx.on_focus(&focus_handle, |log_view, cx| {
420 cx.focus_view(&log_view.editor);
421 });
422
423 let mut this = Self {
424 focus_handle,
425 editor,
426 editor_subscription,
427 project,
428 log_store,
429 current_server_id: None,
430 is_showing_rpc_trace: false,
431 _log_store_subscriptions: vec![
432 model_changes_subscription,
433 events_subscriptions,
434 focus_subscription,
435 ],
436 };
437 if let Some(server_id) = server_id {
438 this.show_logs_for_server(server_id, cx);
439 }
440 this
441 }
442
443 fn editor_for_logs(
444 log_contents: String,
445 cx: &mut ViewContext<Self>,
446 ) -> (View<Editor>, Subscription) {
447 let editor = cx.new_view(|cx| {
448 let mut editor = Editor::multi_line(cx);
449 editor.set_text(log_contents, cx);
450 editor.move_to_end(&MoveToEnd, cx);
451 editor.set_read_only(true);
452 editor.set_show_copilot_suggestions(false);
453 editor
454 });
455 let editor_subscription = cx.subscribe(
456 &editor,
457 |_, _, event: &EditorEvent, cx: &mut ViewContext<'_, LspLogView>| {
458 cx.emit(event.clone())
459 },
460 );
461 (editor, editor_subscription)
462 }
463
464 pub(crate) fn menu_items<'a>(&'a self, cx: &'a AppContext) -> Option<Vec<LogMenuItem>> {
465 let log_store = self.log_store.read(cx);
466 let state = log_store.projects.get(&self.project.downgrade())?;
467 let mut rows = self
468 .project
469 .read(cx)
470 .language_servers()
471 .filter_map(|(server_id, language_server_name, worktree_id)| {
472 let worktree = self.project.read(cx).worktree_for_id(worktree_id, cx)?;
473 let state = state.servers.get(&server_id)?;
474 Some(LogMenuItem {
475 server_id,
476 server_name: language_server_name,
477 worktree_root_name: worktree.read(cx).root_name().to_string(),
478 rpc_trace_enabled: state.rpc_state.is_some(),
479 rpc_trace_selected: self.is_showing_rpc_trace
480 && self.current_server_id == Some(server_id),
481 logs_selected: !self.is_showing_rpc_trace
482 && self.current_server_id == Some(server_id),
483 })
484 })
485 .chain(
486 self.project
487 .read(cx)
488 .supplementary_language_servers()
489 .filter_map(|(&server_id, (name, _))| {
490 let state = state.servers.get(&server_id)?;
491 Some(LogMenuItem {
492 server_id,
493 server_name: name.clone(),
494 worktree_root_name: "supplementary".to_string(),
495 rpc_trace_enabled: state.rpc_state.is_some(),
496 rpc_trace_selected: self.is_showing_rpc_trace
497 && self.current_server_id == Some(server_id),
498 logs_selected: !self.is_showing_rpc_trace
499 && self.current_server_id == Some(server_id),
500 })
501 }),
502 )
503 .collect::<Vec<_>>();
504 rows.sort_by_key(|row| row.server_id);
505 rows.dedup_by_key(|row| row.server_id);
506 Some(rows)
507 }
508
509 fn show_logs_for_server(&mut self, server_id: LanguageServerId, cx: &mut ViewContext<Self>) {
510 let log_contents = self
511 .log_store
512 .read(cx)
513 .server_logs(&self.project, server_id)
514 .map(log_contents);
515 if let Some(log_contents) = log_contents {
516 self.current_server_id = Some(server_id);
517 self.is_showing_rpc_trace = false;
518 let (editor, editor_subscription) = Self::editor_for_logs(log_contents, cx);
519 self.editor = editor;
520 self.editor_subscription = editor_subscription;
521 cx.notify();
522 }
523 cx.focus(&self.focus_handle);
524 }
525
526 fn show_rpc_trace_for_server(
527 &mut self,
528 server_id: LanguageServerId,
529 cx: &mut ViewContext<Self>,
530 ) {
531 let rpc_log = self.log_store.update(cx, |log_store, _| {
532 log_store
533 .enable_rpc_trace_for_language_server(&self.project, server_id)
534 .map(|state| log_contents(&state.rpc_messages))
535 });
536 if let Some(rpc_log) = rpc_log {
537 self.current_server_id = Some(server_id);
538 self.is_showing_rpc_trace = true;
539 let (editor, editor_subscription) = Self::editor_for_logs(rpc_log, cx);
540 let language = self.project.read(cx).languages().language_for_name("JSON");
541 editor
542 .read(cx)
543 .buffer()
544 .read(cx)
545 .as_singleton()
546 .expect("log buffer should be a singleton")
547 .update(cx, |_, cx| {
548 cx.spawn({
549 let buffer = cx.handle();
550 |_, mut cx| async move {
551 let language = language.await.ok();
552 buffer.update(&mut cx, |buffer, cx| {
553 buffer.set_language(language, cx);
554 })
555 }
556 })
557 .detach_and_log_err(cx);
558 });
559
560 self.editor = editor;
561 self.editor_subscription = editor_subscription;
562 cx.notify();
563 }
564
565 cx.focus(&self.focus_handle);
566 }
567
568 fn toggle_rpc_trace_for_server(
569 &mut self,
570 server_id: LanguageServerId,
571 enabled: bool,
572 cx: &mut ViewContext<Self>,
573 ) {
574 self.log_store.update(cx, |log_store, cx| {
575 if enabled {
576 log_store.enable_rpc_trace_for_language_server(&self.project, server_id);
577 } else {
578 log_store.disable_rpc_trace_for_language_server(&self.project, server_id, cx);
579 }
580 });
581 if !enabled && Some(server_id) == self.current_server_id {
582 self.show_logs_for_server(server_id, cx);
583 cx.notify();
584 }
585 }
586}
587
588fn log_contents(lines: &VecDeque<String>) -> String {
589 let (a, b) = lines.as_slices();
590 let log_contents = a.join("\n");
591 if b.is_empty() {
592 log_contents
593 } else {
594 log_contents + "\n" + &b.join("\n")
595 }
596}
597
598impl Render for LspLogView {
599 fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
600 self.editor
601 .update(cx, |editor, cx| editor.render(cx).into_any_element())
602 }
603}
604
605impl FocusableView for LspLogView {
606 fn focus_handle(&self, _: &AppContext) -> FocusHandle {
607 self.focus_handle.clone()
608 }
609}
610
611impl Item for LspLogView {
612 type Event = EditorEvent;
613
614 fn to_item_events(event: &Self::Event, f: impl FnMut(workspace::item::ItemEvent)) {
615 Editor::to_item_events(event, f)
616 }
617
618 fn tab_content(&self, _: Option<usize>, selected: bool, _: &WindowContext<'_>) -> AnyElement {
619 Label::new("LSP Logs")
620 .color(if selected {
621 Color::Default
622 } else {
623 Color::Muted
624 })
625 .into_any_element()
626 }
627
628 fn as_searchable(&self, handle: &View<Self>) -> Option<Box<dyn SearchableItemHandle>> {
629 Some(Box::new(handle.clone()))
630 }
631}
632
633impl SearchableItem for LspLogView {
634 type Match = <Editor as SearchableItem>::Match;
635
636 fn clear_matches(&mut self, cx: &mut ViewContext<Self>) {
637 self.editor.update(cx, |e, cx| e.clear_matches(cx))
638 }
639
640 fn update_matches(&mut self, matches: Vec<Self::Match>, cx: &mut ViewContext<Self>) {
641 self.editor
642 .update(cx, |e, cx| e.update_matches(matches, cx))
643 }
644
645 fn query_suggestion(&mut self, cx: &mut ViewContext<Self>) -> String {
646 self.editor.update(cx, |e, cx| e.query_suggestion(cx))
647 }
648
649 fn activate_match(
650 &mut self,
651 index: usize,
652 matches: Vec<Self::Match>,
653 cx: &mut ViewContext<Self>,
654 ) {
655 self.editor
656 .update(cx, |e, cx| e.activate_match(index, matches, cx))
657 }
658
659 fn select_matches(&mut self, matches: Vec<Self::Match>, cx: &mut ViewContext<Self>) {
660 self.editor
661 .update(cx, |e, cx| e.select_matches(matches, cx))
662 }
663
664 fn find_matches(
665 &mut self,
666 query: Arc<project::search::SearchQuery>,
667 cx: &mut ViewContext<Self>,
668 ) -> gpui::Task<Vec<Self::Match>> {
669 self.editor.update(cx, |e, cx| e.find_matches(query, cx))
670 }
671
672 fn replace(&mut self, _: &Self::Match, _: &SearchQuery, _: &mut ViewContext<Self>) {
673 // Since LSP Log is read-only, it doesn't make sense to support replace operation.
674 }
675 fn supported_options() -> workspace::searchable::SearchOptions {
676 workspace::searchable::SearchOptions {
677 case: true,
678 word: true,
679 regex: true,
680 // LSP log is read-only.
681 replacement: false,
682 }
683 }
684 fn active_match_index(
685 &mut self,
686 matches: Vec<Self::Match>,
687 cx: &mut ViewContext<Self>,
688 ) -> Option<usize> {
689 self.editor
690 .update(cx, |e, cx| e.active_match_index(matches, cx))
691 }
692}
693
694impl EventEmitter<ToolbarItemEvent> for LspLogToolbarItemView {}
695
696impl ToolbarItemView for LspLogToolbarItemView {
697 fn set_active_pane_item(
698 &mut self,
699 active_pane_item: Option<&dyn ItemHandle>,
700 cx: &mut ViewContext<Self>,
701 ) -> workspace::ToolbarItemLocation {
702 if let Some(item) = active_pane_item {
703 if let Some(log_view) = item.downcast::<LspLogView>() {
704 self.log_view = Some(log_view.clone());
705 self._log_view_subscription = Some(cx.observe(&log_view, |_, _, cx| {
706 cx.notify();
707 }));
708 return ToolbarItemLocation::PrimaryLeft;
709 }
710 }
711 self.log_view = None;
712 self._log_view_subscription = None;
713 ToolbarItemLocation::Hidden
714 }
715}
716
717impl Render for LspLogToolbarItemView {
718 fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
719 let Some(log_view) = self.log_view.clone() else {
720 return div();
721 };
722 let (menu_rows, current_server_id) = log_view.update(cx, |log_view, cx| {
723 let menu_rows = log_view.menu_items(cx).unwrap_or_default();
724 let current_server_id = log_view.current_server_id;
725 (menu_rows, current_server_id)
726 });
727
728 let current_server = current_server_id.and_then(|current_server_id| {
729 if let Ok(ix) = menu_rows.binary_search_by_key(¤t_server_id, |e| e.server_id) {
730 Some(menu_rows[ix].clone())
731 } else {
732 None
733 }
734 });
735
736 let log_toolbar_view = cx.view().clone();
737 let lsp_menu = popover_menu("LspLogView")
738 .anchor(AnchorCorner::TopLeft)
739 .trigger(Button::new(
740 "language_server_menu_header",
741 current_server
742 .and_then(|row| {
743 Some(Cow::Owned(format!(
744 "{} ({}) - {}",
745 row.server_name.0,
746 row.worktree_root_name,
747 if row.rpc_trace_selected {
748 RPC_MESSAGES
749 } else {
750 SERVER_LOGS
751 },
752 )))
753 })
754 .unwrap_or_else(|| "No server selected".into()),
755 ))
756 .menu(move |cx| {
757 let menu_rows = menu_rows.clone();
758 let log_view = log_view.clone();
759 let log_toolbar_view = log_toolbar_view.clone();
760 ContextMenu::build(cx, move |mut menu, cx| {
761 for (ix, row) in menu_rows.into_iter().enumerate() {
762 let server_selected = Some(row.server_id) == current_server_id;
763 menu = menu
764 .header(format!(
765 "{} ({})",
766 row.server_name.0, row.worktree_root_name
767 ))
768 .entry(
769 SERVER_LOGS,
770 None,
771 cx.handler_for(&log_view, move |view, cx| {
772 view.show_logs_for_server(row.server_id, cx);
773 }),
774 );
775 if server_selected && row.logs_selected {
776 let selected_ix = menu.select_last();
777 debug_assert_eq!(
778 Some(ix * 3 + 1),
779 selected_ix,
780 "Could not scroll to a just added LSP menu item"
781 );
782 }
783
784 menu = menu.custom_entry(
785 {
786 let log_toolbar_view = log_toolbar_view.clone();
787 move |cx| {
788 h_stack()
789 .w_full()
790 .justify_between()
791 .child(Label::new(RPC_MESSAGES))
792 .child(
793 div().z_index(120).child(
794 Checkbox::new(
795 ix,
796 if row.rpc_trace_enabled {
797 Selection::Selected
798 } else {
799 Selection::Unselected
800 },
801 )
802 .on_click(cx.listener_for(
803 &log_toolbar_view,
804 move |view, selection, cx| {
805 let enabled = matches!(
806 selection,
807 Selection::Selected
808 );
809 view.toggle_logging_for_server(
810 row.server_id,
811 enabled,
812 cx,
813 );
814 cx.stop_propagation();
815 },
816 )),
817 ),
818 )
819 .into_any_element()
820 }
821 },
822 cx.handler_for(&log_view, move |view, cx| {
823 view.show_rpc_trace_for_server(row.server_id, cx);
824 }),
825 );
826 if server_selected && row.rpc_trace_selected {
827 let selected_ix = menu.select_last();
828 debug_assert_eq!(
829 Some(ix * 3 + 2),
830 selected_ix,
831 "Could not scroll to a just added LSP menu item"
832 );
833 }
834 }
835 menu
836 })
837 .into()
838 });
839
840 h_stack().size_full().child(lsp_menu).child(
841 div()
842 .child(
843 Button::new("clear_log_button", "Clear").on_click(cx.listener(
844 |this, _, cx| {
845 if let Some(log_view) = this.log_view.as_ref() {
846 log_view.update(cx, |log_view, cx| {
847 log_view.editor.update(cx, |editor, cx| {
848 editor.set_read_only(false);
849 editor.clear(cx);
850 editor.set_read_only(true);
851 });
852 })
853 }
854 },
855 )),
856 )
857 .ml_2(),
858 )
859 }
860}
861
862const RPC_MESSAGES: &str = "RPC Messages";
863const SERVER_LOGS: &str = "Server Logs";
864
865impl LspLogToolbarItemView {
866 pub fn new() -> Self {
867 Self {
868 log_view: None,
869 _log_view_subscription: None,
870 }
871 }
872
873 fn toggle_logging_for_server(
874 &mut self,
875 id: LanguageServerId,
876 enabled: bool,
877 cx: &mut ViewContext<Self>,
878 ) {
879 if let Some(log_view) = &self.log_view {
880 log_view.update(cx, |log_view, cx| {
881 log_view.toggle_rpc_trace_for_server(id, enabled, cx);
882 if !enabled && Some(id) == log_view.current_server_id {
883 log_view.show_logs_for_server(id, cx);
884 cx.notify();
885 }
886 cx.focus(&log_view.focus_handle);
887 });
888 }
889 cx.notify();
890 }
891}
892
893pub enum Event {
894 NewServerLogEntry {
895 id: LanguageServerId,
896 entry: String,
897 is_rpc: bool,
898 },
899}
900
901impl EventEmitter<Event> for LogStore {}
902impl EventEmitter<Event> for LspLogView {}
903impl EventEmitter<EditorEvent> for LspLogView {}
904impl EventEmitter<SearchEvent> for LspLogView {}