vaadin-framework/src/com/itmill/toolkit/terminal/gwt/client/UIDL.java

/* 
@ITMillApache2LicenseForJavaFiles@
 */

package com.itmill.toolkit.terminal.gwt.client;

import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import com.google.gwt.json.client.JSONArray;
import com.google.gwt.json.client.JSONBoolean;
import com.google.gwt.json.client.JSONNumber;
import com.google.gwt.json.client.JSONObject;
import com.google.gwt.json.client.JSONString;
import com.google.gwt.json.client.JSONValue;
import com.google.gwt.user.client.DOM;
import com.google.gwt.user.client.ui.Tree;
import com.google.gwt.user.client.ui.TreeItem;
import com.google.gwt.user.client.ui.TreeListener;

public class UIDL {

    JSONArray json;

    public UIDL(JSONArray json) {
        this.json = json;
    }

    public String getId() {
        final JSONValue val = ((JSONObject) json.get(1)).get("id");
        if (val == null) {
            return null;
        }
        return ((JSONString) val).stringValue();
    }

    public String getTag() {
        return ((JSONString) json.get(0)).stringValue();
    }

    public String getStringAttribute(String name) {
        final JSONValue val = ((JSONObject) json.get(1)).get(name);
        if (val == null) {
            return null;
        }
        return ((JSONString) val).stringValue();
    }

    public Set getAttributeNames() {
        final HashSet attrs = new HashSet(((JSONObject) json.get(1)).keySet());
        attrs.remove("v");
        return attrs;
    }

    public int getIntAttribute(String name) {
        final JSONValue val = ((JSONObject) json.get(1)).get(name);
        if (val == null) {
            return 0;
        }
        final double num = ((JSONNumber) val).getValue();
        return (int) num;
    }

    public long getLongAttribute(String name) {
        final JSONValue val = ((JSONObject) json.get(1)).get(name);
        if (val == null) {
            return 0;
        }
        final double num = ((JSONNumber) val).getValue();
        return (long) num;
    }

    public float getFloatAttribute(String name) {
        final JSONValue val = ((JSONObject) json.get(1)).get(name);
        if (val == null) {
            return 0;
        }
        final double num = ((JSONNumber) val).getValue();
        return (float) num;
    }

    public double getDoubleAttribute(String name) {
        final JSONValue val = ((JSONObject) json.get(1)).get(name);
        if (val == null) {
            return 0;
        }
        final double num = ((JSONNumber) val).getValue();
        return num;
    }

    public boolean getBooleanAttribute(String name) {
        final JSONValue val = ((JSONObject) json.get(1)).get(name);
        if (val == null) {
            return false;
        }
        return ((JSONBoolean) val).booleanValue();
    }

    public String[] getStringArrayAttribute(String name) {
        final JSONArray a = (JSONArray) ((JSONObject) json.get(1)).get(name);
        final String[] s = new String[a.size()];
        for (int i = 0; i < a.size(); i++) {
            s[i] = ((JSONString) a.get(i)).stringValue();
        }
        return s;
    }

    public int[] getIntArrayAttribute(String name) {
        final JSONArray a = (JSONArray) ((JSONObject) json.get(1)).get(name);
        final int[] s = new int[a.size()];
        for (int i = 0; i < a.size(); i++) {
            s[i] = Integer.parseInt(((JSONString) a.get(i)).stringValue());
        }
        return s;
    }

    public HashSet getStringArrayAttributeAsSet(String string) {
        final JSONArray a = getArrayVariable(string);
        final HashSet s = new HashSet();
        for (int i = 0; i < a.size(); i++) {
            s.add(((JSONString) a.get(i)).stringValue());
        }
        return s;
    }

    /**
     * Get attributes value as string whateever the type is
     * 
     * @param name
     * @return string presentation of attribute
     */
    private String getAttribute(String name) {
        return json.get(1).isObject().get(name).toString();
    }

