diff --git a/src/chisel.jl b/src/chisel.jl index c41654b..07b3d68 100644 --- a/src/chisel.jl +++ b/src/chisel.jl @@ -23,6 +23,10 @@ end function is_assignment(node::JuliaSyntax.GreenNode) return JuliaSyntax.is_prec_assignment(node) end +function is_infix_op_call(node::JuliaSyntax.GreenNode) + return JuliaSyntax.kind(node) === K"call" && + JuliaSyntax.is_infix_op_call(node) +end function is_comparison_leaf(node::JuliaSyntax.GreenNode) return is_leaf(node) && JuliaSyntax.is_prec_comparison(node) end diff --git a/src/runestone.jl b/src/runestone.jl index d10952c..9bbd34d 100644 --- a/src/runestone.jl +++ b/src/runestone.jl @@ -269,7 +269,7 @@ end # This pass handles spaces around infix operator calls and comparison chains function spaces_around_operators(ctx::Context, node::JuliaSyntax.GreenNode) if !( - JuliaSyntax.is_infix_op_call(node) || + is_infix_op_call(node) || (JuliaSyntax.kind(node) === K"comparison" && !JuliaSyntax.is_trivia(node)) ) return nothing