diff -r 000000000000 -r 72b543305e3a email/imum/Mtms/Src/Imap4SizeAndFetchOp.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/email/imum/Mtms/Src/Imap4SizeAndFetchOp.cpp	Thu Dec 17 08:44:11 2009 +0200
@@ -0,0 +1,263 @@
+/*
+* 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