add rule to postprocessor

danilopedraza · Apr 12, 2024 · fc8bf5b · fc8bf5b
1 parent d458e8f
commit fc8bf5b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/semantic.rs b/src/semantic.rs
@@ -15,6 +15,7 @@ pub fn postprocess(node: ASTNode) -> ASTNode {
     match node {
         ASTNode::Infix(InfixOperator::Correspondence, params, proc) => function(*params, *proc),
         ASTNode::Infix(InfixOperator::Call, called, args) => call(*called, *args),
+        ASTNode::Infix(op, lhs, rhs) => infix(op, lhs, rhs),
         ASTNode::For(ident, iter, proc) => {
             ASTNode::For(ident, postprocessed_box(iter), postprocessed_vec(proc))
         }
@@ -39,6 +40,10 @@ pub fn postprocess(node: ASTNode) -> ASTNode {
     }
 }
 
+fn infix(op: InfixOperator, lhs: Box<ASTNode>, rhs: Box<ASTNode>) -> ASTNode {
+    ASTNode::Infix(op, postprocessed_box(lhs), postprocessed_box(rhs))
+}
+
 fn call(called_node: ASTNode, args_node: ASTNode) -> ASTNode {
     let called = postprocess(called_node);
     let args = match postprocess(args_node) {