1use collections::{HashMap, VecDeque};
2use copilot::Copilot;
3use editor::{actions::MoveToEnd, scroll::Autoscroll, Editor, EditorEvent};
4use futures::{channel::mpsc, StreamExt};
5use gpui::{
6 actions, div, AppContext, Context, Corner, EventEmitter, FocusHandle, FocusableView,
7 IntoElement, Model, ModelContext, ParentElement, Render, Styled, Subscription, View,
8 ViewContext, VisualContext, WeakModel, WindowContext,
9};
10use language::{language_settings::SoftWrap, LanguageServerId};
11use lsp::{
12 notification::SetTrace, IoKind, LanguageServer, LanguageServerName, MessageType,
13 SetTraceParams, TraceValue,
14};
15use project::{search::SearchQuery, Project, WorktreeId};
16use std::{borrow::Cow, sync::Arc};
17use ui::{prelude::*, Button, Checkbox, ContextMenu, Label, PopoverMenu, ToggleState};
18use workspace::{
19 item::{Item, ItemHandle},
20 searchable::{SearchEvent, SearchableItem, SearchableItemHandle},
21 SplitDirection, ToolbarItemEvent, ToolbarItemLocation, ToolbarItemView, Workspace, WorkspaceId,
22};
23
24const SEND_LINE: &str = "// Send:";
25const RECEIVE_LINE: &str = "// Receive:";
26const MAX_STORED_LOG_ENTRIES: usize = 2000;
27
28pub struct LogStore {
29 projects: HashMap<WeakModel<Project>, ProjectState>,
30 language_servers: HashMap<LanguageServerId, LanguageServerState>,
31 copilot_log_subscription: Option<lsp::Subscription>,
32 _copilot_subscription: Option<gpui::Subscription>,
33 io_tx: mpsc::UnboundedSender<(LanguageServerId, IoKind, String)>,
34}
35
36struct ProjectState {
37 _subscriptions: [gpui::Subscription; 2],
38}
39
40trait Message: AsRef<str> {
41 type Level: Copy + std::fmt::Debug;
42 fn should_include(&self, _: Self::Level) -> bool {
43 true
44 }
45}
46
47struct LogMessage {
48 message: String,
49 typ: MessageType,
50}
51
52impl AsRef<str> for LogMessage {
53 fn as_ref(&self) -> &str {
54 &self.message
55 }
56}
57
58impl Message for LogMessage {
59 type Level = MessageType;
60
61 fn should_include(&self, level: Self::Level) -> bool {
62 match (self.typ, level) {
63 (MessageType::ERROR, _) => true,
64 (_, MessageType::ERROR) => false,
65 (MessageType::WARNING, _) => true,
66 (_, MessageType::WARNING) => false,
67 (MessageType::INFO, _) => true,
68 (_, MessageType::INFO) => false,
69 _ => true,
70 }
71 }
72}
73
74struct TraceMessage {
75 message: String,
76}
77
78impl AsRef<str> for TraceMessage {
79 fn as_ref(&self) -> &str {
80 &self.message
81 }
82}
83
84impl Message for TraceMessage {
85 type Level = ();
86}
87
88struct RpcMessage {
89 message: String,
90}
91
92impl AsRef<str> for RpcMessage {
93 fn as_ref(&self) -> &str {
94 &self.message
95 }
96}
97
98impl Message for RpcMessage {
99 type Level = ();
100}
101
102struct LanguageServerState {
103 name: Option<LanguageServerName>,
104 worktree_id: Option<WorktreeId>,
105 kind: LanguageServerKind,
106 log_messages: VecDeque<LogMessage>,
107 trace_messages: VecDeque<TraceMessage>,
108 rpc_state: Option<LanguageServerRpcState>,
109 trace_level: TraceValue,
110 log_level: MessageType,
111 io_logs_subscription: Option<lsp::Subscription>,
112}
113
114#[derive(PartialEq, Clone)]
115pub enum LanguageServerKind {
116 Local { project: WeakModel<Project> },
117 Remote { project: WeakModel<Project> },
118 Global,
119}
120
121impl LanguageServerKind {
122 fn is_remote(&self) -> bool {
123 matches!(self, LanguageServerKind::Remote { .. })
124 }
125}
126
127impl std::fmt::Debug for LanguageServerKind {
128 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
129 match self {
130 LanguageServerKind::Local { .. } => write!(f, "LanguageServerKind::Local"),
131 LanguageServerKind::Remote { .. } => write!(f, "LanguageServerKind::Remote"),
132 LanguageServerKind::Global => write!(f, "LanguageServerKind::Global"),
133 }
134 }
135}
136
137impl LanguageServerKind {
138 fn project(&self) -> Option<&WeakModel<Project>> {
139 match self {
140 Self::Local { project } => Some(project),
141 Self::Remote { project } => Some(project),
142 Self::Global { .. } => None,
143 }
144 }
145}
146
147struct LanguageServerRpcState {
148 rpc_messages: VecDeque<RpcMessage>,
149 last_message_kind: Option<MessageKind>,
150}
151
152pub struct LspLogView {
153 pub(crate) editor: View<Editor>,
154 editor_subscriptions: Vec<Subscription>,
155 log_store: Model<LogStore>,
156 current_server_id: Option<LanguageServerId>,
157 active_entry_kind: LogKind,
158 project: Model<Project>,
159 focus_handle: FocusHandle,
160 _log_store_subscriptions: Vec<Subscription>,
161}
162
163pub struct LspLogToolbarItemView {
164 log_view: Option<View<LspLogView>>,
165 _log_view_subscription: Option<Subscription>,
166}
167
168#[derive(Copy, Clone, PartialEq, Eq)]
169enum MessageKind {
170 Send,
171 Receive,
172}
173
174#[derive(Clone, Copy, Debug, Default, PartialEq)]
175pub enum LogKind {
176 Rpc,
177 Trace,
178 #[default]
179 Logs,
180 ServerInfo,
181}
182
183impl LogKind {
184 fn label(&self) -> &'static str {
185 match self {
186 LogKind::Rpc => RPC_MESSAGES,
187 LogKind::Trace => SERVER_TRACE,
188 LogKind::Logs => SERVER_LOGS,
189 LogKind::ServerInfo => SERVER_INFO,
190 }
191 }
192}
193
194#[derive(Clone, Debug, PartialEq)]
195pub(crate) struct LogMenuItem {
196 pub server_id: LanguageServerId,
197 pub server_name: LanguageServerName,
198 pub worktree_root_name: String,
199 pub rpc_trace_enabled: bool,
200 pub selected_entry: LogKind,
201 pub trace_level: lsp::TraceValue,
202 pub server_kind: LanguageServerKind,
203}
204
205actions!(debug, [OpenLanguageServerLogs]);
206
207pub fn init(cx: &mut AppContext) {
208 let log_store = cx.new_model(LogStore::new);
209
210 cx.observe_new_views(move |workspace: &mut Workspace, cx| {
211 let project = workspace.project();
212 if project.read(cx).is_local() || project.read(cx).is_via_ssh() {
213 log_store.update(cx, |store, cx| {
214 store.add_project(project, cx);
215 });
216 }
217
218 let log_store = log_store.clone();
219 workspace.register_action(move |workspace, _: &OpenLanguageServerLogs, cx| {
220 let project = workspace.project().read(cx);
221 if project.is_local() || project.is_via_ssh() {
222 workspace.split_item(
223 SplitDirection::Right,
224 Box::new(cx.new_view(|cx| {
225 LspLogView::new(workspace.project().clone(), log_store.clone(), cx)
226 })),
227 cx,
228 );
229 }
230 });
231 })
232 .detach();
233}
234
235impl LogStore {
236 pub fn new(cx: &mut ModelContext<Self>) -> Self {
237 let (io_tx, mut io_rx) = mpsc::unbounded();
238
239 let copilot_subscription = Copilot::global(cx).map(|copilot| {
240 let copilot = &copilot;
241 cx.subscribe(copilot, |this, copilot, inline_completion_event, cx| {
242 if let copilot::Event::CopilotLanguageServerStarted = inline_completion_event {
243 if let Some(server) = copilot.read(cx).language_server() {
244 let server_id = server.server_id();
245 let weak_this = cx.weak_model();
246 this.copilot_log_subscription =
247 Some(server.on_notification::<copilot::request::LogMessage, _>(
248 move |params, mut cx| {
249 weak_this
250 .update(&mut cx, |this, cx| {
251 this.add_language_server_log(
252 server_id,
253 MessageType::LOG,
254 ¶ms.message,
255 cx,
256 );
257 })
258 .ok();
259 },
260 ));
261 let name = LanguageServerName::new_static("copilot");
262 this.add_language_server(
263 LanguageServerKind::Global,
264 server.server_id(),
265 Some(name),
266 None,
267 Some(server.clone()),
268 cx,
269 );
270 }
271 }
272 })
273 });
274
275 let this = Self {
276 copilot_log_subscription: None,
277 _copilot_subscription: copilot_subscription,
278 projects: HashMap::default(),
279 language_servers: HashMap::default(),
280 io_tx,
281 };
282
283 cx.spawn(|this, mut cx| async move {
284 while let Some((server_id, io_kind, message)) = io_rx.next().await {
285 if let Some(this) = this.upgrade() {
286 this.update(&mut cx, |this, cx| {
287 this.on_io(server_id, io_kind, &message, cx);
288 })?;
289 }
290 }
291 anyhow::Ok(())
292 })
293 .detach_and_log_err(cx);
294 this
295 }
296
297 pub fn add_project(&mut self, project: &Model<Project>, cx: &mut ModelContext<Self>) {
298 let weak_project = project.downgrade();
299 self.projects.insert(
300 project.downgrade(),
301 ProjectState {
302 _subscriptions: [
303 cx.observe_release(project, move |this, _, _| {
304 this.projects.remove(&weak_project);
305 this.language_servers
306 .retain(|_, state| state.kind.project() != Some(&weak_project));
307 }),
308 cx.subscribe(project, |this, project, event, cx| {
309 let server_kind = if project.read(cx).is_via_ssh() {
310 LanguageServerKind::Remote {
311 project: project.downgrade(),
312 }
313 } else {
314 LanguageServerKind::Local {
315 project: project.downgrade(),
316 }
317 };
318
319 match event {
320 project::Event::LanguageServerAdded(id, name, worktree_id) => {
321 this.add_language_server(
322 server_kind,
323 *id,
324 Some(name.clone()),
325 *worktree_id,
326 project
327 .read(cx)
328 .lsp_store()
329 .read(cx)
330 .language_server_for_id(*id),
331 cx,
332 );
333 }
334 project::Event::LanguageServerRemoved(id) => {
335 this.remove_language_server(*id, cx);
336 }
337 project::Event::LanguageServerLog(id, typ, message) => {
338 this.add_language_server(server_kind, *id, None, None, None, cx);
339 match typ {
340 project::LanguageServerLogType::Log(typ) => {
341 this.add_language_server_log(*id, *typ, message, cx);
342 }
343 project::LanguageServerLogType::Trace(_) => {
344 this.add_language_server_trace(*id, message, cx);
345 }
346 }
347 }
348 _ => {}
349 }
350 }),
351 ],
352 },
353 );
354 }
355
356 fn get_language_server_state(
357 &mut self,
358 id: LanguageServerId,
359 ) -> Option<&mut LanguageServerState> {
360 self.language_servers.get_mut(&id)
361 }
362
363 fn add_language_server(
364 &mut self,
365 kind: LanguageServerKind,
366 server_id: LanguageServerId,
367 name: Option<LanguageServerName>,
368 worktree_id: Option<WorktreeId>,
369 server: Option<Arc<LanguageServer>>,
370 cx: &mut ModelContext<Self>,
371 ) -> Option<&mut LanguageServerState> {
372 let server_state = self.language_servers.entry(server_id).or_insert_with(|| {
373 cx.notify();
374 LanguageServerState {
375 name: None,
376 worktree_id: None,
377 kind,
378 rpc_state: None,
379 log_messages: VecDeque::with_capacity(MAX_STORED_LOG_ENTRIES),
380 trace_messages: VecDeque::with_capacity(MAX_STORED_LOG_ENTRIES),
381 trace_level: TraceValue::Off,
382 log_level: MessageType::LOG,
383 io_logs_subscription: None,
384 }
385 });
386
387 if let Some(name) = name {
388 server_state.name = Some(name);
389 }
390 if let Some(worktree_id) = worktree_id {
391 server_state.worktree_id = Some(worktree_id);
392 }
393
394 if let Some(server) = server
395 .clone()
396 .filter(|_| server_state.io_logs_subscription.is_none())
397 {
398 let io_tx = self.io_tx.clone();
399 let server_id = server.server_id();
400 server_state.io_logs_subscription = Some(server.on_io(move |io_kind, message| {
401 io_tx
402 .unbounded_send((server_id, io_kind, message.to_string()))
403 .ok();
404 }));
405 }
406
407 Some(server_state)
408 }
409
410 fn add_language_server_log(
411 &mut self,
412 id: LanguageServerId,
413 typ: MessageType,
414 message: &str,
415 cx: &mut ModelContext<Self>,
416 ) -> Option<()> {
417 let language_server_state = self.get_language_server_state(id)?;
418
419 let log_lines = &mut language_server_state.log_messages;
420 Self::add_language_server_message(
421 log_lines,
422 id,
423 LogMessage {
424 message: message.trim_end().to_string(),
425 typ,
426 },
427 language_server_state.log_level,
428 LogKind::Logs,
429 cx,
430 );
431 Some(())
432 }
433
434 fn add_language_server_trace(
435 &mut self,
436 id: LanguageServerId,
437 message: &str,
438 cx: &mut ModelContext<Self>,
439 ) -> Option<()> {
440 let language_server_state = self.get_language_server_state(id)?;
441
442 let log_lines = &mut language_server_state.trace_messages;
443 Self::add_language_server_message(
444 log_lines,
445 id,
446 TraceMessage {
447 message: message.trim_end().to_string(),
448 },
449 (),
450 LogKind::Trace,
451 cx,
452 );
453 Some(())
454 }
455
456 fn add_language_server_message<T: Message>(
457 log_lines: &mut VecDeque<T>,
458 id: LanguageServerId,
459 message: T,
460 current_severity: <T as Message>::Level,
461 kind: LogKind,
462 cx: &mut ModelContext<Self>,
463 ) {
464 while log_lines.len() >= MAX_STORED_LOG_ENTRIES {
465 log_lines.pop_front();
466 }
467 let entry: &str = message.as_ref();
468 let entry = entry.to_string();
469 let visible = message.should_include(current_severity);
470 log_lines.push_back(message);
471
472 if visible {
473 cx.emit(Event::NewServerLogEntry { id, entry, kind });
474 cx.notify();
475 }
476 }
477
478 fn remove_language_server(&mut self, id: LanguageServerId, cx: &mut ModelContext<Self>) {
479 self.language_servers.remove(&id);
480 cx.notify();
481 }
482
483 fn server_logs(&self, server_id: LanguageServerId) -> Option<&VecDeque<LogMessage>> {
484 Some(&self.language_servers.get(&server_id)?.log_messages)
485 }
486
487 fn server_trace(&self, server_id: LanguageServerId) -> Option<&VecDeque<TraceMessage>> {
488 Some(&self.language_servers.get(&server_id)?.trace_messages)
489 }
490
491 fn server_ids_for_project<'a>(
492 &'a self,
493 lookup_project: &'a WeakModel<Project>,
494 ) -> impl Iterator<Item = LanguageServerId> + 'a {
495 self.language_servers
496 .iter()
497 .filter_map(move |(id, state)| match &state.kind {
498 LanguageServerKind::Local { project } | LanguageServerKind::Remote { project } => {
499 if project == lookup_project {
500 Some(*id)
501 } else {
502 None
503 }
504 }
505 LanguageServerKind::Global => Some(*id),
506 })
507 }
508
509 fn enable_rpc_trace_for_language_server(
510 &mut self,
511 server_id: LanguageServerId,
512 ) -> Option<&mut LanguageServerRpcState> {
513 let rpc_state = self
514 .language_servers
515 .get_mut(&server_id)?
516 .rpc_state
517 .get_or_insert_with(|| LanguageServerRpcState {
518 rpc_messages: VecDeque::with_capacity(MAX_STORED_LOG_ENTRIES),
519 last_message_kind: None,
520 });
521 Some(rpc_state)
522 }
523
524 pub fn disable_rpc_trace_for_language_server(
525 &mut self,
526 server_id: LanguageServerId,
527 ) -> Option<()> {
528 self.language_servers.get_mut(&server_id)?.rpc_state.take();
529 Some(())
530 }
531
532 fn on_io(
533 &mut self,
534 language_server_id: LanguageServerId,
535 io_kind: IoKind,
536 message: &str,
537 cx: &mut ModelContext<Self>,
538 ) -> Option<()> {
539 let is_received = match io_kind {
540 IoKind::StdOut => true,
541 IoKind::StdIn => false,
542 IoKind::StdErr => {
543 let message = format!("stderr: {}", message.trim());
544 self.add_language_server_log(language_server_id, MessageType::LOG, &message, cx);
545 return Some(());
546 }
547 };
548
549 let state = self
550 .get_language_server_state(language_server_id)?
551 .rpc_state
552 .as_mut()?;
553 let kind = if is_received {
554 MessageKind::Receive
555 } else {
556 MessageKind::Send
557 };
558
559 let rpc_log_lines = &mut state.rpc_messages;
560 if state.last_message_kind != Some(kind) {
561 let line_before_message = match kind {
562 MessageKind::Send => SEND_LINE,
563 MessageKind::Receive => RECEIVE_LINE,
564 };
565 rpc_log_lines.push_back(RpcMessage {
566 message: line_before_message.to_string(),
567 });
568 cx.emit(Event::NewServerLogEntry {
569 id: language_server_id,
570 entry: line_before_message.to_string(),
571 kind: LogKind::Rpc,
572 });
573 }
574
575 while rpc_log_lines.len() >= MAX_STORED_LOG_ENTRIES {
576 rpc_log_lines.pop_front();
577 }
578 let message = message.trim();
579 rpc_log_lines.push_back(RpcMessage {
580 message: message.to_string(),
581 });
582 cx.emit(Event::NewServerLogEntry {
583 id: language_server_id,
584 entry: message.to_string(),
585 kind: LogKind::Rpc,
586 });
587 cx.notify();
588 Some(())
589 }
590}
591
592impl LspLogView {
593 pub fn new(
594 project: Model<Project>,
595 log_store: Model<LogStore>,
596 cx: &mut ViewContext<Self>,
597 ) -> Self {
598 let server_id = log_store
599 .read(cx)
600 .language_servers
601 .iter()
602 .find(|(_, server)| server.kind.project() == Some(&project.downgrade()))
603 .map(|(id, _)| *id);
604
605 let weak_project = project.downgrade();
606 let model_changes_subscription = cx.observe(&log_store, move |this, store, cx| {
607 let first_server_id_for_project =
608 store.read(cx).server_ids_for_project(&weak_project).next();
609 if let Some(current_lsp) = this.current_server_id {
610 if !store.read(cx).language_servers.contains_key(¤t_lsp) {
611 if let Some(server_id) = first_server_id_for_project {
612 match this.active_entry_kind {
613 LogKind::Rpc => this.show_rpc_trace_for_server(server_id, cx),
614 LogKind::Trace => this.show_trace_for_server(server_id, cx),
615 LogKind::Logs => this.show_logs_for_server(server_id, cx),
616 LogKind::ServerInfo => this.show_server_info(server_id, cx),
617 }
618 } else {
619 this.current_server_id = None;
620 this.editor.update(cx, |editor, cx| {
621 editor.set_read_only(false);
622 editor.clear(cx);
623 editor.set_read_only(true);
624 });
625 cx.notify();
626 }
627 }
628 } else if let Some(server_id) = first_server_id_for_project {
629 match this.active_entry_kind {
630 LogKind::Rpc => this.show_rpc_trace_for_server(server_id, cx),
631 LogKind::Trace => this.show_trace_for_server(server_id, cx),
632 LogKind::Logs => this.show_logs_for_server(server_id, cx),
633 LogKind::ServerInfo => this.show_server_info(server_id, cx),
634 }
635 }
636
637 cx.notify();
638 });
639 let events_subscriptions = cx.subscribe(&log_store, |log_view, _, e, cx| match e {
640 Event::NewServerLogEntry { id, entry, kind } => {
641 if log_view.current_server_id == Some(*id) && *kind == log_view.active_entry_kind {
642 log_view.editor.update(cx, |editor, cx| {
643 editor.set_read_only(false);
644 let last_point = editor.buffer().read(cx).len(cx);
645 let newest_cursor_is_at_end =
646 editor.selections.newest::<usize>(cx).start >= last_point;
647 editor.edit(
648 vec![
649 (last_point..last_point, entry.trim()),
650 (last_point..last_point, "\n"),
651 ],
652 cx,
653 );
654 let entry_length = entry.len();
655 if entry_length > 1024 {
656 editor.fold_ranges(
657 vec![last_point + 1024..last_point + entry_length],
658 false,
659 cx,
660 );
661 }
662
663 if newest_cursor_is_at_end {
664 editor.request_autoscroll(Autoscroll::bottom(), cx);
665 }
666 editor.set_read_only(true);
667 });
668 }
669 }
670 });
671 let (editor, editor_subscriptions) = Self::editor_for_logs(String::new(), cx);
672
673 let focus_handle = cx.focus_handle();
674 let focus_subscription = cx.on_focus(&focus_handle, |log_view, cx| {
675 cx.focus_view(&log_view.editor);
676 });
677
678 let mut this = Self {
679 focus_handle,
680 editor,
681 editor_subscriptions,
682 project,
683 log_store,
684 current_server_id: None,
685 active_entry_kind: LogKind::Logs,
686 _log_store_subscriptions: vec![
687 model_changes_subscription,
688 events_subscriptions,
689 focus_subscription,
690 ],
691 };
692 if let Some(server_id) = server_id {
693 this.show_logs_for_server(server_id, cx);
694 }
695 this
696 }
697
698 fn editor_for_logs(
699 log_contents: String,
700 cx: &mut ViewContext<Self>,
701 ) -> (View<Editor>, Vec<Subscription>) {
702 let editor = cx.new_view(|cx| {
703 let mut editor = Editor::multi_line(cx);
704 editor.set_text(log_contents, cx);
705 editor.move_to_end(&MoveToEnd, cx);
706 editor.set_read_only(true);
707 editor.set_show_inline_completions(Some(false), cx);
708 editor.set_soft_wrap_mode(SoftWrap::EditorWidth, cx);
709 editor
710 });
711 let editor_subscription = cx.subscribe(
712 &editor,
713 |_, _, event: &EditorEvent, cx: &mut ViewContext<LspLogView>| cx.emit(event.clone()),
714 );
715 let search_subscription = cx.subscribe(
716 &editor,
717 |_, _, event: &SearchEvent, cx: &mut ViewContext<LspLogView>| cx.emit(event.clone()),
718 );
719 (editor, vec![editor_subscription, search_subscription])
720 }
721
722 fn editor_for_server_info(
723 server: &LanguageServer,
724 cx: &mut ViewContext<Self>,
725 ) -> (View<Editor>, Vec<Subscription>) {
726 let editor = cx.new_view(|cx| {
727 let mut editor = Editor::multi_line(cx);
728 let server_info = format!(
729 "* Server: {NAME} (id {ID})
730
731* Binary: {BINARY:#?}
732
733* Running in project: {PATH:?}
734
735* Capabilities: {CAPABILITIES}
736
737* Configuration: {CONFIGURATION}",
738 NAME = server.name(),
739 ID = server.server_id(),
740 BINARY = server.binary(),
741 PATH = server.root_path(),
742 CAPABILITIES = serde_json::to_string_pretty(&server.capabilities())
743 .unwrap_or_else(|e| format!("Failed to serialize capabilities: {e}")),
744 CONFIGURATION = serde_json::to_string_pretty(server.configuration())
745 .unwrap_or_else(|e| format!("Failed to serialize configuration: {e}")),
746 );
747 editor.set_text(server_info, cx);
748 editor.set_read_only(true);
749 editor.set_show_inline_completions(Some(false), cx);
750 editor.set_soft_wrap_mode(SoftWrap::EditorWidth, cx);
751 editor
752 });
753 let editor_subscription = cx.subscribe(
754 &editor,
755 |_, _, event: &EditorEvent, cx: &mut ViewContext<LspLogView>| cx.emit(event.clone()),
756 );
757 let search_subscription = cx.subscribe(
758 &editor,
759 |_, _, event: &SearchEvent, cx: &mut ViewContext<LspLogView>| cx.emit(event.clone()),
760 );
761 (editor, vec![editor_subscription, search_subscription])
762 }
763
764 pub(crate) fn menu_items<'a>(&'a self, cx: &'a AppContext) -> Option<Vec<LogMenuItem>> {
765 let log_store = self.log_store.read(cx);
766
767 let unknown_server = LanguageServerName::new_static("unknown server");
768
769 let mut rows = log_store
770 .language_servers
771 .iter()
772 .map(|(server_id, state)| match &state.kind {
773 LanguageServerKind::Local { .. } | LanguageServerKind::Remote { .. } => {
774 let worktree_root_name = state
775 .worktree_id
776 .and_then(|id| self.project.read(cx).worktree_for_id(id, cx))
777 .map(|worktree| worktree.read(cx).root_name().to_string())
778 .unwrap_or_else(|| "Unknown worktree".to_string());
779
780 LogMenuItem {
781 server_id: *server_id,
782 server_name: state.name.clone().unwrap_or(unknown_server.clone()),
783 server_kind: state.kind.clone(),
784 worktree_root_name,
785 rpc_trace_enabled: state.rpc_state.is_some(),
786 selected_entry: self.active_entry_kind,
787 trace_level: lsp::TraceValue::Off,
788 }
789 }
790
791 LanguageServerKind::Global => LogMenuItem {
792 server_id: *server_id,
793 server_name: state.name.clone().unwrap_or(unknown_server.clone()),
794 server_kind: state.kind.clone(),
795 worktree_root_name: "supplementary".to_string(),
796 rpc_trace_enabled: state.rpc_state.is_some(),
797 selected_entry: self.active_entry_kind,
798 trace_level: lsp::TraceValue::Off,
799 },
800 })
801 .chain(
802 self.project
803 .read(cx)
804 .supplementary_language_servers(cx)
805 .filter_map(|(server_id, name)| {
806 let state = log_store.language_servers.get(&server_id)?;
807 Some(LogMenuItem {
808 server_id,
809 server_name: name.clone(),
810 server_kind: state.kind.clone(),
811 worktree_root_name: "supplementary".to_string(),
812 rpc_trace_enabled: state.rpc_state.is_some(),
813 selected_entry: self.active_entry_kind,
814 trace_level: lsp::TraceValue::Off,
815 })
816 }),
817 )
818 .collect::<Vec<_>>();
819 rows.sort_by_key(|row| row.server_id);
820 rows.dedup_by_key(|row| row.server_id);
821 Some(rows)
822 }
823
824 fn show_logs_for_server(&mut self, server_id: LanguageServerId, cx: &mut ViewContext<Self>) {
825 let typ = self
826 .log_store
827 .read_with(cx, |v, _| {
828 v.language_servers.get(&server_id).map(|v| v.log_level)
829 })
830 .unwrap_or(MessageType::LOG);
831 let log_contents = self
832 .log_store
833 .read(cx)
834 .server_logs(server_id)
835 .map(|v| log_contents(v, typ));
836 if let Some(log_contents) = log_contents {
837 self.current_server_id = Some(server_id);
838 self.active_entry_kind = LogKind::Logs;
839 let (editor, editor_subscriptions) = Self::editor_for_logs(log_contents, cx);
840 self.editor = editor;
841 self.editor_subscriptions = editor_subscriptions;
842 cx.notify();
843 }
844 cx.focus(&self.focus_handle);
845 }
846
847 fn update_log_level(
848 &self,
849 server_id: LanguageServerId,
850 level: MessageType,
851 cx: &mut ViewContext<Self>,
852 ) {
853 let log_contents = self.log_store.update(cx, |this, _| {
854 if let Some(state) = this.get_language_server_state(server_id) {
855 state.log_level = level;
856 }
857
858 this.server_logs(server_id).map(|v| log_contents(v, level))
859 });
860
861 if let Some(log_contents) = log_contents {
862 self.editor.update(cx, move |editor, cx| {
863 editor.set_text(log_contents, cx);
864 editor.move_to_end(&MoveToEnd, cx);
865 });
866 cx.notify();
867 }
868
869 cx.focus(&self.focus_handle);
870 }
871
872 fn show_trace_for_server(&mut self, server_id: LanguageServerId, cx: &mut ViewContext<Self>) {
873 let log_contents = self
874 .log_store
875 .read(cx)
876 .server_trace(server_id)
877 .map(|v| log_contents(v, ()));
878 if let Some(log_contents) = log_contents {
879 self.current_server_id = Some(server_id);
880 self.active_entry_kind = LogKind::Trace;
881 let (editor, editor_subscriptions) = Self::editor_for_logs(log_contents, cx);
882 self.editor = editor;
883 self.editor_subscriptions = editor_subscriptions;
884 cx.notify();
885 }
886 cx.focus(&self.focus_handle);
887 }
888
889 fn show_rpc_trace_for_server(
890 &mut self,
891 server_id: LanguageServerId,
892 cx: &mut ViewContext<Self>,
893 ) {
894 let rpc_log = self.log_store.update(cx, |log_store, _| {
895 log_store
896 .enable_rpc_trace_for_language_server(server_id)
897 .map(|state| log_contents(&state.rpc_messages, ()))
898 });
899 if let Some(rpc_log) = rpc_log {
900 self.current_server_id = Some(server_id);
901 self.active_entry_kind = LogKind::Rpc;
902 let (editor, editor_subscriptions) = Self::editor_for_logs(rpc_log, cx);
903 let language = self.project.read(cx).languages().language_for_name("JSON");
904 editor
905 .read(cx)
906 .buffer()
907 .read(cx)
908 .as_singleton()
909 .expect("log buffer should be a singleton")
910 .update(cx, |_, cx| {
911 cx.spawn({
912 let buffer = cx.handle();
913 |_, mut cx| async move {
914 let language = language.await.ok();
915 buffer.update(&mut cx, |buffer, cx| {
916 buffer.set_language(language, cx);
917 })
918 }
919 })
920 .detach_and_log_err(cx);
921 });
922
923 self.editor = editor;
924 self.editor_subscriptions = editor_subscriptions;
925 cx.notify();
926 }
927
928 cx.focus(&self.focus_handle);
929 }
930
931 fn toggle_rpc_trace_for_server(
932 &mut self,
933 server_id: LanguageServerId,
934 enabled: bool,
935 cx: &mut ViewContext<Self>,
936 ) {
937 self.log_store.update(cx, |log_store, _| {
938 if enabled {
939 log_store.enable_rpc_trace_for_language_server(server_id);
940 } else {
941 log_store.disable_rpc_trace_for_language_server(server_id);
942 }
943 });
944 if !enabled && Some(server_id) == self.current_server_id {
945 self.show_logs_for_server(server_id, cx);
946 cx.notify();
947 }
948 }
949
950 fn update_trace_level(
951 &self,
952 server_id: LanguageServerId,
953 level: TraceValue,
954 cx: &mut ViewContext<Self>,
955 ) {
956 if let Some(server) = self
957 .project
958 .read(cx)
959 .lsp_store()
960 .read(cx)
961 .language_server_for_id(server_id)
962 {
963 self.log_store.update(cx, |this, _| {
964 if let Some(state) = this.get_language_server_state(server_id) {
965 state.trace_level = level;
966 }
967 });
968
969 server
970 .notify::<SetTrace>(&SetTraceParams { value: level })
971 .ok();
972 }
973 }
974
975 fn show_server_info(&mut self, server_id: LanguageServerId, cx: &mut ViewContext<Self>) {
976 let lsp_store = self.project.read(cx).lsp_store();
977 let Some(server) = lsp_store.read(cx).language_server_for_id(server_id) else {
978 return;
979 };
980 self.current_server_id = Some(server_id);
981 self.active_entry_kind = LogKind::ServerInfo;
982 let (editor, editor_subscriptions) = Self::editor_for_server_info(&server, cx);
983 self.editor = editor;
984 self.editor_subscriptions = editor_subscriptions;
985 cx.notify();
986 cx.focus(&self.focus_handle);
987 }
988}
989
990fn log_filter<T: Message>(line: &T, cmp: <T as Message>::Level) -> Option<&str> {
991 if line.should_include(cmp) {
992 Some(line.as_ref())
993 } else {
994 None
995 }
996}
997
998fn log_contents<T: Message>(lines: &VecDeque<T>, cmp: <T as Message>::Level) -> String {
999 let (a, b) = lines.as_slices();
1000 let a = a.iter().filter_map(move |v| log_filter(v, cmp));
1001 let b = b.iter().filter_map(move |v| log_filter(v, cmp));
1002 a.chain(b).fold(String::new(), |mut acc, el| {
1003 acc.push_str(el);
1004 acc.push('\n');
1005 acc
1006 })
1007}
1008
1009impl Render for LspLogView {
1010 fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
1011 self.editor
1012 .update(cx, |editor, cx| editor.render(cx).into_any_element())
1013 }
1014}
1015
1016impl FocusableView for LspLogView {
1017 fn focus_handle(&self, _: &AppContext) -> FocusHandle {
1018 self.focus_handle.clone()
1019 }
1020}
1021
1022impl Item for LspLogView {
1023 type Event = EditorEvent;
1024
1025 fn to_item_events(event: &Self::Event, f: impl FnMut(workspace::item::ItemEvent)) {
1026 Editor::to_item_events(event, f)
1027 }
1028
1029 fn tab_content_text(&self, _cx: &WindowContext) -> Option<SharedString> {
1030 Some("LSP Logs".into())
1031 }
1032
1033 fn telemetry_event_text(&self) -> Option<&'static str> {
1034 None
1035 }
1036
1037 fn as_searchable(&self, handle: &View<Self>) -> Option<Box<dyn SearchableItemHandle>> {
1038 Some(Box::new(handle.clone()))
1039 }
1040
1041 fn clone_on_split(
1042 &self,
1043 _workspace_id: Option<WorkspaceId>,
1044 cx: &mut ViewContext<Self>,
1045 ) -> Option<View<Self>>
1046 where
1047 Self: Sized,
1048 {
1049 Some(cx.new_view(|cx| {
1050 let mut new_view = Self::new(self.project.clone(), self.log_store.clone(), cx);
1051 if let Some(server_id) = self.current_server_id {
1052 match self.active_entry_kind {
1053 LogKind::Rpc => new_view.show_rpc_trace_for_server(server_id, cx),
1054 LogKind::Trace => new_view.show_trace_for_server(server_id, cx),
1055 LogKind::Logs => new_view.show_logs_for_server(server_id, cx),
1056 LogKind::ServerInfo => new_view.show_server_info(server_id, cx),
1057 }
1058 }
1059 new_view
1060 }))
1061 }
1062}
1063
1064impl SearchableItem for LspLogView {
1065 type Match = <Editor as SearchableItem>::Match;
1066
1067 fn clear_matches(&mut self, cx: &mut ViewContext<Self>) {
1068 self.editor.update(cx, |e, cx| e.clear_matches(cx))
1069 }
1070
1071 fn update_matches(&mut self, matches: &[Self::Match], cx: &mut ViewContext<Self>) {
1072 self.editor
1073 .update(cx, |e, cx| e.update_matches(matches, cx))
1074 }
1075
1076 fn query_suggestion(&mut self, cx: &mut ViewContext<Self>) -> String {
1077 self.editor.update(cx, |e, cx| e.query_suggestion(cx))
1078 }
1079
1080 fn activate_match(
1081 &mut self,
1082 index: usize,
1083 matches: &[Self::Match],
1084 cx: &mut ViewContext<Self>,
1085 ) {
1086 self.editor
1087 .update(cx, |e, cx| e.activate_match(index, matches, cx))
1088 }
1089
1090 fn select_matches(&mut self, matches: &[Self::Match], cx: &mut ViewContext<Self>) {
1091 self.editor
1092 .update(cx, |e, cx| e.select_matches(matches, cx))
1093 }
1094
1095 fn find_matches(
1096 &mut self,
1097 query: Arc<project::search::SearchQuery>,
1098 cx: &mut ViewContext<Self>,
1099 ) -> gpui::Task<Vec<Self::Match>> {
1100 self.editor.update(cx, |e, cx| e.find_matches(query, cx))
1101 }
1102
1103 fn replace(&mut self, _: &Self::Match, _: &SearchQuery, _: &mut ViewContext<Self>) {
1104 // Since LSP Log is read-only, it doesn't make sense to support replace operation.
1105 }
1106 fn supported_options() -> workspace::searchable::SearchOptions {
1107 workspace::searchable::SearchOptions {
1108 case: true,
1109 word: true,
1110 regex: true,
1111 // LSP log is read-only.
1112 replacement: false,
1113 selection: false,
1114 }
1115 }
1116 fn active_match_index(
1117 &mut self,
1118 matches: &[Self::Match],
1119 cx: &mut ViewContext<Self>,
1120 ) -> Option<usize> {
1121 self.editor
1122 .update(cx, |e, cx| e.active_match_index(matches, cx))
1123 }
1124}
1125
1126impl EventEmitter<ToolbarItemEvent> for LspLogToolbarItemView {}
1127
1128impl ToolbarItemView for LspLogToolbarItemView {
1129 fn set_active_pane_item(
1130 &mut self,
1131 active_pane_item: Option<&dyn ItemHandle>,
1132 cx: &mut ViewContext<Self>,
1133 ) -> workspace::ToolbarItemLocation {
1134 if let Some(item) = active_pane_item {
1135 if let Some(log_view) = item.downcast::<LspLogView>() {
1136 self.log_view = Some(log_view.clone());
1137 self._log_view_subscription = Some(cx.observe(&log_view, |_, _, cx| {
1138 cx.notify();
1139 }));
1140 return ToolbarItemLocation::PrimaryLeft;
1141 }
1142 }
1143 self.log_view = None;
1144 self._log_view_subscription = None;
1145 ToolbarItemLocation::Hidden
1146 }
1147}
1148
1149impl Render for LspLogToolbarItemView {
1150 fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
1151 let Some(log_view) = self.log_view.clone() else {
1152 return div();
1153 };
1154 let (menu_rows, current_server_id) = log_view.update(cx, |log_view, cx| {
1155 let menu_rows = log_view.menu_items(cx).unwrap_or_default();
1156 let current_server_id = log_view.current_server_id;
1157 (menu_rows, current_server_id)
1158 });
1159
1160 let current_server = current_server_id.and_then(|current_server_id| {
1161 if let Ok(ix) = menu_rows.binary_search_by_key(¤t_server_id, |e| e.server_id) {
1162 Some(menu_rows[ix].clone())
1163 } else {
1164 None
1165 }
1166 });
1167 let available_language_servers: Vec<_> = menu_rows
1168 .iter()
1169 .map(|row| {
1170 (
1171 row.server_id,
1172 row.server_name.clone(),
1173 row.worktree_root_name.clone(),
1174 row.selected_entry,
1175 )
1176 })
1177 .collect();
1178 let log_toolbar_view = cx.view().clone();
1179 let lsp_menu = PopoverMenu::new("LspLogView")
1180 .anchor(Corner::TopLeft)
1181 .trigger(Button::new(
1182 "language_server_menu_header",
1183 current_server
1184 .as_ref()
1185 .map(|row| {
1186 Cow::Owned(format!(
1187 "{} ({})",
1188 row.server_name.0, row.worktree_root_name,
1189 ))
1190 })
1191 .unwrap_or_else(|| "No server selected".into()),
1192 ))
1193 .menu({
1194 let log_view = log_view.clone();
1195 move |cx| {
1196 let log_view = log_view.clone();
1197 ContextMenu::build(cx, |mut menu, cx| {
1198 for (server_id, name, worktree_root, active_entry_kind) in
1199 available_language_servers.iter()
1200 {
1201 let label = format!("{} ({})", name, worktree_root);
1202 let server_id = *server_id;
1203 let active_entry_kind = *active_entry_kind;
1204 menu = menu.entry(
1205 label,
1206 None,
1207 cx.handler_for(&log_view, move |view, cx| {
1208 view.current_server_id = Some(server_id);
1209 view.active_entry_kind = active_entry_kind;
1210 match view.active_entry_kind {
1211 LogKind::Rpc => {
1212 view.toggle_rpc_trace_for_server(server_id, true, cx);
1213 view.show_rpc_trace_for_server(server_id, cx);
1214 }
1215 LogKind::Trace => view.show_trace_for_server(server_id, cx),
1216 LogKind::Logs => view.show_logs_for_server(server_id, cx),
1217 LogKind::ServerInfo => view.show_server_info(server_id, cx),
1218 }
1219 cx.notify();
1220 }),
1221 );
1222 }
1223 menu
1224 })
1225 .into()
1226 }
1227 });
1228 let view_selector = current_server.map(|server| {
1229 let server_id = server.server_id;
1230 let is_remote = server.server_kind.is_remote();
1231 let rpc_trace_enabled = server.rpc_trace_enabled;
1232 let log_view = log_view.clone();
1233 PopoverMenu::new("LspViewSelector")
1234 .anchor(Corner::TopLeft)
1235 .trigger(Button::new(
1236 "language_server_menu_header",
1237 server.selected_entry.label(),
1238 ))
1239 .menu(move |cx| {
1240 let log_toolbar_view = log_toolbar_view.clone();
1241 let log_view = log_view.clone();
1242 Some(ContextMenu::build(cx, move |this, cx| {
1243 this.entry(
1244 SERVER_LOGS,
1245 None,
1246 cx.handler_for(&log_view, move |view, cx| {
1247 view.show_logs_for_server(server_id, cx);
1248 }),
1249 )
1250 .when(!is_remote, |this| {
1251 this.entry(
1252 SERVER_TRACE,
1253 None,
1254 cx.handler_for(&log_view, move |view, cx| {
1255 view.show_trace_for_server(server_id, cx);
1256 }),
1257 )
1258 .custom_entry(
1259 {
1260 let log_toolbar_view = log_toolbar_view.clone();
1261 move |cx| {
1262 h_flex()
1263 .w_full()
1264 .justify_between()
1265 .child(Label::new(RPC_MESSAGES))
1266 .child(
1267 div().child(
1268 Checkbox::new(
1269 "LspLogEnableRpcTrace",
1270 if rpc_trace_enabled {
1271 ToggleState::Selected
1272 } else {
1273 ToggleState::Unselected
1274 },
1275 )
1276 .on_click(cx.listener_for(
1277 &log_toolbar_view,
1278 move |view, selection, cx| {
1279 let enabled = matches!(
1280 selection,
1281 ToggleState::Selected
1282 );
1283 view.toggle_rpc_logging_for_server(
1284 server_id, enabled, cx,
1285 );
1286 cx.stop_propagation();
1287 },
1288 )),
1289 ),
1290 )
1291 .into_any_element()
1292 }
1293 },
1294 cx.handler_for(&log_view, move |view, cx| {
1295 view.show_rpc_trace_for_server(server_id, cx);
1296 }),
1297 )
1298 })
1299 .entry(
1300 SERVER_INFO,
1301 None,
1302 cx.handler_for(&log_view, move |view, cx| {
1303 view.show_server_info(server_id, cx);
1304 }),
1305 )
1306 }))
1307 })
1308 });
1309 h_flex()
1310 .size_full()
1311 .justify_between()
1312 .child(
1313 h_flex()
1314 .child(lsp_menu)
1315 .children(view_selector)
1316 .child(log_view.update(cx, |this, _| match this.active_entry_kind {
1317 LogKind::Trace => {
1318 let log_view = log_view.clone();
1319 div().child(
1320 PopoverMenu::new("lsp-trace-level-menu")
1321 .anchor(Corner::TopLeft)
1322 .trigger(Button::new(
1323 "language_server_trace_level_selector",
1324 "Trace level",
1325 ))
1326 .menu({
1327 let log_view = log_view.clone();
1328
1329 move |cx| {
1330 let id = log_view.read(cx).current_server_id?;
1331
1332 let trace_level = log_view.update(cx, |this, cx| {
1333 this.log_store.update(cx, |this, _| {
1334 Some(
1335 this.get_language_server_state(id)?
1336 .trace_level,
1337 )
1338 })
1339 })?;
1340
1341 ContextMenu::build(cx, |mut menu, _| {
1342 let log_view = log_view.clone();
1343
1344 for (option, label) in [
1345 (TraceValue::Off, "Off"),
1346 (TraceValue::Messages, "Messages"),
1347 (TraceValue::Verbose, "Verbose"),
1348 ] {
1349 menu = menu.entry(label, None, {
1350 let log_view = log_view.clone();
1351 move |cx| {
1352 log_view.update(cx, |this, cx| {
1353 if let Some(id) =
1354 this.current_server_id
1355 {
1356 this.update_trace_level(
1357 id, option, cx,
1358 );
1359 }
1360 });
1361 }
1362 });
1363 if option == trace_level {
1364 menu.select_last();
1365 }
1366 }
1367
1368 menu
1369 })
1370 .into()
1371 }
1372 }),
1373 )
1374 }
1375 LogKind::Logs => {
1376 let log_view = log_view.clone();
1377 div().child(
1378 PopoverMenu::new("lsp-log-level-menu")
1379 .anchor(Corner::TopLeft)
1380 .trigger(Button::new(
1381 "language_server_log_level_selector",
1382 "Log level",
1383 ))
1384 .menu({
1385 let log_view = log_view.clone();
1386
1387 move |cx| {
1388 let id = log_view.read(cx).current_server_id?;
1389
1390 let log_level = log_view.update(cx, |this, cx| {
1391 this.log_store.update(cx, |this, _| {
1392 Some(
1393 this.get_language_server_state(id)?
1394 .log_level,
1395 )
1396 })
1397 })?;
1398
1399 ContextMenu::build(cx, |mut menu, _| {
1400 let log_view = log_view.clone();
1401
1402 for (option, label) in [
1403 (MessageType::LOG, "Log"),
1404 (MessageType::INFO, "Info"),
1405 (MessageType::WARNING, "Warning"),
1406 (MessageType::ERROR, "Error"),
1407 ] {
1408 menu = menu.entry(label, None, {
1409 let log_view = log_view.clone();
1410 move |cx| {
1411 log_view.update(cx, |this, cx| {
1412 if let Some(id) =
1413 this.current_server_id
1414 {
1415 this.update_log_level(
1416 id, option, cx,
1417 );
1418 }
1419 });
1420 }
1421 });
1422 if option == log_level {
1423 menu.select_last();
1424 }
1425 }
1426
1427 menu
1428 })
1429 .into()
1430 }
1431 }),
1432 )
1433 }
1434 _ => div(),
1435 })),
1436 )
1437 .child(
1438 div()
1439 .child(
1440 Button::new("clear_log_button", "Clear").on_click(cx.listener(
1441 |this, _, cx| {
1442 if let Some(log_view) = this.log_view.as_ref() {
1443 log_view.update(cx, |log_view, cx| {
1444 log_view.editor.update(cx, |editor, cx| {
1445 editor.set_read_only(false);
1446 editor.clear(cx);
1447 editor.set_read_only(true);
1448 });
1449 })
1450 }
1451 },
1452 )),
1453 )
1454 .ml_2(),
1455 )
1456 }
1457}
1458
1459const RPC_MESSAGES: &str = "RPC Messages";
1460const SERVER_LOGS: &str = "Server Logs";
1461const SERVER_TRACE: &str = "Server Trace";
1462const SERVER_INFO: &str = "Server Info";
1463
1464impl Default for LspLogToolbarItemView {
1465 fn default() -> Self {
1466 Self::new()
1467 }
1468}
1469
1470impl LspLogToolbarItemView {
1471 pub fn new() -> Self {
1472 Self {
1473 log_view: None,
1474 _log_view_subscription: None,
1475 }
1476 }
1477
1478 fn toggle_rpc_logging_for_server(
1479 &mut self,
1480 id: LanguageServerId,
1481 enabled: bool,
1482 cx: &mut ViewContext<Self>,
1483 ) {
1484 if let Some(log_view) = &self.log_view {
1485 log_view.update(cx, |log_view, cx| {
1486 log_view.toggle_rpc_trace_for_server(id, enabled, cx);
1487 if !enabled && Some(id) == log_view.current_server_id {
1488 log_view.show_logs_for_server(id, cx);
1489 cx.notify();
1490 } else if enabled {
1491 log_view.show_rpc_trace_for_server(id, cx);
1492 cx.notify();
1493 }
1494 cx.focus(&log_view.focus_handle);
1495 });
1496 }
1497 cx.notify();
1498 }
1499}
1500
1501pub enum Event {
1502 NewServerLogEntry {
1503 id: LanguageServerId,
1504 entry: String,
1505 kind: LogKind,
1506 },
1507}
1508
1509impl EventEmitter<Event> for LogStore {}
1510impl EventEmitter<Event> for LspLogView {}
1511impl EventEmitter<EditorEvent> for LspLogView {}
1512impl EventEmitter<SearchEvent> for LspLogView {}