Annotations.java

/**
 * Copyright © 2006-2016 Web Cohesion (info@webcohesion.com)
 *
 * 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 com.webcohesion.enunciate.javac.decorations;

import com.webcohesion.enunciate.javac.decorations.type.DecoratedTypeMirror;
import com.webcohesion.enunciate.javac.decorations.type.TypeMirrorUtils;

import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.MirroredTypeException;
import java.util.concurrent.Callable;

/**
 * @author Ryan Heaton
 */
@SuppressWarnings ( "unchecked" )
public class Annotations {

  private Annotations(){}

  public static DecoratedTypeMirror mirrorOf(Callable<Class<?>> annotationValueFunction, DecoratedProcessingEnvironment env) {
    return mirrorOf(annotationValueFunction, env, null);
  }

  public static DecoratedTypeMirror mirrorOf(Callable<Class<?>> annotationValueFunction, DecoratedProcessingEnvironment env, Class<?> emptyClass) {
    try {
      Class<?> clazz = annotationValueFunction.call();
      if (emptyClass != null && emptyClass.equals(clazz)) {
        return null;
      }
      return TypeMirrorUtils.mirrorOf(clazz, env);
    }
    catch (MirroredTypeException e) {
      DecoratedTypeMirror typeMirror = (DecoratedTypeMirror) TypeMirrorDecorator.decorate(e.getTypeMirror(), env);
      if (emptyClass != null && same(typeMirror, emptyClass)) {
        return null;
      }
      return typeMirror;
    }
    catch (RuntimeException e) {
      throw e;
    }
    catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  private static boolean same(DecoratedTypeMirror typeMirror, Class<?> emptyClass) {
    if (typeMirror instanceof DeclaredType) {
      Element element = ((DeclaredType) typeMirror).asElement();
      if (element instanceof TypeElement) {
        String fqn = ((TypeElement) element).getQualifiedName().toString();
        if (fqn.equals(emptyClass.getCanonicalName())) {
          return true;
        }
      }
    }
    return false;
  }

}