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