plugins/org.eclipse.wst.common.core/src-search/org/eclipse/wst/common/core/search/internal/SearchParticipantRegistry.java

/*******************************************************************************
 * Copyright (c) 2004, 2006 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - Initial API and implementation
 *******************************************************************************/

package org.eclipse.wst.common.core.search.internal;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.eclipse.core.expressions.EvaluationContext;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.wst.common.core.search.SearchParticipant;
import org.eclipse.wst.common.core.search.pattern.SearchPattern;

public class SearchParticipantRegistry
{

	protected Map idMap = new HashMap(); // maps searchParticipant id to a
											// searchParticipant descriptor

	protected Map searchParticipantsMap = new HashMap(); // maps content type
															// a set of
															// searchParticipants
															// ids

	protected Map contentTypesMap = new HashMap(); // maps searchParticipant id
													// a set of content types

	public SearchParticipantRegistry()
	{
	}

	public void putSearchParticipant(String id,
			SearchParticipantDescriptor searchParticipantDescriptor)
	{
		idMap.put(id, searchParticipantDescriptor);
	}

	public String[] getSupportedContentTypes()
	{
		Set contentIds = searchParticipantsMap.keySet();
		return (String[]) contentIds.toArray(new String[contentIds.size()]);
	}

	public Set getSupportedContentTypes(String searchParticipantId)
	{
		Object contentTypes = contentTypesMap.get(searchParticipantId);
		if (!(contentTypes instanceof Set))
		{
			contentTypes = new HashSet();
		}

		return (Set) contentTypes;
	}

	public String[] getSearchParticipantIds()
	{
		Set ids = idMap.keySet();
		return (String[]) ids.toArray(new String[ids.size()]);
	}

	public Collection getSearchParticipants()
	{
		return idMap.values();
	}

	public void putSearchParticipantForContentType(String contentType,
			String searchParticipantId)
	{

		Object searchParticipants = searchParticipantsMap.get(contentType);
		if (!(searchParticipants instanceof Set))
		{
			searchParticipants = new HashSet();
		}
		((Set) searchParticipants).add(searchParticipantId);
		searchParticipantsMap.put(contentType, searchParticipants);

		// cross reference in the contentTypesMap
		Object contentTypes = contentTypesMap.get(searchParticipantId);
		if (!(contentTypes instanceof Set))
		{
			contentTypes = new HashSet();
		}
		((Set) contentTypes).add(contentType);
		contentTypesMap.put(searchParticipantId, contentTypes);

	}

	public SearchParticipant getSearchParticipant(String id)
	{
		SearchParticipantDescriptor searchParticipantDescriptor = null;
		if (id != null)
		{
			searchParticipantDescriptor = (SearchParticipantDescriptor) idMap
					.get(id);
		}
		return searchParticipantDescriptor != null ? searchParticipantDescriptor
				.getSearchParticipant()
				: null;

	}

	public String[] getSearchParticipantIds(String contentTypeId)
	{
		Object searchParticipants = searchParticipantsMap.get(contentTypeId);
		if (searchParticipants instanceof Set)
		{
			Set set = (Set) searchParticipants;
			return (String[]) (set.toArray(new String[set.size()]));
		}
		return new String[0];
	}

	public SearchParticipant[] getParticipants(SearchPattern pattern)
	{

		EvaluationContext evalContext = createEvaluationContext(pattern);
		List result = new ArrayList();
		for (Iterator iter = getSearchParticipants().iterator(); iter.hasNext();)
		{
			SearchParticipantDescriptor descriptor = (SearchParticipantDescriptor) iter
					.next();
			try
			{
				if (descriptor.matches(evalContext))
				{
					try
					{
						SearchParticipant participant = descriptor
								.getSearchParticipant();
						if (!SearchParticipant.class.isInstance(participant))
							throw new ClassCastException();
						if (participant.initialize(pattern, descriptor
								.getSupportedContentTypes()))
						{
							result.add(participant);
						}
					} catch (ClassCastException e)
					{
						iter.remove();
					}
				}

			} catch (CoreException e)
			{
				iter.remove();
			}

		}

		return (SearchParticipant[]) result
				.toArray(new SearchParticipant[result.size()]);
	}

	private static EvaluationContext createEvaluationContext(
			SearchPattern pattern)
	{
		EvaluationContext result = new EvaluationContext(null, pattern);
		result.addVariable("pattern", pattern); //$NON-NLS-1$
		return result;
	}

}