1use anyhow::Context as _;
  2use gpui::{App, Context, Entity, Window};
  3use language::Language;
  4use project::lsp_store::lsp_ext_command::SwitchSourceHeaderResult;
  5use rpc::proto;
  6use url::Url;
  7use workspace::{OpenOptions, OpenVisible};
  8
  9use crate::lsp_ext::find_specific_language_server_in_selection;
 10
 11use crate::{Editor, SwitchSourceHeader, element::register_action};
 12
 13use project::lsp_store::clangd_ext::CLANGD_SERVER_NAME;
 14
 15fn is_c_language(language: &Language) -> bool {
 16    language.name() == "C++".into() || language.name() == "C".into()
 17}
 18
 19pub fn switch_source_header(
 20    editor: &mut Editor,
 21    _: &SwitchSourceHeader,
 22    window: &mut Window,
 23    cx: &mut Context<Editor>,
 24) {
 25    let Some(project) = &editor.project else {
 26        return;
 27    };
 28    let Some(workspace) = editor.workspace() else {
 29        return;
 30    };
 31
 32    let Some((_, _, server_to_query, buffer)) =
 33        find_specific_language_server_in_selection(editor, cx, is_c_language, CLANGD_SERVER_NAME)
 34    else {
 35        return;
 36    };
 37    let project = project.clone();
 38    let upstream_client = project.read(cx).lsp_store().read(cx).upstream_client();
 39    cx.spawn_in(window, async move |_editor, cx| {
 40        let source_file = buffer.read_with(cx, |buffer, _| {
 41            buffer.file().map(|file| file.path()).map(|path| path.to_string_lossy().to_string()).unwrap_or_else(|| "Unknown".to_string())
 42        })?;
 43
 44        let switch_source_header = if let Some((client, project_id)) = upstream_client {
 45            let buffer_id = buffer.read_with(cx, |buffer, _| buffer.remote_id())?;
 46            let request = proto::LspExtSwitchSourceHeader {
 47                project_id,
 48                buffer_id: buffer_id.to_proto(),
 49            };
 50            let response = client
 51                .request(request)
 52                .await
 53                .context("lsp ext switch source header proto request")?;
 54            SwitchSourceHeaderResult(response.target_file)
 55        } else {
 56            project.update(cx, |project, cx| {
 57                project.request_lsp(
 58                    buffer,
 59                    project::LanguageServerToQuery::Other(server_to_query),
 60                    project::lsp_store::lsp_ext_command::SwitchSourceHeader,
 61                    cx,
 62                )
 63            })?.await.with_context(|| format!("Switch source/header LSP request for path \"{source_file}\" failed"))?
 64        };
 65
 66        if switch_source_header.0.is_empty() {
 67            log::info!("Clangd returned an empty string when requesting to switch source/header from \"{source_file}\"" );
 68            return Ok(());
 69        }
 70
 71        let goto = Url::parse(&switch_source_header.0).with_context(|| {
 72            format!(
 73                "Parsing URL \"{}\" returned from switch source/header failed",
 74                switch_source_header.0
 75            )
 76        })?;
 77
 78        let path = goto.to_file_path().map_err(|()| {
 79            anyhow::anyhow!("URL conversion to file path failed for \"{goto}\"")
 80        })?;
 81
 82        workspace
 83            .update_in(cx, |workspace, window, cx| {
 84                workspace.open_abs_path(path, OpenOptions { visible: Some(OpenVisible::None), ..Default::default() }, window, cx)
 85            })
 86            .with_context(|| {
 87                format!(
 88                    "Switch source/header could not open \"{goto}\" in workspace"
 89                )
 90            })?
 91            .await
 92            .map(|_| ())
 93    })
 94    .detach_and_log_err(cx);
 95}
 96
 97pub fn apply_related_actions(editor: &Entity<Editor>, window: &mut Window, cx: &mut App) {
 98    if editor
 99        .read(cx)
100        .buffer()
101        .read(cx)
102        .all_buffers()
103        .into_iter()
104        .filter_map(|buffer| buffer.read(cx).language())
105        .any(|language| is_c_language(language))
106    {
107        register_action(editor, window, switch_source_header);
108    }
109}