aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_ide/src/assists.rs
blob: b60b1a60d61c8cd7941845f32a7e1902ff08d515 (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
51
52
53
54
55
//! FIXME: write short doc here

use either::Either;
use ra_assists::{resolved_assists, AssistAction, AssistLabel};
use ra_db::{FilePosition, FileRange};
use ra_ide_db::RootDatabase;

use crate::{FileId, SourceChange, SourceFileEdit};

pub use ra_assists::AssistId;

#[derive(Debug)]
pub struct Assist {
    pub id: AssistId,
    pub label: String,
    pub change_data: Either<SourceChange, Vec<SourceChange>>,
}

pub(crate) fn assists(db: &RootDatabase, frange: FileRange) -> Vec<Assist> {
    resolved_assists(db, frange)
        .into_iter()
        .map(|assist| {
            let file_id = frange.file_id;
            let assist_label = &assist.label;
            Assist {
                id: assist_label.id,
                label: assist_label.label.clone(),
                change_data: match assist.action_data {
                    Either::Left(action) => {
                        Either::Left(action_to_edit(action, file_id, assist_label))
                    }
                    Either::Right(actions) => Either::Right(
                        actions
                            .into_iter()
                            .map(|action| action_to_edit(action, file_id, assist_label))
                            .collect(),
                    ),
                },
            }
        })
        .collect()
}

fn action_to_edit(
    action: AssistAction,
    file_id: FileId,
    assist_label: &AssistLabel,
) -> SourceChange {
    let file_edit = SourceFileEdit { file_id, edit: action.edit };
    SourceChange::source_file_edit(
        action.label.unwrap_or_else(|| assist_label.label.clone()),
        file_edit,
    )
    .with_cursor_opt(action.cursor_position.map(|offset| FilePosition { offset, file_id }))
}