#include <e32test.h>

#include <f32file.h>

#include <s32file.h>

#include <cntdb.h>

#include <cntdef.h>

#include <cntitem.h>

#include <cntfield.h>

#include <cntfldst.h>

#include <bautils.h>

#include <barsc.h>

#include <barsread.h>

#include <versit.h>

#include <vcard.h>

#include <e32def.h>

#include <coreappstest/testserver.h>

// User includes

// Flags

#define OOM_TESTS

// Globals

static RTest TheTest(_L("T_GENERALADDRESS - Test General Address Export and Import"));

static CCntTest* TheTestHelper;

// Type definitions

typedef void (*TGAFunction)(TBool& aPrintedTestTitle);

// Constants

const TInt KIntContactFieldVCardMapGENERALADDRESS = 0;

const TUid KUidContactFieldVCardMapGENERALADDRESS = { KIntContactFieldVCardMapGENERALADDRESS };

// Literal constants

_LIT(KTestDatabaseFile, "C:GeneralAddressDatabase.cdb");

//

_LIT(KTestNameFamily, "Bloggs");

_LIT(KTestNameGiven, "Joe");

//

_LIT(KTestAddressHOME, "_HOME");

_LIT(KTestAddressWORK, "_WORK");

_LIT(KTestAddressPREF, "_PREF");

_LIT(KTestAddressGENERALADDRESS, "_GENERALADDRESS");

//

_LIT(KTestAddressPOSTOFFICE, "POSTOFFICE");

_LIT(KTestAddressEXTENDEDADR, "EXTENDEDADR");

_LIT(KTestAddressADR, "ADR");

_LIT(KTestAddressLOCALITY, "LOCALITY");

_LIT(KTestAddressREGION, "REGION");

_LIT(KTestAddressPOSTCODE, "POSTCODE");

_LIT(KTestAddressCOUNTRY, "COUNTRY");

//

_LIT(KTestVCardExportFileName, "C:\\T_GENERALADDRESS_EXPORTED.TXT");

_LIT(KTestVCardImportFileName, "C:\\T_GENERALADDRESS_IMPORTED.TXT");

//

_LIT8(KVersitParam8GeneralAddress, "");

// Utility methods

/**

 * Create a simple contact with just the name fields populated

 */

static TContactItemId CreateContactL(CContactDatabase& aDatabase)

	{

	const TContactItemId templateId = aDatabase.TemplateId();

	//

	CContactItem* templateCard = aDatabase.ReadContactLC(templateId);

	CContactCard* card = CContactCard::NewL(templateCard);

	CleanupStack::PushL(card);

	//

	SetNameL(*card, KUidContactFieldFamilyName, KUidContactFieldVCardMapUnusedN, KTestNameFamily, EFalse);

	SetNameL(*card, KUidContactFieldGivenName, KUidContactFieldVCardMapUnusedN, KTestNameGiven, EFalse);

	//

	const TContactItemId id = aDatabase.AddNewContactL(*card);

	CleanupStack::PopAndDestroy(2, templateCard);

	return id;

	}

/**

 * Combine a value and a postfix to create a real unified value

 */

static HBufC* ConstructAddressValueLC(const TDesC& aValue, const TDesC& aPostFix)

	{

	HBufC* value = HBufC::NewLC(aValue.Length() + aPostFix.Length());

	TPtr pValue(value->Des());

	pValue = aValue;

	pValue.Append(aPostFix);

	//

	return value;

	}

/**

 * Add a specific address field to a vCard

 */

static void AddAddressFieldL(CContactItem& aContact, TFieldType aFieldType, TUid aMapping, TFieldType aExtraFieldType, const TDesC& aValue, const TDesC& aValuePostFix)

	{

	CContentType* content = CContentType::NewL(aFieldType, aMapping);

	CleanupStack::PushL(content);

	content->AddFieldTypeL(aExtraFieldType);

	//

	CContactItemField* contactItemField = CContactItemField::NewLC(KStorageTypeText, *content);

	//

	HBufC* value = ConstructAddressValueLC(aValue, aValuePostFix);

	//

	contactItemField->TextStorage()->SetStandardTextL(*value);

	CleanupStack::PopAndDestroy(value);

	//

	aContact.AddFieldL(*contactItemField);

	CleanupStack::Pop(contactItemField);

	//

	CleanupStack::PopAndDestroy(content);

	}

