// 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 <e32std.h>
#include <e32test.h>
#include <s32file.h>

#include <push/pushmessage.h>
#include <push/cmultipartbiniterator.h>
#include <push/cmultiparttextiterator.h>

#include <push/pushlog.h>
#include "testlog.h"
#include "t_pushmessage.h"

/** literals */
_LIT8(KWBContentType, "X-WBML-Content-Type: ");
_LIT8(KHeaderBoundary, "\r\n\r\n");
_LIT(KTestTitle,"T_PushMessage");
TChar KEOL('\n');
TChar KCOMMA(',');
TChar KSEMICOLON(';');

/** global variables */
RTest gTest(KTestTitle);
RFs gFs;



/** 
factory construction method to allocate and construct a new CPushMessageTester 
@param aWapLog Implementation of logging interface
*/
CPushMessageTester* CPushMessageTester::NewLC(CWapPushLog& aWapLog)
	{
	CPushMessageTester* self =  new (ELeave) CPushMessageTester(aWapLog);
	CleanupStack::PushL(self);
	return self;
	}

/** 
factory construction method to allocate and construct a new CPushMessageTester 
@param aWapLog Implementation of logging interface
*/
CPushMessageTester* CPushMessageTester::NewL(CWapPushLog& aWapLog)
	{
	CPushMessageTester* self = CPushMessageTester::NewLC(aWapLog);
	CleanupStack::Pop();
	return self;
	}

/** 
constructor 
@param aWapLog Implementation of logging interface
*/
CPushMessageTester::CPushMessageTester(CWapPushLog& aWapLog)
:iLog(aWapLog)
	{
	}

/** destructor */
CPushMessageTester::~CPushMessageTester()
	{
	delete iPushMessage1;   
	delete iPushMessage2;
	delete iPushMessage3;
	delete iPushMessage4;
	delete iPushMessage5;
	delete iPushMessage6;
	delete iPushMessage7;
	delete iPushMultiMessage;
	}


/* creates a descriptors containing the data from arrays */
HBufC8* CPushMessageTester::CreateMsgDataLC(const TUint8* aMessage, TInt aLength)
	{
	HBufC8* msgData = HBufC8::NewLC(aLength);
	TBuf8<140> tempBuf;
	TUint8 value=0;
	
	for (TInt i=0; i<aLength; i++)
		{
		value = aMessage[i];
		tempBuf.Append(value);
		}

	msgData->Des().Copy(tempBuf);
	return msgData;
	}


/** create push messages for the tests */
void CPushMessageTester::CreateMessagesL()
	{
	__LOG_ALWAYS("Test Results");
	__LOG_ALWAYS(" ");

	HBufC8* header1 = CreateMsgDataLC(KMessageHeader2,15);
	HBufC8* body1 = KLongPushMessageBody().AllocLC();
	HBufC8* address1 = KPushMessageServerAddress().AllocLC();

	iPushMessage1 = CPushMessage::NewL(header1, body1, 22, address1);
	CleanupStack::Pop(3, header1); //header1, body1, address1

	HBufC8* header2 = CreateMsgDataLC(KMessageHeader3,9);
	HBufC8* body2 = KShortPushMessageBody().AllocLC();
	iPushMessage2 = CPushMessage::NewL(header2, body2);
	CleanupStack::Pop(2, header2); //header2, body2
	
	HBufC8* header3 = CreateMsgDataLC(KMessageHeader4,32);
	HBufC8* body3 = KShortPushMessageBody().AllocLC();
	iPushMessage3 = CPushMessage::NewL(header3, body3);
	CleanupStack::Pop(2, header3); //header3, body3
	
	HBufC8* header4 = CreateMsgDataLC(KMessageHeaderMultipart,42);
	HBufC8* body4 = CreateMsgDataLC(KMessageBodyMultipart,42);
	iPushMultiMessage = CPushMessage::NewL(header4, body4);
	CleanupStack::Pop(2, header4); // header4, body4

	HBufC8* header5 = CreateMsgDataLC(KMessageHeader5,22);
	HBufC8* body5 = KLongPushMessageBody().AllocLC();
	HBufC8* address2 = KPushMessageServerAddress().AllocLC();
	iPushMessage4 = CPushMessage::NewL(header5, body5, 255, address2);
	CleanupStack::Pop(3, header5); // header5, body5, address2

	HBufC8* header6 = CreateMsgDataLC(KMessageHeader6,51);
	HBufC8* body6 = KLongPushMessageBody().AllocLC();
	iPushMessage5 = CPushMessage::NewL(header6, body6);
	CleanupStack::Pop(2, header6); // header6, body6

	HBufC8* header7 = CreateMsgDataLC(KMessageHeader7,9);
	HBufC8* body7 = KLongPushMessageBody().AllocLC();
	iPushMessage6 = CPushMessage::NewL(header7, body7);
	CleanupStack::Pop(2, header7); // header7, body7

	HBufC8* header8 = CreateMsgDataLC(KMessageHeader8,139);
	HBufC8* body8 = KLongPushMessageBody().AllocLC();
	iPushMessage7 = CPushMessage::NewL(header8, body8);
	CleanupStack::Pop(2, header8); // header8, body8
	}


