package rep.handler;

import java.io.IOException;
import java.util.LinkedList;
import java.util.List;

import rep.PacketSet;
import rep.REP;
import rep.REPCommand;
import rep.ServerMainLoop;
import rep.SessionManager;
import rep.channel.REPSelectionKey;
import rep.channel.REPSocketChannel;
import rep.optimizers.*;

public class Editor extends Forwarder {

	private Translator translator;
	// REPCommands we are going to send to the next editor
	private List<REPCommand> sentList = new LinkedList<REPCommand>();
	protected List<PacketSet> waitingCommandInMerge= new LinkedList<PacketSet>();
	private REPCommand quit2=null;
	private boolean merging;
	private REPCommand preMergeCommand;
	public static boolean noMergeMode=false;
	static final boolean doOptimize = true;

	public Editor(SessionManager manager,int editorNo){
		// no translator case
		super(manager, null);
	}

	public Editor(int editorNo, SessionManager manager,REPSocketChannel<REPCommand> channel){
		super(editorNo,manager,channel);
		eid = editorNo;
		REPCommandOptimizer optimizer;
		if (doOptimize) optimizer = new DeleteInsertOptimizer(); //タカノがつくったおぷてぃまいざ
		else            optimizer = new NullOptimizer();         //なにもしないけどOptimizer.
		translator = new Translator(eid,optimizer);
	}
	

	public void translate(REPCommand command){				
		switch(command.cmd) {
		case REPCMD_INSERT_ACK:
		case REPCMD_DELETE_ACK:
			if (command.eid==eid) {
				// Second Phase が終わって同期が終了。
				return;
			}
			if (waitingRequired(command)) return;
			checkReturnedCommand(command);
			return;
		case REPCMD_INSERT_USER:
			command.cmd = REP.REPCMD_INSERT;
			userEditorCommand(command);
			return;
		case REPCMD_DELETE_USER:
			command.cmd = REP.REPCMD_INSERT;
			userEditorCommand(command);
			return;
		case REPCMD_INSERT:
		case REPCMD_DELETE:
			if (command.eid == REP.MERGE_EID.id){
				//マージコマンドが返ってきた
				checkDouble(sentList);
				if(translator.checkMergeConflict(command)){
					//マージ中にエディタからの割り込みがあった場合
					translator.getMergeAgain(this);
				}
				checkEndMerge();
				checkDouble(sentList);
				return;
			} else if (command.eid == eid){ 
				// 編集コマンドが一周して来た
				if (waitingRequired(command)) return;
				checkReturnedCommand(command);
				checkDouble(sentList);
				return;
			}

			//他のエディタからの編集コマンド
			if (waitingRequired(command)) return;
			translator.transReceiveCmd(next,command);
			if(command.cmd==REP.REPCMD_DELETE) {
				// delete のundo用の文字列は、外に出す意味はない
				command.string=null;
			}
			sendEditorCommand(command);
			return;
		default:
			assert(false);
		}
	}

	private void userEditorCommand(REPCommand command) {
		//エディタからの新たな編集コマンド
		if (next==this) return; // singleton case
		translator.transSendCmd(command);
		sendEditorCommand(command);
		checkDouble(sentList);
		return;
	}
	
	private void checkDouble(List<REPCommand> sentList) {
		if (sentList.size()==0) return;
		int count = 0;
		REPCommand f = sentList.get(0); 
		for(REPCommand c:sentList) {
			if (c.eid==f.eid&&c.seq==f.seq) {
				count++;
			}
		}
		assert(count==1);
		if (true) return;
		count = 0;
		for(PacketSet c:waitingCommandInMerge) {
			for(REPCommand g:sentList) { 
				if (c.command.eid==g.eid&&c.command.seq==g.seq) {
					count++;
				}
			}
		}
		assert(count==0);
	}

