mirror of
https://github.com/rust-lang/rust-analyzer.git
synced 2025-09-30 05:45:12 +00:00
**Unwrap Block** supports stand-alone blocks
This commit is contained in:
parent
10fa9c595a
commit
8c6f933773
1 changed files with 68 additions and 1 deletions
|
@ -3,7 +3,7 @@ use syntax::{
|
||||||
self,
|
self,
|
||||||
edit::{AstNodeEdit, IndentLevel},
|
edit::{AstNodeEdit, IndentLevel},
|
||||||
},
|
},
|
||||||
AstNode, TextRange, T,
|
AstNode, SyntaxKind, TextRange, T,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::{utils::unwrap_trivial_block, AssistContext, AssistId, AssistKind, Assists};
|
use crate::{utils::unwrap_trivial_block, AssistContext, AssistId, AssistKind, Assists};
|
||||||
|
@ -37,6 +37,15 @@ pub(crate) fn unwrap_block(acc: &mut Assists, ctx: &AssistContext) -> Option<()>
|
||||||
parent = parent.ancestors().find(|it| ast::MatchExpr::can_cast(it.kind()))?
|
parent = parent.ancestors().find(|it| ast::MatchExpr::can_cast(it.kind()))?
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if matches!(parent.kind(), SyntaxKind::BLOCK_EXPR | SyntaxKind::EXPR_STMT) {
|
||||||
|
return acc.add(assist_id, assist_label, target, |builder| {
|
||||||
|
builder.replace(
|
||||||
|
block.syntax().text_range(),
|
||||||
|
update_expr_string(block.to_string(), &[' ', '{', '\n']),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
let parent = ast::Expr::cast(parent)?;
|
let parent = ast::Expr::cast(parent)?;
|
||||||
|
|
||||||
match parent.clone() {
|
match parent.clone() {
|
||||||
|
@ -109,6 +118,64 @@ mod tests {
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn unwrap_tail_expr_block() {
|
||||||
|
check_assist(
|
||||||
|
unwrap_block,
|
||||||
|
r#"
|
||||||
|
fn main() {
|
||||||
|
<|>{
|
||||||
|
92
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
r#"
|
||||||
|
fn main() {
|
||||||
|
92
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn unwrap_stmt_expr_block() {
|
||||||
|
check_assist(
|
||||||
|
unwrap_block,
|
||||||
|
r#"
|
||||||
|
fn main() {
|
||||||
|
<|>{
|
||||||
|
92;
|
||||||
|
}
|
||||||
|
()
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
r#"
|
||||||
|
fn main() {
|
||||||
|
92;
|
||||||
|
()
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
);
|
||||||
|
// Pedantically, we should add an `;` here...
|
||||||
|
check_assist(
|
||||||
|
unwrap_block,
|
||||||
|
r#"
|
||||||
|
fn main() {
|
||||||
|
<|>{
|
||||||
|
92
|
||||||
|
}
|
||||||
|
()
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
r#"
|
||||||
|
fn main() {
|
||||||
|
92
|
||||||
|
()
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn simple_if() {
|
fn simple_if() {
|
||||||
check_assist(
|
check_assist(
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue