summaryrefslogtreecommitdiff
path: root/support/texlab/src/completion/preselect.rs
diff options
context:
space:
mode:
Diffstat (limited to 'support/texlab/src/completion/preselect.rs')
-rw-r--r--support/texlab/src/completion/preselect.rs55
1 files changed, 55 insertions, 0 deletions
diff --git a/support/texlab/src/completion/preselect.rs b/support/texlab/src/completion/preselect.rs
new file mode 100644
index 0000000000..dd4265f5c6
--- /dev/null
+++ b/support/texlab/src/completion/preselect.rs
@@ -0,0 +1,55 @@
+use crate::range::RangeExt;
+use crate::syntax::*;
+use crate::workspace::*;
+use futures_boxed::boxed;
+use lsp_types::{CompletionItem, CompletionParams};
+
+#[derive(Debug)]
+pub struct PreselectCompletionProvider<F> {
+ provider: F,
+}
+
+impl<F> PreselectCompletionProvider<F> {
+ pub fn new(provider: F) -> Self {
+ Self { provider }
+ }
+}
+
+impl<F> FeatureProvider for PreselectCompletionProvider<F>
+where
+ F: FeatureProvider<Params = CompletionParams, Output = Vec<CompletionItem>> + Send + Sync,
+{
+ type Params = CompletionParams;
+ type Output = Vec<CompletionItem>;
+
+ #[boxed]
+ async fn execute<'a>(&'a self, request: &'a FeatureRequest<Self::Params>) -> Self::Output {
+ let mut items = self.provider.execute(request).await;
+ if let SyntaxTree::Latex(tree) = &request.document().tree {
+ for environment in &tree.env.environments {
+ if let Some(name) = environment.left.name() {
+ let right_args = &environment.right.command.args[0];
+ let cond1 = right_args
+ .range()
+ .contains_exclusive(request.params.text_document_position.position);
+ let cond2 = right_args.right.is_none()
+ && right_args
+ .range()
+ .contains(request.params.text_document_position.position);
+
+ if cond1 || cond2 {
+ for item in &mut items {
+ item.preselect = Some(false);
+ if item.label == name.text() {
+ item.preselect = Some(true);
+ break;
+ }
+ }
+ }
+ }
+ }
+ }
+
+ items
+ }
+}