summaryrefslogtreecommitdiff
path: root/support/texlab/crates/texlab/src/features/rename.rs
blob: 297547e7e5686c62b30634a1e516b1f4366457e5 (plain)
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
use std::collections::HashMap;

use base_db::{FeatureParams, Workspace};
use rename::RenameParams;

use crate::util::line_index_ext::LineIndexExt;

pub fn prepare_rename_all(
    workspace: &Workspace,
    params: &lsp_types::TextDocumentPositionParams,
) -> Option<lsp_types::Range> {
    let params = create_params(workspace, params)?;
    let range = rename::prepare_rename(&params)?;
    params.feature.document.line_index.line_col_lsp_range(range)
}

pub fn rename_all(
    workspace: &Workspace,
    params: &lsp_types::RenameParams,
) -> Option<lsp_types::WorkspaceEdit> {
    let new_name = &params.new_name;
    let params = create_params(workspace, &params.text_document_position)?;
    let result = rename::rename(params);

    let mut changes = HashMap::default();
    for (document, ranges) in result.changes {
        let mut edits = Vec::new();
        ranges
            .into_iter()
            .filter_map(|range| document.line_index.line_col_lsp_range(range))
            .for_each(|range| edits.push(lsp_types::TextEdit::new(range, new_name.clone())));

        changes.insert(document.uri.clone(), edits);
    }

    Some(lsp_types::WorkspaceEdit::new(changes))
}

fn create_params<'db>(
    workspace: &'db Workspace,
    params: &lsp_types::TextDocumentPositionParams,
) -> Option<RenameParams<'db>> {
    let document = workspace.lookup(&params.text_document.uri)?;
    let inner = FeatureParams::new(workspace, document);
    let offset = document.line_index.offset_lsp(params.position)?;
    Some(RenameParams {
        feature: inner,
        offset,
    })
}