Reuse print_subexpression outside of expr.rs

dtolnay · Jan 4, 2025 · 5c935d2 · 5c935d2
1 parent 505ab00
commit 5c935d2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/expr.rs b/src/expr.rs
@@ -3161,7 +3161,7 @@ pub(crate) mod printing {
         );
     }
 
-    fn print_subexpression(
+    pub(crate) fn print_subexpression(
         expr: &Expr,
         needs_group: bool,
         tokens: &mut TokenStream,

diff --git a/src/stmt.rs b/src/stmt.rs
@@ -453,11 +453,12 @@ pub(crate) mod printing {
             self.pat.to_tokens(tokens);
             if let Some(init) = &self.init {
                 init.eq_token.to_tokens(tokens);
-                if init.diverge.is_some() && classify::expr_trailing_brace(&init.expr) {
-                    token::Paren::default().surround(tokens, |tokens| init.expr.to_tokens(tokens));
-                } else {
-                    init.expr.to_tokens(tokens);
-                }
+                expr::printing::print_subexpression(
+                    &init.expr,
+                    init.diverge.is_some() && classify::expr_trailing_brace(&init.expr),
+                    tokens,
+                    FixupContext::NONE,
+                );
                 if let Some((else_token, diverge)) = &init.diverge {
                     else_token.to_tokens(tokens);
                     match &**diverge {