com.heroku.api.request.RequestConfig Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of heroku-api Show documentation
There is a newer version: 0.46
package com.heroku.api.request;

import com.heroku.api.Heroku;
import com.heroku.api.parser.Json;

import java.util.EnumMap;
import java.util.HashMap;
import java.util.Map;

/**
 * Wrapper class for an {@link EnumMap} to configure a {@link Request}. A {@link com.heroku.api.Heroku.RequestKey} serves as
 * a predetermined key for a name/value parameter.
 *
 * @author Naaman Newbold
 */
public class RequestConfig {

    private String appName;

    private final Map config = new EnumMap(Heroku.RequestKey.class);

    /**
     * Sets the {Heroku.RequestKey.AppName} parameter.
     * @param appName Name of the app to specify in the config.
     * @return A new {@link RequestConfig}
     */
    public RequestConfig app(String appName) {
        this.appName = appName;
        return this;
    }

    public String getAppName() {
        return this.appName;
    }

    /**
     * Sets a {@link com.heroku.api.Heroku.RequestKey} parameter.
     * @param key Heroku request key
     * @param value value of the property
     * @return A new {@link RequestConfig}
     */
    public RequestConfig with(Heroku.RequestKey key, String value) {
        RequestConfig newConfig = copy();
        newConfig.config.put(key, new Either(value));
        return newConfig;
    }

    /**
     * Sets a {@link com.heroku.api.Heroku.RequestKey} parameter.
     * @param key Heroku request key
     * @param value value of the property
     * @return A new {@link RequestConfig}
     */
    public RequestConfig with(Heroku.RequestKey key, Map value) {
        RequestConfig newConfig = copy();
        newConfig.config.put(key, new Either(value));
        return newConfig;
    }

    /**
     * Sets a {@link com.heroku.api.Heroku.RequestKey} parameter.
     * @param key Heroku request key
     * @param data arbitrary key/value map
     * @return A new {@link RequestConfig}
     */
    public RequestConfig withOptions(Heroku.RequestKey key, Map data) {
        RequestConfig newConfig = copy();
        newConfig.config.put(key, new Either(new Data(data)));
        return newConfig;
    }

    public String get(Heroku.RequestKey key) {
        return config.get(key).string();
    }

    public Map getMap(Heroku.RequestKey key) {
        return config.get(key).map();
    }

    public boolean has(Heroku.RequestKey key){
        return config.containsKey(key);
    }

    public String asJson() {
        return Json.encode(asMap());
    }

    public Map asMap() {
        return stringifyMap(config);
    }

    private Map stringifyMap(Map map) {
        Map jsonMap = new HashMap<>();
        for (Heroku.RequestKey key : map.keySet()) {
            RequestConfig.Either either = map.get(key);
            if (either.is(String.class)) {
                jsonMap.put(key.queryParameter, either.string());
            } else if (either.is(Boolean.class)) {
                jsonMap.put(key.queryParameter, either.bool());
            } else if (either.is(Data.class)) {
                jsonMap.put(key.queryParameter, either.data());
            } else {
                jsonMap.put(key.queryParameter, stringifyMap(either.map()));
            }
        }
        return jsonMap;
    }

    private RequestConfig copy() {
        RequestConfig newConfig = new RequestConfig();
        newConfig.app(this.appName);
        newConfig.config.putAll(config);
        return newConfig;
    }

    public static class Either {

        private Class type;

        private String string;

        private Boolean bool;

        private Data data;

        private Map map;

        public Either(Boolean value) {
            this.type = Boolean.class;
            this.bool = value;
        }

        public Either(String value) {
            this.type = String.class;
            this.string = value;
        }

        public Either(Data data) {
            this.type = Data.class;
            this.data = data;
        }

        public Either(Map value) {
            this.type = Map.class;
            this.map = value;
        }

        public String string() {
            return string;
        }

        public Boolean bool() {
            return bool;
        }

        public Map data() {
            return data.map();
        }

        public Map map() {
            return map;
        }

        public Boolean is(Class c) {
            return c.equals(type);
        }
    }

    public static class Data {

        private Map map;

        public Data(Map map) {
            this.map = map;
        }

        public Map map() {
            return map;
        }

    }
}