TypeClass.java

/**
 * (c) 2011, Alejandro Serrano
 * Released under the terms of the EPL.
 */
package net.sf.eclipsefp.haskell.browser.items;

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

/**
 * Represents a "class" declaration.
 * 
 * @author Alejandro Serrano
 */
public class TypeClass extends Declaration {
	String[] context;
	String[] vars;
	String[] fundeps;

	public TypeClass(String doc, String[] context, String name, String[] vars,
			String[] fundeps) {
		this.setDoc(doc);
		this.setType(DeclarationType.TYPE_CLASS);
		this.setInfo(context, name, vars, fundeps);
	}

	public TypeClass(JSONObject o) throws JSONException {
		this.setDoc(o);
		this.setType(DeclarationType.TYPE_CLASS);
		this.setInfo(o);
	}

	protected void setInfo(String[] context, String name, String[] vars,
			String[] fundeps) {
		this.context = context;
		this.setName(name);
		this.vars = vars;
		this.fundeps = fundeps;
	}

	protected void setInfo(JSONObject o) throws JSONException {
		this.setDoc(o);

		JSONObject head = o.getJSONObject("head");
		this.setInfo(Util.getStringArray(o.getJSONArray("context")),
				head.getString("name"),
				Util.getStringArray(head.getJSONArray("vars")),
				Util.getStringArray(o.getJSONArray("fundeps")));
	}

	public String[] getContext() {
		return this.context;
	}

	public String[] getTypeVariables() {
		return this.vars;
	}

	public String[] getFunctionalDependencies() {
		return this.fundeps;
	}
	
	@Override
	public String getCompleteDefinition() {
		StringBuilder builder = new StringBuilder("class");
		builder.append(' ');
		if (this.getContext().length > 1) {
			builder.append('(');
			builder.append(this.getContext()[0]);
			for (int i = 1; i < this.getContext().length; i++) {
				builder.append(", ");
				builder.append(this.getContext()[i]);
			}
			builder.append(") ");
		} else if (this.getContext().length == 1) {
			builder.append(this.getContext()[0]);
			builder.append(' ');
		}
		if (this.getContext().length > 0)
			builder.append("=> ");
		builder.append(this.getName());
		for (String tvar : this.getTypeVariables()) {
			builder.append(' ');
			builder.append(tvar);
		}
		String[] fdeps = this.getFunctionalDependencies();
		if (fdeps.length > 0) {
			builder.append(" | ");
			for (int i = 0; i < fdeps.length; i++) {
				if (i != 0)
					builder.append(", ");
				builder.append(fdeps[i]);
			}
		}
		return builder.toString();
	}
	
	@Override
	public String getShownName() {
		StringBuilder name = new StringBuilder(this.name);
		for (String var : this.vars) {
			name.append(' ');
			name.append(var);
		}
		return name.toString();
	}
}