#ifndef _TRAYREG_H
#define _TRAYREG_H

#include "dpa.h"

#define SZ_TRAYNOTIFY_REGKEY        TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\TrayNotify")
#define SZ_EXPLORER_REGKEY          TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer")
#define SZ_AUTOTRAY_REGVALUE        TEXT("EnableAutoTray")
#define SZ_ITEMSTREAMS_REGVALUE     TEXT("IconStreams")
#define SZ_ICONSTREAMS_REGVALUE     TEXT("PastIconsStream")
#define SZ_INFOTIP_REGVALUE         TEXT("BalloonTip")

#define SZ_ICON_COUNTDOWN_VALUE             TEXT("TrayIconResidentInterval")
#define SZ_ICONCLEANUP_VALUE                TEXT("Icon Cleanup Time")

#define SZ_ICONDEMOTE_TIMER_TICK_VALUE      TEXT("IconDemoteTimerTick")
#define SZ_INFOTIP_TIMER_TICK_VALUE         TEXT("InfoTipTimerTick")

// Parameters to control the aging algorithm
#ifdef FULL_DEBUG
#define TT_ICON_COUNTDOWN_INTERVAL                   15000
#else
#define TT_ICON_COUNTDOWN_INTERVAL                 3600000      // 1 hour
#endif
#define TT_ICONCLEANUP_INTERVAL                          6      // 6 months


#define TT_ICON_COUNTDOWN_INCREMENT           8*60*60*1000      // 8 hours

// Show the Chevron info tip 5 times, once per session for the first 5 sessions, or
// until the user clicks on the balloon, whichever comes first.
#define MAX_CHEVRON_INFOTIP_SHOWN                        5


// Flags decide which user tracking timer to use - the balloon tips or the icons
#define TF_ICONDEMOTE_TIMER                              1
#define TF_INFOTIP_TIMER                                 2

// What is the timer interval for the UserEventTimer for each of the timer cases ?
// Once every 5 minutes, for the tray items...
#define UET_ICONDEMOTE_TIMER_TICK_INTERVAL                300000
// Once every 5 seconds, for balloon tips...
#define UET_INFOTIP_TIMER_TICK_INTERVAL                     5000


//
// header for persistent streams storing the tray notify info
//
typedef struct tagTNPersistStreamHeader
{
    DWORD   dwSize;         // Size of the header structure..
    DWORD   dwVersion;      // Version number for the header
    DWORD   dwSignature;    // Signature to identify this version of the header
    DWORD   cIcons;         // Number of Icons that have been stored in the stream
    DWORD   dwOffset;       // Offset to the start of the first icondata in the stream
} TNPersistStreamHeader;

#define INVALID_IMAGE_INDEX         -1

typedef struct tagTNPersistStreamData
{
    TCHAR       szExeName[MAX_PATH]; 
    UINT        uID;
    BOOL        bDemoted;
    DWORD       dwUserPref;
    WORD        wYear;
    WORD        wMonth;
    TCHAR       szIconText[MAX_PATH];
    UINT        uNumSeconds;
    BOOL        bStartupIcon;
    INT         nImageIndex;        // Index of the image in the Past Items image list.
    GUID        guidItem;
} TNPersistStreamData;

typedef struct tagTNPersistentIconStreamHeader
{
    DWORD   dwSize;         // Size of the header structure
    DWORD   dwVersion;      // Version number of the header
    DWORD   dwSignature;    // This signature must be the same as TNPersistStreamHeader.dwSignature
    DWORD   cIcons;         // Number of icons stored in the stream
    DWORD   dwOffset;       // Offset into the first item in the stream
} TNPersistentIconStreamHeader;

#define TNH_VERSION_ONE     1
#define TNH_VERSION_TWO     2
#define TNH_VERSION_THREE   3
#define TNH_VERSION_FOUR    4
#define TNH_VERSION_FIVE    5

#define TNH_SIGNATURE       0x00010001

class CNotificationItem;
class CTrayItem;

// Any client asking the TrayNotify class for an item passes these flags as one of the 
// parameters to the callback functions...
typedef enum TRAYCBARG
{
    TRAYCBARG_PTI,
    TRAYCBARG_HICON,
    TRAYCBARG_ALL
} TRAYCBARG;

typedef struct TRAYCBRET
{
    CTrayItem * pti;
    HICON hIcon;
} TRAYCBRET;


typedef BOOL (CALLBACK * PFNTRAYNOTIFYCALLBACK)(INT_PTR nIndex, void *pCallbackData, 
        TRAYCBARG trayCallbackArg, TRAYCBRET  *pOutData);


class CTrayItemRegistry
{
    public:
        CTrayItemRegistry() : _himlPastItemsIconList(NULL) { }
        ~CTrayItemRegistry() { }

        void InitRegistryValues(UINT uIconListFlags);
        void InitTrayItemStream(DWORD dwStreamMode, PFNTRAYNOTIFYCALLBACK pfnTrayNotifyCB, void *pCBData);

