summaryrefslogtreecommitdiff
path: root/support/texlab/src/features/link.rs
blob: 0a02f7819eb3075d855d8e2a1685dc9e6667e332 (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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
use lsp_types::{DocumentLink, DocumentLinkParams};

use crate::LineIndexExt;

use super::FeatureRequest;

pub fn find_document_links(request: FeatureRequest<DocumentLinkParams>) -> Vec<DocumentLink> {
    let mut links = Vec::new();
    let main_document = request.main_document();
    if let Some(data) = main_document.data.as_latex() {
        for include in &data.extras.explicit_links {
            for target in &include.targets {
                if request
                    .workspace
                    .documents_by_uri
                    .values()
                    .any(|document| document.uri.as_ref() == target.as_ref())
                {
                    links.push(DocumentLink {
                        range: main_document
                            .line_index
                            .line_col_lsp_range(include.stem_range),
                        target: Some(target.as_ref().clone()),
                        tooltip: None,
                        data: None,
                    });
                    break;
                }
            }
        }
    }
    links
}

#[cfg(test)]
mod tests {
    use super::*;

    use lsp_types::Range;

    use crate::{features::testing::FeatureTester, RangeExt};

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

        let items = find_document_links(request);
        assert!(items.is_empty());
    }

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

        let items = find_document_links(request);
        assert!(items.is_empty());
    }

    #[test]
    fn test_includes() {
        let tester = FeatureTester::builder()
            .files(vec![("foo.tex", r#"\input{bar.tex}"#), ("bar.tex", r#""#)])
            .main("foo.tex")
            .build();
        let target = tester.uri("bar.tex");

        let actual_items = find_document_links(tester.link());

        let expected_items = vec![DocumentLink {
            range: Range::new_simple(0, 7, 0, 14),
            target: Some(target.as_ref().clone()),
            tooltip: None,
            data: None,
        }];
        assert_eq!(actual_items, expected_items);
    }

    #[test]
    fn test_imports() {
        let tester = FeatureTester::builder()
            .files(vec![
                ("foo.tex", r#"\import{bar/}{baz}"#),
                ("bar/baz.tex", r#""#),
            ])
            .main("foo.tex")
            .build();
        let target = tester.uri("bar/baz.tex");

        let actual_items = find_document_links(tester.link());

        let expected_items = vec![DocumentLink {
            range: Range::new_simple(0, 14, 0, 17),
            target: Some(target.as_ref().clone()),
            tooltip: None,
            data: None,
        }];
        assert_eq!(actual_items, expected_items);
    }
}