From 40950ddb27ff5fcfdf64dc9a4fa9fc98fd98d267 Mon Sep 17 00:00:00 2001
From: David Tolnay <dtolnay@gmail.com>
Date: Sat, 11 May 2024 11:59:24 -0700
Subject: [PATCH] Remove recursion from printing of Expr::If

---
 src/expr.rs | 35 ++++++++++++++++++++++++++---------
 1 file changed, 26 insertions(+), 9 deletions(-)

diff --git a/src/expr.rs b/src/expr.rs
index 873cfca954..265316a466 100644
--- a/src/expr.rs
+++ b/src/expr.rs
@@ -3334,16 +3334,33 @@ pub(crate) mod printing {
     impl ToTokens for ExprIf {
         fn to_tokens(&self, tokens: &mut TokenStream) {
             outer_attrs_to_tokens(&self.attrs, tokens);
-            self.if_token.to_tokens(tokens);
-            wrap_bare_struct(tokens, &self.cond);
-            self.then_branch.to_tokens(tokens);
-            if let Some((else_token, else_)) = &self.else_branch {
+
+            let mut expr = self;
+            loop {
+                expr.if_token.to_tokens(tokens);
+                wrap_bare_struct(tokens, &expr.cond);
+                expr.then_branch.to_tokens(tokens);
+
+                let (else_token, else_) = match &expr.else_branch {
+                    Some(else_branch) => else_branch,
+                    None => break,
+                };
+
                 else_token.to_tokens(tokens);
-                // If we are not one of the valid expressions to exist in an else
-                // clause, wrap ourselves in a block.
-                match **else_ {
-                    Expr::If(_) | Expr::Block(_) => else_.to_tokens(tokens),
-                    _ => token::Brace::default().surround(tokens, |tokens| else_.to_tokens(tokens)),
+                match &**else_ {
+                    Expr::If(next) => {
+                        expr = next;
+                    }
+                    Expr::Block(last) => {
+                        last.to_tokens(tokens);
+                        break;
+                    }
+                    // If this is not one of the valid expressions to exist in
+                    // an else clause, wrap it in a block.
+                    other => {
+                        token::Brace::default().surround(tokens, |tokens| other.to_tokens(tokens));
+                        break;
+                    }
                 }
             }
         }