MapAdapter.java

/*
 * Scriptographer
 *
 * This file is part of Scriptographer, a Scripting Plugin for Adobe Illustrator
 * http://scriptographer.org/
 *
 * Copyright (c) 2002-2010, Juerg Lehni
 * http://scratchdisk.com/
 *
 * All rights reserved. See LICENSE file for details.
 * 
 * File created on Apr 10, 2007.
 */

package com.scratchdisk.script.rhino;

import org.mozilla.javascript.Context;
import org.mozilla.javascript.NativeArray;
import org.mozilla.javascript.Scriptable;
import org.mozilla.javascript.ScriptableObject;
import org.mozilla.javascript.Wrapper;

import com.scratchdisk.util.AbstractMap;

/**
 * MapAdapter wraps a Rhino ScriptableObject instance in a Map interface. All
 * methods are implemented, even entrySet() / keySet() This is the opposite of
 * {@link MapWrapper}.
 * 
 * This class was made redundant by Rhino's support for the java.util.Map
 * interface in ScriptableObject and my extensions to it, ported over
 * from MapAdapter. I still keep it around though.
 * 
 * @author lehni
 */
public class MapAdapter extends AbstractMap {
	Scriptable object;
	
	public MapAdapter(Scriptable object) {
		this.object = object;
	}

	protected Object[] keys() {
		return object.getIds();
	}

	public Object get(Object key) {
		Object value;
		if (key instanceof Integer)
			value = ScriptableObject.getProperty(object, ((Integer) key).intValue());
		else if (key instanceof String)
			value = ScriptableObject.getProperty(object, (String) key);
		else
			value = null;
		if (value instanceof Wrapper)
			value = ((Wrapper) value).unwrap();
		else if (value == ScriptableObject.NOT_FOUND)
			value = null;
		else if (value instanceof NativeArray) {
			// Convert to a normal array
			// TODO: see if we need to convert the other way in put?
			NativeArray array = (NativeArray) value;
			int length = (int) array.getLength();
			Object[] list = new Object[length];
			for (int i = 0; i < length; i++) {
				Object obj = array.get(i, array);
				if (obj instanceof Wrapper)
					obj = ((Wrapper) obj).unwrap();
				list[i] = obj;
			}
			return list;
		}
		return value;
	}

	public Object put(Object key, Object value) {
		// Wrap the value if it is not already
		if (value != null && !(value instanceof Scriptable)) {
			Context cx = Context.getCurrentContext();
			value = cx.getWrapFactory().wrap(cx, object, value, value.getClass());
		}
		Object prev = get(key);
		if (key instanceof Integer)
			object.put(((Integer) key).intValue(), object, value);
		else if (key instanceof String)
			object.put((String) key, object, value);
		else
			prev = null;
		return prev;
	}

	public Object remove(Object key) {
		if (containsKey(key)) {
			Object prev = get(key);
			if (key instanceof Integer)
				object.delete(((Integer) key).intValue());
			else if (key instanceof String)
				object.delete((String) key);
			return prev;
		}
		return null;
	}

	public boolean containsKey(Object key) {
		if (key instanceof Integer)
			return object.has(((Integer) key).intValue(), object);
		else if (key instanceof String)
			return object.has((String) key, object);
		else
			return false;
	}
}