Save capabilities to ServerContext, check them when loading configs

Author: taminomara

crates/emmylua_ls/src/context/mod.rs

@@ -10,6 +10,7 @@ pub use client_id::{ClientId, get_client_id};
 use emmylua_code_analysis::EmmyLuaAnalysis;
 pub use file_diagnostic::FileDiagnostic;
 use lsp_server::{Connection, ErrorCode, Message, RequestId, Response};
+use lsp_types::ClientCapabilities;
 pub use snapshot::ServerContextSnapshot;
 pub use status_bar::ProgressTask;
 pub use status_bar::StatusBar;
@@ -29,10 +30,11 @@ pub struct ServerContext {
     file_diagnostic: Arc<FileDiagnostic>,
     workspace_manager: Arc<RwLock<WorkspaceManager>>,
     status_bar: Arc<StatusBar>,
+    client_capabilities: Arc<ClientCapabilities>,
 }
 
 impl ServerContext {
-    pub fn new(conn: Connection) -> Self {
+    pub fn new(conn: Connection, client_capabilities: Arc<ClientCapabilities>) -> Self {
         let client = Arc::new(ClientProxy::new(Connection {
             sender: conn.sender.clone(),
             receiver: conn.receiver.clone(),
@@ -60,6 +62,7 @@ impl ServerContext {
             cancllations: Arc::new(Mutex::new(HashMap::new())),
             workspace_manager,
             status_bar,
+            client_capabilities,
         }
     }
 
@@ -70,6 +73,7 @@ impl ServerContext {
             file_diagnostic: self.file_diagnostic.clone(),
             workspace_manager: self.workspace_manager.clone(),
             status_bar: self.status_bar.clone(),
+            client_capabilities: self.client_capabilities.clone(),
         }
     }
 

crates/emmylua_ls/src/context/snapshot.rs

@@ -1,3 +1,4 @@
+use lsp_types::ClientCapabilities;
 use std::sync::Arc;
 use tokio::sync::RwLock;
 
@@ -15,4 +16,5 @@ pub struct ServerContextSnapshot {
     pub file_diagnostic: Arc<FileDiagnostic>,
     pub workspace_manager: Arc<RwLock<WorkspaceManager>>,
     pub status_bar: Arc<StatusBar>,
+    pub client_capabilities: Arc<ClientCapabilities>,
 }

crates/emmylua_ls/src/handlers/configuration/mod.rs

@@ -22,8 +22,15 @@ pub async fn on_did_change_configuration(
     let client_id = workspace_manager.client_config.client_id;
     drop(workspace_manager);
 
+    let supports_config_request = context
+        .client_capabilities
+        .workspace
+        .as_ref()?
+        .configuration
+        .unwrap_or_default();
+
     log::info!("change config client_id: {:?}", client_id);
-    let new_client_config = get_client_config(&context, client_id).await;
+    let new_client_config = get_client_config(&context, client_id, supports_config_request).await;
     let mut config_manager = context.workspace_manager.write().await;
     config_manager.client_config = new_client_config;
 

crates/emmylua_ls/src/handlers/initialized/client_config/mod.rs

@@ -20,6 +20,7 @@ pub struct ClientConfig {
 pub async fn get_client_config(
     context: &ServerContextSnapshot,
     client_id: ClientId,
+    supports_config_request: bool,
 ) -> ClientConfig {
     let mut config = ClientConfig {
         client_id,
@@ -29,11 +30,16 @@ pub async fn get_client_config(
         partial_emmyrcs: None,
     };
     match client_id {
-        ClientId::VSCode => get_client_config_vscode(context, &mut config).await,
+        ClientId::VSCode => {
+            get_client_config_vscode(context, &mut config).await;
+        }
         ClientId::Neovim => {
-            get_client_config_default(context, &mut config, Some(&["Lua", "emmylua"])).await
+            get_client_config_default(context, &mut config, Some(&["Lua", "emmylua"])).await;
+        }
+        _ if supports_config_request => {
+            get_client_config_default(context, &mut config, None).await;
         }
-        _ => get_client_config_default(context, &mut config, None).await,
+        _ => {}
     };
 
     config

crates/emmylua_ls/src/handlers/initialized/mod.rs

@@ -42,6 +42,12 @@ pub async fn initialized_handler(
     log::info!("main root: {:?}", main_root);
 
     let client_id = get_client_id(&params.client_info);
+    let supports_config_request = params
+        .capabilities
+        .workspace
+        .as_ref()?
+        .configuration
+        .unwrap_or_default();
     log::info!("client_id: {:?}", client_id);
 
     {
@@ -51,7 +57,7 @@ pub async fn initialized_handler(
         log::info!("workspace folders set");
     }
 
-    let client_config = get_client_config(&context, client_id).await;
+    let client_config = get_client_config(&context, client_id, supports_config_request).await;
     log::info!("client_config: {:?}", client_config);
 
     let params_json = serde_json::to_string_pretty(&params).unwrap();

crates/emmylua_ls/src/lib.rs

@@ -5,17 +5,17 @@ mod logger;
 mod meta_text;
 mod util;
 
+use crate::handlers::{
+    initialized_handler, on_notification_handler, on_req_handler, on_response_handler,
+};
 pub use clap::Parser;
 pub use cmd_args::*;
 use handlers::server_capabilities;
 use lsp_server::{Connection, Message};
 use lsp_types::InitializeParams;
+use std::sync::Arc;
 use std::{env, error::Error};
 
-use crate::handlers::{
-    initialized_handler, on_notification_handler, on_req_handler, on_response_handler,
-};
-
 #[macro_use]
 extern crate rust_i18n;
 rust_i18n::i18n!("./locales", fallback = "en");
@@ -60,10 +60,13 @@ async fn main_loop(
     params: InitializeParams,
     cmd_args: CmdArgs,
 ) -> Result<(), Box<dyn Error + Sync + Send>> {
-    let mut server_context = context::ServerContext::new(Connection {
-        sender: connection.sender.clone(),
-        receiver: connection.receiver.clone(),
-    });
+    let mut server_context = context::ServerContext::new(
+        Connection {
+            sender: connection.sender.clone(),
+            receiver: connection.receiver.clone(),
+        },
+        Arc::new(params.capabilities.clone()),
+    );
 
     let server_context_snapshot = server_context.snapshot();
     tokio::spawn(async move {