diff --git a/tooling/nargo_fmt/src/visitor/expr.rs b/tooling/nargo_fmt/src/visitor/expr.rs index a0128042e1d..395f9f72476 100644 --- 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 index 13b53170da6..c01969f0502 100644 --- 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; }