// fcontacts.cpp

// 

// Copyright (c) 2008 - 2010 Accenture. All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the "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:

// Accenture - Initial contribution

//

#include <fshell/ioutils.h>

#include <CNTDB.H>

#include <CNTITEM.H>

#include <CNTFLDST.H>

// Some things that v9.1 doesn't have

#include "fcontacts_support.h"

using namespace IoUtils;

class CCmdContacts : public CCommandBase

	{

public:

	static CCommandBase* NewLC();

	~CCmdContacts();

private:

	CCmdContacts();

	TInt CheckDuplicateAndUpdate(CContactItemFieldSet &aFieldSet, CContactItemField &aField);	

	void FillContactFieldSetL(CContactItemFieldSet &aFieldSet, TUid aType);

	void PrintContact(CContactItem* aContact);

	void PrintGroupContentsL(CContactItem* aContactGroup, TInt aVerboseLevel);

	void PrintGroupsJoined(CContactItem* aContact, TInt aVerboseLevel);	

	void PrintTime(const TTime& aTime, TBool aNewline=EFalse);

	void PrintAcceptedOptions();	

	static const TDesC* UidName(TUid aUid);

private: // From CCommandBase.

	virtual const TDesC& Name() const;

	virtual void DoRunL();

	virtual void ArgumentsL(RCommandArgumentList& aArguments);

	virtual void OptionsL(RCommandOptionList& aOptions);

	void DoListL();

	void DoAddL();

	void DoDeleteL();

	void DoChangeL();

	void DoDeleteAllL();

	void DoCreateGroupL();

	void DoAddToGroupL();		

	void DoCompactL();

private:

	CContactDatabase* iDb;

	RArray<TBool> iVerbose;

	TInt iOptContactId;

	TInt iOptDestGrpId;

	HBufC* iMatch;

	TBool iShowFirstPhoneOnly;

	HBufC* iOptFirstName;

	HBufC* iOptLastName;

	HBufC* iOptTelephone;

	HBufC* iOptMobile;

	TFileName2 iOptPhotoFileName;

	HBufC* iOptGroupName;

	HBufC* iOptStr;

	enum

		{

		EList, EAdd, EDelete, EChange, EDeleteAll, ECreateGroup, EAddToGroup, ECompact

		} iCommand;

	};

CCommandBase* CCmdContacts::NewLC()

	{

	CCmdContacts* self = new(ELeave) CCmdContacts();

	CleanupStack::PushL(self);

	self->BaseConstructL();

	return self;

	}

CCmdContacts::~CCmdContacts()

	{

	delete iOptFirstName;

	delete iOptLastName;

	delete iOptTelephone;

	delete iOptMobile;

	delete iOptGroupName;

	delete iOptStr;

	delete iMatch;

	delete iDb;

	iVerbose.Close();

	}

CCmdContacts::CCmdContacts()

	{

	}

const TDesC& CCmdContacts::Name() const

	{

	_LIT(KName, "fcontacts");

	return KName;

	}

void CCmdContacts::ArgumentsL(RCommandArgumentList& aArguments)

	{

	aArguments.AppendEnumL((TInt&)iCommand, _L("command"));

	}

void CCmdContacts::OptionsL(RCommandOptionList& aOptions)

	{

	aOptions.AppendBoolL(iVerbose, _L("verbose"));

	aOptions.AppendUintL((TUint&)iOptContactId, _L("contact-id"));

	aOptions.AppendUintL((TUint&)iOptDestGrpId, _L("group-id"));

	aOptions.AppendStringL(iMatch, _L("match"));

	aOptions.AppendBoolL(iShowFirstPhoneOnly, _L("first-phone-only"));

	aOptions.AppendStringL(iOptFirstName, _L("first-name"));

	aOptions.AppendStringL(iOptLastName, _L("last-name"));

	aOptions.AppendStringL(iOptTelephone, _L("telephone"));

	aOptions.AppendStringL(iOptMobile, _L("mobile"));

	aOptions.AppendFileNameL(iOptPhotoFileName, _L("photo"));

	aOptions.AppendStringL(iOptGroupName, _L("group"));

	}

