org.picketlink.idm.jpa.internal.mappers.AttributedValueMapper Maven / Gradle / Ivy

Go to download
/*
 * JBoss, Home of Professional Open Source
 *
 * Copyright 2013 Red Hat, Inc. and/or its affiliates.
 *
 * 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 org.picketlink.idm.jpa.internal.mappers;

import org.picketlink.common.properties.Property;
import org.picketlink.common.properties.query.AnnotatedPropertyCriteria;
import org.picketlink.common.properties.query.PropertyQueries;
import org.picketlink.idm.IDMMessages;
import org.picketlink.idm.jpa.annotations.AttributeClass;
import org.picketlink.idm.jpa.annotations.AttributeValue;
import org.picketlink.idm.jpa.annotations.IdentityClass;
import org.picketlink.idm.jpa.annotations.PartitionClass;
import org.picketlink.idm.jpa.annotations.RelationshipClass;

import static org.picketlink.common.util.StringUtil.isNullOrEmpty;

/**
 * @author pedroigor
 */
public class AttributedValueMapper extends AbstractIdentityManagedMapper {

    @Override
    public boolean supports(Class entityType) {
        return !PropertyQueries.createQuery(entityType)
                .addCriteria(new AnnotatedPropertyCriteria(AttributeValue.class))
                .getResultList().isEmpty()
                && PropertyQueries.createQuery(entityType)
                .addCriteria(new AnnotatedPropertyCriteria(AttributeClass.class))
                .getResultList().isEmpty()
                && PropertyQueries.createQuery(entityType)
                .addCriteria(new AnnotatedPropertyCriteria(IdentityClass.class))
                .getResultList().isEmpty()
                && PropertyQueries.createQuery(entityType)
                .addCriteria(new AnnotatedPropertyCriteria(PartitionClass.class))
                .getResultList().isEmpty()
                && PropertyQueries.createQuery(entityType)
                .addCriteria(new AnnotatedPropertyCriteria(RelationshipClass.class))
                .getResultList().isEmpty();
    }

    @Override
    public EntityMapping configure(Class managedType, Class entityType) {
        EntityMapping entityMapping = new EntityMapping(managedType);

        for (Property mappedProperty : getAnnotatedProperties(AttributeValue.class, entityType)) {
            AttributeValue attributeValue = mappedProperty.getAnnotatedElement().getAnnotation(AttributeValue.class);
            String propertyName = mappedProperty.getName();

            if (!isNullOrEmpty(attributeValue.name())) {
                propertyName = attributeValue.name();
            }

            Property property = getNamedProperty(propertyName, managedType);

            if (property == null) {
                throw IDMMessages.MESSAGES.configJpaStoreMappedPropertyNotFound(entityType, propertyName, managedType);
            }

            entityMapping.addProperty(property, mappedProperty);
        }

        entityMapping.addOwnerProperty(entityType);

        return entityMapping;
    }

}