PYamlObject.java

package org.mapfish.print.wrapper.yaml;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.mapfish.print.ExceptionUtils;
import org.mapfish.print.wrapper.PAbstractObject;
import org.mapfish.print.wrapper.PArray;
import org.mapfish.print.wrapper.PElement;
import org.mapfish.print.wrapper.PObject;
import org.mapfish.print.wrapper.json.PJsonObject;

import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;


/**
 * Object wrapper for Yaml parsing.
 */
public class PYamlObject extends PAbstractObject {
    private final Map<String, Object> obj;

    /**
     * Constructor.
     *
     * @param obj the internal json element
     * @param contextName the field name of this element in the parent.
     */
    public PYamlObject(final Map<String, Object> obj, final String contextName) {
        this(null, obj, contextName);
    }

    /**
     * Constructor.
     *
     * @param parent the parent element
     * @param obj the internal json element
     * @param contextName the field name of this element in the parent.
     */
    public PYamlObject(final PElement parent, final Map<String, Object> obj, final String contextName) {
        super(parent, contextName);
        this.obj = obj;
    }

    @SuppressWarnings("unchecked")
    @Override
    public final Object opt(final String key) {
        final Object value = this.obj.get(key);
        if (value instanceof Map<?, ?>) {
            return new PYamlObject(this, (Map<String, Object>) value, key);
        } else if (value instanceof List<?>) {
            return new PYamlArray(this, (List<Object>) value, key);
        } else if (value != null && value.getClass().isArray()) {
            return new PYamlArray(this, Arrays.asList((Object[]) value), key);
        } else {
            return value;
        }
    }

    @Override
    public final String optString(final String key) {
        Object value = this.obj.get(key);
        if (value == null || value instanceof String) {
            return (String) value;
        } else if (value instanceof Map) {
            return new JSONObject((Map) value).toString();
        } else if (value instanceof Collection) {
            return new JSONArray((Collection) value).toString();
        } else {
            return value.toString();
        }
    }

    @Override
    public final Integer optInt(final String key) {
        return (Integer) this.obj.get(key);
    }

    @Override
    public final Long optLong(final String key) {
        return (Long) this.obj.get(key);
    }

    @Override
    public final Double optDouble(final String key) {
        Number result = (Number) this.obj.get(key);
        return result == null ? null : result.doubleValue();
    }

    @Override
    public final Float optFloat(final String key) {
        Number result = (Number) this.obj.get(key);
        return result == null ? null : result.floatValue();
    }

    @Override
    public final Boolean optBool(final String key) {
        return (Boolean) this.obj.get(key);
    }

    @Override
    public final PObject optObject(final String key) {
        @SuppressWarnings("unchecked")
        final Map<String, Object> val = (Map<String, Object>) this.obj.get(key);
        return val == null ? null : new PYamlObject(this, val, key);
    }

    @Override
    public final PArray optArray(final String key) {
        @SuppressWarnings("unchecked")
        final List<Object> val = (List<Object>) this.obj.get(key);
        return val == null ? null : new PYamlArray(this, val, key);
    }

    @Override
    public final boolean isArray(final String key) {
        return this.obj.get(key) instanceof List<?>;
    }

    @Override
    public final Iterator<String> keys() {
        return this.obj.keySet().iterator();
    }

    @Override
    public final int size() {
        return this.obj.keySet().size();
    }

    @Override
    public final boolean has(final String key) {
        return this.obj.containsKey(key);
    }

    @Override
    public final String toString() {
        try {
            return "PYaml(" + this.getCurrentPath() + ":" + toJSON().getInternalObj().toString(2) + ")";
        } catch (JSONException e) {
            throw ExceptionUtils.getRuntimeException(e);
        }
    }

    /**
     * Convert this object to a json object.
     */
    public final PJsonObject toJSON() {
        try {
            JSONObject json = new JSONObject();
            for (String key : this.obj.keySet()) {
                Object opt = opt(key);
                if (opt instanceof PYamlObject) {
                    opt = ((PYamlObject) opt).toJSON().getInternalObj();
                } else if (opt instanceof PYamlArray) {
                    opt = ((PYamlArray) opt).toJSON().getInternalArray();
                }
                json.put(key, opt);
            }
            return new PJsonObject(json, this.getContextName());
        } catch (Throwable e) {
            throw ExceptionUtils.getRuntimeException(e);
        }
    }
}