// 
// Decompiled by Procyon v0.6.0
// 

package org.bson.codecs.pojo;

import org.bson.BsonType;
import org.bson.BsonObjectId;
import org.bson.types.ObjectId;

final class ConventionObjectIdGeneratorsImpl implements Convention
{
    @Override
    public void apply(final ClassModelBuilder<?> classModelBuilder) {
        if (classModelBuilder.getIdGenerator() == null && classModelBuilder.getIdPropertyName() != null) {
            final PropertyModelBuilder<?> idProperty = classModelBuilder.getProperty(classModelBuilder.getIdPropertyName());
            if (idProperty != null) {
                final Class<?> idType = idProperty.getTypeData().getType();
                if (classModelBuilder.getIdGenerator() == null && idType.equals(ObjectId.class)) {
                    classModelBuilder.idGenerator(IdGenerators.OBJECT_ID_GENERATOR);
                }
                else if (classModelBuilder.getIdGenerator() == null && idType.equals(BsonObjectId.class)) {
                    classModelBuilder.idGenerator(IdGenerators.BSON_OBJECT_ID_GENERATOR);
                }
                else if (classModelBuilder.getIdGenerator() == null && idType.equals(String.class) && idProperty.getBsonRepresentation() == BsonType.OBJECT_ID) {
                    classModelBuilder.idGenerator(IdGenerators.STRING_ID_GENERATOR);
                }
            }
        }
    }
}