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