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.");
  	}
  
  }