Fix infix_op_call predicat

fredrikekre · May 26, 2024 · a0ba874 · a0ba874
1 parent 0db3c54
commit a0ba874
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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
@@ -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