/**

 * Return the post fix descriptor for a given field type (HOME, WORK, PREF, General)

 */

static TPtrC PostFixForFieldType(TFieldType aType)

	{

	TPtrC postFix(KNullDesC);

	switch(aType.iUid)

		{

	case KIntContactFieldVCardMapHOME:

		postFix.Set(KTestAddressHOME);

		break;

	case KIntContactFieldVCardMapWORK:

		postFix.Set(KTestAddressWORK);

		break;

	case KIntContactFieldVCardMapPREF:

		postFix.Set(KTestAddressPREF);

		break;

	case KIntContactFieldVCardMapGENERALADDRESS:

		postFix.Set(KTestAddressGENERALADDRESS);

		break;

	default:

		TheTest(EFalse);

		break;

		}

	//

	return postFix;

	}

/**

 * Add a specific address type to a contact card

 */

static void AddAddressL(CContactDatabase& aDatabase, TContactItemId aId, TFieldType aExtraFieldType)

	{

	const TPtrC postFix(PostFixForFieldType(aExtraFieldType));

	//

	CContactItem* card = aDatabase.OpenContactLX(aId);

	CleanupStack::PushL(card);

	//

	AddAddressFieldL(*card, KUidContactFieldPostOffice, KUidContactFieldVCardMapPOSTOFFICE, aExtraFieldType, KTestAddressPOSTOFFICE, postFix);

	AddAddressFieldL(*card, KUidContactFieldExtendedAddress, KUidContactFieldVCardMapEXTENDEDADR, aExtraFieldType, KTestAddressEXTENDEDADR, postFix);

	AddAddressFieldL(*card, KUidContactFieldAddress, KUidContactFieldVCardMapADR, aExtraFieldType, KTestAddressADR, postFix);

	AddAddressFieldL(*card, KUidContactFieldLocality, KUidContactFieldVCardMapLOCALITY, aExtraFieldType, KTestAddressLOCALITY, postFix);

	AddAddressFieldL(*card, KUidContactFieldRegion, KUidContactFieldVCardMapREGION, aExtraFieldType, KTestAddressREGION, postFix);

	AddAddressFieldL(*card, KUidContactFieldPostcode, KUidContactFieldVCardMapPOSTCODE, aExtraFieldType, KTestAddressPOSTCODE, postFix);

	AddAddressFieldL(*card, KUidContactFieldCountry, KUidContactFieldVCardMapCOUNTRY, aExtraFieldType, KTestAddressCOUNTRY, postFix);

	//

	aDatabase.CommitContactL(*card);

	CleanupStack::PopAndDestroy(card);

	CleanupStack::Pop(); // lock record

	}

/**

 * Export a specific contact to a vCard file

 */

static void ExportContactAsVCardL(CContactDatabase& aDatabase, TContactItemId aId, RFs& aFsSession, const TDesC& aVCardFileName)

	{

	CContactIdArray* idArray = CContactIdArray::NewLC();

	idArray->AddL(aId);

	//

	RFile outfile;

	const TInt error = outfile.Replace(aFsSession, aVCardFileName, EFileWrite);

	User::LeaveIfError(error);

	CleanupClosePushL(outfile);

	//

	RFileWriteStream writeStream(outfile);

	CleanupClosePushL(writeStream);

	//

	const TUid KVersitVCardUid = { KUidVCardConvDefaultImpl };

	aDatabase.ExportSelectedContactsL(KVersitVCardUid, *idArray, writeStream, CContactDatabase::EDefault);

	writeStream.CommitL();

	CleanupStack::PopAndDestroy(3, idArray); // writeStream, outfile, idArray

	}

/**

 * Parse a file using the Symbian vCard parser and return the parsed data structures

 * to the caller

 */

