rust_analyzer_ext.rs

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