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

//

#include "emailsmokecommands.h"

#ifndef _NO_IAP_PREFS

#include <iapprefs.h>

#endif

_LIT(K_T_MSG_STRUCT_FILE, "c:\\logs\\email\\Entry_Structure.txt");

_LIT(KCountText, "[%4d] Children of '%S' counted -\n\t\t\t      Services: %4d, Folders: %4d, Messages: %4d, Attachments: %4d\n");

_LIT(KErrService, "[%4d] Error! No service has been created or used!");

_LIT(KErrSelectText, "Error! Cannot find entry '%S'.\n");

const TInt KDefaultWaitTime = 5000000;

const TInt KNoBodyText = -1;

class TLineReader : public CBase

	{

private:

	RFile& iFile;

	TBuf8<512> iBuffer;

	TInt iBufferIndex;

	TBool Peek(char& rChar)

		{

		TBool characterFound = ETrue;

		if (iBufferIndex == iBuffer.Size())

			{

			// Refresh buffer

			TInt readError = iFile.Read(iBuffer);

			iBufferIndex = 0;

			if ((readError != KErrNone) || (iBuffer.Size() == 0))

				{

				characterFound = EFalse;

				}

			}

		if (characterFound)

			{

			rChar = iBuffer[iBufferIndex];

			}

		return characterFound;

		};

	TBool GetCharacter(char& rChar)

		{

		TBool characterFound = Peek(rChar);

		if (characterFound)

			{

			iBufferIndex++;

			}

		return characterFound;

		}

public:

	TLineReader(RFile& rFile) : iFile(rFile), iBufferIndex(0)

		{};

	HBufC8* GetLineLC()

		{

		// Find next cr or lf (and ignore subsequent cr's or lf's)

		HBufC8* newLine = HBufC8::NewLC(512);

		char ch;

		TBool characterFound = GetCharacter(ch);

		TBool eolFound = EFalse;

		while ((characterFound) && (!eolFound))

			{

			if ((ch == 0x0d) || (ch == 0x0a))

				// cr or lf found, ignore subsequent cr and lf's

				{

				eolFound = ETrue;

				while ((characterFound) && ((ch == 0x0d) || (ch == 0x0a)))

					{

					characterFound = Peek(ch);

					if ((characterFound) && ((ch == 0x0d) || (ch == 0x0a)))

						{

						characterFound = GetCharacter(ch);

						}

					}

				}

			if ((characterFound) && (!eolFound))

				{

				newLine->Des().Append(ch);

				characterFound = GetCharacter(ch);

				}

			}

		return newLine;

		};

	};

//

//

// CMsvTestCleanMessageFolder

//

void CCleanMessageFolder::StartL(TRequestStatus& aStatus)

	{

	TInt err=KErrNone;

	TRAP(err, iTestUtils.CleanMessageFolderL());

	TRAP(err,iTestUtils.ClearEmailAccountsL());

	TRequestStatus* status = &aStatus;

	User::RequestComplete(status, err);

	}

CCleanMessageFolder::CCleanMessageFolder(CEmailTestUtils& aTestUtils) : iTestUtils(aTestUtils)

	{

	}

//

//

// CDumpMailStore

//

void CDumpMailStore::StartL(TRequestStatus& aStatus)

	{

	TFileName filepath(iPath);

	filepath.Insert(0, _L("dump\\"));

	TParse parsedFileName;

	iTestUtils.ResolveLogFile(filepath, parsedFileName);

	// Clean up any existing logs

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

	CleanupStack::PushL(fileMan);

	TInt err = fileMan->RmDir(parsedFileName.FullName()); 

	if(err == KErrNone || err == KErrNotFound)

		{

		iTestUtils.ResolveLogFile(filepath, parsedFileName); // create logs dir again

		TRAP(err, iTestUtils.FindChildrenL(KMsvRootIndexEntryId, parsedFileName.FullName()));

		}

	CleanupStack::PopAndDestroy(); // fileman

	TRequestStatus* status = &aStatus;

	User::RequestComplete(status, err);

	}

