Autoclose blocks when typing `{`

1 files changed, 46 insertions, 1 deletions

diff --git a/crates/ide/src/typing.rs b/crates/ide/src/typing.rs
index 11408d445..de65632e3 100644
--- a/crates/ide/src/typing.rs
+++ b/crates/ide/src/typing.rs
@@ -33,7 +33,8 @@ use crate::SourceChange;
 
 pub(crate) use on_enter::on_enter;
 
-pub(crate) const TRIGGER_CHARS: &str = ".=>";
+// Don't forget to add new trigger characters to `server_capabilities` in `caps.rs`.
+pub(crate) const TRIGGER_CHARS: &str = ".=>{";
 
 // Feature: On Typing Assists
 //
@@ -70,10 +71,47 @@ fn on_char_typed_inner(file: &SourceFile, offset: TextSize, char_typed: char) ->
         '.' => on_dot_typed(file, offset),
         '=' => on_eq_typed(file, offset),
         '>' => on_arrow_typed(file, offset),
+        '{' => on_opening_brace_typed(file, offset),
         _ => unreachable!(),
     }
 }
 
+/// Inserts a closing `}` when the user types an opening `{`, wrapping an existing expression in a
+/// block.
+fn on_opening_brace_typed(file: &SourceFile, offset: TextSize) -> Option<TextEdit> {
+    assert_eq!(file.syntax().text().char_at(offset), Some('{'));
+    let brace_token = file.syntax().token_at_offset(offset).right_biased()?;
+    let block = ast::BlockExpr::cast(brace_token.parent()?)?;
+
+    // We expect a block expression enclosing exactly 1 preexisting expression. It can be parsed as
+    // either the trailing expr or an ExprStmt.
+    let offset = {
+        match block.tail_expr() {
+            Some(expr) => {
+                if block.statements().next().is_some() {
+                    return None;
+                }
+                expr.syntax().text_range().end()
+            }
+            None => {
+                if block.statements().count() != 1 {
+                    return None;
+                }
+
+                match block.statements().next()? {
+                    ast::Stmt::ExprStmt(it) => {
+                        // Use the expression span to place `}` before the `;`
+                        it.expr()?.syntax().text_range().end()
+                    }
+                    _ => return None,
+                }
+            }
+        }
+    };
+
+    Some(TextEdit::insert(offset, "}".to_string()))
+}
+
 /// Returns an edit which should be applied after `=` was typed. Primarily,
 /// this works when adding `let =`.
 // FIXME: use a snippet completion instead of this hack here.
@@ -373,4 +411,11 @@ fn main() {
     fn adds_space_after_return_type() {
         type_char('>', "fn foo() -$0{ 92 }", "fn foo() -> { 92 }")
     }
+
+    #[test]
+    fn adds_closing_brace() {
+        type_char('{', r"fn f() { match () { _ => $0() } }", r"fn f() { match () { _ => {()} } }");
+        type_char('{', r"fn f() { $0(); }", r"fn f() { {()}; }");
+        type_char('{', r"fn f() { let x = $0(); }", r"fn f() { let x = {()}; }");
+    }
 }