MongoIdTypeConverter.java

package org.springframework.roo.addon.layers.repository.mongo;

import java.math.BigInteger;
import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;

import org.apache.commons.lang3.StringUtils;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Service;
import org.springframework.roo.shell.Completion;
import org.springframework.roo.shell.Converter;
import org.springframework.roo.shell.MethodTarget;

/**
 * Custom id type converter for {@link MongoIdType} to limit options in
 * {@link MongoCommands}
 * 
 * @author Stefan Schmidt
 * @since 1.2.0
 */
@Component
@Service
public class MongoIdTypeConverter implements Converter<MongoIdType> {

    public MongoIdType convertFromText(final String value,
            final Class<?> targetType, final String optionContext) {
        if (StringUtils.isBlank(value)) {
            return null;
        }
        return new MongoIdType(value);
    }

    public boolean getAllPossibleValues(final List<Completion> completions,
            final Class<?> targetType, final String existingData,
            final String optionContext, final MethodTarget target) {
        final SortedSet<String> types = new TreeSet<String>();
        types.add(BigInteger.class.getName());
        types.add("org.bson.types.ObjectId");

        for (final String type : types) {
            if (type.startsWith(existingData) || existingData.startsWith(type)) {
                completions.add(new Completion(type));
            }
        }
        return false;
    }

    public boolean supports(final Class<?> type, final String optionContext) {
        return MongoIdType.class.isAssignableFrom(type);
    }
}