/*

* Copyright (c) 2004-2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "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 <f32file.h>

#include <s32file.h>

#include <caf/caferr.h>

#include <caf/supplieroutputfile.h>

#include "testagentimportfile.h"

using namespace ContentAccess;

CTestAgentImportFile* CTestAgentImportFile::NewL(const TDesC8& aMimeType, const CMetaDataArray& aMetaDataArray, const TDesC& aOutputDirectory, const TDesC& aSuggestedFileName)

	{

	CTestAgentImportFile* self=new(ELeave) CTestAgentImportFile(ETrue);

	CleanupStack::PushL(self);

	self->ConstructL(aMimeType, aMetaDataArray, aOutputDirectory, aSuggestedFileName);

	CleanupStack::Pop(self);

	return self;

	}

CTestAgentImportFile* CTestAgentImportFile::NewL(const TDesC8& aMimeType, const CMetaDataArray& aMetaDataArray)

	{

	CTestAgentImportFile* self=new(ELeave) CTestAgentImportFile(EFalse);

	CleanupStack::PushL(self);

	self->ConstructL(aMimeType, aMetaDataArray, KNullDesC(), KNullDesC());

	CleanupStack::Pop(self);

	return self;

	}

CTestAgentImportFile::CTestAgentImportFile(TBool aAgentCreatesOutputFiles) : iAgentCreatesOutputFiles(aAgentCreatesOutputFiles)

	{

	iImportStatus = EInProgress;

	}

CTestAgentImportFile::~CTestAgentImportFile()

	{

	if(iFileOpen)

		{

		iFile.Close();

		iFileOpen = EFalse;

		}

	if(iOutputFileName)

		{

		iFs.Delete(iOutputFileName->Des());

		delete iOutputFileName;

		}

	iFs.Close();

	delete iOutputDirectory;

	delete iSuggestedFileName;

	delete iLastWriteData;

	iOutputFiles.ResetAndDestroy();

	}

void CTestAgentImportFile::ConstructL(const TDesC8& aMimeType, const CMetaDataArray& /*aMetaDataArray*/, const TDesC& aOutputDirectory, const TDesC& aSuggestedFileName)

	{

	_LIT8(KDrmMimeType, "application/testagent.drm");

	TBuf8 <KMaxDataTypeLength> mimeType;

	mimeType.Copy(aMimeType);

	mimeType.LowerCase();

	if(mimeType != KDrmMimeType())

		{

		User::Leave(KErrCANotSupported);

		}

	// create buffer to store any output path and filename supplied by the client

	iOutputDirectory = aOutputDirectory.AllocL();

	iSuggestedFileName = aSuggestedFileName.AllocL();

	User::LeaveIfError(iFs.Connect());

	}

TInt CTestAgentImportFile::OpenOutputFile()

	{

	TInt err = KErrNone;

	if(iOutputFileName)

		{

		return KErrInUse;

		}

	if(iAgentCreatesOutputFiles)

		{

		// Figure out a name including full path for the output file

		TRAP(err, iOutputFileName = HBufC::NewL(iOutputDirectory->Des().Length() + iSuggestedFileName->Des().Length()));

		if(err == KErrNone)

			{

			TPtr fileNamePtr = iOutputFileName->Des();

			fileNamePtr.Copy(*iOutputDirectory);

			fileNamePtr.Append(*iSuggestedFileName);

			// create output file

			err = iFile.Create(iFs, fileNamePtr, EFileShareAny  | EFileStream | EFileWrite);

			if(err == KErrNone)

				{

				iFileOpen = ETrue;

				}

			}

		return err;

		}

	else

		{

		// The application must supply a new file handle for us to adopt

		return KErrCANewFileHandleRequired;

		}

	}

TInt CTestAgentImportFile::WriteData(const TDesC8& aData)

	{

	TInt err = KErrNone;

	if(!iFileOpen)

		{

		err = OpenOutputFile();

		}

	if(err == KErrNone)

		{

		err = iFile.Write(aData);

		}

	// check if the output file creation or file writing needs a new file handle

	if(err == KErrCANewFileHandleRequired)

		{

		// save write parameters

		delete iLastWriteData;

		iLastWriteData = aData.Alloc();

		if (NULL == iLastWriteData)

			{

			err = KErrNoMemory;

			}

		}

	return err;

	}

TInt CTestAgentImportFile::WriteDataComplete()

	{

	TRAPD(err, WriteDataCompleteL());

	return err;

	}