	private boolean waitingRequired(REPCommand command) {
		if (hasWaitingCommand()) {
			// We cannot do this operation before watingCommandQueue.
			addWaitingCommand(new PacketSet(channel, this, new REPCommand(command)));
			checkDouble(sentList);
			return true;
		} else if (isMerging()) { 
			addWaitingCommand(new PacketSet(getChannel(), this, new REPCommand(command)));
			checkDouble(sentList);
			return true;
		} 
		checkDouble(sentList);
		ServerMainLoop.logger.writeLog("Editor eid:"+eid+" no waiting");
		return false;
	}

	public void addWaitingCommand(PacketSet set) {
		if (preMergeCommand!=null) {
			if (preMergeCommand.eid==set.command.eid
					&& preMergeCommand.seq==set.command.seq) {
				assert(false);
			}
		}
		waitingCommandInMerge.add(set);
	}
	
	private void sendEditorCommand(REPCommand command) {
		REPCommand keep = new REPCommand(command);
		sentList.add(keep);
		ServerMainLoop.logger.writeLog("Editor eid:"+eid+" sentList = "+sentList);
		checkDouble(sentList);
		assert(sentList.size()<limit);
		next.send(command);
	}

	boolean merge(REPCommand command) {
		//マージして送信
		return translator.catchOwnCommand(this);
	}

	/**
	 * 一周して来たcommandの処理。
	 * @param command
	 */
	void checkReturnedCommand(REPCommand command) {
		assert(!merging);
		if (sentList.size()==0) {
			ServerMainLoop.logger.writeLog("Editor eid="+eid+" looped command not registered: "+command);
			assert(false);
		}
		REPCommand prev = sentList.remove(0);
		ServerMainLoop.logger.writeLog("Editor eid="+eid+" remove sentList:"+(prev==null?"null":prev));
		if (prev==null || prev.seq != command.seq || prev.eid!=command.eid) {
			String err = "Editor eid="+eid+" checkReturnedCommand() : command = " + command + " prev="+
				(prev==null?"null":prev)+" sentList=";
			err += sentList;
			ServerMainLoop.logger.writeLog(err);
			assert(false);
		}
		startMerge(command);
		return;
	}

	private void startMerge(REPCommand command) {
		preMergeCommand = new REPCommand(command);
		preMergeCommand.string = "";
		checkDouble(sentList);
		// merge は必須だが、EditorのCommand実装をテストするには邪魔なので、off に出来るようにする。
		if (noMergeMode) {
			endMerge();
			return;
		}
		// START_MERGE を送る
		//    送らないで良い場合もある?
		REPCommand cmd = new REPCommand(REP.SMCMD_START_MERGE,command.sid,REP.SM_EID.id,seq(),0,"");
		send(cmd);
		merging = true;
		// Session Manager 側で、このeditorへの他のeditorからの
		// 入力を止めて、merge にそなえる。merge は、eidtor 側から
		// ACKが来てから始まる。
		checkDouble(sentList);
		translator.startMerge(cmd);
	}

	@Override
	public void setQuit2(REPCommand cmd) {
		quit2 = cmd;
		checkQuit();
		// do not send quit2 until we received all pending
		// command
	}

	@Override
	public void setEID(int eid) {
		this.eid = eid;
		if (translator!=null)
			translator.setEid(eid);
	}
	
	public String toString(){
		return ("Editor eid="+eid+" sid="+sid+" " + host  + ":" + file);
	}

	public boolean isMerging() {
		return translator.isMerging();
	}

	

	void checkEndMerge() {
		if (merging) {
			if(translator.isMerging()) return;
			endMerge();
			merging = false;
		}
		if (quit2!=null) checkQuit();
	}

	private void endMerge() {
		REPCommand mergeEnd = new REPCommand(REP.SMCMD_END_MERGE,sid,eid,seq(),0,"");
		send(mergeEnd);
		if (preMergeCommand.eid==eid) {
			// First Phase End, send ACK
			REPCommand keep = new REPCommand(preMergeCommand);
			switch(keep.cmd) {
			case REPCMD_INSERT: keep.cmd = REP.REPCMD_INSERT_ACK;break;
			case REPCMD_DELETE: keep.cmd = REP.REPCMD_DELETE_ACK;break;
			}
			sentList.add(keep);
			ServerMainLoop.logger.writeLog("Editor eid:"+eid+" sentList = "+sentList);
			assert(sentList.size()<limit);
			checkDouble(sentList);
			next.send(keep);
		} else {
			next.send(preMergeCommand);
		}
		preMergeCommand = null;
	}

