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
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 fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
599 self.editor
600 .update(cx, |editor, cx| editor.render(cx).into_any_element())
601 }
602}
603
604impl FocusableView for LspLogView {
605 fn focus_handle(&self, _: &AppContext) -> FocusHandle {
606 self.focus_handle.clone()
607 }
608}
609
610impl Item for LspLogView {
611 type Event = EditorEvent;
612
613 fn to_item_events(event: &Self::Event, f: impl FnMut(workspace::item::ItemEvent)) {
614 Editor::to_item_events(event, f)
615 }
616
617 fn tab_content(&self, _: Option<usize>, selected: bool, _: &WindowContext<'_>) -> AnyElement {
618 Label::new("LSP Logs")
619 .color(if selected {
620 Color::Default
621 } else {
622 Color::Muted
623 })
624 .into_any_element()
625 }
626
627 fn as_searchable(&self, handle: &View<Self>) -> Option<Box<dyn SearchableItemHandle>> {
628 Some(Box::new(handle.clone()))
629 }
630}
631
632impl SearchableItem for LspLogView {
633 type Match = <Editor as SearchableItem>::Match;
634
635 fn clear_matches(&mut self, cx: &mut ViewContext<Self>) {
636 self.editor.update(cx, |e, cx| e.clear_matches(cx))
637 }
638
639 fn update_matches(&mut self, matches: Vec<Self::Match>, cx: &mut ViewContext<Self>) {
640 self.editor
641 .update(cx, |e, cx| e.update_matches(matches, cx))
642 }
643
644 fn query_suggestion(&mut self, cx: &mut ViewContext<Self>) -> String {
645 self.editor.update(cx, |e, cx| e.query_suggestion(cx))
646 }
647
648 fn activate_match(
649 &mut self,
650 index: usize,
651 matches: Vec<Self::Match>,
652 cx: &mut ViewContext<Self>,
653 ) {
654 self.editor
655 .update(cx, |e, cx| e.activate_match(index, matches, cx))
656 }
657
658 fn select_matches(&mut self, matches: Vec<Self::Match>, cx: &mut ViewContext<Self>) {
659 self.editor
660 .update(cx, |e, cx| e.select_matches(matches, cx))
661 }
662
663 fn find_matches(
664 &mut self,
665 query: Arc<project::search::SearchQuery>,
666 cx: &mut ViewContext<Self>,
667 ) -> gpui::Task<Vec<Self::Match>> {
668 self.editor.update(cx, |e, cx| e.find_matches(query, cx))
669 }
670
671 fn replace(&mut self, _: &Self::Match, _: &SearchQuery, _: &mut ViewContext<Self>) {
672 // Since LSP Log is read-only, it doesn't make sense to support replace operation.
673 }
674 fn supported_options() -> workspace::searchable::SearchOptions {
675 workspace::searchable::SearchOptions {
676 case: true,
677 word: true,
678 regex: true,
679 // LSP log is read-only.
680 replacement: false,
681 }
682 }
683 fn active_match_index(
684 &mut self,
685 matches: Vec<Self::Match>,
686 cx: &mut ViewContext<Self>,
687 ) -> Option<usize> {
688 self.editor
689 .update(cx, |e, cx| e.active_match_index(matches, cx))
690 }
691}
692
693impl EventEmitter<ToolbarItemEvent> for LspLogToolbarItemView {}
694
695impl ToolbarItemView for LspLogToolbarItemView {
696 fn set_active_pane_item(
697 &mut self,
698 active_pane_item: Option<&dyn ItemHandle>,
699 cx: &mut ViewContext<Self>,
700 ) -> workspace::ToolbarItemLocation {
701 if let Some(item) = active_pane_item {
702 if let Some(log_view) = item.downcast::<LspLogView>() {
703 self.log_view = Some(log_view.clone());
704 self._log_view_subscription = Some(cx.observe(&log_view, |_, _, cx| {
705 cx.notify();
706 }));
707 return ToolbarItemLocation::PrimaryLeft;
708 }
709 }
710 self.log_view = None;
711 self._log_view_subscription = None;
712 ToolbarItemLocation::Hidden
713 }
714}
715
716impl Render for LspLogToolbarItemView {
717 fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
718 let Some(log_view) = self.log_view.clone() else {
719 return div();
720 };
721 let (menu_rows, current_server_id) = log_view.update(cx, |log_view, cx| {
722 let menu_rows = log_view.menu_items(cx).unwrap_or_default();
723 let current_server_id = log_view.current_server_id;
724 (menu_rows, current_server_id)
725 });
726
727 let current_server = current_server_id.and_then(|current_server_id| {
728 if let Ok(ix) = menu_rows.binary_search_by_key(¤t_server_id, |e| e.server_id) {
729 Some(menu_rows[ix].clone())
730 } else {
731 None
732 }
733 });
734
735 let log_toolbar_view = cx.view().clone();
736 let lsp_menu = popover_menu("LspLogView")
737 .anchor(AnchorCorner::TopLeft)
738 .trigger(Button::new(
739 "language_server_menu_header",
740 current_server
741 .and_then(|row| {
742 Some(Cow::Owned(format!(
743 "{} ({}) - {}",
744 row.server_name.0,
745 row.worktree_root_name,
746 if row.rpc_trace_selected {
747 RPC_MESSAGES
748 } else {
749 SERVER_LOGS
750 },
751 )))
752 })
753 .unwrap_or_else(|| "No server selected".into()),
754 ))
755 .menu(move |cx| {
756 let menu_rows = menu_rows.clone();
757 let log_view = log_view.clone();
758 let log_toolbar_view = log_toolbar_view.clone();
759 ContextMenu::build(cx, move |mut menu, cx| {
760 for (ix, row) in menu_rows.into_iter().enumerate() {
761 let server_selected = Some(row.server_id) == current_server_id;
762 menu = menu
763 .header(format!(
764 "{} ({})",
765 row.server_name.0, row.worktree_root_name
766 ))
767 .entry(
768 SERVER_LOGS,
769 None,
770 cx.handler_for(&log_view, move |view, cx| {
771 view.show_logs_for_server(row.server_id, cx);
772 }),
773 );
774 if server_selected && row.logs_selected {
775 debug_assert_eq!(
776 Some(ix * 3 + 1),
777 menu.select_last(),
778 "Could not scroll to a just added LSP menu item"
779 );
780 }
781
782 menu = menu.custom_entry(
783 {
784 let log_toolbar_view = log_toolbar_view.clone();
785 move |cx| {
786 h_stack()
787 .w_full()
788 .justify_between()
789 .child(Label::new(RPC_MESSAGES))
790 .child(
791 div().z_index(120).child(
792 Checkbox::new(
793 ix,
794 if row.rpc_trace_enabled {
795 Selection::Selected
796 } else {
797 Selection::Unselected
798 },
799 )
800 .on_click(cx.listener_for(
801 &log_toolbar_view,
802 move |view, selection, cx| {
803 let enabled = matches!(
804 selection,
805 Selection::Selected
806 );
807 view.toggle_logging_for_server(
808 row.server_id,
809 enabled,
810 cx,
811 );
812 cx.stop_propagation();
813 },
814 )),
815 ),
816 )
817 .into_any_element()
818 }
819 },
820 cx.handler_for(&log_view, move |view, cx| {
821 view.show_rpc_trace_for_server(row.server_id, cx);
822 }),
823 );
824 if server_selected && row.rpc_trace_selected {
825 debug_assert_eq!(
826 Some(ix * 3 + 2),
827 menu.select_last(),
828 "Could not scroll to a just added LSP menu item"
829 );
830 }
831 }
832 menu
833 })
834 .into()
835 });
836
837 h_stack().size_full().child(lsp_menu).child(
838 div()
839 .child(
840 Button::new("clear_log_button", "Clear").on_click(cx.listener(
841 |this, _, cx| {
842 if let Some(log_view) = this.log_view.as_ref() {
843 log_view.update(cx, |log_view, cx| {
844 log_view.editor.update(cx, |editor, cx| {
845 editor.set_read_only(false);
846 editor.clear(cx);
847 editor.set_read_only(true);
848 });
849 })
850 }
851 },
852 )),
853 )
854 .ml_2(),
855 )
856 }
857}
858
859const RPC_MESSAGES: &str = "RPC Messages";
860const SERVER_LOGS: &str = "Server Logs";
861
862impl LspLogToolbarItemView {
863 pub fn new() -> Self {
864 Self {
865 log_view: None,
866 _log_view_subscription: None,
867 }
868 }
869
870 fn toggle_logging_for_server(
871 &mut self,
872 id: LanguageServerId,
873 enabled: bool,
874 cx: &mut ViewContext<Self>,
875 ) {
876 if let Some(log_view) = &self.log_view {
877 log_view.update(cx, |log_view, cx| {
878 log_view.toggle_rpc_trace_for_server(id, enabled, cx);
879 if !enabled && Some(id) == log_view.current_server_id {
880 log_view.show_logs_for_server(id, cx);
881 cx.notify();
882 }
883 cx.focus(&log_view.focus_handle);
884 });
885 }
886 cx.notify();
887 }
888}
889
890pub enum Event {
891 NewServerLogEntry {
892 id: LanguageServerId,
893 entry: String,
894 is_rpc: bool,
895 },
896}
897
898impl EventEmitter<Event> for LogStore {}
899impl EventEmitter<Event> for LspLogView {}
900impl EventEmitter<EditorEvent> for LspLogView {}
901impl EventEmitter<SearchEvent> for LspLogView {}