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