#include <memspy/engine/memspydevicewideoperations.h>

class CMemSpyDwOperationTracker;

NONSHARABLE_CLASS( CShutdown ) : public CTimer

    {

    enum {KMyShutdownDelay=10 * 1000000};       // 10s

public:

    inline CShutdown();

    inline void ConstructL();

    inline void Start();

private:

    void RunL();

    };

    CMemSpyDwOperationTracker* CurrentOperationTracker() const { return iCurrentOperationTracker; }

    void SetCurrentOperationTracker(CMemSpyDwOperationTracker* aTracker) { iCurrentOperationTracker = aTracker; }

    CMemSpyEngine& Engine() { return iEngine; } 

    void AddSession(TBool aCliRequest);

    void DropSession(TBool aCliRequest);

    CMemSpyDwOperationTracker* iCurrentOperationTracker;

    TInt iSessionCount;

    TBool iCliConnected;

    CShutdown iShutdown;

	void CreateL();

    void DoAsyncServiceL( const RMessage2& aMessage );

    void DoUiServiceL( const RMessage2& aMessage );

    void DoCmdServiceL( const RMessage2& aMessage );

    void StartDeviceWideOperationL(CMemSpyDeviceWideOperations::TOperation aOperation, const RMessage2& aMessage);

    inline CMemSpyEngineServer& Server() const { return *static_cast<CMemSpyEngineServer*>(const_cast<CServer2*>(CSession2::Server())); }

    TBool iIsCliRequest;

/**

 * CMemSpyDwOperationTracker

 * Tracks device wide operation progress and calls iOperationMessage.Complete upon completion. 

 */

NONSHARABLE_CLASS( CMemSpyDwOperationTracker ) : public MMemSpyDeviceWideOperationsObserver

	{

public:

	static CMemSpyDwOperationTracker* NewL(CMemSpyDeviceWideOperations::TOperation aOperation, 

			const RMessage2& aOperationMessage,

			CMemSpyEngineServer& aServer);

	~CMemSpyDwOperationTracker();

	void AddNotificationL(const RMessage2& aMessage);

	void Cancel();

public: // From MMemSpyDeviceWideOperationsObserver

	void HandleDeviceWideOperationEvent(TEvent aEvent, TInt aParam1, const TDesC& aParam2);

private:

	CMemSpyDwOperationTracker(const RMessage2& aOperationMessage, CMemSpyEngineServer& aServer);

	void ConstructL(CMemSpyDeviceWideOperations::TOperation aOperation);

private:

	RMessage2 iOperationMessage;

	CMemSpyEngineServer& iServer;

	CArrayFixFlat<RMessage2>* iPendingNotifications;

	CMemSpyDeviceWideOperations* iOperation;

	TInt iProgress;

	};