CDumpMailStore::CDumpMailStore(CEmailTestUtils& aTestUtils, const TDesC& aPath) : iTestUtils(aTestUtils)

	{

	iPath = aPath;

	}

//

//

// CCompareFiles

//

//  Compares two files passed in as arguments, the first residing in \MsgLogs\t_email\dump\ and the second

//  residing in \msgtest\emailsmoke\reference\ .

void CCompareFiles::LogCommentFormat(TRefByValue<const TDesC> format,...)

	{

	// Build parameter list.

	VA_LIST list;

	VA_START(list, format);

	TBuf<0x100> buf;

	buf.FormatList(format, list);

	// Log the debug buffer.

	iTestUtils.WriteComment(buf);

	}

void CCompareFiles::StartL(TRequestStatus& aStatus)

	{

	TInt err=KErrNone;

	RFileReadStream file1;

	RFileReadStream file2;

	HBufC8* line1Buffer = HBufC8::NewLC(1024);

	TPtr8 line1 = line1Buffer->Des();

	TBuf8<1> aChar1;

	HBufC8* line2Buffer = HBufC8::NewLC(1024);

	TPtr8 line2 = line2Buffer->Des();

	TBuf8<1> aChar2;

	// Generate the file names, first the result file and then the reference file.

	TFileName file1Name(iPath1);

	file1Name.Insert(0, _L("dump\\"));

	TParse parsedFileName1;

	iTestUtils.ResolveLogFile(file1Name, parsedFileName1);

	TFileName file2Name(iPath2);

	TParse parsedFileName2;

	iTestUtils.ResolveFile(_L("emailsmoke\\reference"), file2Name, parsedFileName2);

	TInt error1=KErrNone;

	TInt error2=KErrNone;

	error1=file1.Open(iTestUtils.FileSession(), parsedFileName1.FullName(), EFileShareAny);

	if(error1!=KErrNone)

		{

		LogCommentFormat( _L("\nERROR %d opening file %s"), error1, file1Name);

		err = KErrUnknown;

		}

	else

		{

		error2=file2.Open(iTestUtils.FileSession(), parsedFileName2.FullName(), EFileShareAny);

		if(error2!=KErrNone)

			{

			LogCommentFormat( _L("\nERROR %d opening file %s"), error2, file2Name);

			err = KErrUnknown;

			}

			else

				LogCommentFormat( _L("\nFiles open"));

		}

	if((error1==KErrNone)&&(error2==KErrNone))

		{

		// read the file into the conversion object

		TInt file1LineCounter = 0;

		TBool finished = EFalse;

		do	{

			line1.FillZ();

			line1.SetLength(0);

			// compile the lines one char at a time

			do	{

				TRAPD( error, file1.ReadL(aChar1, 1) ); 

				if (error!=KErrEof)

					line1.Append(aChar1);

				else

					{

					finished = ETrue;

					break;

					}

				// stop at the end of line or no more data

				}

			while((aChar1[0]!=0x0A)&&(line1.Length()<1024));

			line2.FillZ();

			line2.SetLength(0);

			do	{

				TRAPD( error, file2.ReadL(aChar2, 1) ); 

				if (error!=KErrEof)

					line2.Append(aChar2);

				else

					{

					finished = ETrue;

					break;

					}

				// stop at the end of line or no more data

				}

			while((aChar2[0]!=0x0A)&&(line2.Length()<1024));

			if(!finished)

				{

				line1.TrimRight();

				line2.TrimRight();

				if(aChar1[0]==0x0A) 

					file1LineCounter++;

				if(line1.Compare(line2))

					{

					LogCommentFormat(_L("\nERROR - Line %d is different "), file1LineCounter);

					err = KErrUnknown;

					}

				}

			}

		while(!finished);

		}		

	file1.Close();

	file2.Close();

	CleanupStack::PopAndDestroy(2);  //line 1 , line 2

	TRequestStatus* status = &aStatus;

	User::RequestComplete(status, err);

	}

