// Copyright (c) 2001-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:

//

// System includes

#include <http/framework/mrxdataobserver.h>

#include <http/mhttpdatasupplier.h>

#include <http/rhttptransaction.h>

#include <wsperror.h>

// User includes

#include "mwspcorxdatacallback.h"

#include "cwspcotransaction.h"

#include "cwspheaderutils.h"

#include "wsppanic.h"

// Class signature

#include "cwspcorxdata.h"

CWspCORxData* CWspCORxData::NewL(

								CProtTransaction&		aTransaction, 

								MRxDataObserver&		aObserver, 

								MWspCORxDataCallback&	aMethodCallback

								)

	{

	return new (ELeave) CWspCORxData(aTransaction, aObserver, aMethodCallback);

	}

CWspCORxData::CWspCORxData(

						  CProtTransaction&			aTransaction, 

						  MRxDataObserver&			aObserver, 

						  MWspCORxDataCallback&		aMethodCallback

						  )

: CRxData(aTransaction, aObserver), iMethodCallback(aMethodCallback), iOverallDataSize(KErrNotFound)

	{

	}

CWspCORxData::~CWspCORxData()

	{

	}

void CWspCORxData::SetResponseDataL(const TDesC8& aResponseHeaders, MHTTPDataSupplier& aResponseBody, TBool aMoreResponseData)

	{

	// Store the more data flag

	iMoreData = aMoreResponseData;

	// Set the header data

	SetHeaderDataL(aResponseHeaders);

	// Set the body data

	SetBodyDataL(aResponseBody);

	// Is the response complete? Is there any data to be passed to the client?

	if( !iMoreData && iBodyData.Length() == 0 )

		{

		// Have received the entire response - inform the client.

		ResponseCompleteL();

		}

	}

void CWspCORxData::UpdateResponseDataL(const TDesC8& aTrailerHeaders, TBool aMoreResponseData)

	{

	__ASSERT_DEBUG( iMoreData, Panic(KWspPanicNotExpectingMoreResponseData) );

	// Store the more data flag

	iMoreData = aMoreResponseData;

	// Is there trailer headers?

	if( aTrailerHeaders.Length() > 0 )

		{

		// Is this the last primitive

		if( iMoreData )

			{

			// This is not the last primitive so no trailer headers allowed.

			User::Leave(KWspErrTrailerHeadersNotExpected);

			}

		// Update the header data

		UpdateHeaderDataL(aTrailerHeaders);

		}

	// Update the body data

	UpdateBodyDataL();

	// Is the response complete? Is there any data to be passed to the client?

	if( !iMoreData && iBodyData.Length() == 0 )

		{

		// Have received the entire response - inform the client.

		ResponseCompleteL();

		}

	}

void CWspCORxData::SetHeaderDataL(const TDesC8& aResponseHeaders)

	{

	// Was any header data received in this primitive.

	if( aResponseHeaders.Length() > 0 )

		{

		// Segment the header data into the separate header fields in the response

		// object's headers.

		RHTTPTransaction trans	= iProtTrans->Transaction();

		RHTTPHeaders headers	= trans.Response().GetHeaderCollection();

		STATIC_CAST(CWspCOTransaction*, 

					iProtTrans)->GetWspHeaderUtils().DecodeReplyHeadersL(

																		trans.Session().StringPool(),

																		aResponseHeaders,

																		headers

																		);

		}

	// Inform the client that all the response headers have been received

	RHTTPTransaction tr = iProtTrans->Transaction();

	tr.SendEventL(

				 THTTPEvent::EGotResponseHeaders,

				 THTTPEvent::EIncoming, 

				 THTTPFilterHandle(THTTPFilterHandle::EProtocolHandler)

				 );

	}

void CWspCORxData::SetBodyDataL(MHTTPDataSupplier& aResponseBody)

	{

	__ASSERT_DEBUG( !iGotBodyData, Panic(KWspPanicRxDataObjectNotReset) );

	// Store the body data supplier

	iBodyDataSupplier = &aResponseBody;

	// Update the overall data size

	iOverallDataSize = iBodyDataSupplier->OverallDataSize();

   	// Add the body data supplier to the response object

   	iProtTrans->Transaction().Response().SetBody(*this);

	// Get the data...

	UpdateBodyDataL();

	}

void CWspCORxData::UpdateHeaderDataL(const TDesC8& aTrailerHeaders)

	{

	// This is the last S-MethodResultData primitive and there is header data -

	// these are probably Trailer Headers.

	// Segment the header data into the separate header fields in the response

	// object's headers.

	RHTTPTransaction trans	= iProtTrans->Transaction();

	RHTTPHeaders headers	= trans.Response().GetHeaderCollection();

	STATIC_CAST(CWspCOTransaction*, 

				iProtTrans)->GetWspHeaderUtils().DecodeHeadersL(

															   trans.Session().StringPool(),

															   aTrailerHeaders,

															   headers

															   );

	iHasTrailer = ETrue;

	}

