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
|
use std::collections::HashMap;
use lsp_types::{Range, RenameParams, TextEdit, WorkspaceEdit};
use rowan::{TextRange, TextSize};
use crate::{
features::cursor::{CursorContext, HasPosition},
syntax::latex,
LineIndexExt,
};
pub fn prepare_command_rename<P: HasPosition>(context: &CursorContext<P>) -> Option<Range> {
Some(
context
.request
.main_document()
.line_index
.line_col_lsp_range(context.cursor.command_range(context.offset)?),
)
}
pub fn rename_command(context: &CursorContext<RenameParams>) -> Option<WorkspaceEdit> {
prepare_command_rename(context)?;
let name = context.cursor.as_latex()?.text();
let mut changes = HashMap::new();
for document in context.request.workspace.documents_by_uri.values() {
if let Some(data) = document.data.as_latex() {
let edits = latex::SyntaxNode::new_root(data.green.clone())
.descendants_with_tokens()
.filter_map(|element| element.into_token())
.filter(|token| token.kind().is_command_name() && token.text() == name)
.map(|token| {
let range = token.text_range();
let range = document.line_index.line_col_lsp_range(TextRange::new(
range.start() + TextSize::from(1),
range.end(),
));
TextEdit::new(range, context.request.params.new_name.clone())
})
.collect();
changes.insert(document.uri.as_ref().clone(), edits);
}
}
Some(WorkspaceEdit::new(changes))
}
#[cfg(test)]
mod tests {
use crate::{features::testing::FeatureTester, RangeExt};
use super::*;
#[test]
fn test_command() {
let tester = FeatureTester::builder()
.files(vec![
("foo.tex", r#"\baz\include{bar.tex}"#),
("bar.tex", r#"\baz"#),
])
.main("foo.tex")
.line(0)
.character(2)
.new_name("qux")
.build();
let uri1 = tester.uri("foo.tex");
let uri2 = tester.uri("bar.tex");
let req = tester.rename();
let context = CursorContext::new(req);
let actual_edit = rename_command(&context).unwrap();
let mut expected_changes = HashMap::new();
expected_changes.insert(
uri1.as_ref().clone(),
vec![TextEdit::new(Range::new_simple(0, 1, 0, 4), "qux".into())],
);
expected_changes.insert(
uri2.as_ref().clone(),
vec![TextEdit::new(Range::new_simple(0, 1, 0, 4), "qux".into())],
);
let expected_edit = WorkspaceEdit::new(expected_changes);
assert_eq!(actual_edit, expected_edit);
}
}
|