#ifndef __SHAMAPYSNC__
#define __SHAMAPSYNC__

#include "SHAMap.h"
#include "Application.h"

// Sync filters allow low-level SHAMapSync code to interact correctly with
// higher-level structures such as caches and transaction stores

// This class is needed on both add and check functions
class ConsensusTransSetSF : public SHAMapSyncFilter
{ // sync filter for transaction sets during consensus building
public:
	ConsensusTransSetSF() { ; }
	virtual void gotNode(const SHAMapNode& id, const uint256& nodeHash,
		const std::vector<unsigned char>& nodeData, SHAMapTreeNode::TNType)
	{
		// WRITEME: If 'isLeaf' is true, this is a transaction
		theApp->getTempNodeCache().store(nodeHash, nodeData);
	}
	virtual bool haveNode(const SHAMapNode& id, const uint256& nodeHash, std::vector<unsigned char>& nodeData)
	{
		// WRITEME: We could check our own map, we could check transaction tables
		return theApp->getTempNodeCache().retrieve(nodeHash, nodeData);
	}
};

// This class is only needed on add functions
class AccountStateSF : public SHAMapSyncFilter
{ // sync filter for account state nodes during ledger sync
protected:
	uint32 mLedgerSeq;

public:
	AccountStateSF(uint32 ledgerSeq) : mLedgerSeq(ledgerSeq)
	{ ; }

	virtual void gotNode(const SHAMapNode& id, const uint256& nodeHash,
		const std::vector<unsigned char>& nodeData, SHAMapTreeNode::TNType)
	{
		theApp->getHashedObjectStore().store(hotACCOUNT_NODE, mLedgerSeq, nodeData, nodeHash);
	}
	virtual bool haveNode(const SHAMapNode& id, const uint256& nodeHash, std::vector<unsigned char>& nodeData)
	{ // fetchNodeExternal already tried
		return false;
	}
};

// This class is only needed on add functions
class TransactionStateSF : public SHAMapSyncFilter
{ // sync filter for transactions tree during ledger sync
protected:
	uint32 mLedgerSeq;

public:
	TransactionStateSF(uint32 ledgerSeq) : mLedgerSeq(ledgerSeq)
	{ ; }

	virtual void gotNode(const SHAMapNode& id, const uint256& nodeHash,
		const std::vector<unsigned char>& nodeData, SHAMapTreeNode::TNType type)
	{
		theApp->getHashedObjectStore().store(
			(type == SHAMapTreeNode::tnTRANSACTION_NM) ? hotTRANSACTION : hotTRANSACTION_NODE,
			mLedgerSeq, nodeData, nodeHash);
	}
	virtual bool haveNode(const SHAMapNode& id, const uint256& nodeHash, std::vector<unsigned char>& nodeData)
	{ // fetchNodeExternal already tried
		return false;
	}
};

#endif