/** dump relevant bits of the message to the console */
void CPushMessageTester::OutputMessageDataL()
	{
	__LOG_MSG_DEBUG(*iPushMessage1);
	__LOG_MSG_DEBUG(*iPushMessage2);
	__LOG_MSG_DEBUG(*iPushMessage3);
	__LOG_MSG_DEBUG(*iPushMessage4);
	__LOG_MSG_DEBUG(*iPushMessage5);
	__LOG_MSG_DEBUG(*iPushMessage6);
	__LOG_MSG_DEBUG(*iPushMessage7);

	TPtrC contentType;
	iPushMultiMessage->GetContentType(contentType);

	_LIT(KMultipartMixedBin, "application/vnd.wap.multipart.mixed");
	if (contentType==KMultipartMixedBin)
		{
		CMultipartBinIterator* multiMsg = CMultipartBinIterator::NewL(*iPushMultiMessage);
		CleanupStack::PushL(multiMsg);
		do 
			{
			CPushMessage * pm = multiMsg->PartL();
			CleanupStack::PushL(pm);
			__LOG_MSG_DEBUG(*pm);
			CleanupStack::PopAndDestroy(pm); 
			} while (multiMsg->NextL());
		CleanupStack::PopAndDestroy(multiMsg); 
		}
	}






/** create multipart messages from text? */
void CPushMessageTester::TestMultiTextMsgFromFilesL()
	{
	_LIT(KPushTxtFilePath, "C:\\tpushmessage\\"); 
	CDir* dir;
	TInt err = gFs.GetDir(KPushTxtFilePath, KEntryAttMatchMask, ESortByName, dir);
	CleanupStack::PushL(dir);
	if (err == KErrPathNotFound)
		User::LeaveIfError(gFs.MkDirAll(KPushTxtFilePath));		
	else 
		User::LeaveIfError(err);
	CleanupStack::PopAndDestroy(dir);


	// Set the session path for the RFs
    gFs.SetSessionPath(KPushTxtFilePath);
	_LIT(KPushTxtFilePattern,"*.txt");
	dir=NULL;
	User::LeaveIfError(gFs.GetDir(KPushTxtFilePattern, KEntryAttNormal, ESortByName, dir));
	CleanupStack::PushL(dir);
    TInt count = dir->Count();
    if(count != 0)
        {
		TFileName currentFile;
		TBuf8<256> msgType;
		for(TInt i = 0; i < count; i++)
			{
			currentFile=( (*dir)[i].iName );
			GenMessageTypeL(currentFile, msgType);

			//if (msgType[0] == 0x8C && msgType[0] <= 0x8E) 
				{
				CPushMessage* pushMessage = CreateMultiTextPushMessageL(currentFile, msgType);
				CleanupStack::PushL(pushMessage);

				CMultipartTextIterator* multiMsg = CMultipartTextIterator::NewL(*pushMessage);
				CleanupStack::PushL(multiMsg);
				do 
					{
					CPushMessage * pm = multiMsg->PartL();
					__LOG_MSG_DEBUG(*pm);
					delete pm;
					} while (multiMsg->NextL());
				CleanupStack::PopAndDestroy(2); //multiMsg, pushMessage
				}
			}
		}
	CleanupStack::PopAndDestroy(dir);
    }




