/*

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

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "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:  DRM3 Engine manages all DRM related database operations.

*

*/

// INCLUDE FILES

#include <e32std.h>

#include <s32mem.h> 

#include "DRMNotifierSession.h"

#include "DRMEventAddRemove.h"

#include "DRMEventModify.h"

#include "DRMEventTimeChange.h"

#include "drmnotifierclientserver.h"

// EXTERNAL DATA STRUCTURES

// EXTERNAL FUNCTION PROTOTYPES  

// CONSTANTS

// MACROS

#ifdef _DRM_TESTING

_LIT( KDateTimeFormat, "%F%Y%M%D%H%T%S%C" );

#define _LOGBUFNUM( a, b ) { TBuf8< 64 > __b( a ); __b.AppendNum( b ); file.Write( __b ); }

#define LOGBUFNUM( a, b ) { TBuf8< 64 > __b( a ); __b.AppendNum( b ); Log( __b ); }

#define LOG( a ) Log( a )

#else

#define _LOGBUFNUM( a, b )

#define LOGBUFNUM( a, b )

#define LOG( a )

#endif

// LOCAL CONSTANTS AND MACROS

// MODULE DATA STRUCTURES

// LOCAL FUNCTION PROTOTYPES

// FORWARD DECLARATIONS

// ============================= LOCAL FUNCTIONS ===============================

// ============================ MEMBER FUNCTIONS ===============================

// -----------------------------------------------------------------------------

// CDRMMessageStorage::NewLC

// Two-phased constructor.

// -----------------------------------------------------------------------------

//

CDRMMessageStorage* CDRMMessageStorage::NewL()

    {

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

    CleanupStack::PushL( self );

    self->ConstructL();

    CleanupStack::Pop( self );

    return self;

    }

// -----------------------------------------------------------------------------

// Destructor

// -----------------------------------------------------------------------------

CDRMMessageStorage::~CDRMMessageStorage() 

    {

    for ( TInt i = 0; i < iMessages.Count();i++ )

        {

        delete iMessages[i]->iData;

        delete iMessages[i]->iMessageData;

        }

    iMessages.ResetAndDestroy();

    iSessions.Reset();

    LOG( _L8( "Dying..." ) );

#ifdef _DRM_TESTING

    delete iLog;

#endif

    }

// -----------------------------------------------------------------------------

// CDRMMessageStorage::AddSession

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::AddSession( CDRMNotifierSession* aSession ) 

    {

    iSessions.Append( aSession );

    };

// -----------------------------------------------------------------------------

// CDRMMessageStorage::UpdateMessage

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::UpdateMessage( TMessageData* aMessage )

    {

    TInt count = 0;

    aMessage->iRefCount = aMessage->iRefCount - 1;

    if(aMessage->iRefCount == 0)

        {

        for( count = 0; count < iMessages.Count(); count++ )

            {

            if( iMessages[count] == aMessage )

                {

                delete iMessages[count]->iMessageData;

                delete iMessages[count]->iData;

                iMessages[count]->iMessageData = 0;

                iMessages[count]->iData = 0;

                }

            iMessages.Remove(count);

            count = iMessages.Count();

            }

        }

    }

// -----------------------------------------------------------------------------

// CDRMMessageStorage::GetEventObjectLC

// -----------------------------------------------------------------------------

//

MDRMEvent* CDRMMessageStorage::GetEventObjectLC( TDRMEventType aEventType )

    {

    MDRMEvent* event = NULL;

    switch( aEventType)

        {

        case KEventAddRemove:

            event = CDRMEventAddRemove::NewLC(ERightsObjectRecieved);

            break;

        case KEventModify:

            event = CDRMEventModify::NewLC();

            break;

        case KEventTimeChange:

            event = CDRMEventTimeChange::NewLC();

            break;

	    default:

	        User::Leave(KErrArgument);

	        break;

        }           

    return event;

    }

// -----------------------------------------------------------------------------

// CDRMMessageStorage::NotifyL

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::NotifyL(TDRMEventType& aEventType, const RMessage2& aMessage) 

    {

    TInt queued = EFalse;

    TMessageData* message = new (ELeave) TMessageData;

    CleanupStack::PushL(message);

    TUint8* eventBuf = static_cast<TUint8*>(User::AllocLC(DRMNotifier::KDRMSizeOfMessage));

    TPtr8 event( eventBuf,0,DRMNotifier::KDRMSizeOfMessage );

    message->iRefCount = 0;

    message->iEventType = aEventType;

    message->iData = 0;

    message->iMessageData = eventBuf;

    // Read the buffer:

    aMessage.ReadL( 0, event);

    RMemReadStream input(eventBuf, DRMNotifier::KDRMSizeOfMessage);

    MDRMEvent* eventobj = GetEventObjectLC(aEventType);

    eventobj->InternalizeL(input);

    // We need to extract the uri from the message

    // for those that it is needed

    switch( aEventType )

        {

        case KEventAddRemove:

            message->iData = static_cast<CDRMEventAddRemove*>(eventobj)->GetContentIDL();

            break;

        case KEventModify:

            message->iData = static_cast<CDRMEventModify*>(eventobj)->GetContentIDL();

            break;

        case KEventTimeChange:

            break;

        default:

            User::Leave(KErrArgument);

            break;

        }

    CleanupStack::PopAndDestroy(); // eventobj

    CleanupStack::PushL(message->iData);

    for( TInt i = 0; i < iSessions.Count();i++ )

        {

        queued = EFalse;

        TRAPD(error, queued = iSessions[i]->SendNotificationL( message ) );

        if( error && error != KErrNotFound )

            {

            User::LeaveIfError(error);

            }

        else if( queued )

            {

            message->iRefCount++;

            }

        }

    if( !message->iRefCount ) 

        {

        CleanupStack::PopAndDestroy(); // message->iData;

        CleanupStack::PopAndDestroy(); // eventBuf;

        CleanupStack::PopAndDestroy(); // message

        }

    else

        {

        iMessages.Append( message );

        CleanupStack::Pop(); // message->iData;

        CleanupStack::Pop(); // eventBuf

        CleanupStack::Pop(); // message

        }

    };

// -----------------------------------------------------------------------------

// CDRMMessageStorage::CancelL

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::CancelL(CDRMNotifierSession* aSession)

    {

    TInt count = 0;

    for ( count = 0; count < iSessions.Count(); count++ )

        {

        if( iSessions[count] == aSession)

            {

            iSessions.Remove(count);

            return;

            }

        } 

    User::Leave(KErrNotFound);

    }

// -----------------------------------------------------------------------------

// CDRMMessageStorage::CDRMMessageStorage

// C++ default constructor can NOT contain any code, that

// might leave.

// -----------------------------------------------------------------------------

//

CDRMMessageStorage::CDRMMessageStorage() : iSessions(4), iMessages(4)

    {

    // Nothing

    }

// -----------------------------------------------------------------------------

// CDRMMessageStorage::ConstructL

// Symbian 2nd phase constructor can leave.

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::ConstructL() 

    {

    }

#ifdef _DRM_TESTING

// -----------------------------------------------------------------------------

// CDRMMessageStorage::Log

// Logging operation.

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::Log( const TDesC8& aLog ) const

    {

    iLog->Log( aLog );

    }

// -----------------------------------------------------------------------------

// CDRMMessageStorage::Log

// Logging operation.

// -----------------------------------------------------------------------------

//

void CDRMMessageStorage::Log( const TDesC& aLog ) const

    {

    iLog->Log( aLog );

    }

#endif

// ========================== OTHER EXPORTED FUNCTIONS =========================

// End of File