com.venky.swf.db.model.io.json.JSONFormatter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swf-db-io-json Show documentation
Succinct Web Framework JSON plugin
There is a newer version: 2.11
package com.venky.swf.db.model.io.json;
import java.io.IOException;
import java.io.StringWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import com.venky.core.string.StringUtil;
import com.venky.swf.db.Database;
import com.venky.swf.db.JdbcTypeHelper;
import org.json.simple.JSONArray;
import org.json.simple.JSONAware;
import org.json.simple.JSONObject;

/**
 * Created by venky on 11/5/16.
 */
public class JSONFormatter  {
	public JSONFormatter(){
		this(2);
	}
    public JSONFormatter(int spacing){
    	SPACES=spacing;
    }
    public String toString(JSONAware jsonAware){
		if (jsonAware instanceof JSONArray){
			return toString((JSONArray) jsonAware);
		}else if (jsonAware instanceof JSONObject){
			return toString((JSONObject) jsonAware);
		}else {
			return jsonAware.toString();
		}
	}
    public String toString(JSONArray jsonArray){
		StringWriter writer = new StringWriter();
		writePrettyJsonArray(jsonArray,writer);
		return writer.toString();
	}
	public String toString(JSONObject jsonObject){
		StringWriter writer = new StringWriter();
		writePrettyJson(jsonObject,writer);
		return writer.toString();
	}
    public void writePrettyJsonArray(JSONArray jsonArray, Writer w) {
		try {
			w.append("[");
			for (int i = 0 ; i < jsonArray.size() ; i ++ ){
				Object o = jsonArray.get(i);
				if (i > 0){
					w.append(",");
				}
				writeObject(o, w);
			}
			w.append("]");
		}catch (IOException ex){
			throw new RuntimeException(ex);
		}
    }
    public void writePrettyJson(JSONObject obj,Writer w) {
		try {
			w.append("{");
			writeAttributes(obj, w);
			w.append("}");
		}catch (IOException ex){
			throw new RuntimeException(ex);
		}
    }
    private int indent = 0;
    private int SPACES = 2;
    private void indent(Writer w) throws IOException{
    	indent += SPACES;
    	for (int i = 0 ; i< indent ; i ++){
			w.append(" ");
		}
    }
    private void backIndent(Writer w) throws IOException {
    	indent -=SPACES;
    }
    private void writeObject(Object o ,Writer w) throws IOException{
    	if (o instanceof JSONObject){
    		writePrettyJson((JSONObject)o, w);
    	}else if (o instanceof JSONArray){
    		writePrettyJsonArray((JSONArray)o, w);
    	}else {
    		w.append('"').append(JSONObject.escape(o.toString())).append("\"");
    	}
    	
    }
    private static Set nonStrings = new HashSet(){{
    	add(Boolean.class);
    	add(Integer.class);
    	add(Long.class);
    	add(Double.class);
		add(Short.class);
		add(Float.class);
	}};
	private void writeAttributes(JSONObject obj, Writer w) throws IOException {
		boolean first = true;
		List keys = new ArrayList();
		keys.addAll(obj.keySet());
		Collections.sort(keys);
		for (Object k : keys){
			Object v = obj.get(k);
            w.append("\n");
            indent(w);
            if (!first) {
                w.append(",");
            }else {
                first=false;
            }
			w.append("\"").append(JSONObject.escape(k.toString())).append("\" : ");
			if (v instanceof JSONObject){
				writePrettyJson((JSONObject)v, w);
			}else if (v instanceof JSONArray) {
				writePrettyJsonArray((JSONArray)v, w);
			}else if (v != null){
				if (v.getClass().isPrimitive() || nonStrings.contains(v.getClass())){
					w.append(StringUtil.valueOf(v));
				}else {
					w.append('"').append(JSONObject.escape(StringUtil.valueOf(v))).append('"');
				}
			}else {
				w.append("null");
			}
			backIndent(w);
		}
        w.append("\n");
        backIndent(w);
        indent(w);
	}
    
    
}