// Copyright (c) 1998-2009 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of "Eclipse Public License v1.0"
// which accompanies this distribution, and is available
// at the URL "http://www.eclipse.org/legal/epl-v10.html".
//
// Initial Contributors:
// Nokia Corporation - initial contribution.
//
// Contributors:
//
// Description:
//

#include "UT_STD.H"
#include <s32mem.h>

#define UNUSED_VAR(a) a = a

const TInt KStreamDictGranularity=16;

NONSHARABLE_CLASS(HDictionaryStoreBuf) : public MStreamBuf
	{
public:
	HDictionaryStoreBuf(MStreamBuf& aBuf,CDictionaryStore& aDic,TStreamId aStreamId,TUid aUid);
private:
	// from MStreamBuf
	void DoSynchL(); // only update the stream dic during synch
	// just delegate the rest straight through...
	void DoRelease();
	TInt DoReadL(TAny* aPtr,TInt aMaxLength);
	TInt DoReadL(TDes8& aDes,TInt aMaxLength,TRequestStatus& aStatus);
	TStreamTransfer DoReadL(MStreamInput& anInput,TStreamTransfer aTransfer);
	void DoWriteL(const TAny* aPtr,TInt aLength);
	TInt DoWriteL(const TDesC8& aDes,TInt aMaxLength,TRequestStatus& aStatus);
	TStreamTransfer DoWriteL(MStreamOutput& anOutput,TStreamTransfer aTransfer);
	TStreamPos DoSeekL(TMark aMark,TStreamLocation aLocation,TInt anOffset);
private:
	MStreamBuf* iRealBuf; // delegate everything to this
	CDictionaryStore* iDictionary;
	TStreamId iStreamId; // id of the actual stream in the store
	TUid iStreamUid; // Uid to be paired with the stream id in the dic
	};

NONSHARABLE_CLASS(HNullBuf) : public TMemBuf
	{
public:
	HNullBuf();
private:
	void DoRelease();
	};

////////////////////////////////////////
// CStreamDictionary
////////////////////////////////////////

EXPORT_C CStreamDictionary* CStreamDictionary::NewL()
/** Allocates, constructs and returns a pointer to a new stream dictionary.

The function leaves if it cannot complete successfully.

@return A pointer to the new stream dictionary object. */
	{
	return new(ELeave) CStreamDictionary;
	}

EXPORT_C CStreamDictionary* CStreamDictionary::NewLC()
/** Allocates, constructs and returns a pointer to a new stream dictionary, putting 
the pointer onto the cleanup stack.

Putting the pointer onto the cleanup stack allows the object and allocated 
resources to be cleaned up if a subsequent leave occurs.

The function leaves if it cannot complete successfully.

@return A pointer to the new stream dictionary object. */
	{
	CStreamDictionary* dict=NewL();
	CleanupStack::PushL(dict);
	return dict;
	}

EXPORT_C CStreamDictionary::CStreamDictionary()
	: iArray(KStreamDictGranularity)
	{}

EXPORT_C CStreamDictionary::~CStreamDictionary()
/** Frees resources owned by the object, prior to its destruction. */
	{}

EXPORT_C void CStreamDictionary::AssignL(TUid aUid,TStreamId anId)
/** Creates or changes an association between a UID and a stream id.

If the stream dictionary already contains an entry with the same UID as aUid, 
then the associated stream id in that entry is replaced by anId.

If anId has the value KNullStreamId, then the function attempts to remove 
the entry whose unique identifier matches aUid.

@param aUid The UID
@param anId The stream id.
@see CStreamDictionary::Remove() */
	{
	if (anId==KNullStreamId)
		{
		Remove(aUid); // default associated stream is null
		return;
		}
//
	TEntry entry;
	entry.iUid=aUid;
	TKeyArrayFix key(_FOFF(TEntry,iUid),ECmpTUint32);
	TInt i;
	if (iArray.Find(entry,key,i)==0)
		{
		iArray[i].iId=anId;
		return;
		}
//
	__ASSERT_DEBUG(i==iArray.Count(),User::Invariant());
	entry.iId=anId;
	iArray.InsertL(i,entry);
	}

EXPORT_C void CStreamDictionary::Remove(TUid aUid)
/** Removes an association from the stream dictionary.

The function searches the stream dictionary for an entry whose UID matches 
aUid. If a match is found, the entry is removed. If no match is found, the 
stream dictionary remains unchanged.

@param aUid The UID whose matching entry is to be removed from the stream 
dictionary. */
	{
	TEntry entry;
	entry.iUid=aUid;
	TKeyArrayFix key(_FOFF(TEntry,iUid),ECmpTUint32);
	TInt i;
	if (iArray.Find(entry,key,i)==0)
		iArray.Delete(i);
	}

