// Copyright (c) 1999-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:

//

#ifdef _DEBUG

#undef _NO_SERVER_LOGGING_

#endif

#include <s32file.h>

#include <bautils.h>

#include <barsc.h>

#include <msgs.rsg>

#include "indexcontext.h"

#include "MSVSERV.H"

#include "MSVUTILS.H"

#include "MSVRBLD.H"

// will this header be expoerted to epoc\include

#include <sqldb.h>

#include "msvindexadapter.h"

#if (defined SYMBIAN_MESSAGESTORE_UNIT_TESTCODE)

	_LIT(KDBFileName,"\\messaging.db");

#else

	_LIT(KDBFileName,"[1000484B]messaging.db");

#endif

_LIT(KServerResourceFile,"\\resource\\messaging\\msgs.rsc");

#if (defined SYMBIAN_MESSAGESTORE_UNIT_TESTCODE)

	_LIT(KStoreDeletedFile,"\\private\\1000484b\\storedeleted.tmp");

#else

	_LIT(KStoreDeletedFile,"storedeleted.tmp");

#endif

const TInt KMsvInitDelayTime = 0x00000001; // as soon as possible

const TUid KUidMsvIndexFile = {0x10003C6B};

extern const TInt KMsvIndexStoreVersionNumber = 1;   // See MsvIndex.cpp

//**********************************

// CMsvIndexContext

//**********************************

CMsvIndexContext* CMsvIndexContext::NewL(CMsvServer& aServer, MMsvContextObserver& aObserver)

	{

	CMsvIndexContext* self = CMsvIndexContext::NewLC(aServer, aObserver);

	CleanupStack::Pop();

	return self;

	}

CMsvIndexContext* CMsvIndexContext::NewLC(CMsvServer& aServer, MMsvContextObserver& aObserver)

	{

	CMsvIndexContext* self = new(ELeave)CMsvIndexContext(aServer, aObserver);

	CleanupStack::PushL(self);

	self->ConstructL();

	return self;

	}

void CMsvIndexContext::ConstructL()

	{

	iUpdateEntryStreams = new(ELeave)CMsvEntrySelection;

	iRemoveEntries = new(ELeave)CMsvEntrySelection;

	TFileName fileName(KServerResourceFile);

	MsvUtils::AddPathAndExtensionToResFileL(fileName);

 	BaflUtils::NearestLanguageFile(iServer.FileSession(), fileName);

	// Load the initial index structure from resource file

	RResourceFile resource;

	resource.OpenL(iServer.FileSession(), fileName);

	CleanupClosePushL(resource);

	// Get ready to read resource

	iBuf = resource.AllocReadL(R_SERVER_INDEX_STARTUP);

	CleanupStack::PopAndDestroy(); //resource

	}

CMsvIndexContext::~CMsvIndexContext()

	{

	Cancel();

	delete iBuf;

	delete iIndexAdapter;

	delete iUpdateEntryStreams;

	delete iRemoveEntries;

	}

CMsvIndexContext::CMsvIndexContext(CMsvServer& aServer, MMsvContextObserver& aObserver)

: CActive(EPriorityStandard), iServer(aServer), iObserver(aObserver)

	{

	CActiveScheduler::Add(this);

	}

// CODE USED AFTER PREQ 557.

#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

/**

 * CreateIndex()

 * 

 * Code added for PREQ 557.

 */

void CMsvIndexContext::CreateIndexL()

	{

	__ASSERT_DEBUG(iProgress.iState == TMsvIndexLoadProgress::EIndexNotLoaded, PanicServer(EMsvBadIndexState));

	DoCreateIndexL();

	IndexLoadingCompleted();	

	}

void CMsvIndexContext::IndexLoadingCompleted()

	{

	iProgress.iError = KErrNone;

	IndexLoaded(EMsvNullNotification);

	// Tell the observer that we've finshed

	iObserver.ContextComplete(KErrNone, iRunMailInit);

	}

/**

 * DoCreateIndexL()

 *

 * Create the index adapter object.

 * Code added in PREQ 557.

 */

void CMsvIndexContext::DoCreateIndexL()

	{	

	// Remember stuff

	iObserverStatus = NULL;

	iRunMailInit = EFalse;

	// Update drive status in preferred drive list.

	TDriveState driveStatus = EMsvInvalidDriveStatus;

	for(TUint index=0; index<CMsvPreferredDriveList::GetDriveList()->Count(); index++)

		{

		UpdateDriveStatusL(index, driveStatus);

		}

	// Look for the current drive.

	// The first drive in the preferred drive list, whose 

	// status is either EMsvMessageStoreAvailable or 

	// EMsvMessageStoreUnavailable will be the current drive.

	TMsvPreferredDrive driveEntry;

	CMsvPreferredDriveList *driveList = CMsvPreferredDriveList::GetDriveList();

	for(TInt currentDriveIndex=0; currentDriveIndex<driveList->Count(); ++currentDriveIndex)

		{

		driveEntry = (*driveList)[currentDriveIndex];

		if( (EMsvMessageStoreAvailableStatus   == driveEntry.status) ||

			(EMsvMessageStoreUnavailableStatus == driveEntry.status)

		  )

			{

			// Set the current drive index to server.

			CMsvPreferredDriveList::GetDriveList()->SetCurrentDriveIndex(currentDriveIndex);

			TParse parse;

			TPtrC drive(TDriveUnit(driveEntry.driveNum).Name());

			parse.Set(KMsvDefaultIndexFile2(), &drive, NULL);

			iMessageFolder = parse.DriveAndPath();

			// If the drive already has a message store...

			if(EMsvMessageStoreUnavailableStatus == driveEntry.status)

				{

				// Create the database and all standard table.

				CMsvDBAdapter::CreateDBL(driveEntry.driveNum);

				// Update the drive status.

				CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(currentDriveIndex, EMsvMessageStoreAvailableStatus);

				ResetAndCreateNewMailStoreL();

				iRunMailInit = ETrue;

				// Creating index adapter object.

				iIndexAdapter = CMsvIndexAdapter::NewL(iServer);

				// Get updated drive id.

				CreateStandardEntriesFromResourceFileL(KCurrentDriveId);		

				iIndexAdapter->SetLocalServiceComplete();

				}

			else

				{

				// The drive already has a message store.

				// Creating index adapter object.

				iIndexAdapter = CMsvIndexAdapter::NewL(iServer);

				ResetAndCreateNewMailStoreL(EFalse);						

				}

			break;

			}

		}	

	}	

/**

 * UpdateDriveStatusL()

 * @param TUint: The index of the said drive in preferred

 *               drive list.

 */	

