net.radai.beanz.codecs.ArrayCodec.java Source code

Introduction

Here is the source code for net.radai.beanz.codecs.ArrayCodec.java

Source

/*
 * Copyright (c) 2016 Radai Rosenblatt.
 * This file is part of Beanz.
 *
 * Beanz is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Beanz is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Beanz.  If not, see <http://www.gnu.org/licenses/>.
 */

package net.radai.beanz.codecs;

import net.radai.beanz.api.Codec;
import net.radai.beanz.util.ReflectionUtil;
import org.apache.commons.lang3.ClassUtils;
import org.apache.commons.lang3.text.StrBuilder;

import java.lang.reflect.Array;
import java.lang.reflect.Type;
import java.util.Collection;

import static net.radai.beanz.util.ReflectionUtil.erase;
import static net.radai.beanz.util.ReflectionUtil.prettyPrint;

/**
 * Created by Radai Rosenblatt
 */
public class ArrayCodec implements Codec {
    private Type type;
    private Type elementType;
    private Codec elementCodec;

    public ArrayCodec(Type type, Type elementType, Codec elementCodec) {
        if (type == null || elementType == null || elementCodec == null
                || !ClassUtils.isAssignable(erase(elementCodec.getType()), erase(elementType), true)
                || !ReflectionUtil.isArray(type)) {
            throw new IllegalArgumentException();
        }
        this.type = type;
        this.elementType = elementType;
        this.elementCodec = elementCodec;
    }

    @Override
    public Type getType() {
        return type;
    }

    @Override
    public Object decode(String encoded) {
        if (encoded == null || (encoded = encoded.trim()).isEmpty()) {
            return null;
        }
        if (!(encoded.startsWith("[") && encoded.endsWith("]"))) {
            throw new IllegalArgumentException();
        }
        String[] elements = encoded.substring(1, encoded.length() - 1).split("\\s*,\\s*");
        Class<?> erased = erase(getElementType());
        Object array = Array.newInstance(erased, elements.length);
        for (int i = 0; i < elements.length; i++) {
            Array.set(array, i, elementCodec.decode(elements[i]));
        }
        return array;
    }

    @Override
    public String encode(Object object) {
        if (object == null) {
            return null;
        }
        if (!ReflectionUtil.isArray(object.getClass())) {
            throw new IllegalArgumentException();
        }
        int size = Array.getLength(object);
        if (size == 0) {
            return "[]";
        }
        StrBuilder sb = new StrBuilder();
        sb.append("[");
        for (int i = 0; i < size; i++) {
            Object element = Array.get(object, i);
            String encoded = elementCodec.encode(element);
            sb.append(encoded).append(", ");
        }
        sb.delete(sb.length() - 2, sb.length()); //last ", "
        sb.append("]");
        return sb.toString();
    }

    public Object decodeArray(Collection<String> strCollection) {
        Object result = ReflectionUtil.instatiateArray(getElementType(), strCollection.size());
        int i = 0;
        for (String strValue : strCollection) {
            Object value = elementCodec.decode(strValue);
            Array.set(result, i++, value); //this is primitive-safe
        }
        return result;
    }

    public Type getElementType() {
        return elementType;
    }

    public Codec getElementCodec() {
        return elementCodec;
    }

    @Override
    public String toString() {
        return prettyPrint(getType()) + " codec: via " + elementCodec;
    }
}