void CWspCORxData::UpdateBodyDataL()

	{

	__ASSERT_DEBUG( !iGotBodyData, Panic(KWspPanicResponseDataNotReleased) );

	__ASSERT_DEBUG( iBodyDataSupplier, Panic(KWspPanicNoResponseDataReceived) );

	// Is there more data for this SDU?

	iLastChunk = iBodyDataSupplier->GetNextDataPart(iBodyData);

	// Check to see there is some data

	if( iBodyData.Length() == 0 )

		{

		// Are there more chunks? Keep trying to there is some data..

		while( !iLastChunk && iBodyData.Length() == 0 )

			{

			// Release the data in the body data supplier

			iBodyDataSupplier->ReleaseData();

			// Get the next chunk

			iLastChunk = iBodyDataSupplier->GetNextDataPart(iBodyData);

			}

		// Was some data received?

		if( iBodyData.Length() == 0 )

			{

			// Release the data in the body data supplier

			iBodyDataSupplier->ReleaseData();

			// No body data received - is this the last primitive? It is possible

			// to have no body data if the last primitive had the trailer headers

			// only.

			if( iMoreData )

				{

				// Something has gone wrong - more data expected, but none sent in 

				// this SDU. Response must be corrupt.

				User::Leave(KWspErrExpectingBodyData);

				}

			// Inform method callback to send the response to the primitive

			iMethodCallback.SendResponsePrimitive();

			// Nothing more to do

			return;

			}

		}

	// Set the flag to indicate that there is some response body data

	iGotBodyData = ETrue;

	// Inform the client that there is some (more) response data.

	RHTTPTransaction tr = iProtTrans->Transaction();

	tr.SendEventL(

				 THTTPEvent::EGotResponseBodyData,

				 THTTPEvent::EIncoming, 

				 THTTPFilterHandle(THTTPFilterHandle::EProtocolHandler)

				 );

	}

void CWspCORxData::ResponseCompleteL()

	{

	if( iHasTrailer )

		{

		// Inform the client that there are trailer headers

		RHTTPTransaction tr = iProtTrans->Transaction();

		tr.SendEventL(

					 THTTPEvent::EGotResponseTrailerHeaders,

					 THTTPEvent::EIncoming, 

					 THTTPFilterHandle(THTTPFilterHandle::EProtocolHandler)

					 );

		iHasTrailer = EFalse;

		}

	// Tell the observer that the response is complete - this object will

	// be deleted by this call, so is the last thing to be done.

	iObserver->SetStatusL(*this, THTTPEvent::EResponseComplete);

	}

/*

 * Methods from CRxData

 */

void CWspCORxData::ResetRxData()

	{

	User::Panic(KWspPanicCategory, KErrNotSupported);	

	}

/*

 * Methods from MHTTPDataSupplier

 */

TBool CWspCORxData::GetNextDataPart(TPtrC8& aDataPart)

	{

	__ASSERT_DEBUG( iGotBodyData, Panic(KWspPanicNoResponseDataReceived) );

	// Set the data part to the given chunk.

	aDataPart.Set(iBodyData);

	// Set the return value

	return (iLastChunk && !iMoreData);

	}

void CWspCORxData::ReleaseData()

	{

	__ASSERT_DEBUG( iGotBodyData, Panic(KWspPanicNoResponseDataReceived) );

	// Release the data in the body data supplier

	iBodyDataSupplier->ReleaseData();

	// Clear flag that indicates that there is a current data part.

	iGotBodyData = EFalse;

	// Has all the data for this SDU been passed to the client?

	TInt error = KErrNone;

	if( iLastChunk )

		{

		// Ok, all the response data has been received and the client has been 

		// passed all of it - it has just released the last batch. Inform method

		// callback to send the response to the primitive

		iMethodCallback.SendResponsePrimitive();

		// Are there subsequent S-MethodResultData primitives to follow?

		if( !iMoreData )

			{

			// No, have received all data from server - inform observer that the 

			// response is complete.

			TRAP(error, ResponseCompleteL());

			}

		}

	else

		{

		// No, data supplier still has got more - go get it.

		TRAP(error, UpdateBodyDataL());

		}

	// Any problems...?

	if( error != KErrNone )

		{

		// Could not send event - need to abort the method

		iMethodCallback.AbortResponse();

		}

	}

TInt CWspCORxData::OverallDataSize()

	{

	return iOverallDataSize;

	}

TInt CWspCORxData::Reset()

	{

	__ASSERT_DEBUG( iBodyDataSupplier, Panic(KWspPanicNoResponseDataReceived) );

	return iBodyDataSupplier->Reset();

	}