package appeng.me;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.HashSet;
import java.util.Set;

import net.minecraft.nbt.CompressedStreamTools;
import net.minecraft.nbt.NBTTagCompound;

import org.bouncycastle.util.encoders.Base64;

import appeng.api.networking.IGrid;
import appeng.api.networking.IGridStorage;
import appeng.core.WorldSettings;

public class GridStorage implements IGridStorage
{

	IGrid myGrid = null;

	final long myID;
	final NBTTagCompound data;

	public boolean isDirty = false;
	private Set<GridStorage> divlist = new HashSet();
	final GridStorageSearch mySearchEntry; // keep myself in the list until I'm lost...

	/**
	 * for use with world settings
	 * 
	 * @param id
	 * @param gss
	 */
	public GridStorage(long id, GridStorageSearch gss) {
		myID = id;
		mySearchEntry = gss;
		data = new NBTTagCompound();
	}

	/**
	 * for use with world settings
	 * 
	 * @param input
	 * @param id
	 * @param gss
	 */
	public GridStorage(String input, long id, GridStorageSearch gss) {
		myID = id;
		mySearchEntry = gss;
		NBTTagCompound myTag = null;

		try
		{
			byte[] dbata = Base64.decode( input );
			myTag = CompressedStreamTools.readCompressed( new ByteArrayInputStream( dbata ) );
		}
		catch (Throwable t)
		{
			myTag = new NBTTagCompound();
		}

		data = myTag;
	}

	/**
	 * fake storage.
	 */
	public GridStorage() {
		myID = 0;
		mySearchEntry = null;
		data = new NBTTagCompound();
	}

	public String getValue()
	{
		isDirty = false;

		if ( myGrid != null )
		{
			((Grid) myGrid).saveState();
		}

		try
		{
			ByteArrayOutputStream out = new ByteArrayOutputStream();
			CompressedStreamTools.writeCompressed( data, out );
			return new String( Base64.encode( out.toByteArray() ), "UTF-8" );
		}
		catch (IOException e)
		{
			e.printStackTrace();
		}

		return "";
	}

	@Override
	public NBTTagCompound dataObject()
	{
		return data;
	}

	@Override
	public long getID()
	{
		return myID;
	}

	public void markDirty()
	{
		isDirty = true;
	}

	public IGrid getGrid()
	{
		return myGrid;
	}

	public void setGrid(Grid grid)
	{
		myGrid = grid;
	}

	public void addDivided(GridStorage gs)
	{
		divlist.add( gs );
	}

	public boolean hasDivided(GridStorage myStorage)
	{
		return divlist.contains( myStorage );
	}

	public void remove()
	{
		WorldSettings.getInstance().destroyGridStorage( getID() );
	}

}