name = "ra_assists"
version = "0.1.0"
dependencies = [
+ "itertools 0.8.0 (registry+https://github.com/rust-lang/crates.io-index)",
"join_to_string 0.1.3 (registry+https://github.com/rust-lang/crates.io-index)",
"ra_db 0.1.0",
"ra_fmt 0.1.0",
[dependencies]
join_to_string = "0.1.3"
+itertools = "0.8.0"
ra_syntax = { path = "../ra_syntax" }
ra_text_edit = { path = "../ra_text_edit" }
use crate::{AssistCtx, Assist};
-pub(crate) fn add_derive(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn add_derive(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let nominal = ctx.node_at_offset::<ast::NominalDef>()?;
let node_start = derive_insertion_offset(nominal)?;
- ctx.build("add `#[derive]`", |edit| {
+ ctx.add_action("add `#[derive]`", |edit| {
let derive_attr = nominal
.attrs()
.filter_map(|x| x.as_call())
};
edit.target(nominal.syntax().range());
edit.set_cursor(offset)
- })
+ });
+
+ ctx.build()
}
// Insert `derive` after doc comments.
use crate::{AssistCtx, Assist};
-pub(crate) fn add_impl(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn add_impl(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let nominal = ctx.node_at_offset::<ast::NominalDef>()?;
let name = nominal.name()?;
- ctx.build("add impl", |edit| {
+ ctx.add_action("add impl", |edit| {
edit.target(nominal.syntax().range());
let type_params = nominal.type_param_list();
let start_offset = nominal.syntax().range().end();
edit.set_cursor(start_offset + TextUnit::of_str(&buf));
buf.push_str("\n}");
edit.insert(start_offset, buf);
- })
+ });
+
+ ctx.build()
}
#[cfg(test)]
use crate::{AssistLabel, AssistAction};
+#[derive(Clone, Debug)]
pub(crate) enum Assist {
- Unresolved(AssistLabel),
- Resolved(AssistLabel, AssistAction),
+ Unresolved(Vec<AssistLabel>),
+ Resolved(Vec<(AssistLabel, AssistAction)>),
}
/// `AssistCtx` allows to apply an assist or check if it could be applied.
pub(crate) frange: FileRange,
source_file: &'a SourceFile,
should_compute_edit: bool,
+ assist: Assist,
}
impl<'a, DB> Clone for AssistCtx<'a, DB> {
frange: self.frange,
source_file: self.source_file,
should_compute_edit: self.should_compute_edit,
+ assist: self.assist.clone(),
}
}
}
F: FnOnce(AssistCtx<DB>) -> T,
{
let source_file = &db.parse(frange.file_id);
- let ctx = AssistCtx { db, frange, source_file, should_compute_edit };
+ let assist =
+ if should_compute_edit { Assist::Resolved(vec![]) } else { Assist::Unresolved(vec![]) };
+
+ let ctx = AssistCtx { db, frange, source_file, should_compute_edit, assist };
f(ctx)
}
- pub(crate) fn build(
- self,
+ pub(crate) fn add_action(
+ &mut self,
label: impl Into<String>,
f: impl FnOnce(&mut AssistBuilder),
- ) -> Option<Assist> {
+ ) -> &mut Self {
let label = AssistLabel { label: label.into() };
- if !self.should_compute_edit {
- return Some(Assist::Unresolved(label));
+ match &mut self.assist {
+ Assist::Unresolved(labels) => labels.push(label),
+ Assist::Resolved(labels_actions) => {
+ let action = {
+ let mut edit = AssistBuilder::default();
+ f(&mut edit);
+ edit.build()
+ };
+ labels_actions.push((label, action));
+ }
}
- let action = {
- let mut edit = AssistBuilder::default();
- f(&mut edit);
- edit.build()
- };
- Some(Assist::Resolved(label, action))
+ self
+ }
+
+ pub(crate) fn build(self) -> Option<Assist> {
+ Some(self.assist)
}
pub(crate) fn leaf_at_offset(&self) -> LeafAtOffset<&'a SyntaxNode> {
}
}
-pub(crate) fn auto_import(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn auto_import(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let node = ctx.covering_node();
let current_file = node.ancestors().find_map(ast::SourceFile::cast)?;
return None;
}
- ctx.build(format!("import {} in the current file", fmt_segments(&segments)), |edit| {
+ ctx.add_action(format!("import {} in the current file", fmt_segments(&segments)), |edit| {
let action = best_action_for_target(current_file.syntax(), path, &segments);
make_assist(&action, segments.as_slice(), edit);
if let Some(last_segment) = path.segment() {
last_segment.syntax().range().start(),
));
}
- })
+ });
+
+ ctx.build()
}
#[cfg(test)]
add_vis(ctx)
}
-fn add_vis(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+fn add_vis(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let item_keyword = ctx.leaf_at_offset().find(|leaf| match leaf.kind() {
FN_KW | MOD_KW | STRUCT_KW | ENUM_KW | TRAIT_KW => true,
_ => false,
(vis_offset(field.syntax()), ident.range())
};
- ctx.build("make pub(crate)", |edit| {
+ ctx.add_action("make pub(crate)", |edit| {
edit.target(target);
edit.insert(offset, "pub(crate) ");
edit.set_cursor(offset);
- })
+ });
+
+ ctx.build()
}
fn vis_offset(node: &SyntaxNode) -> TextUnit {
.unwrap_or(node.range().start())
}
-fn change_vis(ctx: AssistCtx<impl HirDatabase>, vis: &ast::Visibility) -> Option<Assist> {
+fn change_vis(
mut ctx: AssistCtx<impl HirDatabase>, vis: &ast::Visibility) -> Option<Assist> {
if vis.syntax().text() == "pub" {
- return ctx.build("change to pub(crate)", |edit| {
+ ctx.add_action("change to pub(crate)", |edit| {
edit.target(vis.syntax().range());
edit.replace(vis.syntax().range(), "pub(crate)");
- edit.set_cursor(vis.syntax().range().start());
+ edit.set_cursor(vis.syntax().range().start())
});
+
+ return ctx.build();
}
if vis.syntax().text() == "pub(crate)" {
- return ctx.build("change to pub", |edit| {
+ ctx.add_action("change to pub", |edit| {
edit.target(vis.syntax().range());
edit.replace(vis.syntax().range(), "pub");
edit.set_cursor(vis.syntax().range().start());
});
+
+ return ctx.build();
}
None
}
use crate::{AssistCtx, Assist};
-pub(crate) fn fill_match_arms(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn fill_match_arms(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let match_expr = ctx.node_at_offset::<ast::MatchExpr>()?;
// We already have some match arms, so we don't provide any assists.
let enum_name = enum_def.name(ctx.db)?;
let db = ctx.db;
- ctx.build("fill match arms", |edit| {
+ ctx.add_action("fill match arms", |edit| {
let mut buf = format!("match {} {{\n", expr.syntax().text().to_string());
let variants = enum_def.variants(db);
for variant in variants {
edit.target(match_expr.syntax().range());
edit.set_cursor(expr.syntax().range().start());
edit.replace_node_and_indent(match_expr.syntax(), buf);
- })
+ });
+
+ ctx.build()
}
#[cfg(test)]
use crate::{AssistCtx, Assist, non_trivia_sibling};
-pub(crate) fn flip_comma(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn flip_comma(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let comma = ctx.leaf_at_offset().find(|leaf| leaf.kind() == COMMA)?;
let prev = non_trivia_sibling(comma, Direction::Prev)?;
let next = non_trivia_sibling(comma, Direction::Next)?;
- ctx.build("flip comma", |edit| {
+ ctx.add_action("flip comma", |edit| {
edit.target(comma.range());
edit.replace(prev.range(), next.text());
edit.replace(next.range(), prev.text());
- })
+ });
+
+ ctx.build()
}
#[cfg(test)]
use crate::{AssistCtx, Assist};
-pub(crate) fn introduce_variable(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn introduce_variable(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let node = ctx.covering_node();
if !valid_covering_node(node) {
return None;
if indent.kind() != WHITESPACE {
return None;
}
- ctx.build("introduce variable", move |edit| {
+ ctx.add_action("introduce variable", move |edit| {
let mut buf = String::new();
let cursor_offset = if wrap_in_block {
}
}
edit.set_cursor(anchor_stmt.range().start() + cursor_offset);
- })
+ });
+
+ ctx.build()
}
fn valid_covering_node(node: &SyntaxNode) -> bool {
mod assist_ctx;
+use itertools::Itertools;
+
use ra_text_edit::TextEdit;
use ra_syntax::{TextRange, TextUnit, SyntaxNode, Direction};
use ra_db::FileRange;
pub(crate) use crate::assist_ctx::{AssistCtx, Assist};
-#[derive(Debug)]
+#[derive(Debug, Clone)]
pub struct AssistLabel {
/// Short description of the assist, as shown in the UI.
pub label: String,
}
+#[derive(Debug, Clone)]
pub struct AssistAction {
pub edit: TextEdit,
pub cursor_position: Option<TextUnit>,
.iter()
.filter_map(|f| f(ctx.clone()))
.map(|a| match a {
- Assist::Unresolved(label) => label,
+ Assist::Unresolved(labels) => labels,
Assist::Resolved(..) => unreachable!(),
})
- .collect()
+ .concat()
})
}
.iter()
.filter_map(|f| f(ctx.clone()))
.map(|a| match a {
- Assist::Resolved(label, action) => (label, action),
+ Assist::Resolved(labels_actions) => labels_actions,
Assist::Unresolved(..) => unreachable!(),
})
- .collect::<Vec<(AssistLabel, AssistAction)>>();
+ .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,
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);
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 action = match assist {
+ let labels_actions = match assist {
Assist::Unresolved(_) => unreachable!(),
- Assist::Resolved(_, it) => it,
+ 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
assert_eq_text!(after, &actual);
}
- pub(crate) fn check_assist_range(
+ 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 action = match assist {
+ let labels_actions = match assist {
Assist::Unresolved(_) => unreachable!(),
- Assist::Resolved(_, it) => it,
+ 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(
+ 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);
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 action = match assist {
+ let labels_actions = match assist {
Assist::Unresolved(_) => unreachable!(),
- Assist::Resolved(_, it) => it,
+ 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(
+ 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 action = match assist {
+ let labels_actions = match assist {
Assist::Unresolved(_) => unreachable!(),
- Assist::Resolved(_, it) => it,
+ 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);
}
};
use crate::{AssistCtx, Assist};
-pub(crate) fn remove_dbg(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn remove_dbg(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let macro_call = ctx.node_at_offset::<ast::MacroCall>()?;
if !is_valid_macrocall(macro_call, "dbg")? {
macro_args.text().slice(start..end).to_string()
};
- ctx.build("remove dbg!()", |edit| {
+ ctx.add_action("remove dbg!()", |edit| {
edit.target(macro_call.syntax().range());
edit.replace(macro_range, macro_content);
edit.set_cursor(cursor_pos);
- })
+ });
+
+ ctx.build()
}
/// Verifies that the given macro_call actually matches the given name
use crate::{AssistCtx, Assist};
-pub(crate) fn replace_if_let_with_match(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn replace_if_let_with_match(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let if_expr: &ast::IfExpr = ctx.node_at_offset()?;
let cond = if_expr.condition()?;
let pat = cond.pat()?;
ast::ElseBranchFlavor::IfExpr(_) => return None,
};
- ctx.build("replace with match", |edit| {
+ ctx.add_action("replace with match", |edit| {
let match_expr = build_match_expr(expr, pat, then_block, else_block);
edit.target(if_expr.syntax().range());
edit.replace_node_and_indent(if_expr.syntax(), match_expr);
edit.set_cursor(if_expr.syntax().range().start())
- })
+ });
+
+ ctx.build()
}
fn build_match_expr(
use crate::{AssistCtx, Assist};
-pub(crate) fn split_import(ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
+pub(crate) fn split_import(
mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> {
let colon_colon = ctx.leaf_at_offset().find(|leaf| leaf.kind() == COLONCOLON)?;
let path = colon_colon.parent().and_then(ast::Path::cast)?;
let top_path = generate(Some(path), |it| it.parent_path()).last()?;
None => top_path.syntax().range().end(),
};
- ctx.build("split import", |edit| {
+ ctx.add_action("split import", |edit| {
edit.target(colon_colon.range());
edit.insert(l_curly, "{");
edit.insert(r_curly, "}");
edit.set_cursor(l_curly + TextUnit::of_str("{"));
- })
+ });
+
+ ctx.build()
}
#[cfg(test)]
projects / rust.git / commitdiff