mirror of
https://github.com/redocnib/docker-tray
synced 2024-11-26 04:14:40 +01:00
Initial setup
This commit is contained in:
Kiran
commit
5cde3d3c50
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
docker-tray
|
||||||
|
.vscode/
|
||||||
164
docker-service-tray.cc
Normal file
164
docker-service-tray.cc
Normal file
@ -0,0 +1,164 @@
|
|||||||
|
//g++ docker-service-tray.cc -o docker-tray `pkg-config --cflags --libs gtk+-3.0 appindicator3-0.1` -lsystemd
|
||||||
|
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
#include <string>
|
||||||
|
#include <cctype>
|
||||||
|
#include <unistd.h> //Header file for sleep(). man 3 sleep for details.
|
||||||
|
#include <pthread.h>
|
||||||
|
#include <systemd/sd-bus.h>
|
||||||
|
#include <libgen.h>
|
||||||
|
# define TRAY_APPINDICATOR 1
|
||||||
|
#include "tray.h"
|
||||||
|
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
static void docker_switch_db(struct tray_menu * item);
|
||||||
|
static void quit_cb(struct tray_menu * item);
|
||||||
|
|
||||||
|
void * sdbusHookServiceAsync(void * vargp);
|
||||||
|
|
||||||
|
static struct tray tray = {
|
||||||
|
.menu =
|
||||||
|
(struct tray_menu[]) {
|
||||||
|
{
|
||||||
|
.text = "Please Wait...", .cb = docker_switch_db
|
||||||
|
},
|
||||||
|
{
|
||||||
|
.text = "Quit",
|
||||||
|
.cb = quit_cb
|
||||||
|
}, {
|
||||||
|
.text = NULL
|
||||||
|
}
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
bool QUITTING = false; //Shared state for thread, no mutex for now
|
||||||
|
|
||||||
|
string TRAY_ICON1 = "/home/beep/Desktop/Projects/c/test1/active.png";
|
||||||
|
string TRAY_ICON2 = "/home/beep/Desktop/Projects/c/test1/inactive.png";
|
||||||
|
string TRAY_ICON3 = "/home/beep/Desktop/Projects/c/test1/unknown.png";
|
||||||
|
|
||||||
|
|
||||||
|
/* Application Entry Point */
|
||||||
|
int main() {
|
||||||
|
char exePath[PATH_MAX];
|
||||||
|
ssize_t len = ::readlink("/proc/self/exe", exePath, sizeof(exePath));
|
||||||
|
if (len == -1 || len == sizeof(exePath))
|
||||||
|
len = 0;
|
||||||
|
exePath[len] = '\0';
|
||||||
|
string base_path = dirname(exePath);
|
||||||
|
TRAY_ICON1 = base_path + "/icons/active.png";
|
||||||
|
TRAY_ICON2 = base_path + "/icons/inactive.png";
|
||||||
|
TRAY_ICON3 = base_path + "/icons/unknown.png";
|
||||||
|
|
||||||
|
tray.icon = const_cast<char*>(TRAY_ICON3.c_str());
|
||||||
|
|
||||||
|
if (tray_init( & tray) < 0) {
|
||||||
|
printf("failed to create tray\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
pthread_t thread_id;
|
||||||
|
pthread_create( & thread_id, NULL, sdbusHookServiceAsync, NULL);
|
||||||
|
while (tray_loop(1) == 0) {
|
||||||
|
//Tray Updates
|
||||||
|
}
|
||||||
|
QUITTING = true;
|
||||||
|
(void) pthread_join(thread_id, NULL);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
Threaded sd_bus event lister
|
||||||
|
*/
|
||||||
|
void * sdbusHookServiceAsync(void * vargp) {
|
||||||
|
sd_bus* bus = NULL;
|
||||||
|
sd_bus_error err = SD_BUS_ERROR_NULL;
|
||||||
|
char* msg = NULL;
|
||||||
|
sd_bus_message* m = NULL;
|
||||||
|
void* userdata = NULL;
|
||||||
|
|
||||||
|
sd_bus_default_system(&bus);
|
||||||
|
|
||||||
|
sd_bus_match_signal(
|
||||||
|
bus, /* bus */
|
||||||
|
NULL, /* slot */
|
||||||
|
NULL, /* sender */
|
||||||
|
"/org/freedesktop/systemd1/unit/docker_2eservice", /* path */
|
||||||
|
"org.freedesktop.DBus.Properties", /* interface */
|
||||||
|
"PropertiesChanged", /* member */
|
||||||
|
NULL, //message_callback , /* callback */
|
||||||
|
userdata
|
||||||
|
);
|
||||||
|
|
||||||
|
while (!QUITTING) {
|
||||||
|
m = NULL;
|
||||||
|
int r = sd_bus_process(bus, &m);
|
||||||
|
if (r < 0) {
|
||||||
|
printf("Failed to process requests: %d", r);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (r == 0) {
|
||||||
|
r = sd_bus_wait(bus, 1000000);
|
||||||
|
if (r < 0) {
|
||||||
|
printf("Failed to wait: %d", r);
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (!m) continue;
|
||||||
|
sd_bus_get_property_string(
|
||||||
|
bus, /* bus */
|
||||||
|
"org.freedesktop.systemd1", /* destination */
|
||||||
|
"/org/freedesktop/systemd1/unit/docker_2eservice", /* path */
|
||||||
|
"org.freedesktop.systemd1.Unit", /* interface */
|
||||||
|
"ActiveState", /* member */ &err, &msg);
|
||||||
|
|
||||||
|
//printf("State Update From Service: %s\n", msg);
|
||||||
|
|
||||||
|
if (strcmp(msg, "active") == 0) {
|
||||||
|
tray.icon = const_cast<char*>(TRAY_ICON1.c_str());
|
||||||
|
tray.menu[0].text = "Stop Docker Service";
|
||||||
|
tray.menu[0].disabled = 0;
|
||||||
|
} else if (strcmp(msg, "inactive") == 0) {
|
||||||
|
tray.icon = const_cast<char*>(TRAY_ICON2.c_str());
|
||||||
|
tray.menu[0].text = "Start Docker Service";
|
||||||
|
tray.menu[0].disabled = 0;
|
||||||
|
} else {
|
||||||
|
tray.icon = const_cast<char*>(TRAY_ICON3.c_str());
|
||||||
|
tray.menu[0].text = "Working...";
|
||||||
|
tray.menu[0].disabled = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
tray_update( & tray);
|
||||||
|
sd_bus_message_unref(m);
|
||||||
|
free(msg);
|
||||||
|
}
|
||||||
|
sd_bus_error_free( & err);
|
||||||
|
sd_bus_unref(bus);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
Tray Callbacks
|
||||||
|
*/
|
||||||
|
|
||||||
|
static void docker_switch_db(struct tray_menu * item) {
|
||||||
|
(void) item;
|
||||||
|
if (strcmp(tray.icon, const_cast<char*>(TRAY_ICON1.c_str())) == 0) {
|
||||||
|
system("systemctl stop docker");
|
||||||
|
} else {
|
||||||
|
system("systemctl start docker");
|
||||||
|
}
|
||||||
|
tray_update( & tray);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void quit_cb(struct tray_menu * item) {
|
||||||
|
(void) item;
|
||||||
|
tray_exit();
|
||||||
|
}
|
||||||
|
|
||||||
BIN
icons/active.png
Normal file
BIN
icons/active.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 3.5 KiB |
BIN
icons/inactive.png
Normal file
BIN
icons/inactive.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 2.6 KiB |
BIN
icons/unknown.png
Normal file
BIN
icons/unknown.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 2.7 KiB |
35
readme.md
Normal file
35
readme.md
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
# Docker-Tray
|
||||||
|
|
||||||
|
Docker-Tray is a C++ application to monitor docker.service and allows to start and stop the service from system tray.
|
||||||
|
|
||||||
|
## Execution
|
||||||
|
|
||||||
|
Running the application.
|
||||||
|
|
||||||
|
```bash
|
||||||
|
g++ docker-service-tray.cc -o docker-tray `pkg-config --cflags --libs gtk+-3.0 appindicator3-0.1` -lsystemd
|
||||||
|
./docker-tray
|
||||||
|
```
|
||||||
|
|
||||||
|
## Dependencies
|
||||||
|
```
|
||||||
|
gtk+-3.0
|
||||||
|
appindicator3-0.1
|
||||||
|
lsystemd
|
||||||
|
```
|
||||||
|
The application utalizes sd_bus for hooking service states.
|
||||||
|
|
||||||
|
|
||||||
|
## Contributing
|
||||||
|
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
|
||||||
|
|
||||||
|
To do:
|
||||||
|
* Implement make, build and packing
|
||||||
|
* Enable/Disable service from tray.
|
||||||
|
* Enable/disable application on startup
|
||||||
|
* Refactor the code, implement proper testing.
|
||||||
|
|
||||||
|
Please make sure to update tests as appropriate.
|
||||||
|
|
||||||
|
## License
|
||||||
|
[MIT](https://choosealicense.com/licenses/mit/)
|
||||||
371
tray.h
Normal file
371
tray.h
Normal file
@ -0,0 +1,371 @@
|
|||||||
|
//https://github.com/zserge/tray
|
||||||
|
#ifndef TRAY_H
|
||||||
|
#define TRAY_H
|
||||||
|
|
||||||
|
struct tray_menu;
|
||||||
|
|
||||||
|
struct tray {
|
||||||
|
char *icon;
|
||||||
|
struct tray_menu *menu;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct tray_menu {
|
||||||
|
char *text;
|
||||||
|
int disabled;
|
||||||
|
int checked;
|
||||||
|
|
||||||
|
void (*cb)(struct tray_menu *);
|
||||||
|
void *context;
|
||||||
|
|
||||||
|
struct tray_menu *submenu;
|
||||||
|
};
|
||||||
|
|
||||||
|
static void tray_update(struct tray *tray);
|
||||||
|
|
||||||
|
#if defined(TRAY_APPINDICATOR)
|
||||||
|
|
||||||
|
#include <gtk/gtk.h>
|
||||||
|
#include <libappindicator/app-indicator.h>
|
||||||
|
|
||||||
|
#define TRAY_APPINDICATOR_ID "tray-id"
|
||||||
|
|
||||||
|
static AppIndicator *indicator = NULL;
|
||||||
|
static int loop_result = 0;
|
||||||
|
|
||||||
|
static void _tray_menu_cb(GtkMenuItem *item, gpointer data) {
|
||||||
|
(void)item;
|
||||||
|
struct tray_menu *m = (struct tray_menu *)data;
|
||||||
|
m->cb(m);
|
||||||
|
}
|
||||||
|
|
||||||
|
static GtkMenuShell *_tray_menu(struct tray_menu *m) {
|
||||||
|
GtkMenuShell *menu = (GtkMenuShell *)gtk_menu_new();
|
||||||
|
for (; m != NULL && m->text != NULL; m++) {
|
||||||
|
GtkWidget *item;
|
||||||
|
if (strcmp(m->text, "-") == 0) {
|
||||||
|
item = gtk_separator_menu_item_new();
|
||||||
|
} else {
|
||||||
|
if (m->submenu != NULL) {
|
||||||
|
item = gtk_menu_item_new_with_label(m->text);
|
||||||
|
gtk_menu_item_set_submenu(GTK_MENU_ITEM(item),
|
||||||
|
GTK_WIDGET(_tray_menu(m->submenu)));
|
||||||
|
} else {
|
||||||
|
item = gtk_check_menu_item_new_with_label(m->text);
|
||||||
|
gtk_check_menu_item_set_active(GTK_CHECK_MENU_ITEM(item), !!m->checked);
|
||||||
|
}
|
||||||
|
gtk_widget_set_sensitive(item, !m->disabled);
|
||||||
|
if (m->cb != NULL) {
|
||||||
|
g_signal_connect(item, "activate", G_CALLBACK(_tray_menu_cb), m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
gtk_widget_show(item);
|
||||||
|
gtk_menu_shell_append(menu, item);
|
||||||
|
}
|
||||||
|
return menu;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tray_init(struct tray *tray) {
|
||||||
|
if (gtk_init_check(0, NULL) == FALSE) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
indicator = app_indicator_new(TRAY_APPINDICATOR_ID, tray->icon,
|
||||||
|
APP_INDICATOR_CATEGORY_APPLICATION_STATUS);
|
||||||
|
app_indicator_set_status(indicator, APP_INDICATOR_STATUS_ACTIVE);
|
||||||
|
tray_update(tray);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tray_loop(int blocking) {
|
||||||
|
gtk_main_iteration_do(blocking);
|
||||||
|
return loop_result;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tray_update(struct tray *tray) {
|
||||||
|
app_indicator_set_icon(indicator, tray->icon);
|
||||||
|
// GTK is all about reference counting, so previous menu should be destroyed
|
||||||
|
// here
|
||||||
|
app_indicator_set_menu(indicator, GTK_MENU(_tray_menu(tray->menu)));
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tray_exit() { loop_result = -1; }
|
||||||
|
|
||||||
|
#elif defined(TRAY_APPKIT)
|
||||||
|
|
||||||
|
#include <objc/objc-runtime.h>
|
||||||
|
#include <limits.h>
|
||||||
|
|
||||||
|
static id app;
|
||||||
|
static id pool;
|
||||||
|
static id statusBar;
|
||||||
|
static id statusItem;
|
||||||
|
static id statusBarButton;
|
||||||
|
|
||||||
|
static id _tray_menu(struct tray_menu *m) {
|
||||||
|
id menu = objc_msgSend((id)objc_getClass("NSMenu"), sel_registerName("new"));
|
||||||
|
objc_msgSend(menu, sel_registerName("autorelease"));
|
||||||
|
objc_msgSend(menu, sel_registerName("setAutoenablesItems:"), false);
|
||||||
|
|
||||||
|
for (; m != NULL && m->text != NULL; m++) {
|
||||||
|
if (strcmp(m->text, "-") == 0) {
|
||||||
|
objc_msgSend(menu, sel_registerName("addItem:"),
|
||||||
|
objc_msgSend((id)objc_getClass("NSMenuItem"), sel_registerName("separatorItem")));
|
||||||
|
} else {
|
||||||
|
id menuItem = objc_msgSend((id)objc_getClass("NSMenuItem"), sel_registerName("alloc"));
|
||||||
|
objc_msgSend(menuItem, sel_registerName("autorelease"));
|
||||||
|
objc_msgSend(menuItem, sel_registerName("initWithTitle:action:keyEquivalent:"),
|
||||||
|
objc_msgSend((id)objc_getClass("NSString"), sel_registerName("stringWithUTF8String:"), m->text),
|
||||||
|
sel_registerName("menuCallback:"),
|
||||||
|
objc_msgSend((id)objc_getClass("NSString"), sel_registerName("stringWithUTF8String:"), ""));
|
||||||
|
|
||||||
|
objc_msgSend(menuItem, sel_registerName("setEnabled:"), (m->disabled ? false : true));
|
||||||
|
objc_msgSend(menuItem, sel_registerName("setState:"), (m->checked ? 1 : 0));
|
||||||
|
objc_msgSend(menuItem, sel_registerName("setRepresentedObject:"),
|
||||||
|
objc_msgSend((id)objc_getClass("NSValue"), sel_registerName("valueWithPointer:"), m));
|
||||||
|
|
||||||
|
objc_msgSend(menu, sel_registerName("addItem:"), menuItem);
|
||||||
|
|
||||||
|
if (m->submenu != NULL) {
|
||||||
|
objc_msgSend(menu, sel_registerName("setSubmenu:forItem:"), _tray_menu(m->submenu), menuItem);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return menu;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void menu_callback(id self, SEL cmd, id sender) {
|
||||||
|
struct tray_menu *m =
|
||||||
|
(struct tray_menu *)objc_msgSend(objc_msgSend(sender, sel_registerName("representedObject")),
|
||||||
|
sel_registerName("pointerValue"));
|
||||||
|
|
||||||
|
if (m != NULL && m->cb != NULL) {
|
||||||
|
m->cb(m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tray_init(struct tray *tray) {
|
||||||
|
pool = objc_msgSend((id)objc_getClass("NSAutoreleasePool"),
|
||||||
|
sel_registerName("new"));
|
||||||
|
|
||||||
|
objc_msgSend((id)objc_getClass("NSApplication"),
|
||||||
|
sel_registerName("sharedApplication"));
|
||||||
|
|
||||||
|
Class trayDelegateClass = objc_allocateClassPair(objc_getClass("NSObject"), "Tray", 0);
|
||||||
|
class_addProtocol(trayDelegateClass, objc_getProtocol("NSApplicationDelegate"));
|
||||||
|
class_addMethod(trayDelegateClass, sel_registerName("menuCallback:"), (IMP)menu_callback, "v@:@");
|
||||||
|
objc_registerClassPair(trayDelegateClass);
|
||||||
|
|
||||||
|
id trayDelegate = objc_msgSend((id)trayDelegateClass,
|
||||||
|
sel_registerName("new"));
|
||||||
|
|
||||||
|
app = objc_msgSend((id)objc_getClass("NSApplication"),
|
||||||
|
sel_registerName("sharedApplication"));
|
||||||
|
|
||||||
|
objc_msgSend(app, sel_registerName("setDelegate:"), trayDelegate);
|
||||||
|
|
||||||
|
statusBar = objc_msgSend((id)objc_getClass("NSStatusBar"),
|
||||||
|
sel_registerName("systemStatusBar"));
|
||||||
|
|
||||||
|
statusItem = objc_msgSend(statusBar, sel_registerName("statusItemWithLength:"), -1.0);
|
||||||
|
|
||||||
|
objc_msgSend(statusItem, sel_registerName("retain"));
|
||||||
|
objc_msgSend(statusItem, sel_registerName("setHighlightMode:"), true);
|
||||||
|
statusBarButton = objc_msgSend(statusItem, sel_registerName("button"));
|
||||||
|
tray_update(tray);
|
||||||
|
objc_msgSend(app, sel_registerName("activateIgnoringOtherApps:"), true);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tray_loop(int blocking) {
|
||||||
|
id until = (blocking ?
|
||||||
|
objc_msgSend((id)objc_getClass("NSDate"), sel_registerName("distantFuture")) :
|
||||||
|
objc_msgSend((id)objc_getClass("NSDate"), sel_registerName("distantPast")));
|
||||||
|
|
||||||
|
id event = objc_msgSend(app, sel_registerName("nextEventMatchingMask:untilDate:inMode:dequeue:"),
|
||||||
|
ULONG_MAX,
|
||||||
|
until,
|
||||||
|
objc_msgSend((id)objc_getClass("NSString"),
|
||||||
|
sel_registerName("stringWithUTF8String:"),
|
||||||
|
"kCFRunLoopDefaultMode"),
|
||||||
|
true);
|
||||||
|
if (event) {
|
||||||
|
objc_msgSend(app, sel_registerName("sendEvent:"), event);
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tray_update(struct tray *tray) {
|
||||||
|
objc_msgSend(statusBarButton, sel_registerName("setImage:"),
|
||||||
|
objc_msgSend((id)objc_getClass("NSImage"), sel_registerName("imageNamed:"),
|
||||||
|
objc_msgSend((id)objc_getClass("NSString"), sel_registerName("stringWithUTF8String:"), tray->icon)));
|
||||||
|
|
||||||
|
objc_msgSend(statusItem, sel_registerName("setMenu:"), _tray_menu(tray->menu));
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tray_exit() { objc_msgSend(app, sel_registerName("terminate:"), app); }
|
||||||
|
|
||||||
|
#elif defined(TRAY_WINAPI)
|
||||||
|
#include <windows.h>
|
||||||
|
|
||||||
|
#include <shellapi.h>
|
||||||
|
|
||||||
|
#define WM_TRAY_CALLBACK_MESSAGE (WM_USER + 1)
|
||||||
|
#define WC_TRAY_CLASS_NAME "TRAY"
|
||||||
|
#define ID_TRAY_FIRST 1000
|
||||||
|
|
||||||
|
static WNDCLASSEX wc;
|
||||||
|
static NOTIFYICONDATA nid;
|
||||||
|
static HWND hwnd;
|
||||||
|
static HMENU hmenu = NULL;
|
||||||
|
|
||||||
|
static LRESULT CALLBACK _tray_wnd_proc(HWND hwnd, UINT msg, WPARAM wparam,
|
||||||
|
LPARAM lparam) {
|
||||||
|
switch (msg) {
|
||||||
|
case WM_CLOSE:
|
||||||
|
DestroyWindow(hwnd);
|
||||||
|
return 0;
|
||||||
|
case WM_DESTROY:
|
||||||
|
PostQuitMessage(0);
|
||||||
|
return 0;
|
||||||
|
case WM_TRAY_CALLBACK_MESSAGE:
|
||||||
|
if (lparam == WM_LBUTTONUP || lparam == WM_RBUTTONUP) {
|
||||||
|
POINT p;
|
||||||
|
GetCursorPos(&p);
|
||||||
|
SetForegroundWindow(hwnd);
|
||||||
|
WORD cmd = TrackPopupMenu(hmenu, TPM_LEFTALIGN | TPM_RIGHTBUTTON |
|
||||||
|
TPM_RETURNCMD | TPM_NONOTIFY,
|
||||||
|
p.x, p.y, 0, hwnd, NULL);
|
||||||
|
SendMessage(hwnd, WM_COMMAND, cmd, 0);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case WM_COMMAND:
|
||||||
|
if (wparam >= ID_TRAY_FIRST) {
|
||||||
|
MENUITEMINFO item = {
|
||||||
|
.cbSize = sizeof(MENUITEMINFO), .fMask = MIIM_ID | MIIM_DATA,
|
||||||
|
};
|
||||||
|
if (GetMenuItemInfo(hmenu, wparam, FALSE, &item)) {
|
||||||
|
struct tray_menu *menu = (struct tray_menu *)item.dwItemData;
|
||||||
|
if (menu != NULL && menu->cb != NULL) {
|
||||||
|
menu->cb(menu);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return DefWindowProc(hwnd, msg, wparam, lparam);
|
||||||
|
}
|
||||||
|
|
||||||
|
static HMENU _tray_menu(struct tray_menu *m, UINT *id) {
|
||||||
|
HMENU hmenu = CreatePopupMenu();
|
||||||
|
for (; m != NULL && m->text != NULL; m++, (*id)++) {
|
||||||
|
if (strcmp(m->text, "-") == 0) {
|
||||||
|
InsertMenu(hmenu, *id, MF_SEPARATOR, TRUE, "");
|
||||||
|
} else {
|
||||||
|
MENUITEMINFO item;
|
||||||
|
memset(&item, 0, sizeof(item));
|
||||||
|
item.cbSize = sizeof(MENUITEMINFO);
|
||||||
|
item.fMask = MIIM_ID | MIIM_TYPE | MIIM_STATE | MIIM_DATA;
|
||||||
|
item.fType = 0;
|
||||||
|
item.fState = 0;
|
||||||
|
if (m->submenu != NULL) {
|
||||||
|
item.fMask = item.fMask | MIIM_SUBMENU;
|
||||||
|
item.hSubMenu = _tray_menu(m->submenu, id);
|
||||||
|
}
|
||||||
|
if (m->disabled) {
|
||||||
|
item.fState |= MFS_DISABLED;
|
||||||
|
}
|
||||||
|
if (m->checked) {
|
||||||
|
item.fState |= MFS_CHECKED;
|
||||||
|
}
|
||||||
|
item.wID = *id;
|
||||||
|
item.dwTypeData = m->text;
|
||||||
|
item.dwItemData = (ULONG_PTR)m;
|
||||||
|
|
||||||
|
InsertMenuItem(hmenu, *id, TRUE, &item);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return hmenu;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tray_init(struct tray *tray) {
|
||||||
|
memset(&wc, 0, sizeof(wc));
|
||||||
|
wc.cbSize = sizeof(WNDCLASSEX);
|
||||||
|
wc.lpfnWndProc = _tray_wnd_proc;
|
||||||
|
wc.hInstance = GetModuleHandle(NULL);
|
||||||
|
wc.lpszClassName = WC_TRAY_CLASS_NAME;
|
||||||
|
if (!RegisterClassEx(&wc)) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
hwnd = CreateWindowEx(0, WC_TRAY_CLASS_NAME, NULL, 0, 0, 0, 0, 0, 0, 0, 0, 0);
|
||||||
|
if (hwnd == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
UpdateWindow(hwnd);
|
||||||
|
|
||||||
|
memset(&nid, 0, sizeof(nid));
|
||||||
|
nid.cbSize = sizeof(NOTIFYICONDATA);
|
||||||
|
nid.hWnd = hwnd;
|
||||||
|
nid.uID = 0;
|
||||||
|
nid.uFlags = NIF_ICON | NIF_MESSAGE;
|
||||||
|
nid.uCallbackMessage = WM_TRAY_CALLBACK_MESSAGE;
|
||||||
|
Shell_NotifyIcon(NIM_ADD, &nid);
|
||||||
|
|
||||||
|
tray_update(tray);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int tray_loop(int blocking) {
|
||||||
|
MSG msg;
|
||||||
|
if (blocking) {
|
||||||
|
GetMessage(&msg, NULL, 0, 0);
|
||||||
|
} else {
|
||||||
|
PeekMessage(&msg, NULL, 0, 0, PM_REMOVE);
|
||||||
|
}
|
||||||
|
if (msg.message == WM_QUIT) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
TranslateMessage(&msg);
|
||||||
|
DispatchMessage(&msg);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tray_update(struct tray *tray) {
|
||||||
|
HMENU prevmenu = hmenu;
|
||||||
|
UINT id = ID_TRAY_FIRST;
|
||||||
|
hmenu = _tray_menu(tray->menu, &id);
|
||||||
|
SendMessage(hwnd, WM_INITMENUPOPUP, (WPARAM)hmenu, 0);
|
||||||
|
HICON icon;
|
||||||
|
ExtractIconEx(tray->icon, 0, NULL, &icon, 1);
|
||||||
|
if (nid.hIcon) {
|
||||||
|
DestroyIcon(nid.hIcon);
|
||||||
|
}
|
||||||
|
nid.hIcon = icon;
|
||||||
|
Shell_NotifyIcon(NIM_MODIFY, &nid);
|
||||||
|
|
||||||
|
if (prevmenu != NULL) {
|
||||||
|
DestroyMenu(prevmenu);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void tray_exit() {
|
||||||
|
Shell_NotifyIcon(NIM_DELETE, &nid);
|
||||||
|
if (nid.hIcon != 0) {
|
||||||
|
DestroyIcon(nid.hIcon);
|
||||||
|
}
|
||||||
|
if (hmenu != 0) {
|
||||||
|
DestroyMenu(hmenu);
|
||||||
|
}
|
||||||
|
PostQuitMessage(0);
|
||||||
|
UnregisterClass(WC_TRAY_CLASS_NAME, GetModuleHandle(NULL));
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
static int tray_init(struct tray *tray) { return -1; }
|
||||||
|
static int tray_loop(int blocking) { return -1; }
|
||||||
|
static void tray_update(struct tray *tray) {}
|
||||||
|
static void tray_exit();
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /* TRAY_H */
|
||||||
Loading…
Reference in New Issue
Block a user