/*
 * Copyright 2010 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
 *
 *      http://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.gradle.api.internal.artifacts.ivyservice;

import org.gradle.api.GradleException;
import org.gradle.api.artifacts.*;
import org.gradle.api.internal.artifacts.CachingDependencyResolveContext;
import org.gradle.api.internal.artifacts.ArtifactDependencyResolver;
import org.gradle.api.internal.artifacts.configurations.ConfigurationInternal;
import org.gradle.api.specs.Spec;
import org.gradle.util.CollectionUtils;

import java.io.File;
import java.util.LinkedHashSet;
import java.util.Set;

public class SelfResolvingDependencyResolver implements ArtifactDependencyResolver {
    private final ArtifactDependencyResolver resolver;

    public SelfResolvingDependencyResolver(ArtifactDependencyResolver resolver) {
        this.resolver = resolver;
    }

    public ArtifactDependencyResolver getResolver() {
        return resolver;
    }

    public ResolvedConfiguration resolve(final ConfigurationInternal configuration) {
        final ResolvedConfiguration resolvedConfiguration = resolver.resolve(configuration);
        final Set<Dependency> dependencies = configuration.getAllDependencies();

        return new ResolvedConfiguration() {
            private final CachingDependencyResolveContext resolveContext = new CachingDependencyResolveContext(configuration.isTransitive());

            public Set<File> getFiles(Spec<? super Dependency> dependencySpec) {
                Set<File> files = new LinkedHashSet<File>();

                Set<Dependency> selectedDependencies = CollectionUtils.filter(dependencies, dependencySpec);
                for (Dependency dependency : selectedDependencies) {
                    resolveContext.add(dependency);
                }
                files.addAll(resolveContext.resolve().getFiles());
                files.addAll(resolvedConfiguration.getFiles(dependencySpec));
                return files;
            }

            public Set<ResolvedArtifact> getResolvedArtifacts() {
                return resolvedConfiguration.getResolvedArtifacts();
            }

            public Set<ResolvedDependency> getFirstLevelModuleDependencies() {
                return resolvedConfiguration.getFirstLevelModuleDependencies();
            }

            public Set<ResolvedDependency> getFirstLevelModuleDependencies(Spec<? super Dependency> dependencySpec) throws ResolveException {
                return resolvedConfiguration.getFirstLevelModuleDependencies(dependencySpec);
            }

            public boolean hasError() {
                return resolvedConfiguration.hasError();
            }

            public LenientConfiguration getLenientConfiguration() {
                return resolvedConfiguration.getLenientConfiguration();
            }

            public void rethrowFailure() throws GradleException {
                resolvedConfiguration.rethrowFailure();
            }
        };
    }
}
