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

package org.bson.codecs.pojo;

import org.bson.codecs.configuration.CodecConfigurationException;
import java.util.HashSet;
import java.util.ArrayList;
import org.bson.BsonType;
import org.bson.codecs.DecoderContext;
import org.bson.BsonReader;
import java.util.Iterator;
import org.bson.codecs.EncoderContext;
import org.bson.BsonWriter;
import java.util.Collection;
import org.bson.codecs.Codec;

final class CollectionPropertyCodecProvider implements PropertyCodecProvider
{
    @Override
    public <T> Codec<T> get(final TypeWithTypeParameters<T> type, final PropertyCodecRegistry registry) {
        if (Collection.class.isAssignableFrom(type.getType()) && type.getTypeParameters().size() == 1) {
            return (Codec<T>)new CollectionCodec((Class<Collection<Object>>)type.getType(), registry.get((TypeWithTypeParameters<Object>)type.getTypeParameters().get(0)));
        }
        return null;
    }
    
    private static class CollectionCodec<T> implements Codec<Collection<T>>
    {
        private final Class<Collection<T>> encoderClass;
        private final Codec<T> codec;
        
        CollectionCodec(final Class<Collection<T>> encoderClass, final Codec<T> codec) {
            this.encoderClass = encoderClass;
            this.codec = codec;
        }
        
        @Override
        public void encode(final BsonWriter writer, final Collection<T> collection, final EncoderContext encoderContext) {
            writer.writeStartArray();
            for (final T value : collection) {
                if (value == null) {
                    writer.writeNull();
                }
                else {
                    this.codec.encode(writer, value, encoderContext);
                }
            }
            writer.writeEndArray();
        }
        
        @Override
        public Collection<T> decode(final BsonReader reader, final DecoderContext context) {
            final Collection<T> collection = this.getInstance();
            reader.readStartArray();
            while (reader.readBsonType() != BsonType.END_OF_DOCUMENT) {
                if (reader.getCurrentBsonType() == BsonType.NULL) {
                    collection.add(null);
                    reader.readNull();
                }
                else {
                    collection.add(this.codec.decode(reader, context));
                }
            }
            reader.readEndArray();
            return collection;
        }
        
        @Override
        public Class<Collection<T>> getEncoderClass() {
            return this.encoderClass;
        }
        
        private Collection<T> getInstance() {
            if (this.encoderClass.isInterface()) {
                if (this.encoderClass.isAssignableFrom(ArrayList.class)) {
                    return new ArrayList<T>();
                }
                if (this.encoderClass.isAssignableFrom(HashSet.class)) {
                    return new HashSet<T>();
                }
                throw new CodecConfigurationException(String.format("Unsupported Collection interface of %s!", this.encoderClass.getName()));
            }
            else {
                try {
                    return this.encoderClass.getDeclaredConstructor((Class<?>[])new Class[0]).newInstance(new Object[0]);
                }
                catch (final Exception e) {
                    throw new CodecConfigurationException(e.getMessage(), e);
                }
            }
        }
    }
}