#ifndef __LEDGERACQUIRE__
#define __LEDGERACQUIRE__

#include <vector>
#include <map>

#include <boost/enable_shared_from_this.hpp>
#include <boost/function.hpp>
#include <boost/asio.hpp>
#include <boost/thread/mutex.hpp>

#include "Ledger.h"
#include "Peer.h"
#include "../obj/src/newcoin.pb.h"

class PeerSet
{
protected:
	uint256 mHash;
	int mTimerInterval;
	bool mComplete, mFailed;

	boost::recursive_mutex				mLock;
	boost::asio::deadline_timer			mTimer;
	std::list< boost::weak_ptr<Peer> >	mPeers;

	PeerSet(const uint256& hash, int interval);

public:
	const uint256& getHash() const		{ return mHash; }
	bool isComplete() const				{ return mComplete; }
	bool isFailed() const				{ return mFailed; }

	void peerHas(Peer::pointer);
	void badPeer(Peer::pointer);
	void resetTimer();

protected:
	virtual void newPeer(Peer::pointer) = 0;
	virtual boost::function<void (boost::system::error_code)> getTimerLamba() = 0;

	void setComplete()					{ mComplete = true; }
	void setFailed()					{ mFailed = true; }
};

class LedgerAcquire : public PeerSet, public boost::enable_shared_from_this<LedgerAcquire>
{ // A ledger we are trying to acquire
public:
	typedef boost::shared_ptr<LedgerAcquire> pointer;

protected:
	Ledger::pointer mLedger;
	bool mHaveBase, mHaveState, mHaveTransactions;

	std::vector< boost::function<void (LedgerAcquire::pointer)> > mOnComplete;

	void done();
	void onTimer() { trigger(Peer::pointer()); }

	void sendRequest(boost::shared_ptr<newcoin::TMGetLedger> message);
	void sendRequest(boost::shared_ptr<newcoin::TMGetLedger> message, Peer::pointer peer);
	void newPeer(Peer::pointer peer) { trigger(peer); }
	void trigger(Peer::pointer);
	virtual boost::function<void (boost::system::error_code)> getTimerLamba();
	static void LATimerEntry(boost::weak_ptr<LedgerAcquire>, const boost::system::error_code&);

public:
	LedgerAcquire(const uint256& hash);

	bool isBase() const					{ return mHaveBase; }
	bool isAcctStComplete() const		{ return mHaveState; }
	bool isTransComplete() const		{ return mHaveTransactions; }
	Ledger::pointer getLedger()			{ return mLedger; }

	void addOnComplete(boost::function<void (LedgerAcquire::pointer)>);

	bool takeBase(const std::string& data, Peer::pointer);
	bool takeTxNode(const std::list<SHAMapNode>& IDs, const std::list<std::vector<unsigned char> >& data,
		Peer::pointer);
	bool takeAsNode(const std::list<SHAMapNode>& IDs, const std::list<std::vector<unsigned char> >& data,
		Peer::pointer);
};

class LedgerAcquireMaster
{
protected:
	boost::mutex mLock;
	std::map<uint256, LedgerAcquire::pointer> mLedgers;

public:
	LedgerAcquireMaster() { ; }

	LedgerAcquire::pointer findCreate(const uint256& hash);
	LedgerAcquire::pointer find(const uint256& hash);
	bool hasLedger(const uint256& ledgerHash);
	bool dropLedger(const uint256& ledgerHash);
	bool gotLedgerData(newcoin::TMLedgerData& packet, Peer::pointer);
};

#endif
// vim:ts=4