language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-09-28 21:05:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Simplify

Browse source
This commit is contained in:
Aleksey Kladov 2020-02-07 14:53:50 +01:00
parent 8337dcd9e2
commit b831b17b3d
2 changed files with 29 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

16
crates/ra_assists/src/assist_ctx.rs
View file

@ -69,23 +69,11 @@ impl<'a> Clone for AssistCtx<'a> {
}
impl<'a> AssistCtx<'a> {
pub(crate) fn with_ctx<F, T>(
db: &RootDatabase,
frange: FileRange,
should_compute_edit: bool,
f: F,
) -> T
where
F: FnOnce(AssistCtx) -> T,
{
pub fn new(db: &RootDatabase, frange: FileRange, should_compute_edit: bool) -> AssistCtx {
let parse = db.parse(frange.file_id);
let ctx = AssistCtx { db, frange, source_file: parse.tree(), should_compute_edit };
f(ctx)
AssistCtx { db, frange, source_file: parse.tree(), should_compute_edit }
}
}
impl<'a> AssistCtx<'a> {
pub(crate) fn add_assist(
self,
id: AssistId,

19
crates/ra_assists/src/lib.rs
View file

@ -37,6 +37,7 @@ pub struct AssistAction {
pub label: Option<String>,
pub edit: TextEdit,
pub cursor_position: Option<TextUnit>,
// FIXME: This belongs to `AssistLabel`
pub target: Option<TextRange>,
}
@ -60,7 +61,7 @@ impl ResolvedAssist {
/// Assists are returned in the "unresolved" state, that is only labels are
/// returned, without actual edits.
pub fn unresolved_assists(db: &RootDatabase, range: FileRange) -> Vec<AssistLabel> {
AssistCtx::with_ctx(db, range, false, |ctx| {
let ctx = AssistCtx::new(db, range, false);
assists::all()
.iter()
.filter_map(|f| f(ctx.clone()))
@ -69,7 +70,6 @@ pub fn unresolved_assists(db: &RootDatabase, range: FileRange) -> Vec<AssistLabe
Assist::Resolved { .. } => unreachable!(),
})
.collect()
})
}
/// Return all the assists applicable at the given position.
@ -77,7 +77,7 @@ pub fn unresolved_assists(db: &RootDatabase, range: FileRange) -> Vec<AssistLabe
/// Assists are returned in the "resolved" state, that is with edit fully
/// computed.
pub fn resolved_assists(db: &RootDatabase, range: FileRange) -> Vec<ResolvedAssist> {
AssistCtx::with_ctx(db, range, true, |ctx| {
let ctx = AssistCtx::new(db, range, true);
let mut a = assists::all()
.iter()
.filter_map(|f| f(ctx.clone()))
@ -88,7 +88,6 @@ pub fn resolved_assists(db: &RootDatabase, range: FileRange) -> Vec<ResolvedAssi
.collect();
sort_assists(&mut a);
a
})
}
fn sort_assists(assists: &mut Vec<ResolvedAssist>) {
@ -192,7 +191,7 @@ mod helpers {
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");
assist(AssistCtx::new(&db, frange, true)).expect("code action is not applicable");
let action = match assist {
Assist::Unresolved { .. } => unreachable!(),
Assist::Resolved { assist } => assist.get_first_action(),
@ -219,7 +218,7 @@ mod helpers {
let (db, file_id) = 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");
assist(AssistCtx::new(&db, frange, true)).expect("code action is not applicable");
let action = match assist {
Assist::Unresolved { .. } => unreachable!(),
Assist::Resolved { assist } => assist.get_first_action(),
@ -242,7 +241,7 @@ mod helpers {
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");
assist(AssistCtx::new(&db, frange, true)).expect("code action is not applicable");
let action = match assist {
Assist::Unresolved { .. } => unreachable!(),
Assist::Resolved { assist } => assist.get_first_action(),
@ -261,7 +260,7 @@ mod helpers {
let (db, file_id) = 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");
assist(AssistCtx::new(&db, frange, true)).expect("code action is not applicable");
let action = match assist {
Assist::Unresolved { .. } => unreachable!(),
Assist::Resolved { assist } => assist.get_first_action(),
@ -279,7 +278,7 @@ mod helpers {
let (db, file_id) = 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);
let assist = assist(AssistCtx::new(&db, frange, true));
assert!(assist.is_none());
}
@ -290,7 +289,7 @@ mod helpers {
let (range, before) = extract_range(before);
let (db, file_id) = with_single_file(&before);
let frange = FileRange { file_id, range };
let assist = AssistCtx::with_ctx(&db, frange, true, assist);
let assist = assist(AssistCtx::new(&db, frange, true));
assert!(assist.is_none());
}
}