package com.aurea.maven.plugins.util;

 import java.util.HashSet;

 import java.util.Set;

 import org.apache.maven.AbstractMavenLifecycleParticipant;

 import org.apache.maven.artifact.Artifact;

 import org.apache.maven.artifact.handler.ArtifactHandler;

 import org.apache.maven.artifact.handler.manager.ArtifactHandlerManager;

 import org.apache.maven.execution.MavenSession;

 import org.apache.maven.project.MavenProject;

 import org.codehaus.plexus.component.annotations.Component;

 import org.codehaus.plexus.component.annotations.Requirement;

 import org.codehaus.plexus.logging.Logger;

 @Component(role = AbstractMavenLifecycleParticipant.class, hint = "ArtifactHandlerExt")

 public class ArtifactHandlerExtension extends AbstractMavenLifecycleParticipant {

         /**

          * The {@link ArtifactHandlerManager} into which any extension

          * {@link ArtifactHandler} instances should have been injected when the

          * extensions were loaded.

          * 

          * @component

          */

         private ArtifactHandlerManager artifactHandlerManager;

         @Requirement

         private Logger logger;

         // @Requirement

         // private RuntimeInformation runtime;

         public void afterProjectsRead(MavenSession session) {

                 logger.info(" __  __");

                 logger.info("|  \\/  |__ _Aurea-__ ___");

                 logger.info("| |\\/| / _` \\ V / -_) ' \\  ~ Artifact Loader Extension ~ 1.0");

                 logger.info("|_|  |_\\__,_|\\_/\\___|_||_|  ");

                 logger.info("");

                 MavenProject project = session.getProjects().get(0);

                 logger.info(

                                 "Setting ArtifactHandler on project-artifact: "

                                                 + project.getId() + "...");

                 Set<Artifact> artifacts = new HashSet<Artifact>();

                 artifacts.add(project.getArtifact());

                 Set<Artifact> dependencyArtifacts = project.getDependencyArtifacts();

                 if (dependencyArtifacts != null && !dependencyArtifacts.isEmpty()) {

                         artifacts.addAll(dependencyArtifacts);

                 }

                 for (Artifact artifact : artifacts) {

                         String type = artifact.getType();

                         ArtifactHandler handler = artifactHandlerManager

                                         .getArtifactHandler(type);

                         logger.debug("Artifact: " + artifact.getId() + "\nType: " + type

                                         + "\nArtifactHandler extension: " + handler.getExtension());

                         artifact.setArtifactHandler(handler);

                 }

                 logger.info("...done.");

         }

 }