tray icon added

cnucky · Mar 7, 2016 · 607336d · 607336d
1 parent 6383fc3
commit 607336d
Show file tree

Hide file tree

Showing 2 changed files with 92 additions and 6 deletions.
diff --git a/Win32/Win32App.cpp b/Win32/Win32App.cpp
@@ -1,19 +1,107 @@
 #include <string.h>
+#include <windows.h>
+#include <shellapi.h>
 #include "Win32App.h"
 
+#define ID_ABOUT 2000
+#define ID_EXIT 2001
+
+void ShowPopupMenu (HWND hWnd, POINT *curpos, int wDefaultItem)
+{
+    HMENU hPopup = CreatePopupMenu();
+    InsertMenu (hPopup, 0, MF_BYPOSITION | MF_STRING, ID_ABOUT, "About...");
+    InsertMenu (hPopup, 1, MF_BYPOSITION | MF_STRING, ID_EXIT , "Exit");
+    SetMenuDefaultItem (hPopup, ID_ABOUT, FALSE);
+    SetFocus (hWnd);
+    SendMessage (hWnd, WM_INITMENUPOPUP, (WPARAM)hPopup, 0);
+
+    POINT p;
+    if (!curpos)
+    {
+        GetCursorPos (&p);
+        curpos = &p;
+    }
+
+    WORD cmd = TrackPopupMenu (hPopup, TPM_LEFTALIGN | TPM_RIGHTBUTTON | TPM_RETURNCMD | TPM_NONOTIFY, curpos->x, curpos->y, 0, hWnd, NULL);
+    SendMessage (hWnd, WM_COMMAND, cmd, 0);
+
+    DestroyMenu(hPopup);
+}
+
+void AddTrayIcon (HWND hWnd, UINT uID, UINT uCallbackMsg, UINT uIcon)
+{
+    NOTIFYICONDATA nid;
+    nid.hWnd = hWnd;
+    nid.uID = uID;
+    nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
+    nid.uCallbackMessage = uCallbackMsg;
+    nid.hIcon = LoadIcon (GetModuleHandle(NULL), IDI_APPLICATION);
+    strcpy (nid.szTip, "i2pd");
+    Shell_NotifyIcon(NIM_ADD, &nid );
+}
+
+void RemoveTrayIcon (HWND hWnd, UINT uID)
+{
+    NOTIFYICONDATA nid;
+    nid.hWnd = hWnd;
+    nid.uID = uID;
+    Shell_NotifyIcon (NIM_DELETE, &nid);
+}
 
 static LRESULT CALLBACK WndProc (HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
 {
     switch (uMsg)
     {
+        case WM_CREATE:
+        {
+            AddTrayIcon (hWnd, 1, WM_APP, 0);
+            return 0;
+        }
+        case WM_CLOSE:
+        {
+            RemoveTrayIcon (hWnd, 1);
+            PostQuitMessage (0);
+            return DefWindowProc (hWnd, uMsg, wParam, lParam);
+        }
+        case WM_COMMAND:
+        {
+            switch (LOWORD(wParam))
+            {
+                case ID_ABOUT:
+                {
+                    MessageBox( hWnd, TEXT("i2pd"), TEXT("About"), MB_ICONINFORMATION | MB_OK );
+                    return 0;
+                }
+                case ID_EXIT:
+                {
+                    PostMessage (hWnd, WM_CLOSE, 0, 0);
+                    return 0;
+                }
+            }
+            break;
+        }
+        case WM_APP:
+        {
+            switch (lParam)
+            {
+                case WM_RBUTTONUP:
+                {
+                    SetForegroundWindow (hWnd);
+                    ShowPopupMenu(hWnd, NULL, -1);
+                    PostMessage (hWnd, WM_APP + 1, 0, 0);
+                    return 0;
+                }
+            }
+            break;
+        }
     }
     return DefWindowProc( hWnd, uMsg, wParam, lParam);
 }
 
  int WINAPI WinMain (HINSTANCE hInst, HINSTANCE prev, LPSTR cmdline, int show)
  {
     // check if tunning already
-    if (FindWindow (I2PD_WIN32_CLASSNAME, TEXT("Title")))
+    if (FindWindow (I2PD_WIN32_CLASSNAME, TEXT("i2pd")))
     {
         MessageBox(NULL, TEXT("I2Pd is running already"), TEXT("Warning"), MB_OK);
         return 0;
@@ -27,15 +115,15 @@ static LRESULT CALLBACK WndProc (HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lPa
     wclx.cbClsExtra = 0;
     wclx.cbWndExtra = 0;
     wclx.hInstance = hInst;
-    //wclx.hIcon = LoadIcon( hInstance, MAKEINTRESOURCE( IDI_TRAYICON ) );
-    //wclx.hIconSm = LoadSmallIcon( hInstance, IDI_TRAYICON );
+    wclx.hIcon = LoadIcon (hInst, IDI_APPLICATION);
+    wclx.hIconSm = LoadIcon (hInst, IDI_APPLICATION);
     wclx.hCursor = LoadCursor (NULL, IDC_ARROW);
     wclx.hbrBackground = (HBRUSH)(COLOR_BTNFACE + 1);
     wclx.lpszMenuName = NULL;
     wclx.lpszClassName = I2PD_WIN32_CLASSNAME;
     RegisterClassEx (&wclx);
     // create new window
-    if (!CreateWindow(I2PD_WIN32_CLASSNAME, TEXT("Title"), WS_OVERLAPPEDWINDOW | WS_VISIBLE, 100, 100, 250, 150, NULL, NULL, hInst, NULL))
+    if (!CreateWindow(I2PD_WIN32_CLASSNAME, TEXT("i2pd"), WS_OVERLAPPEDWINDOW | WS_VISIBLE, 100, 100, 250, 150, NULL, NULL, hInst, NULL))
     {
         MessageBox(NULL, "Failed to create main window", TEXT("Warning!"), MB_ICONERROR | MB_OK | MB_TOPMOST);
         return 1;

diff --git a/Win32/Win32App.h b/Win32/Win32App.h
@@ -1,8 +1,6 @@
 #ifndef WIN32APP_H__
 #define WIN32APP_H__
 
-#include <windows.h>
-
 #define I2PD_WIN32_CLASSNAME "i2pd main window"
 
 #endif // WIN32APP_H__