EXPORT_C TStreamId CStreamDictionary::At(TUid aUid) const
/** Returns the stream id associated with the specified UID.

@param aUid The UID whose associated stream id is to be returned.
@return The stream id associated with the specified UID.KNullStreamId, if no 
entry in the stream dictionary contains the specified UID. */
	{
	TEntry entry;
	entry.iUid=aUid;
	TKeyArrayFix key(_FOFF(TEntry,iUid),ECmpTUint32);
	TInt i;
	if (iArray.Find(entry,key,i)!=0)
		return KNullStreamId;
//
	return iArray[i].iId;
	}

EXPORT_C TBool CStreamDictionary::IsNull() const
/** Tests whether the stream dictionary is empty.

@return True, if the stream dictionary is empty. False if the stream dictionary 
contains at least one entry. */
	{
	return iArray.Count()==0;
	}

EXPORT_C void CStreamDictionary::ExternalizeL(RWriteStream& aStream) const
/** Externalises an object of this class to a write stream.

The presence of this function means that the standard templated operator<<() 
can be used to externalise objects of this class.

@param aStream Stream to which the object should be externalised */
	{
	aStream<<iArray;
	}

EXPORT_C void CStreamDictionary::InternalizeL(RReadStream& aStream)
/** Internalises an object of this class from a read stream.

The presence of this function means that the standard templated operator>>() 
can be used to internalise objects of this class.

Note that this function has assignment semantics. It replaces the old value 
of the object with a new value read from the read stream.

@param aStream Stream from which the object should be internalised. */
	{
	aStream>>iArray;
	}

void CStreamDictionary::TEntry::ExternalizeL(RWriteStream& aStream) const
	{
	aStream<<iUid;
	aStream<<iId;
	}

void CStreamDictionary::TEntry::InternalizeL(RReadStream& aStream)
	{
	aStream>>iUid;
	aStream>>iId;
	}


////////////////////////////////////////
// CDictionaryStore
////////////////////////////////////////

EXPORT_C CDictionaryStore::~CDictionaryStore()
/** Frees resources owned by the object, prior to its destruction. */
	{
	delete iDictionary;
	delete iStore;
	}


EXPORT_C void CDictionaryStore::ConstructL()
	{
	iDictionary = CStreamDictionary::NewL();
	RStoreWriteStream stream;
	iStore->SetRootL(stream.CreateLC(*iStore));
	stream<<*iDictionary;
	stream.CommitL();
	CleanupStack::PopAndDestroy(); // dictionary stream
	iStore->CommitL();
	}


EXPORT_C void CDictionaryStore::Remove(TUid aUid)
/** Removes the stream associated with the specified UID from the dictionary store.

If there is no stream associated with the specified UID, then the dictionary 
store remains unchanged.

The function cannot leave; it returns whether or not it is succesful.

@param aUid The UID whose associated stream is to be removed from the dictionary 
store. */
	{
	TRAPD(ignore,RemoveL(aUid));
    UNUSED_VAR(ignore);
	}


EXPORT_C void CDictionaryStore::RemoveL(TUid aUid)
/** Removes the stream associated with the specified UID from the dictionary store 
and leaves if unsuccessful.

If there is no stream associated with the specified UID, then the dictionary 
store remains unchanged.

@param aUid The UID whose associated stream is to be removed from the dictionary 
store. */
	{
	TStreamId id = DictionaryL()->At(aUid);
	if (id!=KNullStreamId)
		{
		DictionaryL()->Remove(aUid);
		iStore->DeleteL(id);
		iDictionaryHasChanged=ETrue;
		}
	}


EXPORT_C TInt CDictionaryStore::Commit()
/** Commits changes.

It establishes a new commit point and then compacts the dictionary store. 
Typically, this is done after changes to new or existing streams are complete 
and the streams themselves have been committed.

Establishing a new commit point makes changes to the store permanent. Until 
such changes are committed, they can be rolled back, effectively causing the 
store to revert back to its state before the changes were made.

This ensures that persistent data moves from one consistent state to another 
and guarantees the integrity of persistent store data in the event of failures. 
In particular, if a process terminates or a media failure occurs, the store 
reverts automatically to its state at the last successful commit point.

@return KErrNone if successful, otherwise another of the system-wide error 
codes. */
	{
	TRAPD(ret,CommitL());
	return ret;
	}


EXPORT_C void CDictionaryStore::CommitL()
/** Commits changes and leaves if unsuccessful.

@see CDictionaryStore::Commit() */
	{
	if (iDictionaryHasChanged)
		{
		// rewrite the root stream
		RStoreWriteStream stream;
		stream.ReplaceLC(*iStore,iStore->Root());
		stream<< *DictionaryL();
		stream.CommitL();
		CleanupStack::PopAndDestroy(); // dictionary stream
		}
	// commit the store
	iStore->CommitL();
	// compact the store, ignoring any failure (commit was good)
	TRAPD(
		ignore,
			{
			if (iStore->ReclaimL() >= KDictionaryCommitThreshold)
				{
				iStore->CompactL();
				}
			iStore->CommitL();
			} );
    UNUSED_VAR(ignore);
	}


