package org.apache.maven.plugin.ear.util;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.
   */
  
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.apache.maven.plugin.ear.EarModuleFactory;
  import org.apache.maven.plugin.ear.EarPluginException;
  import org.apache.maven.plugin.ear.UnknownArtifactTypeException;
  import org.codehaus.plexus.configuration.PlexusConfiguration;
  import org.codehaus.plexus.configuration.PlexusConfigurationException;
  
  /**
   * Allows to map custom artifact type to standard type.
   *
   * @author <a href="snicoll@apache.org">Stephane Nicoll</a>
   * @version $Id: ArtifactTypeMappingService.java 1228846 2012-01-08 13:53:34Z rfscholte $
   */
  public class ArtifactTypeMappingService
  {
      static final String ARTIFACT_TYPE_MAPPING_ELEMENT = "artifactTypeMapping";
  
      static final String TYPE_ATTRIBUTE = "type";
  
      static final String MAPPING_ATTRIBUTE = "mapping";
  
      // A standard type to a list of customType
      private final Map<String, List<String>> typeMappings;
  
      // The user-defined mapping for direct access
      private final Map<String, String> customMappings;
  
      public ArtifactTypeMappingService()
      {
          this.typeMappings = new HashMap<String, List<String>>();
          this.customMappings = new HashMap<String, String>();
          init();
      }
  
      public void configure( final PlexusConfiguration plexusConfiguration )
          throws EarPluginException, PlexusConfigurationException
      {
  
          // No user defined configuration
          if ( plexusConfiguration == null )
          {
              return;
          }
  
          // Inject users configuration
          final PlexusConfiguration[] artifactTypeMappings =
              plexusConfiguration.getChildren( ARTIFACT_TYPE_MAPPING_ELEMENT );
          for ( int i = 0; i < artifactTypeMappings.length; i++ )
          {
              PlexusConfiguration artifactTypeMapping = artifactTypeMappings[i];
              final String customType = artifactTypeMapping.getAttribute( TYPE_ATTRIBUTE );
              final String mapping = artifactTypeMapping.getAttribute( MAPPING_ATTRIBUTE );
  
              if ( customType == null )
              {
                  throw new EarPluginException( "Invalid artifact type mapping, type attribute should be set." );
              }
              else if ( mapping == null )
              {
                  throw new EarPluginException( "Invalid artifact type mapping, mapping attribute should be set." );
              }
              else if ( !EarModuleFactory.isStandardArtifactType( mapping ) )
              {
                  throw new EarPluginException(
                      "Invalid artifact type mapping, mapping[" + mapping + "] must be a standard Ear artifact type["
                          + EarModuleFactory.getStandardArtifactTypes() + "]" );
              }
              else if ( customMappings.containsKey( customType ) )
              {
                  throw new EarPluginException(
                      "Invalid artifact type mapping, type[" + customType + "] is already registered." );
              }
              else
              {
                 // Add the custom mapping
                 customMappings.put( customType, mapping );
 
                 // Register the custom mapping to its standard type
                 List<String> typeMapping = typeMappings.get( mapping );
                 typeMapping.add( customType );
             }
         }
     }
 
     /**
      * Specify whether the <tt>customType</tt> could be mapped to the
      * <tt>standardType</tt>.
      *
      * @param standardType the standard type (ejb, jar, war, ...)
      * @param customType   a user-defined type
      * @return true if the customType could be mapped to the standard type
      */
     public boolean isMappedToType( final String standardType, final String customType )
     {
         if ( !EarModuleFactory.isStandardArtifactType( standardType ) )
         {
             throw new IllegalStateException(
                 "Artifact type[" + standardType + "] is not a standard Ear artifact type["
                     + EarModuleFactory.getStandardArtifactTypes() + "]" );
         }
         final List<String> typeMappings = this.typeMappings.get( standardType );
         return typeMappings.contains( customType );
 
     }
 
     /**
      * Returns the standard type for the specified <tt>type</tt>. If
      * the specified type is already a standard type, the orignal type
      * is returned.
      *
      * @param type a type
      * @return the standard type (ejb, jar, war, ...) for this type
      */
     public String getStandardType( final String type )
         throws UnknownArtifactTypeException
     {
         if ( type == null )
         {
             throw new IllegalStateException( "custom type could not be null." );
         }
         else if ( EarModuleFactory.getStandardArtifactTypes().contains( type ) )
         {
             return type;
         }
         else if ( !customMappings.containsKey( type ) )
         {
             throw new UnknownArtifactTypeException( "Unknown artifact type[" + type + "]" );
         }
         else
         {
             return (String) customMappings.get( type );
         }
     }
 
     private void init()
     {
         // Initialize the typeMappings
         typeMappings.clear();
 
         // Clear the customMappings
         customMappings.clear();
 
         // Initialize the mapping with the standard artifact types
         for ( String type : EarModuleFactory.getStandardArtifactTypes() )
         {
             List<String> typeMapping = new ArrayList<String>();
             typeMapping.add( type );
             this.typeMappings.put( type, typeMapping );
         }
     }
 }