Beans

package fr.ifremer.quadrige2.core.dao.technical;

/*-
 * #%L
 * Quadrige2 Core :: Quadrige2 Core Shared
 * $Id:$
 * $HeadURL:$
 * %%
 * Copyright (C) 2017 Ifremer
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */


import com.google.common.base.Function;
import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import fr.ifremer.quadrige2.core.exception.Quadrige2TechnicalException;
import org.apache.commons.beanutils.PropertyUtils;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.MapUtils;
import org.apache.commons.lang3.StringUtils;

import java.lang.reflect.InvocationTargetException;
import java.util.*;

/**
 * helper class for beans (split by property, make sure list exists, ...)
 * Created by blavenie on 13/10/15.
 */
public class Beans {

    /**
     * <p>getList.</p>
     *
     * @param list a {@link java.util.Collection} object.
     * @param <E> a E object.
     * @return a {@link java.util.List} object.
     */
    public static <E> List<E> getList(Collection<E> list) {
        if (CollectionUtils.isEmpty(list)) {
            return Lists.newArrayList();
        } else {
            return Lists.newArrayList(list);
        }
    }

    /**
     * <p>getListWithoutNull.</p>
     *
     * @param list a {@link java.util.Collection} object.
     * @param <E> a E object.
     * @return a {@link java.util.List} object.
     */
    public static <E> List<E> getListWithoutNull(Collection<E> list) {
        List<E> result = getList(list);
        result.removeAll(Collections.singleton((E) null));
        return result;
    }

    /**
     * <p>getSet.</p>
     *
     * @param list a {@link java.util.Collection} object.
     * @param <E> a E object.
     * @return a {@link java.util.Set} object.
     */
    public static <E> Set<E> getSet(Collection<E> list) {
        if (CollectionUtils.isEmpty(list)) {
            return Sets.newHashSet();
        } else {
            return Sets.newHashSet(list);
        }
    }

    /**
     * <p>getSetWithoutNull.</p>
     *
     * @param list a {@link java.util.Collection} object.
     * @param <E> a E object.
     * @return a {@link java.util.Set} object.
     */
    public static <E> Set<E> getSetWithoutNull(Collection<E> list) {
        Set<E> result = getSet(list);
        result.removeAll(Collections.singleton((E) null));
        return result;
    }

    /**
     * <p>getMap.</p>
     *
     * @param map a {@link java.util.Map} object.
     * @param <K> a K object.
     * @param <V> a V object.
     * @return a {@link java.util.Map} object.
     */
    public static <K, V> Map<K, V> getMap(Map<K, V> map) {
        if (MapUtils.isEmpty(map)) {
            return Maps.newHashMap();
        } else {
            return Maps.newHashMap(map);
        }
    }

    /**
     * <p>splitByProperty.</p>
     *
     * @param list a {@link java.lang.Iterable} object.
     * @param propertyName a {@link java.lang.String} object.
     * @param <K> a K object.
     * @param <V> a V object.
     * @return a {@link java.util.Map} object.
     */
    public static <K, V> Map<K, V> splitByProperty(Iterable<V> list, String propertyName) {
        Preconditions.checkArgument(StringUtils.isNotBlank(propertyName));
        Function<V, K> function = newPropertyFunction(propertyName);
        return getMap(Maps.uniqueIndex(list, function));
    }

    /**
     * <p>collectProperties.</p>
     *
     * @param collection a {@link java.util.Collection} object.
     * @param propertyName a {@link java.lang.String} object.
     * @param <K> a K object.
     * @param <V> a V object.
     * @return a {@link java.util.List} object.
     */
    public static <K, V> List<K> collectProperties(Collection<V> collection, String propertyName) {
        List<V> list;
        if (collection != null && collection instanceof List) {
            list = (List<V>) collection;
        } else {
            list = getList(collection);
        }
        Preconditions.checkArgument(StringUtils.isNotBlank(propertyName));
        Function<V, K> function = newPropertyFunction(propertyName);
        return Lists.transform(list, function);
    }

    private static <K, V> Function<V, K> newPropertyFunction(final String propertyName) {
        return input -> getProperty(input, propertyName);
    }

    /**
     * <p>getProperty.</p>
     *
     * @param object       a K object.
     * @param propertyName a {@link java.lang.String} object.
     * @param <K>          a K object.
     * @param <V>          a V object.
     * @return a V object.
     */
    @SuppressWarnings("unchecked")
    public static <K, V> V getProperty(K object, String propertyName) {
        try {
            return (V) PropertyUtils.getProperty(object, propertyName);
        } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
            throw new Quadrige2TechnicalException(String.format("Could not get property %1s on object of type %2s", propertyName, object.getClass().getName()), e);
        }
    }

    /**
     * <p>setProperty.</p>
     *
     * @param object       a K object.
     * @param propertyName a {@link java.lang.String} object.
     * @param value        a V object.
     * @param <K>          a K object.
     * @param <V>          a V object.
     */
    public static <K, V> void setProperty(K object, String propertyName, V value) {
        try {
            PropertyUtils.setProperty(object, propertyName, value);
        } catch (InvocationTargetException | NoSuchMethodException | IllegalAccessException e) {
            throw new Quadrige2TechnicalException(String.format("Could not set property %1s not found on object of type %2s", propertyName, object.getClass().getName()), e);
        }
    }

    public static Integer[] asIntegerArray(Collection<Integer> values) {
        if (CollectionUtils.isEmpty(values)) {
            return null;
        }
        return values.toArray(new Integer[values.size()]);
    }

    public static String[] asStringArray(Collection<String> values) {
        if (CollectionUtils.isEmpty(values)) {
            return null;
        }
        return values.toArray(new String[values.size()]);
    }

    public static String[] asStringArray(String value, String delimiter) {
        if (StringUtils.isBlank(value)) return new String[0];
        StringTokenizer tokenizer = new StringTokenizer(value, delimiter);
        String[] values = new String[tokenizer.countTokens()];
        int i=0;
        while (tokenizer.hasMoreTokens()) {
            values[i] = tokenizer.nextToken();
            i++;
        }
        return values;
    }
}