ResultMessage.java

package com.mygeopay.stratumj.messages;

import org.json.JSONArray;
import org.json.JSONException;

/**
 * @author John L. Jegutanis
 */
public class ResultMessage extends BaseMessage {
    protected ResultMessage(String source) throws JSONException {
        super(source);
    }

    public static ResultMessage fromJson(String json) throws JSONException {
        return new ResultMessage(json);
    }

    public JSONArray getResult() {
        if (has("result")) {
            if (opt("result") instanceof JSONArray) {
                try {
                    return getJSONArray("result");
                } catch (JSONException e) {
                    // Should not happen
                    throw new RuntimeException(e);
                }
            } else {
                JSONArray result = new JSONArray();
                try {
                    result.put(get("result"));
                } catch (JSONException e) {
                    // Should not happen
                    throw new RuntimeException(e);
                }
                return result;
            }
        } else {
            return new JSONArray();
        }
    }
}