package gregtech.api.objects;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Objects;

import com.google.common.collect.Collections2;

public class GTArrayList<E> extends ArrayList<E> {

    private static final long serialVersionUID = 1L;
    private int size_sS;

    private final boolean mAllowNulls;

    public GTArrayList(boolean aAllowNulls, int aCapacity) {
        super(aCapacity);
        mAllowNulls = aAllowNulls;
    }

    @SafeVarargs
    public GTArrayList(boolean aAllowNulls, E... aArray) {
        super(Arrays.asList(aArray));
        mAllowNulls = aAllowNulls;
        if (!mAllowNulls) {
            size_sS = size();
            for (int i = 0; i < size_sS; i++) if (get(i) == null) {
                remove(i--);
                size_sS = size();
            }
        }
    }

    public GTArrayList(boolean aAllowNulls, Collection<? extends E> aList) {
        super(aList);
        mAllowNulls = aAllowNulls;
        if (!mAllowNulls) {
            size_sS = size();
            for (int i = 0; i < size_sS; i++) if (get(i) == null) {
                remove(i--);
                size_sS = size();
            }
        }
    }

    @Override
    public E set(int aIndex, E aElement) {
        if (mAllowNulls || aElement != null) return super.set(aIndex, aElement);
        return null;
    }

    @Override
    public boolean add(E aElement) {
        if (mAllowNulls || aElement != null) return super.add(aElement);
        return false;
    }

    @Override
    public void add(int aIndex, E aElement) {
        if (mAllowNulls || aElement != null) super.add(aIndex, aElement);
    }

    @Override
    public boolean addAll(Collection<? extends E> aList) {
        return super.addAll(Collections2.filter(aList, Objects::nonNull));
    }

    @Override
    public boolean addAll(int aIndex, Collection<? extends E> aList) {
        return super.addAll(aIndex, Collections2.filter(aList, Objects::nonNull));
    }
}