/*

* Copyright (c) 2006 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:  Message store encrypt handler implementation.

*

*/

#include "MessageStoreEncryptHandler.h"

#include "ContainerStoreGeneralTable.h"

#include "ContainerStoreUtils.h"

// ======================

// METHOD IMPLEMENTATIONS

// ======================

// ==========================================================================

// FUNCTION: NewL

// ==========================================================================

CMessageStoreEncryptHandler* CMessageStoreEncryptHandler::NewL( CContainerStoreGeneralTable& aGeneralTable,

                                                                  CContainerStoreUtils&        aUtils,

                                                                  TInt                         aPriority )

    {

    CMessageStoreEncryptHandler* self = new(ELeave) CMessageStoreEncryptHandler( aGeneralTable, aUtils, aPriority );

    CleanupStack::PushL( self );

    self->ConstructL();

    CleanupStack::Pop( self );

    return self;

    }

// ==========================================================================

// FUNCTION: CMessageStoreEncryptHandler

// ==========================================================================

CMessageStoreEncryptHandler::CMessageStoreEncryptHandler( CContainerStoreGeneralTable&    aGeneralTable,

                                                            CContainerStoreUtils&           aUtils,

                                                            TInt                            aPriority )

: CActive( aPriority ),

  iGeneralTable( aGeneralTable ),

  iUtils( aUtils )

    {

    __LOG_CONSTRUCT( "msg", "CMessageStoreEncryptHandler" )

    }

// ==========================================================================

// FUNCTION: ConstructL

// ==========================================================================

void CMessageStoreEncryptHandler::ConstructL()

    {

    InitState();

    CActiveScheduler::Add(this);    

    }

// ==========================================================================

// FUNCTION: Destructor

// ==========================================================================

CMessageStoreEncryptHandler::~CMessageStoreEncryptHandler()

    {

    __LOG_DESTRUCT

    Cancel();

    iEncryptableTables.Close();

    }

// ==========================================================================

// FUNCTION: AddEncryptableTable

// ==========================================================================

void CMessageStoreEncryptHandler::AddEncryptableTableL( MEncryptableTable* aEncryptableTable )

    {

    iEncryptableTables.AppendL( aEncryptableTable );

    }

// ==========================================================================

// FUNCTION: ResumeIfNeededL

// ==========================================================================

void CMessageStoreEncryptHandler::StartL()

    {

    __LOG_ENTER_SUPPRESS( "StartL" )

    iEncryptionState = iGeneralTable.EncryptionStateL();

    if ( iEncryptionState != EMsgStoreESIdle )

        {

        __LOG_WRITE_INFO("Start/Resume Encrypt/Decrypt")

        InitState();

        ScheduleNext();

        }

    }

// ==========================================================================

// FUNCTION: EncryptL

// ==========================================================================

void CMessageStoreEncryptHandler::EncryptL()

    {

    if ( iEncryptionState != EMsgStoreESIdle  )

        {

        User::Leave( KErrInUse );

        }

    iGeneralTable.SetEncryptionStateL( EMsgStoreESEncrypting );

    StartL();

    }

// ==========================================================================

// FUNCTION: DecryptL

// ==========================================================================

void CMessageStoreEncryptHandler::DecryptL()

    {

    if ( iEncryptionState != EMsgStoreESIdle  )

        {

        User::Leave( KErrInUse );

        }

    iGeneralTable.SetEncryptionStateL( EMsgStoreESDecrypting );

    StartL();

    }

void CMessageStoreEncryptHandler::ScheduleNext()

    {

    // Schedule next operation (yield, in case higher priority active objects have something to do).

    SetActive();

    TRequestStatus* status = &iStatus;

    User::RequestComplete( status, KErrNone );      

    }

// ==========================================================================

// FUNCTION: RunL

// ==========================================================================

void CMessageStoreEncryptHandler::RunL()

    {

    DoNextOperationL();

    }

// ==========================================================================

// FUNCTION: DoCancel

// ==========================================================================

void CMessageStoreEncryptHandler::DoCancel()

    {

    //Nothing to do

    }

// ==========================================================================

// FUNCTION: StartL

// ==========================================================================

void CMessageStoreEncryptHandler::DoNextOperationL()

    {

    __LOG_ENTER_SUPPRESS( "DoNextOperationL" )

    if ( iEncryptionState != EMsgStoreESIdle )

        {

        iUtils.SuspendCompactionLC();

        iUtils.BeginDatabaseTransactionLC();

        MEncryptableTable* curTable = iEncryptableTables[iCurTableIndex];

        TBool hasMore = EFalse;

        if ( iIsFirstInTable )

            {

            if ( iEncryptionState == EMsgStoreESEncrypting )

                {

                hasMore = curTable->EncryptFirstL( iBookmark );

                }

            else

                {

                hasMore = curTable->DecryptFirstL( iBookmark );

                }

            iIsFirstInTable = EFalse;

            }

        else

            {

            if ( iEncryptionState == EMsgStoreESEncrypting )

                {

                hasMore = curTable->EncryptNextL( iBookmark );

                }

            else

                {

                hasMore = curTable->DecryptNextL( iBookmark );

                }

            }

        if ( !hasMore )

            {

            iCurTableIndex++;

            if ( iCurTableIndex >= iEncryptableTables.Count() )

                {

                //we are done, update the table and do NOT schedule next

                iGeneralTable.SetEncryptionStateL( EMsgStoreESIdle );

                iEncryptionState = EMsgStoreESIdle;

                }

            else

                {

                //start encrypt/decrypt the next table in next RunL()

                iIsFirstInTable = ETrue;

                ScheduleNext();

                }

            }

        else

            {

            ScheduleNext();

            }

        iUtils.CommitDatabaseTransactionL();

        iUtils.ResumeCompaction();

        }

    }

void CMessageStoreEncryptHandler::InitState()

    {

    iCurTableIndex = 0;

    iIsFirstInTable = ETrue;

    }