[转载]Win32 API 实现系统托盘程序

#include <windows.h>
#include "resource.h"

LPCTSTR szAppName = TEXT("TrayHelper");
LPCTSTR szCaption = TEXT("TrayIcon");

LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
static HMENU hMenu;
static UINT WM_TASKBARCREATED;

POINT point;
HINSTANCE hInstance;
NOTIFYICONDATA nid;

switch (message)
{
case WM_CREATE:
// 不要修改TaskbarCreated 字符串，这是系统任务栏自定义的消息
WM_TASKBARCREATED = RegisterWindowMessage(TEXT("TaskbarCreated"));

hInstance = ((LPCREATESTRUCT)lParam)->hInstance;

hMenu = LoadMenu(hInstance, MAKEINTRESOURCE(IDR_TRAYMENU));
hMenu = GetSubMenu(hMenu, 0);

nid.cbSize = sizeof(nid);
nid.hWnd = hWnd;
nid.uID = IDI_PIRAMIDE;
nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
nid.uCallbackMessage = WM_USER;
nid.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_PIRAMIDE));
strcpy(nid.szTip, szAppName);
Shell_NotifyIcon(NIM_ADD, &nid);
break;

case WM_USER:
if (lParam == WM_RBUTTONDOWN)
{
GetCursorPos(&point);
SetForegroundWindow(hWnd); // 修正当用户按下ESCAPE 键或者在菜单之外单击鼠标时菜单不会消失的情况
TrackPopupMenu(hMenu, TPM_RIGHTBUTTON, point.x, point.y, 0, hWnd, NULL);
}
break;

case WM_COMMAND:
switch (LOWORD(wParam)) {
case IDM_TRAYSETTINGS:
MessageBox (hWnd, TEXT("Settings not yet implemented!"), szAppName, MB_ICONEXCLAMATION | MB_OK) ;
return 0;

case IDM_TRAYHELP:
MessageBox (hWnd, TEXT("Help not yet implemented!"), szAppName, MB_ICONEXCLAMATION | MB_OK) ;
return 0;

case IDM_TRAYABOUT:
MessageBox (hWnd, TEXT("About not yet implemented!"), szAppName, MB_ICONEXCLAMATION | MB_OK) ;
return 0;

case IDM_TRAYEXIT:
SendMessage(hWnd, WM_CLOSE, 0, 0);
return 0;
}
break;

case WM_DESTROY:
nid.uID = IDI_PIRAMIDE;  // 修正点击Exit 菜单退出后图标仍在托盘区显示，
nid.hWnd = hWnd;        //  要把鼠标在图标上面过一下才会消失。
Shell_NotifyIcon(NIM_DELETE, &nid);
PostQuitMessage(0);
break;

default:
/*
* 防止当Explorer.exe 崩溃以后，程序在系统托盘中的图标就消失
*
* 原理：Explorer.exe 重新载入后会重建系统任务栏。当系统任务栏建立的时候会向系统内所有
* 注册接收TaskbarCreated 消息的顶级窗口发送一条消息，我们只需要捕捉这个消息，并重建系
* 统托盘的图标即可。
*/
if (message == WM_TASKBARCREATED)
SendMessage(hWnd, WM_CREATE, wParam, lParam);
break;
}
return DefWindowProc(hWnd, message, wParam, lParam);
}

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR szCmdLine, int iCmdShow)
{
HWND hWnd;
MSG msg;
WNDCLASS wndclass;

HWND handle = FindWindow(NULL, szCaption);
if (handle != NULL)
{
MessageBox(NULL, TEXT("Application is already running"), szAppName, MB_ICONERROR);
return 0;
}

wndclass.style = CS_HREDRAW | CS_VREDRAW;
wndclass.lpfnWndProc = WndProc;
wndclass.cbClsExtra = 0;
wndclass.cbWndExtra = 0;
wndclass.hInstance = hInstance;
wndclass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
wndclass.hCursor = LoadCursor(NULL, IDC_ARROW);
wndclass.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);
wndclass.lpszMenuName = NULL;
wndclass.lpszClassName = szAppName;

if (!RegisterClass(&wndclass))
{
MessageBox(NULL, TEXT("This program requires Windows NT!"), szAppName, MB_ICONERROR);
return 0;
}

// 此处使用WS_EX_TOOLWINDOW 属性来隐藏显示在任务栏上的窗口程序按钮
hWnd = CreateWindowEx(WS_EX_TOOLWINDOW,
szAppName, szCaption,
WS_POPUP,
CW_USEDEFAULT,
CW_USEDEFAULT,
CW_USEDEFAULT,
CW_USEDEFAULT,
NULL, NULL, hInstance, NULL);

ShowWindow(hWnd, iCmdShow);
UpdateWindow(hWnd);

while (GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}

return msg.wParam;
}