feat: add support for functions

0xPolygonMiden · Aug 8, 2023 · 7806077 · 7806077
1 parent 80c1a88
commit 7806077
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/parser/src/transforms/inlining.rs b/parser/src/transforms/inlining.rs
@@ -307,17 +307,13 @@ impl<'a> Inlining<'a> {
                 Ok(vec![Statement::Expr(expr)])
             }
 
-            Statement::Return(mut expr) => {
-                match expr {
-                    Expr::Call(call) => {
-                        self.expand_call(call)
-                    }
-                    _ => {
-                        self.rewrite_expr(&mut expr)?;
-                        Ok(vec![Statement::Return(expr)])
-                    }
-                }   
-            }
+            Statement::Return(mut expr) => match expr {
+                Expr::Call(call) => self.expand_call(call),
+                _ => {
+                    self.rewrite_expr(&mut expr)?;
+                    Ok(vec![Statement::Return(expr)])
+                }
+            },
         }
     }