summaryrefslogtreecommitdiff
path: root/support/texlab/src/definition/latex_command.rs
diff options
context:
space:
mode:
Diffstat (limited to 'support/texlab/src/definition/latex_command.rs')
-rw-r--r--support/texlab/src/definition/latex_command.rs106
1 files changed, 106 insertions, 0 deletions
diff --git a/support/texlab/src/definition/latex_command.rs b/support/texlab/src/definition/latex_command.rs
new file mode 100644
index 0000000000..7f91412ef7
--- /dev/null
+++ b/support/texlab/src/definition/latex_command.rs
@@ -0,0 +1,106 @@
+use crate::syntax::*;
+use crate::workspace::*;
+use futures_boxed::boxed;
+use lsp_types::{LocationLink, TextDocumentPositionParams};
+
+pub struct LatexCommandDefinitionProvider;
+
+impl FeatureProvider for LatexCommandDefinitionProvider {
+ type Params = TextDocumentPositionParams;
+ type Output = Vec<LocationLink>;
+
+ #[boxed]
+ async fn execute<'a>(&'a self, request: &'a FeatureRequest<Self::Params>) -> Self::Output {
+ let mut definitions = Vec::new();
+ if let SyntaxTree::Latex(tree) = &request.document().tree {
+ if let Some(LatexNode::Command(command)) = tree.find(request.params.position).last() {
+ for document in request.related_documents() {
+ if let SyntaxTree::Latex(tree) = &document.tree {
+ tree.command_definitions
+ .iter()
+ .filter(|def| def.definition.name.text() == command.name.text())
+ .map(|def| LocationLink {
+ origin_selection_range: Some(command.range()),
+ target_uri: document.uri.clone().into(),
+ target_range: def.range(),
+ target_selection_range: def.range(),
+ })
+ .for_each(|def| definitions.push(def));
+
+ tree.math
+ .operators
+ .iter()
+ .filter(|op| op.definition.name.text() == command.name.text())
+ .map(|op| LocationLink {
+ origin_selection_range: Some(command.range()),
+ target_uri: document.uri.clone().into(),
+ target_range: op.range(),
+ target_selection_range: op.range(),
+ })
+ .for_each(|def| definitions.push(def));
+ }
+ }
+ }
+ }
+ definitions
+ }
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+ use crate::range::RangeExt;
+ use lsp_types::{Position, Range};
+
+ #[test]
+ fn test_command_definition() {
+ let links = test_feature(
+ LatexCommandDefinitionProvider,
+ FeatureSpec {
+ files: vec![
+ FeatureSpec::file("foo.tex", "\\include{bar}\n\\foo"),
+ FeatureSpec::file("bar.tex", "\\newcommand{\\foo}{bar}"),
+ FeatureSpec::file("baz.tex", "\\newcommand{\\foo}{baz}"),
+ ],
+ main_file: "foo.tex",
+ position: Position::new(1, 3),
+ ..FeatureSpec::default()
+ },
+ );
+ assert_eq!(
+ links,
+ vec![LocationLink {
+ origin_selection_range: Some(Range::new_simple(1, 0, 1, 4)),
+ target_uri: FeatureSpec::uri("bar.tex"),
+ target_range: Range::new_simple(0, 0, 0, 22),
+ target_selection_range: Range::new_simple(0, 0, 0, 22),
+ }]
+ );
+ }
+
+ #[test]
+ fn test_math_operator() {
+ let links = test_feature(
+ LatexCommandDefinitionProvider,
+ FeatureSpec {
+ files: vec![
+ FeatureSpec::file("foo.tex", "\\include{bar}\n\\foo"),
+ FeatureSpec::file("bar.tex", "\\DeclareMathOperator{\\foo}{bar}"),
+ FeatureSpec::file("baz.tex", "\\DeclareMathOperator{\\foo}{baz}"),
+ ],
+ main_file: "foo.tex",
+ position: Position::new(1, 3),
+ ..FeatureSpec::default()
+ },
+ );
+ assert_eq!(
+ links,
+ vec![LocationLink {
+ origin_selection_range: Some(Range::new_simple(1, 0, 1, 4)),
+ target_uri: FeatureSpec::uri("bar.tex"),
+ target_range: Range::new_simple(0, 0, 0, 31),
+ target_selection_range: Range::new_simple(0, 0, 0, 31)
+ }]
+ );
+ }
+}