1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
use cancellation::CancellationToken;
use cstree::TextRange;
use lsp_types::CompletionParams;
use crate::{
features::cursor::CursorContext,
syntax::{latex, CstNode},
};
use super::types::{InternalCompletionItem, InternalCompletionItemData};
pub fn complete_user_environments<'a>(
context: &'a CursorContext<CompletionParams>,
items: &mut Vec<InternalCompletionItem<'a>>,
cancellation_token: &CancellationToken,
) -> Option<()> {
cancellation_token.result().ok()?;
let token = context.cursor.as_latex()?;
let group = latex::CurlyGroupWord::cast(token.parent())
.filter(|group| context.is_inside_latex_curly(group))?;
if !matches!(group.syntax().parent()?.kind(), latex::BEGIN | latex::END) {
return None;
}
let range = if token.kind() == latex::WORD {
token.text_range()
} else {
TextRange::empty(context.offset)
};
for document in &context.request.subset.documents {
if let Some(data) = document.data.as_latex() {
for name in data
.extras
.environment_names
.iter()
.filter(|name| name.as_str() != token.text())
{
cancellation_token.result().ok()?;
items.push(InternalCompletionItem::new(
range,
InternalCompletionItemData::UserEnvironment { name },
));
}
}
}
Some(())
}
#[cfg(test)]
mod tests {
use cstree::TextRange;
use crate::features::testing::FeatureTester;
use super::*;
#[test]
fn test_empty_latex_document() {
let request = FeatureTester::builder()
.files(vec![("main.tex", "")])
.main("main.tex")
.line(0)
.character(0)
.build()
.completion();
let context = CursorContext::new(request);
let mut actual_items = Vec::new();
complete_user_environments(&context, &mut actual_items, CancellationToken::none());
assert!(actual_items.is_empty());
}
#[test]
fn test_empty_bibtex_document() {
let request = FeatureTester::builder()
.files(vec![("main.bib", "")])
.main("main.bib")
.line(0)
.character(0)
.build()
.completion();
let context = CursorContext::new(request);
let mut actual_items = Vec::new();
complete_user_environments(&context, &mut actual_items, CancellationToken::none());
assert!(actual_items.is_empty());
}
#[test]
fn test_latex_simple() {
let request = FeatureTester::builder()
.files(vec![(
"main.tex",
"\\begin{foo}\\end{foo} \\begin{bar}\\end{bar}",
)])
.main("main.tex")
.line(0)
.character(7)
.build()
.completion();
let context = CursorContext::new(request);
let mut actual_items = Vec::new();
complete_user_environments(&context, &mut actual_items, CancellationToken::none());
assert_eq!(actual_items.len(), 1);
for item in actual_items {
assert_eq!(item.range, TextRange::new(7.into(), 10.into()));
}
}
}
|