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
|
use cancellation::CancellationToken;
use lsp_types::{GotoDefinitionParams, LocationLink};
use crate::{
features::cursor::CursorContext, find_label_definition, render_label, syntax::CstNode,
LineIndexExt,
};
pub fn goto_label_definition(
context: &CursorContext<GotoDefinitionParams>,
cancellation_token: &CancellationToken,
) -> 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.subset.documents {
cancellation_token.result().ok()?;
if let Some(data) = document.data.as_latex() {
if let Some(definition) = find_label_definition(&data.root, &name_text) {
let target_selection_range = definition.name()?.key()?.small_range();
let target_range =
render_label(&context.request.subset, &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().into(),
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, CancellationToken::none());
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, CancellationToken::none());
assert!(actual_links.is_none());
}
}
|