use crate::{ find_lsp_workspace, jsonrpc, transport::{Payload, Transport}, Call, Error, OffsetEncoding, Result, }; use helix_core::{find_workspace, ChangeSet, Rope}; use helix_loader::{self, VERSION_AND_GIT_HASH}; use lsp::{ notification::DidChangeWorkspaceFolders, DidChangeWorkspaceFoldersParams, OneOf, PositionEncodingKind, WorkspaceFolder, WorkspaceFoldersChangeEvent, }; use lsp_types as lsp; use parking_lot::Mutex; use serde::Deserialize; use serde_json::Value; use std::future::Future; use std::process::Stdio; use std::sync::{ atomic::{AtomicU64, Ordering}, Arc, }; use std::{collections::HashMap, path::PathBuf}; use tokio::{ io::{BufReader, BufWriter}, process::{Child, Command}, sync::{ mpsc::{channel, UnboundedReceiver, UnboundedSender}, Notify, OnceCell, }, }; fn workspace_for_uri(uri: lsp::Url) -> WorkspaceFolder { lsp::WorkspaceFolder { name: uri .path_segments() .and_then(|segments| segments.last()) .map(|basename| basename.to_string()) .unwrap_or_default(), uri, } } #[derive(Debug)] pub struct Client { id: usize, _process: Child, server_tx: UnboundedSender, request_counter: AtomicU64, pub(crate) capabilities: OnceCell, config: Option, root_path: std::path::PathBuf, root_uri: Option, workspace_folders: Mutex>, initalize_notify: Arc, /// workspace folders added while the server is still initalizing req_timeout: u64, } impl Client { pub fn try_add_doc( self: &Arc, root_markers: &[String], manual_roots: &[PathBuf], doc_path: Option<&std::path::PathBuf>, may_support_workspace: bool, ) -> bool { let (workspace, workspace_is_cwd) = find_workspace(); let root = find_lsp_workspace( doc_path .and_then(|x| x.parent().and_then(|x| x.to_str())) .unwrap_or("."), root_markers, manual_roots, &workspace, workspace_is_cwd, ); let root_uri = root .as_ref() .and_then(|root| lsp::Url::from_file_path(root).ok()); if self.root_path == root.unwrap_or(workspace) || root_uri.as_ref().map_or(false, |root_uri| { self.workspace_folders .lock() .iter() .any(|workspace| &workspace.uri == root_uri) }) { // workspace URI is already registered so we can use this client return true; } // this server definitly doesn't support multiple workspace, no need to check capabilities if !may_support_workspace { return false; } let Some(capabilities) = self.capabilities.get() else { let client = Arc::clone(self); // initalization hasn't finished yet, deal with this new root later // TODO: In the edgecase that a **new root** is added // for an LSP that **doesn't support workspace_folders** before initaliation is finished // the new roots are ignored. // That particular edgecase would require retroactively spawning new LSP // clients and therefore also require us to retroactively update the corresponding // documents LSP client handle. It's doable but a pretty weird edgecase so let's // wait and see if anyone ever runs into it. tokio::spawn(async move { client.initalize_notify.notified().await; if let Some(workspace_folders_caps) = client .capabilities() .workspace .as_ref() .and_then(|cap| cap.workspace_folders.as_ref()) .filter(|cap| cap.supported.unwrap_or(false)) { client.add_workspace_folder( root_uri, &workspace_folders_caps.change_notifications, ); } }); return true; }; if let Some(workspace_folders_caps) = capabilities .workspace .as_ref() .and_then(|cap| cap.workspace_folders.as_ref()) .filter(|cap| cap.supported.unwrap_or(false)) { self.add_workspace_folder(root_uri, &workspace_folders_caps.change_notifications); true } else { // the server doesn't support multi workspaces, we need a new client false } } fn add_workspace_folder( &self, root_uri: Option, change_notifications: &Option>, ) { // root_uri is None just means that there isn't really any LSP workspace // associated with this file. For servers that support multiple workspaces // there is just one server so we can always just use that shared instance. // No need to add a new workspace root here as there is no logical root for this file // let the server deal with this let Some(root_uri) = root_uri else { return; }; // server supports workspace folders, let's add the new root to the list self.workspace_folders .lock() .push(workspace_for_uri(root_uri.clone())); if &Some(OneOf::Left(false)) == change_notifications { // server specifically opted out of DidWorkspaceChange notifications // let's assume the server will request the workspace folders itself // and that we can therefore reuse the client (but are done now) return; } tokio::spawn(self.did_change_workspace(vec![workspace_for_uri(root_uri)], Vec::new())); } #[allow(clippy::type_complexity)] #[allow(clippy::too_many_arguments)] pub fn start( cmd: &str, args: &[String], config: Option, server_environment: HashMap, root_markers: &[String], manual_roots: &[PathBuf], id: usize, req_timeout: u64, doc_path: Option<&std::path::PathBuf>, ) -> Result<(Self, UnboundedReceiver<(usize, Call)>, Arc)> { // Resolve path to the binary let cmd = which::which(cmd).map_err(|err| anyhow::anyhow!(err))?; let process = Command::new(cmd) .envs(server_environment) .args(args) .stdin(Stdio::piped()) .stdout(Stdio::piped()) .stderr(Stdio::piped()) // make sure the process is reaped on drop .kill_on_drop(true) .spawn(); let mut process = process?; // TODO: do we need bufreader/writer here? or do we use async wrappers on unblock? let writer = BufWriter::new(process.stdin.take().expect("Failed to open stdin")); let reader = BufReader::new(process.stdout.take().expect("Failed to open stdout")); let stderr = BufReader::new(process.stderr.take().expect("Failed to open stderr")); let (server_rx, server_tx, initialize_notify) = Transport::start(reader, writer, stderr, id); let (workspace, workspace_is_cwd) = find_workspace(); let root = find_lsp_workspace( doc_path .and_then(|x| x.parent().and_then(|x| x.to_str())) .unwrap_or("."), root_markers, manual_roots, &workspace, workspace_is_cwd, ); // `root_uri` and `workspace_folder` can be empty in case there is no workspace // `root_url` can not, use `workspace` as a fallback let root_path = root.clone().unwrap_or_else(|| workspace.clone()); let root_uri = root.and_then(|root| lsp::Url::from_file_path(root).ok()); let workspace_folders = root_uri .clone() .map(|root| vec![workspace_for_uri(root)]) .unwrap_or_default(); let client = Self { id, _process: process, server_tx, request_counter: AtomicU64::new(0), capabilities: OnceCell::new(), config, req_timeout, root_path, root_uri, workspace_folders: Mutex::new(workspace_folders), initalize_notify: initialize_notify.clone(), }; Ok((client, server_rx, initialize_notify)) } pub fn id(&self) -> usize { self.id } fn next_request_id(&self) -> jsonrpc::Id { let id = self.request_counter.fetch_add(1, Ordering::Relaxed); jsonrpc::Id::Num(id) } fn value_into_params(value: Value) -> jsonrpc::Params { use jsonrpc::Params; match value { Value::Null => Params::None, Value::Bool(_) | Value::Number(_) | Value::String(_) => Params::Array(vec![value]), Value::Array(vec) => Params::Array(vec), Value::Object(map) => Params::Map(map), } } pub fn is_initialized(&self) -> bool { self.capabilities.get().is_some() } pub fn capabilities(&self) -> &lsp::ServerCapabilities { self.capabilities .get() .expect("language server not yet initialized!") } pub fn offset_encoding(&self) -> OffsetEncoding { self.capabilities() .position_encoding .as_ref() .and_then(|encoding| match encoding.as_str() { "utf-8" => Some(OffsetEncoding::Utf8), "utf-16" => Some(OffsetEncoding::Utf16), "utf-32" => Some(OffsetEncoding::Utf32), encoding => { log::error!("Server provided invalid position encording {encoding}, defaulting to utf-16"); None }, }) .unwrap_or_default() } pub fn config(&self) -> Option<&Value> { self.config.as_ref() } pub async fn workspace_folders( &self, ) -> parking_lot::MutexGuard<'_, Vec> { self.workspace_folders.lock() } /// Execute a RPC request on the language server. async fn request(&self, params: R::Params) -> Result where R::Params: serde::Serialize, R::Result: core::fmt::Debug, // TODO: temporary { // a future that resolves into the response let json = self.call::(params).await?; let response = serde_json::from_value(json)?; Ok(response) } /// Execute a RPC request on the language server. fn call( &self, params: R::Params, ) -> impl Future> where R::Params: serde::Serialize, { let server_tx = self.server_tx.clone(); let id = self.next_request_id(); let timeout_secs = self.req_timeout; async move { use std::time::Duration; use tokio::time::timeout; let params = serde_json::to_value(params)?; let request = jsonrpc::MethodCall { jsonrpc: Some(jsonrpc::Version::V2), id: id.clone(), method: R::METHOD.to_string(), params: Self::value_into_params(params), }; let (tx, mut rx) = channel::>(1); server_tx .send(Payload::Request { chan: tx, value: request, }) .map_err(|e| Error::Other(e.into()))?; // TODO: delay other calls until initialize success timeout(Duration::from_secs(timeout_secs), rx.recv()) .await .map_err(|_| Error::Timeout(id))? // return Timeout .ok_or(Error::StreamClosed)? } } /// Send a RPC notification to the language server. pub fn notify( &self, params: R::Params, ) -> impl Future> where R::Params: serde::Serialize, { let server_tx = self.server_tx.clone(); async move { let params = serde_json::to_value(params)?; let notification = jsonrpc::Notification { jsonrpc: Some(jsonrpc::Version::V2), method: R::METHOD.to_string(), params: Self::value_into_params(params), }; server_tx .send(Payload::Notification(notification)) .map_err(|e| Error::Other(e.into()))?; Ok(()) } } /// Reply to a language server RPC call. pub fn reply( &self, id: jsonrpc::Id, result: core::result::Result, ) -> impl Future> { use jsonrpc::{Failure, Output, Success, Version}; let server_tx = self.server_tx.clone(); async move { let output = match result { Ok(result) => Output::Success(Success { jsonrpc: Some(Version::V2), id, result: serde_json::to_value(result)?, }), Err(error) => Output::Failure(Failure { jsonrpc: Some(Version::V2), id, error, }), }; server_tx .send(Payload::Response(output)) .map_err(|e| Error::Other(e.into()))?; Ok(()) } } // ------------------------------------------------------------------------------------------- // General messages // ------------------------------------------------------------------------------------------- pub(crate) async fn initialize(&self) -> Result { if let Some(config) = &self.config { log::info!("Using custom LSP config: {}", config); } #[allow(deprecated)] let params = lsp::InitializeParams { process_id: Some(std::process::id()), workspace_folders: Some(self.workspace_folders.lock().clone()), // root_path is obsolete, but some clients like pyright still use it so we specify both. // clients will prefer _uri if possible root_path: self.root_path.to_str().map(|path| path.to_owned()), root_uri: self.root_uri.clone(), initialization_options: self.config.clone(), capabilities: lsp::ClientCapabilities { workspace: Some(lsp::WorkspaceClientCapabilities { configuration: Some(true), did_change_configuration: Some(lsp::DynamicRegistrationClientCapabilities { dynamic_registration: Some(false), }), workspace_folders: Some(true), apply_edit: Some(true), symbol: Some(lsp::WorkspaceSymbolClientCapabilities { dynamic_registration: Some(false), ..Default::default() }), execute_command: Some(lsp::DynamicRegistrationClientCapabilities { dynamic_registration: Some(false), }), inlay_hint: Some(lsp::InlayHintWorkspaceClientCapabilities { refresh_support: Some(false), }), workspace_edit: Some(lsp::WorkspaceEditClientCapabilities { document_changes: Some(true), resource_operations: Some(vec![ lsp::ResourceOperationKind::Create, lsp::ResourceOperationKind::Rename, lsp::ResourceOperationKind::Delete, ]), failure_handling: Some(lsp::FailureHandlingKind::Abort), normalizes_line_endings: Some(false), change_annotation_support: None, }), ..Default::default() }), text_document: Some(lsp::TextDocumentClientCapabilities { completion: Some(lsp::CompletionClientCapabilities { completion_item: Some(lsp::CompletionItemCapability { snippet_support: Some(true), resolve_support: Some(lsp::CompletionItemCapabilityResolveSupport { properties: vec![ String::from("documentation"), String::from("detail"), String::from("additionalTextEdits"), ], }), insert_replace_support: Some(true), deprecated_support: Some(true), tag_support: Some(lsp::TagSupport { value_set: vec![lsp::CompletionItemTag::DEPRECATED], }), ..Default::default() }), completion_item_kind: Some(lsp::CompletionItemKindCapability { ..Default::default() }), context_support: None, // additional context information Some(true) ..Default::default() }), hover: Some(lsp::HoverClientCapabilities { // if not specified, rust-analyzer returns plaintext marked as markdown but // badly formatted. content_format: Some(vec![lsp::MarkupKind::Markdown]), ..Default::default() }), signature_help: Some(lsp::SignatureHelpClientCapabilities { signature_information: Some(lsp::SignatureInformationSettings { documentation_format: Some(vec![lsp::MarkupKind::Markdown]), parameter_information: Some(lsp::ParameterInformationSettings { label_offset_support: Some(true), }), active_parameter_support: Some(true), }), ..Default::default() }), rename: Some(lsp::RenameClientCapabilities { dynamic_registration: Some(false), prepare_support: Some(true), prepare_support_default_behavior: None, honors_change_annotations: Some(false), }), code_action: Some(lsp::CodeActionClientCapabilities { code_action_literal_support: Some(lsp::CodeActionLiteralSupport { code_action_kind: lsp::CodeActionKindLiteralSupport { value_set: [ lsp::CodeActionKind::EMPTY, lsp::CodeActionKind::QUICKFIX, lsp::CodeActionKind::REFACTOR, lsp::CodeActionKind::REFACTOR_EXTRACT, lsp::CodeActionKind::REFACTOR_INLINE, lsp::CodeActionKind::REFACTOR_REWRITE, lsp::CodeActionKind::SOURCE, lsp::CodeActionKind::SOURCE_ORGANIZE_IMPORTS, ] .iter() .map(|kind| kind.as_str().to_string()) .collect(), }, }), ..Default::default() }), publish_diagnostics: Some(lsp::PublishDiagnosticsClientCapabilities { version_support: Some(true), ..Default::default() }), inlay_hint: Some(lsp::InlayHintClientCapabilities { dynamic_registration: Some(false), resolve_support: None, }), ..Default::default() }), window: Some(lsp::WindowClientCapabilities { work_done_progress: Some(true), ..Default::default() }), general: Some(lsp::GeneralClientCapabilities { position_encodings: Some(vec![ PositionEncodingKind::UTF32, PositionEncodingKind::UTF8, PositionEncodingKind::UTF16, ]), ..Default::default() }), ..Default::default() }, trace: None, client_info: Some(lsp::ClientInfo { name: String::from("helix"), version: Some(String::from(VERSION_AND_GIT_HASH)), }), locale: None, // TODO }; self.request::(params).await } pub async fn shutdown(&self) -> Result<()> { self.request::(()).await } pub fn exit(&self) -> impl Future> { self.notify::(()) } /// Tries to shut down the language server but returns /// early if server responds with an error. pub async fn shutdown_and_exit(&self) -> Result<()> { self.shutdown().await?; self.exit().await } /// Forcefully shuts down the language server ignoring any errors. pub async fn force_shutdown(&self) -> Result<()> { if let Err(e) = self.shutdown().await { log::warn!("language server failed to terminate gracefully - {}", e); } self.exit().await } // ------------------------------------------------------------------------------------------- // Workspace // ------------------------------------------------------------------------------------------- pub fn did_change_configuration(&self, settings: Value) -> impl Future> { self.notify::( lsp::DidChangeConfigurationParams { settings }, ) } pub fn did_change_workspace( &self, added: Vec, removed: Vec, ) -> impl Future> { self.notify::(DidChangeWorkspaceFoldersParams { event: WorkspaceFoldersChangeEvent { added, removed }, }) } // ------------------------------------------------------------------------------------------- // Text document // ------------------------------------------------------------------------------------------- pub fn text_document_did_open( &self, uri: lsp::Url, version: i32, doc: &Rope, language_id: String, ) -> impl Future> { self.notify::(lsp::DidOpenTextDocumentParams { text_document: lsp::TextDocumentItem { uri, language_id, version, text: String::from(doc), }, }) } pub fn changeset_to_changes( old_text: &Rope, new_text: &Rope, changeset: &ChangeSet, offset_encoding: OffsetEncoding, ) -> Vec { let mut iter = changeset.changes().iter().peekable(); let mut old_pos = 0; let mut new_pos = 0; let mut changes = Vec::new(); use crate::util::pos_to_lsp_pos; use helix_core::Operation::*; // this is dumb. TextEdit describes changes to the initial doc (concurrent), but // TextDocumentContentChangeEvent describes a series of changes (sequential). // So S -> S1 -> S2, meaning positioning depends on the previous edits. // // Calculation is therefore a bunch trickier. use helix_core::RopeSlice; fn traverse(pos: lsp::Position, text: RopeSlice) -> lsp::Position { let lsp::Position { mut line, mut character, } = pos; let mut chars = text.chars().peekable(); while let Some(ch) = chars.next() { // LSP only considers \n, \r or \r\n as line endings if ch == '\n' || ch == '\r' { // consume a \r\n if ch == '\r' && chars.peek() == Some(&'\n') { chars.next(); } line += 1; character = 0; } else { character += ch.len_utf16() as u32; } } lsp::Position { line, character } } let old_text = old_text.slice(..); while let Some(change) = iter.next() { let len = match change { Delete(i) | Retain(i) => *i, Insert(_) => 0, }; let mut old_end = old_pos + len; match change { Retain(i) => { new_pos += i; } Delete(_) => { let start = pos_to_lsp_pos(new_text, new_pos, offset_encoding); let end = traverse(start, old_text.slice(old_pos..old_end)); // deletion changes.push(lsp::TextDocumentContentChangeEvent { range: Some(lsp::Range::new(start, end)), text: "".to_string(), range_length: None, }); } Insert(s) => { let start = pos_to_lsp_pos(new_text, new_pos, offset_encoding); new_pos += s.chars().count(); // a subsequent delete means a replace, consume it let end = if let Some(Delete(len)) = iter.peek() { old_end = old_pos + len; let end = traverse(start, old_text.slice(old_pos..old_end)); iter.next(); // replacement end } else { // insert start }; changes.push(lsp::TextDocumentContentChangeEvent { range: Some(lsp::Range::new(start, end)), text: s.to_string(), range_length: None, }); } } old_pos = old_end; } changes } pub fn text_document_did_change( &self, text_document: lsp::VersionedTextDocumentIdentifier, old_text: &Rope, new_text: &Rope, changes: &ChangeSet, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support document sync. let sync_capabilities = match capabilities.text_document_sync { Some( lsp::TextDocumentSyncCapability::Kind(kind) | lsp::TextDocumentSyncCapability::Options(lsp::TextDocumentSyncOptions { change: Some(kind), .. }), ) => kind, // None | SyncOptions { changes: None } _ => return None, }; let changes = match sync_capabilities { lsp::TextDocumentSyncKind::FULL => { vec![lsp::TextDocumentContentChangeEvent { // range = None -> whole document range: None, //Some(Range) range_length: None, // u64 apparently deprecated text: new_text.to_string(), }] } lsp::TextDocumentSyncKind::INCREMENTAL => { Self::changeset_to_changes(old_text, new_text, changes, self.offset_encoding()) } lsp::TextDocumentSyncKind::NONE => return None, kind => unimplemented!("{:?}", kind), }; Some(self.notify::( lsp::DidChangeTextDocumentParams { text_document, content_changes: changes, }, )) } pub fn text_document_did_close( &self, text_document: lsp::TextDocumentIdentifier, ) -> impl Future> { self.notify::(lsp::DidCloseTextDocumentParams { text_document, }) } // will_save / will_save_wait_until pub fn text_document_did_save( &self, text_document: lsp::TextDocumentIdentifier, text: &Rope, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); let include_text = match &capabilities.text_document_sync { Some(lsp::TextDocumentSyncCapability::Options(lsp::TextDocumentSyncOptions { save: Some(options), .. })) => match options { lsp::TextDocumentSyncSaveOptions::Supported(true) => false, lsp::TextDocumentSyncSaveOptions::SaveOptions(lsp_types::SaveOptions { include_text, }) => include_text.unwrap_or(false), // Supported(false) _ => return None, }, // unsupported _ => return None, }; Some(self.notify::( lsp::DidSaveTextDocumentParams { text_document, text: include_text.then_some(text.into()), }, )) } pub fn completion( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support completion. capabilities.completion_provider.as_ref()?; let params = lsp::CompletionParams { text_document_position: lsp::TextDocumentPositionParams { text_document, position, }, // TODO: support these tokens by async receiving and updating the choice list work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, partial_result_params: lsp::PartialResultParams { partial_result_token: None, }, context: None, // lsp::CompletionContext { trigger_kind: , trigger_character: Some(), } }; Some(self.call::(params)) } pub fn resolve_completion_item( &self, completion_item: lsp::CompletionItem, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support resolving completion items. match capabilities.completion_provider { Some(lsp::CompletionOptions { resolve_provider: Some(true), .. }) => (), _ => return None, } Some(self.call::(completion_item)) } pub fn text_document_signature_help( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support signature help. capabilities.signature_help_provider.as_ref()?; let params = lsp::SignatureHelpParams { text_document_position_params: lsp::TextDocumentPositionParams { text_document, position, }, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, context: None, // lsp::SignatureHelpContext }; Some(self.call::(params)) } pub fn text_document_range_inlay_hints( &self, text_document: lsp::TextDocumentIdentifier, range: lsp::Range, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); match capabilities.inlay_hint_provider { Some( lsp::OneOf::Left(true) | lsp::OneOf::Right(lsp::InlayHintServerCapabilities::Options(_)), ) => (), _ => return None, } let params = lsp::InlayHintParams { text_document, range, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, }; Some(self.call::(params)) } pub fn text_document_hover( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support hover. match capabilities.hover_provider { Some( lsp::HoverProviderCapability::Simple(true) | lsp::HoverProviderCapability::Options(_), ) => (), _ => return None, } let params = lsp::HoverParams { text_document_position_params: lsp::TextDocumentPositionParams { text_document, position, }, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, // lsp::SignatureHelpContext }; Some(self.call::(params)) } // formatting pub fn text_document_formatting( &self, text_document: lsp::TextDocumentIdentifier, options: lsp::FormattingOptions, work_done_token: Option, ) -> Option>>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support formatting. match capabilities.document_formatting_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, }; // merge FormattingOptions with 'config.format' let config_format = self .config .as_ref() .and_then(|cfg| cfg.get("format")) .and_then(|fmt| HashMap::::deserialize(fmt).ok()); let options = if let Some(mut properties) = config_format { // passed in options take precedence over 'config.format' properties.extend(options.properties); lsp::FormattingOptions { properties, ..options } } else { options }; let params = lsp::DocumentFormattingParams { text_document, options, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, }; let request = self.call::(params); Some(async move { let json = request.await?; let response: Option> = serde_json::from_value(json)?; Ok(response.unwrap_or_default()) }) } pub fn text_document_range_formatting( &self, text_document: lsp::TextDocumentIdentifier, range: lsp::Range, options: lsp::FormattingOptions, work_done_token: Option, ) -> Option>>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support range formatting. match capabilities.document_range_formatting_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, }; let params = lsp::DocumentRangeFormattingParams { text_document, range, options, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, }; let request = self.call::(params); Some(async move { let json = request.await?; let response: Option> = serde_json::from_value(json)?; Ok(response.unwrap_or_default()) }) } pub fn text_document_document_highlight( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support document highlight. match capabilities.document_highlight_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, } let params = lsp::DocumentHighlightParams { text_document_position_params: lsp::TextDocumentPositionParams { text_document, position, }, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, partial_result_params: lsp::PartialResultParams { partial_result_token: None, }, }; Some(self.call::(params)) } fn goto_request< T: lsp::request::Request< Params = lsp::GotoDefinitionParams, Result = Option, >, >( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> impl Future> { let params = lsp::GotoDefinitionParams { text_document_position_params: lsp::TextDocumentPositionParams { text_document, position, }, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, partial_result_params: lsp::PartialResultParams { partial_result_token: None, }, }; self.call::(params) } pub fn goto_definition( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support goto-definition. match capabilities.definition_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, } Some(self.goto_request::( text_document, position, work_done_token, )) } pub fn goto_declaration( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support goto-declaration. match capabilities.declaration_provider { Some( lsp::DeclarationCapability::Simple(true) | lsp::DeclarationCapability::RegistrationOptions(_) | lsp::DeclarationCapability::Options(_), ) => (), _ => return None, } Some(self.goto_request::( text_document, position, work_done_token, )) } pub fn goto_type_definition( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support goto-type-definition. match capabilities.type_definition_provider { Some( lsp::TypeDefinitionProviderCapability::Simple(true) | lsp::TypeDefinitionProviderCapability::Options(_), ) => (), _ => return None, } Some(self.goto_request::( text_document, position, work_done_token, )) } pub fn goto_implementation( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support goto-definition. match capabilities.implementation_provider { Some( lsp::ImplementationProviderCapability::Simple(true) | lsp::ImplementationProviderCapability::Options(_), ) => (), _ => return None, } Some(self.goto_request::( text_document, position, work_done_token, )) } pub fn goto_reference( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, work_done_token: Option, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support goto-reference. match capabilities.references_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, } let params = lsp::ReferenceParams { text_document_position: lsp::TextDocumentPositionParams { text_document, position, }, context: lsp::ReferenceContext { include_declaration: true, }, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token }, partial_result_params: lsp::PartialResultParams { partial_result_token: None, }, }; Some(self.call::(params)) } pub fn document_symbols( &self, text_document: lsp::TextDocumentIdentifier, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support document symbols. match capabilities.document_symbol_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, } let params = lsp::DocumentSymbolParams { text_document, work_done_progress_params: lsp::WorkDoneProgressParams::default(), partial_result_params: lsp::PartialResultParams::default(), }; Some(self.call::(params)) } pub fn prepare_rename( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); match capabilities.rename_provider { Some(lsp::OneOf::Right(lsp::RenameOptions { prepare_provider: Some(true), .. })) => (), _ => return None, } let params = lsp::TextDocumentPositionParams { text_document, position, }; Some(self.call::(params)) } // empty string to get all symbols pub fn workspace_symbols(&self, query: String) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support workspace symbols. match capabilities.workspace_symbol_provider { Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) => (), _ => return None, } let params = lsp::WorkspaceSymbolParams { query, work_done_progress_params: lsp::WorkDoneProgressParams::default(), partial_result_params: lsp::PartialResultParams::default(), }; Some(self.call::(params)) } pub fn code_actions( &self, text_document: lsp::TextDocumentIdentifier, range: lsp::Range, context: lsp::CodeActionContext, ) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the server does not support code actions. match capabilities.code_action_provider { Some( lsp::CodeActionProviderCapability::Simple(true) | lsp::CodeActionProviderCapability::Options(_), ) => (), _ => return None, } let params = lsp::CodeActionParams { text_document, range, context, work_done_progress_params: lsp::WorkDoneProgressParams::default(), partial_result_params: lsp::PartialResultParams::default(), }; Some(self.call::(params)) } pub fn supports_rename(&self) -> bool { let capabilities = self.capabilities.get().unwrap(); matches!( capabilities.rename_provider, Some(lsp::OneOf::Left(true) | lsp::OneOf::Right(_)) ) } pub fn rename_symbol( &self, text_document: lsp::TextDocumentIdentifier, position: lsp::Position, new_name: String, ) -> Option>> { if !self.supports_rename() { return None; } let params = lsp::RenameParams { text_document_position: lsp::TextDocumentPositionParams { text_document, position, }, new_name, work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token: None, }, }; let request = self.call::(params); Some(async move { let json = request.await?; let response: Option = serde_json::from_value(json)?; Ok(response.unwrap_or_default()) }) } pub fn command(&self, command: lsp::Command) -> Option>> { let capabilities = self.capabilities.get().unwrap(); // Return early if the language server does not support executing commands. capabilities.execute_command_provider.as_ref()?; let params = lsp::ExecuteCommandParams { command: command.command, arguments: command.arguments.unwrap_or_default(), work_done_progress_params: lsp::WorkDoneProgressParams { work_done_token: None, }, }; Some(self.call::(params)) } }