--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/email/imap4mtm/imapprotocolcontroller/src/cimapupdateflagoperation.cpp	Wed Nov 03 22:55:44 2010 +0530
@@ -0,0 +1,341 @@
+/*
+* Copyright (c) 2009 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:
+*
+*/
+
+
+#include "cimapupdateflagoperation.h"
+#include "cimapfolder.h"
+#include "cimapsyncmanager.h"
+#include "cimapsessionconsts.h"
+#include "cimaplogger.h"
+#include "imappaniccodes.h"
+
+_LIT8(KImapAddFlags, "+FLAGS");
+_LIT8(KStoreFlagsClearCommand, "-FLAGS");
+_LIT8(KImapFlagSeen, "(\\Seen)");
+
+
+CImapUpdateFlagOperation::CImapUpdateFlagOperation(CImapSyncManager& aSyncManager,
+						 				 CMsvServerEntry& aServerEntry,
+						 				 CImapSettings& aImapSettings) :
+	CImapCompoundBase(aSyncManager, aServerEntry, aImapSettings)
+	
+	{
+	
+	}
+
+CImapUpdateFlagOperation::~CImapUpdateFlagOperation()
+	{
+	
+
+	iMessageUids.Reset();
+	iFlagInfoResponseArray.Reset();
+	}
+	
+CImapUpdateFlagOperation* CImapUpdateFlagOperation::NewL(CImapSyncManager& aSyncManager,
+						 				  	   CMsvServerEntry& aServerEntry,
+						 				  	   CImapSettings& aImapSettings)
+	{
+	CImapUpdateFlagOperation* self = new (ELeave) CImapUpdateFlagOperation(aSyncManager,
+																 aServerEntry,
+																 aImapSettings
+																 );
+	CleanupStack::PushL(self);
+	self->ConstructL();
+	CleanupStack::Pop(self);
+	return self;
+	}
+
+void CImapUpdateFlagOperation::ConstructL()
+	{
+	
+	// Add to the active scheduler
+	CActiveScheduler::Add(this);
+	}
+
+// NewL for Selection
+CImapUpdateFlagOperation* CImapUpdateFlagOperation::NewL(CImapSyncManager& aSyncManager,
+                                                         CMsvServerEntry& aServerEntry,
+                                                         CImapSettings& aImapSettings, 
+                                                         const CMsvEntrySelection& aSourceSel , TBool aFlagChanged)
+
+    {
+    
+    CImapUpdateFlagOperation* self = new (ELeave) CImapUpdateFlagOperation(aSyncManager,
+                                                                     aServerEntry,
+                                                                     aImapSettings);
+    CleanupStack::PushL(self);
+    self->ConstructL(aSourceSel,aFlagChanged);
+    CleanupStack::Pop(self);
+    return self;
+    }
+//// ConstructL for Selection   
+void CImapUpdateFlagOperation::ConstructL(const CMsvEntrySelection& aSourceSel, TBool aFlagChanged)
+    {
+  
+    iSourceSel=new (ELeave) CMsvEntrySelection;
+    CheckSelectionL(aSourceSel, iSourceSel, ETrue, EFalse, EFalse, ETrue);
+
+    // Initialise the progress counters 
+    iProgressMsgsToDo=iSourceSel->Count();
+    iProgressMsgsDone=0;
+    iMarkFlag = aFlagChanged;
+    iEntrySelection = ETrue;
+    // Add to the active scheduler
+    CActiveScheduler::Add(this);
+    }
+
+
+
+void CImapUpdateFlagOperation::StartOperation(TRequestStatus& aStatus, CImapSession& aSession)
+//void CImapUpdateFlagOperation::StartOperation(TRequestStatus& aStatus)
+	{
+	iSession = &aSession;
+	__LOG_TEXT(iSession->LogId(), "CImapUpdateFlagOperation::StartOperation()");
+	iNextStep = EUpdateFlag;
+	Queue(aStatus);
+	CompleteSelf();
+	}
+
+/**
+Handles the compound operation state machine
+
+@return ETrue if compound operation is completed, 
+		EFalse otherwise.
+*/	
+TBool CImapUpdateFlagOperation::DoRunLoopL()
+	{
+	SetCurrentStep();
+	switch (iCurrentStep)
+		{
+	
+	case EUpdateFlag:
+		{
+		HBufC8* uidSeq = NULL;
+        if(!iEntrySelection) //Entry iMarkFlag is Flase
+            {
+            TMsvEmailEntry entry = iServerEntry.Entry();
+            TUint id = entry.UID();
+            iMessageUids.AppendL(id);
+            if(entry.Unread())
+                {
+                entry.SetSeenIMAP4Flag(EFalse);
+                }
+            else
+                {
+                entry.SetSeenIMAP4Flag(ETrue);	
+                }
+                    
+                    
+            User::LeaveIfError(iServerEntry.ChangeEntry(entry));
+    
+            uidSeq = CImapSession::CreateSequenceSetLC(iMessageUids);
+    
+            // finished with the UID array. Clear it.
+            iMessageUids.Reset();
+            if(entry.SeenIMAP4Flag())
+                {
+                // issue the store command
+                iSession->StoreL(iStatus, uidSeq->Des(), KImapAddFlags, KImapFlagSeen, ETrue, iFlagInfoResponseArray);
+                
+                }
+            else
+                {
+                // issue the store command
+                iSession->StoreL(iStatus, uidSeq->Des(), KStoreFlagsClearCommand, KImapFlagSeen, ETrue, iFlagInfoResponseArray);
+                }
+            }
+        else  //Selection iMarkFlag is Trure
+            {
+            TMsvEntry *aNewEntry1;
+			
+			TInt  count = iSourceSel->Count();
+			while (count--)
+				{
+				iServerEntry.GetEntryFromId(iSourceSel->At(count), aNewEntry1);
+				TMsvEmailEntry entry(*aNewEntry1);
+				iMessageUids.AppendL(entry.UID());
+					
+				if(iMarkFlag) // True  means  Unread
+					 {
+					 entry.SetSeenIMAP4Flag(EFalse);
+					 }
+				 else
+					 {
+					 entry.SetSeenIMAP4Flag(ETrue);  
+					 }
+				SetEntryL((*iSourceSel)[count]);
+				User::LeaveIfError(iServerEntry.ChangeEntry(entry));
+				}// end of while
+
+			uidSeq = CImapSession::CreateSequenceSetLC(iMessageUids);
+	
+			// finished with the UID array. Clear it.
+			iMessageUids.Reset();
+			if(!iMarkFlag) //read
+				{
+				// issue the store command
+				iSession->StoreL(iStatus, uidSeq->Des(), KImapAddFlags, KImapFlagSeen, ETrue, iFlagInfoResponseArray);
+				}
+			else   //Unread
+				{
+				// issue the store command
+				iSession->StoreL(iStatus, uidSeq->Des(), KStoreFlagsClearCommand, KImapFlagSeen, ETrue, iFlagInfoResponseArray);
+				}
+             } //End of else (Selection)
+	
+		iProgressState = TImap4GenericProgress::EBusy;
+		CleanupStack::PopAndDestroy(uidSeq);
+		iNextStep = EFinished;
+		break;
+		
+		}
+	
+	
+	case EFinished:
+		{
+		__LOG_TEXT(iSession->LogId(), "CImapUpdateFlagOperation::Completing OK");
+		iProgressState = TImap4GenericProgress::EIdle;
+		Complete(KErrNone);
+		return ETrue;
+		}
+		
+	default:
+		{
+		__ASSERT_DEBUG(EFalse, 
+					   TImapServerPanic::ImapPanic(TImapServerPanic::ECreateCompoundUnexpectedState));
+		// unexpected state - complete the request
+		iProgressState = TImap4GenericProgress::EIdle;
+		return ETrue;
+		}
+		}
+
+	if (!IsActive())
+		{
+		SetActive();
+		}
+	return EFalse;
+	}
+
+void CImapUpdateFlagOperation::DoCancel()
+	{
+	iProgressErrorCode = KErrCancel;
+	
+	switch (iCurrentStep)
+		{
+	case EUpdateFlag:
+		{
+		// outstanding request is on session
+		iSession->Cancel();
+		break;
+		}
+	case EFinished:
+		{
+		// self-completed or no outstanding request.
+		break;
+		}
+
+	default:
+		{
+		__ASSERT_DEBUG(EFalse,
+		               TImapServerPanic::ImapPanic(TImapServerPanic::ECreateCompoundCancelUnexpectedState));
+		break;
+		}
+		} // end switch (iCurrentStep)
+	
+	CMsgActive::DoCancel();
+	}
+
+
+
+
+void CImapUpdateFlagOperation::Progress(TImap4CompoundProgress& aCompoundProgress)
+	{
+	// Create does not set iOperation, it just sets iState to EBusy 
+	// when doing the creation
+	aCompoundProgress.iGenericProgress.iState = iProgressState;
+
+	
+	// Put error into progress buffer
+	if( aCompoundProgress.iGenericProgress.iErrorCode == KErrNone )
+		{
+		aCompoundProgress.iGenericProgress.iErrorCode = iProgressErrorCode;
+		}
+	}
+
+
+/**
+Handles NO/BAD responses according to current step.
+Negative server responses are not fatal - the error is reported in the 
+progress report, and the operation completes.
+
+@return KErrNone if the error has been handled
+		Completion error code otherwise.
+*/
+TInt CImapUpdateFlagOperation::ProcessNegativeServerResponse()
+	{
+	TInt err = iStatus.Int();
+	switch (iCurrentStep)
+		{
+	case EUpdateFlag:
+		iNextStep = EFinished;
+		break;
+	case EFinished:
+	default:
+		{
+		// positive error code not expected,
+		// self-completed states or no outstanding request.
+		TImapServerPanic::ImapPanic(TImapServerPanic::ECreateCompoundUnexpectedState);
+		break;
+		}
+		
+		} // end switch (iCurrentStep)
+	iProgressErrorCode = err;
+	return KErrNone;
+	}
+
+/**
+Resume the operation following a bearer migration.
+*/
+void CImapUpdateFlagOperation::ResumeOperationL(TRequestStatus& aStatus, CImapSession& aSession)
+	{
+	iSession = &aSession;
+	__LOG_TEXT(iSession->LogId(), "CImapUpdateFlagOperation::Resuming");
+	__ASSERT_DEBUG(iCurrentStep==ESuspendedForMigrate, TImapServerPanic::ImapPanic(TImapServerPanic::EUpdateCompoundUnexpectedState));
+	iStopForMigrate = EFalse;
+
+	switch (iNextStep)
+		{
+		case EUpdateFlag:
+			{
+			// just return to the main state machine
+			CompleteSelf();
+			break;
+			}
+		case EFinished:
+		default:
+			{
+			__ASSERT_DEBUG(EFalse, TImapServerPanic::ImapPanic(TImapServerPanic::EUpdateCompoundUnexpectedState));
+			// abandon the compound operation
+			iNextStep=EFinished;
+			CompleteSelf();
+			break;
+			}
+		}
+	Queue(aStatus);
+	CompleteSelf();
+	}
+