/******************************************************************************

Copyright (c) 1999-2000 Microsoft Corporation

Module Name:
    rstrmgr.h

Abstract:
    This file contains the declaration of the CRestoreManager class, which
    controls overall restoration process and provides methods to control &
    help user experience flow.

Revision History:
    Seong Kook Khang (SKKhang)  05/10/00
        created

******************************************************************************/

#ifndef _RSTRMGR_H__INCLUDED_
#define _RSTRMGR_H__INCLUDED_

#pragma once


/////////////////////////////////////////////////////////////////////////////
//
// Constant Definitions
//
/////////////////////////////////////////////////////////////////////////////

// SR Restore Start Mode
enum
{
    SRRSM_NORMAL = 0,
    SRRSM_SUCCESS,
    SRRSM_FAIL,
    SRRSM_FAILLOWDISK
};

// SR Restore Manager Status
enum
{
    SRRMS_NONE = 0,
    SRRMS_STARTED,
    SRRMS_INITIALIZING,
    SRRMS_CREATING_MAP,
    SRRMS_RESTORING,
    SRRMS_FINISHED
};

// Functionality Chosen in the Main Page
enum
{
    RMO_RESTORE = 0,
    RMO_CREATERP,
    RMO_UNDO,
    RMO_MAX
};

// SR Restore Manager Flags
#define SRRMF_CANNAVIGATEPAGE       0x00000001
#define SRRMF_ISUNDO                0x00000002
#define SRRMF_ISRPSELECTED          0x00000004


/////////////////////////////////////////////////////////////////////////////
//
// CSRTime Definitions
//
/////////////////////////////////////////////////////////////////////////////

class CSRTime
{
public:
    CSRTime();

public:
    const CSRTime& operator=( const CSRTime &cSrc );

public:
    int  GetYear()   {  return( m_st.wYear );  }
    int  GetMonth()  {  return( m_st.wMonth );  }
    int  GetDay()    {  return( m_st.wDay );  }
    operator PSYSTEMTIME() const;
    //operator (FILETIME*)();
    PSYSTEMTIME  GetTime();
    void         GetTime( PSYSTEMTIME pst );
    BOOL         GetTime( PFILETIME pft );

public:
    int   Compare( CSRTime &time );
    int   CompareDate( CSRTime &time );
    BOOL  SetTime( PFILETIME ft, BOOL fLocal=TRUE );
    void  SetTime( PSYSTEMTIME st );
    void  SetToCurrent();

protected:
    SYSTEMTIME  m_st;   // Date/Time in UTC
};


/////////////////////////////////////////////////////////////////////////////
//
// Structure Definitions
//
/////////////////////////////////////////////////////////////////////////////

struct SRestorePointInfo
{
    DWORD      dwType;
    DWORD      dwNum;
    CSRStr     strDir;
    CSRStr     strName;
    CSRTime    stTimeStamp;
    DWORD      dwFlags;
};

typedef SRestorePointInfo  *PSRPI;
typedef CSRDynPtrArray<PSRPI, 32>  CDPA_RPI;

struct SRenamedFolderInfo
{
    CSRStr  strOld;
    CSRStr  strNew;
    CSRStr  strLoc;
};

typedef SRenamedFolderInfo  *PSRFI;
typedef CSRDynPtrArray<PSRFI, 16>  CDPA_RFI;


/////////////////////////////////////////////////////////////////////////////
//
// CRestoreManager
//
/////////////////////////////////////////////////////////////////////////////