	private boolean checkQuit() {
		if (sentList.size()==0&&!isMerging()) {
			send(quit2);
			quit2 = null;
			return true;
		}
		return false;
	}

	@Override
	public boolean manage(REPCommand command) {
		
		
		switch(command.cmd){
		// Editor Command

		case REPCMD_DELETE:
		case REPCMD_INSERT:
		case REPCMD_DELETE_USER:
		case REPCMD_INSERT_USER:
		case REPCMD_DELETE_ACK:
		case REPCMD_INSERT_ACK:
		{
			translate(command);
			break;
		}

		case SMCMD_START_MERGE_ACK:
		{
			// マージの処理と次のエディタへコマンドを送信する処理
			translator.mergeAck();
			if (!merge(command)) {
				// nothing to do, send END_MERGE
				checkEndMerge();
			}
			break;
		}
		
		case SMCMD_SYNC:
			if (isMaster()) 
				send(command);
			else
				next.send(command);
			
		case SMCMD_QUIT:
		{
			next.send(command);
			break;
		}
		case SMCMD_QUIT_2:
		{
			// QUIT_2 is returned.
			if (command.eid!=eid) {
				// stop this editor unless this is the start, starter will stopped
				// by QUIT_2_ACK
				manager.remove(this);
			}
			// don't send quit_2 directly to the editor until all pending
			// merge is processed.
			//   this does not work in distributed case.
			if (next.isDirect())
				next.setQuit2(command);
			else
				next.send(command);
			break;
		}
		case SMCMD_QUIT_2_ACK:
		{
			manager.remove(this);
			break;
		}
		default:
			assert false;
			return false;
		}
		return true;
	}
	

	@Override
	public void handle(REPCommand command, REPSelectionKey<REPCommand> key) throws IOException {
		ServerMainLoop.logger.writeLog("Manager "+manager.getId()+" read : command = " + command 
				+" from "+manager.editorList.editorByChannel(channel));
		if (command.cmd==REP.SMCMD_JOIN||command.cmd==REP.SMCMD_PUT) {
			assert false;
			// 若干問題があるらしい
			next = new Forwarder(manager,next.channel);
			REPNode first = new FirstConnector(manager,channel);
			first.handle(command, key);
			key.attach(new Dispatcher(manager,channel));
			return;
		}
		if (manager.sessionManage(this, command)) return;
		checkDouble(sentList);
		manage(command);
		checkDouble(sentList);
	}

	@Override
	public void cancel(REPSocketChannel<REPCommand> socketChannel) {
		manager.remove(socketChannel);
	}

	public boolean isMaster() {
		return mode==REP.SMCMD_PUT;
	}

	
	/* Handle special case first, usually these cases
	 * are handled in the next Editor in a session manager, but 
	 * it is forwarded here.
	 */
	public void forwardedCommandManage(REPCommand command) {
		if (command.cmd==REP.SMCMD_QUIT_2) {
			// we have to wait next editor's finishing before sending this.
			// this is odd, but the editor itself does not know it's merging
			// state. Only this session manager knows it.
			setQuit2(command);
			return;
		} 
		send(command);
	}

	/**
	 * Check waiting command in merge
	 * @return true if there is a processed waiting command
	 * @throws IOException
	 */
	public void checkWaitingCommandInMerge() {
		int count = waitingCommandInMerge.size();
		if (count==0) return;
		if (isMerging()) return;
		while(count-->0) {
			PacketSet p = waitingCommandInMerge.remove(0);
			try {
				//					if (manager.sessionManage(e, p.command)) { // we don't need this
				//						assert false;
				//						return;
				//					}
				manage(p.command);
			} catch (Exception e1) {
				// should be e.close()?
				manager.close(p.channel);
			}
		}
	}


	public boolean hasWaitingCommand() {
		return waitingCommandInMerge.size()>0;
	}
}