diff -r ebe688cedc25 -r 7fdbb852d323 email/mail/EditorSrc/cmsgmailrestoreattasop.h
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/email/mail/EditorSrc/cmsgmailrestoreattasop.h	Wed Sep 01 12:31:54 2010 +0100
@@ -0,0 +1,148 @@
+/*
+* Copyright (c) 2002-2007 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:  
+*           State machine -based operation for restoring message's attachments.
+*
+*/
+
+
+
+#ifndef CMSGMAILRESTOREATTASOP_H
+#define CMSGMAILRESTOREATTASOP_H
+
+//  INCLUDES
+#include "cmsgmailbaseop.h"
+#include <msvapi.h>
+
+// DATA TYPES
+
+// FORWARD DECLARATIONS
+class MMsvAttachmentManager;
+class CMsgMailRestoreAttaOp;
+
+// CLASS DECLARATION
+/**
+* State machine -based operation for restoring message's attachments.
+* This operation involves restoring each individual attachments, and
+* furthermore initialization and finalization phase.
+*/
+NONSHARABLE_CLASS( CMsgMailRestoreAttasOp ) : public CMsgMailBaseOp
+    {
+    protected:
+    
+        enum TLaunchState
+            {
+            EInitialize = EFirstFreeState, // initialize operation
+            ERestoreAtta, // restore single atta
+            EFinalize // finalize entire operation
+            };
+    
+    public:  // Constructors and destructor
+    
+        /**
+        * Two-phased constructor.
+        * @param aDocument Editor's document
+        */
+        static CMsgMailRestoreAttasOp* NewL(
+            CMsgMailEditorDocument& aDocument );
+
+        /**
+        * Destructor. Cancels operation.
+        */
+        ~CMsgMailRestoreAttasOp();
+        
+    public: // API
+            
+        /**
+        * Method for querying the number of all attachments involved in
+        * this operation, including any discarded attachments.
+        * To get correct result this method should be called once the
+        * operation has completed.        
+        * @return count of all attachments, i.e. 
+        */
+        TInt CountAllAttas() const;          
+    
+        /**
+        * Method for querying the number of attachments that were
+        * removed due to DRM restrictions.
+        * To get correct result this method should be called once the
+        * operation has completed.
+        * @return count of removed DRM restricted attachments
+        */        
+        TInt CountDRMAttas() const;
+        
+        /**
+        * Method for querying the number of invalid attachments. This
+        * number does not include DRM attachments.
+        * To get correct result this method should be called once the
+        * operation has completed.
+        * @return count of removed invalid attachments
+        */        
+        TInt CountInvalidAttas() const;   
+
+    protected: // from CMsgMailBaseOp
+    
+        void SetNextState();
+        
+        void HandleStateActionL();
+        
+        void HandleOperationCancel();
+        
+        TBool HandleStateActionError( TInt aError );
+        
+    private:  // Implementation
+        
+        void InitializeL();
+        
+        void RestoreAttaL();
+        
+        void Finalize();
+
+    private:
+    
+        /**
+        * Constructor.
+        * @param aDocument Editor's document        
+        */
+        CMsgMailRestoreAttasOp( CMsgMailEditorDocument& aDocument );
+
+        /**
+        * By default Symbian 2nd phase constructor is private.
+        */
+        void ConstructL();
+        
+    private:
+    
+        // Count of all attachments, including discarded attachments
+        TInt iAllAttaCount;
+    
+        // Count of DRM restricted files
+        TInt iDRMAttaCount;    
+        
+        // Count of invalid attachments (excluding DRM problems)
+        TInt iInvalidAttaCount;
+    
+        // Index for iterating attachment manager's attachments
+        TInt iIndex;
+    
+        // Attachment manager, not own
+        MMsvAttachmentManager* iManager;    
+        
+        // Currently restored attachment, own
+        CMsgMailRestoreAttaOp* iRestoreOneAtta;
+    };
+
+#endif      // CMSGMAILRESTOREATTASOP_H
+
+// End of File