void CTestAgentImportFile::WriteDataCompleteL()

	{

	CSupplierOutputFile *temp = NULL; 

	if(iFileOpen)

		{	

		iFile.Close();

		iFileOpen = EFalse;

		}

	// add a new entry to the array of output files

	if(iOutputFileName)

		{

		temp = CSupplierOutputFile::NewL(iOutputFileName->Des(), EContent, _L8("application/testagent.drm"));

		CleanupStack::PushL(temp);

		iOutputFiles.AppendL(temp);

		CleanupStack::Pop(temp);

		delete iOutputFileName;

		iOutputFileName = NULL;

		}

	if(!iAgentCreatesOutputFiles)

		{

		iWaitingForReceiptHandle = ETrue;

		User::Leave(KErrCANewFileHandleRequired);

		}

	else

		{

		// Figure out receipt name including full path for the output file

		iOutputFileName = HBufC::NewL(KMaxFileName);

		TPtr fileNamePtr = iOutputFileName->Des();

		fileNamePtr.Copy(*iOutputDirectory);

		fileNamePtr.Append(_L("receipt.txt"));

		// create output file

		User::LeaveIfError(iFile.Create(iFs, fileNamePtr, EFileShareAny  | EFileStream | EFileWrite));

		CleanupClosePushL(iFile);

		RFileWriteStream stream(iFile);

		CleanupClosePushL(stream);

		stream.WriteL(_L("This is a receipt"));

		stream.CommitL();

		CleanupStack::PopAndDestroy(2, &iFile);  // stream, file

		// Add to list of output files

		temp = CSupplierOutputFile::NewL(iOutputFileName->Des(), EReceipt, _L8("text/plain"));

		CleanupStack::PushL(temp);

		iOutputFiles.AppendL(temp);

		CleanupStack::Pop(temp);

		delete iOutputFileName;

		iOutputFileName = NULL;	

		}

	}

void CTestAgentImportFile::WriteData(const TDesC8& aData, TRequestStatus &aStatus)

	{

	TRequestStatus *ptr = &aStatus;

	// call synchronous WriteData

	TInt rVal = WriteData(aData);

	// complete the request

	User::RequestComplete(ptr,rVal);

	}

void CTestAgentImportFile::WriteDataComplete(TRequestStatus &aStatus)

	{

	TRequestStatus *ptr = &aStatus;

	// call synchronous WriteDataComplete

	TInt rVal = WriteDataComplete();

	// complete the request

	User::RequestComplete(ptr,rVal);

	}

TInt CTestAgentImportFile::OutputFileCountL() const

	{

	return iOutputFiles.Count();

	}

CSupplierOutputFile& CTestAgentImportFile::OutputFileL(TInt aIndex)

	{

	return *iOutputFiles[aIndex];

	}

TImportStatus CTestAgentImportFile::GetImportStatus() const

	{

	return iImportStatus;

	}	

TInt CTestAgentImportFile::GetSuggestedOutputFileExtension(TDes& aFileExtension)

	{

	if(iWaitingForReceiptHandle)

		{

		aFileExtension.Copy(_L(".TXT"));		

		}

	else

		{

		aFileExtension.Copy(_L(".DRM"));		

		}

	return KErrNone;

	}

TInt CTestAgentImportFile::GetSuggestedOutputFileName(TDes& aFileName)

	{

	if(iWaitingForReceiptHandle)

		{

		aFileName.Copy(_L("RECEIPT.TXT"));

		}

	else

		{

		aFileName.Copy(_L("OUTPUTFILE.DRM"));

		}

	return KErrNone;

	}

TInt CTestAgentImportFile::ContinueWithNewOutputFile(RFile& aFile, const TDesC& aFileName)

	{

	TRAPD(err, ContinueWithNewOutputFileL(aFile, aFileName));

	return err;

	}

void CTestAgentImportFile::ContinueWithNewOutputFileL(RFile& aFile, const TDesC& aFileName)

	{

	User::LeaveIfError( iFile.Duplicate(aFile) );

	if(iWaitingForReceiptHandle)

		{

		RFileWriteStream stream(iFile);

		stream.WriteL(_L("This is a receipt"));

		stream.CommitL();

		stream.Close();

		iFile.Close();

		CSupplierOutputFile *temp =

			CSupplierOutputFile::NewL(aFileName, EReceipt, _L8("text/plain"));

		CleanupStack::PushL(temp);

		iOutputFiles.AppendL(temp);

		CleanupStack::Pop(temp);

		}

	else

		{

		iFileOpen = ETrue;

		iOutputFileName = aFileName.AllocL();

		User::LeaveIfError(WriteData(*iLastWriteData));

		}

	}

void CTestAgentImportFile::ContinueWithNewOutputFile(RFile& aFile, const TDesC& aFileName, TRequestStatus& aStatus)

	{

	TRequestStatus *ptr = &aStatus;

	// call synchronous WriteDataComplete

	TInt rVal = ContinueWithNewOutputFile(aFile, aFileName);

	// complete the request

	User::RequestComplete(ptr,rVal);

	}

TBool CTestAgentImportFile::ContentMimeTypeL(TDes8& aContentMime)

	{

	//Just set aContentMime to text/plain

	//Real agent should implement this function for discovering content type based on available buffer

	//Please refer to reftestagent as an example.

	_LIT8(KPlainText, "text/plain");

	aContentMime.Copy(KPlainText);

	return ETrue;

	}