Add braces around let-else's else expression if needed

dtolnay · May 15, 2024 · c644808 · c644808
1 parent 9597ebc
commit c644808
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/stmt.rs b/src/stmt.rs
@@ -412,7 +412,7 @@ pub(crate) mod parsing {
 #[cfg(feature = "printing")]
 mod printing {
     use crate::classify;
-    use crate::expr;
+    use crate::expr::{self, Expr};
     use crate::stmt::{Block, Local, Stmt, StmtMacro};
     use crate::token;
     use proc_macro2::TokenStream;
@@ -457,7 +457,11 @@ mod printing {
                 }
                 if let Some((else_token, diverge)) = &init.diverge {
                     else_token.to_tokens(tokens);
-                    diverge.to_tokens(tokens);
+                    match &**diverge {
+                        Expr::Block(diverge) => diverge.to_tokens(tokens),
+                        _ => token::Brace::default()
+                            .surround(tokens, |tokens| diverge.to_tokens(tokens)),
+                    }
                 }
             }
             self.semi_token.to_tokens(tokens);