static CParserVCard* ParseVCardFileLC(RFs& aFsSession, const TDesC& aVCardFileName)

	{

	CParserVCard* parser = CParserVCard::NewL();

	CleanupStack::PushL(parser);

	//

	RFile readFile;

	const TInt error = readFile.Open(aFsSession, aVCardFileName, EFileRead);

	User::LeaveIfError(error);

	CleanupClosePushL(readFile);

	//

	RFileReadStream readStream(readFile);

	CleanupClosePushL(readStream);

	parser->InternalizeL(readStream);

	CleanupStack::PopAndDestroy(2); // readStream, readFile

	//

	return parser;

	}

/**

 * Returns the property parameter name value used by Versit for this particular

 * Contacts Model field type (E.g. HOME, WORK, PREF, etc). Returns KNullDesC in

 * the case of General Address. Panic (fails the test) if the type is not recognised.

 */

static TPtrC8 VersitPropertyParameterNameForContactsModelFieldType(TFieldType aAddressType)

	{

	TPtrC8 propParamValue(KNullDesC8);

	//

	switch(aAddressType.iUid)

		{

	case KIntContactFieldVCardMapHOME:

		propParamValue.Set(KVersitParam8Home);

		break;

	case KIntContactFieldVCardMapWORK:

		propParamValue.Set(KVersitParam8Work);

		break;

	case KIntContactFieldVCardMapPREF:

		propParamValue.Set(KVersitParam8Pref);

		break;

	case KIntContactFieldVCardMapGENERALADDRESS:

		break;

	default:

		TheTest(EFalse);

		break;

		}

	//

	return propParamValue;

	}

/**

 * Search a vCard property for the presence of a specific address type, e.g. HOME, WORK, PREF or "General"

 */

static TBool ValidateVCardPropertyForAddressType(const CParserProperty& aProperty, const TDesC8& aPropertyParameterType)

	{

	TBool validated = EFalse;

	// Have to perform more substancial checks if we're looking for just a general address...

	if	(aPropertyParameterType.Length())

		{

		// HOME, WORK, PREF

		validated = (aProperty.Param(aPropertyParameterType) != NULL);

		}

	else

		{

		// In the case of General Address, which won't have an associated property parameter

		// we must check that none of the other types are present for this address. If none are

		// matched, then this address can be considered general.

		const TBool foundHOME = (aProperty.Param(KVersitParam8Home) != NULL);

		const TBool foundWORK = (aProperty.Param(KVersitParam8Work) != NULL);

		const TBool foundPREF = (aProperty.Param(KVersitParam8Pref) != NULL);

		//

		validated = !(foundHOME || foundWORK || foundPREF);

		}

	//

	return validated;

	}

/**

 * Check that the specified actual value found in the vCard matches the expected value

 * which is constructed from the raw value and post fix.

 */

static TBool ValidateAddressSubFieldL(const TDesC& aActualValue, const TDesC& aExpectedRawValue, const TDesC& aPostFix)

	{

	HBufC* expectedValue = ConstructAddressValueLC(aExpectedRawValue, aPostFix);

	const TBool match = (expectedValue->CompareF(aActualValue) == 0);

	CleanupStack::PopAndDestroy(expectedValue);

	//

	return match;

	}

/**

 * Check that the specified array contains exact matches for each of the 7 address

 * sub field values.

 */

static TBool ValidateAddressSubFieldValuesL(const CDesCArray& aArray, const TDesC& aPostFix)

	{

	TPtrC actualValue;

	//

	actualValue.Set(aArray[0]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressPOSTOFFICE, aPostFix))

		return EFalse;

	actualValue.Set(aArray[1]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressEXTENDEDADR, aPostFix))

		return EFalse;

	actualValue.Set(aArray[2]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressADR, aPostFix))

		return EFalse;

	actualValue.Set(aArray[3]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressLOCALITY, aPostFix))

		return EFalse;

	actualValue.Set(aArray[4]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressREGION, aPostFix))

		return EFalse;

	actualValue.Set(aArray[5]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressPOSTCODE, aPostFix))

		return EFalse;

	actualValue.Set(aArray[6]);

	if	(!ValidateAddressSubFieldL(actualValue, KTestAddressCOUNTRY, aPostFix))

		return EFalse;

	//

	return ETrue;

	}

