/*
 * Copyright (c) 2012 Mateusz Piękos
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

#include <Swiften/Whiteboard/WhiteboardClient.h>
#include <Swiften/Whiteboard/WhiteboardTransformer.h>
#include <boost/smart_ptr/make_shared.hpp>

namespace Swift {
	WhiteboardOperation::ref WhiteboardClient::handleLocalOperationReceived(WhiteboardOperation::ref operation) {
		localOperations_.push_back(operation);
//		if (bridge_.size() > 0) {
		WhiteboardOperation::ref op = boost::make_shared<WhiteboardInsertOperation>(*boost::dynamic_pointer_cast<WhiteboardInsertOperation>(operation).get());
		if (bridge_.size() > 0) {
			op->setParentID(bridge_.back()->getID());
		}
		bridge_.push_back(op);
//		}
		if (lastSentOperationID_.empty())
		{
			WhiteboardOperation::ref op = boost::make_shared<WhiteboardInsertOperation>(*boost::dynamic_pointer_cast<WhiteboardInsertOperation>(operation).get());
			if (serverOperations_.size() > 0) {
				op->setParentID(serverOperations_.back()->getID());
			}
			lastSentOperationID_ = operation->getID();
			return op;
		} else {
			return WhiteboardOperation::ref();
		}
	}

	std::pair<WhiteboardOperation::ref, WhiteboardOperation::ref> WhiteboardClient::handleServerOperationReceived(WhiteboardOperation::ref operation) {
		serverOperations_.push_back(operation);
		WhiteboardOperation::ref clientOp;
		WhiteboardOperation::ref serverOp;
//		if (localOperations_.empty()) {// || localOperations_.back()->getID() == operation->getParentID()) {
		//Situation where client and server are in sync
		if (localOperations_.size() == serverOperations_.size()-1) {
			localOperations_.push_back(operation);
			clientOp = operation;
		} else if (lastSentOperationID_ == operation->getID()) {
			//Client received confirmation about own operation and it sends next operation to server
			if (bridge_.size() > 0 && lastSentOperationID_ == bridge_.front()->getID()) {
				bridge_.erase(bridge_.begin());
			}
			/*std::list<WhiteboardOperation::ref>::iterator it;
			for (it = bridge_.begin(); it != bridge_.end(); ++it) {
				if ((*it)->getParentID() == lastSentOperationID_) {
					lastSentOperationID_ = (*it)->getID();
					serverOp = *it;
					serverOp->setOrigin(WhiteboardOperation::Other);
					break;
				}
				}*/

			if (bridge_.size() > 0 && (bridge_.front())->getParentID() == lastSentOperationID_) {
				lastSentOperationID_ = (bridge_.front())->getID();
				serverOp = bridge_.front();
				serverOp->setOrigin(WhiteboardOperation::Other);
			}

			if (!serverOp) {
				lastSentOperationID_.clear();
			}
		} else {
			if (bridge_.size() > 0 && bridge_.front()->getParentID() == operation->getParentID()) {
				std::list<WhiteboardOperation::ref>::iterator it = bridge_.begin();
				std::pair<WhiteboardInsertOperation::ref, WhiteboardInsertOperation::ref> opPair;
				WhiteboardInsertOperation::ref temp;
				opPair = WhiteboardTransformer::transform(boost::dynamic_pointer_cast<WhiteboardInsertOperation>(*it), boost::dynamic_pointer_cast<WhiteboardInsertOperation>(operation));
				temp = opPair.first;

				*it = opPair.second;
				std::string previousID = (*it)->getID();
				++it;
				for (; it != bridge_.end(); ++it) {
					opPair = WhiteboardTransformer::transform(boost::dynamic_pointer_cast<WhiteboardInsertOperation>(*it), temp);
					temp = opPair.first;
					*it = opPair.second;
					(*it)->setParentID(previousID);
					previousID = (*it)->getID();
				}

				temp->setParentID(localOperations_.back()->getID());
				localOperations_.push_back(temp);
				clientOp = temp;
			} else {
//doesn't get executed
				std::list<WhiteboardOperation::ref>::reverse_iterator it;
				std::pair<WhiteboardInsertOperation::ref, WhiteboardInsertOperation::ref> opPair;
				WhiteboardInsertOperation::ref temp = boost::dynamic_pointer_cast<WhiteboardInsertOperation>(operation);
				bool end = false;
				for (it = localOperations_.rbegin(); it != localOperations_.rend(); ++it) {
					while ((*it)->getParentID() == temp->getParentID()) {
						opPair = WhiteboardTransformer::transform(boost::dynamic_pointer_cast<WhiteboardInsertOperation>(*it), temp);
						if (bridge_.size() > 0) {
							opPair.second->setParentID(bridge_.back()->getID());
						}
						temp = opPair.first;
						bridge_.push_back(opPair.second);
						if (it == localOperations_.rbegin()) {
							localOperations_.push_back(temp);
							clientOp = temp;
							end = true;
							break;
						} else {
							--it;
						}
					}
					if (end) {
						break;
					}
				}
			}
		}

		return std::pair<WhiteboardOperation::ref, WhiteboardOperation::ref>(clientOp, serverOp);
	}

	void WhiteboardClient::print() {
		std::list<WhiteboardOperation::ref>::iterator it;
		std::cout << "Client" << std::endl;
		for(it = localOperations_.begin(); it != localOperations_.end(); ++it) {
			std::cout << (*it)->getID() << " " << (*it)->getPos() << std::endl;
		}

		std::cout << "Server" << std::endl;
		for(it = serverOperations_.begin(); it != serverOperations_.end(); ++it) {
			std::cout << (*it)->getID() << " " << (*it)->getPos() << std::endl;
		}
	}
}