1use ::serde::{Deserialize, Serialize};
 2use gpui::{PromptLevel, WeakEntity};
 3use lsp::LanguageServer;
 4
 5use crate::{LanguageServerPromptRequest, LspStore, LspStoreEvent};
 6
 7pub const RUST_ANALYZER_NAME: &str = "rust-analyzer";
 8
 9/// Experimental: Informs the end user about the state of the server
10///
11/// [Rust Analyzer Specification](https://github.com/rust-lang/rust-analyzer/blob/master/docs/dev/lsp-extensions.md#server-status)
12#[derive(Debug)]
13enum ServerStatus {}
14
15/// Other(String) variant to handle unknown values due to this still being experimental
16#[derive(Debug, PartialEq, Deserialize, Serialize, Clone)]
17#[serde(rename_all = "camelCase")]
18enum ServerHealthStatus {
19    Ok,
20    Warning,
21    Error,
22    Other(String),
23}
24
25#[derive(Debug, PartialEq, Deserialize, Serialize, Clone)]
26#[serde(rename_all = "camelCase")]
27struct ServerStatusParams {
28    pub health: ServerHealthStatus,
29    pub message: Option<String>,
30}
31
32impl lsp::notification::Notification for ServerStatus {
33    type Params = ServerStatusParams;
34    const METHOD: &'static str = "experimental/serverStatus";
35}
36
37pub fn register_notifications(lsp_store: WeakEntity<LspStore>, language_server: &LanguageServer) {
38    let name = language_server.name();
39    let server_id = language_server.server_id();
40
41    let this = lsp_store;
42
43    language_server
44        .on_notification::<ServerStatus, _>({
45            let name = name.to_string();
46            move |params, cx| {
47                let this = this.clone();
48                let name = name.to_string();
49                if let Some(ref message) = params.message {
50                    let message = message.trim();
51                    if !message.is_empty() {
52                        let formatted_message = format!(
53                            "Language server {name} (id {server_id}) status update: {message}"
54                        );
55                        match params.health {
56                            ServerHealthStatus::Ok => log::info!("{}", formatted_message),
57                            ServerHealthStatus::Warning => log::warn!("{}", formatted_message),
58                            ServerHealthStatus::Error => {
59                                log::error!("{}", formatted_message);
60                                let (tx, _rx) = smol::channel::bounded(1);
61                                let request = LanguageServerPromptRequest {
62                                    level: PromptLevel::Critical,
63                                    message: params.message.unwrap_or_default(),
64                                    actions: Vec::new(),
65                                    response_channel: tx,
66                                    lsp_name: name.clone(),
67                                };
68                                let _ = this
69                                    .update(cx, |_, cx| {
70                                        cx.emit(LspStoreEvent::LanguageServerPrompt(request));
71                                    })
72                                    .ok();
73                            }
74                            ServerHealthStatus::Other(status) => {
75                                log::info!("Unknown server health: {status}\n{formatted_message}")
76                            }
77                        }
78                    }
79                }
80            }
81        })
82        .detach();
83}