/**

 * Check a vCard for the presence of a specific address type. Returns ETrue

 * if the type is found within the vCard.

 */

static TBool ValidateVCardForAddressL(CParserVCard& aVCard, TFieldType aAddressType)

	{

	TBool validated = EFalse;

	const TUid KVersitAddressPropertyValueType = { KVersitPropertyCDesCArrayUid };

	const TPtrC8 propValueType = VersitPropertyParameterNameForContactsModelFieldType(aAddressType);

	//

	CArrayPtr<CParserProperty>* properties = aVCard.PropertyL(KVersitTokenADR, KVersitAddressPropertyValueType, EFalse);

	if	(properties)

		{

		CleanupStack::PushL(properties);

		const TInt count = properties->Count();

		for(TInt i=0; i<count || !validated; i++)

			{

			// First verify the property parameters

			CParserProperty& property = *properties->At(i);

			if	(ValidateVCardPropertyForAddressType(property, propValueType))

				{

				// Next verify the property values

				CParserPropertyValueCDesCArray* value = static_cast<CParserPropertyValueCDesCArray*>(property.Value());

				CDesCArray* array = value->Value();

				// There must be 7 sub-fields

				if	(array->MdcaCount() == 7)

					{

					// Check each sub-field for the correct value

					const TPtrC postFix = PostFixForFieldType(aAddressType);

					validated = ValidateAddressSubFieldValuesL(*array, postFix);

					}

				}

			}

		CleanupStack::PopAndDestroy(properties);

		}

	return validated;

	}

/**

 * Create a vCard N property and add it to the specified vCard parser object

 */

static void AddNameToVCardL(CParserVCard& aVCard)

	{

	// For the property value itself

	CDesCArrayFlat* desArray = new (ELeave)CDesCArrayFlat(4);

	CleanupStack::PushL(desArray);

	// Add the names

	desArray->AppendL(KTestNameFamily);

	desArray->AppendL(KTestNameGiven);

	desArray->AppendL(KNullDesC);

	desArray->AppendL(KNullDesC);

	desArray->AppendL(KNullDesC);

	// Create the property value

	CParserPropertyValue* value = new(ELeave) CParserPropertyValueCDesCArray(desArray);

	CleanupStack::Pop(desArray);

	CleanupStack::PushL(value);

	//

	CParserGroupedProperty* property = CParserGroupedProperty::NewL(*value, KVersitTokenN, NULL, NULL);

	CleanupStack::Pop(value); // value

	// This pushes property before anything can leave...

	aVCard.AddPropertyL(property);

	}

/**

 * Combine the address data and the post fix to form a new combined address data value

 * and append it to the specified array

 */

static void AddAddressSubFieldToArrayL(CDesCArray& aArray, const TDesC& aAddress, const TDesC& aPostFix)

	{

	HBufC* value = ConstructAddressValueLC(aAddress, aPostFix);

	aArray.AppendL(*value);

	CleanupStack::PopAndDestroy(value);

	}

/**

 * Create a vCard ADR property and add it to the specified vCard parser object

 */