void CCmdContacts::DoRunL()

	{

	TRAPD(err, iDb = CContactDatabase::OpenL());

	LeaveIfErr(err, _L("Couldn't open contacts database"));

	switch (iCommand)

		{

	case EList:

		DoListL();

		break;

	case EAdd:

		DoAddL();

		break;

	case EDelete:

		DoDeleteL();

		break;

	case EChange:

		DoChangeL();

		break;

	case EDeleteAll:

		DoDeleteAllL();

		break;

	case ECreateGroup:

		DoCreateGroupL();

		break;

	case EAddToGroup:

		DoAddToGroupL();	

		break;

	case ECompact:

		DoCompactL();

		break;

		}

	}

void CCmdContacts::DoListL()

	{

	if (iOptContactId)

		{

		CContactItem* contact = NULL;

		TRAPD(err, contact = iDb->ReadContactL(iOptContactId));

		LeaveIfErr(err, _L("Couldn't read contact %d"), iOptContactId);

		PrintContact(contact);

		delete contact;

		}

	else

		{

		// List all

		TContactIter iter(*iDb);

		TContactItemId id;

		TBool foundMatch = EFalse;

		while ((id = iter.NextL()) != KNullContactId)

			{

			CContactItem* contact = iDb->ReadContactLC(id);

			TBool show = ETrue;

			if (iMatch)

				{

				show = EFalse;

				CContactItemFieldSet& fields = contact->CardFields();

				TInt num = fields.Count();

				for (TInt i = 0; i < num; i++)

					{

					CContactItemField& field = fields[i];

					switch (field.StorageType())

						{

						case KStorageTypeText:

							if (field.TextStorage()->Text().MatchF(*iMatch) != KErrNotFound)

								{

								show = ETrue;

								foundMatch = ETrue;

								}

							break;

						default:

							break;

						}

					if (show) break;

					}

				}

			if (show)

				{

				PrintContact(contact);

				}

			CleanupStack::PopAndDestroy(contact);

			if (show && iShowFirstPhoneOnly)

				{

				break;

				}

			}

		if (iMatch && !foundMatch)

			{

			User::Leave(KErrNotFound);

			}

		}

	}

void CCmdContacts::DoAddL()

	{

	TContactItemId newId = 0;

	//create a "contact card" contact item

	CContactItem* contact = CContactCard::NewLC();

	CContactItemFieldSet& fieldSet = contact->CardFields();

	FillContactFieldSetL(fieldSet, KUidContactCard);

	TRAPD(err, newId = iDb->AddNewContactL(*contact));

	LeaveIfErr(err, _L("Could not add this contact"));

	Printf(_L("Contact Added\r\nContact ID:%d\r\n"), newId);

	CleanupStack::PopAndDestroy(contact);

	}

void CCmdContacts::DoDeleteL()

	{

	if (iOptContactId)

		{

		TRAPD(err, iDb->DeleteContactL(iOptContactId));

		if (err == KErrNone)

			{

			Printf(_L("Contact %d deleted\r\n"), iOptContactId);

			}

		else

			{

			LeaveIfErr(err, _L("Could not detelet contact item %d"), iOptContactId);

			}

		}

	else

		{

		LeaveIfErr(KErrArgument, _L("No contact id specified. Use -i to specify a contact ID"));						

		}	

	}

void CCmdContacts::DoChangeL()

	{

	if (iOptContactId) 

		{

		PrintAcceptedOptions();		

		CContactItem* contact = NULL;

		TRAPD(err, contact = iDb->OpenContactL(iOptContactId));

		LeaveIfErr(err, _L("could not open contact item %d"), iOptContactId);

		CleanupStack::PushL(contact);

		CContactItemFieldSet& FieldSet = contact->CardFields();

		FillContactFieldSetL(FieldSet, KUidContactCard);

		TRAP(err, iDb->CommitContactL(*contact));

		LeaveIfErr(err, _L("Could not commit the change for item %d"), iOptContactId);

		CleanupStack::PopAndDestroy(contact);

		}

	else

		{

		LeaveIfErr(KErrArgument, _L("No contact id specified. Use -i to specify a contact ID"));

		}		

	}

void CCmdContacts::DoDeleteAllL()

	{

	TInt TotalErrors = 0;	//how many items are not deleted due to error 

	TInt ContactItemCnt = iDb->CountL();

	Printf(_L("Database contains %d items \r\n"), ContactItemCnt);

	// List all

	TContactIter iter(*iDb);

	TContactItemId id;

	while ((id = iter.NextL()) != KNullContactId)

		{

		Printf(_L("Deleting contact item %d ..."), id);

		TRAPD(err, iDb->DeleteContactL(id));

		if (err==KErrNone)

			Printf(_L("done \r\n"), id);

		else 

			{

			TotalErrors++;

			Printf(_L("fail, error code=%d \r\n"), err);

			}		

		}

	if (TotalErrors > 0)

		{

		LeaveIfErr(KErrGeneral, _L("%d items are not deleted due to error"), TotalErrors);

		}	

	}

