1 /*

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

     3 * All rights reserved.

     4 * This component and the accompanying materials are made available

     5 * under the terms of the License "Eclipse Public License v1.0"

     6 * which accompanies this distribution, and is available

     7 * at the URL "http://www.eclipse.org/legal/epl-v10.html".

     8 *

     9 * Initial Contributors:

    10 * Nokia Corporation - initial contribution.

    11 *

    12 * Contributors:

    13 *

    14 * Description: 

    15 *

    16 */

    17 

    18 

    19 #include <e32std.h>

    20 #include <f32file.h>

    21 #include <s32file.h>

    22 #include <caf/caferr.h>

    23 #include <caf/supplieroutputfile.h>

    24 #include "testagentimportfile.h"

    25 

    26 

    27 

    28 using namespace ContentAccess;

    29 

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

    31 	{

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

    33 	CleanupStack::PushL(self);

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

    35 	CleanupStack::Pop(self);

    36 	return self;

    37 	}

    38 

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

    40 	{

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

    42 	CleanupStack::PushL(self);

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

    44 	CleanupStack::Pop(self);

    45 	return self;

    46 	}

    47 	

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

    49 	{

    50 	iImportStatus = EInProgress;

    51 	}

    52 

    53 CTestAgentImportFile::~CTestAgentImportFile()

    54 	{

    55 	if(iFileOpen)

    56 		{

    57 		iFile.Close();

    58 		iFileOpen = EFalse;

    59 		}

    60 	if(iOutputFileName)

    61 		{

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

    63 		delete iOutputFileName;

    64 		}

    65 	iFs.Close();

    66 	delete iOutputDirectory;

    67 	delete iSuggestedFileName;

    68 	delete iLastWriteData;

    69 	iOutputFiles.ResetAndDestroy();

    70 	}

    71   

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

    73 	{

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

    75 	TBuf8 <KMaxDataTypeLength> mimeType;

    76 	mimeType.Copy(aMimeType);

    77 	mimeType.LowerCase();

    78 	if(mimeType != KDrmMimeType())

    79 		{

    80 		User::Leave(KErrCANotSupported);

    81 		}

    82 

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

    84 	iOutputDirectory = aOutputDirectory.AllocL();

    85 	iSuggestedFileName = aSuggestedFileName.AllocL();

    86 

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

    88 	}

    89 

    90 TInt CTestAgentImportFile::OpenOutputFile()

    91 	{

    92 	TInt err = KErrNone;

    93 	if(iOutputFileName)

    94 		{

    95 		return KErrInUse;

    96 		}

    97 

    98 	if(iAgentCreatesOutputFiles)

    99 		{

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

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

   102 		if(err == KErrNone)

   103 			{

   104 			TPtr fileNamePtr = iOutputFileName->Des();

   105 			fileNamePtr.Copy(*iOutputDirectory);

   106 			fileNamePtr.Append(*iSuggestedFileName);

   107 

   108 			// create output file

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

   110 			if(err == KErrNone)

   111 				{

   112 				iFileOpen = ETrue;

   113 				}

   114 			}

   115 		return err;

   116 		}

   117 	else

   118 		{

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

   120 		return KErrCANewFileHandleRequired;

   121 		}

   122 	}

   123 

   124 TInt CTestAgentImportFile::WriteData(const TDesC8& aData)

   125 	{

   126 	TInt err = KErrNone;

   127 

   128 	if(!iFileOpen)

   129 		{

   130 		err = OpenOutputFile();

   131 		}

   132 	if(err == KErrNone)

   133 		{

   134 		err = iFile.Write(aData);

   135 		}

   136 

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

   138 	if(err == KErrCANewFileHandleRequired)

   139 		{

   140 		// save write parameters

   141 

   142 		delete iLastWriteData;

   143 		iLastWriteData = aData.Alloc();

   144 		if (NULL == iLastWriteData)

   145 			{

   146 			err = KErrNoMemory;

   147 			}

   148 		}

   149 	return err;

   150 	}

   151 

   152 TInt CTestAgentImportFile::WriteDataComplete()

   153 	{

   154 	TRAPD(err, WriteDataCompleteL());

   155 	return err;

   156 	}

   157 

   158 void CTestAgentImportFile::WriteDataCompleteL()

   159 	{

   160 	CSupplierOutputFile *temp = NULL; 

   161 	if(iFileOpen)

   162 		{	

   163 		iFile.Close();

   164 		iFileOpen = EFalse;

   165 		}

   166 

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

   168 	if(iOutputFileName)

   169 		{

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

   171 

   172 		CleanupStack::PushL(temp);

   173 		iOutputFiles.AppendL(temp);

   174 		CleanupStack::Pop(temp);

   175 

   176 		delete iOutputFileName;

   177 		iOutputFileName = NULL;

   178 		}

   179 

   180 	if(!iAgentCreatesOutputFiles)

   181 		{

   182 		iWaitingForReceiptHandle = ETrue;

   183 		User::Leave(KErrCANewFileHandleRequired);

   184 		}

   185 	else

   186 		{

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

   188 		iOutputFileName = HBufC::NewL(KMaxFileName);

   189 		TPtr fileNamePtr = iOutputFileName->Des();

   190 		fileNamePtr.Copy(*iOutputDirectory);

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

   192 

   193 		// create output file

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

   195 		CleanupClosePushL(iFile);

   196 

   197 		RFileWriteStream stream(iFile);

   198 		CleanupClosePushL(stream);

   199 

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

   201 		stream.CommitL();

   202 

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

   204 		

   205 		// Add to list of output files

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

   207 

   208 		CleanupStack::PushL(temp);

   209 		iOutputFiles.AppendL(temp);

   210 		CleanupStack::Pop(temp);

   211 

   212 		delete iOutputFileName;

   213 		iOutputFileName = NULL;	

   214 		}

   215 	}

   216 		

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

   218 	{

   219 	TRequestStatus *ptr = &aStatus;

   220 	

   221 	// call synchronous WriteData

   222 	TInt rVal = WriteData(aData);

   223 	

   224 	// complete the request

   225 	User::RequestComplete(ptr,rVal);

   226 	}

   227 

   228 void CTestAgentImportFile::WriteDataComplete(TRequestStatus &aStatus)

   229 	{

   230 	TRequestStatus *ptr = &aStatus;

   231 	// call synchronous WriteDataComplete

   232 	TInt rVal = WriteDataComplete();

   233 	

   234 	// complete the request

   235 	User::RequestComplete(ptr,rVal);

   236 	}

   237 	

   238 TInt CTestAgentImportFile::OutputFileCountL() const

   239 	{

   240 	return iOutputFiles.Count();

   241 	}

   242 

   243 CSupplierOutputFile& CTestAgentImportFile::OutputFileL(TInt aIndex)

   244 	{

   245 	return *iOutputFiles[aIndex];

   246 	}

   247 

   248 TImportStatus CTestAgentImportFile::GetImportStatus() const

   249 	{

   250 	return iImportStatus;

   251 	}	

   252 

   253 TInt CTestAgentImportFile::GetSuggestedOutputFileExtension(TDes& aFileExtension)

   254 	{

   255 	if(iWaitingForReceiptHandle)

   256 		{

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

   258 		}

   259 	else

   260 		{

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

   262 		}

   263 	return KErrNone;

   264 	}

   265 

   266 TInt CTestAgentImportFile::GetSuggestedOutputFileName(TDes& aFileName)

   267 	{

   268 	if(iWaitingForReceiptHandle)

   269 		{

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

   271 		}

   272 	else

   273 		{

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

   275 		}

   276 	return KErrNone;

   277 	}

   278 

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

   280 	{

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

   282 	return err;

   283 	}

   284 

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

   286 	{

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

   288 

   289 	if(iWaitingForReceiptHandle)

   290 		{

   291 		RFileWriteStream stream(iFile);

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

   293 		stream.CommitL();

   294 		stream.Close();

   295 		iFile.Close();

   296 		

   297 		CSupplierOutputFile *temp =

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

   299 		CleanupStack::PushL(temp);

   300 		iOutputFiles.AppendL(temp);

   301 		CleanupStack::Pop(temp);

   302 		}

   303 	else

   304 		{

   305 		iFileOpen = ETrue;

   306 		iOutputFileName = aFileName.AllocL();

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

   308 		}

   309 	}

   310 

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

   312 	{

   313 	TRequestStatus *ptr = &aStatus;

   314 	// call synchronous WriteDataComplete

   315 	TInt rVal = ContinueWithNewOutputFile(aFile, aFileName);

   316 	

   317 	// complete the request

   318 	User::RequestComplete(ptr,rVal);

   319 	}

   320 

   321 

   322 TBool CTestAgentImportFile::ContentMimeTypeL(TDes8& aContentMime)

   323 	{

   324 	//Just set aContentMime to text/plain

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

   326 	//Please refer to reftestagent as an example.

   327 	_LIT8(KPlainText, "text/plain");

   328 	aContentMime.Copy(KPlainText);

   329 	

   330 	return ETrue;

   331 	}

   332