org.hisrc.jscm.codemodel.statement.impl.IfStatementImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of js-codemodel Show documentation

The newest version!

package org.hisrc.jscm.codemodel.statement.impl;

import org.hisrc.jscm.codemodel.JSCodeModel;
import org.hisrc.jscm.codemodel.expression.JSExpression;
import org.hisrc.jscm.codemodel.lang.Validate;
import org.hisrc.jscm.codemodel.statement.JSIfStatement;
import org.hisrc.jscm.codemodel.statement.JSStatement;
import org.hisrc.jscm.codemodel.statement.JSStatementGenerator;
import org.hisrc.jscm.codemodel.statement.JSStatementVisitor;

public class IfStatementImpl extends StatementImpl implements JSIfStatement {

	private final JSExpression _if;

	private JSStatement _then;
	private JSStatement _else;

	private final JSStatementGenerator thenGenerator;
	private final JSStatementGenerator elseGenerator;

	public IfStatementImpl(JSCodeModel codeModel, JSExpression expression) {
		super(codeModel);
		Validate.notNull(expression);
		this._if = expression;
		this._then = new EmptyStatementImpl(getCodeModel());
		this._else = null;
		this.thenGenerator = new StatementGeneratorImpl(codeModel) {

			protected  S add(S statement) {
				Validate.notNull(statement);
				IfStatementImpl.this._then = statement;
				return statement;
			}
		};
		this.elseGenerator = new StatementGeneratorImpl(codeModel) {

			protected  S add(S statement) {
				Validate.notNull(statement);
				IfStatementImpl.this._else = statement;
				return statement;
			}
		};
	}

	public JSExpression getIf() {
		return _if;
	}

	public JSStatement getThen() {
		return _then;
	}

	@Override
	public JSStatement getElse() {
		return _else;
	}

	@Override
	public JSStatementGenerator _then() {
		return thenGenerator;
	}

	@Override
	public JSStatementGenerator _else() {
		return elseGenerator;
	}

	@Override
	public  V acceptStatementVisitor(
			JSStatementVisitor visitor) throws E {
		return visitor.visitIf(this);
	}

}