CCompareFiles::CCompareFiles(CMsvTestUtils& aTestUtils, const TDesC& aPath1, const TDesC& aPath2) : iTestUtils(aTestUtils)

	{

	iPath1 = aPath1;

	iPath2 = aPath2;

	}

//

//

// CDumpToFile

//

void CDumpToFile::WriteToFileL(RFile& aFile, const TDesC& aInputLine)

	{

	HBufC8* pOutputLine = HBufC8::NewLC(aInputLine.Length()*2);

	TPtr8 pOutput = pOutputLine->Des();

	pOutput.Zero();

	TUint lowChar;

	TUint highChar;

	TInt i=0;

	while(i<aInputLine.Length())

		{

		lowChar=((TUint)aInputLine[i])&0x00FF;

		pOutput.Append(lowChar);

		highChar=((TUint)aInputLine[i++])&0xFF00;

		pOutput.Append(highChar);

		}

	aFile.Write(pOutput);

	CleanupStack::PopAndDestroy(); // pBodyText

	}

//

//

// CDumpBodyText

//

void CDumpBodyText::StartL(TRequestStatus& aStatus)

	{

	TInt err = KErrNone;

	TFileName filepath(iPath);

	filepath.Insert(0, _L("dump\\"));

	TParse parsedFileName;

	iTestUtils.ResolveLogFile(filepath, parsedFileName);

	// delete any previous logs 

	err = iTestUtils.FileSession().Delete(parsedFileName.FullName());

	if(err == KErrNone || err == KErrNotFound)

		{

		RFile file;

		err = file.Replace(iTestUtils.FileSession(), parsedFileName.FullName(), EFileShareAny | EFileStreamText | EFileWrite);

		if(err == KErrNone)

			{

			TInt offset=0;

			err = file.Seek(ESeekEnd, offset);

			if (err == KErrNone)

				{

				TInt numEmails = iSelection.Count();

				TMsvId entryId;

				// dump the contents of every message in the selection to the file

				for(TInt i=0; i<numEmails; i++)

					{

					entryId = iSelection.At(i);

					DumpBodyTextToFileL(file, entryId);

					}

				}

			file.Close();

			}

		}

	TRequestStatus* status = &aStatus;

	User::RequestComplete(status, err);

	}

void CDumpBodyText::DumpBodyTextToFileL(RFile& aFile, TMsvId& aEntryId)

	{

	TBuf<100> buf;

	CParaFormatLayer* paraFormatLayer=CParaFormatLayer::NewL();

	CleanupStack::PushL(paraFormatLayer);

	CCharFormatLayer* charFormatLayer=CCharFormatLayer::NewL(); 

	CleanupStack::PushL(charFormatLayer);

	CRichText* bodyText=CRichText::NewL(paraFormatLayer, charFormatLayer, CEditableText::EFlatStorage, 256);

	CleanupStack::PushL(bodyText);

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*(iTestUtils.iMsvEntry));

	imEmailMessage->GetBodyTextL(aEntryId, CImEmailMessage::EThisMessageOnly, *bodyText, *paraFormatLayer, *charFormatLayer);

	HBufC* pBodyText = HBufC::NewLC(bodyText->DocumentLength()+(bodyText->DocumentLength()/70)+1);

	TPtr pBody = pBodyText->Des();

	bodyText->Extract(pBody, 0);

	buf.Zero();

	//buf.AppendFormat(_L("*** %d *************** RichText Data ***************\n"), aEntryId);

	buf.AppendFormat(_L("************************** RichText Data ***************\n"));

	buf.AppendFormat(_L("Size >>> %d\n"), bodyText->DocumentLength());	

	WriteToFileL(aFile, buf);

	RemoveRichtextFormating(pBody);

	WriteToFileL(aFile, pBody);

	buf.Zero();

	buf.AppendFormat(_L("\n********************* end of Body ***********************\n"));

	WriteToFileL(aFile, buf);

	CleanupStack::PopAndDestroy(); // pBodyText

	CleanupStack::PopAndDestroy();  // imEmailMessage

	CleanupStack::PopAndDestroy(3); // bodyText, charFormatLayer, paraFormatLayer

	}

