org.purplejrank.mods.CollectionsJrankInput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of purplejrank Show documentation
There is a newer version: 1.1.2
package org.purplejrank.mods;

import java.io.IOException;
import java.io.InputStream;
import java.io.NotActiveException;
import java.io.ObjectInputStream;
import java.nio.channels.ReadableByteChannel;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import org.purplejrank.JrankClass;
import org.purplejrank.JrankConstants;
import org.purplejrank.JrankContext;
import org.purplejrank.JrankGetFields;
import org.purplejrank.PurpleJrankInput;

/**
 * Implementation of {@link ObjectInputStream} that deserializes all arrays
 * to {@link List} and all objects to {@link Map}
 * @author robin
 *
 */
public class CollectionsJrankInput extends PurpleJrankInput {

	public CollectionsJrankInput(ReadableByteChannel in) throws IOException {
		super(in);
	}

	public CollectionsJrankInput(ReadableByteChannel in, ClassLoader cl)
			throws IOException {
		super(in, cl);
	}

	public CollectionsJrankInput(InputStream in, ClassLoader cl)
			throws IOException {
		super(in, cl);
	}

	public CollectionsJrankInput(InputStream in) throws IOException {
		super(in);
	}

	@Override
	protected Object newArray(JrankClass desc, int size) {
		return Arrays.asList(new Object[size]);
	}
	
	@Override
	protected Object newOrdinaryObject(JrankClass desc) {
		Map ret = new TreeMap();
		ret.put("-class", desc.getName());
		ret.put("-desc", desc);
		return ret;
	}
	
	@Override
	protected Class resolveClass(String name) throws IOException, ClassNotFoundException {
		try {
			return super.resolveClass(name);
		} catch(ClassNotFoundException e) {
			return Map.class;
		}
	}
	
	@Override
	protected void skipOptionalData() throws IOException, ClassNotFoundException {
		setBlockMode(false);
		for(byte b = peek(); b != JrankConstants.J_WALL; b = peek()) {
			switch(b) {
			case JrankConstants.J_BLOCK_DATA:
				setBlockMode(true);
				setBlockMode(false);
				break;
			case JrankConstants.J_FIELDS:
				defaultReadObject();
				break;
			default:
				readObject0(true);
			}
			setBlockMode(false);
		}
	}
	
	@SuppressWarnings("unchecked")
	@Override
	public void defaultReadObject() throws IOException, ClassNotFoundException {
		JrankContext ctx = context.peekLast();
		if(ctx == JrankContext.NO_CONTEXT)
			throw new NotActiveException();
		JrankGetFields fields = readFields();
		for(String name : ctx.getType().getFieldNames()) {
			((Map) ctx.getObject()).put(name, fields.get(name, null));
		}
	}
	
	@Override
	protected void checkSerialVersion(JrankClass desc, Class cls) throws ClassNotFoundException {
	}
	
	@SuppressWarnings("unchecked")
	@Override
	protected void setArrayElement(Object array, int index, Object value) {
		((List