void CPushMessageTester::GenMessageTypeL(const TDesC& aFileName, TDes8& rMsgType)
	{
	TBuf8<256>		lineBuffer;   // Buffer from Read function.
	TInt			aPos = 0;

	HBufC8*			contentType = NULL;

	RFile			file;
	TInt err = file.Open(gFs,aFileName,EFileStreamText|EFileRead|EFileShareAny);

	 // Didn't find the file, so leave - should only get valid filenames!
	if(err != KErrNone)  
        User::Leave(KErrNotFound);


	file.Seek(ESeekAddress,aPos);
	err = file.Read(lineBuffer, 256);// Read upto 256 chars, '\n' and all...
	if (err==KErrNone)				// Made a valid read,
		if (lineBuffer.Length()==0) // but read 0 chars
			err=KErrEof;			// so set err value to end processing

	if(err == KErrNone)
		{
		// now we'ver read in a line 
		TInt ctOffset = lineBuffer.FindF(KWBContentType);
		if (ctOffset != KErrNotFound)
			{
			contentType = lineBuffer.Mid(ctOffset).AllocL();
			// we've found content-type ... make sure we've read the whole line
			TInt eolOffset = contentType->Locate(KEOL);
			if (ctOffset == KErrNotFound)
				{
				// must get the rest of the line
				err = file.Read(lineBuffer, 256);// Read upto 256 chars, '\n' and all...
				eolOffset = contentType->Locate(KEOL);
				contentType = contentType->ReAllocL(contentType->Length() + eolOffset);
				contentType->Des().Append(lineBuffer.Mid(eolOffset));
				}
			}
		}
	
	// set ContentType the whole  WBContent line 
	TInt eolOffset = contentType->Locate(KEOL);
	TPtrC8 wbContentType = contentType->Mid(KWBContentType().Length(), eolOffset - KWBContentType().Length() - 1);

	// First value is the binary value of content type;
	TInt commaOffset = wbContentType.Locate(KCOMMA);
	TPtrC8	binType = wbContentType.Mid(0, commaOffset);

	TLex8 ctTypeLex(binType);
	TUint8 ctVal = 0;
	ctTypeLex.Val(ctVal, EHex);

	TBuf8<256> binHeader;
	binHeader.Zero();
	binHeader.Append(ctVal);

	// is this a multipart/
	if (ctVal >= 0x8C && ctVal <= 0x8E)
		{
		// next field is boundary
		TPtrC8	boundary = wbContentType.Mid(commaOffset+1);
		binHeader.Append(KBoundaryParam);
		binHeader.Append(0);
		binHeader.Append(boundary);
		binHeader.Append(0);
		}

	rMsgType.Zero();

	TInt len = binHeader.Length();
	if (len <= 30)
		rMsgType.Append(len);
	else if (len < 128)
		{
		// make sure the top bit isn't set
		TUint8 wapUint = 31;
		len &= 0x7F;
		rMsgType.Append(wapUint);
		rMsgType.Append(len);
		}
	else
		{
		

		}

	rMsgType.Append(binHeader);
	delete contentType;
	file.Close();
	}


