mirror of https://codeberg.org/topola/topola.git
contracts: Visit nested expressions
Forgot to add this earlier.
This commit is contained in:
parent
0695b1deab
commit
894fa654cf
|
|
@ -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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue