// 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:

// HEADER FILES

// 

//

 #include "msvsearchsortdbwrapper.h"

 #include "MSVSTD.H"

 #include "msvindexadapter.h"

 #if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)

 #include "MSVSERV.H"

 #endif

 const TInt KMaxQuerySize = 4000;

 const TInt KMaxInCount = 50;

/**

 * LITERAL DEFINITION

 */

_LIT8(KQuote, "\"");

_LIT8(KComma, ", ");

_LIT8(KAnd, " AND ");

_LIT8(KLike, " LIKE ");

_LIT8(KPercent, "%");

_LIT8(KUnderScore, "_");

_LIT8(KDescription, "description");	

_LIT8(KDetails, "details");

_LIT8(KDate, "date");

_LIT8(KSize, "size");

_LIT8(KMtmId, "mtmId");

_LIT8(KType,"type");

_LIT8(KCloseBrace, ")");

_LIT8(KAscending," ASC");

_LIT8(KDescending," DESC");

_LIT8(KDelimiter,";");

_LIT8(KOrderBy," ORDER by ");

_LIT8(KBitwiseData,"data & ");

_LIT8(KId, "id");

_LIT8(KSelect, "SELECT ");

#if (!defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)

_LIT8(KIdIn, "id IN(");

#endif

_LIT8(KWhere, " WHERE ");

_LIT8(KNull, "NULL");

_LIT8(KPrioritySort, "data & 3");

_LIT8(KAttachmentSort, "data & 32768");

_LIT8(KNewMsgSort, "data & 16");

_LIT8(KUnReadSort, "data & 32");

_LIT8(KIndexSortInsert, "INSERT OR IGNORE INTO SearchSortEntry (Qid, id, sortfield) ");

_LIT8(KGreaterThan, " > ");

_LIT8(KLessThan, " < ");

_LIT8(KGreaterThanEqualTo," >= ");

_LIT8(KLessThanEqualTo," <= ");

_LIT8(KNotEqualTo," != ");

_LIT8(KEqualTo, " = ");

#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

	_LIT8(KFromDb," FROM DB");

	_LIT8(KIndexSortSelectRemovable, ".IndexEntry WHERE ");

	_LIT8(KIndexSortSelectNoParentRemovable, ".IndexEntry ");

	_LIT8(KIdQuerySelect, "SELECT DISTINCT id FROM DB");

#else

	_LIT8(KIdQuery, "SELECT DISTINCT id ");

	_LIT8(KIdQueryNoParent, "SELECT DISTINCT id FROM IndexEntry ");

	_LIT8(KIndexSortSelect, " FROM IndexEntry WHERE parentId = ");

	_LIT8(KIndexSortSelectNoParent, " FROM IndexEntry ");

#endif

/**

 * TSearchSortDbWrapper

 */

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)	

TSearchSortDbWrapper* TSearchSortDbWrapper::NewL(CMsvServer& aServer)

	{

	//Create an Search sort db adaption object 

	TSearchSortDbWrapper* self = new(ELeave) TSearchSortDbWrapper(aServer);

	return self;

	}

TSearchSortDbWrapper::TSearchSortDbWrapper(CMsvServer& aServer) : iServer(aServer)

	{

	iDBadapter = iServer.Context().IndexAdapter()->GetDbAdapter();

	iTotalCount = -1;

	iProcessedIndexQueries = 0;

	}

void TSearchSortDbWrapper::CloseResources()

	{

	// relese the resources

	iMtmIdArray.Close();	

	}

#else 	// SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB

/**

 NewL()

It returns an instance of TSearchSortDbWrapper class. 

@param CMsvDBAdapter& aDBadapter: reference to CMsvDBAdapter Class.

@return The newly created Msv DBadaption Object.

@leave: KErrNone if successful or any of system wide leave codes if not.

*/

TSearchSortDbWrapper* TSearchSortDbWrapper::NewL(CMsvDBAdapter& aDBadapter)

	{

	//Create an Search sort db adaption object 

	TSearchSortDbWrapper* self = new(ELeave) TSearchSortDbWrapper(aDBadapter);

	return self;

	}

/**

 TSearchSortDbWrapper()

Constructor for the TSearchSortDbWrapper class.

@param 	None.

@return None.

*/

TSearchSortDbWrapper::TSearchSortDbWrapper(CMsvDBAdapter& aDBadapter):iDBadapter(&aDBadapter), iTotalCount(-1), iProcessedIndexQueries(0)

	{

	}

#endif // if (!defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)

/** 

  FindL()

Depending on the CMsvSearchSortCacheEntry passed to this function it constructs the DB query

and gets the resultant TMsvIds for the query so formed.

@param CMsvSearchSortCacheEntry& aQuery:	Contains all the SearchSort related information(What to search on, what to search for).

@param RArray<TMsvId>& aIdResult: RArray which holds the resultant TMsvIds.

@param TBool aSortOnHeader: This specifies whether the sort field is a part of the Header(To, Bcc, Cc) or TMsvEntry(Date, Size, priority).

@leave: KErrNone if successful or any of system wide leave codes or SQL leave codes if not.

*/ 

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

void TSearchSortDbWrapper::FindL(const CMsvSearchSortCacheEntry& aQuery, RArray<TMsvId>& aIdResult, const TBool aSortOnHeader, TInt aTypeOfQuery, TInt& aStatusOfQuery)

#else

void TSearchSortDbWrapper::FindL(const CMsvSearchSortCacheEntry& aQuery, RArray<TMsvId>& aIdResult, const TBool aSortOnHeader/*DEFAULT = EFalse*/)

#endif

	{

	RBuf8 queryString;

	queryString.CreateL(KMaxQuerySize);

	queryString.CleanupClosePushL();

	RArray<TMsvId> folderIdList;

	CleanupClosePushL(folderIdList);

	TMsvId parentId;

#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

	parentId = UnmaskTMsvId(aQuery.iParentId);

#else

	parentId = aQuery.iParentId;

#endif  

	//Append the parentid on which the search/sort is intiated into the folder list

	folderIdList.AppendL(parentId);

	iSortOnHeader = aSortOnHeader;

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)	

	iTypeOfQuery = aTypeOfQuery;

	iTotalNumMsgPartsOnHeader = 0;

	iNumOfMsgParts = 0;

	ParseQueryForHeaderFields(aQuery);

#endif	

	for(TInt folderindex = 0; folderindex < folderIdList.Count(); ++folderindex)

		{

		//If subfolder search is enabled then all the subfolders under the parent and its 

		// childrens are searched for a match.

		if(aQuery.iSubFolderOption && parentId != KMsvRootIndexEntryIdValue)

			{

			iDBadapter->GetFolderIdsL(folderIdList[folderindex], folderIdList);	

			}

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)	

		// get MtmIds for corresponding parentId

		TRAPD(error, ProcessPreQueryRequestL(folderIdList[folderindex], aStatusOfQuery));

		if(error == KErrArgument)

			{

			queryString.Delete(0,KMaxQuerySize);

			continue;

			}

#endif

		//Create the SQLquery from the CMsvSearchSortCacheEntry and the parentid specified

		CreateQueryL(aQuery, queryString, folderIdList[folderindex]);

		//If the sort is done on a Header specific part(To, Cc, Bcc), sortfield is not stored in DB

		//It has to be fetched from client side, hence sending the TMsvIds to client without storing.

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

		if(iStatusOfQuery == KNewQuery)

#else

		if(iSortOnHeader)

#endif				

			{

			iDBadapter->SearchTMsvIdL(queryString, aIdResult);

			}

		//If the sort is done on a TMsvEntry specific part(size, date, mtmtype), sortfield is stored in DB

		else  // (iStatusOfQuery == KPartialResult || iStatusOfQuery == KFinalResult)

			{

			iDBadapter->AddEntriesfromIndexTableL(queryString);

			}

		queryString.Delete(0,KMaxQuerySize);

		}

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

	if(iStatusOfQuery == KPartialResult || iStatusOfQuery == KFinalResult)

#else

	if(!iSortOnHeader)

#endif

		{

		//If the sort is done on a TMsvEntry fetch all the TMsvIds matching this query and pass it to client

		iDBadapter->GetSearchSortEntriesL(aQuery.iQueryID, aIdResult, aQuery.IsAscendingSort(), IsSortAsNumRequired(aQuery.iMsgExplicitSortPart));	

		}

	iProcessedIndexQueries = 0;

	CleanupStack::PopAndDestroy(&folderIdList);  // for folderIdList;

	CleanupStack::PopAndDestroy(); // for  queryString;

	}

/** 

  UpdateNewIdsL()

Depending on the CMsvSearchSortCacheEntry and the TMsvIds passed to this function it constructs the DB query

and gets the resultant TMsvIds for the query so formed.

@param CMsvSearchSortCacheEntry& aQuery:	Contains all the SearchSort related information(What to search on, what to search for).

@param RArray<TMsvId>& aIdFilter: RArray holding the entries to be filtered with

@param RArray<TMsvId>& aIdResult: RArray which holds the resultant TMsvIds.

@param const TBool aSortOnHeader: This specifies whether the sort field is a part of the Header(To, Bcc, Cc) or TMsvEntry(Date, Size, priority).

@leave: KErrNone if successful or any of system wide leave codes or SQL leave codes if not.

*/ 

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

void TSearchSortDbWrapper::UpdateNewIdsL(const CMsvSearchSortCacheEntry& aQuery,const RArray<TMsvId>& aIdFilter, RArray<TMsvId>& aIdResult,const TBool aSortOnHeader, TInt aTypeOfQuery, TInt& aStatusOfQuery)

#else

void TSearchSortDbWrapper::UpdateNewIdsL(const CMsvSearchSortCacheEntry& aQuery,const RArray<TMsvId>& aIdFilter, RArray<TMsvId>& aIdResult,const TBool aSortOnHeader/*DEFAULT = EFalse*/)

#endif

	{

	if(aIdFilter.Count() != 0)

		{

		TInt idArrayCount = 0;

		TInt initQuerySize = 0;

		TInt idsAdded = 0;

		RBuf8 queryString;

		queryString.CleanupClosePushL();

		//Folderlist for subfolder search

		RArray<TMsvId> folderIdList;

		TMsvId parentId;

	#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

		parentId = UnmaskTMsvId(aQuery.iParentId);

	#else

		parentId = aQuery.iParentId;

	#endif  

		//Append the parentid on which the search/sort is intiated into the folder list

		iSortOnHeader = aSortOnHeader;

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)	

		iTypeOfQuery = aTypeOfQuery;

		iTotalNumMsgPartsOnHeader = 0;

		iNumOfMsgParts = 0;

		ParseQueryForHeaderFields(aQuery);

#endif

		for(TInt folderindex = 0; folderindex < folderIdList.Count(); ++folderindex)

			{

			//If subfolder search is enabled then all the subfolders under the parent and its 

			// childrens are searched for a match.

			if(aQuery.iSubFolderOption && parentId != KMsvRootIndexEntryIdValue)

			{

			iDBadapter->GetFolderIdsL(folderIdList[folderindex], folderIdList);	

			}

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)	

			// get MtmIds for corresponding parentId

			ProcessPreQueryRequestL(folderIdList[folderindex], aStatusOfQuery);

#endif

			//Create the SQLquery from the CMsvSearchSortCacheEntry and the parentid specified

			CreateQueryL(aQuery, queryString, folderIdList[folderindex], ETrue);

			//initQuerySize is made one less than entire size to remove the delimiter(;)

			initQuerySize = queryString.Size()-1;

			//Reset Query string to Append New ID's

			queryString.Delete(initQuerySize,KMaxQuerySize);

			if(iProcessedIndexQueries == 0 && folderIdList[folderindex] == KMsvRootIndexEntryIdValue)

				{

				queryString.Append(KWhere);

				}

			else

				{

				queryString.Append(KAnd);

				}

			//initQuerySize

			initQuerySize = queryString.Size();

			while(idArrayCount < aIdFilter.Count())

				{

				//Reset Query string to Append New ID's

				queryString.Delete(initQuerySize,KMaxQuerySize);

				//Append Id's into the In statement

				DoAppendIdConstraint(aIdFilter, queryString, idArrayCount);

				//If the Sort is on Header, need not store the result as it doesnt have the sortfield

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

				if(iStatusOfQuery == KNewQuery)

#else

				if(iSortOnHeader)

#endif	

					{

					iDBadapter->SearchTMsvIdL(queryString, aIdResult);

					}

				//If the Sort is on index we need to store the result and then send the id's

				else

					{

					idsAdded += iDBadapter->AddEntriesfromIndexTableL(queryString);

					}

				}

			iProcessedIndexQueries = 0;

			idArrayCount = 0;

			queryString.Delete(0,KMaxQuerySize);

			}

		//Get only the updated results for this type of query

		// not the previously stored results

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

		if(iStatusOfQuery == KNewQuery)

#else

		if (aQuery.iQueryType == ECombinedQuery && !aQuery.iSortOnHeader)

#endif		

			{

			iDBadapter->GetLastNTMsvIdsfromTableL(aQuery.iQueryID, aIdResult, idsAdded,aQuery.IsAscendingSort(), IsSortAsNumRequired(aQuery.iMsgExplicitSortPart));

			}

		folderIdList.Close();

		CleanupStack::PopAndDestroy();

		}

	//Get the final list of sorted ids from the SearchSortEntry table

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)

	if(iStatusOfQuery == KPartialResult || iStatusOfQuery == KFinalResult)

#else

	if(!iSortOnHeader)

#endif

		{

		iDBadapter->GetSearchSortEntriesL(aQuery.iQueryID, aIdResult, aQuery.IsAscendingSort(), IsSortAsNumRequired(aQuery.iMsgExplicitSortPart));	

		}

	}

/** 

  CreateQueryL()

Depending on the CMsvSearchSortCacheEntry and the TMsvIds passed to this function it constructs the DB query

and gets the resultant TMsvIds for the query so formed.

@param CMsvSearchSortCacheEntry&:	Contains all the SearchSort related information(What to search on, what to search for).

@param RBuf8&: SQL query created depending on the CMsvSearchSortCacheEntry

@param TMsvId& : ParentId on which the the search/sort is performed on

@param TBool: This Specifies whether sorting is done on Header or TMsvEntry field

@param TBool: This Specifies whether the sort option should be processed or not

@leave: KErrNone if successful, KErrNotSupported if the Search/Sort message passed is not supported

		 or any of system wide leave codes or SQL leave codes.

@return none.

*/

void TSearchSortDbWrapper::CreateQueryL(const CMsvSearchSortCacheEntry& aQuery, RBuf8& aQueryString, const TMsvId& aParentId, TBool aBypassSort)

	{

	// If sort field is on Index part, result should be stored on the DB

	if(!iSortOnHeader) //it's on TMsvEntry

		{

		DoCreateInsertSortQueryL(aQuery, aQueryString, aParentId);

		}

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)

	// If sort field is on Header part and if Header table/s is availble or partiallly available 

	// result should be stored on the DB

	else if(iSortOnHeader && (iHeaderTableAtDB == EAvailable || iHeaderTableAtDB == EPartiallyAvailable))

		{

		DoCreateInsertSortOnHeaderQueryL(aQuery, aQueryString);

		}

