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

EXPORT_C  CMultipartTextIterator* CMultipartTextIterator::NewL(CPushMessage& aPushMessage)
/** Allocates and creates a new MIME multipart message iterator.

@param aPushMessage MIME multipart message to process
@return New MIME multipart message iterator */
	{
	CMultipartTextIterator* self =  new (ELeave) CMultipartTextIterator(aPushMessage);
	CleanupStack::PushL(self);
	self->ConstructL(); 
	CleanupStack::Pop();
	return self;
	}


/**
 * CMultipartTextMessage
 * @param aPushMessage A message determined to be text/multipart. 
 */
CMultipartTextIterator::CMultipartTextIterator(CPushMessage& aPushMessage) : 
	CMultipartIteratorBase(aPushMessage)
	{
	}

/**
 * Fully instantiates this class.  
 * Finds the boundary and sets the internal pointer to the first part
 */
void CMultipartTextIterator::ConstructL()  
	{
	BoundaryParamL();
	FirstL();
	}

EXPORT_C CMultipartTextIterator::~CMultipartTextIterator()
/** Destructor. */
	{
	delete iBoundary;
	}



/**
 * Finds the boundary from the WAP Binary Header.  
 * The actually boundary conforms to RFC 2061.  
 * @leave KErrCorrupt if the message does not contain a boundary
 * @leave HBufC8::NewL
*/
void CMultipartTextIterator::BoundaryParamL()
	{
	TPtrC contentType;
	iPushMessage.GetContentType(contentType);
	__ASSERT_DEBUG( contentType.FindF(KMultipartText) == 0 , User::Panic(KPushInvalidContentType,0));

	const TInt KMaxBoundaryLen(74);	// Max Boundary = 70 + preceeding "--" and at the end "--"
	TBuf8<KMaxBoundaryLen> cleanCtHdrValue;
	TPtrC8 ctHdrValue;
	TPtrC8 bndLabel(KBoundaryParam);
	iPushMessage.GetBinaryHeaderField(EHttpContentType, ctHdrValue);


	TInt paramOffset = ctHdrValue.FindF(KBoundaryParam);
	if (paramOffset != KErrNotFound)
		{
		ctHdrValue.Set(ctHdrValue.Mid(paramOffset + bndLabel.Length()));

		// remove the NULL terminators
		cleanCtHdrValue.Copy(ctHdrValue);
		TInt nullTerm = cleanCtHdrValue.Locate(0);
		for ( ; nullTerm != KErrNotFound; nullTerm = cleanCtHdrValue.Locate(0))
			cleanCtHdrValue.Delete(nullTerm,1);

		// remove start & end quotes
		if (cleanCtHdrValue[0] == '\"')
			{
			cleanCtHdrValue.Delete(0,1);
			TInt endIndex = cleanCtHdrValue.Length() - 1;
			if (cleanCtHdrValue[endIndex] == '\"')
				cleanCtHdrValue.Delete(endIndex,1);
			else
				User::Leave(KErrCorrupt);
			}
		}
	else
		User::Leave(KErrCorrupt);

	iBoundary = HBufC8::NewL(sizeof(KBoundaryStartEnd) - 1 + ctHdrValue.Length());
	iBoundary->Des().Copy(KBoundaryStartEnd);
	iBoundary->Des().Append(cleanCtHdrValue);	
	}


EXPORT_C TBool CMultipartTextIterator::NextL()
/** Moves the iterator to the next message part.

@return ETrue if there is a next part, EFalse otherwise */
	{
	TPtrC8 hdrPtr;
	TInt offset = 0;
	
	hdrPtr.Set(iIndexPtr);

	// find the boudary which starts our next part
	offset = hdrPtr.Find(*iBoundary);
	
	if (offset == KErrNotFound )
			return EFalse;

	// point to the end of the boudary string
	hdrPtr.Set(hdrPtr.Mid(offset + iBoundary->Length()));

	// did we encounter the last "--"
	if (hdrPtr.Find(KBoundaryStartEnd) == 0)
			return EFalse;

	// see if this is really the boundary.  Expecting to see \r\n, no spurious text following it
	if (hdrPtr.Find(KCRLF) != 0)
			return EFalse;
	
	// point to the beginning of the new part 
	TPtrC8 ptrCRLF(KCRLF);
	hdrPtr.Set(hdrPtr.Mid(ptrCRLF.Length()));
	iIndexPtr.Set(hdrPtr);

	return ETrue;
	}


EXPORT_C void CMultipartTextIterator::FirstL()
/** Resets the iterator the beginning of the first part of the message. */
	{
	// Set iIndexPtr to the start of the body
	iPushMessage.GetMessageBody(iIndexPtr);

	// need to call this to get past the first boundary
	NextL();
	}


EXPORT_C CPushMessage*	CMultipartTextIterator::PartL()
/** Gets the message part that the iterator is currently referencing.

@return Message part */
	{
	// iIndexPtr should be pointing to line after the boudary

	// check if there are boudary headers
	TPtrC8 hdrSep(KHdrSeparator);
	TPtrC8 hdrStart(iIndexPtr);

	TInt offset = hdrStart.Find(KHdrSeparator);
	if (offset == KErrNotFound)
		User::Leave(KErrNotFound);
	
	HBufC8* headerBuf = hdrStart.Mid(0, offset).AllocLC();

	// now find the body
	TPtrC8 bodyStart;
	bodyStart.Set(hdrStart.Mid(offset + hdrSep.Length()));

	offset = bodyStart.Find(*iBoundary);
	if (offset == KErrNotFound)
		User::Leave(KErrNotFound);

	HBufC8* bodyBuf = bodyStart.Mid(0, offset).AllocLC();
	
	CPushMessage* pm = CPushMessage::NewL( headerBuf,  bodyBuf);
	CleanupStack::Pop(2, headerBuf);  // bodyBuf, headerBuf
	return pm;
	}