feat: format infix expressions (#3001)

noir-lang · Oct 5, 2023 · 7926ada · 7926ada
1 parent a38b15f
commit 7926ada
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;
 }