reactos/base/applications/kbswitch/kbswitch.c

   1 /*
   2  * PROJECT:         Keyboard Layout Switcher
   3  * FILE:            base\applications\kbswitch\kbswitch.c
   4  * PURPOSE:         Switching Keyboard Layouts
   5  * PROGRAMMERS:     Dmitry Chapyshev (dmitry@reactos.org)
   6  *                  Colin Finck (mail@colinfinck.de)
   7  */
   8
   9 #include "kbswitch.h"
  10
  11 #define WM_NOTIFYICONMSG (WM_USER + 248)
  12
  13 PROC KbSwitchSetHooks    = NULL;
  14 PROC KbSwitchDeleteHooks = NULL;
  15
  16
  17 static BOOL
  18 GetLayoutID(LPTSTR szLayoutNum, LPTSTR szLCID);
  19
  20 static BOOL
  21 GetLayoutName(LPTSTR szLayoutNum, LPTSTR szName);
  22
  23 HINSTANCE hInst;
  24 HANDLE    hProcessHeap;
  25 HMODULE   hDllLib;
  26 ULONG     ulCurrentLayoutNum = 1;
  27
  28 static HICON
  29 CreateTrayIcon(LPTSTR szLCID)
  30 {
  31     LANGID lId;
  32     TCHAR szBuf[3];
  33     HDC hdc, hdcsrc;
  34     HBITMAP hBitmap, hBmpNew, hBmpOld;
  35     RECT rect;
  36     DWORD bkColor, bkText;
  37     HFONT hFont = NULL;
  38     ICONINFO IconInfo;
  39     HICON hIcon = NULL;
  40
  41     lId = (LANGID)_tcstoul(szLCID, NULL, 16);
  42     if (GetLocaleInfo(lId,
  43                       LOCALE_SISO639LANGNAME,
  44                       szBuf,
  45                       sizeof(szBuf) / sizeof(TCHAR)) == 0)
  46     {
  47         lstrcpy(szBuf, _T("??\0"));
  48     }
  49
  50     hdcsrc = GetDC(NULL);
  51     hdc = CreateCompatibleDC(hdcsrc);
  52     hBitmap = CreateCompatibleBitmap(hdcsrc, 16, 16);
  53     ReleaseDC(NULL, hdcsrc);
  54
  55     if (hdc && hBitmap)
  56     {
  57         hBmpNew = CreateBitmap(16, 16, 1, 1, NULL);
  58         if (hBmpNew)
  59         {
  60             hBmpOld = SelectObject(hdc, hBitmap);
  61             rect.right = 16;
  62             rect.left = 0;
  63             rect.bottom = 16;
  64             rect.top = 0;
  65
  66             bkColor = SetBkColor(hdc, GetSysColor(COLOR_HIGHLIGHT));
  67             bkText  = SetTextColor(hdc, GetSysColor(COLOR_HIGHLIGHTTEXT));
  68
  69             ExtTextOut(hdc, rect.left, rect.top, ETO_OPAQUE, &rect, _T(""), 0, NULL);
  70
  71             hFont = CreateFont(-11, 0, 0, 0, FW_NORMAL, FALSE, FALSE, FALSE, ANSI_CHARSET,
  72                                OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS,
  73                                DEFAULT_QUALITY, FF_DONTCARE, _T("Tahoma"));
  74
  75             SelectObject(hdc, hFont);
  76             DrawText(hdc, _tcsupr(szBuf), 2, &rect, DT_SINGLELINE|DT_CENTER|DT_VCENTER);
  77             SelectObject(hdc, hBmpNew);
  78             PatBlt(hdc, 0, 0, 16, 16, BLACKNESS);
  79             SelectObject(hdc, hBmpOld);
  80
  81             IconInfo.hbmColor = hBitmap;
  82             IconInfo.hbmMask = hBmpNew;
  83             IconInfo.fIcon = TRUE;
  84
  85             hIcon = CreateIconIndirect(&IconInfo);
  86
  87             DeleteObject(hBmpNew);
  88             DeleteObject(hBmpOld);
  89             DeleteObject(hFont);
  90         }
  91     }
  92
  93     DeleteDC(hdc);
  94     DeleteObject(hBitmap);
  95
  96     return hIcon;
  97 }
  98
  99 static VOID
 100 AddTrayIcon(HWND hwnd)
 101 {
 102     NOTIFYICONDATA tnid;
 103     TCHAR szLCID[CCH_LAYOUT_ID + 1];
 104     TCHAR szName[MAX_PATH];
 105
 106     GetLayoutID(_T("1"), szLCID);
 107     GetLayoutName(_T("1"), szName);
 108
 109     tnid.cbSize = sizeof(NOTIFYICONDATA);
 110     tnid.hWnd = hwnd;
 111     tnid.uID = 1;
 112     tnid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
 113     tnid.uCallbackMessage = WM_NOTIFYICONMSG;
 114     tnid.hIcon = CreateTrayIcon(szLCID);
 115
 116     lstrcpyn(tnid.szTip, szName, sizeof(tnid.szTip));
 117
 118     Shell_NotifyIcon(NIM_ADD, &tnid);
 119 }
 120
 121 static VOID
 122 DelTrayIcon(HWND hwnd)
 123 {
 124     NOTIFYICONDATA tnid;
 125
 126     tnid.cbSize = sizeof(NOTIFYICONDATA);
 127     tnid.hWnd = hwnd;
 128     tnid.uID = 1;
 129
 130     Shell_NotifyIcon(NIM_DELETE, &tnid);
 131 }
 132
 133 static VOID
 134 UpdateTrayIcon(HWND hwnd, LPTSTR szLCID, LPTSTR szName)
 135 {
 136     NOTIFYICONDATA tnid;
 137
 138     tnid.cbSize = sizeof(NOTIFYICONDATA);
 139     tnid.hWnd = hwnd;
 140     tnid.uID = 1;
 141     tnid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
 142     tnid.uCallbackMessage = WM_NOTIFYICONMSG;
 143     tnid.hIcon = CreateTrayIcon(szLCID);
 144
 145     lstrcpyn(tnid.szTip, szName, sizeof(tnid.szTip));
 146
 147     Shell_NotifyIcon(NIM_MODIFY, &tnid);
 148 }
 149
 150 static BOOL
 151 GetLayoutID(LPTSTR szLayoutNum, LPTSTR szLCID)
 152 {
 153     DWORD dwBufLen;
 154     DWORD dwRes;
 155     HKEY hKey;
 156     TCHAR szTempLCID[CCH_LAYOUT_ID + 1];
 157
 158     // Get the Layout ID
 159     if (RegOpenKeyEx(HKEY_CURRENT_USER, _T("Keyboard Layout\\Preload"), 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
 160     {
 161         dwBufLen = sizeof(szTempLCID);
 162         dwRes = RegQueryValueEx(hKey, szLayoutNum, NULL, NULL, (LPBYTE)szTempLCID, &dwBufLen);
 163
 164         if (dwRes != ERROR_SUCCESS)
 165         {
 166             RegCloseKey(hKey);
 167             return FALSE;
 168         }
 169
 170         RegCloseKey(hKey);
 171     }
 172
 173     // Look for a substitude of this layout
 174     if (RegOpenKeyEx(HKEY_CURRENT_USER, _T("Keyboard Layout\\Substitutes"), 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
 175     {
 176         dwBufLen = sizeof(szTempLCID);
 177
 178         if (RegQueryValueEx(hKey, szTempLCID, NULL, NULL, (LPBYTE)szLCID, &dwBufLen) != ERROR_SUCCESS)
 179         {
 180             // No substitute found, then use the old LCID
 181             lstrcpy(szLCID, szTempLCID);
 182         }
 183
 184         RegCloseKey(hKey);
 185     }
 186     else
 187     {
 188         // Substitutes key couldn't be opened, so use the old LCID
 189         lstrcpy(szLCID, szTempLCID);
 190     }
 191
 192     return TRUE;
 193 }
 194
 195 VOID
 196 GetLayoutIDByHkl(HKL hKl, LPTSTR szLayoutID)
 197 {
 198     /*
 199         FIXME!!! This way of getting layout ID incorrect!
 200                  This will not work correctly for 0001040a, 00010410, etc
 201     */
 202     wsprintf(szLayoutID, _T("%08x"), LOWORD(hKl));
 203 }
 204
 205 static BOOL
 206 GetLayoutName(LPTSTR szLayoutNum, LPTSTR szName)
 207 {
 208     HKEY hKey;
 209     DWORD dwBufLen;
 210     TCHAR szBuf[MAX_PATH], szDispName[MAX_PATH], szIndex[MAX_PATH], szPath[MAX_PATH];
 211     TCHAR szLCID[CCH_LAYOUT_ID + 1];
 212     HANDLE hLib;
 213     UINT i, j, k;
 214
 215     if(!GetLayoutID(szLayoutNum, szLCID))
 216         return FALSE;
 217
 218     wsprintf(szBuf, _T("SYSTEM\\CurrentControlSet\\Control\\Keyboard Layouts\\%s"), szLCID);
 219
 220     if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, (LPCTSTR)szBuf, 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
 221     {
 222         dwBufLen = sizeof(szBuf);
 223
 224         if (RegQueryValueEx(hKey, _T("Layout Display Name"), NULL, NULL, (LPBYTE)szDispName, &dwBufLen) == ERROR_SUCCESS)
 225         {
 226             if (szDispName[0] == '@')
 227             {
 228                 for (i = 0; i < _tcslen(szDispName); i++)
 229                 {
 230                     if ((szDispName[i] == ',') && (szDispName[i + 1] == '-'))
 231                     {
 232                         for (j = i + 2, k = 0; j < _tcslen(szDispName)+1; j++, k++)
 233                         {
 234                             szIndex[k] = szDispName[j];
 235                         }
 236                         szDispName[i - 1] = '\0';
 237                         break;
 238                     }
 239                     else szDispName[i] = szDispName[i + 1];
 240                 }
 241
 242                 if (ExpandEnvironmentStrings(szDispName, szPath, MAX_PATH))
 243                 {
 244                     hLib = LoadLibrary(szPath);
 245                     if (hLib)
 246                     {
 247                         if (LoadString(hLib, _ttoi(szIndex), szPath, sizeof(szPath) / sizeof(TCHAR)) != 0)
 248                         {
 249                             _tcscpy(szName, szPath);
 250                             RegCloseKey(hKey);
 251                             return TRUE;
 252                         }
 253                         FreeLibrary(hLib);
 254                     }
 255                 }
 256             }
 257         }
 258
 259         dwBufLen = sizeof(szBuf);
 260
 261         if (RegQueryValueEx(hKey, _T("Layout Text"), NULL, NULL, (LPBYTE)szName, &dwBufLen) == ERROR_SUCCESS)
 262         {
 263             RegCloseKey(hKey);
 264             return TRUE;
 265         }
 266     }
 267
 268     return FALSE;
 269 }
 270
 271 BOOL CALLBACK
 272 EnumWindowsProc(HWND hwnd, LPARAM lParam)
 273 {
 274     PostMessage(hwnd, WM_INPUTLANGCHANGEREQUEST, 0, lParam);
 275     return TRUE;
 276 }
 277
 278 static VOID
 279 ActivateLayout(HWND hwnd, ULONG uLayoutNum)
 280 {
 281     HKL hKl;
 282     TCHAR szLayoutNum[CCH_ULONG_DEC + 1];
 283     TCHAR szLCID[CCH_LAYOUT_ID + 1];
 284     TCHAR szLangName[MAX_PATH];
 285
 286     _ultot(uLayoutNum, szLayoutNum, 10);
 287     GetLayoutID(szLayoutNum, szLCID);
 288     CreateTrayIcon(szLCID);
 289
 290     // Switch to the new keyboard layout
 291     GetLocaleInfo((LANGID)_tcstoul(szLCID, NULL, 16), LOCALE_SLANGUAGE, (LPTSTR)szLangName, sizeof(szLangName) / sizeof(TCHAR));
 292     UpdateTrayIcon(hwnd, szLCID, szLangName);
 293     hKl = LoadKeyboardLayout(szLCID, KLF_ACTIVATE);
 294
 295     EnumWindows(EnumWindowsProc, (LPARAM) hKl);
 296
 297     ulCurrentLayoutNum = uLayoutNum;
 298 }
 299
 300 static HMENU
 301 BuildLeftPopupMenu()
 302 {
 303     HMENU hMenu;
 304     HKEY hKey;
 305     DWORD dwIndex, dwSize;
 306     TCHAR szLayoutNum[CCH_ULONG_DEC + 1];
 307     TCHAR szName[MAX_PATH];
 308
 309     hMenu = CreatePopupMenu();
 310
 311     // Add the keyboard layouts to the popup menu
 312     if (RegOpenKeyEx(HKEY_CURRENT_USER, _T("Keyboard Layout\\Preload"), 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
 313     {
 314         for(dwIndex = 0; ; dwIndex++)
 315         {
 316             dwSize = sizeof(szLayoutNum);
 317             if(RegEnumValue(hKey, dwIndex, szLayoutNum, &dwSize, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
 318                 break;
 319
 320             if(!GetLayoutName(szLayoutNum, szName))
 321                 break;
 322
 323             AppendMenu(hMenu, MF_STRING, _ttoi(szLayoutNum), szName);
 324         }
 325
 326         (void)CheckMenuItem(hMenu, ulCurrentLayoutNum, MF_CHECKED);
 327
 328         RegCloseKey(hKey);
 329     }
 330
 331     return hMenu;
 332 }
 333
 334 BOOL
 335 SetHooks()
 336 {
 337     hDllLib = LoadLibrary(_T("kbsdll.dll"));
 338     if (!hDllLib) return FALSE;
 339
 340     KbSwitchSetHooks    = (PROC) GetProcAddress(hDllLib, MAKEINTRESOURCEA(1));
 341     KbSwitchDeleteHooks = (PROC) GetProcAddress(hDllLib, MAKEINTRESOURCEA(2));
 342
 343     if ((KbSwitchSetHooks == NULL)||(KbSwitchDeleteHooks == NULL))
 344         return FALSE;
 345
 346     return KbSwitchSetHooks();
 347 }
 348
 349 VOID
 350 DeleteHooks()
 351 {
 352     if (KbSwitchDeleteHooks) KbSwitchDeleteHooks();
 353     if (hDllLib) FreeLibrary(hDllLib);
 354 }
 355
 356 ULONG
 357 GetNextLayout()
 358 {
 359     TCHAR szLayoutNum[3 + 1], szLayoutID[CCH_LAYOUT_ID + 1];
 360     ULONG Ret = ulCurrentLayoutNum;
 361
 362     _ultot(ulCurrentLayoutNum, szLayoutNum, 10);
 363     if (!GetLayoutID(szLayoutNum, szLayoutID))
 364     {
 365         return -1;
 366     }
 367
 368     _ultot(Ret + 1, szLayoutNum, 10);
 369
 370     if (GetLayoutID(szLayoutNum, szLayoutID))
 371     {
 372         return (Ret + 1);
 373     }
 374     else
 375     {
 376         _ultot(Ret - 1, szLayoutNum, 10);
 377         if (GetLayoutID(szLayoutNum, szLayoutID))
 378             return (Ret - 1);
 379         else
 380             return -1;
 381     }
 382
 383     return -1;
 384 }
 385
 386 LRESULT CALLBACK
 387 WndProc(HWND hwnd, UINT Message, WPARAM wParam, LPARAM lParam)
 388 {
 389     static HMENU hRightPopupMenu;
 390     static TCHAR szLCID[MAX_PATH], szLangName[MAX_PATH];
 391
 392     switch (Message)
 393     {
 394         case WM_CREATE:
 395         {
 396             SetHooks();
 397             AddTrayIcon(hwnd);
 398             hRightPopupMenu = GetSubMenu(LoadMenu(hInst, MAKEINTRESOURCE(IDR_POPUP)), 0);
 399
 400             ActivateLayout(hwnd, ulCurrentLayoutNum);
 401
 402             return 0;
 403         }
 404
 405         case WM_LANG_CHANGED:
 406         {
 407             GetLayoutIDByHkl((HKL)lParam, szLCID);
 408             GetLocaleInfo((LANGID)_tcstoul(szLCID, NULL, 16), LOCALE_SLANGUAGE, (LPTSTR)szLangName, sizeof(szLangName) / sizeof(TCHAR));
 409             UpdateTrayIcon(hwnd, szLCID, szLangName);
 410
 411             return 0;
 412         }
 413
 414         case WM_LOAD_LAYOUT:
 415         {
 416             ActivateLayout(hwnd, GetNextLayout());
 417
 418             return 0;
 419         }
 420
 421         case WM_WINDOW_ACTIVATE:
 422         {
 423             GetLayoutIDByHkl(GetKeyboardLayout(GetWindowThreadProcessId((HWND)wParam, 0)), szLCID);
 424             GetLocaleInfo((LANGID)_tcstoul(szLCID, NULL, 16), LOCALE_SLANGUAGE, (LPTSTR)szLangName, sizeof(szLangName) / sizeof(TCHAR));
 425             UpdateTrayIcon(hwnd, szLCID, szLangName);
 426
 427             return 0;
 428         }
 429
 430         case WM_NOTIFYICONMSG:
 431             switch (lParam)
 432             {
 433                 case WM_RBUTTONDOWN:
 434                 case WM_LBUTTONDOWN:
 435                 {
 436                     POINT pt;
 437
 438                     GetCursorPos(&pt);
 439                     SetForegroundWindow(hwnd);
 440
 441                     if (lParam == WM_LBUTTONDOWN)
 442                     {
 443                         HMENU hLeftPopupMenu;
 444
 445                         /* Rebuild the left popup menu on every click to take care of keyboard layout changes */
 446                         hLeftPopupMenu = BuildLeftPopupMenu();
 447                         TrackPopupMenu(hLeftPopupMenu, 0, pt.x, pt.y, 0, hwnd, NULL);
 448                         DestroyMenu(hLeftPopupMenu);
 449                     }
 450                     else
 451                     {
 452                         TrackPopupMenu(hRightPopupMenu, 0, pt.x, pt.y, 0, hwnd, NULL);
 453                     }
 454
 455                     PostMessage(hwnd, WM_NULL, 0, 0);
 456
 457                     return 0;
 458                 }
 459             }
 460             break;
 461
 462         case WM_COMMAND:
 463             switch (LOWORD(wParam))
 464             {
 465                 case ID_EXIT:
 466                     SendMessage(hwnd, WM_CLOSE, 0, 0);
 467                     return 0;
 468
 469                 case ID_PREFERENCES:
 470                 {
 471                     SHELLEXECUTEINFO shInputDll = {0};
 472
 473                     shInputDll.cbSize = sizeof(shInputDll);
 474                     shInputDll.hwnd = hwnd;
 475                     shInputDll.lpVerb = _T("open");
 476                     shInputDll.lpFile = _T("rundll32.exe");
 477                     shInputDll.lpParameters = _T("shell32.dll,Control_RunDLL input.dll");
 478
 479                     if (!ShellExecuteEx(&shInputDll))
 480                         MessageBox(hwnd, _T("Can't start input.dll"), NULL, MB_OK | MB_ICONERROR);
 481                 }
 482
 483                 default:
 484                     ActivateLayout(hwnd, LOWORD(wParam));
 485                     return 0;
 486             }
 487             break;
 488
 489         case WM_SETTINGCHANGE:
 490         {
 491             if (wParam == SPI_SETDEFAULTINPUTLANG)
 492             {
 493                 //FIXME: Should detect default language changes by CPL applet or by other tools and update UI
 494             }
 495         }
 496         break;
 497
 498         case WM_DESTROY:
 499         {
 500             DeleteHooks();
 501             DestroyMenu(hRightPopupMenu);
 502             DelTrayIcon(hwnd);
 503             PostQuitMessage(0);
 504
 505             return 0;
 506         }
 507     }
 508
 509     return DefWindowProc(hwnd, Message, wParam, lParam);
 510 }
 511
 512 INT WINAPI
 513 _tWinMain(HINSTANCE hInstance, HINSTANCE hPrevInst, LPTSTR lpCmdLine, INT nCmdShow)
 514 {
 515     WNDCLASS WndClass = {0};
 516     MSG msg;
 517     HANDLE hMutex;
 518
 519     hMutex = CreateMutex(NULL, FALSE, szKbSwitcherName);
 520     if ((!hMutex) || (GetLastError() == ERROR_ALREADY_EXISTS))
 521         return 1;
 522
 523     hInst = hInstance;
 524     hProcessHeap = GetProcessHeap();
 525
 526     WndClass.style = 0;
 527     WndClass.lpfnWndProc   = (WNDPROC)WndProc;
 528     WndClass.cbClsExtra    = 0;
 529     WndClass.cbWndExtra    = 0;
 530     WndClass.hInstance     = hInstance;
 531     WndClass.hIcon         = NULL;
 532     WndClass.hCursor       = NULL;
 533     WndClass.hbrBackground = NULL;
 534     WndClass.lpszMenuName  = NULL;
 535     WndClass.lpszClassName = szKbSwitcherName;
 536
 537     if (!RegisterClass(&WndClass))
 538         return 1;
 539
 540     CreateWindow(szKbSwitcherName, NULL, 0, 0, 0, 1, 1, HWND_DESKTOP, NULL, hInstance, NULL);
 541
 542     while(GetMessage(&msg,NULL,0,0))
 543     {
 544         TranslateMessage(&msg);
 545         DispatchMessage(&msg);
 546     }
 547
 548     if (hMutex) CloseHandle(hMutex);
 549
 550     return 0;
 551 }