void CMsvIndexContext::UpdateDriveStatusL(TUint aDriveIndex, TDriveState& aStoreStatus)

	{

	aStoreStatus = EMsvInvalidDriveStatus;

	TBool deleteStore = EFalse;

	TMsvPreferredDrive driveEntry;

	CMsvPreferredDriveList::GetDriveList()->DriveInfoL(aDriveIndex, driveEntry);

	// Check if the media is available in the drive.

	TVolumeInfo volume;

	if (iServer.FileSession().Volume(volume, driveEntry.driveNum) != KErrNone)

		{

		CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvDriveDiskNotAvailableStatus);

		aStoreStatus = EMsvDriveDiskNotAvailableStatus;

		return;

		}

	// Validate the database. The function opens the database

	// and check its version and returns appropriate error.

	TRAPD(err, CMsvDBAdapter::ValidateDatabaseL(driveEntry.driveNum));

	// If no error, database is available.

	if(KErrNone == err)

		{

		CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvMessageStoreAvailableStatus);

		return;

		}

	// If the database is not found in the drive.

	if(KErrNotFound == err)

		{

		// Check if index file exists.

		RFile file;

		TParse parse;

		TPtrC drive(TDriveUnit(driveEntry.driveNum).Name());

		parse.Set(KMsvDefaultIndexFile2, &drive, NULL);

		TFileName indexFileName = parse.FullName();

		err = file.Open(iServer.FileSession(), indexFileName, EFileShareAny|EFileWrite);

		//file.Close();

		// If index file exists, set drive status to NOT SUPPORTED.

		if(KErrNone == err)

			{

			CPermanentFileStore* fileStoreIndex = NULL;

			TRAPD(permError, fileStoreIndex = CPermanentFileStore::FromL(file));

			if(KErrNone == permError)

				{

				// Check if the store is corrupt. If so then delete it.

				if (fileStoreIndex->Type() != TUidType(KPermanentFileStoreLayoutUid, KUidMsvIndexFile))

					{

					deleteStore = ETrue;

					}

				}

			// There was an error in getting a permanent filestore object.

			// Mark the message store for deletion.

			else

				{

				deleteStore = ETrue;

				}

			delete fileStoreIndex;

			//If message store is corrupted, wipe it.

			if(deleteStore)

				{

				TFileName mail2Folder = parse.DriveAndPath();

				CFileMan* fileMan = CFileMan::NewL(iServer.FileSession());

				// Remove the readonly attribute..

				(void)fileMan->Attribs(mail2Folder, 0, KEntryAttReadOnly, TTime(0), CFileMan::ERecurse);

				CleanupStack::PushL(fileMan);

				//Check if the mailfolder exists..

				TBool mail2FolderExists = BaflUtils::FileExists(iServer.FileSession(), mail2Folder);

				if(mail2FolderExists)

					{

					// Remove old message store if exists..

					User::LeaveIfError(fileMan->RmDir(mail2Folder));

					}

				CleanupStack::PopAndDestroy(fileMan);

				CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvMessageStoreUnavailableStatus);

				aStoreStatus = EMsvMessageStoreCorruptStatus;

				// Create store delete file in the respective drive.

				CreateStoreDeleteFile(driveEntry.driveNum);

				}

			//Else only set status as EMsvMessageStoreNotSupportedStatus.

			else

				{

				CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvMessageStoreNotSupportedStatus);

				aStoreStatus = EMsvMessageStoreNotSupportedStatus;

				}

			file.Close();

			return;

			}

		// If index file does not exists, set drive status to STORE UNAVAILABLE.

		if((KErrNotFound == err) || (KErrPathNotFound == err))

			{

			CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvMessageStoreUnavailableStatus);

			aStoreStatus = EMsvMessageStoreUnavailableStatus;

			return;

			}

		User::Leave(err);

		}

	//If a database exists, but is corrupt.

	if( (KSqlErrCorrupt == err) ||

		(KSqlErrNotDb   == err)

	  )

		{

		// If the database is corrupt, delete

		// the database and old message store.

		// Delete the database.

		TParse parse;

		TPtrC drive(TDriveUnit(driveEntry.driveNum).Name());

		parse.Set(KDBFileName, &drive, NULL);

		TFileName dBFile = parse.FullName();

		RSqlDatabase::Delete(dBFile);

		// Delete the message store. //ISSUE Can we use ResetAndCreateNewMailStore()?

		parse.Set(KMsvDefaultIndexFile2, &drive, NULL);

		TFileName mail2Folder = parse.DriveAndPath();

		CFileMan* fileMan = CFileMan::NewL(iServer.FileSession());

		// Remove the readonly attribute..

		(void)fileMan->Attribs(mail2Folder, 0, KEntryAttReadOnly, TTime(0), CFileMan::ERecurse);

		CleanupStack::PushL(fileMan);

		//Check if the mailfolder exists..

		TBool mail2FolderExists = BaflUtils::FileExists(iServer.FileSession(), mail2Folder);

		if( mail2FolderExists)

			{

			// Remove old message store if exists..

			User::LeaveIfError(fileMan->RmDir(mail2Folder));

			}

		CleanupStack::PopAndDestroy(fileMan);

		// Update the drive status.

		CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvMessageStoreUnavailableStatus);

		aStoreStatus = EMsvMessageStoreCorruptStatus;

		// Create store delete file in the respective drive.

		CreateStoreDeleteFile(driveEntry.driveNum);

		return;

		}

	//If the database version is not supported.

	if(KErrNotSupported == err)

		{

		// If the drive as old version of database.

		CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aDriveIndex, EMsvMessageStoreNotSupportedStatus);

		aStoreStatus = EMsvMessageStoreNotSupportedStatus;

		return;

		}

	// Do not return DB error.

	if(err <= KSqlErrGeneral)

		{

		// If DB Error, return KErrGeneral.

		User::LeaveIfError(KErrGeneral);

		}

	else

		{

		User::LeaveIfError(err);

		}

	}

TInt CMsvIndexContext::ChangeDrive(TUint aNewDriveIndex, TRequestStatus& aStatus)

	{

	TInt err = ChangeDrive(aNewDriveIndex, ETrue, ETrue);

	// Remember the status which we'll complete

	aStatus = KRequestPending;

	iObserverStatus = &aStatus;

	return err;

	}