#endif		

	else

		{

		if(aParentId != KMsvRootIndexEntryIdValue)

			{

		#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

			aQueryString.Append(KIdQuerySelect);

			aQueryString.AppendNum(KCurrentDriveId);

			aQueryString.Append(KIndexSortSelectRemovable);

		#else

			aQueryString.Append(KIdQuery);

		#endif

			}

		else

			{

		#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

			aQueryString.Append(KIdQuerySelect);

			aQueryString.AppendNum(KCurrentDriveId);

			aQueryString.Append(KIndexSortSelectNoParentRemovable);

		#else

			aQueryString.Append(KIdQueryNoParent);

		#endif

			} 

	#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)

		UpadeteQueryWithParentIdAndMsgType(aQueryString, KCurrentDriveId, aParentId);

	#else

		UpadeteQueryWithParentIdAndMsgType(aQueryString, aParentId);

	#endif		

		}

	//Checks whether the query has a search part specified, 

	//if yes then SQL query is created for the same

	if(!aQuery.IsSortQuery())

		{

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)		

		if((iTypeOfQuery & EQueryOnHeader) && (iHeaderTableAtDB == EAvailable || iHeaderTableAtDB == EPartiallyAvailable ))

			{

			DoUpdateSearchQueryForHeaderFieldsL(aQuery, aQueryString, aParentId);

			}

		if(iTypeOfQuery & EQueryOnIndexEntry)

#endif		

			{

		#if (defined SYMBIAN_MSGS_ENHANCED_REMOVABLE_MEDIA_SUPPORT)	

			UpdateQueryWithTMsvEntrySearchFieldsL(aQuery, aQueryString, KCurrentDriveId, aParentId);

		#else

			UpdateQueryWithTMsvEntrySearchFieldsL(aQuery, aQueryString, aParentId);

		#endif

			}

		}

	//Checks whether the query has a sort part specified

	//if yes then SQL query is created for the same

	if ((aQuery.IsExplicitSortReq() || aQuery.IsSortQuery()) && !aBypassSort)

		{

		if(!iSortOnHeader)

			{

			DoCreateSortQueryL(aQuery, aQueryString);

			}

#if (defined SYMBIAN_MESSAGESTORE_HEADER_BODY_USING_SQLDB)			

		if(iSortOnHeader && (iHeaderTableAtDB == EAvailable || iHeaderTableAtDB == EPartiallyAvailable))

			{

			UpdateQueryWithSortFieldL(aQuery, aQueryString, KCurrentDriveId);

			}

#endif			

		}

	//Adds the delimiter(;)

	aQueryString.Append(KDelimiter);

	//Replaces Symbian WildCards(?, *) with SQL Wildcards (_, %)

	if(aQuery.IsWildCharacter())

		{

		ReplaceWildCardCharacters(aQueryString,aQuery);

		}

	}

/** 

  ReplaceWildCardCharacters()

The Function replaces the '?' in the query with '_' and '*' in the string with a '%'.

@param TDes8& aString: string on which the replace operation has to be done on.

@return none.

*/

void TSearchSortDbWrapper::ReplaceWildCardCharacters(TDes8& aString,const CMsvSearchSortCacheEntry& aQuery)

	{

	for(TInt pos=0; pos < aString.Size(); pos++)

		{

		if(aString[pos] == '?')

			{

			aString.Replace(pos,1,KUnderScore);

			}

		else if(aString[pos] == '*')

			{

			if(!aQuery.IsCaseSensetive())

			    aString.Replace(pos,1,KPercent);

			}

		}

	}

/** 

  DoAddRelationalOperatorL()

The function Appends the required relational operator to the query string passed 

	depending on the enum type specified

@param RBuf8& aQuery: string on which the replace operation has to be done on.

@param TMsvRelationOp aRelationOp: Relational operator which has to be appended to the string

@return none.

@leave: KErrNone if successful, KErrNotSupported if the Search/Sort message passed is not supported.

*/

void TSearchSortDbWrapper::DoAddRelationalOperatorL(RBuf8& aQuery,TMsvRelationOp aRelationOp)

	{