package gr.uoa.di.webui.search;

import java.util.List;

public class QueryCollectionEnhancer implements QueryEnhancer{

	private List<String> collectionIds = null;

	public void setCollectionIds(List<String> collectionIds) {
		this.collectionIds = collectionIds;
	}


	public List<String> getCollectionIds() {
		return collectionIds;
	}
	
	@Override
	public String enhanceQuery(String query) {
		
		if(!collectionIds.isEmpty()){
			query += " AND (";
		}
		
		int i = 0;
		for(String collectionId : collectionIds){
			query += " (collection = \"" + collectionId +"\")";
			
			i++;
			
			if (i < collectionIds.size()){
				query += " OR";
			} else {
				query += " )";
			}
			
		}
		
		return query;
	}



}