/*******************************************************************************
* Copyright Technophobia Ltd 2012
*
* This file is part of the Substeps Eclipse Plugin.
*
* The Substeps Eclipse Plugin is free software: you can redistribute it and/or modify
* it under the terms of the Eclipse Public License v1.0.
*
* The Substeps Eclipse Plugin is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* Eclipse Public License for more details.
*
* You should have received a copy of the Eclipse Public License
* along with the Substeps Eclipse Plugin. If not, see <http://www.eclipse.org/legal/epl-v10.html>.
******************************************************************************/
package com.technophobia.substeps.step;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IWorkspace;
import com.technophobia.substeps.supplier.Transformer;
public class ProvidedSuggestionManager implements ContextualSuggestionManager, ProjectStepImplementationProvider {
private final Transformer<IResource, IProject> resourceToProjectTransformer;
private final Map<SuggestionSource, Set<ProjectSuggestionProvider>> projectSuggestionProviders;
public ProvidedSuggestionManager(final Transformer<IResource, IProject> resourceToProjectTransformer) {
this.resourceToProjectTransformer = resourceToProjectTransformer;
this.projectSuggestionProviders = new HashMap<SuggestionSource, Set<ProjectSuggestionProvider>>();
}
public void addProvider(final SuggestionSource source, final ProjectSuggestionProvider provider) {
if (!projectSuggestionProviders.containsKey(source)) {
projectSuggestionProviders.put(source, new HashSet<ProjectSuggestionProvider>());
}
projectSuggestionProviders.get(source).add(provider);
}
public Collection<ProjectSuggestionProvider> providersOfSource(final SuggestionSource source) {
return projectSuggestionProviders.get(source);
}
public void load(final IWorkspace workspace) {
for (final Set<ProjectSuggestionProvider> providers : projectSuggestionProviders.values()) {
for (final ProjectSuggestionProvider provider : providers) {
provider.load(workspace);
}
}
}
@Override
public List<Suggestion> suggestionsFor(final IResource resource) {
final Set<Suggestion> suggestions = new HashSet<Suggestion>();
final IProject project = resourceToProjectTransformer.from(resource);
for (final Entry<SuggestionSource, Set<ProjectSuggestionProvider>> entry : projectSuggestionProviders
.entrySet()) {
for (final ProjectSuggestionProvider provider : entry.getValue()) {
suggestions.addAll(provider.suggestionsFor(project));
}
}
final List<Suggestion> suggestionsList = new ArrayList<Suggestion>(suggestions);
Collections.sort(suggestionsList);
return suggestionsList;
}
@Override
public Collection<String> stepImplementationClasses(final IProject project) {
final Collection<String> classes = new HashSet<String>();
for (final Set<ProjectSuggestionProvider> providers : projectSuggestionProviders.values()) {
for (final ProjectSuggestionProvider provider : providers) {
if (provider instanceof ProjectStepImplementationProvider) {
classes.addAll(((ProjectStepImplementationProvider) provider).stepImplementationClasses(project));
}
}
}
return classes;
}
}