void CMsvIndexContext::ChangeDriveCompleted(TInt aError)

	{

	// Remember error

	iProgress.iError = aError;	

	if (iObserverStatus)

		{

		User::RequestComplete(iObserverStatus, aError);

		}

	}

/**

 * ChangeDriveL()

 * @param TUint: Index of the new current drive in the preferred drive list.

 * @param TBool: True, if old drive is still present in the preferred drive list.

 *               False, otherwise.

 * 

 */

TInt CMsvIndexContext::ChangeDrive(TUint aNewDriveIndex, TBool aIsRemovePartial /*= ETrue*/, TBool aAsync /*=EFalse */)

	{

	TDriveNumber newDriveNumber;

	TRAPD(err, DoChangeDriveL(aNewDriveIndex, aIsRemovePartial, newDriveNumber));

	// Add a function ChangeDriveComplete() in server

	// which should be called here. The function should 

	// run mailInit and recreate backup object.

	iObserver.ChangeDriveComplete(err, iRunMailInit, newDriveNumber);

	if(aAsync)

		{

		iStatus = KRequestPending;

		SetActive();

		TRequestStatus* st = &iStatus;

		User::RequestComplete(st, err);

		}

	return err;

	}

void CMsvIndexContext::DoChangeDriveL(TUint aNewDriveIndex, TBool aIsRemovePartial, TDriveNumber& aNewDriveNumber)

	{

	iObserverStatus = NULL;

	TMsvPreferredDrive driveEntry;

	CMsvPreferredDriveList::GetDriveList()->DriveInfoL(aNewDriveIndex, driveEntry);

	aNewDriveNumber = driveEntry.driveNum;

	if(EMsvMessageStoreUnavailableStatus == driveEntry.status)

		{

		// Create the database and all standard table.

		CMsvDBAdapter::CreateDBL(driveEntry.driveNum);

		// Update the drive status.

		CMsvPreferredDriveList::GetDriveList()->UpdateDriveStatusL(aNewDriveIndex, EMsvMessageStoreAvailableStatus);

		TParse parse;

		TPtrC drive(TDriveUnit(driveEntry.driveNum).Name());

		parse.Set(KMsvDefaultIndexFile2(), &drive, NULL);

		iMessageFolder = parse.DriveAndPath();

		ResetAndCreateNewMailStoreL();

		iRunMailInit = ETrue;

		// Perform ChangeDrive operation.

		iIndexAdapter->ChangeDriveL(aNewDriveIndex, aIsRemovePartial);

		// Read initial entries from resources

		CreateStandardEntriesFromResourceFileL(KCurrentDriveId);

		iIndexAdapter->SetLocalServiceComplete();

		}

	else

		{

		// Update the message folder local variable.

		TParse parse;

		TPtrC drive(TDriveUnit(driveEntry.driveNum).Name());

		parse.Set(KMsvDefaultIndexFile2(), &drive, NULL);

		iMessageFolder = parse.DriveAndPath();

		ResetAndCreateNewMailStoreL(EFalse);

		// Perform ChangeDrive operation.

		iIndexAdapter->ChangeDriveL(aNewDriveIndex, aIsRemovePartial);			

		iRunMailInit = EFalse;

		}

	// Create service directories in new drive.

	TRAP_IGNORE(DoCreateServiceDirsL(KCurrentDriveId));

	TRAP_IGNORE(LocalizeStandardFoldersL());

	}

void CMsvIndexContext::GetInPreparationIds(TUint aDriveId)

	{

	delete iRemoveEntries;

	iRemoveEntries = new CMsvEntrySelection;

	iIndexAdapter->GetInPreparationIds(*iRemoveEntries, aDriveId);

	}

/**

 * CreateStandardEntriesFromResourceFileL()

 * @param None:

 *

 * Will read messaging resource file and create entries.

 */	

void CMsvIndexContext::CreateStandardEntriesFromResourceFileL(TUint aDriveId)

	{

	// Read initial entries from resources

	TResourceReader reader;

	reader.SetBuffer(iBuf);	

	const TInt numberOfEntries = reader.ReadInt16();

	for (TInt index=0; index < numberOfEntries; ++index)

		{

		TMsvEntry newEntry;

		// Values from resource file

		newEntry.iId = MaskTMsvId(aDriveId, reader.ReadInt32());