// SystemIcon.cpp: implementation of the CSystemIcon class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "SystemIcon.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
IMPLEMENT_DYNAMIC(CSystemIcon, CObject) 
CSystemIcon::CSystemIcon()
{
   	 memset(&m_tnd, 0, sizeof(m_tnd)); 

    m_bEnabled = FALSE; 

    m_bHidden  = FALSE;   

}
CSystemIcon::CSystemIcon(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szToolTip, 

                     HICON icon, UINT uID) 

{ 

 
    Create(pWnd, uCallbackMessage, szToolTip, icon, uID); 

    m_bHidden = FALSE; 

} 


CSystemIcon::~CSystemIcon()
{
DeleteIcon(); 
}
BOOL CSystemIcon::Create(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szToolTip, 

                       HICON icon, UINT uID) 

{ 

    //文件只能使用在WINDOW 95以上的版本中 

    VERIFY(m_bEnabled = ( GetVersion() & 0xff ) >= 4); 

    if (!m_bEnabled) return FALSE; 

    //確認通知窗口有效 

    VERIFY(m_bEnabled = (pWnd && ::IsWindow(pWnd->GetSafeHwnd()))); 

    if (!m_bEnabled) return FALSE; 

    //確認自定義消息大于WM_USER 

    ASSERT(uCallbackMessage >= WM_USER); 

    //確定提示文本長度小于64 

    ASSERT(_tcslen(szToolTip) <= 64); 

    //定義NOTIFYICONDATA結構的數據項 

    m_tnd.cbSize = sizeof(NOTIFYICONDATA); 

    m_tnd.hWnd     = pWnd->GetSafeHwnd(); 

    m_tnd.uID     = uID; 

    m_tnd.hIcon  = icon; 

    m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP; 

    m_tnd.uCallbackMessage = uCallbackMessage; 

    strcpy (m_tnd.szTip, szToolTip); 

    //設置圖標 

    VERIFY(m_bEnabled = Shell_NotifyIcon(NIM_ADD, &m_tnd)); 

    return m_bEnabled; 

} 

void CSystemIcon::DeleteIcon() 

{ 

    if (!m_bEnabled) return; 

  

    m_tnd.uFlags = 0; 

    Shell_NotifyIcon(NIM_DELETE, &m_tnd); 

    m_bEnabled = FALSE; 

} 

  

void CSystemIcon::HideIcon() 

{ 

    if (m_bEnabled && !m_bHidden) { 

        m_tnd.uFlags = NIF_ICON; 

        Shell_NotifyIcon (NIM_DELETE, &m_tnd); 

        m_bHidden = TRUE; 

    } 

} 

  

void CSystemIcon::ShowIcon() 

{ 

    if (m_bEnabled && m_bHidden) { 

        m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP; 

        Shell_NotifyIcon(NIM_ADD, &m_tnd); 

        m_bHidden = FALSE; 

    } 

} 

  

BOOL CSystemIcon::SetIcon(HICON hIcon) 

{ 

    if (!m_bEnabled) return FALSE; 

    m_tnd.uFlags = NIF_ICON; 

    m_tnd.hIcon = hIcon; 

    return Shell_NotifyIcon(NIM_MODIFY, &m_tnd); 

} 

  

BOOL CSystemIcon::SetIcon(LPCTSTR lpszIconName) 

{ 

    HICON hIcon = AfxGetApp()->LoadIcon(lpszIconName); 

    return SetIcon(hIcon); 

} 

  

BOOL CSystemIcon::SetIcon(UINT nIDResource) 

{ 

    HICON hIcon = AfxGetApp()->LoadIcon(nIDResource); 

    return SetIcon(hIcon); 

} 

  

BOOL CSystemIcon::SetStandardIcon(LPCTSTR lpIconName) 

{ 

    HICON hIcon = LoadIcon(NULL, lpIconName); 

    return SetIcon(hIcon); 

} 

  

BOOL CSystemIcon::SetStandardIcon(UINT nIDResource) 

{ 

    HICON hIcon = LoadIcon(NULL, MAKEINTRESOURCE(nIDResource)); 

    return SetIcon(hIcon); 

} 

  

HICON CSystemIcon::GetIcon() const 

{ 

    HICON hIcon = NULL; 

    if (m_bEnabled) 

        hIcon = m_tnd.hIcon; 

    return hIcon; 

} 

  

BOOL CSystemIcon::SetTooltipText(LPCTSTR pszTip) 

{ 

    if (!m_bEnabled) return FALSE; 

  

    m_tnd.uFlags = NIF_TIP; 

    _tcscpy(m_tnd.szTip, pszTip); 

    return Shell_NotifyIcon(NIM_MODIFY, &m_tnd); 

} 

  

BOOL CSystemIcon::SetTooltipText(UINT nID) 

{ 

    CString strText; 

    VERIFY(strText.LoadString(nID)); 

    return SetTooltipText(strText); 

} 

  

CString CSystemIcon::GetTooltipText() const 

{ 

    CString strText; 

    if (m_bEnabled) 

        strText = m_tnd.szTip; 

    return strText; 

} 

  

BOOL CSystemIcon::SetNotificationWnd(CWnd* pWnd) 

{ 

    if (!m_bEnabled) return FALSE; 

    ASSERT(pWnd && ::IsWindow(pWnd->GetSafeHwnd())); 

    m_tnd.hWnd = pWnd->GetSafeHwnd(); 

    m_tnd.uFlags = 0; 

    return Shell_NotifyIcon(NIM_MODIFY, &m_tnd); 

} 

  

CWnd* CSystemIcon::GetNotificationWnd() const 

{ 

    return CWnd::FromHandle(m_tnd.hWnd); 

} 

  

LRESULT CSystemIcon::OnIconNotification(WPARAM wParam, LPARAM lParam) 

{ 

    if (wParam != m_tnd.uID) 

        return 0L; 

    CMenu menu, *pSubMenu; 

    //單擊右鍵彈出菜單 

    if (LOWORD(lParam) == WM_RBUTTONUP) 

    {    

        if (!menu.LoadMenu(m_tnd.uID)) return 0; 

        if (!(pSubMenu = menu.GetSubMenu(0))) return 0; 

        //設定第一項為缺省 

        ::SetMenuDefaultItem(pSubMenu->m_hMenu, 0, TRUE); 

        //定義彈出菜單 

        CPoint pos; 

        GetCursorPos(&pos); 

        ::SetForegroundWindow(m_tnd.hWnd);  

        ::TrackPopupMenu(pSubMenu->m_hMenu, 0, pos.x, pos.y, 0, m_tnd.hWnd, NULL); 

        ::PostMessage(m_tnd.hWnd, WM_NULL, 0, 0); 

        menu.DestroyMenu(); 

    } 

    else if (LOWORD(lParam) == WM_LBUTTONDBLCLK) 

    { 

        if (!menu.LoadMenu(m_tnd.uID)) return 0; 

        if (!(pSubMenu = menu.GetSubMenu(0))) return 0; 

        //雙擊左鍵起動缺省菜單 

        ::SetForegroundWindow(m_tnd.hWnd); 

        ::SendMessage(m_tnd.hWnd, WM_COMMAND, pSubMenu->GetMenuItemID(0), 0); 

        menu.DestroyMenu(); 

    } 

  

    return 1; 

}