package code.ast;

public class IfStatement extends Statement {

    private Expression expression;
    private Statement thenStatement;
    private Statement elseStatement;

    public Expression getExpression() {
        return expression;
    }

    public void setExpression(Expression expression) {
        this.expression = expression;
    }

    public Statement getThenStatement() {
        return thenStatement;
    }

    public void setThenStatement(Statement thenStatement) {
        this.thenStatement = thenStatement;
    }

    public Statement getElseStatement() {
        return elseStatement;
    }

    public void setElseStatement(Statement elseStatement) {
        this.elseStatement = elseStatement;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append("if (").append(expression).append(") ");
        sb.append(thenStatement);
        if (elseStatement != null) {
            sb.append(" else ").append(elseStatement);
        }
        return sb.toString();
    }
}