344 lines
12 KiB
Rust
344 lines
12 KiB
Rust
//! `ra_assists` crate provides a bunch of code assists, also known as code
|
|
//! actions (in LSP) or intentions (in IntelliJ).
|
|
//!
|
|
//! An assist is a micro-refactoring, which is automatically activated in
|
|
//! certain context. For example, if the cursor is over `,`, a "swap `,`" assist
|
|
//! becomes available.
|
|
|
|
mod assist_ctx;
|
|
mod marks;
|
|
pub mod ast_editor;
|
|
|
|
use hir::db::HirDatabase;
|
|
use itertools::Itertools;
|
|
use ra_db::FileRange;
|
|
use ra_syntax::{TextRange, TextUnit};
|
|
use ra_text_edit::TextEdit;
|
|
|
|
pub(crate) use crate::assist_ctx::{Assist, AssistCtx};
|
|
pub use crate::assists::auto_import::auto_import_text_edit;
|
|
|
|
/// Unique identifier of the assist, should not be shown to the user
|
|
/// directly.
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub struct AssistId(pub &'static str);
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct AssistLabel {
|
|
/// Short description of the assist, as shown in the UI.
|
|
pub label: String,
|
|
pub id: AssistId,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct AssistAction {
|
|
pub edit: TextEdit,
|
|
pub cursor_position: Option<TextUnit>,
|
|
pub target: Option<TextRange>,
|
|
}
|
|
|
|
/// Return all the assists eapplicable at the given position.
|
|
///
|
|
/// Assists are returned in the "unresolved" state, that is only labels are
|
|
/// returned, without actual edits.
|
|
pub fn applicable_assists<H>(db: &H, range: FileRange) -> Vec<AssistLabel>
|
|
where
|
|
H: HirDatabase + 'static,
|
|
{
|
|
AssistCtx::with_ctx(db, range, false, |ctx| {
|
|
assists::all()
|
|
.iter()
|
|
.filter_map(|f| f(ctx.clone()))
|
|
.map(|a| match a {
|
|
Assist::Unresolved(labels) => labels,
|
|
Assist::Resolved(..) => unreachable!(),
|
|
})
|
|
.concat()
|
|
})
|
|
}
|
|
|
|
/// Return all the assists applicable at the given position.
|
|
///
|
|
/// Assists are returned in the "resolved" state, that is with edit fully
|
|
/// computed.
|
|
pub fn assists<H>(db: &H, range: FileRange) -> Vec<(AssistLabel, AssistAction)>
|
|
where
|
|
H: HirDatabase + 'static,
|
|
{
|
|
use std::cmp::Ordering;
|
|
|
|
AssistCtx::with_ctx(db, range, true, |ctx| {
|
|
let mut a = assists::all()
|
|
.iter()
|
|
.filter_map(|f| f(ctx.clone()))
|
|
.map(|a| match a {
|
|
Assist::Resolved(labels_actions) => labels_actions,
|
|
Assist::Unresolved(..) => unreachable!(),
|
|
})
|
|
.concat();
|
|
a.sort_by(|a, b| match (a.1.target, b.1.target) {
|
|
(Some(a), Some(b)) => a.len().cmp(&b.len()),
|
|
(Some(_), None) => Ordering::Less,
|
|
(None, Some(_)) => Ordering::Greater,
|
|
(None, None) => Ordering::Equal,
|
|
});
|
|
a
|
|
})
|
|
}
|
|
|
|
mod assists {
|
|
use crate::{Assist, AssistCtx};
|
|
use hir::db::HirDatabase;
|
|
|
|
mod add_derive;
|
|
mod add_explicit_type;
|
|
mod add_impl;
|
|
mod flip_comma;
|
|
mod flip_binexpr;
|
|
mod change_visibility;
|
|
mod fill_match_arms;
|
|
mod merge_match_arms;
|
|
mod introduce_variable;
|
|
mod inline_local_variable;
|
|
mod raw_string;
|
|
mod replace_if_let_with_match;
|
|
mod split_import;
|
|
mod remove_dbg;
|
|
pub(crate) mod auto_import;
|
|
mod add_missing_impl_members;
|
|
mod move_guard;
|
|
mod move_bounds;
|
|
|
|
pub(crate) fn all<DB: HirDatabase>() -> &'static [fn(AssistCtx<DB>) -> Option<Assist>] {
|
|
&[
|
|
add_derive::add_derive,
|
|
add_explicit_type::add_explicit_type,
|
|
add_impl::add_impl,
|
|
change_visibility::change_visibility,
|
|
fill_match_arms::fill_match_arms,
|
|
merge_match_arms::merge_match_arms,
|
|
flip_comma::flip_comma,
|
|
flip_binexpr::flip_binexpr,
|
|
introduce_variable::introduce_variable,
|
|
replace_if_let_with_match::replace_if_let_with_match,
|
|
split_import::split_import,
|
|
remove_dbg::remove_dbg,
|
|
auto_import::auto_import,
|
|
add_missing_impl_members::add_missing_impl_members,
|
|
add_missing_impl_members::add_missing_default_members,
|
|
inline_local_variable::inline_local_varialbe,
|
|
move_guard::move_guard_to_arm_body,
|
|
move_guard::move_arm_cond_to_match_guard,
|
|
move_bounds::move_bounds_to_where_clause,
|
|
raw_string::add_hash,
|
|
raw_string::make_raw_string,
|
|
raw_string::make_usual_string,
|
|
raw_string::remove_hash,
|
|
]
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod helpers {
|
|
use hir::mock::MockDatabase;
|
|
use ra_db::FileRange;
|
|
use ra_syntax::TextRange;
|
|
use test_utils::{add_cursor, assert_eq_text, extract_offset, extract_range};
|
|
|
|
use crate::{Assist, AssistCtx};
|
|
|
|
pub(crate) fn check_assist(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
after: &str,
|
|
) {
|
|
check_assist_nth_action(assist, before, after, 0)
|
|
}
|
|
|
|
pub(crate) fn check_assist_range(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
after: &str,
|
|
) {
|
|
check_assist_range_nth_action(assist, before, after, 0)
|
|
}
|
|
|
|
pub(crate) fn check_assist_target(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
target: &str,
|
|
) {
|
|
check_assist_target_nth_action(assist, before, target, 0)
|
|
}
|
|
|
|
pub(crate) fn check_assist_range_target(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
target: &str,
|
|
) {
|
|
check_assist_range_target_nth_action(assist, before, target, 0)
|
|
}
|
|
|
|
pub(crate) fn check_assist_nth_action(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
after: &str,
|
|
index: usize,
|
|
) {
|
|
let (before_cursor_pos, before) = extract_offset(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange =
|
|
FileRange { file_id, range: TextRange::offset_len(before_cursor_pos, 0.into()) };
|
|
let assist =
|
|
AssistCtx::with_ctx(&db, frange, true, assist).expect("code action is not applicable");
|
|
let labels_actions = match assist {
|
|
Assist::Unresolved(_) => unreachable!(),
|
|
Assist::Resolved(labels_actions) => labels_actions,
|
|
};
|
|
|
|
let (_, action) = labels_actions.get(index).expect("expect assist action at index");
|
|
let actual = action.edit.apply(&before);
|
|
let actual_cursor_pos = match action.cursor_position {
|
|
None => action
|
|
.edit
|
|
.apply_to_offset(before_cursor_pos)
|
|
.expect("cursor position is affected by the edit"),
|
|
Some(off) => off,
|
|
};
|
|
let actual = add_cursor(&actual, actual_cursor_pos);
|
|
assert_eq_text!(after, &actual);
|
|
}
|
|
|
|
pub(crate) fn check_assist_range_nth_action(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
after: &str,
|
|
index: usize,
|
|
) {
|
|
let (range, before) = extract_range(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange = FileRange { file_id, range };
|
|
let assist =
|
|
AssistCtx::with_ctx(&db, frange, true, assist).expect("code action is not applicable");
|
|
let labels_actions = match assist {
|
|
Assist::Unresolved(_) => unreachable!(),
|
|
Assist::Resolved(labels_actions) => labels_actions,
|
|
};
|
|
|
|
let (_, action) = labels_actions.get(index).expect("expect assist action at index");
|
|
let mut actual = action.edit.apply(&before);
|
|
if let Some(pos) = action.cursor_position {
|
|
actual = add_cursor(&actual, pos);
|
|
}
|
|
assert_eq_text!(after, &actual);
|
|
}
|
|
|
|
pub(crate) fn check_assist_target_nth_action(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
target: &str,
|
|
index: usize,
|
|
) {
|
|
let (before_cursor_pos, before) = extract_offset(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange =
|
|
FileRange { file_id, range: TextRange::offset_len(before_cursor_pos, 0.into()) };
|
|
let assist =
|
|
AssistCtx::with_ctx(&db, frange, true, assist).expect("code action is not applicable");
|
|
let labels_actions = match assist {
|
|
Assist::Unresolved(_) => unreachable!(),
|
|
Assist::Resolved(labels_actions) => labels_actions,
|
|
};
|
|
|
|
let (_, action) = labels_actions.get(index).expect("expect assist action at index");
|
|
let range = action.target.expect("expected target on action");
|
|
assert_eq_text!(&before[range.start().to_usize()..range.end().to_usize()], target);
|
|
}
|
|
|
|
pub(crate) fn check_assist_range_target_nth_action(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
target: &str,
|
|
index: usize,
|
|
) {
|
|
let (range, before) = extract_range(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange = FileRange { file_id, range };
|
|
let assist =
|
|
AssistCtx::with_ctx(&db, frange, true, assist).expect("code action is not applicable");
|
|
let labels_actions = match assist {
|
|
Assist::Unresolved(_) => unreachable!(),
|
|
Assist::Resolved(labels_actions) => labels_actions,
|
|
};
|
|
|
|
let (_, action) = labels_actions.get(index).expect("expect assist action at index");
|
|
let range = action.target.expect("expected target on action");
|
|
assert_eq_text!(&before[range.start().to_usize()..range.end().to_usize()], target);
|
|
}
|
|
|
|
pub(crate) fn check_assist_not_applicable(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
) {
|
|
let (before_cursor_pos, before) = extract_offset(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange =
|
|
FileRange { file_id, range: TextRange::offset_len(before_cursor_pos, 0.into()) };
|
|
let assist = AssistCtx::with_ctx(&db, frange, true, assist);
|
|
assert!(assist.is_none());
|
|
}
|
|
|
|
pub(crate) fn check_assist_range_not_applicable(
|
|
assist: fn(AssistCtx<MockDatabase>) -> Option<Assist>,
|
|
before: &str,
|
|
) {
|
|
let (range, before) = extract_range(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange = FileRange { file_id, range };
|
|
let assist = AssistCtx::with_ctx(&db, frange, true, assist);
|
|
assert!(assist.is_none());
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use hir::mock::MockDatabase;
|
|
use ra_db::FileRange;
|
|
use ra_syntax::TextRange;
|
|
use test_utils::{extract_offset, extract_range};
|
|
|
|
#[test]
|
|
fn assist_order_field_struct() {
|
|
let before = "struct Foo { <|>bar: u32 }";
|
|
let (before_cursor_pos, before) = extract_offset(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange =
|
|
FileRange { file_id, range: TextRange::offset_len(before_cursor_pos, 0.into()) };
|
|
let assists = super::assists(&db, frange);
|
|
let mut assists = assists.iter();
|
|
|
|
assert_eq!(assists.next().expect("expected assist").0.label, "make pub(crate)");
|
|
assert_eq!(assists.next().expect("expected assist").0.label, "add `#[derive]`");
|
|
}
|
|
|
|
#[test]
|
|
fn assist_order_if_expr() {
|
|
let before = "
|
|
pub fn test_some_range(a: int) -> bool {
|
|
if let 2..6 = <|>5<|> {
|
|
true
|
|
} else {
|
|
false
|
|
}
|
|
}";
|
|
let (range, before) = extract_range(before);
|
|
let (db, _source_root, file_id) = MockDatabase::with_single_file(&before);
|
|
let frange = FileRange { file_id, range };
|
|
let assists = super::assists(&db, frange);
|
|
let mut assists = assists.iter();
|
|
|
|
assert_eq!(assists.next().expect("expected assist").0.label, "introduce variable");
|
|
assert_eq!(assists.next().expect("expected assist").0.label, "replace with match");
|
|
}
|
|
}
|