#include "TransactionMeta.h"

#include <algorithm>

#include <boost/make_shared.hpp>
#include <boost/bind.hpp>
#include <boost/foreach.hpp>

#include "Log.h"
#include "SerializedObject.h"

SETUP_LOG();

TransactionMetaSet::TransactionMetaSet(const uint256& txid, uint32 ledger, const std::vector<unsigned char>& vec) :
	mTransactionID(txid), mLedger(ledger), mNodes(sfAffectedNodes, 32)
{
	Serializer s(vec);
	SerializerIterator sit(s);

	std::auto_ptr<SerializedType> pobj = STObject::deserialize(sit, sfAffectedNodes);
	STObject *obj = static_cast<STObject*>(pobj.get());
	if (!obj)
		throw std::runtime_error("bad metadata");

	mResult = obj->getFieldU8(sfTransactionResult);
	mIndex = obj->getFieldU32(sfTransactionIndex);
	mNodes = * dynamic_cast<STArray*>(&obj->getField(sfAffectedNodes));
}

bool TransactionMetaSet::isNodeAffected(const uint256& node) const
{
	BOOST_FOREACH(const STObject& it, mNodes)
		if (it.getFieldH256(sfLedgerIndex) == node)
			return true;
	return false;
}

void TransactionMetaSet::setAffectedNode(const uint256& node, SField::ref type, uint16 nodeType)
{ // make sure the node exists and force its type
	BOOST_FOREACH(STObject& it, mNodes)
	{
		if (it.getFieldH256(sfLedgerIndex) == node)
		{
			it.setFName(type);
			it.setFieldU16(sfLedgerEntryType, nodeType);
			return;
		}
	}

	mNodes.push_back(STObject(type));
	STObject& obj = mNodes.back();

	assert(obj.getFName() == type);
	obj.setFieldH256(sfLedgerIndex, node);
	obj.setFieldU16(sfLedgerEntryType, nodeType);
}

static void addIfUnique(std::vector<RippleAddress>& vector, const RippleAddress& address)
{
	BOOST_FOREACH(const RippleAddress& a, vector)
		if (a == address)
			return;
	vector.push_back(address);
}

std::vector<RippleAddress> TransactionMetaSet::getAffectedAccounts()
{
	std::vector<RippleAddress> accounts;
	accounts.reserve(10);

	BOOST_FOREACH(const STObject& it, mNodes)
	{
		int index = it.getFieldIndex((it.getFName() == sfCreatedNode) ? sfNewFields : sfFinalFields);
		if (index != -1)
		{
			const STObject *inner = dynamic_cast<const STObject*>(&it.peekAtIndex(index));
			if (inner)
			{
				BOOST_FOREACH(const SerializedType& field, inner->peekData())
				{
					const STAccount* sa = dynamic_cast<const STAccount*>(&field);
					if (sa)
						addIfUnique(accounts, sa->getValueNCA());
					else if ((field.getFName() == sfLowLimit) || (field.getFName() == sfHighLimit))
					{
						const STAmount* lim = dynamic_cast<const STAmount*>(&field);
						if (lim != NULL)
						{
							uint160 issuer = lim->getIssuer();
							if (issuer.isNonZero())
							{
								RippleAddress na;
								na.setAccountID(issuer);
								addIfUnique(accounts, na);
							}
						}
						else
						{
							cLog(lsFATAL) << "limit is not amount " << field.getJson(0);
						}
					}
				}
			}
			else assert(false);
		}
	}

	return accounts;
}

STObject& TransactionMetaSet::getAffectedNode(SLE::ref node, SField::ref type)
{
	assert(&type);
	uint256 index = node->getIndex();
	BOOST_FOREACH(STObject& it, mNodes)
	{
		if (it.getFieldH256(sfLedgerIndex) == index)
			return it;
	}
	mNodes.push_back(STObject(type));
	STObject& obj = mNodes.back();

	assert(obj.getFName() == type);
	obj.setFieldH256(sfLedgerIndex, index);
	obj.setFieldU16(sfLedgerEntryType, node->getFieldU16(sfLedgerEntryType));

	return obj;
}

STObject& TransactionMetaSet::getAffectedNode(const uint256& node)
{
	BOOST_FOREACH(STObject& it, mNodes)
	{
		if (it.getFieldH256(sfLedgerIndex) == node)
			return it;
	}
	assert(false);
	throw std::runtime_error("Affected node not found");
}

const STObject& TransactionMetaSet::peekAffectedNode(const uint256& node) const
{
	BOOST_FOREACH(const STObject& it, mNodes)
		if (it.getFieldH256(sfLedgerIndex) == node)
			return it;
	throw std::runtime_error("Affected node not found");
}

void TransactionMetaSet::init(const uint256& id, uint32 ledger)
{
	mTransactionID = id;
	mLedger = ledger;
	mNodes = STArray(sfAffectedNodes, 32);
}

void TransactionMetaSet::swap(TransactionMetaSet& s)
{
	assert((mTransactionID == s.mTransactionID) && (mLedger == s.mLedger));
	mNodes.swap(s.mNodes);
}

bool TransactionMetaSet::thread(STObject& node, const uint256& prevTxID, uint32 prevLgrID)
{
	if (node.getFieldIndex(sfPreviousTxnID) == -1)
	{
		assert(node.getFieldIndex(sfPreviousTxnLgrSeq) == -1);
		node.setFieldH256(sfPreviousTxnID, prevTxID);
		node.setFieldU32(sfPreviousTxnLgrSeq, prevLgrID);
		return true;
	}
	assert(node.getFieldH256(sfPreviousTxnID) == prevTxID);
	assert(node.getFieldU32(sfPreviousTxnLgrSeq) == prevLgrID);
	return false;
}

static bool compare(const STObject& o1, const STObject& o2)
{
	return o1.getFieldH256(sfLedgerIndex) < o2.getFieldH256(sfLedgerIndex);
}

STObject TransactionMetaSet::getAsObject() const
{
	STObject metaData(sfTransactionMetaData);
	assert(mResult != 255);
	metaData.setFieldU8(sfTransactionResult, mResult);
	metaData.setFieldU32(sfTransactionIndex, mIndex);
	metaData.addObject(mNodes);
	return metaData;
}

void TransactionMetaSet::addRaw(Serializer& s, TER result, uint32 index)
{
	mResult = static_cast<int>(result);
	mIndex = index;
	assert((mResult == 0) || ((mResult > 100) && (mResult <= 255)));

	mNodes.sort(compare);

	getAsObject().add(s);
}

// vim:ts=4