WTL::右键弹出菜单和托盘图标

 　WTL嵌入类的架构可以使整个程序很清晰，系统托盘气泡提示、上下文菜单处理可以分别独立出来，以下是实现代码，很清晰就不注释了。基本上这两个类很少需要修改，因此我把它们放到了系统包含文件搜索路径中。

//ContextMenu.h
#pragma once
template <class T>
class CContextMenu
{
public:
             BOOL CreateContextMenu(UINT ID_Menu)
             {
                           T* pT = static_cast<T*>(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 T, class MenuT, int MenuID>
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<T*>(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<T*>(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<T*>(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 , WPARAM , LPARAM , BOOL& )
             {
                           T* pT = static_cast<T*>(this);
                           if (pT->IsIconic()) pT->ShowWindow(SW_HIDE);
                           return 0;
             }
};

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

             LRESULT OnQuit(WORD , WORD wID, HWND , BOOL& )
             {
                           PostQuitMessage(0);
                           return 0;
             }
};

 

　　CContextMenu的使用

　　如果你的窗口需要上下文菜单，只需要从CContextMenu派生一个类，例如：template <class T> class CMyContextMenu : public CContextMenu<T>

　　然后完成消息映射，例如CMyIconMenu所做的。在你的窗口类（假设为CMyWindowClass）的继承列表里面添加public CMyContextMenu<CMyWindowClass>，在消息映射表中添加CHAIN_MSG_MAP(CMyContextMenu<CMyWindowClass>)

　　接下来只要在需要显示上下文菜单的地方，调用CreatContextMenu(UINT ID_Menu)就可以了。

　　CShellIcon的使用

　　首先要把stdafx.h中IE的版本改成5或者6，例如：

#define _WIN32_IE           0x0600

　　然后加入CString支持，可能每个项目都几乎用到CString，大家不妨直接修改模板文件，让向导直接把CString加进去算了。

　　从CContextMenu派生一个类来完成菜单映射，例如CMyIconMenu所作的。

　　以下用模式对话框来演示如何给其加入托盘功能，蓝色的部分为添加的部分。

#pragma once
#include "MyIconMenu.h"
#include <ShellIcon.h>
class CMainDlg : public CDialogImpl<CMainDlg>, public CShellIcon<CMainDlg, CMyIconMenu<CMainDlg>, IDR_ICONMENU>
{
public:
             enum { IDD = IDD_MAINDLG };
             typedef CShellIcon<CMainDlg, CMyIconMenu<CMainDlg>, IDR_ICONMENU> CMyShellIcon;
             BEGIN_MSG_MAP(CMainDlg)
                           MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
                           COMMAND_ID_HANDLER(ID_APP_ABOUT, OnAppAbout)
                           COMMAND_ID_HANDLER(IDOK, OnOK)
                           COMMAND_ID_HANDLER(IDCANCEL, OnCancel)
                           CHAIN_MSG_MAP(CMyShellIcon)
             END_MSG_MAP()
             LRESULT OnInitDialog(UINT , WPARAM , LPARAM , BOOL& )
             {
                           // center the dialog on the screen
                           CenterWindow();
                           // set icons
                           HICON hIcon = (HICON)::LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDR_MAINFRAME), 
                                         IMAGE_ICON, ::GetSystemMetrics(SM_CXICON), ::GetSystemMetrics(SM_CYICON), LR_DEFAULTCOLOR);
                           SetIcon(hIcon, TRUE);
                           HICON hIconSmall = (HICON)::LoadImage(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDR_MAINFRAME), 
                                         IMAGE_ICON, ::GetSystemMetrics(SM_CXSMICON), ::GetSystemMetrics(SM_CYSMICON), LR_DEFAULTCOLOR);
                           SetIcon(hIconSmall, FALSE);
                           CreateShellIcon();
                           return TRUE;
             }
……………………
};

　　在想弹出气泡的时候调用BOOL BalloonToolTips(LPCTSTR info)