contracts: Visit nested expressions

Forgot to add this earlier.
This commit is contained in:
Mikolaj Wielgus 2023-09-20 00:31:35 +02:00
parent 0695b1deab
commit 894fa654cf
1 changed files with 5 additions and 3 deletions

View File

@ -383,10 +383,12 @@ struct ReturnReplacer;
impl syn::visit_mut::VisitMut for ReturnReplacer { impl syn::visit_mut::VisitMut for ReturnReplacer {
fn visit_expr_mut(&mut self, node: &mut Expr) { fn visit_expr_mut(&mut self, node: &mut Expr) {
if let Expr::Return(retexpr) = node { if let Expr::Return(ret_expr) = node {
let retexprexpr = retexpr.expr.clone(); let ret_expr_expr = ret_expr.expr.clone();
*node = syn::parse_quote!(break 'run #retexprexpr); *node = syn::parse_quote!(break 'run #ret_expr_expr);
} }
syn::visit_mut::visit_expr_mut(self, node);
} }
} }