package org.hibernate.validator.internal.util.annotation;
import static org.hibernate.validator.internal.util.CollectionHelper.newHashMap;
import java.io.Serializable;
import java.lang.annotation.Annotation;
import java.lang.invoke.MethodHandles;
import java.lang.reflect.Method;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.SortedSet;
import java.util.TreeSet;
import org.hibernate.validator.internal.util.CollectionHelper;
import org.hibernate.validator.internal.util.StringHelper;
import org.hibernate.validator.internal.util.logging.Log;
import org.hibernate.validator.internal.util.logging.LoggerFactory;
import org.hibernate.validator.internal.util.privilegedactions.GetAnnotationAttributes;
import org.hibernate.validator.internal.util.privilegedactions.GetDeclaredMethods;
public class AnnotationDescriptor<A extends Annotation> implements Serializable {
private static final Log LOG = LoggerFactory.make( MethodHandles.lookup() );
private final Class<A> type;
private final Map<String, Object> attributes;
private final int hashCode;
private final A annotation;
@SuppressWarnings("unchecked")
public AnnotationDescriptor(A annotation) {
this.type = (Class<A>) annotation.annotationType();
this.attributes = run( GetAnnotationAttributes.action( annotation ) );
this.annotation = annotation;
this.hashCode = buildHashCode();
}
public AnnotationDescriptor(AnnotationDescriptor<A> descriptor) {
this.type = descriptor.type;
this.attributes = descriptor.attributes;
this.hashCode = descriptor.hashCode;
this.annotation = descriptor.annotation;
}
private AnnotationDescriptor(Class<A> annotationType, Map<String, Object> attributes) {
this.type = annotationType;
this.attributes = CollectionHelper.toImmutableMap( attributes );
this.hashCode = buildHashCode();
this.annotation = AnnotationFactory.create( this );
}
public Class<A> getType() {
return type;
}
public Map<String, Object> getAttributes() {
return attributes;
}
@SuppressWarnings("unchecked")
public <T> T getMandatoryAttribute(String attributeName, Class<T> attributeType) {
Object attribute = attributes.get( attributeName );
if ( attribute == null ) {
throw LOG.getUnableToFindAnnotationAttributeException( type, attributeName, null );
}
if ( !attributeType.isAssignableFrom( attribute.getClass() ) ) {
throw LOG.getWrongAnnotationAttributeTypeException( type, attributeName, attributeType, attribute.getClass() );
}
return (T) attribute;
}
@SuppressWarnings("unchecked")
public <T> T getAttribute(String attributeName, Class<T> attributeType) {
Object attribute = attributes.get( attributeName );
if ( attribute == null ) {
return null;
}
if ( !attributeType.isAssignableFrom( attribute.getClass() ) ) {
throw LOG.getWrongAnnotationAttributeTypeException( type, attributeName, attributeType, attribute.getClass() );
}
return (T) attribute;
}
public Object getAttribute(String attributeName) {
return attributes.get( attributeName );
}
public A getAnnotation() {
return annotation;
}
@Override
public boolean equals(Object obj) {
if ( this == obj ) {
return true;
}
if ( obj == null || !( obj instanceof AnnotationDescriptor ) ) {
return false;
}
AnnotationDescriptor<?> other = (AnnotationDescriptor<?>) obj;
if ( !type.equals( other.type ) ) {
return false;
}
if ( attributes.size() != other.attributes.size() ) {
return false;
}
for ( Entry<String, Object> member : attributes.entrySet() ) {
Object value = member.getValue();
Object otherValue = other.attributes.get( member.getKey() );
if ( !areEqual( value, otherValue ) ) {
return false;
}
}
return true;
}
@Override
public int hashCode() {
return hashCode;
}
@Override
public String toString() {
StringBuilder result = new StringBuilder();
result.append( '@' ).append( StringHelper.toShortString( type ) ).append( '(' );
for ( String s : getRegisteredAttributesInAlphabeticalOrder() ) {
result.append( s ).append( '=' ).append( attributes.get( s ) ).append( ", " );
}
if ( attributes.size() > 0 ) {
result.delete( result.length() - 2, result.length() );
result.append( ")" );
}
else {
result.delete( result.length() - 1, result.length() );
}
return result.toString();
}
private SortedSet<String> getRegisteredAttributesInAlphabeticalOrder() {
return new TreeSet<String>( attributes.keySet() );
}
private int buildHashCode() {
int hashCode = 0;
for ( Entry<String, Object> member : attributes.entrySet() ) {
Object value = member.getValue();
int nameHashCode = member.getKey().hashCode();
int valueHashCode = !value.getClass().isArray() ? value.hashCode()
: value.getClass() == boolean[].class ? Arrays.hashCode( (boolean[]) value )
: value.getClass() == byte[].class ? Arrays.hashCode( (byte[]) value )
: value.getClass() == char[].class ? Arrays.hashCode( (char[]) value )
: value.getClass() == double[].class ? Arrays.hashCode( (double[]) value )
: value.getClass() == float[].class ? Arrays.hashCode( (float[]) value )
: value.getClass() == int[].class ? Arrays.hashCode( (int[]) value )
: value.getClass() == long[].class ? Arrays.hashCode( (long[]) value )
: value.getClass() == short[].class ? Arrays.hashCode( (short[]) value )
: Arrays.hashCode( (Object[]) value );
hashCode += 127 * nameHashCode ^ valueHashCode;
}
return hashCode;
}
private boolean areEqual(Object o1, Object o2) {
return !o1.getClass().isArray() ? o1.equals( o2 )
: o1.getClass() == boolean[].class ? Arrays.equals( (boolean[]) o1, (boolean[]) o2 )
: o1.getClass() == byte[].class ? Arrays.equals( (byte[]) o1, (byte[]) o2 )
: o1.getClass() == char[].class ? Arrays.equals( (char[]) o1, (char[]) o2 )
: o1.getClass() == double[].class ? Arrays.equals( (double[]) o1, (double[]) o2 )
: o1.getClass() == float[].class ? Arrays.equals( (float[]) o1, (float[]) o2 )
: o1.getClass() == int[].class ? Arrays.equals( (int[]) o1, (int[]) o2 )
: o1.getClass() == long[].class ? Arrays.equals( (long[]) o1, (long[]) o2 )
: o1.getClass() == short[].class ? Arrays.equals( (short[]) o1, (short[]) o2 )
: Arrays.equals( (Object[]) o1, (Object[]) o2 );
}
public static class Builder<S extends Annotation> {
private final Class<S> type;
private final Map<String, Object> attributes;
public Builder(Class<S> type) {
this.type = type;
this.attributes = new HashMap<String, Object>();
}
public Builder(Class<S> type, Map<String, Object> attributes) {
this.type = type;
this.attributes = new HashMap<String, Object>( attributes );
}
@SuppressWarnings("unchecked")
public Builder(S annotation) {
this.type = (Class<S>) annotation.annotationType();
this.attributes = new HashMap<String, Object>( run( GetAnnotationAttributes.action( annotation ) ) );
}
public void setAttribute(String attributeName, Object value) {
attributes.put( attributeName, value );
}
public boolean hasAttribute(String key) {
return attributes.containsKey( key );
}
public Class<S> getType() {
return type;
}
public AnnotationDescriptor<S> build() {
return new AnnotationDescriptor<>( type, getAnnotationAttributes() );
}
private Map<String, Object> getAnnotationAttributes() {
Map<String, Object> result = newHashMap( attributes.size() );
int processedValuesFromDescriptor = 0;
final Method[] declaredMethods = run( GetDeclaredMethods.action( type ) );
for ( Method m : declaredMethods ) {
Object elementValue = attributes.get( m.getName() );
if ( elementValue != null ) {
result.put( m.getName(), elementValue );
processedValuesFromDescriptor++;
}
else if ( m.getDefaultValue() != null ) {
result.put( m.getName(), m.getDefaultValue() );
}
else {
throw LOG.getNoValueProvidedForAnnotationAttributeException(
m.getName(),
type
);
}
}
if ( processedValuesFromDescriptor != attributes.size() ) {
Set<String> unknownAttributes = attributes.keySet();
unknownAttributes.removeAll( result.keySet() );
throw LOG.getTryingToInstantiateAnnotationWithUnknownAttributesException(
type,
unknownAttributes
);
}
return result;
}
@Override
public String toString() {
StringBuilder result = new StringBuilder();
result.append( '@' ).append( StringHelper.toShortString( type ) ).append( '(' );
for ( String s : getRegisteredAttributesInAlphabeticalOrder() ) {
result.append( s ).append( '=' ).append( attributes.get( s ) ).append( ", " );
}
if ( attributes.size() > 0 ) {
result.delete( result.length() - 2, result.length() );
result.append( ")" );
}
else {
result.delete( result.length() - 1, result.length() );
}
return result.toString();
}
private SortedSet<String> getRegisteredAttributesInAlphabeticalOrder() {
return new TreeSet<String>( attributes.keySet() );
}
}
private static <V> V run(PrivilegedAction<V> action) {
return System.getSecurityManager() != null ? AccessController.doPrivileged( action ) : action.run();
}
}