feat: format infix expressions (noir-lang#3001)

Sakapoi · Oct 19, 2023 · de73069 · de73069
1 parent aa11272
commit de73069
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/tooling/nargo_fmt/src/visitor/expr.rs b/tooling/nargo_fmt/src/visitor/expr.rs
@@ -25,6 +25,14 @@ impl FmtVisitor<'_> {
             ExpressionKind::Prefix(prefix) => {
                 format!("{}{}", prefix.operator, self.format_expr(prefix.rhs))
             }
+            ExpressionKind::Infix(infix) => {
+                format!(
+                    "{} {} {}",
+                    self.format_expr(infix.lhs),
+                    infix.operator.contents.as_string(),
+                    self.format_expr(infix.rhs)
+                )
+            }
             // TODO:
             _expr => slice!(self, span.start(), span.end()).to_string(),
         }

diff --git a/tooling/nargo_fmt/tests/input/infix.nr b/tooling/nargo_fmt/tests/input/infix.nr
@@ -1,5 +1,5 @@
 fn foo() {
     40 + 2;
-    !40 + 2;
-    40 + 2 == 42;
+    !40+2;
+    40 +  2 == 42;
 }