// Copyright (c) 2000-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 "emailtestutils.h"

#include "scripttestutils.h"

_LIT8(K_IM_CR_LF, "\r\n");

_LIT(K_IM_LOG_FILE_NAME, "c:\\logs\\email\\imlog%d.txt");

_LIT(K_IM_END_OF_FILE, "END OF FILE");

_LIT(K_IM_LOG_TEMP_FILE_NAME, "temp.out");

_LIT8(K_IM_READ_CANCELLED, "** Read cancelled");

_LIT8(K_IM_TEST_DATE, "Date: ");

_LIT8(K_IM_TEST_DATE_REPLACEMENT, "date-replacement-string-");

_LIT8(K_IM_TEST_MESSAGE_ID, "Message-ID: ");

_LIT8(K_IM_TEST_MESSAGE_ID_REPLACEMENT, "message-id-replacement-string-");

_LIT8(K_IM_TEST_CONTENT_ID, "Content-ID: ");

_LIT8(K_IM_TEST_CONTENT_ID_REPLACEMENT, "content-id-replacement-string-");

_LIT8(K_IM_TEST_BOUNDARY, "  boundary=");

_LIT8(K_IM_TEST_BOUNDARY_REPLACEMENT, "boundary-replacement-string-");

_LIT8(K_IM_TEST_HELO, "HELO ");

_LIT8(K_IM_TEST_HELO_REPLACEMENT, "");

const TInt K_IM_TEST_STRING_MAX = 128;

const TInt K_IM_TEST_ID_LEN = 5;

CScriptTestUtils::CImTestField::~CImTestField()

	{

	delete iFieldName;

	delete iReplacementString;

	}

CScriptTestUtils::CImFindReplace::~CImFindReplace()

	{

	delete iFind;

	delete iReplace;

	}

EXPORT_C void CScriptTestUtils::LogToInputFileL(const TDesC& aSourceFileName, const TDesC& aDestFileName, TBool aStandardiseDateFields)

// Create an email script file from a log file.

	{

	StripLogFileL(aSourceFileName, aDestFileName, aStandardiseDateFields, EImLogLineIn);

	}

EXPORT_C void CScriptTestUtils::LogToOutputFileL(const TDesC& aSourceFileName, const TDesC& aDestFileName, TBool aStandardiseDateFields)

// Create an email output file from a log file.

	{

	StripLogFileL(aSourceFileName, aDestFileName, aStandardiseDateFields, EImLogLineOut);

	}

CScriptTestUtils::CImFindReplace* CScriptTestUtils::FindVariableDataL(TDes8& aLine)

// Find any variable data fields in the given line.

// The field value and its replacement is returned.

// If the line doesn't contain any variable data then a NULL pointer is returned.

	{

	CImFindReplace* replacement = NULL;

	TInt index = iFieldList->Count();

	HBufC8* fieldName;

	TPtrC8 startOfLine;

	TInt lineIndex = 0;

	while (index--)

	// Look for each variable field in turn.

		{

		fieldName = ((*iFieldList)[index]->iFieldName);

		lineIndex = fieldName->Des().Size();

		if (lineIndex <= aLine.Size())

			{

			startOfLine.Set(aLine.Left(lineIndex));

			if (fieldName->Des().CompareF(startOfLine) == 0)

				{

				TPtrC8 fieldValue;

				fieldValue.Set(aLine.Mid(lineIndex, aLine.Size() - lineIndex));

				if (fieldValue.Size() > 1)

				// Don't include enclosing '<' '>' '"' characters in the value part.

					{

					if (((fieldValue[0] == '<') && (fieldValue[fieldValue.Size() - 1] == '>'))

						|| ((fieldValue[0] == '"') && (fieldValue[fieldValue.Size() - 1] == '"'))

						|| ((fieldValue[0] == '[') && (fieldValue[fieldValue.Size() - 1] == ']')))

						{

						fieldValue.Set(fieldValue.Mid(1, fieldValue.Size() - 2));

						}

					}

				// Set 

				if (fieldValue.Size() > 0)

					{

					replacement = new (ELeave) CImFindReplace;

					replacement->iFind = HBufC8::NewL(fieldValue.Size());

					*(replacement->iFind) = fieldValue;

					replacement->iReplace = HBufC8::NewL((*iFieldList)[index]->iReplacementString->Des().Size() + K_IM_TEST_ID_LEN);

					*(replacement->iReplace) = (*iFieldList)[index]->iReplacementString->Des();

					// Append the ID number to the replacement string.

					TPtr8 replacementString = replacement->iReplace->Des();

					TBuf8<K_IM_TEST_ID_LEN> numberString;

					(*iFieldList)[index]->iNumberFound++;

					numberString.Num((*iFieldList)[index]->iNumberFound);

					if ((replacementString.Size() != 0) && (index != 0))

						replacementString.Append(numberString);

					}

				}

			}

		}

	return replacement;

	}

