com.brettonw.bedrock.bag.SelectKey Maven / Gradle / Ivy

Go to download
package com.brettonw.bedrock.bag;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Supplier;

public class SelectKey {
    private static final Logger log = LogManager.getLogger (SelectKey.class);

    public static final String AS_KEY = "as";
    public static final String KEYS_KEY = "keys";
    public static final String TYPE_KEY = "type";
    public static final SelectType DEFAULT_TYPE = SelectType.INCLUDE;

    private Map keys;
    private SelectType type;

    public SelectKey () {
        this (DEFAULT_TYPE, (String[]) null, (String[]) null);
    }

    public SelectKey (SelectType type, String... keysArray) {
        this (type, keysArray, null);
    }

    public SelectKey (SelectType type, String[] keysArray, String[] asArray) {
        this.type = type;
        setKeys (keysArray, asArray);
    }

    public SelectKey (String... keysArray) {
        this (DEFAULT_TYPE, keysArray);
    }

    public SelectKey (BagArray bagArray) {
        this (DEFAULT_TYPE, bagArray);
    }

    public SelectKey (SelectType type, BagArray keysArray) {
        this (type, keysArray.toArray (String.class), null);
    }

    public SelectKey (SelectType type, BagArray keysArray, BagArray asArray) {
        this (type,
                (keysArray != null) ? keysArray.toArray (String.class) : null,
                (asArray != null) ? asArray.toArray (String.class) : null);
    }

    public SelectKey (BagObject bagObject) {
        this (bagObject.getEnum (TYPE_KEY, SelectType.class, () -> DEFAULT_TYPE), bagObject.getBagArray (KEYS_KEY), bagObject.getBagArray (AS_KEY));
    }

    public String select (String key, Supplier notFound) {
        if (key != null) {
            switch (type) {
                case INCLUDE:
                    // in the include case, we can map the requested to to an "as" key
                    if (keys.containsKey (key)) return keys.get (key);
                    break;
                case EXCLUDE:
                    if (!keys.containsKey (key)) return key;
                    break;
            }
        }
        return notFound.get ();
    }

    public String select (String key) {
        return select (key, () -> null);
    }

    public SelectKey setType (SelectType type) {
        this.type = type;
        return this;
    }

    public SelectType getType () {
        return type;
    }

    public SelectKey setKeys (String... keysArray) {
        return setKeys (keysArray, null);
    }

    public SelectKey setKeys (String[] keysArray, String[] asArray) {
        keys = new HashMap<> ();
        return addKeys (keysArray, asArray);
    }

    public SelectKey addKeys (String... keysArray) {
        return addKeys (keysArray, null);
    }
    public SelectKey addKeys (String[] keysArray, String[] asArray) {
        if (keysArray != null) {
            // if asArray is not supplied, use the keysArray itself
            if (asArray == null) asArray = keysArray;

            // check that the two arrays are the same size, and build the map if they are.
            if (keysArray.length == asArray.length) {
                List keysList = Arrays.asList (keysArray);
                for (int i = 0, end = keysArray.length; i < end; ++i) {
                    keys.put (keysArray[i], asArray[i]);
                }
            } else {
                // what? throw an exception
                log.error ("Invalid 'AS' mapping specified");
            }

        }
        return this;
    }
}