java-commons/muscript/src/main/java/io/gitlab/jfronny/muscript/ast/compare/Equal.java

package io.gitlab.jfronny.muscript.ast.compare;

import io.gitlab.jfronny.muscript.data.dynamic.Dynamic;
import io.gitlab.jfronny.muscript.ast.BoolExpr;
import io.gitlab.jfronny.muscript.ast.Expr;

import java.util.Objects;

public class Equal extends BoolExpr {
    private final Expr<?> left;
    private final Expr<?> right;

    public Equal(int chStart, int chEnd, Expr<?> left, Expr<?> right) {
        super(chStart, chEnd);
        this.left = left;
        this.right = right;
    }

    @Override
    public Boolean get(Dynamic<?> dataRoot) {
        return Objects.equals(unwrap(left.get(dataRoot)), unwrap(right.get(dataRoot)));
    }

    private Object unwrap(Object o) {
        if (o instanceof Dynamic<?> a) return unwrap(a.getValue());
        return o;
    }

    @Override
    public BoolExpr optimize() {
        Expr<?> left = this.left.optimize();
        Expr<?> right = this.right.optimize();
        if (left.equals(right)) return Expr.literal(chStart, chEnd, true);
        return new Equal(chStart, chEnd, left, right);
    }

    @Override
    public boolean equals(Object obj) {
        return obj instanceof Equal equal && left.equals(equal.left) && right.equals(equal.right);
    }
}