From 894fa654cfa5e6cb31b7b1dd19ec8f82955ea805 Mon Sep 17 00:00:00 2001 From: Mikolaj Wielgus Date: Wed, 20 Sep 2023 00:31:35 +0200 Subject: [PATCH] contracts: Visit nested expressions Forgot to add this earlier. --- vendor/contracts/src/implementation/codegen.rs | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/vendor/contracts/src/implementation/codegen.rs b/vendor/contracts/src/implementation/codegen.rs index fc23feb..3d82bf2 100644 --- a/vendor/contracts/src/implementation/codegen.rs +++ b/vendor/contracts/src/implementation/codegen.rs @@ -383,10 +383,12 @@ struct ReturnReplacer; impl syn::visit_mut::VisitMut for ReturnReplacer { fn visit_expr_mut(&mut self, node: &mut Expr) { - if let Expr::Return(retexpr) = node { - let retexprexpr = retexpr.expr.clone(); - *node = syn::parse_quote!(break 'run #retexprexpr); + if let Expr::Return(ret_expr) = node { + let ret_expr_expr = ret_expr.expr.clone(); + *node = syn::parse_quote!(break 'run #ret_expr_expr); } + + syn::visit_mut::visit_expr_mut(self, node); } }