/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtCore module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "cntrelationshipgroup.h"

#include <cntitem.h>

#include <QDebug>

/*!

 * Conatructor

 * 

 * \a contactDatabase CContactDatabase with established connection to the database

 * \a managerUri current contact manager uri

 */

CntRelationshipGroup::CntRelationshipGroup(CContactDatabase *contactDatabase, const QString &managerUri) :

    CntAbstractRelationship(contactDatabase, managerUri, QContactRelationship::HasMember)

{

}

/*!

 * Destructor

 */

CntRelationshipGroup::~CntRelationshipGroup()

{

}

/*!

 * Save a single relatonship

 * 

 * \a affectedContactIds will include the affected contact ids 

 * \a relationship to be saved

 * \a error t

 */

bool CntRelationshipGroup::saveRelationshipL(QSet<QContactLocalId> *affectedContactIds, QContactRelationship* relationship, QContactManager::Error* error)

{

    //get the ids of the relationship

    QScopedPointer<QContactId> groupId(new QContactId(relationship->first()));

    QScopedPointer<QContactId> contactId(new QContactId(relationship->second()));

    //cntmodel accepts contact id 0, which is considered an error in qt contacts

    database()->AddContactToGroupL(TContactItemId(contactId->localId()), TContactItemId(groupId->localId()));

    //add the group  and group member to the list of affected contacts

    //note if the value already exists in the QSet nothing happens

    affectedContactIds->insert(groupId->localId());

    affectedContactIds->insert(contactId->localId());

    *error = QContactManager::NoError;

    return true;

}

bool CntRelationshipGroup::removeRelationshipL(QSet<QContactLocalId> *affectedContactIds, const QContactRelationship& relationship, QContactManager::Error* error)

{

    //get the ids of the relationship

    QScopedPointer<QContactId> groupId(new QContactId(relationship.first()));

    QScopedPointer<QContactId> contactId(new QContactId(relationship.second()));

    //read the contacts from the database

    CContactItem* groupContact = database()->ReadContactLC(groupId->localId());

    CContactItem* contact      = database()->ReadContactLC(contactId->localId());

    //Check if contact is part of the group

    isGroupMemberL(contact, groupContact->Id());

    //remove contact doesn't return an error if the group is not part of the group

    database()->RemoveContactFromGroupL(contact->Id(), groupContact->Id());

    //add the removed group member to the list of affected contacts

    affectedContactIds->insert(groupId->localId());

    affectedContactIds->insert(contactId->localId());

    CleanupStack::PopAndDestroy(contact);

    CleanupStack::PopAndDestroy(groupContact);

    *error = QContactManager::NoError;

    return true;

}

bool CntRelationshipGroup::validateRelationship(const QContactRelationship &relationship, QContactManager::Error* error)

{

    // check that "second" is in this manager

    if (!relationship.second().managerUri().isEmpty() && relationship.second().managerUri() != managerUri()) {

        *error = QContactManager::InvalidRelationshipError;

        return false;

    }

    return true;

}

/*

 * Function will leave with KErrNotFound if contact is not part of the specific goup

 * 

 */

void CntRelationshipGroup::isGroupMemberL(const CContactItem* contactItem, const TContactItemId groupId) const

{

    if(contactItem &&

            (contactItem->Type() == KUidContactCard

            || contactItem->Type() == KUidContactOwnCard

            || contactItem->Type() == KUidContactGroup

            || contactItem->Type() == KUidContactICCEntry))

    {

        //cast the contact item to base class for groups and contacts

        const CContactItemPlusGroup *contact = static_cast<const CContactItemPlusGroup *>(contactItem);

        const CContactIdArray *idArray = contact->GroupsJoined();

        //array is null if contact is not part of any groups

        if(idArray)

        {

            //find the group from the array

            TInt arrayIndex = idArray->Find(groupId);

            //if it doesn't exist leave

            if(arrayIndex == KErrNotFound)

            {

                User::Leave(KErrNotFound);

            }

        }

    }

    else

    {

        User::Leave(KErrNotFound);

    }

}

//retrieve all the groups that the contact is part of

