summaryrefslogtreecommitdiff
path: root/support/texlab/src/definition/latex_citation.rs
diff options
context:
space:
mode:
Diffstat (limited to 'support/texlab/src/definition/latex_citation.rs')
-rw-r--r--support/texlab/src/definition/latex_citation.rs119
1 files changed, 119 insertions, 0 deletions
diff --git a/support/texlab/src/definition/latex_citation.rs b/support/texlab/src/definition/latex_citation.rs
new file mode 100644
index 0000000000..daefa8b2ef
--- /dev/null
+++ b/support/texlab/src/definition/latex_citation.rs
@@ -0,0 +1,119 @@
+use crate::range::RangeExt;
+use crate::syntax::*;
+use crate::workspace::*;
+use futures_boxed::boxed;
+use lsp_types::{LocationLink, TextDocumentPositionParams};
+
+#[derive(Debug, PartialEq, Eq, Clone)]
+pub struct LatexCitationDefinitionProvider;
+
+impl FeatureProvider for LatexCitationDefinitionProvider {
+ type Params = TextDocumentPositionParams;
+ type Output = Vec<LocationLink>;
+
+ #[boxed]
+ async fn execute<'a>(&'a self, request: &'a FeatureRequest<Self::Params>) -> Self::Output {
+ let mut links = Vec::new();
+ if let Some(reference) = Self::find_reference(&request) {
+ for document in request.related_documents() {
+ Self::find_definitions(&document, &reference, &mut links);
+ }
+ }
+ links
+ }
+}
+
+impl LatexCitationDefinitionProvider {
+ fn find_definitions(
+ document: &Document,
+ reference: &LatexToken,
+ links: &mut Vec<LocationLink>,
+ ) {
+ if let SyntaxTree::Bibtex(tree) = &document.tree {
+ for entry in tree.entries() {
+ if let Some(key) = &entry.key {
+ if key.text() == reference.text() {
+ let link = LocationLink {
+ origin_selection_range: Some(reference.range()),
+ target_uri: document.uri.clone().into(),
+ target_range: entry.range(),
+ target_selection_range: key.range(),
+ };
+ links.push(link);
+ }
+ }
+ }
+ }
+ }
+
+ fn find_reference(request: &FeatureRequest<TextDocumentPositionParams>) -> Option<&LatexToken> {
+ if let SyntaxTree::Latex(tree) = &request.document().tree {
+ tree.citations
+ .iter()
+ .flat_map(LatexCitation::keys)
+ .find(|key| key.range().contains(request.params.position))
+ } else {
+ None
+ }
+ }
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+ use lsp_types::{Position, Range};
+
+ #[test]
+ fn test_has_definition() {
+ let links = test_feature(
+ LatexCitationDefinitionProvider,
+ FeatureSpec {
+ files: vec![
+ FeatureSpec::file("foo.tex", "\\addbibresource{baz.bib}\n\\cite{foo}"),
+ FeatureSpec::file("bar.bib", "@article{foo, bar = {baz}}"),
+ FeatureSpec::file("baz.bib", "@article{foo, bar = {baz}}"),
+ ],
+ main_file: "foo.tex",
+ position: Position::new(1, 6),
+ ..FeatureSpec::default()
+ },
+ );
+ assert_eq!(
+ links,
+ vec![LocationLink {
+ origin_selection_range: Some(Range::new_simple(1, 6, 1, 9)),
+ target_uri: FeatureSpec::uri("baz.bib"),
+ target_range: Range::new_simple(0, 0, 0, 26),
+ target_selection_range: Range::new_simple(0, 9, 0, 12)
+ }]
+ );
+ }
+
+ #[test]
+ fn test_no_definition_latex() {
+ let links = test_feature(
+ LatexCitationDefinitionProvider,
+ FeatureSpec {
+ files: vec![FeatureSpec::file("foo.tex", "")],
+ main_file: "foo.tex",
+ position: Position::new(0, 0),
+ ..FeatureSpec::default()
+ },
+ );
+ assert!(links.is_empty());
+ }
+
+ #[test]
+ fn test_no_definition_bibtex() {
+ let links = test_feature(
+ LatexCitationDefinitionProvider,
+ FeatureSpec {
+ files: vec![FeatureSpec::file("foo.bib", "")],
+ main_file: "foo.bib",
+ position: Position::new(0, 0),
+ ..FeatureSpec::default()
+ },
+ );
+ assert!(links.is_empty());
+ }
+}