Add implementation for clangd's textDocument/switchSourceHeader (#1105)

Author: martskins

autoload/LanguageClient.vim

@@ -808,6 +808,15 @@ function! LanguageClient#findLocations(...) abort
     return LanguageClient#Call('languageClient/findLocations', l:params, l:Callback)
 endfunction
 
+function! LanguageClient#textDocument_switchSourceHeader(...) abort
+    let l:Callback = get(a:000, 1, v:null)
+    let l:params = {
+                \ 'filename': LSP#filename(),
+                \ }
+    call extend(l:params, get(a:000, 0, {}))
+    return LanguageClient#Call('textDocument/switchSourceHeader', l:params, l:Callback)
+endfunction
+
 function! LanguageClient#textDocument_definition(...) abort
     let l:params = {
                 \ 'method': 'textDocument/definition',

doc/LanguageClient.txt

@@ -974,6 +974,10 @@ Signature: LanguageClient#diagnosticsPrevious()
 
 Moves the cursor to the previous diagnostic in the buffer, relative to the current cursor position.
 
+*LanguageClient#textDocument_switchSourceHeader*
+Signature: LanguageClient#textDocument_switchSourceHeader(...)
+
+Calls clangd's `textDocument/switchSourceHeader` extension request.
 
 ==============================================================================
 5. Mappings                                           *LanguageClientMappings*

plugin/LanguageClient.vim

@@ -146,6 +146,10 @@ function! LanguageClient_explainErrorAtPoint(...)
     return call('LanguageClient#explainErrorAtPoint', a:000)
 endfunction
 
+function! LanguageClient_textDocument_switchSourceHeader(...)
+    return call('LanguageClient#textDocument_switchSourceHeader', a:000)
+endfunction
+
 command! -nargs=* LanguageClientStart :call LanguageClient#startServer(<f-args>)
 command! LanguageClientStop :call LanguageClient#exit()
 

src/extensions/clangd.rs

@@ -0,0 +1,27 @@
+use crate::{language_client::LanguageClient, utils::ToUrl};
+use anyhow::Result;
+use jsonrpc_core::Value;
+use lsp_types::TextDocumentIdentifier;
+
+pub mod request {
+    pub const SWITCH_SOURCE_HEADER: &str = "textDocument/switchSourceHeader";
+}
+
+impl LanguageClient {
+    pub fn text_document_switch_source_header(&self, params: &Value) -> Result<Value> {
+        let filename = self.vim()?.get_filename(params)?;
+        let language_id = self.vim()?.get_language_id(&filename, &Value::Null)?;
+        let params = TextDocumentIdentifier {
+            uri: filename.to_url()?,
+        };
+
+        let response: String = self
+            .get_client(&Some(language_id))?
+            .call(request::SWITCH_SOURCE_HEADER, params)?;
+
+        let path = std::path::Path::new(&response);
+        self.vim()?.edit(&None, path)?;
+
+        Ok(Value::Null)
+    }
+}

src/extensions/mod.rs

@@ -1,3 +1,4 @@
+pub mod clangd;
 pub mod gopls;
 pub mod java;
 pub mod rust_analyzer;

src/rpchandler.rs

@@ -1,3 +1,4 @@
+use crate::extensions::clangd;
 use crate::{language_client::LanguageClient, language_server_protocol::Direction, types::*};
 use anyhow::{anyhow, Result};
 use log::*;
@@ -118,6 +119,10 @@ impl LanguageClient {
             REQUEST_SEMANTIC_SCOPES => self.semantic_scopes(&params),
             REQUEST_SHOW_SEMANTIC_HL_SYMBOLS => self.semantic_highlight_symbols(&params),
 
+            clangd::request::SWITCH_SOURCE_HEADER => {
+                self.text_document_switch_source_header(&params)
+            }
+
             _ => {
                 let language_id_target = if language_id.is_some() {
                     // Message from language server. No handler found.