summaryrefslogtreecommitdiff
path: root/support/texlab/src/features/definition/label.rs
blob: e5ccd49937988620cd3c2186337627da75c167cb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
use lsp_types::{GotoDefinitionParams, LocationLink};

use crate::{
    features::cursor::CursorContext, find_label_definition, render_label, syntax::latex,
    LineIndexExt,
};

pub fn goto_label_definition(
    context: &CursorContext<GotoDefinitionParams>,
) -> Option<Vec<LocationLink>> {
    let main_document = context.request.main_document();

    let (name_text, name_range) = context
        .find_label_name_key()
        .or_else(|| context.find_label_name_command())?;

    let origin_selection_range = main_document.line_index.line_col_lsp_range(name_range);

    for document in context.request.workspace.documents_by_uri.values() {
        if let Some(data) = document.data.as_latex() {
            if let Some(definition) =
                find_label_definition(&latex::SyntaxNode::new_root(data.green.clone()), &name_text)
            {
                let target_selection_range = latex::small_range(&definition.name()?.key()?);
                let target_range =
                    render_label(&context.request.workspace, &name_text, Some(definition))
                        .map(|label| label.range)
                        .unwrap_or(target_selection_range);

                return Some(vec![LocationLink {
                    origin_selection_range: Some(origin_selection_range),
                    target_uri: document.uri.as_ref().clone(),
                    target_range: document.line_index.line_col_lsp_range(target_range),
                    target_selection_range: document
                        .line_index
                        .line_col_lsp_range(target_selection_range),
                }]);
            }
        }
    }

    None
}

#[cfg(test)]
mod tests {
    use crate::features::testing::FeatureTester;

    use super::*;

    #[test]
    fn test_empty_latex_document() {
        let request = FeatureTester::builder()
            .files(vec![("main.tex", "")])
            .main("main.tex")
            .line(0)
            .character(0)
            .build()
            .definition();

        let context = CursorContext::new(request);
        let actual_links = goto_label_definition(&context);

        assert!(actual_links.is_none());
    }

    #[test]
    fn test_empty_bibtex_document() {
        let request = FeatureTester::builder()
            .files(vec![("main.bib", "")])
            .main("main.bib")
            .line(0)
            .character(0)
            .build()
            .definition();

        let context = CursorContext::new(request);
        let actual_links = goto_label_definition(&context);

        assert!(actual_links.is_none());
    }
}