org.springframework.ldap.query.DefaultContainerCriteria Maven / Gradle / Ivy

/*
 * Copyright 2005-2013 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.ldap.query;

import java.util.LinkedHashSet;
import java.util.Set;

import javax.naming.Name;

import org.springframework.ldap.filter.Filter;

/**
 * @author Mattias Hellborg Arthursson
 * @since 2.0
 */
class DefaultContainerCriteria implements AppendableContainerCriteria {

	private final Set filters = new LinkedHashSet();

	private final LdapQuery topQuery;

	private CriteriaContainerType type;

	DefaultContainerCriteria(LdapQuery topQuery) {
		this.topQuery = topQuery;
	}

	@Override
	public DefaultContainerCriteria append(Filter filter) {
		this.filters.add(filter);
		return this;
	}

	DefaultContainerCriteria withType(CriteriaContainerType newType) {
		this.type = newType;
		return this;
	}

	@Override
	public ConditionCriteria and(String attribute) {
		CriteriaContainerType.AND.validateSameType(this.type);
		this.type = CriteriaContainerType.AND;

		return new DefaultConditionCriteria(this, attribute);
	}

	@Override
	public ConditionCriteria or(String attribute) {
		CriteriaContainerType.OR.validateSameType(this.type);
		this.type = CriteriaContainerType.OR;

		return new DefaultConditionCriteria(this, attribute);
	}

	@Override
	public ContainerCriteria and(ContainerCriteria nested) {
		if (this.type == CriteriaContainerType.OR) {
			return new DefaultContainerCriteria(this.topQuery).withType(CriteriaContainerType.AND).append(this.filter())
					.append(nested.filter());
		}
		else {
			this.type = CriteriaContainerType.AND;
			this.filters.add(nested.filter());
			return this;
		}
	}

	@Override
	public ContainerCriteria or(ContainerCriteria nested) {
		if (this.type == CriteriaContainerType.AND) {
			return new DefaultContainerCriteria(this.topQuery).withType(CriteriaContainerType.OR).append(this.filter())
					.append(nested.filter());
		}
		else {
			this.type = CriteriaContainerType.OR;
			this.filters.add(nested.filter());
			return this;
		}
	}

	@Override
	public Filter filter() {
		if (this.filters.size() == 1) {
			// No need to wrap in And/OrFilter if there's just one condition.
			return this.filters.iterator().next();
		}

		return this.type.constructFilter().appendAll(this.filters);
	}

	@Override
	public Name base() {
		return this.topQuery.base();
	}

	@Override
	public SearchScope searchScope() {
		return this.topQuery.searchScope();
	}

	@Override
	public Integer timeLimit() {
		return this.topQuery.timeLimit();
	}

	@Override
	public Integer countLimit() {
		return this.topQuery.countLimit();
	}

	@Override
	public String[] attributes() {
		return this.topQuery.attributes();
	}

}