ebean/src/main/java/com/avaje/ebeaninternal/server/deploy/BeanSqlSelect.java

package com.avaje.ebeaninternal.server.deploy;

import javax.persistence.PersistenceException;

import com.avaje.ebeaninternal.server.lib.util.StringHelper;

public class BeanSqlSelect {

	public static final String HAVING_PREDICATES = "${HAVING_PREDICATES}";

	public static final String WHERE_PREDICATES = "${WHERE_PREDICATES}";

	public static final String AND_PREDICATES = "${AND_PREDICATES}";

	public static final String ORDER_BY = "${ORDER_BY}";

	public enum PredicatesType {
		HAVING, WHERE, AND, NONE
	}

	final String sql;

	final PredicatesType predicatesType;

	final boolean hasOrderBy;

	public BeanSqlSelect(String sql, PredicatesType predicatesType, boolean hasOrderBy) {
		this.sql = sql;
		this.predicatesType = predicatesType;
		this.hasOrderBy = hasOrderBy;
	}

	public String getSql() {
		return sql;
	}

	public PredicatesType getPredicatesType() {
		return predicatesType;
	}

	public boolean hasOrderBy() {
		return hasOrderBy;
	}

	public String addPredicates(String query, String predicates) {
		if (predicates == null) {
			switch (predicatesType) {
			case HAVING:
				return StringHelper.replaceString(query, HAVING_PREDICATES, "");
			case WHERE:
				return StringHelper.replaceString(query, WHERE_PREDICATES, "");
			case AND:
				return StringHelper.replaceString(query, AND_PREDICATES, "");
			case NONE:
				return query;

			default:
				throw new PersistenceException("predicatesType " + predicatesType + " not handled");
			}
		} else {
			switch (predicatesType) {
			case HAVING:
				return StringHelper
						.replaceString(query, HAVING_PREDICATES, " HAVING " + predicates);
			case WHERE:
				return StringHelper.replaceString(query, WHERE_PREDICATES, " WHERE " + predicates);
			case AND:
				return StringHelper.replaceString(query, AND_PREDICATES, " AND " + predicates);
			case NONE:
				return query;

			default:
				throw new PersistenceException("predicatesType " + predicatesType + " not handled");
			}
		}
	}

	public String addOrderBy(String query, String orderBy) {
		if (!hasOrderBy) {
			return query;
		}
		if (orderBy == null) {
			return StringHelper.replaceString(query, ORDER_BY, "");
		} else {
			return StringHelper.replaceString(query, ORDER_BY, " ORDER BY " + orderBy);
		}
	}
}