aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_ide/src/assists.rs
blob: 2b5d11681af79fe725f86bc89411a4ab505b3264 (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
//! FIXME: write short doc here

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 group_label: Option<String>,
    pub source_change: 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(),
                group_label: assist.group_label.map(|it| it.0),
                source_change: action_to_edit(assist.action, file_id, assist_label),
            }
        })
        .collect()
}

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