    public boolean hasAttribute(String name) {
        return ((JSONObject) json.get(1)).get(name) != null;
    }

    public UIDL getChildUIDL(int i) {

        final JSONValue c = json.get(i + 2);
        if (c == null) {
            return null;
        }
        if (c.isArray() != null) {
            return new UIDL(c.isArray());
        }
        throw new IllegalStateException("Child node " + i
                + " is not of type UIDL");
    }

    public String getChildString(int i) {

        final JSONValue c = json.get(i + 2);
        if (c.isString() != null) {
            return ((JSONString) c).stringValue();
        }
        throw new IllegalStateException("Child node " + i
                + " is not of type String");
    }

    public Iterator getChildIterator() {

        return new Iterator() {

            int index = 2;

            public void remove() {
                throw new UnsupportedOperationException();
            }

            public Object next() {

                if (json.size() > index) {
                    final JSONValue c = json.get(index++);
                    if (c.isString() != null) {
                        return c.isString().stringValue();
                    } else if (c.isArray() != null) {
                        return new UIDL(c.isArray());
                    } else if (c.isObject() != null) {
                        return new XML(c.isObject());
                    } else {
                        throw new IllegalStateException("Illegal child " + c
                                + " in tag " + getTag() + " at index " + index);
                    }
                }
                return null;
            }

            public boolean hasNext() {
                return json.size() > index;
            }

        };
    }

    public int getNumberOfChildren() {
        return json.size() - 2;
    }

    public String toString() {
        String s = "<" + getTag();

        for (final Iterator i = getAttributeNames().iterator(); i.hasNext();) {
            final String name = i.next().toString();
            s += " " + name + "=";
            final JSONValue v = ((JSONObject) json.get(1)).get(name);
            if (v.isString() != null) {
                s += v;
            } else {
                s += "\"" + v + "\"";
            }
        }

        s += ">\n";

        final Iterator i = getChildIterator();
        while (i.hasNext()) {
            final Object c = i.next();
            s += c.toString();
        }

        s += "</" + getTag() + ">\n";

        return s;
    }

    public String getChildrenAsXML() {
        String s = "";
        final Iterator i = getChildIterator();
        while (i.hasNext()) {
            final Object c = i.next();
            s += c.toString();
        }
        return s;
    }

    public UIDLBrowser print_r() {
        return new UIDLBrowser();
    }

    private class UIDLBrowser extends Tree {
        public UIDLBrowser() {

            DOM.setStyleAttribute(getElement(), "position", "");

            final TreeItem root = new TreeItem(getTag());
            addItem(root);
            root.addItem("");
            addTreeListener(new TreeListener() {

                public void onTreeItemStateChanged(TreeItem item) {
                    if (item == root) {
                        removeItem(root);
                        UIDLBrowser.this.removeTreeListener(this);
                        addItem(dir());
                        final Iterator it = treeItemIterator();
                        while (it.hasNext()) {
                            ((TreeItem) it.next()).setState(true);
                        }
                    }
                }

                public void onTreeItemSelected(TreeItem item) {
                }

            });

        }

        protected boolean isKeyboardNavigationEnabled(TreeItem currentItem) {
            return false;
        }

    }