void CScriptTestUtils::ReplaceVariableFields(TDes8& aLine)

// Replaces any variable data in the given descriptor with a standard equivalent.

	{

	TInt index = 0;

	index = iFindReplaceList->Count();

	TInt stringPosition;

	HBufC8* findString;

	HBufC8* replaceString;

	while (index--)

	// Search for each bit of variable data in turn

		{

		findString = (*iFindReplaceList)[index]->iFind;

		replaceString = (*iFindReplaceList)[index]->iReplace;

		stringPosition = aLine.FindF(findString->Des());

		if (stringPosition != KErrNotFound)

			{

			aLine.Replace(stringPosition, findString->Des().Size(), replaceString->Des());

			}

		}

	}

void CScriptTestUtils::StandardiseVariableFieldsL(const TDesC& aSourceFileName)

// Write over all the MIME variable field values with standard values.

	{

	TInt index = 0;

	index = iFieldList->Count();

	while (index--)

		{

		(*iFieldList)[index]->iNumberFound = 0;

		}

	// Open and create the files

	RFile sourceFile;

	RFile tempFile;

	TFileName tempFileName;

	index = 0;

	while (aSourceFileName[index] != L'.')

		{

		tempFileName.Append(aSourceFileName[index]);

		index++;

		}

	tempFileName.Append(_L(".tmp"));

	TInt err = tempFile.Replace(iTestUtils.FileSession(), tempFileName, EFileShareAny);

	// Open the log file.

	err = sourceFile.Open(iTestUtils.FileSession(), aSourceFileName, EFileShareReadersOnly);

	TFileReader fileReader(sourceFile);

	HBufC8* sourceLine = GetLineL(fileReader);

	CImFindReplace* findReplace;

	while (0 != sourceLine)

	// Fill in the find and replace table.

		{

		TPtr8 linePtr = sourceLine->Des();

		CleanupStack::PushL(sourceLine);

		findReplace = FindVariableDataL(linePtr);

		if (findReplace)

		// Add the find and replace strings to the list

			{

			iFindReplaceList->AppendL(findReplace);

			}

		CleanupStack::PopAndDestroy(); // sourceLine

		sourceLine = GetLineL(fileReader);

		}

	// Go back to the begining of the source file.

	fileReader.Seek(0);

	sourceLine = GetLineL(fileReader);

	// Replace all of the variable field values with the appropriate fixed string.

	while (0 != sourceLine)

		{

		TPtr8 linePtr = sourceLine->Des();

		CleanupStack::PushL(sourceLine);

		TPtr8 sourceLinePtr = sourceLine->Des();

		ReplaceVariableFields(sourceLinePtr);

		tempFile.Write(sourceLinePtr);

		tempFile.Write(K_IM_CR_LF);

		CleanupStack::PopAndDestroy(); // sourceLine

		sourceLine = GetLineL(fileReader);

		if (sourceLine)

			linePtr = sourceLine->Des();

		}

	// Close the files.

	sourceFile.Close();

	tempFile.Close();

	// Delete the source file.

	err = iTestUtils.FileSession().Delete(aSourceFileName);

	// Rename the temp file as the source file.

	err = iTestUtils.FileSession().Rename(tempFileName, aSourceFileName);

	}