QList<QContactRelationship> CntRelationshipGroup::relationshipsL(const QContactId& participantId, QContactRelationship::Role role, QContactManager::Error* error)

{

    QList<QContactRelationship> returnValue;

    if (participantId != QContactId())

    {

        //role is a group

        if(role == QContactRelationship::First || role == QContactRelationship::Either)

        {

            fetchGroupMembersL(participantId.localId(), &returnValue);

        }

        //role is member of a group

        if(role == QContactRelationship::Second || role == QContactRelationship::Either)

        {

            fetchMemberOfGroupsL(participantId.localId(), &returnValue);

        }

    }

    else

    {

        // If participantId is the default-constructed id, role is ignored and

        // all relationships of the given relationshipType are returned.

        fetchMembersOfAllGroupsL(&returnValue);

    }

    if (returnValue.isEmpty())

        *error = QContactManager::DoesNotExistError;

    return returnValue;

}

void CntRelationshipGroup::fetchGroupMembersL(const QContactLocalId& participantId, QList<QContactRelationship> *relationships)

{

    //fetch the contact item from the database

    CContactItem *contactItem = database()->ReadContactLC(TContactItemId(participantId));

    //make sure it's a group

    if(contactItem && contactItem->Type() == KUidContactGroup)

    {

        //cast the contact to a group

        CContactGroup *groupContact = static_cast<CContactGroup*>( contactItem );

        //create the relationship

        QContactRelationship relationship;

        relationship.setRelationshipType(QContactRelationship::HasMember);

        //get the group contacts

        const CContactIdArray *idArray = groupContact->ItemsContained();

        QContactId groupId;

        //loop through all the contacts and add them to the list

        for(int i = 0; i < idArray->Count(); i++ )

        {

            //set participant id as first id

            QContactId first;

            first.setLocalId(participantId);

            first.setManagerUri(managerUri());

            relationship.setFirst(first);

            //set the manager uri

            groupId.setManagerUri(managerUri());

            groupId.setLocalId(idArray->operator[](i));

            relationship.setSecond(groupId);

            *relationships << relationship;

        }

    }

    CleanupStack::PopAndDestroy(contactItem);

}

void CntRelationshipGroup::fetchMemberOfGroupsL(const QContactLocalId& participantId, QList<QContactRelationship> *relationships)

{

    CContactItem *contactItem = database()->ReadContactLC(TContactItemId(participantId));

    if(contactItem &&

            (contactItem->Type() == KUidContactCard

            || contactItem->Type() == KUidContactOwnCard

            || contactItem->Type() == KUidContactGroup

            || contactItem->Type() == KUidContactICCEntry))

    {

        //cast the contact item to base class for groups and contacts

        CContactItemPlusGroup *contact = static_cast<CContactItemPlusGroup *>(contactItem);

        //create the relationship

        QContactRelationship relationship;

        relationship.setRelationshipType(QContactRelationship::HasMember);

        const CContactIdArray *idArray = contact->GroupsJoined();

        if(idArray) {

            QContactId groupId;

            //loop through all the contacts and add them to the list

            for(int i = 0; i < idArray->Count(); i++ )

            {

               //set the manager uri

               groupId.setManagerUri(managerUri());

               groupId.setLocalId(idArray->operator[](i));

               //set the group as first

               relationship.setFirst(groupId);

               //set participant id as member of group

               QContactId second;

               second.setLocalId(participantId);

               second.setManagerUri(managerUri());

               relationship.setSecond(second);

               *relationships << relationship;

            }

        }

    }

    CleanupStack::PopAndDestroy(contactItem);

}

void CntRelationshipGroup::fetchMembersOfAllGroupsL(QList<QContactRelationship> *relationships)

{

    CContactIdArray *groupIdList = database()->GetGroupIdListL();

    CleanupStack::PushL(groupIdList);

    int count = groupIdList->Count();

    for (int i=0; i<count; i++)

    {

        fetchGroupMembersL(QContactLocalId((*groupIdList)[i]), relationships);

    }

    CleanupStack::PopAndDestroy(groupIdList);

}