TnbTasktrayIcon.h

[詳解]

#pragma once
#include "TnbSync.h"
#include "TnbStr.h"
 
 
 
#include <shellapi.h>
#ifndef _WIN32_WCE
 #pragma comment(lib, "shell32.lib")
#endif
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CTasktrayIcon : CCopyImpossible
{
public:
 
    struct IListener
    {
        virtual ~IListener(void) {}
 
        virtual void OnNotifyTasktrayIcon(UINT uMsg, const POINT& tPoint) = 0;
 
    private:
        friend class CTasktrayIcon; 
    };
 
    explicit CTasktrayIcon(UINT uID = 0)
    {
        m_icondata.uID              = uID;
        m_icondata.uCallbackMessage = ::RegisterWindowMessage(CStr::Fmt(_T("__SSA_%u"), uID));
        m_icondata.ResetFlag();
        m_boShow                    = false;
        m_piListener                = NULL;
        m_uTaskbarCreatedMessage    = ::RegisterWindowMessage(_T("TaskbarCreated"));
        m_idMenu                    = 0;
        m_hMenu                     = NULL;
    }
 
    virtual ~CTasktrayIcon(void)
    {
        Hide();
    }
 
    void SetListener(IListener* piListener)
    {
        m_piListener = piListener;
    }
    
    bool RelayEvent(UINT message, WPARAM wParam, LPARAM lParam)
    {
        if ( message == m_uTaskbarCreatedMessage )
        {
            //Explorerの再起動など。ICONの再設定が必要
            if ( m_boShow )
            {
                m_Modify(m_icondata);
            }
            return false;
        }
        if ( message != m_icondata.uCallbackMessage )
        {
            return false;
        }
        //TRACE1("RelayEvent = 0x%08X\n", lParam);
        switch ( lParam )
        {
        case WM_LBUTTONDOWN:
        case WM_LBUTTONUP:
        case WM_LBUTTONDBLCLK:
        case WM_RBUTTONDOWN:
        case WM_RBUTTONUP:
        case WM_RBUTTONDBLCLK:
            break;
        default:
            return false;
        }
        if ( ! m_syncClick.Lock(0) )
        {
            return false;
        }
        //
        bool rc = false;
        POINT tPoint = { 0x7FFF, 0x7FFF };
        ::GetCursorPos(&tPoint);
        ::SetForegroundWindow(m_icondata.hWnd);
        ::SetFocus(m_icondata.hWnd);
        if ( m_piListener != NULL )
        {
            m_piListener->OnNotifyTasktrayIcon(static_cast<UINT>(lParam), tPoint);
        }
        #ifndef _WIN32_WCE
            if ( lParam == WM_RBUTTONDOWN )
            {
                rc = true;
            }
            if ( lParam == WM_RBUTTONUP )
            {
                rc = !! ShowMenu(tPoint, TPM_LEFTALIGN | TPM_RIGHTBUTTON);
            }
        #else
            if ( lParam == WM_LBUTTONUP )
            {
                rc = !! ShowMenu(tPoint, TPM_LEFTALIGN);
            }
        #endif
        m_syncClick.Unlock();
        return rc;
    }
 
    bool Create(HWND hWnd, HICON hIcon, LPCTSTR lpTip = NULL)
    {
        m_icondata.hIcon = hIcon;
        m_icondata.hWnd = hWnd;
        m_icondata.SetTip(lpTip);
        m_icondata.ResetFlag();
        m_Modify(m_icondata);
        m_boShow = true;
        return true;
    }
 
    void Hide(void)
    {
        if ( m_boShow )
        {
            ::Shell_NotifyIcon(NIM_DELETE, m_icondata);
            m_boShow = false;
        }
    }
 
    void SetMenu(int id = 0)
    {
        m_idMenu = id;
        m_hMenu = NULL; 
    }
 
    void SetMenu(HMENU h = NULL)
    {
        m_idMenu = 0;
        m_hMenu = h;
    }
 
    void ModifyIcon(HICON hIcon)
    {
        if ( m_boShow )
        {
            m_icondata.hIcon = hIcon;
            m_icondata.ResetFlag();
            m_Modify(m_icondata);
        }
    }
 
    void ModifyTip(LPCTSTR lpTip)
    {
        if ( m_boShow )
        {
            m_icondata.SetTip(lpTip);
            m_icondata.ResetFlag();
            m_Modify(m_icondata);
        }
    }
 
#if _WIN32_IE >= 0x0500
    enum EInfoIcon
    {
        EII_None        = 0x00000000,   
        EII_Info        = 0x00000001,   
        EII_Warning     = 0x00000002,   
        EII_Error       = 0x00000003,   
        EII_NsNone      = 0x00000010,   
        EII_NsInfo      = 0x00000011,   
        EII_NsWarning   = 0x00000012,   
        EII_NsError     = 0x00000013,   
    };
 
    void ShowInformation(EInfoIcon eIcon, LPCTSTR lpszTitle, LPCTSTR lpszMessage, DWORD dwTimeout = 0)
    {
        if ( m_boShow )
        {
            CStr s;
            s = lpszTitle;
            STRLIB::Copy(m_icondata.szInfoTitle, s.Left(sizeof(m_icondata.szInfoTitle) - 1));
            s = lpszMessage;
            STRLIB::Copy(m_icondata.szInfo, s.Left(sizeof(m_icondata.szInfo) - 1));
            m_icondata.dwInfoFlags = eIcon;
            m_icondata.uTimeout = dwTimeout;
            //
            m_icondata.uFlags |= 0x00000010/*NIF_INFO*/;
            m_Modify(m_icondata);
            m_icondata.uFlags &= ~0x00000010/*NIF_INFO*/;
        }
    }
 
    void HideInformation(void)
    {
        if ( m_boShow )
        {
            m_icondata.szInfoTitle[0] = 0;
            m_icondata.szInfo[0] = 0;
            m_icondata.dwInfoFlags = EII_None;
            m_icondata.uTimeout = 0;
            //
            m_icondata.uFlags |= 0x00000010/*NIF_INFO*/;
            m_Modify(m_icondata);
            m_icondata.uFlags &= ~0x00000010/*NIF_INFO*/;
        }
    }
#endif
 
    #ifndef _WIN32_WCE
    DWORD ShowMenu(const POINT& po, UINT flag = TPM_LEFTALIGN | TPM_RIGHTBUTTON)
    #else
    DWORD ShowMenu(const POINT& po, UINT flag = TPM_LEFTALIGN)
    #endif
    {
        if ( m_idMenu != 0 )
        {
            HINSTANCE hIns = TNB::GetInstanceHandleByTnb();
            HMENU hMenu = ::LoadMenu(hIns, MAKEINTRESOURCE(m_idMenu));
            if ( hMenu != NULL )
            {
                HMENU h = ::GetSubMenu(hMenu, 0);
                DWORD r = ::TrackPopupMenuEx(h, flag, po.x, po.y, m_icondata.hWnd, NULL);
                ::DestroyMenu(h);
                ::DestroyMenu(hMenu);
                return r;
            }
        }
        else if ( m_hMenu != NULL )
        {
            return ::TrackPopupMenuEx(m_hMenu, flag, po.x, po.y, m_icondata.hWnd, NULL);
        }
        return 0;
    }
 
private:
 
    class CIconData
    {
    public:
        DWORD   cbSize;
        HWND    hWnd;
        UINT    uID;
        UINT    uFlags;
        UINT    uCallbackMessage;
        HICON   hIcon;
        #if _WIN32_IE < 0x0500
            TCHAR   szTip[64];
        #else
            TCHAR   szTip[128];
            DWORD   dwState;
            DWORD   dwStateMask;
            TCHAR   szInfo[256];
            union {
                UINT    uTimeout;
                UINT    uVersion;
            } DUMMYUNIONNAME;
            TCHAR   szInfoTitle[64];
            DWORD   dwInfoFlags;
            #if _WIN32_IE >= 0x0600
                GUID guidItem;
            #endif
        #endif
        // コンストラクタ
        CIconData(void)
        {
            Zero(*this);
            cbSize = sizeof(CIconData);
        }
        operator NOTIFYICONDATA*(void)
        {
            return reinterpret_cast<NOTIFYICONDATA*>(this);
        }
        void SetTip(LPCTSTR lpTip)
        {
            CStr s = lpTip;
            STRLIB::Copy(szTip, s.Left(sizeof(szTip) - 1));
        }
        void ResetFlag(void)
        {
            uFlags = NIF_MESSAGE;
            if ( szTip[0] != 0 )
            {
                uFlags |= NIF_TIP;
            }
            if ( hIcon != NULL )
            {
                uFlags |= NIF_ICON;
            }
        }
    };
 
    void m_Modify(NOTIFYICONDATA* P)
    {
        if ( m_boShow && ::Shell_NotifyIcon(NIM_MODIFY, P) )
        {
            return;
        }
        while ( true )
        {
            if ( ::Shell_NotifyIcon(NIM_ADD, P) )
            {
                break;/*成功*/ 
            }
            if ( _GetLastError("Shell_NotifyIcon") != ERROR_TIMEOUT )
            {
                break;/*失敗*/ 
            }
            ::Sleep(500);
            if ( ::Shell_NotifyIcon(NIM_MODIFY, P) )
            {
                break;/*成功*/ 
            }
        }
    }
    //
    CIconData       m_icondata;                 
    int             m_idMenu;                   
    HMENU           m_hMenu;                    
    bool            m_boShow;                   
    IListener*      m_piListener;               
    CSyncCounter    m_syncClick;                
    UINT            m_uTaskbarCreatedMessage;   
};
 
 
 
}; // TNB