package code.ast;

public class WhileStatement extends Statement {

    private Statement expression;
    private Statement body;

    public Statement getExpression() {
        return expression;
    }

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

    public Statement getBody() {
        return body;
    }

    public void setBody(Statement body) {
        this.body = body;
    }

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