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

import java.util.concurrent.atomic.AtomicReference;

import org.omg.CORBA.TCKind;

import fj.data.List;
import fj.data.TreeMap;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.persistent.ChangeList;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.persistent.ChangeListWriter;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.ChangeSet;
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.operations.TreeOperation;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.AtomicReservableReference;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.DefaultJungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.DefaultTransactionManager;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.IndexJungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.TreeContext;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.traverser.InterfaceTraverser;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Pair;


public class DefaultJungleTree implements JungleTree
{
	private final AtomicReservableReference<TreeContext> repository;
	private final String uuid;
	private final ChangeListWriter writer;
	private final TreeEditor editor;
	
	public DefaultJungleTree(TreeContext tc,String uuid,ChangeListWriter writer,TreeEditor editor)
	{
		this.repository = new AtomicReservableReference<TreeContext>(tc);
		this.uuid = uuid;
		this.writer = writer;
		this.editor = editor;
	}

	@Override
	public JungleTreeEditor getTreeEditor()
	{
		TreeContext tc = repository.get();
		DefaultTransactionManager txManager = new DefaultTransactionManager(writer,tc,repository,uuid);
		TreeNode root = tc.getTreeNode();
		return new DefaultJungleTreeEditor(root,txManager,editor);
	}
	
	@Override
	public IndexJungleTreeEditor getIndexTreeEditor()
	{
		TreeContext tc = repository.get();
		DefaultTransactionManager txManager = new DefaultTransactionManager(writer,tc,repository,uuid);
		TreeNode root = tc.getTreeNode();
		TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> newIndex = getIndex();
		return new IndexJungleTreeEditor(root,txManager,editor,newIndex);
	}
	

	@Override
	public JungleTreeEditor getLocalTreeEditor()
	{
		return getTreeEditor();
	}
	
	@Override
	public TreeNode getRootNode()
	{
		TreeContext tc = repository.get();
		ChangeSet cs = tc.getChangeSet();
		return cs.getRoot();
	}

	@Override
	public InterfaceTraverser getTraverser() {
		AtomicReservableReference<TreeContext>.Reservation reservation = repository.getReservation();
		TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> index = getIndex();
		if (index != null)
			return new InterfaceTraverser(getRootNode(), index, reservation);
		return new InterfaceTraverser(getRootNode(),reservation);
	}
	
	@Override
	public TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> getIndex() {
		TreeContext tc = repository.get();
		ChangeSet cs = tc.getChangeSet();
		return cs.getIndex();
	}
	
	@Override
	public int getLogSize(){
		TreeContext tc = repository.get();
		ChangeSet cs = tc.getChangeSet();
		ChangeList cl = cs.getChangeList();
		return cl.getLogSize();
	}

	@Override
	public Iterable<TreeOperation> getLog() {
		TreeContext tc = repository.get();
		ChangeSet cs = tc.getChangeSet();
		return cs.getOperations();
	}
	
}