CDumpBodyText::CDumpBodyText(CMsvEntrySelection& aSelection, CMsvTestUtils& aTestUtils, const TDesC& aPath) : 

	iSelection(aSelection),

	iTestUtils(aTestUtils)

	{

	iPath = aPath;

	}

void CDumpBodyText::RemoveRichtextFormating(TDes& aSourceLine)

	{

	TUint8* ptr = (TUint8*)aSourceLine.Ptr();

	TUint8* start = ptr;

	TInt totalLength = aSourceLine.Length()*2;

	do {

		if(*ptr==CEditableText::EParagraphDelimiter || *ptr==CEditableText::ELineBreak || *ptr==CEditableText::EPageBreak)

			*ptr=0x0A;

		} while((++ptr-start)<totalLength);

	}

//

//

// CGetAttachmentNames

//

void CGetAttachmentNames::StartL(TRequestStatus& aStatus)

	{

	TInt err = KErrNone;

	TFileName filepath(iPath);

	filepath.Insert(0, _L("dump\\"));

	TParse parsedFileName;

	iTestUtils.ResolveLogFile(filepath, parsedFileName);

	// delete any previous logs 

	err = iTestUtils.FileSession().Delete(parsedFileName.FullName());

	if(err == KErrNone || err == KErrNotFound)

		{

		RFile file;

		err = file.Replace(iTestUtils.FileSession(), parsedFileName.FullName(), EFileShareAny | EFileStreamText | EFileWrite);

		if(err == KErrNone)

			{

			TInt offset=0;

			err = file.Seek(ESeekEnd, offset);

			if (err == KErrNone)

				{

				TInt numEmails = iSelection.Count();

				TMsvId entryId;

				// dump the attachment names of every message in the selection to the file

				for(TInt i=0; i<numEmails; i++)

					{

					entryId = iSelection.At(i);

					DumpAttachmentNamesToFileL(file, entryId);

					}

				}

			file.Close();

			}

		}

	TRequestStatus* status = &aStatus;

	User::RequestComplete(status, err);

	}

void CGetAttachmentNames::DumpAttachmentNamesToFileL(RFile& aFile, TMsvId& aEntryId)

	{

	TBuf<100> buf;

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*(iTestUtils.iMsvEntry));

	imEmailMessage->GetAttachmentsListL(aEntryId, CImEmailMessage::EAllAttachments, CImEmailMessage::EThisMessageOnly);

	TPtrC attachName(KNullDesC);

	MMsvAttachmentManager& manager = imEmailMessage->AttachmentManager();

	for(TInt i =0; i<manager.AttachmentCount(); i++)

		{

		CMsvAttachment* attachmentInfo = manager.GetAttachmentInfoL(i);

		CleanupStack::PushL(attachmentInfo);

		attachName.Set(attachmentInfo->AttachmentName());		

		WriteToFileL(aFile, attachName);

		buf.Zero();

		buf.AppendFormat(_L("\n"));

		WriteToFileL(aFile, buf);

		CleanupStack::PopAndDestroy(attachmentInfo);

		}	

	CleanupStack::PopAndDestroy(imEmailMessage);  // imEmailMessage

	}

CGetAttachmentNames::CGetAttachmentNames(CMsvEntrySelection& aSelection, CMsvTestUtils& aTestUtils, const TDesC& aPath) : 

	iSelection(aSelection),

	iTestUtils(aTestUtils)

	{

	iPath = aPath;

	}

//

//

// CDumpHeaderFields

//

void CDumpHeaderFields::StartL(TRequestStatus& aStatus)

	{

	TInt err = KErrNone;