/*
* 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: 
*       Fetch operation with checking of available disk space
*
*/


#include <eikenv.h>
#include <eikrutil.h>

#include "Imap4SizeAndFetchOp.h"
#include "Imap4FetchOp.h"
#include "EmailMsgSizerOperation.h"
#include "EmailUtils.H"
#include "ImumDisconnectOperation.h"
#include "IMAPPRGR.H"
#include "ImumMtmLogging.h"
#include "ImumPanic.h"

// Constants
const TInt KSizeAndFetchPriority = CActive::EPriorityStandard;

// ----------------------------------------------------------------------------
// NewL
// ----------------------------------------------------------------------------
CImap4SizeAndFetchOp* CImap4SizeAndFetchOp::NewL(
    CImumInternalApi& aMailboxApi,
    TRequestStatus& aStatus,
    MMsvProgressReporter& aReporter,
    TMsvId aService,
    TInt aFunctionId,
    const TImImap4GetMailInfo& aGetMailInfo,
    const CMsvEntrySelection& aSel)
    {
    IMUM_STATIC_CONTEXT( CImap4SizeAndFetchOp::NewL, 0, mtm, KImumMtmLog );
    IMUM_IN();
    
    CImap4SizeAndFetchOp* op = new(ELeave) CImap4SizeAndFetchOp(
        aMailboxApi, aStatus, aReporter, aService, aFunctionId, aGetMailInfo);
    CleanupStack::PushL(op);
    op->ConstructL(aSel);
    CleanupStack::Pop();
    IMUM_OUT();
    return op;
    }

// ----------------------------------------------------------------------------
// ~CImap4SizeAndFetchOp
// ----------------------------------------------------------------------------
CImap4SizeAndFetchOp::~CImap4SizeAndFetchOp()
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::~CImap4SizeAndFetchOp, 0, KImumMtmLog );
    IMUM_IN();
    IMUM_OUT();
    
    delete iSelection;
    }

// ----------------------------------------------------------------------------
// GetErrorProgressL
// ----------------------------------------------------------------------------
const TDesC8& CImap4SizeAndFetchOp::GetErrorProgressL(TInt aError)
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::GetErrorProgressL, 0, KImumMtmLog );
    IMUM_IN();
    
    TImap4CompoundProgress& progress = iProgressBuf();
    progress.iGenericProgress.iOperation = TImap4GenericProgress::EPopulate;
    progress.iGenericProgress.iState = TImap4GenericProgress::EFetching;
    progress.iGenericProgress.iErrorCode = aError;
    IMUM_OUT();
    return iProgressBuf;
    }

// ----------------------------------------------------------------------------
// CImap4SizeAndFetchOp
// ----------------------------------------------------------------------------
CImap4SizeAndFetchOp::CImap4SizeAndFetchOp(
    CImumInternalApi& aMailboxApi,
    TRequestStatus& aStatus,
    MMsvProgressReporter& aReporter,
    TMsvId aService,
    TInt aFunctionId,
    const TImImap4GetMailInfo& aGetMailInfo )
    :
    CImumOnlineOperation(aMailboxApi, KSizeAndFetchPriority, aStatus, aReporter),
    iState(EStateIdle),
    iFunctionId(aFunctionId),
    iGetMailInfo(aGetMailInfo)
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::CImap4SizeAndFetchOp, 0, KImumMtmLog );
    IMUM_IN();
    
    iService = aService;
    IMUM_OUT();
    }

// ----------------------------------------------------------------------------
// ConstructL
// ----------------------------------------------------------------------------
void CImap4SizeAndFetchOp::ConstructL(const CMsvEntrySelection& aSel)
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::ConstructL, 0, KImumMtmLog );
    IMUM_IN();
    
    __ASSERT_DEBUG(aSel.Count(),
        User::Panic(KImumMtmUiPanic,EPanicSelectionEmpty));
    if ( aSel.Count() == 0 )
        {
        User::Leave( KErrNotSupported );
        }
    iSelection = aSel.CopyL();