void CCmdContacts::DoCreateGroupL()

	{

	Printf(_L("Creating a new group... \r\n"));

	CContactItem* contact = NULL;

	TRAPD(err, contact = iDb->CreateContactGroupL());

	LeaveIfErr(err, _L("Could not create contact group"));

	CleanupStack::PushL(contact);

	TContactItemId GroupId = contact->Id();

	CleanupStack::PopAndDestroy(contact);

	contact = NULL;

	///////////////////////////////////////////////////

	TRAP(err, contact = iDb->OpenContactL(GroupId));

	LeaveIfErr(err, _L("Could not open contact group for editing"));

	CleanupStack::PushL(contact);

	Printf(_L("Changing group name for contact item %d... \r\n"), GroupId);

	CContactItemFieldSet& FieldSet = contact->CardFields();

	FillContactFieldSetL(FieldSet, KUidContactGroup);

	TRAP(err, iDb->CommitContactL(*contact));

	LeaveIfErr(err, _L("Could not sumbit the change"));

	CleanupStack::PopAndDestroy(contact);

	}

void CCmdContacts::DoCompactL()

	{

	TRAPD(err, iDb->CompactL());

	LeaveIfErr(err, _L("Could not compact the database"));

	}

void CCmdContacts::DoAddToGroupL()

	{

	if (iOptContactId && iOptDestGrpId)

		{

		Printf(_L("Adding contact item %d to destination group item %d \r\n"), 

				iOptContactId, iOptDestGrpId);

		TRAPD(err, iDb->AddContactToGroupL( (TContactItemId)iOptContactId, (TContactItemId)iOptDestGrpId) ) ;

		//this is most common mistake, need to point out especially to remind the user.

		if (err==KErrNotSupported)

			PrintError(KErrNotSupported, _L("Destination is not a group"));

		User::LeaveIfError(err);

		}

	else 

		{

		LeaveIfErr(KErrArgument, _L("No contact id specified. Use -i to specify a contact ID"));						

		}	

	}

// FOR TEST PURPOSE

//to check if a field of the same type has already existed in a FieldSet.

//if yes, delete that field first and then add it

TInt CCmdContacts::CheckDuplicateAndUpdate(CContactItemFieldSet &aFieldSet, CContactItemField &aField) 

	{

	const CContentType &newContentType=aField.ContentType();

	TInt fieldPos=0;

	TInt FieldCnt = aFieldSet.Count();

	for(; fieldPos<FieldCnt ; fieldPos++ )

		{

		CContactItemField& TestedField = aFieldSet[fieldPos];

		if (newContentType == TestedField.ContentType())

			{

				aFieldSet.Remove(fieldPos);

				break;

			}

		}

	if (aField.Storage()->IsFull())

		{

		CContactItemField* field=CContactItemField::NewLC(aField);

		aFieldSet.InsertL(fieldPos,*field);

		CleanupStack::Pop();	// field

		}	

	return 0;

	}

//this function will retrieve infomation from command line arguments and fill them into 

//a field set, FieldSet doesn't have to be empty

//aType: either KUidContactCard or KUidContactGroup

