224 lines
5.3 KiB
C++
224 lines
5.3 KiB
C++
|
// shelltray.cpp
|
||
|
//
|
||
|
// Copyright 2000 Microsoft Corporation, all rights reserved
|
||
|
//
|
||
|
// Created 2-00 - anbrad
|
||
|
//
|
||
|
|
||
|
#include "main.h"
|
||
|
#include "shelltray.h"
|
||
|
|
||
|
#include "resource.h"
|
||
|
#include "netwatch.h"
|
||
|
#include "dsubmit.h"
|
||
|
|
||
|
void OpenContextMenu(HWND hwnd, POINT * pPoint);
|
||
|
void OnTaskBarIconRButtonUp(HWND hwnd);
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// AddTrayIcon
|
||
|
//
|
||
|
void
|
||
|
AddTrayIcon(
|
||
|
HWND hwnd)
|
||
|
{
|
||
|
NOTIFYICONDATA nid;
|
||
|
HICON hiconTray = LoadIcon(g_hInst, MAKEINTRESOURCE(IDI_FACE));
|
||
|
|
||
|
if (hiconTray)
|
||
|
{
|
||
|
nid.uID = 0;
|
||
|
nid.cbSize = sizeof(NOTIFYICONDATA);
|
||
|
nid.hWnd = hwnd;
|
||
|
nid.uCallbackMessage = WM_USER_TRAYCALLBACK;
|
||
|
nid.hIcon = hiconTray;
|
||
|
nid.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
|
||
|
|
||
|
lstrcpy(nid.szTip, TEXT("Double click to save your network trace."));
|
||
|
}
|
||
|
|
||
|
g_fTrayPresent = Shell_NotifyIcon(NIM_ADD, &nid);
|
||
|
}
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// RemoveTrayIcon
|
||
|
//
|
||
|
void
|
||
|
RemoveTrayIcon(
|
||
|
HWND hwnd)
|
||
|
{
|
||
|
NOTIFYICONDATA nid;
|
||
|
|
||
|
if (g_fTrayPresent)
|
||
|
{
|
||
|
nid.uID = 0;
|
||
|
nid.cbSize = sizeof(NOTIFYICONDATA);
|
||
|
nid.hWnd = hwnd;
|
||
|
nid.uCallbackMessage = WM_USER_TRAYCALLBACK;
|
||
|
nid.uFlags = 0;
|
||
|
}
|
||
|
|
||
|
g_fTrayPresent = !(Shell_NotifyIcon(NIM_DELETE, &nid));
|
||
|
}
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// UpdateTrayIcon
|
||
|
//
|
||
|
void
|
||
|
UpdateTrayIcon(
|
||
|
HWND hwnd)
|
||
|
{
|
||
|
NOTIFYICONDATA nid;
|
||
|
|
||
|
if (g_fTrayPresent)
|
||
|
{
|
||
|
nid.uID = 0;
|
||
|
nid.cbSize = sizeof(NOTIFYICONDATA);
|
||
|
nid.hWnd = hwnd;
|
||
|
nid.uCallbackMessage = WM_USER_TRAYCALLBACK;
|
||
|
nid.uFlags = NIF_TIP;
|
||
|
|
||
|
_tcscpy(nid.szTip, TEXT("Double click to save your network trace."));
|
||
|
}
|
||
|
|
||
|
Shell_NotifyIcon(NIM_MODIFY, &nid);
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// ProcessTrayCallback
|
||
|
//
|
||
|
void
|
||
|
ProcessTrayCallback(
|
||
|
HWND hwnd,
|
||
|
WPARAM wParam,
|
||
|
LPARAM lParam)
|
||
|
{
|
||
|
UINT uID = (UINT) wParam;
|
||
|
UINT uMouseMsg = (UINT) lParam;
|
||
|
static bInDialog;
|
||
|
|
||
|
|
||
|
switch(uMouseMsg)
|
||
|
{
|
||
|
case WM_LBUTTONDBLCLK:
|
||
|
|
||
|
if (!bInDialog)
|
||
|
{
|
||
|
bInDialog = TRUE;
|
||
|
StopCapture();
|
||
|
|
||
|
if (IDOK == DialogBox(
|
||
|
g_hInst,
|
||
|
MAKEINTRESOURCE(IDD_SUBMIT),
|
||
|
NULL,
|
||
|
DlgProcSubmit))
|
||
|
{
|
||
|
SaveCapture();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
RestartCapture();
|
||
|
}
|
||
|
|
||
|
bInDialog = FALSE;
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case WM_RBUTTONUP:
|
||
|
OnTaskBarIconRButtonUp(hwnd);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void OnTaskBarIconRButtonUp(HWND hwnd)
|
||
|
{
|
||
|
POINT pt;
|
||
|
|
||
|
GetCursorPos(&pt);
|
||
|
OpenContextMenu(hwnd, &pt);
|
||
|
}
|
||
|
|
||
|
#if (WINVER > 0x0400)
|
||
|
VOID SetIconFocus(HWND hwnd)
|
||
|
{
|
||
|
NOTIFYICONDATA nid;
|
||
|
|
||
|
ZeroMemory (&nid, sizeof(nid));
|
||
|
nid.cbSize = sizeof(NOTIFYICONDATA);
|
||
|
nid.hWnd = hwnd;
|
||
|
nid.uID = 0;
|
||
|
|
||
|
// Shell_NotifyIcon(NIM_SETFOCUS, &nid);
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
void OpenContextMenu(HWND hwnd, POINT * pPoint)
|
||
|
{
|
||
|
HRESULT hr = S_OK;
|
||
|
INT iCmd = 0;
|
||
|
INT iMenu = 0;
|
||
|
HMENU hmenu = 0;
|
||
|
BOOL fDisconnected = FALSE;
|
||
|
INT iIdCustomMin = -1;
|
||
|
INT iIdCustomMax = -1;
|
||
|
BOOL fBranded = FALSE;
|
||
|
|
||
|
// Find the connection info based on the tray icon id.
|
||
|
//
|
||
|
hmenu = LoadMenu(g_hInst, MAKEINTRESOURCE(POPUP_TRAY));
|
||
|
if (hmenu)
|
||
|
{
|
||
|
// Get the first menu from the popup. For some reason, this hack is
|
||
|
// required instead of tracking on the outside menu
|
||
|
//
|
||
|
HMENU hmenuTrack = GetSubMenu(hmenu, 0);
|
||
|
|
||
|
// Set the default menu item
|
||
|
//
|
||
|
SetMenuDefaultItem(hmenuTrack, CMIDM_TRAY_CLOSE, FALSE);
|
||
|
|
||
|
// Set the owner window to be foreground as a hack so the
|
||
|
// popup menu disappears when the user clicks elsewhere.
|
||
|
//
|
||
|
SetForegroundWindow(hwnd);
|
||
|
|
||
|
// Part of the above hack. Bring up the menu and figure out the result
|
||
|
iCmd = TrackPopupMenu(hmenuTrack, TPM_RETURNCMD | TPM_NONOTIFY | TPM_RIGHTBUTTON,
|
||
|
pPoint->x, pPoint->y, 0, hwnd, NULL);
|
||
|
DestroyMenu(hmenu);
|
||
|
|
||
|
MSG msgTmp;
|
||
|
while (PeekMessage(&msgTmp, hwnd, WM_LBUTTONDOWN, WM_LBUTTONUP, PM_REMOVE))
|
||
|
{
|
||
|
DispatchMessage(&msgTmp);
|
||
|
}
|
||
|
|
||
|
// Process the command
|
||
|
//
|
||
|
switch (iCmd)
|
||
|
{
|
||
|
case CMIDM_TRAY_CLOSE:
|
||
|
SendMessage(hwnd, WM_DESTROY, 0, 0);
|
||
|
break;
|
||
|
|
||
|
// Tray menu cancelled without selection
|
||
|
//
|
||
|
case 0:
|
||
|
break;
|
||
|
|
||
|
// Unknown command
|
||
|
//
|
||
|
default:
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
// Shift the focus back to the shell
|
||
|
//
|
||
|
#if (WINVER > 0x0400)
|
||
|
SetIconFocus(hwnd);
|
||
|
#endif
|
||
|
}
|
||
|
}
|
||
|
|