    public TreeItem dir() {

        String nodeName = getTag();
        for (final Iterator i = getAttributeNames().iterator(); i.hasNext();) {
            final String name = i.next().toString();
            final String value = getAttribute(name);
            nodeName += " " + name + "=" + value;
        }
        final TreeItem item = new TreeItem(nodeName);

        try {
            TreeItem tmp = null;
            for (final Iterator i = getVariableHash().keySet().iterator(); i
                    .hasNext();) {
                final String name = i.next().toString();
                String value = "";
                try {
                    value = getStringVariable(name);
                } catch (final Exception e) {
                    try {
                        final JSONArray a = getArrayVariable(name);
                        value = a.toString();
                    } catch (final Exception e2) {
                        try {
                            final int intVal = getIntVariable(name);
                            value = String.valueOf(intVal);
                        } catch (final Exception e3) {
                            value = "unknown";
                        }
                    }
                }
                if (tmp == null) {
                    tmp = new TreeItem("variables");
                }
                tmp.addItem(name + "=" + value);
            }
            if (tmp != null) {
                item.addItem(tmp);
            }
        } catch (final Exception e) {
            // Ingonered, no variables
        }

        final Iterator i = getChildIterator();
        while (i.hasNext()) {
            final Object child = i.next();
            try {
                final UIDL c = (UIDL) child;
                item.addItem(c.dir());

            } catch (final Exception e) {
                item.addItem(child.toString());
            }
        }
        return item;
    }

    private JSONObject getVariableHash() {
        final JSONObject v = (JSONObject) ((JSONObject) json.get(1)).get("v");
        if (v == null) {
            throw new IllegalArgumentException("No variables defined in tag.");
        }
        return v;
    }

    public boolean hasVariable(String name) {
        Object v = null;
        try {
            v = getVariableHash().get(name);
        } catch (final IllegalArgumentException e) {
        }
        return v != null;
    }

    public String getStringVariable(String name) {
        final JSONString t = (JSONString) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return t.stringValue();
    }

    public int getIntVariable(String name) {
        final JSONNumber t = (JSONNumber) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return (int) t.getValue();
    }

    public long getLongVariable(String name) {
        final JSONNumber t = (JSONNumber) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return (long) t.getValue();
    }

    public float getFloatVariable(String name) {
        final JSONNumber t = (JSONNumber) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return (float) t.getValue();
    }

    public double getDoubleVariable(String name) {
        final JSONNumber t = (JSONNumber) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return t.getValue();
    }

    public boolean getBooleanVariable(String name) {
        final JSONBoolean t = (JSONBoolean) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return t.booleanValue();
    }

    private JSONArray getArrayVariable(String name) {
        final JSONArray t = (JSONArray) getVariableHash().get(name);
        if (t == null) {
            throw new IllegalArgumentException("No such variable: " + name);
        }
        return t;
    }

    public String[] getStringArrayVariable(String name) {
        final JSONArray a = getArrayVariable(name);
        final String[] s = new String[a.size()];
        for (int i = 0; i < a.size(); i++) {
            s[i] = ((JSONString) a.get(i)).stringValue();
        }
        return s;
    }

    public Set getStringArrayVariableAsSet(String name) {
        final JSONArray a = getArrayVariable(name);
        final HashSet s = new HashSet();
        for (int i = 0; i < a.size(); i++) {
            s.add(((JSONString) a.get(i)).stringValue());
        }
        return s;
    }

    public int[] getIntArrayVariable(String name) {
        final JSONArray a = getArrayVariable(name);
        final int[] s = new int[a.size()];
        for (int i = 0; i < a.size(); i++) {
            final JSONValue v = a.get(i);
            s[i] = v.isNumber() != null ? (int) ((JSONNumber) v).getValue()
                    : Integer.parseInt(v.toString());
        }
        return s;
    }

    public class XML {
        JSONObject x;

        private XML(JSONObject x) {
            this.x = x;
        }

        public String getXMLAsString() {
            final StringBuffer sb = new StringBuffer();
            for (final Iterator it = x.keySet().iterator(); it.hasNext();) {
                final String tag = (String) it.next();
                sb.append("<");
                sb.append(tag);
                sb.append(">");
                sb.append(x.get(tag).isString().stringValue());
                sb.append("</");
                sb.append(tag);
                sb.append(">");
            }
            return sb.toString();
        }
    }

    public int getChildCount() {
        return json.size() - 2;
    }

    public UIDL getErrors() {
        final JSONArray a = (JSONArray) ((JSONObject) json.get(1)).get("error");
        return new UIDL(a);
    }

}