#ifdef _DEBUG
    const TInt count = aSel.Count();
    for ( TInt loop = 0; loop < count; loop++ )
        {
        }
#endif

    BaseConstructL(KUidMsgTypeIMAP4);
    DoSizeMessageL();
    IMUM_OUT();
    }

// ----------------------------------------------------------------------------
// DoRunL
// ----------------------------------------------------------------------------
void CImap4SizeAndFetchOp::DoRunL()
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::DoRunL, 0, KImumMtmLog );
    IMUM_IN();
    
    switch(iState)
        {
        case EStateSizeMsg:
            // Finished determining message size.
            if(!CheckMessageSizeL())
                {
                // Not enough storage, user already informed.
                CompleteObserver();
                iState = EStateIdle;
                }
            else
                {
                DoSizeNextMessageL();
                }
            break;
        case EStateFetchMsg:
            // Finished fetching message.
            CompleteObserver();
            iState = EStateIdle;
            break;
        case EStateIdle:
        default:
            break;
        }
    IMUM_OUT();
    }

// ----------------------------------------------------------------------------
// DoSizeMessageL
// ----------------------------------------------------------------------------
void CImap4SizeAndFetchOp::DoSizeMessageL()
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::DoSizeMessageL, 0, KImumMtmLog );
    IMUM_IN();
    
    iState = EStateSizeMsg;
    iStatus = KRequestPending;
    delete iOperation;
    iOperation = NULL;
    iOperation = CEmailMsgSizerOperation::NewL(
        iMailboxApi, iStatus, (*iSelection)[ iCurrentMsgToSize ], iMtm);
    SetActive();
    IMUM_OUT();
    }

// ----------------------------------------------------------------------------
// DoSizeNextMessageL
// ----------------------------------------------------------------------------
void CImap4SizeAndFetchOp::DoSizeNextMessageL()
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::DoSizeNextMessageL, 0, KImumMtmLog );
    IMUM_IN();
    
    iCurrentMsgToSize++;
    if ( iCurrentMsgToSize < iSelection->Count() )
        {
        DoSizeMessageL();
        }
    else
        {
        DoFetchMessageL();
        }
    IMUM_OUT();
    }

// ----------------------------------------------------------------------------
// DoFetchMessageL
// ----------------------------------------------------------------------------
void CImap4SizeAndFetchOp::DoFetchMessageL()
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::DoFetchMessageL, 0, KImumMtmLog );
    IMUM_IN();
    
    iState = EStateFetchMsg;
    iStatus = KRequestPending;
    delete iOperation;
    iOperation = NULL;
    iOperation = CImap4FetchOp::NewL(
        iMailboxApi, iStatus, iReporter, iService,
        iFunctionId, iGetMailInfo, *iSelection);
    SetActive();
    IMUM_OUT();
    }

// ----------------------------------------------------------------------------
// CheckMessageSizeL
// ----------------------------------------------------------------------------
TBool CImap4SizeAndFetchOp::CheckMessageSizeL()
    {
    IMUM_CONTEXT( CImap4SizeAndFetchOp::CheckMessageSizeL, 0, KImumMtmLog );
    IMUM_IN();
    
    CEmailMsgSizerOperation& sizer = *STATIC_CAST(CEmailMsgSizerOperation*, iOperation);
    TInt32 msgSize = 0;

    switch(iGetMailInfo.iGetMailBodyParts)
        {
        case EGetImap4EmailBodyText:
            msgSize = sizer.BodyTextSizeL();
            break;
        case EGetImap4EmailBodyAlternativeText:
        case EGetImap4EmailBodyTextAndAttachments:
            msgSize = sizer.TotalMessageSize();
            break;
        case EGetImap4EmailAttachments:
            msgSize = sizer.SizeOfAllAttachmentsL();
            break;
        case EGetImap4EmailHeaders:
        default:
            break;
        }
    iTotalSizeToFetch += msgSize;
    
    IMUM_OUT();
    
    return ETrue;
    }

// End of File