        BOOL GetTrayItem(INT_PTR nIndex, CNotificationItem * pni, BOOL * pbStat);
        BOOL AddToPastItems(CTrayItem * pti, HICON hIcon);

        void IncChevronInfoTipShownInRegistry(BOOL bUserClickedInfoTip = FALSE);
        BOOL SetIsAutoTrayEnabledInRegistry(BOOL bAutoTray);
        BOOL SetPastItemPreference(LPNOTIFYITEM pNotifyItem);

        INT_PTR CheckAndRestorePersistentIconSettings(CTrayItem *pti, LPTSTR pszIconToolTip, HICON hIcon);
        void DeletePastItem(INT_PTR nIndex);
        int DoesIconExistFromPreviousSession(CTrayItem * pti, LPTSTR pszIconToolTip, HICON hIcon);

        void Delete()
        {
            _dpaPersistentItemInfo.DestroyCallback(_DestroyIconInfoCB, NULL);

            _DestroyPastItemsIconList();
        }

        int _AddPastIcon(int nImageIndex, HICON hIcon)
        {
            if (_himlPastItemsIconList && hIcon)
                return ImageList_ReplaceIcon(_himlPastItemsIconList, nImageIndex, hIcon);

            return INVALID_IMAGE_INDEX;
        }
        
        UINT GetTimerTickInterval(int nTimerFlag);

    public:
        BOOL ShouldChevronInfoTipBeShown()
        {
            return _bShowChevronInfoTip;
        }

        // Has the Whistler "Auto" tray been disabled by policy ?
        BOOL IsNoAutoTrayPolicyEnabled() const
        {
            return _fNoAutoTrayPolicyEnabled;
        }

        // If not, has the user turned off Whistler "auto" tray policy ?
        BOOL IsAutoTrayEnabledByUser() const
        {
            return _fAutoTrayEnabledByUser;
        }

        BOOL IsAutoTrayEnabled()
        {
            return (!_fNoAutoTrayPolicyEnabled && _fAutoTrayEnabledByUser);
        }

    public:
        ULONG           _uPrimaryCountdown;
        
    private:
        static int _DestroyIconInfoCB(TNPersistStreamData * pData, LPVOID pData2);

        HRESULT _LoadTrayItemStream(IStream *pstm, PFNTRAYNOTIFYCALLBACK pfnTrayNotifyCB, void *pCBData);
        HRESULT _SaveTrayItemStream(IStream *pstm, PFNTRAYNOTIFYCALLBACK pfnTrayNotifyCB, void *pCBData);
        BOOL _FillPersistData(TNPersistStreamData * ptnPersistData, CTrayItem * pti, HICON hIcon);

        BOOL _IsAutoTrayEnabledInRegistry();

        void _QueryRegValue(HKEY hkey, LPTSTR pszValue, ULONG* puVal, ULONG uDefault);

        void _RestorePersistentIconSettings(TNPersistStreamData * ptnpd, CTrayItem * pti);

        BOOL _IsValidStreamHeaderVersion(DWORD dwVersion)
        {
            return ( (dwVersion == TNH_VERSION_FOUR) || (dwVersion == TNH_VERSION_FIVE) );
        }

        UINT_PTR _SizeOfPersistStreamData(DWORD dwVersion);
        
        inline void _DestroyPastItemsIconList()
        {
            if (_himlPastItemsIconList)
            {
                ImageList_Destroy(_himlPastItemsIconList);
                _himlPastItemsIconList = NULL;
            }
        }

        BOOL _IsIconLastUseValid(WORD wYear, WORD wMonth);

        BOOL _SaveIconsToRegStream();
        BOOL _LoadIconsFromRegStream(DWORD dwItemStreamSignature);
        void UpdateImageIndices(INT_PTR nDeletedImageIndex);

        //
        // Persistent Icon information...
        //
        CDPA<TNPersistStreamData> _dpaPersistentItemInfo;
        DWORD           _dwTimesChevronInfoTipShown;
        BOOL            _bShowChevronInfoTip;
        ULONG           _uValidLastUseTimePeriod;

        // We store this policy in a local cache, since we do not support settings
        // changes during the session. At logon, the settings are renewed, and if the
        // settings change during the session, they dont take effect, until the user
        // has logged off and logged back on to a different session.

        // This policy states that the tray should function like the Windows 2000 tray,
        // and disables all "smart" functionality like aging, and advanced balloon 
        // tips.
        BOOL            _fNoAutoTrayPolicyEnabled;

        // This variable dictates the current setting of the tray, since the user is
        // allowed to specify if he wants the Windows 2000 tray, or the Whistler 
        // auto-tray. 
        BOOL            _fAutoTrayEnabledByUser;

        HIMAGELIST      _himlPastItemsIconList;

        ULONG           _uIconDemoteTimerTickInterval;
        ULONG           _uInfoTipTimerTickInterval;
};

#endif // _TRAYREG_H