// Copyright (c) 2008-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 <upnp/tupnptable.h>

#include <inetprottextutils.h>

#include <upnp/upnpconstants.h>

#include "cupnpheaderwriter.h"

#include "CHeaderField.h"

const TInt KMaxNumDigits = 10;

_LIT8(KTimeoutSecond, "Second-");

_LIT8(KSemiSpaceSep,"; ");

_LIT8(KQuotes, "\"");

_LIT8(KOpenAngleBracket, "<");

_LIT8(KCloseAngleBracket, ">");

CUPnPHeaderWriter* CUPnPHeaderWriter::NewL(RStringPool aStringPool)

	{

	return new (ELeave) CUPnPHeaderWriter(aStringPool);

	}

CUPnPHeaderWriter::~CUPnPHeaderWriter()

	{

	// do nothing.

	}

void CUPnPHeaderWriter::EncodeHeaderL(RHeaderField& aHeader)

	{

	RStringF fieldStr = iStringPool.StringF(aHeader.Name());

	switch( fieldStr.Index(TUPnPTable::Table()) )

		{

		case UPnP::EST:

			EncodeGenericUpnpHeaderL(aHeader, KErrUPnPEncodeST);

			break;

		case UPnP::EUSN:

			EncodeGenericUpnpHeaderL(aHeader, KErrUPnPEncodeUSN);

			break;

		case UPnP::ETimeout:

			EncodeTimeoutHeaderL(aHeader, KErrUPnPEncodeTimeout);

			break;

		case UPnP::EExt:

			EncodeExtL(aHeader);

			break;

		case UPnP::ESID:

			EncodeGenericUpnpHeaderL(aHeader, KErrUPnPEncodeSID);

			break;

		case UPnP::ENTS:

			EncodeGenericUpnpHeaderL(aHeader, KErrUPnPEncodeNTS);	

			break;

		case UPnP::ENT:

			EncodeGenericUpnpHeaderL(aHeader, KErrUPnPEncodeNT);

			break;

		case UPnP::EMX:

			EncodeGenericNumberHeaderL(aHeader, KErrUPnPEncodeMX);	

			break;

		case UPnP::ESEQ:

			EncodeGenericNumberHeaderL(aHeader, KErrUPnPEncodeSEQ);	

			break;

		case UPnP::EMAN:

			EncodeMANL(aHeader, KErrUPnPEncodeMAN);

			break;

		case UPnP::ESoapAction:

			EncodeSoapActionL(aHeader);	

			break;

		case UPnP::ECallback:

			EncodeCallbackL(aHeader);	

			break;

		default:

			User::Leave(KErrNotSupported);

		}	

	}

CUPnPHeaderWriter::CUPnPHeaderWriter(RStringPool aStringPool)

:iStringPool(aStringPool)

	{

	// do nothing.

	}

void CUPnPHeaderWriter::EncodeGenericUpnpHeaderL(RHeaderField& aHeader, TInt aErrorCode) const

	{

	THeaderFieldPartIter iter = aHeader.PartsL();

	iter.First();

	if (iter.AtEnd())

		User::Leave(aErrorCode);

	aHeader.BeginRawDataL();

	const CHeaderFieldPart* part = iter();

	if (part == NULL)

		User::Leave(aErrorCode);

	THTTPHdrVal ptVal = part->Value();

	if (ptVal.Type() != THTTPHdrVal::KStrFVal)

		User::Leave(aErrorCode);

	aHeader.WriteRawDataL(ptVal.StrF().DesC());

	aHeader.CommitRawData();	

	}

void CUPnPHeaderWriter::EncodeGenericNumberHeaderL(RHeaderField& aHeader, TInt aErrorCodeToUseOnFailure) const

	{

	// header = <header> ":" 1*DIGIT

	THeaderFieldPartIter iter1 = aHeader.PartsL();

	iter1.First();

	if (!iter1.AtEnd())

		{

		const CHeaderFieldPart* part = iter1();

		if (part)

			{

			THTTPHdrVal pt1val = part->Value();

			if (pt1val.Type() == THTTPHdrVal::KTIntVal)

				{

				TBuf8<KMaxNumDigits> lenBuf;

				lenBuf.Num(pt1val.Int());

				aHeader.BeginRawDataL();

				aHeader.WriteRawDataL(lenBuf);

				aHeader.CommitRawData();

				return;

				}

			}

		}

	User::Leave(aErrorCodeToUseOnFailure);

	}

void CUPnPHeaderWriter::EncodeTimeoutHeaderL(RHeaderField& aHeader, TInt aErrorCode) const

	{

	THeaderFieldPartIter iter = aHeader.PartsL();

	iter.First();

	if (iter.AtEnd())

		User::Leave(aErrorCode);

	aHeader.BeginRawDataL();

	const CHeaderFieldPart* part = iter();

	if (part == NULL)

		User::Leave(aErrorCode);

	THTTPHdrVal ptVal = part->Value();

	if (ptVal.Type() != THTTPHdrVal::KTIntVal)

		User::Leave(aErrorCode);

	if(ptVal.Int() == -KMaxTInt)

		{

		RStringF infiniteStrF = iStringPool.StringF(UPnP::EInfinite, TUPnPTable::Table());

		aHeader.WriteRawDataL(infiniteStrF.DesC());

		}

	else

		{

		HBufC8* ptValDesC;

		InetProtTextUtils::ConvertIntToDescriptorL(ptVal.Int(), ptValDesC);

		CleanupStack::PushL(ptValDesC);

		RBuf8 buf;

		buf.CreateL(KTimeoutSecond());

		CleanupClosePushL(buf);

		buf.ReAllocL(buf.Length() + ptValDesC->Length());

		buf.Append(ptValDesC->Des());

		aHeader.WriteRawDataL(buf);

		CleanupStack::PopAndDestroy(&buf);

		CleanupStack::PopAndDestroy(ptValDesC);

		}

	aHeader.CommitRawData();	

	}

