/**
 * Copyright 2015-2016 Red Hat, Inc, and individual contributors.
 *
 * 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.wildfly.swarm.plugin.gradle;

import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import org.gradle.api.Project;
import org.gradle.api.artifacts.Configuration;
import org.gradle.api.artifacts.DependencySet;
import org.gradle.api.artifacts.ResolvedArtifact;
import org.gradle.api.internal.artifacts.dependencies.DefaultDependencyArtifact;
import org.gradle.api.internal.artifacts.dependencies.DefaultExternalModuleDependency;
import org.gradle.api.internal.artifacts.dependencies.DefaultProjectDependency;
import org.gradle.api.internal.project.DefaultProjectAccessListener;
import org.gradle.api.internal.project.ProjectInternal;
import org.wildfly.swarm.tools.ArtifactResolvingHelper;
import org.wildfly.swarm.tools.ArtifactSpec;

/**
 * @author Bob McWhirter
 */
public class GradleArtifactResolvingHelper implements ArtifactResolvingHelper {


    private final Project project;

    Map<String, Project> projects;

    public GradleArtifactResolvingHelper(Project project) {
        this.project = project;
        this.projects = project.getRootProject().getAllprojects().stream().collect(Collectors.toMap(p -> p.getGroup() + ":" + p.getName() + ":" + p.getVersion(), p -> p));
        this.project.getRepositories().maven(repo -> {
            repo.setName("jboss-public");
            repo.setUrl("http://repository.jboss.org/nexus/content/groups/public/");
        });
    }

    @Override
    public ArtifactSpec resolve(final ArtifactSpec spec) {
        if (spec.file != null) {
            return spec;
        }

        final Iterator<ResolvedArtifact> iterator =
                doResolve(new HashSet<>(Collections.singletonList(spec)), false).iterator();
        if (iterator.hasNext()) {
            spec.file = iterator.next().getFile();

            return spec;
        }

        return null;
    }

    @Override
    public Set<ArtifactSpec> resolveAll(final Set<ArtifactSpec> specs, boolean transitive, boolean defaultExcludes) throws Exception {
        if (specs.isEmpty()) {
            return specs;
        }

        return doResolve(specs, transitive)
            .stream()
            .map(artifact -> new ArtifactSpec("default",
                                artifact.getModuleVersion().getId().getGroup(),
                                artifact.getModuleVersion().getId().getName(),
                                artifact.getModuleVersion().getId().getVersion(),
                                artifact.getExtension(),
                                artifact.getClassifier(),
                                artifact.getFile()))
            .collect(Collectors.toSet());
    }

    private Collection<ResolvedArtifact> doResolve(final Collection<ArtifactSpec> deps, boolean transitive) {
        final Configuration config = this.project.getConfigurations().detachedConfiguration();
        final DependencySet dependencySet = config.getDependencies();

        config.getResolutionStrategy().setForcedModules(
                this.project.getConfigurations().getByName("compile").getResolutionStrategy().getForcedModules());

        deps.stream()
                .forEach(spec -> {
                    if (projects.containsKey(spec.groupId() + ":" + spec.artifactId() + ":" + spec.version())) {
                        dependencySet.add(new DefaultProjectDependency((ProjectInternal) projects.get(spec.groupId() + ":" + spec.artifactId() + ":" + spec.version()), new DefaultProjectAccessListener(), false));
                    } else {
                        final DefaultExternalModuleDependency d =
                                new DefaultExternalModuleDependency(spec.groupId(), spec.artifactId(), spec.version());
                        final DefaultDependencyArtifact da =
                                new DefaultDependencyArtifact(spec.artifactId(), spec.type(), spec.type(), spec.classifier(), null);
                        d.addArtifact(da);
                        dependencySet.add(d);
                    }
                });

        if (transitive) {
            return config
                .getResolvedConfiguration()
                .getResolvedArtifacts();
        }

        return config
            .getResolvedConfiguration()
            .getFirstLevelModuleDependencies()
            .stream()
            .map(dep -> dep.getModuleArtifacts())
            .flatMap(artifacts -> artifacts.stream())
            .collect(Collectors.toList());
    }

}

@@ -3665,24 +3665,255 @@
 dencies();%0A%0A
+        if ( transitive ) %7B%0A            // if transitive, then dependency-manage everything,%0A            // if non-transitive, we want exactly what we've asked for, and don't force%0A            // to the project's dependencies.%0A    
         conf
@@ -3957,16 +3957,20 @@
 odules(%0A
+    
         
@@ -4076,25 +4076,24 @@
 les());%0A
-%0A
         
 deps.str
@@ -4088,21 +4088,10 @@
     
-deps.stream()
+%7D%0A
 %0A   
@@ -4095,24 +4095,20 @@
         
-        
+deps
 .forEach
@@ -4130,24 +4130,16 @@
         
- 
-       
 if (proj
@@ -4219,32 +4219,24 @@
 rsion())) %7B%0A
-        
             
@@ -4440,24 +4440,16 @@
         
-       
- 
 %7D else %7B
@@ -4449,24 +4449,16 @@
  else %7B%0A
-        
         
@@ -4507,24 +4507,16 @@
 ncy d =%0A
-        
         
@@ -4631,32 +4631,24 @@
             
-     
-   
 final Defaul
@@ -4672,24 +4672,16 @@
 ct da =%0A
-        
         
@@ -4813,24 +4813,16 @@
         
-        
 d.addArt
@@ -4844,32 +4844,24 @@
             
-   
-     
 dependencySe
@@ -4886,26 +4886,10 @@
     
-        %7D%0A        
+%7D%0A
     

Break process, but trying to fix Gradle again.