CPushMessage* CPushMessageTester::CreateMultiTextPushMessageL(const TDesC& aFileName, const TDesC8& aHdr)
	{
	RFile          file;
	TBuf8<256>     lineBuffer;   // Buffer from Read function.
	TInt		   err=KErrNone;
	TInt		   aPos = 0;
	HBufC8*	headers = NULL;
	HBufC8*	body = NULL;
	TBool		bBodyFound = EFalse;

	err = file.Open(gFs,aFileName,EFileStreamText|EFileRead|EFileShareAny);

    if(err != KErrNone)  // Didn't find the file, so leave - should only get valid filenames!
        {
        User::Leave(KErrNotFound);
		}

	headers = aHdr.AllocLC();

	file.Seek(ESeekAddress,aPos);
	err = file.Read(lineBuffer, 256);// Read upto 256 chars, '\n' and all...
	if (err==KErrNone)				// Made a valid read,
		if (lineBuffer.Length()==0) // but read 0 chars
			err=KErrEof;			// so set err value to end processing

	if(err == KErrNone)
	do {
		// now we'ver read in a line 
		TInt bdOffset = lineBuffer.FindF(KHeaderBoundary);
		if ( !bBodyFound && bdOffset == KErrNotFound)
			{
			HBufC8* oldHeaders = headers;
			headers = headers->ReAllocL(headers->Length() + lineBuffer.Length());
			CleanupStack::Pop(oldHeaders);
			CleanupStack::PushL(headers);

			headers->Des().Append(lineBuffer);
			}
		
		else // we've found our break
			{
			if (!bBodyFound)
				{
				HBufC8* oldHeaders = headers;
				headers = headers->ReAllocL(headers->Length() + lineBuffer.Length() - bdOffset );
				CleanupStack::Pop(oldHeaders);
				CleanupStack::PushL(headers);
				
				headers->Des().Append(lineBuffer.Mid(0, lineBuffer.Length() - bdOffset));

				body = lineBuffer.Mid(bdOffset+2, lineBuffer.Length() - bdOffset-2).AllocLC();
				bBodyFound = ETrue;
				}

			else 
				{
				HBufC8* oldBody = body;
				body = body->ReAllocL(body->Length() + lineBuffer.Length() );
				CleanupStack::Pop(oldBody);
				CleanupStack::PushL(body);

				body->Des().Append(lineBuffer);
				}

			}
		err = file.Read(lineBuffer, 256);// Read upto 256 chars, '\n' and all...
		} while ( lineBuffer.Length());

	    
	file.Close();
	CPushMessage* rPush = CPushMessage::NewL(headers, body);
	if (body == NULL)
		CleanupStack::Pop(headers);
	else
		CleanupStack::Pop(2, headers); //headers, body
	return rPush;
	}

void CPushMessageTester::TestFinished()
	{
	__LOG_ALWAYS(" ");
	__LOG_ALWAYS("Tests Completed");
	}

/** what does this do? - i assume this was used to create messages */
LOCAL_C void CalcTimePeriod()
	{
	// Get num bytes encoding the date - 
	// we are positioned at that location in the source descriptor
	// TInt32 time = 0;

	// The WSP Date encoding is the number of seconds since the start of the
	// UNIX epoch (00:00:00.000, 01-Jan-1970), as a long integer
	TDateTime targetDT(2000, EAugust, 15,10,20,0,0);
	TDateTime unixEpocDT(1970, EJanuary, 0, 0, 0, 0, 0);
	TTime targetTime(targetDT);
	TTimeIntervalSeconds timeSeconds;

	targetTime.SecondsFrom(unixEpocDT,timeSeconds);

	timeSeconds.Int();
	}




/** Setup test environment and run tests */
LOCAL_C void doMainL()
	{
	gTest.Start(KTestTitle);
			
	gTest.Printf(_L("@SYMTestCaseID IWS-WAPBROWSER-PUSHUTILS-T_PUSHMESSAGE-0001 "));	
	
	CWapPushLog* log = CWapPushLog::NewL(*gTest.Console());
	log->SetLogFileName(_L("TPushMessage.txt"));
	CleanupStack::PushL(log);
	User::LeaveIfError(gFs.Connect());
	CleanupClosePushL(gFs);
	CalcTimePeriod();

	CPushMessageTester* myTester = CPushMessageTester::NewLC(*log);
	myTester->CreateMessagesL();
	myTester->OutputMessageDataL();
	myTester->TestMultiTextMsgFromFilesL();
	myTester->TestFinished();
	
	CleanupStack::PopAndDestroy(3); //gFs.Close(), log, myTester
	gTest.End();
	gTest.Close();
	}



/** Entry point to test executable */
GLDEF_C TInt E32Main()
	{	
	__UHEAP_MARK;
	CTrapCleanup* theCleanup = CTrapCleanup::New();
	TRAPD(ret,doMainL());		
	gTest(ret==KErrNone);
	delete theCleanup;	
	__UHEAP_MARKEND;
	return(KErrNone);
	}
