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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
use lsp_types::{GotoDefinitionParams, LocationLink};
use rowan::ast::AstNode;
use crate::{features::cursor::CursorContext, syntax::latex, LineIndexExt};
pub fn goto_command_definition(
context: &CursorContext<GotoDefinitionParams>,
) -> Option<Vec<LocationLink>> {
let main_document = context.request.main_document();
let name = context
.cursor
.as_latex()
.filter(|token| token.kind().is_command_name())?;
let origin_selection_range = main_document
.line_index
.line_col_lsp_range(name.text_range());
for document in context.request.workspace.documents_by_uri.values() {
if let Some(data) = document.data.as_latex() {
for node in latex::SyntaxNode::new_root(data.green.clone()).descendants() {
if let Some(defintion) = latex::CommandDefinition::cast(node).filter(|def| {
def.name()
.and_then(|name| name.command())
.map_or(false, |node| node.text() == name.text())
}) {
let target_selection_range = document
.line_index
.line_col_lsp_range(defintion.name()?.command()?.text_range());
let target_range = document
.line_index
.line_col_lsp_range(latex::small_range(&defintion));
return Some(vec![LocationLink {
origin_selection_range: Some(origin_selection_range),
target_uri: document.uri.as_ref().clone(),
target_range,
target_selection_range,
}]);
}
}
}
}
None
}
#[cfg(test)]
mod tests {
use indoc::indoc;
use lsp_types::Range;
use crate::{features::testing::FeatureTester, RangeExt};
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_command_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_command_definition(&context);
assert!(actual_links.is_none());
}
#[test]
fn test_command_definition() {
let tester = FeatureTester::builder()
.files(vec![(
"main.tex",
indoc! {
r#"
\DeclareMathOperator{\foo}{foo}
\foo
"#
},
)])
.main("main.tex")
.line(1)
.character(2)
.build();
let target_uri = tester.uri("main.tex").as_ref().clone();
let request = tester.definition();
let context = CursorContext::new(request);
let actual_links = goto_command_definition(&context).unwrap();
let expected_links = vec![LocationLink {
origin_selection_range: Some(Range::new_simple(1, 0, 1, 4)),
target_uri,
target_range: Range::new_simple(0, 0, 0, 31),
target_selection_range: Range::new_simple(0, 21, 0, 25),
}];
assert_eq!(actual_links, expected_links);
}
}
|