1use collections::{HashMap, VecDeque};
2use editor::{Editor, EditorElement, EditorEvent, MoveToEnd};
3use futures::{channel::mpsc, StreamExt};
4use gpui::{
5 actions, div, AnchorCorner, AnyElement, AppContext, Context, Div, 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::{h_stack, popover_menu, Button, Checkbox, Clickable, 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.build_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.build_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.build_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
453 });
454 let editor_subscription = cx.subscribe(
455 &editor,
456 |_, _, event: &EditorEvent, cx: &mut ViewContext<'_, LspLogView>| {
457 cx.emit(event.clone())
458 },
459 );
460 (editor, editor_subscription)
461 }
462
463 pub(crate) fn menu_items<'a>(&'a self, cx: &'a AppContext) -> Option<Vec<LogMenuItem>> {
464 let log_store = self.log_store.read(cx);
465 let state = log_store.projects.get(&self.project.downgrade())?;
466 let mut rows = self
467 .project
468 .read(cx)
469 .language_servers()
470 .filter_map(|(server_id, language_server_name, worktree_id)| {
471 let worktree = self.project.read(cx).worktree_for_id(worktree_id, cx)?;
472 let state = state.servers.get(&server_id)?;
473 Some(LogMenuItem {
474 server_id,
475 server_name: language_server_name,
476 worktree_root_name: worktree.read(cx).root_name().to_string(),
477 rpc_trace_enabled: state.rpc_state.is_some(),
478 rpc_trace_selected: self.is_showing_rpc_trace
479 && self.current_server_id == Some(server_id),
480 logs_selected: !self.is_showing_rpc_trace
481 && self.current_server_id == Some(server_id),
482 })
483 })
484 .chain(
485 self.project
486 .read(cx)
487 .supplementary_language_servers()
488 .filter_map(|(&server_id, (name, _))| {
489 let state = state.servers.get(&server_id)?;
490 Some(LogMenuItem {
491 server_id,
492 server_name: name.clone(),
493 worktree_root_name: "supplementary".to_string(),
494 rpc_trace_enabled: state.rpc_state.is_some(),
495 rpc_trace_selected: self.is_showing_rpc_trace
496 && self.current_server_id == Some(server_id),
497 logs_selected: !self.is_showing_rpc_trace
498 && self.current_server_id == Some(server_id),
499 })
500 }),
501 )
502 .collect::<Vec<_>>();
503 rows.sort_by_key(|row| row.server_id);
504 rows.dedup_by_key(|row| row.server_id);
505 Some(rows)
506 }
507
508 fn show_logs_for_server(&mut self, server_id: LanguageServerId, cx: &mut ViewContext<Self>) {
509 let log_contents = self
510 .log_store
511 .read(cx)
512 .server_logs(&self.project, server_id)
513 .map(log_contents);
514 if let Some(log_contents) = log_contents {
515 self.current_server_id = Some(server_id);
516 self.is_showing_rpc_trace = false;
517 let (editor, editor_subscription) = Self::editor_for_logs(log_contents, cx);
518 self.editor = editor;
519 self.editor_subscription = editor_subscription;
520 cx.notify();
521 }
522 cx.focus(&self.focus_handle);
523 }
524
525 fn show_rpc_trace_for_server(
526 &mut self,
527 server_id: LanguageServerId,
528 cx: &mut ViewContext<Self>,
529 ) {
530 let rpc_log = self.log_store.update(cx, |log_store, _| {
531 log_store
532 .enable_rpc_trace_for_language_server(&self.project, server_id)
533 .map(|state| log_contents(&state.rpc_messages))
534 });
535 if let Some(rpc_log) = rpc_log {
536 self.current_server_id = Some(server_id);
537 self.is_showing_rpc_trace = true;
538 let (editor, editor_subscription) = Self::editor_for_logs(rpc_log, cx);
539 let language = self.project.read(cx).languages().language_for_name("JSON");
540 editor
541 .read(cx)
542 .buffer()
543 .read(cx)
544 .as_singleton()
545 .expect("log buffer should be a singleton")
546 .update(cx, |_, cx| {
547 cx.spawn({
548 let buffer = cx.handle();
549 |_, mut cx| async move {
550 let language = language.await.ok();
551 buffer.update(&mut cx, |buffer, cx| {
552 buffer.set_language(language, cx);
553 })
554 }
555 })
556 .detach_and_log_err(cx);
557 });
558
559 self.editor = editor;
560 self.editor_subscription = editor_subscription;
561 cx.notify();
562 }
563
564 cx.focus(&self.focus_handle);
565 }
566
567 fn toggle_rpc_trace_for_server(
568 &mut self,
569 server_id: LanguageServerId,
570 enabled: bool,
571 cx: &mut ViewContext<Self>,
572 ) {
573 self.log_store.update(cx, |log_store, cx| {
574 if enabled {
575 log_store.enable_rpc_trace_for_language_server(&self.project, server_id);
576 } else {
577 log_store.disable_rpc_trace_for_language_server(&self.project, server_id, cx);
578 }
579 });
580 if !enabled && Some(server_id) == self.current_server_id {
581 self.show_logs_for_server(server_id, cx);
582 cx.notify();
583 }
584 }
585}
586
587fn log_contents(lines: &VecDeque<String>) -> String {
588 let (a, b) = lines.as_slices();
589 let log_contents = a.join("\n");
590 if b.is_empty() {
591 log_contents
592 } else {
593 log_contents + "\n" + &b.join("\n")
594 }
595}
596
597impl Render for LspLogView {
598 type Element = EditorElement;
599
600 fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
601 self.editor.update(cx, |editor, cx| editor.render(cx))
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>, _: bool, _: &WindowContext<'_>) -> AnyElement {
619 Label::new("LSP Logs").into_any_element()
620 }
621
622 fn as_searchable(&self, handle: &View<Self>) -> Option<Box<dyn SearchableItemHandle>> {
623 Some(Box::new(handle.clone()))
624 }
625}
626
627impl SearchableItem for LspLogView {
628 type Match = <Editor as SearchableItem>::Match;
629
630 fn clear_matches(&mut self, cx: &mut ViewContext<Self>) {
631 self.editor.update(cx, |e, cx| e.clear_matches(cx))
632 }
633
634 fn update_matches(&mut self, matches: Vec<Self::Match>, cx: &mut ViewContext<Self>) {
635 self.editor
636 .update(cx, |e, cx| e.update_matches(matches, cx))
637 }
638
639 fn query_suggestion(&mut self, cx: &mut ViewContext<Self>) -> String {
640 self.editor.update(cx, |e, cx| e.query_suggestion(cx))
641 }
642
643 fn activate_match(
644 &mut self,
645 index: usize,
646 matches: Vec<Self::Match>,
647 cx: &mut ViewContext<Self>,
648 ) {
649 self.editor
650 .update(cx, |e, cx| e.activate_match(index, matches, cx))
651 }
652
653 fn select_matches(&mut self, matches: Vec<Self::Match>, cx: &mut ViewContext<Self>) {
654 self.editor
655 .update(cx, |e, cx| e.select_matches(matches, cx))
656 }
657
658 fn find_matches(
659 &mut self,
660 query: Arc<project::search::SearchQuery>,
661 cx: &mut ViewContext<Self>,
662 ) -> gpui::Task<Vec<Self::Match>> {
663 self.editor.update(cx, |e, cx| e.find_matches(query, cx))
664 }
665
666 fn replace(&mut self, _: &Self::Match, _: &SearchQuery, _: &mut ViewContext<Self>) {
667 // Since LSP Log is read-only, it doesn't make sense to support replace operation.
668 }
669 fn supported_options() -> workspace::searchable::SearchOptions {
670 workspace::searchable::SearchOptions {
671 case: true,
672 word: true,
673 regex: true,
674 // LSP log is read-only.
675 replacement: false,
676 }
677 }
678 fn active_match_index(
679 &mut self,
680 matches: Vec<Self::Match>,
681 cx: &mut ViewContext<Self>,
682 ) -> Option<usize> {
683 self.editor
684 .update(cx, |e, cx| e.active_match_index(matches, cx))
685 }
686}
687
688impl EventEmitter<ToolbarItemEvent> for LspLogToolbarItemView {}
689
690impl ToolbarItemView for LspLogToolbarItemView {
691 fn set_active_pane_item(
692 &mut self,
693 active_pane_item: Option<&dyn ItemHandle>,
694 cx: &mut ViewContext<Self>,
695 ) -> workspace::ToolbarItemLocation {
696 if let Some(item) = active_pane_item {
697 if let Some(log_view) = item.downcast::<LspLogView>() {
698 self.log_view = Some(log_view.clone());
699 self._log_view_subscription = Some(cx.observe(&log_view, |_, _, cx| {
700 cx.notify();
701 }));
702 return ToolbarItemLocation::PrimaryLeft;
703 }
704 }
705 self.log_view = None;
706 self._log_view_subscription = None;
707 ToolbarItemLocation::Hidden
708 }
709}
710
711impl Render for LspLogToolbarItemView {
712 type Element = Div;
713
714 fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
715 let Some(log_view) = self.log_view.clone() else {
716 return div();
717 };
718 let (menu_rows, current_server_id) = log_view.update(cx, |log_view, cx| {
719 let menu_rows = log_view.menu_items(cx).unwrap_or_default();
720 let current_server_id = log_view.current_server_id;
721 (menu_rows, current_server_id)
722 });
723
724 let current_server = current_server_id.and_then(|current_server_id| {
725 if let Ok(ix) = menu_rows.binary_search_by_key(¤t_server_id, |e| e.server_id) {
726 Some(menu_rows[ix].clone())
727 } else {
728 None
729 }
730 });
731
732 let log_toolbar_view = cx.view().clone();
733 let lsp_menu = popover_menu("LspLogView")
734 .anchor(AnchorCorner::TopLeft)
735 .trigger(Button::new(
736 "language_server_menu_header",
737 current_server
738 .and_then(|row| {
739 Some(Cow::Owned(format!(
740 "{} ({}) - {}",
741 row.server_name.0,
742 row.worktree_root_name,
743 if row.rpc_trace_selected {
744 RPC_MESSAGES
745 } else {
746 SERVER_LOGS
747 },
748 )))
749 })
750 .unwrap_or_else(|| "No server selected".into()),
751 ))
752 .menu(move |cx| {
753 let menu_rows = menu_rows.clone();
754 let log_view = log_view.clone();
755 let log_toolbar_view = log_toolbar_view.clone();
756 ContextMenu::build(cx, move |mut menu, cx| {
757 for (ix, row) in menu_rows.into_iter().enumerate() {
758 let server_selected = Some(row.server_id) == current_server_id;
759 menu = menu
760 .header(format!(
761 "{} ({})",
762 row.server_name.0, row.worktree_root_name
763 ))
764 .entry(
765 SERVER_LOGS,
766 cx.handler_for(&log_view, move |view, cx| {
767 view.show_logs_for_server(row.server_id, cx);
768 }),
769 );
770 if server_selected && row.logs_selected {
771 debug_assert_eq!(
772 Some(ix * 3 + 1),
773 menu.select_last(),
774 "Could not scroll to a just added LSP menu item"
775 );
776 }
777
778 menu = menu.custom_entry(
779 {
780 let log_toolbar_view = log_toolbar_view.clone();
781 move |cx| {
782 h_stack()
783 .w_full()
784 .justify_between()
785 .child(Label::new(RPC_MESSAGES))
786 .child(
787 div().z_index(120).child(
788 Checkbox::new(
789 ix,
790 if row.rpc_trace_enabled {
791 Selection::Selected
792 } else {
793 Selection::Unselected
794 },
795 )
796 .on_click(cx.listener_for(
797 &log_toolbar_view,
798 move |view, selection, cx| {
799 let enabled = matches!(
800 selection,
801 Selection::Selected
802 );
803 view.toggle_logging_for_server(
804 row.server_id,
805 enabled,
806 cx,
807 );
808 cx.stop_propagation();
809 },
810 )),
811 ),
812 )
813 .into_any_element()
814 }
815 },
816 cx.handler_for(&log_view, move |view, cx| {
817 view.show_rpc_trace_for_server(row.server_id, cx);
818 }),
819 );
820 if server_selected && row.rpc_trace_selected {
821 debug_assert_eq!(
822 Some(ix * 3 + 2),
823 menu.select_last(),
824 "Could not scroll to a just added LSP menu item"
825 );
826 }
827 }
828 menu
829 })
830 });
831
832 h_stack().size_full().child(lsp_menu).child(
833 div()
834 .child(
835 Button::new("clear_log_button", "Clear").on_click(cx.listener(
836 |this, _, cx| {
837 if let Some(log_view) = this.log_view.as_ref() {
838 log_view.update(cx, |log_view, cx| {
839 log_view.editor.update(cx, |editor, cx| {
840 editor.set_read_only(false);
841 editor.clear(cx);
842 editor.set_read_only(true);
843 });
844 })
845 }
846 },
847 )),
848 )
849 .ml_2(),
850 )
851 }
852}
853
854const RPC_MESSAGES: &str = "RPC Messages";
855const SERVER_LOGS: &str = "Server Logs";
856
857impl LspLogToolbarItemView {
858 pub fn new() -> Self {
859 Self {
860 log_view: None,
861 _log_view_subscription: None,
862 }
863 }
864
865 fn toggle_logging_for_server(
866 &mut self,
867 id: LanguageServerId,
868 enabled: bool,
869 cx: &mut ViewContext<Self>,
870 ) {
871 if let Some(log_view) = &self.log_view {
872 log_view.update(cx, |log_view, cx| {
873 log_view.toggle_rpc_trace_for_server(id, enabled, cx);
874 if !enabled && Some(id) == log_view.current_server_id {
875 log_view.show_logs_for_server(id, cx);
876 cx.notify();
877 }
878 cx.focus(&log_view.focus_handle);
879 });
880 }
881 cx.notify();
882 }
883}
884
885pub enum Event {
886 NewServerLogEntry {
887 id: LanguageServerId,
888 entry: String,
889 is_rpc: bool,
890 },
891}
892
893impl EventEmitter<Event> for LogStore {}
894impl EventEmitter<Event> for LspLogView {}
895impl EventEmitter<EditorEvent> for LspLogView {}
896impl EventEmitter<SearchEvent> for LspLogView {}