class CRestoreManager
{
public:
    CRestoreManager();
    ~CRestoreManager();
    void  Release();

// Properties - main flow
public:
    BOOL     CanRunRestore( BOOL fThawIfFrozen );
    int      GetFirstDayOfWeek();
    BOOL     GetIsRPSelected();
    BOOL     GetIsSafeMode();
    BOOL     GetIsSmgrAvailable();
    BOOL     GetIsUndo();
    int      GetLastRestore();
    int      GetMainOption();
    LPCWSTR  GetManualRPName();
    void     GetMaxDate( PSYSTEMTIME pstMax );
    void     GetMinDate( PSYSTEMTIME pstMin );
    int      GetRealPoint();
    PSRFI    GetRFI( int nIndex );
    int      GetRFICount();
    PSRPI    GetRPI( int nIndex );
    int      GetRPICount();
    void     GetSelectedDate( PSYSTEMTIME pstSel );
    LPCWSTR  GetSelectedName();
    int      GetSelectedPoint();
    int      GetStartMode();
    void     GetToday( PSYSTEMTIME pstToday );
    void     SetIsRPSelected( BOOL fSel );
    void     SetIsUndo( BOOL fUndo );
    BOOL     SetMainOption( int nOpt );
    void     SetManualRPName( LPCWSTR cszRPName );
    void     SetSelectedDate( PSYSTEMTIME pstSel );
    BOOL     SetSelectedPoint( int nRP );
    BOOL     SetStartMode( int nMode );
    void     GetUsedDate( PSYSTEMTIME pstDate );
    LPCWSTR  GetUsedName();
    DWORD    GetUsedType();    

// Properties - HTML UI specific
public:
    BOOL  GetCanNavigatePage();
    void  SetCanNavigatePage( BOOL fCanNav );

// Properties
public:
    PSRPI   GetUsedRP();
    int     GetNewRP();

// Operations - main flow
public:
    BOOL  CheckRestore( BOOL fSilent );
    BOOL  BeginRestore();
    BOOL  Cancel();
    BOOL  CancelRestorePoint();
    BOOL  CreateRestorePoint();
    BOOL  DisableFIFO();
    BOOL  EnableFIFO();
    BOOL  FormatDate( PSYSTEMTIME pst, CSRStr &str, BOOL fLongFmt );
    BOOL  FormatLowDiskMsg( LPCWSTR cszFmt, CSRStr &str );
    BOOL  FormatTime( PSYSTEMTIME pst, CSRStr &str );
    BOOL  GetLocaleDateFormat( PSYSTEMTIME pst, LPCWSTR cszFmt, CSRStr &str );
    BOOL  GetYearMonthStr( int nYear, int nMonth, CSRStr &str );
    BOOL  InitializeAll();
    BOOL  Restore( HWND hwndProgress );

// Operations
public:
    BOOL  AddRenamedFolder( PSRFI pRFI );
    BOOL  SetRPsUsed( int nRPUsed, int nRPNew );
    BOOL  SilentRestore( DWORD dwRP );

// Operations - internal
protected:
    void  Cleanup();
    BOOL  GetDateStr( PSYSTEMTIME pst, CSRStr &str, DWORD dwFlags, LPCWSTR cszFmt );
    BOOL  GetTimeStr( PSYSTEMTIME pst, CSRStr &str, DWORD dwFlags );
    void  UpdateRestorePoint();
    BOOL  UpdateRestorePointList();
    BOOL  CheckForDomainChange (WCHAR *pwszFilename, WCHAR *pszMsg);

// Attributes
public:
    HWND  GetFrameHwnd();
    //int   GetStatus();
    BOOL  DenyClose();
    BOOL  NeedReboot();
    void  SetFrameHwnd( HWND hWnd );
    void  SetIdealSize( int cx, int cy );

protected:
    int      m_nStartMode;
    BOOL     m_fNeedReboot;
    HWND     m_hwndFrame;

    CSRTime  m_stToday;         // Current local date/time
    int      m_nMainOption;     // Option on main screen
    //int      m_nStatus;
    BOOL     m_fDenyClose;
    DWORD    m_dwFlags;
    DWORD    m_dwFlagsEx;
    int      m_nSelectedRP;
    CSRTime  m_stSelected;
    CSRStr   m_strSelected;
    int      m_nRealPoint;
    INT64    m_ullManualRP;
    CSRStr   m_strManualRP;

    int      m_nRPUsed;     // RP used for the last restore
    int      m_nRPNew;      // New "Restore" RP created by the last restore

    // Restore Point specific informations
    CDPA_RPI  m_aryRPI;
    //int       m_nRPI;
    //PSRPI    *m_aryRPI;
    int       m_nLastRestore;
    CSRTime   m_stRPMin;
    CSRTime   m_stRPMax;

    CDPA_RFI  m_aryRFI;
    //int       m_nRFI;
    //PSRFI  *m_aryRFI;
    IRestoreContext  *m_pCtx;    
};

extern CRestoreManager  *g_pRstrMgr;

BOOL  CreateRestoreManagerInstance( CRestoreManager **ppMgr );


#endif //_RSTRMGR_H__INCLUDED_