static void AddAddressToVCardL(CParserVCard& aVCard, const TDesC8& aAddressType)

	{

	// For the property parameters

    CArrayPtr<CParserParam>* arrayOfParams = new(ELeave) CArrayPtrFlat<CParserParam>(5);

	CleanupStack::PushL(TCleanupItem(CleanUpResetAndDestroy, arrayOfParams));

	// If the address type is NULL then its a general address, in which case it doesn't

	// need a property parameter. Otherwise, assume it does

	if	(aAddressType.Length())

		{

		CParserParam* parserParam = CParserParam::NewL(aAddressType, KNullDesC);

		CleanupStack::PushL(parserParam);

		arrayOfParams->AppendL(parserParam);

		CleanupStack::Pop(parserParam);

		}

	// Create the value array

	CDesCArrayFlat* desArray = new (ELeave)CDesCArrayFlat(4);

	CleanupStack::PushL(desArray);

	// Pick the correct postfix

	TPtrC postFix(KNullDesC);

	if	(aAddressType == KVersitParam8Home)

		postFix.Set(KTestAddressHOME);

	else if (aAddressType == KVersitParam8Work)

		postFix.Set(KTestAddressWORK);

	else if (aAddressType == KVersitParam8Pref)

		postFix.Set(KTestAddressPREF);

	else if (aAddressType == KVersitParam8GeneralAddress)

		postFix.Set(KTestAddressGENERALADDRESS);

	// Add the names

	AddAddressSubFieldToArrayL(*desArray, KTestAddressPOSTOFFICE, postFix);

	AddAddressSubFieldToArrayL(*desArray, KTestAddressEXTENDEDADR, postFix);

	AddAddressSubFieldToArrayL(*desArray, KTestAddressADR, postFix);

	AddAddressSubFieldToArrayL(*desArray, KTestAddressLOCALITY, postFix);

	AddAddressSubFieldToArrayL(*desArray, KTestAddressREGION, postFix);

	AddAddressSubFieldToArrayL(*desArray, KTestAddressPOSTCODE, postFix);

	AddAddressSubFieldToArrayL(*desArray, KTestAddressCOUNTRY, postFix);

	// Create the property value

	CParserPropertyValue* value = new (ELeave) CParserPropertyValueCDesCArray(desArray);

	CleanupStack::Pop(desArray);

	CleanupStack::PushL(value);

	// And finally, the property itself

	CParserGroupedProperty* property = CParserGroupedProperty::NewL(*value, KVersitTokenADR, NULL, arrayOfParams);

	CleanupStack::Pop(2, arrayOfParams); // value, arrayOfParams

	// Add to vCard object - takes ownership of property immediately

	aVCard.AddPropertyL(property);

	}

/**

 * Create a simple vCard which contains only an N (Name) property

 */

static CParserVCard* CreateVCardLC()

	{

	CParserVCard* parser = CParserVCard::NewL();

	CleanupStack::PushL(parser);

	//

	AddNameToVCardL(*parser);

	//

	return parser;

	}

/**

 * Export the specified vCard object to a file

 */

static void CreateVCardFileFromVCardObjectL(CParserVCard& aVCard, RFs& aFsSession, const TDesC& aVCardFileName)

	{

	RFile outfile;

	const TInt error = outfile.Replace(aFsSession, aVCardFileName, EFileWrite);

	User::LeaveIfError(error);

	CleanupClosePushL(outfile);

	//

	RFileWriteStream writeStream(outfile);

	CleanupClosePushL(writeStream);

	//

	aVCard.ExternalizeL(writeStream);

	writeStream.CommitL();

	CleanupStack::PopAndDestroy(2); // writeStream, outfile

	}

/**

 * Import the specified vCard to the contacts database and return the newly created contact item

 */

static CContactItem* ImportVCardLC(CContactDatabase& aDatabase, RFs& aFsSession, const TDesC& aVCardFileName)

	{

	RFile readFile;

	const TInt error = readFile.Open(aFsSession, aVCardFileName, EFileRead);

	User::LeaveIfError(error);

	CleanupClosePushL(readFile);

	//

	RFileReadStream readStream(readFile);

	CleanupClosePushL(readStream);

	//

	TBool importSuccessful = EFalse;

	const TInt options = CContactDatabase::EDefault;

	const TUid KVersitVCardUid = { KUidVCardConvDefaultImpl };

	//

	CArrayPtr<CContactItem>* contacts = aDatabase.ImportContactsL(KVersitVCardUid, readStream, importSuccessful, options);

	CleanupStack::PopAndDestroy(2); // readStream, readFile

	// Check import results

	TheTest(importSuccessful);

	TheTest(contacts != NULL);

	TheTest(contacts->Count() == 1);

	// Delete container, but not the contact itself

	CContactItem* item = contacts->At(0);

	CleanupStack::PushL(item);

	delete contacts;

	return item;

	}

/**