void CScriptTestUtils::StripLogFileL(const TFileName& aSourceFileName, const TFileName& aDestFileName, TBool aStandardiseDateFields, TImLogLineType aLogLineType)

	{

	TBuf8<2> crlf = _L8("\r\n");

	RFile rSourceFile;

	RFile rOutputFile;

	User::LeaveIfError(rOutputFile.Replace(iTestUtils.FileSession(), aDestFileName, EFileShareAny));

	// Open the log file.

	User::LeaveIfError(rSourceFile.Open(iTestUtils.FileSession(), aSourceFileName, EFileShareAny));

	TBuf8<2048> scriptLine;

	TFileReader fileReader(rSourceFile);

	// Copy the incoming data from the log to the script file, removing any formatting.

	TInt lineCounter = 0;

	TInt lineIndex;

	TImLogLineType lastLineType = aLogLineType;

	TBool waitingForCrLf = EFalse;

	HBufC8* logLine = GetLineL(fileReader);

	while (logLine != 0)

		{

		CleanupStack::PushL(logLine);

		lineCounter++;

		lineIndex = 0;

		scriptLine.Zero();

		TImLogLineType lineType = LineType(logLine);

		if (EImEmptyLine == lineType)

			{

			if (waitingForCrLf)

				{

				waitingForCrLf = EFalse;

				rOutputFile.Write(crlf);

				}

			}

		else if (aLogLineType == lineType)

			{

			if ((lastLineType != EImLogLineComment) && (waitingForCrLf))

				{

				rOutputFile.Write(crlf);

				}

			// Copy the line of incoming data to the script file.

			if (logLine->Size() > 21)

				{

				// Copy the logLine from character 20 to the end into the output line.

				lineIndex = 21;

				while (lineIndex < logLine->Size())

					{

					scriptLine.Append((*logLine)[lineIndex]);

					lineIndex++;

					}

				// Write the stripped line to the script file

				User::LeaveIfError(rOutputFile.Write(scriptLine));

				}

			waitingForCrLf = ETrue;

				// 

			}

		else if (EImLogLineComment == lineType)

			{}

		else if (EImReadCancelled == lineType)

			{

			if (EImLogLineIn == aLogLineType)

				{

				User::LeaveIfError(rOutputFile.Write(K_IM_READ_CANCELLED));

				rOutputFile.Write(crlf);

				}

			}

		else

			{

			// If we're waiting for a crlf then write one regardless of the last line...

			if (waitingForCrLf)

				{

				rOutputFile.Write(crlf);

				}

			}

		lastLineType = LineType(logLine);

		CleanupStack::PopAndDestroy(); // logLine

		logLine = GetLineL(fileReader);

		}

	if (waitingForCrLf)

		{

		rOutputFile.Write(crlf);

		}

	rSourceFile.Close();

	rOutputFile.Close();

	if (aStandardiseDateFields)

		{

		StandardiseVariableFieldsL(aDestFileName);

		}

	}

EXPORT_C CScriptTestUtils* CScriptTestUtils::NewLC(CEmailTestUtils& aTestUtils)

	{

	CScriptTestUtils* self = new (ELeave) CScriptTestUtils(aTestUtils);

	CleanupStack::PushL(self);

	self->ConstructL();

	return self;

	}

void CScriptTestUtils::ConstructL()

	{

	iFieldList =  new (ELeave) CArrayPtrFlat<CImTestField>(6);

	CImTestField* removeReplaceItem = new (ELeave) CImTestField;

	removeReplaceItem->iFieldName = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iFieldName) = K_IM_TEST_DATE;

	removeReplaceItem->iReplacementString = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iReplacementString) = K_IM_TEST_DATE_REPLACEMENT;

	removeReplaceItem->iNumberFound=0;

	iFieldList->AppendL(removeReplaceItem);

	removeReplaceItem = new (ELeave) CImTestField;

	removeReplaceItem->iFieldName = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iFieldName) = K_IM_TEST_MESSAGE_ID;

	removeReplaceItem->iReplacementString = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iReplacementString) = K_IM_TEST_MESSAGE_ID_REPLACEMENT;

	removeReplaceItem->iNumberFound=0;

	iFieldList->AppendL(removeReplaceItem);

	removeReplaceItem = new (ELeave) CImTestField;

	removeReplaceItem->iFieldName = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iFieldName) = K_IM_TEST_CONTENT_ID;

	removeReplaceItem->iReplacementString = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iReplacementString) = K_IM_TEST_CONTENT_ID_REPLACEMENT;

	removeReplaceItem->iNumberFound=0;

	iFieldList->AppendL(removeReplaceItem);

	removeReplaceItem = new (ELeave) CImTestField;

	removeReplaceItem->iFieldName = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iFieldName) = K_IM_TEST_BOUNDARY;

	removeReplaceItem->iReplacementString = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iReplacementString) = K_IM_TEST_BOUNDARY_REPLACEMENT;

	removeReplaceItem->iNumberFound=0;

	iFieldList->AppendL(removeReplaceItem);

	removeReplaceItem = new (ELeave) CImTestField;

	removeReplaceItem->iFieldName = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iFieldName) = K_IM_TEST_HELO;

	removeReplaceItem->iReplacementString = HBufC8::NewL(K_IM_TEST_STRING_MAX);

	*(removeReplaceItem->iReplacementString) = K_IM_TEST_HELO_REPLACEMENT;

	removeReplaceItem->iNumberFound=0;

	iFieldList->AppendL(removeReplaceItem);

	iFindReplaceList = new (ELeave) CArrayPtrFlat<CImFindReplace>(6);

	}

