package alice.jungle.codesegment;

import java.io.IOException;
import java.util.Iterator;

import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTree;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.core.Node;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.logger.DefaultTreeOperationLog;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.DefaultEither;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.DefaultError;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Either;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Error;
import jungle.test.bbs.JungleManager;
import alice.codesegment.CodeSegment;
import alice.datasegment.CommandType;
import alice.datasegment.Receiver;
import alice.jungle.datasegment.HashSetDataSegment;
import alice.jungle.datasegment.store.operations.DefaultTreeOperationLogContainer;
import alice.jungle.transaction.NetworkDefaultJungleTreeEditor;

public class LogUpdateCodeSegment extends CodeSegment {
	
	Receiver parentLog = ids.create(CommandType.PEEK);
	Receiver host = ids.create(CommandType.PEEK);
	Receiver hashLog = ids.create(CommandType.PEEK);
	
	public LogUpdateCodeSegment() {
		parentLog.setKey("parent", "log");
		host.setKey("host");
		hashLog.setKey("hashLog");
	}
	
	public LogUpdateCodeSegment(int index) {
		parentLog.setKey("parent", "log", index);
		host.setKey("host");
		hashLog.setKey("hashLog");
	}
	
	public void run() {
		System.out.println("--LogUpdateCodeSegment--");
		int index = parentLog.index;
		String h = host.asString();
		DefaultTreeOperationLogContainer container = parentLog.asClass(DefaultTreeOperationLogContainer.class);
		HashSetDataSegment ds = hashLog.asClass(HashSetDataSegment.class);
		if(ds.hash.contains(container.getHashLogString())) {
			ods.update("hashLog", ds);		
			new LogUpdateCodeSegment(index);
			return;
		}
		DefaultTreeOperationLog log = null;
		try {
			log = container.convert();
		} catch (IOException e) {
			e.printStackTrace();
		}
		String treeName = container.getTreeName();
		if (JungleManager.getJungle().getTreeByName(treeName) == null) {
			if(null == JungleManager.getJungle().createNewTree(treeName)){
				throw new IllegalStateException();
			}
		}
		JungleTree tree = JungleManager.getJungle().getTreeByName(treeName);
		JungleTreeEditor editor = tree.getTreeEditor();
		Either<DefaultError, Integer> e = checkTimeStamp(tree.getRootNode(), container.getTimeStamp());
		Either<Error, JungleTreeEditor> either; 
		if(e.isA()) {	
			either = JungleManager.edit(editor, log, container.getPosition());
		} else {
			either = JungleManager.edit(editor, log, e.b());
		}
		if(either.isA()) {
			throw new IllegalStateException();
		}
		editor = either.b();
		either = editor.success();
		if(either.isA()) {
			throw new IllegalStateException();			
		}
		ds.hash.add(container.getHashLogString());
		ods.update("hashLog", ds);		
		ods.put("log", container);
		new LogUpdateCodeSegment(index);
	}

	
	private DefaultEither<DefaultError, Integer> checkTimeStamp(Node node, long newNodeTimeStamp) {
		int count = 0;
		long childTimeStamp = 0;
		for(Iterator<Node> iter = node.getChildren().iterator();iter.hasNext();) {
			Node n = iter.next();
			if(n.getAttributes().get("timestamp") == null) {
				return DefaultEither.newA(new DefaultError());
			}
			if(n.getAttributes().get("timestamp") != null) {
				childTimeStamp = n.getAttributes().get("timestamp").getLong();
				if (newNodeTimeStamp < childTimeStamp) {
					break;
				}
			}
			count++;
		}
		return DefaultEither.newB(count);
	}
}