git.oblomov.eu Git - wine/blob - dlls/shell32/systray.c

   1 /*
   2  *      Systray
   3  *
   4  *      Copyright 1999 Kai Morich       <kai.morich@bigfoot.de>
   5  *
   6  *  Manage the systray window. That it actually appears in the docking
   7  *  area of KDE or GNOME is delegated to windows/x11drv/wnd.c,
   8  *  X11DRV_WND_DockWindow.
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Lesser General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2.1 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public
  21  * License along with this library; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  23  */
  24
  25 #include "config.h"
  26
  27 #ifdef HAVE_UNISTD_H
  28 # include <unistd.h>
  29 #endif
  30 #include <stdlib.h>
  31 #include <string.h>
  32
  33 #include "winbase.h"
  34 #include "winnls.h"
  35 #include "shlobj.h"
  36 #include "shellapi.h"
  37 #include "shell32_main.h"
  38 #include "commctrl.h"
  39 #include "wine/debug.h"
  40
  41 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  42
  43 typedef struct SystrayItem {
  44   HWND                  hWnd;
  45   HWND                  hWndToolTip;
  46   NOTIFYICONDATAA       notifyIcon;
  47   struct SystrayItem    *nextTrayItem;
  48 } SystrayItem;
  49
  50 static SystrayItem *systray=NULL;
  51 static int firstSystray=TRUE; /* defer creation of window class until first systray item is created */
  52
  53 static BOOL SYSTRAY_Delete(PNOTIFYICONDATAA pnid);
  54
  55
  56 #define ICON_SIZE GetSystemMetrics(SM_CXSMICON)
  57 /* space around icon (forces icon to center of KDE systray area) */
  58 #define ICON_BORDER  4
  59
  60
  61
  62 static BOOL SYSTRAY_ItemIsEqual(PNOTIFYICONDATAA pnid1, PNOTIFYICONDATAA pnid2)
  63 {
  64   if (pnid1->hWnd != pnid2->hWnd) return FALSE;
  65   if (pnid1->uID  != pnid2->uID)  return FALSE;
  66   return TRUE;
  67 }
  68
  69 static LRESULT CALLBACK SYSTRAY_WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
  70 {
  71   HDC hdc;
  72   PAINTSTRUCT ps;
  73
  74   switch (message) {
  75   case WM_PAINT:
  76   {
  77     RECT rc;
  78     SystrayItem  *ptrayItem = systray;
  79
  80     while (ptrayItem) {
  81       if (ptrayItem->hWnd==hWnd) {
  82         if (ptrayItem->notifyIcon.hIcon) {
  83           hdc = BeginPaint(hWnd, &ps);
  84           GetClientRect(hWnd, &rc);
  85           if (!DrawIconEx(hdc, rc.left+ICON_BORDER, rc.top+ICON_BORDER, ptrayItem->notifyIcon.hIcon,
  86                           ICON_SIZE, ICON_SIZE, 0, 0, DI_DEFAULTSIZE|DI_NORMAL)) {
  87             ERR("Paint(SystrayWindow %p) failed -> removing SystrayItem %p\n", hWnd, ptrayItem);
  88             SYSTRAY_Delete(&ptrayItem->notifyIcon);
  89           }
  90         }
  91         break;
  92       }
  93       ptrayItem = ptrayItem->nextTrayItem;
  94     }
  95     EndPaint(hWnd, &ps);
  96   }
  97   break;
  98
  99   case WM_MOUSEMOVE:
 100   case WM_LBUTTONDOWN:
 101   case WM_LBUTTONUP:
 102   case WM_RBUTTONDOWN:
 103   case WM_RBUTTONUP:
 104   case WM_MBUTTONDOWN:
 105   case WM_MBUTTONUP:
 106   {
 107     MSG msg;
 108     SystrayItem *ptrayItem = systray;
 109
 110     while ( ptrayItem ) {
 111       if (ptrayItem->hWnd == hWnd) {
 112         msg.hwnd=hWnd;
 113         msg.message=message;
 114         msg.wParam=wParam;
 115         msg.lParam=lParam;
 116         msg.time = GetMessageTime ();
 117         msg.pt.x = LOWORD(GetMessagePos ());
 118         msg.pt.y = HIWORD(GetMessagePos ());
 119
 120         SendMessageA(ptrayItem->hWndToolTip, TTM_RELAYEVENT, 0, (LPARAM)&msg);
 121       }
 122       ptrayItem = ptrayItem->nextTrayItem;
 123     }
 124   }
 125   /* fall through */
 126
 127   case WM_LBUTTONDBLCLK:
 128   case WM_RBUTTONDBLCLK:
 129   case WM_MBUTTONDBLCLK:
 130   {
 131     SystrayItem *ptrayItem = systray;
 132
 133     while (ptrayItem) {
 134       if (ptrayItem->hWnd == hWnd) {
 135         if (ptrayItem->notifyIcon.hWnd && ptrayItem->notifyIcon.uCallbackMessage) {
 136           if (!PostMessageA(ptrayItem->notifyIcon.hWnd, ptrayItem->notifyIcon.uCallbackMessage,
 137                             (WPARAM)ptrayItem->notifyIcon.uID, (LPARAM)message)) {
 138               ERR("PostMessage(SystrayWindow %p) failed -> removing SystrayItem %p\n", hWnd, ptrayItem);
 139               SYSTRAY_Delete(&ptrayItem->notifyIcon);
 140             }
 141         }
 142         break;
 143       }
 144       ptrayItem = ptrayItem->nextTrayItem;
 145     }
 146   }
 147   break;
 148
 149   default:
 150     return (DefWindowProcA(hWnd, message, wParam, lParam));
 151   }
 152   return (0);
 153
 154 }
 155
 156
 157 BOOL SYSTRAY_RegisterClass(void)
 158 {
 159   WNDCLASSA  wc;
 160
 161   wc.style         = CS_SAVEBITS|CS_DBLCLKS;
 162   wc.lpfnWndProc   = (WNDPROC)SYSTRAY_WndProc;
 163   wc.cbClsExtra    = 0;
 164   wc.cbWndExtra    = 0;
 165   wc.hInstance     = 0;
 166   wc.hIcon         = 0;
 167   wc.hCursor       = LoadCursorA(0, IDC_ARROWA);
 168   wc.hbrBackground = (HBRUSH)(COLOR_WINDOW);
 169   wc.lpszMenuName  = NULL;
 170   wc.lpszClassName = "WineSystray";
 171
 172   if (!RegisterClassA(&wc)) {
 173     ERR("RegisterClass(WineSystray) failed\n");
 174     return FALSE;
 175   }
 176   return TRUE;
 177 }
 178
 179
 180 BOOL SYSTRAY_ItemInit(SystrayItem *ptrayItem)
 181 {
 182   RECT rect;
 183
 184   /* Register the class if this is our first tray item. */
 185   if ( firstSystray ) {
 186     firstSystray = FALSE;
 187     if ( !SYSTRAY_RegisterClass() ) {
 188       ERR( "RegisterClass(WineSystray) failed\n" );
 189       return FALSE;
 190     }
 191   }
 192
 193   /* Initialize the window size. */
 194   rect.left   = 0;
 195   rect.top    = 0;
 196   rect.right  = ICON_SIZE+2*ICON_BORDER;
 197   rect.bottom = ICON_SIZE+2*ICON_BORDER;
 198
 199   ZeroMemory( ptrayItem, sizeof(SystrayItem) );
 200   /* Create tray window for icon. */
 201   ptrayItem->hWnd = CreateWindowExA( WS_EX_TRAYWINDOW,
 202                                 "WineSystray", "Wine-Systray",
 203                                 WS_VISIBLE,
 204                                 CW_USEDEFAULT, CW_USEDEFAULT,
 205                                 rect.right-rect.left, rect.bottom-rect.top,
 206                                 0, 0, 0, 0 );
 207   if ( !ptrayItem->hWnd ) {
 208     ERR( "CreateWindow(WineSystray) failed\n" );
 209     return FALSE;
 210   }
 211
 212   /* Create tooltip for icon. */
 213   ptrayItem->hWndToolTip = CreateWindowA( TOOLTIPS_CLASSA,NULL,TTS_ALWAYSTIP,
 214                                      CW_USEDEFAULT, CW_USEDEFAULT,
 215                                      CW_USEDEFAULT, CW_USEDEFAULT,
 216                                      ptrayItem->hWnd, 0, 0, 0 );
 217   if ( !ptrayItem->hWndToolTip ) {
 218     ERR( "CreateWindow(TOOLTIP) failed\n" );
 219     return FALSE;
 220   }
 221   return TRUE;
 222 }
 223
 224
 225 static void SYSTRAY_ItemTerm(SystrayItem *ptrayItem)
 226 {
 227   if(ptrayItem->notifyIcon.hIcon)
 228      DestroyIcon(ptrayItem->notifyIcon.hIcon);
 229   if(ptrayItem->hWndToolTip)
 230       DestroyWindow(ptrayItem->hWndToolTip);
 231   if(ptrayItem->hWnd)
 232     DestroyWindow(ptrayItem->hWnd);
 233   return;
 234 }
 235
 236
 237 void SYSTRAY_ItemSetMessage(SystrayItem *ptrayItem, UINT uCallbackMessage)
 238 {
 239   ptrayItem->notifyIcon.uCallbackMessage = uCallbackMessage;
 240 }
 241
 242
 243 void SYSTRAY_ItemSetIcon(SystrayItem *ptrayItem, HICON hIcon)
 244 {
 245   ptrayItem->notifyIcon.hIcon = CopyIcon(hIcon);
 246   InvalidateRect(ptrayItem->hWnd, NULL, TRUE);
 247 }
 248
 249
 250 void SYSTRAY_ItemSetTip(SystrayItem *ptrayItem, CHAR* szTip, int modify)
 251 {
 252   TTTOOLINFOA ti;
 253
 254   strncpy(ptrayItem->notifyIcon.szTip, szTip, sizeof(ptrayItem->notifyIcon.szTip));
 255   ptrayItem->notifyIcon.szTip[sizeof(ptrayItem->notifyIcon.szTip)-1]=0;
 256
 257   ti.cbSize = sizeof(TTTOOLINFOA);
 258   ti.uFlags = 0;
 259   ti.hwnd = ptrayItem->hWnd;
 260   ti.hinst = 0;
 261   ti.uId = 0;
 262   ti.lpszText = ptrayItem->notifyIcon.szTip;
 263   ti.rect.left   = 0;
 264   ti.rect.top    = 0;
 265   ti.rect.right  = ICON_SIZE+2*ICON_BORDER;
 266   ti.rect.bottom = ICON_SIZE+2*ICON_BORDER;
 267
 268   if(modify)
 269     SendMessageA(ptrayItem->hWndToolTip, TTM_UPDATETIPTEXTA, 0, (LPARAM)&ti);
 270   else
 271     SendMessageA(ptrayItem->hWndToolTip, TTM_ADDTOOLA, 0, (LPARAM)&ti);
 272 }
 273
 274
 275 static BOOL SYSTRAY_Add(PNOTIFYICONDATAA pnid)
 276 {
 277   SystrayItem **ptrayItem = &systray;
 278
 279   /* Find last element. */
 280   while( *ptrayItem ) {
 281     if ( SYSTRAY_ItemIsEqual(pnid, &(*ptrayItem)->notifyIcon) )
 282       return FALSE;
 283     ptrayItem = &((*ptrayItem)->nextTrayItem);
 284   }
 285   /* Allocate SystrayItem for element and add to end of list. */
 286   (*ptrayItem) = ( SystrayItem *)malloc( sizeof(SystrayItem) );
 287
 288   /* Initialize and set data for the tray element. */
 289   SYSTRAY_ItemInit( (*ptrayItem) );
 290   (*ptrayItem)->notifyIcon.uID = pnid->uID; /* only needed for callback message */
 291   (*ptrayItem)->notifyIcon.hWnd = pnid->hWnd; /* only needed for callback message */
 292   SYSTRAY_ItemSetIcon   (*ptrayItem, (pnid->uFlags&NIF_ICON)   ?pnid->hIcon           :0);
 293   SYSTRAY_ItemSetMessage(*ptrayItem, (pnid->uFlags&NIF_MESSAGE)?pnid->uCallbackMessage:0);
 294   SYSTRAY_ItemSetTip    (*ptrayItem, (pnid->uFlags&NIF_TIP)    ?pnid->szTip           :"", FALSE);
 295
 296   TRACE("%p: %p %s\n",  (*ptrayItem), (*ptrayItem)->notifyIcon.hWnd,
 297                                           (*ptrayItem)->notifyIcon.szTip);
 298   return TRUE;
 299 }
 300
 301
 302 static BOOL SYSTRAY_Modify(PNOTIFYICONDATAA pnid)
 303 {
 304   SystrayItem *ptrayItem = systray;
 305
 306   while ( ptrayItem ) {
 307     if ( SYSTRAY_ItemIsEqual(pnid, &ptrayItem->notifyIcon) ) {
 308       if (pnid->uFlags & NIF_ICON)
 309         SYSTRAY_ItemSetIcon(ptrayItem, pnid->hIcon);
 310       if (pnid->uFlags & NIF_MESSAGE)
 311         SYSTRAY_ItemSetMessage(ptrayItem, pnid->uCallbackMessage);
 312       if (pnid->uFlags & NIF_TIP)
 313         SYSTRAY_ItemSetTip(ptrayItem, pnid->szTip, TRUE);
 314
 315       TRACE("%p: %p %s\n", ptrayItem, ptrayItem->notifyIcon.hWnd, ptrayItem->notifyIcon.szTip);
 316       return TRUE;
 317     }
 318     ptrayItem = ptrayItem->nextTrayItem;
 319   }
 320   return FALSE; /* not found */
 321 }
 322
 323
 324 static BOOL SYSTRAY_Delete(PNOTIFYICONDATAA pnid)
 325 {
 326   SystrayItem **ptrayItem = &systray;
 327
 328   while (*ptrayItem) {
 329     if (SYSTRAY_ItemIsEqual(pnid, &(*ptrayItem)->notifyIcon)) {
 330       SystrayItem *next = (*ptrayItem)->nextTrayItem;
 331       TRACE("%p: %p %s\n", *ptrayItem, (*ptrayItem)->notifyIcon.hWnd, (*ptrayItem)->notifyIcon.szTip);
 332       SYSTRAY_ItemTerm(*ptrayItem);
 333
 334       free(*ptrayItem);
 335       *ptrayItem = next;
 336
 337       return TRUE;
 338     }
 339     ptrayItem = &((*ptrayItem)->nextTrayItem);
 340   }
 341
 342   return FALSE; /* not found */
 343 }
 344
 345 /*************************************************************************
 346  *
 347  */
 348 BOOL SYSTRAY_Init(void)
 349 {
 350   return TRUE;
 351 }
 352
 353 /*************************************************************************
 354  * Shell_NotifyIcon                     [SHELL32.296]
 355  * Shell_NotifyIconA                    [SHELL32.297]
 356  */
 357 BOOL WINAPI Shell_NotifyIconA(DWORD dwMessage, PNOTIFYICONDATAA pnid )
 358 {
 359   BOOL flag=FALSE;
 360   TRACE("enter %p %d %ld\n", pnid->hWnd, pnid->uID, dwMessage);
 361   switch(dwMessage) {
 362   case NIM_ADD:
 363     flag = SYSTRAY_Add(pnid);
 364     break;
 365   case NIM_MODIFY:
 366     flag = SYSTRAY_Modify(pnid);
 367     break;
 368   case NIM_DELETE:
 369     flag = SYSTRAY_Delete(pnid);
 370     break;
 371   }
 372   TRACE("leave %p %d %ld=%d\n", pnid->hWnd, pnid->uID, dwMessage, flag);
 373   return flag;
 374 }
 375
 376 /*************************************************************************
 377  * Shell_NotifyIconW                    [SHELL32.298]
 378  */
 379 BOOL WINAPI Shell_NotifyIconW (DWORD dwMessage, PNOTIFYICONDATAW pnid )
 380 {
 381         BOOL ret;
 382
 383         PNOTIFYICONDATAA p = HeapAlloc(GetProcessHeap(),0,sizeof(NOTIFYICONDATAA));
 384         memcpy(p, pnid, sizeof(NOTIFYICONDATAA));
 385         WideCharToMultiByte( CP_ACP, 0, pnid->szTip, -1, p->szTip, sizeof(p->szTip), NULL, NULL );
 386         p->szTip[sizeof(p->szTip)-1] = 0;
 387
 388         ret = Shell_NotifyIconA(dwMessage, p );
 389
 390         HeapFree(GetProcessHeap(),0,p);
 391         return ret;
 392 }