package com.jsoniter.output;

import com.jsoniter.any.Any;
import com.jsoniter.spi.Encoder;
import com.jsoniter.spi.TypeLiteral;

import java.io.IOException;
import java.lang.reflect.Type;
import java.util.List;

class ReflectionListEncoder implements Encoder {

    private final TypeLiteral compTypeLiteral;

    public ReflectionListEncoder(Class clazz, Type[] typeArgs) {
        if (typeArgs.length > 0) {
            compTypeLiteral = TypeLiteral.create(typeArgs[0]);
        } else {
            compTypeLiteral = TypeLiteral.create(Object.class);
        }
    }

    @Override
    public void encode(Object obj, JsonStream stream) throws IOException {
        if (null == obj) {
            stream.writeNull();
            return;
        }
        List list = (List) obj;
        if (list.isEmpty()) {
            stream.writeEmptyArray();
            return;
        }
        stream.writeArrayStart();
        stream.writeVal(compTypeLiteral, list.get(0));
        for (int i = 1; i < list.size(); i++) {
            stream.writeMore();
            stream.writeVal(compTypeLiteral, list.get(i));
        }
        stream.writeArrayEnd();
    }

    @Override
    public Any wrap(Object obj) {
        List col = (List) obj;
        return Any.wrap(col);
    }
}
