系統托盤氣泡提示和右鍵菜單的實現

2019-11-18 11:13:33

字體：大中小

來源：轉載

供稿：網友

WTL嵌入類的架構可以使整個程序很清楚，系統托盤氣泡提示、上下文菜單處理可以分別獨立出來，以下是實現代碼，很清楚就不注釋了。基本上這兩個類很少需要修改，因此我把它們放到了系統包含文件搜索路徑中。

//ContextMenu.h
#PRagma once
template 
class CContextMenu
{
public:
BOOL CreateContextMenu(UINT ID_Menu)
{
T* pT = static_cast(this);
CMenu menu;
menu.LoadMenu(ID_Menu);
CMenu SubMenu(menu.GetSubMenu(0));
POINT pos;
GetCursorPos(&pos);
SubMenu.TrackPopupMenu(TPM_LEFTALIGN  TPM_RIGHTBUTTON, pos.x, pos.y, pT->m_hWnd);
return TRUE;
}
};

//ShellIcon.h
#pragma once
#define WM_ICON WM_USER + 180
template 
class CShellIcon : public MenuT
{
private:
NOTIFYICONDATA m_data;
UINT m_msgTaskbarRestart;
CString m_appName;
public:
CShellIcon()
{
m_appName.LoadString(IDS_APPNAME);
m_msgTaskbarRestart = RegisterWindowMessage(TEXT("TaskbarCreated"));
}
~CShellIcon()
{
Shell_NotifyIcon(NIM_DELETE, &m_data);
}
BOOL CreateShellIcon()
{
T* pT = static_cast(this);
SecureZeroMemory(&m_data, sizeof(m_data));
m_data.cbSize = sizeof(m_data);
m_data.hIcon = LoadIcon(_Module.get_m_hInst(), MAKEINTRESOURCE(IDR_MAINFRAME));
m_data.hWnd = pT->m_hWnd;
m_data.uID = IDR_MAINFRAME;
m_data.uFlags = NIF_ICON  NIF_MESSAGE  NIF_INFO  NIF_Tip;
m_data.UCallbackMessage = WM_ICON;
m_data.dwInfoFlags = NIIF_USER;
strcpy_s(m_data.szInfoTitle, m_appName);
strcpy_s(m_data.szTip, m_appName);
return Shell_NotifyIcon(NIM_ADD, &m_data);
}
void ModifyToolTips(LPCTSTR info)
{
strcpy_s(m_data.szInfo, info);
}
BOOL DispalyToolTips()
{
return Shell_NotifyIcon(NIM_MODIFY, &m_data);
}
BOOL BalloonToolTips(LPCTSTR info)
{
ModifyToolTips(LPCTSTR info);
return DispalyToolTips();
}
BEGIN_MSG_MAP(CShellIcon)
MESSAGE_HANDLER(WM_ICON, OnIcon)
MESSAGE_HANDLER(m_msgTaskbarRestart, OnRestart)
MESSAGE_HANDLER(WM_SIZE, OnSize)
CHAIN_MSG_MAP(MenuT)
END_MSG_MAP()
LRESULT OnIcon(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
T* pT = static_cast(this); char t;
if (wParam != IDR_MAINFRAME) return 1;
switch(lParam)
{
case WM_RBUTTONUP:
t = *m_data.szInfo;
*m_data.szInfo = '/0';
Shell_NotifyIcon(NIM_MODIFY, &m_data);
pT->CreateContextMenu(MenuID);
*m_data.szInfo = t;
break;
case WM_LBUTTONUP:
pT->ShowWindow(SW_SHOW);
OpenIcon(pT->m_hWnd);
break;
//去掉下面的注釋可以使鼠標懸停在圖標上時出現氣泡提示，個人不太喜歡
case WM_MOUSEMOVE:
// DispalyToolTips(); break;
default:
;
}
return 0;
}
//處理EXPlorer外殼崩潰后任務欄重建，你會發現很多程序都沒處理，結果就是Explorer一崩潰
//圖標就找不到了，假如最小化到任務欄更慘，還得用任務治理器關閉，這個函數我沒機會測試，希望有作用
LRESULT OnRestart(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
T* pT = static_cast(this);
SecureZeroMemory(&m_data, sizeof(m_data));
m_data.cbSize = sizeof(m_data);
m_data.hWnd = pT->m_hWnd;
m_data.uID = IDR_MAINFRAME;
Shell_NotifyIcon(NIM_DELETE, &m_data);
CreateShellIcon();
return 0;
}
//最小化到系統托盤
LRESULT OnSize(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
T* pT = static_cast(this);
if (pT->IsIconic()) pT->ShowWindow(SW_HIDE);
return 0;
}
};

//MyIconMenu.h
#pragma once
#include "stdafx.h"
#include 
template 
class CMyIconMenu : public CContextMenu
{
public:
BEGIN_MSG_MAP(CContextMenu)
COMMAND_ID_HANDLER(ID_RESUME, OnResume)
COMMAND_ID_HANDLER(ID_QUIT, OnQuit)
END_MSG_MAP()
LRESULT OnResume(Word /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
T* pT = static_cast(this);
pT->ShowWindow(SW_SHOW);
OpenIcon(pT->m_hWnd);
return 0;
}

LRESULT OnQuit(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
PostQuitMessage(0);
return 0;
}
};