summaryrefslogtreecommitdiff
path: root/support/texlab/src/completion/latex/import.rs
diff options
context:
space:
mode:
Diffstat (limited to 'support/texlab/src/completion/latex/import.rs')
-rw-r--r--support/texlab/src/completion/latex/import.rs110
1 files changed, 110 insertions, 0 deletions
diff --git a/support/texlab/src/completion/latex/import.rs b/support/texlab/src/completion/latex/import.rs
new file mode 100644
index 0000000000..7f333f17ae
--- /dev/null
+++ b/support/texlab/src/completion/latex/import.rs
@@ -0,0 +1,110 @@
+use super::combinators::{self, Parameter};
+use crate::completion::factory;
+use crate::completion::DATABASE;
+use crate::syntax::*;
+use crate::workspace::*;
+use futures_boxed::boxed;
+use lsp_types::{CompletionItem, CompletionParams, TextEdit};
+
+#[derive(Debug, PartialEq, Eq, Clone)]
+pub struct LatexClassImportProvider;
+
+impl FeatureProvider for LatexClassImportProvider {
+ type Params = CompletionParams;
+ type Output = Vec<CompletionItem>;
+
+ #[boxed]
+ async fn execute<'a>(&'a self, request: &'a FeatureRequest<Self::Params>) -> Self::Output {
+ import(request, LatexIncludeKind::Class, factory::class).await
+ }
+}
+
+#[derive(Debug, PartialEq, Eq, Clone)]
+pub struct LatexPackageImportProvider;
+
+impl FeatureProvider for LatexPackageImportProvider {
+ type Params = CompletionParams;
+ type Output = Vec<CompletionItem>;
+
+ #[boxed]
+ async fn execute<'a>(&'a self, request: &'a FeatureRequest<Self::Params>) -> Self::Output {
+ import(request, LatexIncludeKind::Package, factory::package).await
+ }
+}
+
+async fn import<F>(
+ request: &FeatureRequest<CompletionParams>,
+ kind: LatexIncludeKind,
+ factory: F,
+) -> Vec<CompletionItem>
+where
+ F: Fn(&FeatureRequest<CompletionParams>, &'static str, TextEdit) -> CompletionItem,
+{
+ let extension = if kind == LatexIncludeKind::Package {
+ "sty"
+ } else {
+ "cls"
+ };
+
+ let parameters = LANGUAGE_DATA
+ .include_commands
+ .iter()
+ .filter(|cmd| cmd.kind == kind)
+ .map(|cmd| Parameter::new(&cmd.name, cmd.index));
+
+ combinators::argument(request, parameters, |context| {
+ async move {
+ let mut items = Vec::new();
+ for component in &DATABASE.components {
+ for file_name in &component.file_names {
+ if file_name.ends_with(extension) {
+ let stem = &file_name[0..file_name.len() - 4];
+ let text_edit = TextEdit::new(context.range, stem.into());
+ let item = factory(request, stem, text_edit);
+ items.push(item);
+ }
+ }
+ }
+ items
+ }
+ })
+ .await
+}
+
+#[cfg(test)]
+mod tests {
+ use super::*;
+ use lsp_types::Position;
+
+ #[test]
+ fn test_class() {
+ let items = test_feature(
+ LatexClassImportProvider,
+ FeatureSpec {
+ files: vec![FeatureSpec::file("foo.tex", "\\documentclass{}")],
+ main_file: "foo.tex",
+ position: Position::new(0, 15),
+ ..FeatureSpec::default()
+ },
+ );
+
+ assert!(items.iter().any(|item| item.label == "beamer"));
+ assert!(items.iter().all(|item| item.label != "amsmath"));
+ }
+
+ #[test]
+ fn test_package() {
+ let items = test_feature(
+ LatexPackageImportProvider,
+ FeatureSpec {
+ files: vec![FeatureSpec::file("foo.tex", "\\usepackage{}")],
+ main_file: "foo.tex",
+ position: Position::new(0, 12),
+ ..FeatureSpec::default()
+ },
+ );
+
+ assert!(items.iter().all(|item| item.label != "beamer"));
+ assert!(items.iter().any(|item| item.label == "amsmath"));
+ }
+}