CScriptTestUtils::CScriptTestUtils(CEmailTestUtils& aTestUtils) : iTestUtils(aTestUtils)

	{}

CScriptTestUtils::TImLogLineType CScriptTestUtils::LineType(const HBufC8* aLine)

	{

	TImLogLineType lineType = EImLogLineComment;

	if (aLine->Size() == 0)

		{

		lineType = EImEmptyLine;

		}

	else if (aLine->Size() > 19)

		{

		if ((((*aLine)[18]) == '<')

			&& (((*aLine)[19]) == '<'))

			{

			lineType = EImLogLineIn;

			}

		else if ((((*aLine)[18]) == '>')

			&& (((*aLine)[19]) == '>'))

			{

			lineType = EImLogLineOut;

			}

		else if ((aLine->Right(17)).Compare(K_IM_READ_CANCELLED) == 0)

			{

			lineType = EImReadCancelled;

			}

		}

	return lineType;

	}

HBufC8* CScriptTestUtils::GetLineL(TFileReader& aFileReader)

	{

	TBuf8<2048> currentLine;

	char character;

	currentLine.Zero();

	TBool crlfFound = EFalse;

	TBool read = ETrue;

	// Append characters one at a time until we reach a cr/lf pair or the end of the file.

	while ((read) && (crlfFound == EFalse))

		{

		read = aFileReader.Read(character);

		if (read)

			{

			if (0x0D == character)

				{

				read = aFileReader.Read(character);

				if (read)

					{

					if (character == 0x0A)

						{

						crlfFound = ETrue;

						}

					else

						{

						// The cr was not part of the cr/lf pair so go back one space.

						aFileReader.SeekCurrent(-1);

						}

					}

				}

			if ((read) && (EFalse == crlfFound))

				{

				currentLine.Append(character);

				}

			}

		}

	HBufC8* lineCopy;

	if ((currentLine.Size() != 0) || (read))

		{

		lineCopy = HBufC8::NewL(2048);

		(*lineCopy) = currentLine;

		}

	else

		{

		lineCopy = 0;

		}

	return lineCopy;

	}

CScriptTestUtils::TFileReader::TFileReader(RFile &rFile) : iFile(rFile), iBufferIndex(0), iBufferStart(0)

	{

	}

TBool CScriptTestUtils::TFileReader::Read(char &rChar)

	{

	// Is the character we want not in the buffer ?

	TBool read = ETrue;

	if ((iBufferIndex >= iBuffer.Size()) 

		|| (iBufferIndex < 0))

		// If it's not then get a new buffer starting at this position.

		{

		TInt seekPosition = iBufferIndex + iBufferStart;

		TInt err = iFile.Seek(ESeekStart, seekPosition);

		if (KErrNone != err)

			{

			iBuffer.SetLength(0);

			read = EFalse;

			}

		else

			{

			iFile.Read(iBuffer);

			if (iBuffer.Size() == 0)

				{

				read = EFalse;

				}

			iBufferStart += iBufferIndex;

			iBufferIndex = 0;

			}

		if (iBuffer.Size() > 0)

			{

			rChar = iBuffer[0];

			}

		}

	else

		{

		rChar = iBuffer[iBufferIndex];

		}

	if (read)

		{

		iBufferIndex++;

		}

	return read;

	}

void CScriptTestUtils::TFileReader::Seek(TInt aOffset)

	{

	iBufferIndex = aOffset;

	iBuffer.Zero();

	iBufferStart = 0;

	}

void CScriptTestUtils::TFileReader::SeekCurrent(TInt aOffset)

	{

	iBufferIndex += aOffset;

	}