forked from getlantern/systray
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
12 changed files
with
554 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -7,3 +7,4 @@ Debug | |
*.sdf | ||
dll/systray_unsigned.dll | ||
out.txt | ||
.vs |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,48 @@ | ||
======================================================================== | ||
DYNAMIC LINK LIBRARY : systray Project Overview | ||
======================================================================== | ||
|
||
AppWizard has created this systray DLL for you. | ||
|
||
This file contains a summary of what you will find in each of the files that | ||
make up your systray application. | ||
|
||
|
||
systray.vcxproj | ||
This is the main project file for VC++ projects generated using an Application Wizard. | ||
It contains information about the version of Visual C++ that generated the file, and | ||
information about the platforms, configurations, and project features selected with the | ||
Application Wizard. | ||
|
||
systray.vcxproj.filters | ||
This is the filters file for VC++ projects generated using an Application Wizard. | ||
It contains information about the association between the files in your project | ||
and the filters. This association is used in the IDE to show grouping of files with | ||
similar extensions under a specific node (for e.g. ".cpp" files are associated with the | ||
"Source Files" filter). | ||
|
||
systray.cpp | ||
This is the main DLL source file. | ||
|
||
When created, this DLL does not export any symbols. As a result, it | ||
will not produce a .lib file when it is built. If you wish this project | ||
to be a project dependency of some other project, you will either need to | ||
add code to export some symbols from the DLL so that an export library | ||
will be produced, or you can set the Ignore Input Library property to Yes | ||
on the General propert page of the Linker folder in the project's Property | ||
Pages dialog box. | ||
|
||
///////////////////////////////////////////////////////////////////////////// | ||
Other standard files: | ||
|
||
StdAfx.h, StdAfx.cpp | ||
These files are used to build a precompiled header (PCH) file | ||
named systray.pch and a precompiled types file named StdAfx.obj. | ||
|
||
///////////////////////////////////////////////////////////////////////////// | ||
Other notes: | ||
|
||
AppWizard uses "TODO:" comments to indicate parts of the source code you | ||
should add to or customize. | ||
|
||
///////////////////////////////////////////////////////////////////////////// |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
// dllmain.cpp : Defines the entry point for the DLL application. | ||
#include "stdafx.h" | ||
|
||
BOOL APIENTRY DllMain( HMODULE hModule, | ||
DWORD ul_reason_for_call, | ||
LPVOID lpReserved | ||
) | ||
{ | ||
switch (ul_reason_for_call) | ||
{ | ||
case DLL_PROCESS_ATTACH: | ||
case DLL_THREAD_ATTACH: | ||
case DLL_THREAD_DETACH: | ||
case DLL_PROCESS_DETACH: | ||
break; | ||
} | ||
return TRUE; | ||
} | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
// stdafx.cpp : source file that includes just the standard includes | ||
// systray.pch will be the pre-compiled header | ||
// stdafx.obj will contain the pre-compiled type information | ||
|
||
#include "stdafx.h" | ||
|
||
// TODO: reference any additional headers you need in STDAFX.H | ||
// and not in this file |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
// stdafx.h : include file for standard system include files, | ||
// or project specific include files that are used frequently, but | ||
// are changed infrequently | ||
// | ||
|
||
#pragma once | ||
|
||
#include "targetver.h" | ||
|
||
#define WIN32_LEAN_AND_MEAN // Exclude rarely-used stuff from Windows headers | ||
// Windows Header Files: | ||
#include <windows.h> | ||
// Additional required headers | ||
#include <stdio.h> | ||
#include <stdlib.h> | ||
#include <windows.h> | ||
#include <shellapi.h> |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,207 @@ | ||
// systray.cpp : Defines the exported functions for the DLL application. | ||
// | ||
|
||
// dllmain.cpp : Defines the entry point for the DLL application. | ||
#include "stdafx.h" | ||
#include "systray.h" | ||
|
||
// Message posted into message loop when Notification Icon is clicked | ||
#define WM_SYSTRAY_MESSAGE (WM_USER + 1) | ||
|
||
static NOTIFYICONDATA nid; | ||
static HWND hWnd; | ||
static HMENU hTrayMenu; | ||
|
||
void (*systray_on_exit)(int ignored); | ||
void (*systray_menu_item_selected)(int menu_id); | ||
|
||
void reportWindowsError(const char* action) { | ||
LPTSTR pErrMsg = NULL; | ||
DWORD errCode = GetLastError(); | ||
DWORD result = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER| | ||
FORMAT_MESSAGE_FROM_SYSTEM| | ||
FORMAT_MESSAGE_ARGUMENT_ARRAY, | ||
NULL, | ||
errCode, | ||
LANG_NEUTRAL, | ||
pErrMsg, | ||
0, | ||
NULL); | ||
printf("Systray error %s: %d %ls\n", action, errCode, pErrMsg); | ||
} | ||
|
||
void ShowMenu(HWND hWnd) { | ||
POINT p; | ||
if (0 == GetCursorPos(&p)) { | ||
reportWindowsError("get tray menu position"); | ||
return; | ||
}; | ||
SetForegroundWindow(hWnd); // Win32 bug work-around | ||
TrackPopupMenu(hTrayMenu, TPM_BOTTOMALIGN | TPM_LEFTALIGN, p.x, p.y, 0, hWnd, NULL); | ||
|
||
} | ||
|
||
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { | ||
switch (message) { | ||
case WM_COMMAND: | ||
{ | ||
int menuId = LOWORD(wParam); | ||
if (menuId != -1) { | ||
systray_menu_item_selected(menuId); | ||
} | ||
} | ||
break; | ||
case WM_DESTROY: | ||
printf("Window destroyed\n"); | ||
systray_on_exit(0/*ignored*/); | ||
Shell_NotifyIcon(NIM_DELETE, &nid); | ||
PostQuitMessage(0); | ||
break; | ||
case WM_ENDSESSION: | ||
printf("Session ending\n"); | ||
// When the system shuts down (or logs off), we don't receive WM_DESTROY, | ||
// so we capture WM_ENDSESSION instead and call on_exit here too. | ||
systray_on_exit(0/*ignored*/); | ||
Shell_NotifyIcon(NIM_DELETE, &nid); | ||
break; | ||
case WM_SYSTRAY_MESSAGE: | ||
switch(lParam) { | ||
case WM_RBUTTONUP: | ||
ShowMenu(hWnd); | ||
break; | ||
case WM_LBUTTONUP: | ||
ShowMenu(hWnd); | ||
break; | ||
default: | ||
return DefWindowProc(hWnd, message, wParam, lParam); | ||
}; | ||
break; | ||
default: | ||
return DefWindowProc(hWnd, message, wParam, lParam); | ||
} | ||
return 0; | ||
} | ||
|
||
void MyRegisterClass(HINSTANCE hInstance, TCHAR* szWindowClass) { | ||
WNDCLASSEX wcex; | ||
|
||
wcex.cbSize = sizeof(WNDCLASSEX); | ||
wcex.style = CS_HREDRAW | CS_VREDRAW; | ||
wcex.lpfnWndProc = WndProc; | ||
wcex.cbClsExtra = 0; | ||
wcex.cbWndExtra = 0; | ||
wcex.hInstance = hInstance; | ||
wcex.hIcon = LoadIcon(NULL, IDI_APPLICATION); | ||
wcex.hCursor = LoadCursor(NULL, IDC_ARROW); | ||
wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1); | ||
wcex.lpszMenuName = 0; | ||
wcex.lpszClassName = szWindowClass; | ||
wcex.hIconSm = LoadIcon(NULL, IDI_APPLICATION); | ||
|
||
RegisterClassEx(&wcex); | ||
} | ||
|
||
HWND InitInstance(HINSTANCE hInstance, int nCmdShow, TCHAR* szWindowClass) { | ||
HWND hWnd = CreateWindow(szWindowClass, TEXT(""), WS_OVERLAPPEDWINDOW, | ||
CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL); | ||
if (!hWnd) { | ||
return 0; | ||
} | ||
|
||
ShowWindow(hWnd, nCmdShow); | ||
UpdateWindow(hWnd); | ||
|
||
return hWnd; | ||
} | ||
|
||
|
||
BOOL createMenu() { | ||
hTrayMenu = CreatePopupMenu(); | ||
MENUINFO menuInfo; | ||
menuInfo.cbSize = sizeof(MENUINFO); | ||
menuInfo.fMask = MIM_APPLYTOSUBMENUS | MIM_STYLE; | ||
return SetMenuInfo(hTrayMenu, &menuInfo); | ||
} | ||
|
||
BOOL addNotifyIcon() { | ||
nid.cbSize = sizeof(NOTIFYICONDATA); | ||
nid.hWnd = hWnd; | ||
nid.uID = 100; | ||
nid.uCallbackMessage = WM_SYSTRAY_MESSAGE; | ||
nid.uFlags = NIF_MESSAGE; | ||
return Shell_NotifyIcon(NIM_ADD, &nid); | ||
} | ||
|
||
int nativeLoop(void (*systray_ready)(int ignored), | ||
void (*_systray_on_exit)(int ignored), | ||
void (*_systray_menu_item_selected)(int menu_id)) { | ||
systray_on_exit = _systray_on_exit; | ||
systray_menu_item_selected = _systray_menu_item_selected; | ||
|
||
HINSTANCE hInstance = GetModuleHandle(NULL); | ||
TCHAR* szWindowClass = TEXT("SystrayClass"); | ||
MyRegisterClass(hInstance, szWindowClass); | ||
hWnd = InitInstance(hInstance, FALSE, szWindowClass); // Don't show window | ||
if (!hWnd) { | ||
return EXIT_FAILURE; | ||
} | ||
if (!createMenu() || !addNotifyIcon()) { | ||
return EXIT_FAILURE; | ||
} | ||
systray_ready(0); | ||
|
||
MSG msg; | ||
while (GetMessage(&msg, NULL, 0, 0)) { | ||
TranslateMessage(&msg); | ||
DispatchMessage(&msg); | ||
} | ||
return EXIT_SUCCESS; | ||
} | ||
|
||
|
||
void setIcon(const wchar_t* iconFile) { | ||
HICON hIcon = (HICON) LoadImage(NULL, iconFile, IMAGE_ICON, 64, 64, LR_LOADFROMFILE); | ||
if (hIcon == NULL) { | ||
reportWindowsError("load icon image"); | ||
} else { | ||
nid.hIcon = hIcon; | ||
nid.uFlags = NIF_ICON; | ||
Shell_NotifyIcon(NIM_MODIFY, &nid); | ||
} | ||
} | ||
|
||
void setTooltip(const wchar_t* tooltip) { | ||
wcsncpy_s(nid.szTip, tooltip, sizeof(nid.szTip)/sizeof(wchar_t)); | ||
nid.uFlags = NIF_TIP; | ||
Shell_NotifyIcon(NIM_MODIFY, &nid); | ||
} | ||
|
||
void add_or_update_menu_item(int menuId, wchar_t* title, wchar_t* tooltip, short disabled, short checked) { | ||
MENUITEMINFO menuItemInfo; | ||
menuItemInfo.cbSize = sizeof(MENUITEMINFO); | ||
menuItemInfo.fMask = MIIM_FTYPE | MIIM_STRING | MIIM_ID | MIIM_STATE; | ||
menuItemInfo.fType = MFT_STRING; | ||
menuItemInfo.dwTypeData = title; | ||
menuItemInfo.cch = wcslen(title) + 1; | ||
menuItemInfo.wID = menuId; | ||
menuItemInfo.fState = 0; | ||
if (disabled == 1) { | ||
menuItemInfo.fState |= MFS_DISABLED; | ||
} | ||
if (checked == 1) { | ||
menuItemInfo.fState |= MFS_CHECKED; | ||
} | ||
|
||
BOOL setOK = SetMenuItemInfo(hTrayMenu, menuId, FALSE, &menuItemInfo); | ||
if (!setOK) { | ||
InsertMenuItem(hTrayMenu, menuId, TRUE, &menuItemInfo); | ||
} | ||
} | ||
|
||
void hide_menu_item(int menuId) { | ||
DeleteMenu(hTrayMenu, menuId, MF_BYCOMMAND); | ||
} | ||
|
||
void quit() { | ||
PostMessage(hWnd, WM_CLOSE, 0, 0); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,9 @@ | ||
LIBRARY systray | ||
DESCRIPTION "System Tray Support" | ||
EXPORTS | ||
nativeLoop @1 | ||
setIcon @2 | ||
setTooltip @3 | ||
add_or_update_menu_item @4 | ||
hide_menu_item @5 | ||
quit @6 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,13 @@ | ||
#pragma once | ||
|
||
extern "C" { | ||
__declspec(dllexport) int __cdecl nativeLoop(void (*systray_ready)(int ignored), | ||
void (*_systray_on_exit)(int ignored), | ||
void (*_systray_menu_item_selected)(int menu_id)); | ||
|
||
__declspec(dllexport) void __cdecl setIcon(const wchar_t* iconFile); | ||
__declspec(dllexport) void __cdecl setTooltip(const wchar_t* tooltip); | ||
__declspec(dllexport) void __cdecl add_or_update_menu_item(int menuId, wchar_t* title, wchar_t* tooltip, short disabled, short checked); | ||
__declspec(dllexport) void __cdecl hide_menu_item(int menuId); | ||
__declspec(dllexport) void __cdecl quit(); | ||
} |
Oops, something went wrong.