EXPORT_C void CDictionaryStore::Revert()
/** Rolls the dictionary store back to its state at the last commit point.

A commit point is set using the Commit() or CommitL() functions.

The function returns, whether or not it completes successfully.

@see CDictionaryStore::Commit()
@see CDictionaryStore::CommitL() */
	{
	TRAPD(ignore,RevertL());
    UNUSED_VAR(ignore);
	}


EXPORT_C void CDictionaryStore::RevertL()
/** Rolls the dictionary store back to its state at the last commit point and leaves 
if unsuccessful.

A commit point is set using the Commit() or CommitL() functions.

@see CDictionaryStore::Commit()
@see CDictionaryStore::CommitL() */
	{
	delete iDictionary;
	iDictionary = NULL;
	iDictionaryHasChanged = EFalse;
	iStore->RevertL();
	}


CStreamDictionary* CDictionaryStore::DictionaryL() const
// Returns the handle of the stream dictionary, internalizing it first if necessary
	{
	if (!iDictionary)
		{
		// create a temporary dictionary
		CStreamDictionary* dictionary = CStreamDictionary::NewLC();
		// stream into the dictionary
		RStoreReadStream root;
		root.OpenLC(*iStore,iStore->Root());
		root>> *dictionary;
		root.Close();
		CleanupStack::PopAndDestroy(); // root
		// set iDictionary
		MUTABLE_CAST(CStreamDictionary*&,iDictionary) = dictionary;
		CleanupStack::Pop(); // dictionary
		}
	return iDictionary;
	}


EXPORT_C TBool CDictionaryStore::IsNullL() const
/** Tests whether the dictionary stores stream dictionary is empty.

@return True, if the stream dictionary is empty. False, if the stream dictionary 
contains at least one entry.
@see CStreamDictionary */
	{
	return DictionaryL()->IsNull();
	}


EXPORT_C TBool CDictionaryStore::IsPresentL(TUid aUid) const
/** Tests whether the specified UID has an associated stream within this dictionary 
store.

@param aUid The UID.
@return True, if there is a stream associated with the specified UID in the 
dictionary store; false otherwise.
@see CStreamDictionary */
	{
	return DictionaryL()->At(aUid)!=KNullStreamId;
	}


MStreamBuf* CDictionaryStore::GetSourceL(TUid aUid) const
// Opens the stream matching aUid and returns its source.
// If no stream matches aUid an empty stream buf is returned
// Ownership of the source is transfered to the caller.
	{
	// get the id of the stream
	TStreamId id = DictionaryL()->At(aUid);
	// open the stream if it exists, otherwise open a dummy empty stream
	if (id==KNullStreamId)
		return (new(ELeave) HNullBuf);
	else
		{
		// open the stream and steal its source
		RStoreReadStream stream;
		stream.OpenL(*iStore,id);
		return stream.Source();
		}
	}


MStreamBuf* CDictionaryStore::GetSinkL(TUid aUid)
// If a stream matches aUid it is replaced ad its sink returned 
// If no stream matches aUid a new stream is created and its sink returned
// Ownership of the sink is transfered to the caller.
	{
	// get the id of the stream
	TStreamId id = DictionaryL()->At(aUid);
	RStoreWriteStream stream;
	if (id!=KNullStreamId)
		//replace the stream
		stream.ReplaceLC(*iStore,id);
	else
		// create a new stream
		id = stream.CreateLC(*iStore);
	// create a dictionary store buffer to hold the incoming data
	MStreamBuf* buf = new(ELeave) HDictionaryStoreBuf(*stream.Sink(),*this,id,aUid);
	CleanupStack::Pop(); // stream;
	return buf;
	}


//////////////////////////////////////////////
// RDictionaryReadStream
//////////////////////////////////////////////

EXPORT_C void RDictionaryReadStream::OpenL(const CDictionaryStore& aDictStore,TUid aUid)
// Open the stream if it exists, otherwise open a dummy empty stream
/** Opens the stream associated with the specified UID in the specified dictionary 
store, and prepares it for reading.

@param aDictStore The dictionary store containing the stream.
@param aUid The unique identifier associated with the stream to be read. */
	{
	Attach(aDictStore.GetSourceL(aUid));
	}


EXPORT_C void RDictionaryReadStream::OpenLC(const CDictionaryStore& aDictStore,TUid aUid)
/** Opens the stream associated with the specified UID in the specified dictionary 
store, prepares it for reading, and puts a a cleanup item onto the cleanup 
stack..

Placing a cleanup item on the cleanup stack allows allocated resources to 
be cleaned up if a subsequent leave occurs.

@param aDictStore The dictionary store containing the stream.
@param aUid The unique identifier associated with the stream to be read. */
	{
	OpenL(aDictStore,aUid);
	PushL();
	}


