package jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction;

import java.nio.ByteBuffer;

import fj.data.List;
import fj.data.TreeMap;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.TreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.TreeNode;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.logger.DefaultTreeOperationLog;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.logger.LoggingNode;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.logger.OperationLog;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.logger.TreeOperationLog;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.operations.DefaultTreeOperation;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.operations.NodeOperation;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.operations.TreeOperation;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.trasnformer.AppendChildAt;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.trasnformer.DeleteAttribute;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.trasnformer.DeleteChildAt;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.trasnformer.NodeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.trasnformer.PutAttribute;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.DefaultEither;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Either;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Error;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.IterableConverter;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Pair;
import jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.store.index.AppendChildAtIndex;
import jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.store.index.DeleteAttributeIndex;
import jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.store.index.DeleteChildAtIndex;
import jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.store.index.Logging;
import jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.store.index.PutAttributeIndex;

public class DefaultJungleIndexTreeEditor extends DefaultJungleTreeEditor
{
	private final TransactionManager txManager;
	private final TreeNode root;
	private final TreeEditor editor;
	private final TreeOperationLog log;
	private TreeMap<String,List<TreeNode>> index;
	
	public DefaultJungleIndexTreeEditor(TreeNode _root,TransactionManager _txManager,TreeEditor _editor, TreeMap<String,List<TreeNode>> index)
	{
		this(_root,_txManager,_editor,new DefaultTreeOperationLog(), index);
	}
	
	public DefaultJungleIndexTreeEditor(TreeNode newNode,TransactionManager _txManager,TreeEditor _editor,TreeOperationLog _log, TreeMap<String,List<TreeNode>> index)
	{
		super(newNode, _txManager, _editor , _log);
		this.root = newNode;
		this.txManager = _txManager;
		this.editor = _editor;
		this.log = _log;
		this.index = index;
	}
	
	private Either<Error,JungleTreeEditor> _edit(final NodePath _path,NodeEditor _e)
	{
		//LoggingNodeHook hook = new LoggingNodeHook(_e);
		Either<Error,Logging> either = editor.edit(root,_path,_e);
		if(either.isA()){
			return DefaultEither.newA(either.a());
		}
		
		Logging newLogging = either.b();
		OperationLog newLog = newLogging.getOperationLog();
		TreeNode newNode = newLogging.getWrap();
		index = newLogging.getIndex();
		IterableConverter.Converter<TreeOperation,NodeOperation> converter = new IterableConverter.Converter<TreeOperation,NodeOperation>(){
			@Override
			public TreeOperation conv(NodeOperation _b){
				return new DefaultTreeOperation(_path,_b);
			}
		};
		
		Iterable<TreeOperation> iterable = new IterableConverter<TreeOperation,NodeOperation>(newLog,converter);
		DefaultTreeOperationLog treeOperationLog = new DefaultTreeOperationLog(iterable,newLog.length());
		TreeOperationLog newTreeOpLog = log.append(treeOperationLog);
		
		JungleTreeEditor newEditor = new DefaultJungleIndexTreeEditor(newNode,txManager,editor,newTreeOpLog, index);
		return DefaultEither.newB(newEditor);
	}
	
	@Override
	public Either<Error,JungleTreeEditor> addNewChildAt(NodePath _path, int _pos)
	{
		AppendChildAtIndex appendChildAt = new AppendChildAtIndex(_pos, index);
		return _edit(_path,appendChildAt);
	}

	@Override
	public Either<Error,JungleTreeEditor> deleteChildAt(NodePath _path, int _pos)
	{
		DeleteChildAtIndex deleteChildAt = new DeleteChildAtIndex(_pos, index);
		return _edit(_path,deleteChildAt);
	}

	@Override
	public Either<Error,JungleTreeEditor> putAttribute(NodePath _path,String _key,ByteBuffer _value)
	{
		PutAttributeIndex putAttribute = new PutAttributeIndex(_key,_value, index);
		return _edit(_path,putAttribute);
	}

	@Override
	public Either<Error,JungleTreeEditor> deleteAttribute(NodePath _path, String _key)
	{
		DeleteAttributeIndex deleteAttribute = new DeleteAttributeIndex(_key, index);
		return _edit(_path,deleteAttribute);
	}

	@Override
	public Either<Error,JungleTreeEditor> edit(NodePath _path,NodeEditor _editor)
	{
		return _edit(_path,_editor);
	}

	@Override
	public Either<Error,JungleTreeEditor> success()
	{
		Either<Error,TransactionManager> either = txManager.commit(root,log, index);
		if(either.isA()){
			return DefaultEither.newA(either.a());
		}
		
		TransactionManager newTxManager = either.b();
		JungleTreeEditor newTreeEditor = new DefaultJungleIndexTreeEditor(root,newTxManager,editor, index);
		
		return DefaultEither.newB(newTreeEditor);
	}

}