void CUPnPHeaderWriter::EncodeMANL(RHeaderField& aHeader, TInt aErrorCode) const

	{

	THeaderFieldPartIter iter = aHeader.PartsL();

	iter.First();

	if (iter.AtEnd())

		User::Leave(aErrorCode);

	aHeader.BeginRawDataL();

	const CHeaderFieldPart* part = WriteRawStrFMANPartL(aHeader, iter, aErrorCode);

	// Check for parameters

	THeaderFieldParamIter iter2 = part->Parameters();

	iter2.First();

	if(!iter2.AtEnd())

		{

		aHeader.WriteRawDataL(KSemiSpaceSep);

		const CHeaderFieldParam* param = iter2();

		if (!param)

			User::Leave(aErrorCode);

		RStringF nsStr = iStringPool.StringF(UPnP::ENS,TUPnPTable::Table());

		THTTPHdrVal paramVal = param->Value();

		if (param->Name() == nsStr)

			{

			// Get the ns value

			if (paramVal.Type() != THTTPHdrVal::KTIntVal)

				User::Leave(aErrorCode);

			HBufC8* nsValBuf;

			InetProtTextUtils::ConvertIntToDescriptorL(paramVal.Int(), nsValBuf);

			CleanupStack::PushL(nsValBuf);

			aHeader.WriteRawDataL(nsStr.DesC());

			aHeader.WriteRawDataL('=');

			aHeader.WriteRawDataL(*nsValBuf);

			CleanupStack::PopAndDestroy(nsValBuf);				

			}

		else

			{

			// Extension parameter - handle strings only

			if (paramVal.Type() != THTTPHdrVal::KStrFVal)

				User::Leave(aErrorCode);

			RStringF paramName = iStringPool.StringF(param->Name());

			aHeader.WriteRawDataL(paramName.DesC());

			aHeader.WriteRawDataL('=');

			aHeader.WriteRawDataL(paramVal.StrF().DesC());

			}

		}

	// Done

	aHeader.CommitRawData();	

	}

const CHeaderFieldPart* CUPnPHeaderWriter::WriteRawStrFMANPartL(RHeaderField& aHeader, THeaderFieldPartIter& aIter, TInt aErrorCode) const

	{

	const CHeaderFieldPart* part = aIter();

	if (part == NULL)

		User::Leave(aErrorCode);

	THTTPHdrVal ptVal = part->Value();

	if (ptVal.Type() != THTTPHdrVal::KStrFVal)

		User::Leave(aErrorCode);

	aHeader.WriteRawDataL(KQuotes());

	aHeader.WriteRawDataL(ptVal.StrF().DesC());

	aHeader.WriteRawDataL(KQuotes());

	return part;

	}

void CUPnPHeaderWriter::EncodeSoapActionL(RHeaderField& aHeader) const

	{

	THeaderFieldPartIter iter = aHeader.PartsL();

	iter.First();

	if (iter.AtEnd())

		User::Leave(KErrUPnPEncodeSoapAction);

	aHeader.BeginRawDataL();

	const CHeaderFieldPart* part = iter();

	if (part == NULL)

		User::Leave(KErrUPnPEncodeSoapAction);

	THTTPHdrVal ptVal = part->Value();

	if (ptVal.Type() != THTTPHdrVal::KStrFVal)

		User::Leave(KErrUPnPEncodeSoapAction);

	aHeader.WriteRawDataL(KQuotes());

	aHeader.WriteRawDataL(ptVal.StrF().DesC());

	aHeader.WriteRawDataL(KQuotes());

	aHeader.CommitRawData();

	}

void CUPnPHeaderWriter::EncodeCallbackL(RHeaderField& aHeader) const

	{

	THeaderFieldPartIter iter = aHeader.PartsL();

	iter.First();

	if (iter.AtEnd())

		User::Leave(KErrUPnPEncodeCallback);

	aHeader.BeginRawDataL();

	while (!iter.AtEnd())

		{

		const CHeaderFieldPart* part = iter();

		if (part == NULL)

			User::Leave(KErrUPnPEncodeCallback);

		THTTPHdrVal ptVal = part->Value();

		if (ptVal.Type() != THTTPHdrVal::KStrFVal)

			User::Leave(KErrUPnPEncodeCallback);

		aHeader.WriteRawDataL(KOpenAngleBracket());

		aHeader.WriteRawDataL(ptVal.StrF().DesC());

		aHeader.WriteRawDataL(KCloseAngleBracket());

		++iter;

		}

	aHeader.CommitRawData();

	}

void CUPnPHeaderWriter::EncodeExtL(RHeaderField& aHeader)

	{

	aHeader.BeginRawDataL();

	aHeader.WriteRawDataL(KNullDesC8);

	aHeader.CommitRawData();	

	}