/*
    * Copyright 2010-2013 smartics, Kronseder & Reiner GmbH
    *
    * 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 de.smartics.testdoc.collect.extractor;
  
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  import javax.lang.model.element.AnnotationMirror;
  import javax.lang.model.element.AnnotationValue;
  import javax.lang.model.element.Element;
  import javax.lang.model.element.ExecutableElement;
  import javax.lang.model.element.TypeElement;
  
  import org.junit.experimental.categories.Category;
  
  /**
   * Extracts category information.
   *
   * @author <a href="mailto:robert.reiner@smartics.de">Robert Reiner</a>
   * @version $Revision:591 $
   */
  class CategoryExtractor
  {
    // ********************************* Fields *********************************
  
    // --- constants ------------------------------------------------------------
  
    // --- members --------------------------------------------------------------
  
    /**
     * The type element of the enclosing unit test class.
     */
    private final TypeElement unitTestType;
  
    // ****************************** Initializer *******************************
  
    // ****************************** Constructors ******************************
  
    /**
     * Default constructor.
     *
     * @param unitTestType the type element of the enclosing unit test class.
     */
    CategoryExtractor(final TypeElement unitTestType)
    {
      this.unitTestType = unitTestType;
    }
  
    // ****************************** Inner Classes *****************************
  
    // ********************************* Methods ********************************
  
    // --- init -----------------------------------------------------------------
  
    // --- get&set --------------------------------------------------------------
  
    // --- business -------------------------------------------------------------
  
    List<String> calcCategories(final Element element)
    {
      final List<String> categories = new ArrayList<String>();
  
      if (unitTestType != null)
      {
        for (final AnnotationMirror annotation : unitTestType
            .getAnnotationMirrors())
        {
          addCategories(categories, annotation);
        }
      }
      for (final AnnotationMirror annotation : element.getAnnotationMirrors())
      {
        addCategories(categories, annotation);
      }
  
      return categories;
    }
  
    private static void addCategories(final List<String> categories,
        final AnnotationMirror annotation)
    {
      if (Category.class.getName().equals(
          annotation.getAnnotationType().toString()))
      {
        final Map<? extends ExecutableElement, ? extends AnnotationValue> map =
           annotation.getElementValues();
       for (final Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry : map
           .entrySet())
       {
         addCategories(categories, entry);
       }
     }
   }
 
   private static void addCategories(
       final List<String> categories,
       final Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry)
   {
     final String key = entry.getKey().toString();
     if (key != null && key.trim().endsWith("value()"))
     {
       final Object values = entry.getValue().getValue();
       if (values instanceof Iterable)
       {
         for (final Object value : (Iterable<?>) values)
         {
           final String valueString = value.toString();
           final String name = TestDocAnnotationExtractor.typeNameFix(valueString);
           categories.add(name);
         }
       }
     }
   }
 
   // --- object basics --------------------------------------------------------
 
 }