void CCmdContacts::FillContactFieldSetL(CContactItemFieldSet &aFieldSet, TUid aType)

	{	

	if (aType == KUidContactGroup)

		{

		//group name

		if (iOptGroupName && iOptGroupName->Length())

			{		

			CContactItemField* field = NULL;

			field = CContactItemField::NewLC(KStorageTypeText);

			field->SetLabelL(_L("Group Label"));

			field->SetMapping(KUidContactFieldNone);

			field->AddFieldTypeL(KUidContactFieldTemplateLabel);	

			CContactTextField* pFieldText = field->TextStorage();

			pFieldText->SetTextL(*iOptGroupName);

			aFieldSet.UpdateFieldL(*field, 1);

			CleanupStack::PopAndDestroy(field);	

			}

		}

	//Synchronization, for both Group and normal contacts

		{

		CContactItemField* field = NULL;		

		field = CContactItemField::NewLC(KStorageTypeText);

		field->SetLabelL(_L("Synchronization"));

		field->SetMapping(KUidContactFieldVCardMapClass);

		field->AddFieldTypeL(KUidContactFieldClass);	

		CContactTextField* pFieldText = field->TextStorage();

		pFieldText->SetTextL(_L("private"));

		aFieldSet.UpdateFieldL(*field, 1);

		CleanupStack::PopAndDestroy(field);		

		}

	if (aType == KUidContactCard)

		{

		//last name

		if (iOptLastName && iOptLastName->Length())

			{		

			CContactItemField* field = NULL;

			field = CContactItemField::NewLC(KStorageTypeText);

			field->SetLabelL(_L("Last name"));

			field->SetMapping(KUidContactFieldVCardMapUnusedN);

			field->AddFieldTypeL(KUidContactFieldFamilyName);	

			CContactTextField* pFieldText = field->TextStorage();

			pFieldText->SetTextL(*iOptLastName);

			aFieldSet.UpdateFieldL(*field, 1);

			CleanupStack::PopAndDestroy(field);	

			}

		//first name

		if (iOptFirstName && iOptFirstName->Length())

			{	

			CContactItemField* field = NULL;

			field = CContactItemField::NewLC(KStorageTypeText);

			field->SetLabelL(_L("First name"));

			field->SetMapping(KUidContactFieldVCardMapUnusedN);

			field->AddFieldTypeL(KUidContactFieldGivenName);	

			CContactTextField* pFieldText = field->TextStorage();

			pFieldText->SetTextL(*iOptFirstName);

			aFieldSet.UpdateFieldL(*field, 1);

			CleanupStack::PopAndDestroy(field);	

			}			

		//general telephone

		if (iOptTelephone && iOptTelephone->Length())

			{	

			CContactItemField* field = NULL;

			field = CContactItemField::NewLC(KStorageTypeText);

			field->SetLabelL(_L("Telephone"));

			field->SetMapping(KUidContactFieldVCardMapTEL);

			field->AddFieldTypeL(KUidContactFieldPhoneNumber);	

			field->AddFieldTypeL(KUidContactFieldVCardMapVOICE);		

			CContactTextField* pFieldText = field->TextStorage();

			pFieldText->SetTextL(*iOptTelephone);

			aFieldSet.UpdateFieldL(*field, 1);

			CleanupStack::PopAndDestroy(field);	

			}			

		//mobile phone

		if (iOptMobile && iOptMobile->Length())

			{	

			CContactItemField* field = NULL;

			field = CContactItemField::NewLC(KStorageTypeText);

			field->SetLabelL(_L("Mobile"));

			field->SetMapping(KUidContactFieldVCardMapTEL);

			field->AddFieldTypeL(KUidContactFieldPhoneNumber);

			field->AddFieldTypeL(KUidContactFieldVCardMapCELL);

			CContactTextField* pFieldText = field->TextStorage();

			pFieldText->SetTextL(*iOptMobile);

			aFieldSet.UpdateFieldL(*field, 1);

			CleanupStack::PopAndDestroy(field);	

			}

		//photo

		if (iOptPhotoFileName.Length())

			{

			RFile file;

			LeaveIfErr(file.Open(FsL(), iOptPhotoFileName, EFileRead), _L("Couldn't open file \"%S\""), &iOptPhotoFileName);

			TInt fileSize;

			LeaveIfErr(file.Size(fileSize), _L("Couldn't read the size of file \"%S\""), &iOptPhotoFileName);

			RBuf8 buf;

			buf.Create(fileSize);

			buf.CleanupClosePushL();

			LeaveIfErr(file.Read(buf), _L("Couldn't read the contents of file \"%S\""), &iOptPhotoFileName);

			CContactItemField* field = CContactItemField::NewLC(KStorageTypeStore);

			field->SetLabelL(_L("Photo"));

			field->SetMapping(KUidContactFieldVCardMapPHOTO);

			field->AddFieldTypeL(KUidContactFieldPicture);

			CContactStoreField* storeField = field->StoreStorage();

			storeField->SetThingL(buf);

			aFieldSet.UpdateFieldL(*field, 1);

			CleanupStack::PopAndDestroy(2, &buf);	

			}

		}

	}

void CCmdContacts::PrintContact(CContactItem* aContact)

	{

	TContactItemId ContactId = aContact->Id();

	TUid ContactType = aContact->Type();

	CContactItemFieldSet& fields = aContact->CardFields();

	TBool verbose = (iVerbose.Count() > 0);

	TBool veryverbose = (iVerbose.Count() > 1);

	if (verbose)