diff options
Diffstat (limited to 'support/texlab/src/completion/latex/argument.rs')
-rw-r--r-- | support/texlab/src/completion/latex/argument.rs | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/support/texlab/src/completion/latex/argument.rs b/support/texlab/src/completion/latex/argument.rs new file mode 100644 index 0000000000..db05bca824 --- /dev/null +++ b/support/texlab/src/completion/latex/argument.rs @@ -0,0 +1,117 @@ +use super::combinators::{self, Parameter}; +use crate::completion::factory; +use crate::completion::DATABASE; +use crate::workspace::*; +use futures_boxed::boxed; +use lsp_types::*; +use std::iter; + +#[derive(Debug, PartialEq, Eq, Clone, Copy)] +pub struct LatexArgumentCompletionProvider; + +impl FeatureProvider for LatexArgumentCompletionProvider { + type Params = CompletionParams; + type Output = Vec<CompletionItem>; + + #[boxed] + async fn execute<'a>(&'a self, request: &'a FeatureRequest<Self::Params>) -> Self::Output { + let mut all_items = Vec::new(); + for component in DATABASE.related_components(request.related_documents()) { + for command in &component.commands { + let name = format!("\\{}", command.name); + for (i, parameter) in command.parameters.iter().enumerate() { + let mut items = combinators::argument( + request, + iter::once(Parameter::new(&name, i)), + |context| { + async move { + let mut items = Vec::new(); + for argument in ¶meter.0 { + let text_edit = + TextEdit::new(context.range, (&argument.name).into()); + let item = factory::argument( + request, + &argument.name, + text_edit, + argument.image.as_ref().map(AsRef::as_ref), + ); + items.push(item); + } + items + } + }, + ) + .await; + all_items.append(&mut items); + } + } + } + all_items + } +} + +#[cfg(test)] +mod tests { + use super::*; + use crate::range::RangeExt; + use lsp_types::{Position, Range}; + + #[test] + fn test_inside_mathbb_empty() { + let items = test_feature( + LatexArgumentCompletionProvider, + FeatureSpec { + files: vec![FeatureSpec::file( + "foo.tex", + "\\usepackage{amsfonts}\n\\mathbb{}", + )], + main_file: "foo.tex", + position: Position::new(1, 8), + ..FeatureSpec::default() + }, + ); + assert!(!items.is_empty()); + assert_eq!( + items[0].text_edit.as_ref().map(|edit| edit.range), + Some(Range::new_simple(1, 8, 1, 8)) + ); + } + + #[test] + fn test_inside_mathbb_non_empty() { + let items = test_feature( + LatexArgumentCompletionProvider, + FeatureSpec { + files: vec![FeatureSpec::file( + "foo.tex", + "\\usepackage{amsfonts}\n\\mathbb{foo}", + )], + main_file: "foo.tex", + position: Position::new(1, 8), + ..FeatureSpec::default() + }, + ); + assert!(!items.is_empty()); + assert_eq!( + items[0].text_edit.as_ref().map(|edit| edit.range), + Some(Range::new_simple(1, 8, 1, 11)) + ); + } + + #[test] + fn test_outside_mathbb() { + let items = test_feature( + LatexArgumentCompletionProvider, + FeatureSpec { + files: vec![FeatureSpec::file( + "foo.tex", + "\\usepackage{amsfonts}\n\\mathbb{}", + )], + main_file: "foo.tex", + position: Position::new(1, 9), + ..FeatureSpec::default() + }, + ); + assert!(items.is_empty()); + } +} |