//////////////////////////////////////////////
// RDictionaryWriteStream
//////////////////////////////////////////////

EXPORT_C void RDictionaryWriteStream::AssignL(CDictionaryStore& aDictStore,TUid aUid)
// Replace the stream if it exists, otherwise create a new one
/** Prepares a stream in the specified dictionary store for writing.

If no stream is associated with the specified UID, then a new stream is created 
and an association is made between the resulting stream id and the specified 
UID. If a stream is currently associated with the specified UID, then this 
existing stream is prepared for replacement.

@param aDictStore The dictionary store which contains the new or replacement 
stream.
@param aUid The UID associated with the stream.
@see RWriteStream::Release() */
	{
	Attach(aDictStore.GetSinkL(aUid));
	}


EXPORT_C void RDictionaryWriteStream::AssignLC(CDictionaryStore& aDictStore,TUid aUid)
/** Prepares a stream in the specified dictionary store for writing, and places 
a cleanup item for this RDictionaryWriteStream object onto the cleanup stack. 

If no stream is associated with the specified UID, then a new stream is created 
and an association is made between the resulting stream id and the specified 
UID. If a stream is currently associated with the specified UID, then this 
existing stream is prepared for replacement.

Placing a cleanup item onto the cleanup stack allows allocated resources to 
be cleaned up if a subsequent leave occurs. 

@param aDictStore The dictionary store which contains the new or replacement 
stream.
@param aUid The UID associated with the stream. */
	{
	AssignL(aDictStore,aUid);
	PushL();
	}


//////////////////////////////////////////////
// HDictionaryStoreBuf
//////////////////////////////////////////////

HDictionaryStoreBuf::HDictionaryStoreBuf(MStreamBuf& aBuf,CDictionaryStore& aDic,TStreamId aStreamId,TUid aUid)
	:iRealBuf(&aBuf),
	iDictionary(&aDic),
	iStreamId(aStreamId),
	iStreamUid(aUid)
	{}


void HDictionaryStoreBuf::DoSynchL()
// only update the stream dic during synch - this is when the stream gets added to the store
	{
	CStreamDictionary& dic = *iDictionary->DictionaryL();
	TBool newEntry = (dic.At(iStreamUid)==KNullStreamId);
	if (iRealBuf->SizeL()>0)
		{
		dic.AssignL(iStreamUid,iStreamId);
		TInt ret = iRealBuf->Synch();
		if (ret!=KErrNone)
			{// attempt rollback
			if (newEntry)
				dic.Remove(iStreamUid);
			__LEAVE(ret);
			}
		if (newEntry)
			iDictionary->iDictionaryHasChanged=ETrue;
		}
	else
		{
		iRealBuf->SynchL();
		iDictionary->iStore->Delete(iStreamId);
		iDictionary->Remove(iStreamUid);
		}
	}


void HDictionaryStoreBuf::DoRelease()
	{
	iRealBuf->Release();
	delete this;
	}


TInt HDictionaryStoreBuf::DoReadL(TAny* aPtr,TInt aMaxLength)
	{
	return iRealBuf->ReadL(aPtr,aMaxLength);
	}


TInt HDictionaryStoreBuf::DoReadL(TDes8& aDes,TInt aMaxLength,TRequestStatus& aStatus)
	{
	return iRealBuf->ReadL(aDes,aMaxLength,aStatus);
	}


TStreamTransfer HDictionaryStoreBuf::DoReadL(MStreamInput& anInput,TStreamTransfer aTransfer)
	{
	return iRealBuf->ReadL(anInput,aTransfer);
	}


void HDictionaryStoreBuf::DoWriteL(const TAny* aPtr,TInt aLength)
	{
	iRealBuf->WriteL(aPtr,aLength);
	}


TInt HDictionaryStoreBuf::DoWriteL(const TDesC8& aDes,TInt aMaxLength,TRequestStatus& aStatus)
	{
	return iRealBuf->WriteL(aDes,aMaxLength,aStatus);
	}


TStreamTransfer HDictionaryStoreBuf::DoWriteL(MStreamOutput& anOutput,TStreamTransfer aTransfer)
	{
	return iRealBuf->WriteL(anOutput,aTransfer);
	}


TStreamPos HDictionaryStoreBuf::DoSeekL(TMark aMark,TStreamLocation aLocation,TInt anOffset)
	{
	return iRealBuf->SeekL(aMark,aLocation,anOffset);
	}

// class HNullBuf

HNullBuf::HNullBuf()
	{
	Set(0,0,ERead);
	}

void HNullBuf::DoRelease()
	{
	delete this;
	}