io.inugami.commons.transformer.NashornTransformer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of inugami_commons Show documentation
There is a newer version: 3.3.5
/* --------------------------------------------------------------------
 *  Inugami
 * --------------------------------------------------------------------
 *
 * This program 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, version 3.
 *
 * 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 General Public License
 * along with this program. If not, see .
 */
package io.inugami.commons.transformer;

import flexjson.JSONContext;
import flexjson.transformer.AbstractTransformer;
import io.inugami.api.models.JsonBuilder;
import jdk.nashorn.api.scripting.ScriptObjectMirror;

import java.util.Iterator;
import java.util.Map.Entry;

/**
 * OptionalTransformer
 *
 * @author patrick_guillerm
 * @since 7 févr. 2018
 */
@SuppressWarnings({"java:S5738", "java:S5738"})
public class NashornTransformer extends AbstractTransformer {

    // =========================================================================
    // METHODS
    // =========================================================================
    @Override
    public void transform(final Object object) {
        if (object == null) {
            getContext().write(JsonBuilder.VALUE_NULL);
        } else {
            if (object instanceof ScriptObjectMirror) {
                processTransform(object);
            } else {
                getContext().transform(object);
            }
        }
    }

    private void processTransform(final Object object) {
        final ScriptObjectMirror scriptObject = (ScriptObjectMirror) object;

        if (scriptObject.isArray()) {
            processList(scriptObject);
        } else {
            processObject(scriptObject);
        }

    }

    private void processList(final ScriptObjectMirror scriptObject) {
        final JSONContext json = getContext();
        json.writeOpenArray();

        boolean enableComma = false;
        for (final Entry item : scriptObject.entrySet()) {
            if (enableComma) {
                json.writeComma();
            }

            final Object value = item.getValue();
            if (value == null) {
                json.write(JsonBuilder.VALUE_NULL);
            } else {
                json.transform(value);
            }
            enableComma = true;
        }

        json.writeCloseArray();

    }

    private void processObject(final ScriptObjectMirror scriptObject) {
        final JSONContext json = getContext();
        json.writeOpenObject();
        final Iterator> iterator = scriptObject.entrySet().iterator();
        while (iterator.hasNext()) {
            final Entry item = iterator.next();
            json.writeName(item.getKey());

            final Object value = item.getValue();
            if (value == null) {
                json.write(JsonBuilder.VALUE_NULL);
            } else if (isScriptObjectArray(value)) {
                processList((ScriptObjectMirror) value);
            } else {
                json.transform(item.getValue());
            }

            if (iterator.hasNext()) {
                json.writeComma();
            }
        }
        json.writeCloseObject();
    }

    private boolean isScriptObjectArray(final Object value) {
        return value instanceof ScriptObjectMirror && ((ScriptObjectMirror) value).isArray();
    }
}