LiveBeansJsonParser.java

/*******************************************************************************
 *  Copyright (c) 2012 - 2013 VMware, Inc.
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  which accompanies this distribution, and is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  Contributors:
 *      VMware, Inc. - initial API and implementation
 *******************************************************************************/
package org.springframework.ide.eclipse.beans.ui.livegraph.model;

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

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

/**
 * @author Leo Dos Santos
 */
public class LiveBeansJsonParser {

	private final LiveBeansSession session;

	private final String jsonInput;

	private Map<String, LiveBean> beansMap;

	private Map<String, LiveBeansContext> contextMap;

	private Map<String, LiveBeansGroup> resourceMap;

	public LiveBeansJsonParser(LiveBeansSession session, String jsonInput) {
		this.jsonInput = jsonInput;
		this.session = session;
	}

	private void groupByResource() {
		for (LiveBean bean : beansMap.values()) {
			String resource = bean.getResource();
			if (resourceMap.containsKey(resource)) {
				LiveBeansGroup group = resourceMap.get(resource);
				group.addBean(bean);
			}
			else {
				LiveBeansGroup group = new LiveBeansResource(resource);
				group.addBean(bean);
				resourceMap.put(resource, group);
			}
		}
	}

	public LiveBeansModel parse() throws JSONException {
		beansMap = new HashMap<String, LiveBean>();
		contextMap = new HashMap<String, LiveBeansContext>();
		resourceMap = new HashMap<String, LiveBeansGroup>();

		// JSON structure is an array of context descriptions, each containing
		// an array of beans
		JSONArray contextsArray = new JSONArray(jsonInput);
		parseContexts(contextsArray);
		populateContextDependencies(contextsArray);
		groupByResource();

		LiveBeansModel model = new LiveBeansModel(session);
		model.addBeans(beansMap.values());
		model.addContexts(contextMap.values());
		model.addResources(resourceMap.values());
		return model;
	}

	private void parseBeans(LiveBeansContext context, JSONArray beansArray) throws JSONException {
		// construct LiveBeans
		for (int i = 0; i < beansArray.length(); i++) {
			JSONObject beanJson = beansArray.getJSONObject(i);
			if (beanJson != null && beanJson.has(LiveBean.ATTR_BEAN)) {
				LiveBean bean = new LiveBean(session, beanJson.getString(LiveBean.ATTR_BEAN));
				bean.addAttribute(LiveBeansContext.ATTR_CONTEXT, context.getLabel());
				if (beanJson.has(LiveBean.ATTR_SCOPE)) {
					bean.addAttribute(LiveBean.ATTR_SCOPE, beanJson.getString(LiveBean.ATTR_SCOPE));
				}
				if (beanJson.has(LiveBean.ATTR_TYPE)) {
					bean.addAttribute(LiveBean.ATTR_TYPE, beanJson.getString(LiveBean.ATTR_TYPE));
				}
				if (beanJson.has(LiveBean.ATTR_RESOURCE)) {
					bean.addAttribute(LiveBean.ATTR_RESOURCE, beanJson.getString(LiveBean.ATTR_RESOURCE));
				}
				if (session.getApplicationName() != null) {
					bean.addAttribute(LiveBean.ATTR_APPLICATION, session.getApplicationName());
				}
				context.addBean(bean);
				beansMap.put(bean.getId(), bean);
			}
		}
	}

	private void parseContexts(JSONArray contextsArray) throws JSONException {
		// construct LiveBeansContexts
		for (int i = 0; i < contextsArray.length(); i++) {
			JSONObject contextJson = contextsArray.optJSONObject(i);
			if (contextJson != null) {
				LiveBeansContext context = new LiveBeansContext(contextJson.getString(LiveBeansContext.ATTR_CONTEXT));
				JSONArray beansArray = contextJson.optJSONArray(LiveBeansContext.ATTR_BEANS);
				if (beansArray != null) {
					parseBeans(context, beansArray);
				}
				contextMap.put(context.getLabel(), context);
			}
		}
	}

	private void populateBeanDependencies(JSONArray beansArray) throws JSONException {
		// populate LiveBean dependencies
		for (int i = 0; i < beansArray.length(); i++) {
			JSONObject beanJson = beansArray.optJSONObject(i);
			if (beanJson != null && beanJson.has(LiveBean.ATTR_BEAN)) {
				LiveBean bean = beansMap.get(beanJson.getString(LiveBean.ATTR_BEAN));
				JSONArray dependencies = beanJson.optJSONArray(LiveBean.ATTR_DEPENDENCIES);
				if (dependencies != null) {
					for (int j = 0; j < dependencies.length(); j++) {
						String dependency = dependencies.getString(j);
						LiveBean dependencyBean = beansMap.get(dependency);
						if (dependencyBean != null) {
							bean.addDependency(dependencyBean);
						}
						else {
							LiveBean dependentBean = new LiveBean(session, dependency, true);
							if (session.getApplicationName() != null) {
								dependentBean.addAttribute(LiveBean.ATTR_APPLICATION, session.getApplicationName());
							}
							bean.addDependency(dependentBean);
						}
					}
				}
			}
		}
	}

	private void populateContextDependencies(JSONArray contextsArray) throws JSONException {
		// populate LiveBeanContext dependencies
		for (int i = 0; i < contextsArray.length(); i++) {
			JSONObject contextJson = contextsArray.optJSONObject(i);
			if (contextJson != null) {
				LiveBeansContext context = contextMap.get(contextJson.getString(LiveBeansContext.ATTR_CONTEXT));
				if (!contextJson.isNull(LiveBeansContext.ATTR_PARENT)) {
					String parent = contextJson.getString(LiveBeansContext.ATTR_PARENT);
					LiveBeansContext parentContext = contextMap.get(parent);
					if (parentContext != null) {
						context.setParent(parentContext);
					}
				}

				JSONArray beansArray = contextJson.optJSONArray(LiveBeansContext.ATTR_BEANS);
				if (beansArray != null) {
					populateBeanDependencies(beansArray);
				}
			}
		}
	}

}