diff options
Diffstat (limited to 'support/texlab/src/features/definition/document.rs')
-rw-r--r-- | support/texlab/src/features/definition/document.rs | 112 |
1 files changed, 112 insertions, 0 deletions
diff --git a/support/texlab/src/features/definition/document.rs b/support/texlab/src/features/definition/document.rs new file mode 100644 index 0000000000..b2b713f260 --- /dev/null +++ b/support/texlab/src/features/definition/document.rs @@ -0,0 +1,112 @@ +use cancellation::CancellationToken; +use lsp_types::{GotoDefinitionParams, LocationLink, Range}; + +use crate::{features::cursor::CursorContext, LineIndexExt, RangeExt}; + +pub fn goto_document_definition( + context: &CursorContext<GotoDefinitionParams>, + cancellation_token: &CancellationToken, +) -> Option<Vec<LocationLink>> { + let main_document = context.request.main_document(); + if let Some(data) = main_document.data.as_latex() { + for include in data + .extras + .explicit_links + .iter() + .filter(|link| link.stem_range.contains_inclusive(context.offset)) + { + for target in &include.targets { + cancellation_token.result().ok()?; + + if context + .request + .subset + .documents + .iter() + .any(|document| document.uri.as_ref() == target.as_ref()) + { + return Some(vec![LocationLink { + origin_selection_range: Some( + main_document + .line_index + .line_col_lsp_range(include.stem_range), + ), + target_uri: target.as_ref().clone().into(), + target_range: Range::new_simple(0, 0, 0, 0), + target_selection_range: Range::new_simple(0, 0, 0, 0), + }]); + } + } + } + } + None +} + +#[cfg(test)] +mod tests { + 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_document_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_document_definition(&context, CancellationToken::none()); + + assert!(actual_links.is_none()); + } + + #[test] + fn test_simple() { + let tester = FeatureTester::builder() + .files(vec![ + ("foo.tex", r#"\addbibresource{baz.bib}"#), + ("bar.bib", r#"@article{foo, bar = {baz}}"#), + ("baz.bib", r#"@article{foo, bar = {baz}}"#), + ]) + .main("foo.tex") + .line(0) + .character(18) + .build(); + let target_uri = tester.uri("baz.bib").as_ref().clone().into(); + + let request = tester.definition(); + let context = CursorContext::new(request); + let actual_links = goto_document_definition(&context, CancellationToken::none()).unwrap(); + + let expected_links = vec![LocationLink { + origin_selection_range: Some(Range::new_simple(0, 16, 0, 23)), + target_uri, + target_range: Range::new_simple(0, 0, 0, 0), + target_selection_range: Range::new_simple(0, 0, 0, 0), + }]; + + assert_eq!(actual_links, expected_links); + } +} |