base/applications/downloader/main.c

   1 /* PROJECT:         ReactOS Downloader
   2  * LICENSE:         GPL - See COPYING in the top level directory
   3  * FILE:            base/applications/downloader/xml.c
   4  * PURPOSE:         Main program
   5  * PROGRAMMERS:     Maarten Bosma, Lester Kortenhoeven, Dmitry Chapyshev
   6  */
   7
   8 #include <windows.h>
   9 #include <commctrl.h>
  10 #include <richedit.h>
  11 #include <tchar.h>
  12 #include <stdio.h>
  13 #include <shlwapi.h>
  14 #include <shlobj.h>
  15 #include "resources.h"
  16 #include "structures.h"
  17
  18 HWND hwnd, hCategories, hApps, hDownloadButton, hUninstallButton, hUpdateButton, hHelpButton, hProfButton;
  19 HBITMAP hLogo, hUnderline;
  20 WCHAR* DescriptionHeadline = L"";
  21 WCHAR* DescriptionText = L"";
  22 WCHAR ApplicationText[700];
  23
  24 struct Category Root;
  25 struct Application* SelectedApplication;
  26
  27 INT_PTR CALLBACK DownloadProc (HWND, UINT, WPARAM, LPARAM);
  28 BOOL ProcessXML (const char* filename, struct Category* Root);
  29 VOID FreeTree (struct Category* Node);
  30 WCHAR Strings [STRING_COUNT][MAX_STRING_LENGHT];
  31
  32
  33 BOOL
  34 getUninstaller(WCHAR* RegName, WCHAR* Uninstaller) {
  35     HKEY hKey1;
  36     HKEY hKey2;
  37     DWORD Type = 0;
  38     DWORD Size = MAX_PATH;
  39     WCHAR Value[MAX_PATH];
  40     WCHAR KeyName[MAX_PATH];
  41     LONG i = 0;
  42
  43     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall",0,KEY_READ,&hKey1) == ERROR_SUCCESS) {
  44         while (RegEnumKeyExW(hKey1,i,KeyName,&Size,NULL,NULL,NULL,NULL) == ERROR_SUCCESS) {
  45             ++i;
  46             RegOpenKeyExW(hKey1,KeyName,0,KEY_READ,&hKey2);
  47             Size = MAX_PATH;
  48             if (RegQueryValueExW(hKey2,L"DisplayName",0,&Type,(LPBYTE)Value,&Size) == ERROR_SUCCESS) {
  49                 Size = MAX_PATH;
  50                 if (StrCmpW(Value,RegName) == 0) {
  51                     if (RegQueryValueExW(hKey2,L"UninstallString",0,&Type,(LPBYTE)Uninstaller,&Size) == ERROR_SUCCESS) {
  52                         RegCloseKey(hKey2);
  53                         RegCloseKey(hKey1);
  54                         return TRUE;
  55                     } else {
  56                         RegCloseKey(hKey2);
  57                         RegCloseKey(hKey1);
  58                         return FALSE;
  59                     }
  60                 }
  61             }
  62             RegCloseKey(hKey2);
  63             Size = MAX_PATH;
  64         }
  65         RegCloseKey(hKey1);
  66     }
  67     return FALSE;
  68 }
  69
  70 void
  71 ShowMessage (WCHAR* title, WCHAR* message)
  72 {
  73     DescriptionHeadline = title;
  74     DescriptionText = message;
  75     InvalidateRect(hwnd,NULL,TRUE);
  76     UpdateWindow(hwnd);
  77 }
  78
  79 void
  80 AddItems(HWND hwnd, struct Category* Category, struct Category* Parent)
  81 {
  82     TV_INSERTSTRUCTW Insert;
  83
  84     Insert.item.lParam = (LPARAM)Category;
  85     Insert.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;;
  86     Insert.item.pszText = Category->Name;
  87     Insert.item.cchTextMax = lstrlenW(Category->Name);
  88     Insert.item.iImage = Category->Icon;
  89     Insert.item.iSelectedImage = Category->Icon;
  90     Insert.hInsertAfter = TVI_LAST;
  91     Insert.hParent = Category->Parent ? Category->Parent->TreeviewItem : TVI_ROOT;
  92
  93     Category->TreeviewItem = (HTREEITEM)SendMessage(hwnd, TVM_INSERTITEM, 0, (LPARAM)&Insert);
  94
  95     if(Category->Next)
  96         AddItems (hwnd,Category->Next,Parent);
  97
  98     if(Category->Children)
  99         AddItems (hwnd,Category->Children,Category);
 100 }
 101
 102 void
 103 CategoryChoosen(HWND hwnd, struct Category* Category)
 104 {
 105     struct Application* CurrentApplication;
 106     TV_INSERTSTRUCTW Insert;
 107     WCHAR Uninstaller[200];
 108     SelectedApplication = NULL;
 109
 110     if(Category->Children && !Category->Apps)
 111         ShowMessage(Category->Name, Strings[IDS_CHOOSE_SUB]);
 112     else if(!Category->Children && Category->Apps)
 113         ShowMessage(Category->Name, Strings[IDS_CHOOSE_APP]);
 114     else if(Category->Children && Category->Apps)
 115         ShowMessage(Category->Name, Strings[IDS_CHOOSE_BOTH]);
 116     else
 117         ShowMessage(Category->Name, Strings[IDS_NO_APPS]);
 118
 119     (void)TreeView_DeleteItem(hwnd, TVI_ROOT);
 120     (void)TreeView_DeleteItem(hwnd, TVI_ROOT); // Delete twice to bypass bug in windows
 121
 122     Insert.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE;
 123     Insert.hInsertAfter = TVI_LAST;
 124     Insert.hParent = TVI_ROOT;
 125
 126     CurrentApplication = Category->Apps;
 127     while(CurrentApplication)
 128     {
 129         Insert.item.lParam = (LPARAM)CurrentApplication;
 130         Insert.item.pszText = CurrentApplication->Name;
 131         Insert.item.cchTextMax = lstrlenW(CurrentApplication->Name);
 132         Insert.item.iImage = 10;
 133         if(StrCmpW(CurrentApplication->RegName,L"")) {
 134             if(getUninstaller(CurrentApplication->RegName, Uninstaller))
 135                 Insert.item.iImage = 9;
 136         }
 137         SendMessage(hwnd, TVM_INSERTITEM, 0, (LPARAM)&Insert);
 138         CurrentApplication = CurrentApplication->Next;
 139     }
 140 }
 141
 142 BOOL CreateToolTip(HWND hwndTool, HWND hDlg, WCHAR* pText)
 143 {
 144     HWND hwndTip;
 145     TOOLINFO toolInfo;
 146
 147     if (!hwndTool || !hDlg || !pText)
 148         return FALSE;
 149
 150     hwndTip = CreateWindowExW(0, TOOLTIPS_CLASS, NULL,
 151                                    WS_POPUP |TTS_ALWAYSTIP | TTS_BALLOON,
 152                                    CW_USEDEFAULT, CW_USEDEFAULT,
 153                                    CW_USEDEFAULT, CW_USEDEFAULT,
 154                                    hDlg, NULL,
 155                                    GetModuleHandle(NULL), NULL);
 156     if (!hwndTip)
 157         return FALSE;
 158
 159     ZeroMemory(&toolInfo, sizeof(TOOLINFO));
 160     toolInfo.cbSize = sizeof(toolInfo);
 161     toolInfo.hwnd = hDlg;
 162     toolInfo.uFlags = TTF_IDISHWND | TTF_SUBCLASS;
 163     toolInfo.uId = (UINT_PTR)hwndTool;
 164     toolInfo.lpszText = pText;
 165     SendMessage(hwndTip, TTM_ADDTOOL, 0, (LPARAM)&toolInfo);
 166
 167     return TRUE;
 168 }
 169
 170 BOOL
 171 SetupControls (HWND hwnd)
 172 {
 173     TV_INSERTSTRUCTW Insert = {0};
 174     HIMAGELIST hImageList;
 175     HINSTANCE hInstance = GetModuleHandle(NULL);
 176     WCHAR Cats[MAX_STRING_LENGHT], Apps[MAX_STRING_LENGHT];
 177     WCHAR Tooltip1[MAX_STRING_LENGHT], Tooltip2[MAX_STRING_LENGHT], Tooltip3[MAX_STRING_LENGHT];
 178     char Buf[MAX_PATH];
 179
 180         // Getting downloader.xml path
 181     if(!GetSystemDirectoryA(Buf,sizeof(Buf))) return FALSE;
 182     strcat(Buf, "\\downloader.xml");
 183
 184     // Parse the XML file
 185     if (!ProcessXML(Buf, &Root))
 186         return FALSE;
 187
 188     LoadStringW(hInstance, IDS_CATS_TITLE, Cats, MAX_STRING_LENGHT);
 189     LoadStringW(hInstance, IDS_APPS_TITLE, Apps, MAX_STRING_LENGHT);
 190
 191     // Set up the controls
 192     hCategories = CreateWindowExW(0, WC_TREEVIEWW, Cats,
 193                                   WS_CHILD|WS_VISIBLE|WS_BORDER|TVS_HASLINES|TVS_LINESATROOT|TVS_HASBUTTONS|TVS_SHOWSELALWAYS,
 194                                   0, 0, 0, 0, hwnd, NULL, hInstance, NULL);
 195
 196     hApps = CreateWindowExW(0, WC_TREEVIEWW, Apps,
 197                             WS_CHILD|WS_VISIBLE|WS_BORDER|TVS_HASLINES|TVS_LINESATROOT|TVS_HASBUTTONS|TVS_SHOWSELALWAYS,
 198                             0, 0, 0, 0, hwnd, NULL, hInstance, NULL);
 199
 200     hLogo = LoadBitmap(GetModuleHandle(NULL),
 201                        MAKEINTRESOURCE(IDB_LOGO));
 202     hUnderline = LoadBitmap(GetModuleHandle(NULL),
 203                             MAKEINTRESOURCE(IDB_UNDERLINE));
 204
 205     hHelpButton = CreateWindowW(L"Button", L"",
 206                                 WS_CHILD | WS_VISIBLE | BS_ICON,
 207                                 550, 10, 40, 40,
 208                                 hwnd, 0, hInstance, NULL);
 209     LoadString(hInstance, TTT_HELPBUTTON, Tooltip1, MAX_STRING_LENGHT);
 210     CreateToolTip(hHelpButton, hwnd, Tooltip1);
 211
 212     hUpdateButton = CreateWindowW(L"Button", L"",
 213                                   WS_CHILD | WS_VISIBLE | BS_ICON,
 214                                   450, 10, 40, 40,
 215                                   hwnd, 0, hInstance, NULL);
 216     LoadString(hInstance, TTT_UPDATEBUTTON, Tooltip2, MAX_STRING_LENGHT);
 217     CreateToolTip(hUpdateButton, hwnd, Tooltip2);
 218
 219     hProfButton = CreateWindowW(L"Button", L"",
 220                                 WS_CHILD | WS_VISIBLE | BS_ICON,
 221                                 500, 10, 40, 40,
 222                                 hwnd, 0, hInstance, NULL);
 223     LoadString(hInstance, TTT_PROFBUTTON, Tooltip3, MAX_STRING_LENGHT);
 224     CreateToolTip(hProfButton, hwnd, Tooltip3);
 225
 226     hDownloadButton = CreateWindowW(L"Button", L"",
 227                                     WS_CHILD | WS_VISIBLE | BS_BITMAP,
 228                                     330, 505, 140, 33,
 229                                     hwnd, 0, hInstance, NULL);
 230
 231     hUninstallButton = CreateWindowW(L"Button", L"",
 232                                      WS_CHILD | WS_VISIBLE | BS_BITMAP,
 233                                      260, 505, 140, 33,
 234                                      hwnd, 0, hInstance, NULL);
 235
 236     SendMessageW(hProfButton,
 237                 BM_SETIMAGE,
 238                 (WPARAM)IMAGE_ICON,
 239                 (LPARAM)(HANDLE)LoadIcon(hInstance,MAKEINTRESOURCE(IDI_PROF)));
 240     SendMessageW(hHelpButton,
 241                 BM_SETIMAGE,
 242                 (WPARAM)IMAGE_ICON,
 243                 (LPARAM)(HANDLE)LoadIcon(hInstance, MAKEINTRESOURCE(IDI_HELP)));
 244     SendMessageW(hUpdateButton,
 245                 BM_SETIMAGE,
 246                 (WPARAM)IMAGE_ICON,
 247                 (LPARAM)(HANDLE)LoadIcon(hInstance, MAKEINTRESOURCE(IDI_UPDATE)));
 248     SendMessageW(hDownloadButton,
 249                 BM_SETIMAGE,
 250                 (WPARAM)IMAGE_BITMAP,
 251                 (LPARAM)(HANDLE)LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_DOWNLOAD)));
 252     SendMessageW(hUninstallButton,
 253                 BM_SETIMAGE,
 254                 (WPARAM)IMAGE_BITMAP,
 255                 (LPARAM)(HANDLE)LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_UNINSTALL)));
 256     ShowWindow(hUninstallButton, SW_HIDE);
 257
 258     // Set deflaut entry for hApps
 259     Insert.item.mask = TVIF_TEXT|TVIF_IMAGE;
 260     Insert.item.pszText = Strings[IDS_CHOOSE_CATEGORY];
 261     Insert.item.cchTextMax = lstrlenW(Strings[IDS_CHOOSE_CATEGORY]);
 262     Insert.item.iImage = 0;
 263     SendMessage(hApps, TVM_INSERTITEM, 0, (LPARAM)&Insert);
 264
 265     // Create Tree Icons
 266     hImageList = ImageList_Create(16, 16, ILC_COLORDDB, 1, 1);
 267     SendMessageW(hCategories, TVM_SETIMAGELIST, TVSIL_NORMAL, (LPARAM)(HIMAGELIST)hImageList);
 268     SendMessageW(hApps, TVM_SETIMAGELIST, TVSIL_NORMAL, (LPARAM)(HIMAGELIST)hImageList);
 269
 270     ImageList_Add(hImageList,
 271                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_0)), NULL);
 272     ImageList_Add(hImageList,
 273                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_1)), NULL);
 274     ImageList_Add(hImageList,
 275                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_2)), NULL);
 276     ImageList_Add(hImageList,
 277                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_3)), NULL);
 278     ImageList_Add(hImageList,
 279                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_4)), NULL);
 280     ImageList_Add(hImageList,
 281                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_5)), NULL);
 282     ImageList_Add(hImageList,
 283                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_6)), NULL);
 284     ImageList_Add(hImageList,
 285                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_7)), NULL);
 286     ImageList_Add(hImageList,
 287                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_8)), NULL);
 288     ImageList_Add(hImageList,
 289                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_9)), NULL);
 290     ImageList_Add(hImageList,
 291                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_10)), NULL);
 292     ImageList_Add(hImageList,
 293                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_11)), NULL);
 294     ImageList_Add(hImageList,
 295                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_12)), NULL);
 296     ImageList_Add(hImageList,
 297                   LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_TREEVIEW_ICON_13)), NULL);
 298
 299     // Fill the TreeViews
 300     AddItems (hCategories, Root.Children, NULL);
 301
 302     return TRUE;
 303 }
 304
 305 static void
 306 ResizeControl (HWND hwnd, int x1, int y1, int x2, int y2)
 307 {
 308     // Make resizing a little easier
 309     MoveWindow(hwnd, x1, y1, x2-x1, y2-y1, TRUE);
 310 }
 311
 312 static void
 313 DrawBitmap (HDC hdc, int x, int y, HBITMAP hBmp)
 314 {
 315     BITMAP bm;
 316     HDC hdcMem = CreateCompatibleDC(hdc);
 317
 318     SelectObject(hdcMem, hBmp);
 319     GetObject(hBmp, sizeof(bm), &bm);
 320     TransparentBlt(hdc, x, y, bm.bmWidth, bm.bmHeight, hdcMem, 0, 0, bm.bmWidth, bm.bmHeight, 0xFFFFFF);
 321
 322     DeleteDC(hdcMem);
 323 }
 324
 325 static void
 326 DrawDescription (HDC hdc, RECT DescriptionRect)
 327 {
 328     int i;
 329     HFONT Font;
 330     RECT Rect = {DescriptionRect.left+5, DescriptionRect.top+3, DescriptionRect.right-2, DescriptionRect.top+22};
 331
 332     // Backgroud
 333     Rectangle(hdc, DescriptionRect.left, DescriptionRect.top, DescriptionRect.right, DescriptionRect.bottom);
 334
 335     // Underline
 336     for (i=DescriptionRect.left+1;i<DescriptionRect.right-1;i++)
 337         DrawBitmap(hdc, i, DescriptionRect.top+22, hUnderline); // less code then stretching ;)
 338
 339     // Headline
 340     Font = CreateFont(-16 , 0, 0, 0, FW_EXTRABOLD, FALSE, FALSE, FALSE, ANSI_CHARSET,
 341                       OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY, FF_DONTCARE, L"Arial");
 342     SelectObject(hdc, Font);
 343     DrawTextW(hdc, DescriptionHeadline, lstrlenW(DescriptionHeadline), &Rect, DT_SINGLELINE|DT_NOPREFIX);
 344     DeleteObject(Font);
 345
 346     // Description
 347     Font = CreateFont(-13 , 0, 0, 0, FW_NORMAL, FALSE, FALSE, FALSE, ANSI_CHARSET,
 348                       OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY, FF_DONTCARE, L"Arial");
 349     SelectObject(hdc, Font);
 350     Rect.top += 50;
 351     Rect.bottom = DescriptionRect.bottom-2;
 352     DrawTextW(hdc, DescriptionText, lstrlenW(DescriptionText), &Rect, DT_WORDBREAK|DT_NOPREFIX); // ToDo: Call TabbedTextOut to draw a nice table
 353     DeleteObject(Font);
 354
 355 }
 356
 357 void showUninstaller() {
 358     int Split_Vertical = 200;
 359     RECT Rect;
 360
 361     GetClientRect(hwnd,&Rect);
 362     ShowWindow(hUninstallButton,SW_SHOW);
 363     MoveWindow(hDownloadButton,(Split_Vertical+Rect.right-Rect.left)/2,Rect.bottom-Rect.top-45,140,35,TRUE);;
 364 }
 365
 366 void hideUninstaller() {
 367     int Split_Vertical = 200;
 368     RECT Rect;
 369
 370     GetClientRect(hwnd,&Rect);
 371     ShowWindow(hUninstallButton,SW_HIDE);
 372     MoveWindow(hDownloadButton,(Split_Vertical+Rect.right-Rect.left)/2-70,Rect.bottom-Rect.top-45,140,35,TRUE);
 373 }
 374
 375 void startUninstaller(WCHAR* Uninstaller) {
 376     STARTUPINFOW si;
 377     PROCESS_INFORMATION pi;
 378
 379     memset(&si, 0, sizeof(si));
 380     si.cb = sizeof(si);
 381     CreateProcessW(NULL,Uninstaller,NULL,NULL,FALSE,0,NULL,NULL,&si,&pi);
 382     CloseHandle(pi.hThread);
 383     // WaitForSingleObject(pi.hProcess, INFINITE); // If you want to wait for the Unistaller
 384     CloseHandle(pi.hProcess);
 385     hideUninstaller();
 386 }
 387
 388 BOOL
 389 SaveSettings(HWND hwnd)
 390 {
 391     HKEY hKey;
 392     TCHAR szBuf[MAX_PATH];
 393     DWORD dwValue;
 394
 395     if (RegOpenKey(HKEY_LOCAL_MACHINE,
 396                    TEXT("Software\\ReactOS\\Downloader"),
 397                    &hKey) != ERROR_SUCCESS)
 398         return FALSE;
 399
 400     GetDlgItemText(hwnd, IDC_DOWNLOAD_FOLDER_EDIT, szBuf, sizeof(szBuf) / sizeof(TCHAR));
 401     if (GetFileAttributes(szBuf) == 0xFFFFFFFF)
 402         if (!CreateDirectory((LPCTSTR)szBuf,NULL))
 403         {
 404             LoadString(GetModuleHandle(NULL), IDS_UNABLECREATE_FOLDER, szBuf, sizeof(szBuf) / sizeof(TCHAR));
 405             MessageBox(hwnd, (LPCTSTR)szBuf, NULL, MB_ICONSTOP);
 406             return FALSE;
 407         }
 408     if (RegSetValueEx(hKey,
 409                       L"DownloadFolder",
 410                       0,
 411                       REG_SZ,
 412                       (LPBYTE)szBuf,
 413                       (DWORD)(sizeof(szBuf) / sizeof(TCHAR))))
 414         return FALSE;
 415
 416     GetDlgItemText(hwnd, IDC_UPDATE_SERVER_EDIT, szBuf, sizeof(szBuf) / sizeof(TCHAR));
 417     if (RegSetValueEx(hKey,
 418                       L"UpdateServer",
 419                       0,
 420                       REG_SZ,
 421                       (LPBYTE)szBuf,
 422                       (DWORD)(sizeof(szBuf) / sizeof(TCHAR))))
 423         return FALSE;
 424
 425     if (SendDlgItemMessage(hwnd, IDC_DELINST_FILES_CHECKBOX, BM_GETCHECK, 0, 0) == BST_CHECKED)
 426         dwValue = 0x1;
 427     else
 428         dwValue = 0x0;
 429     if (RegSetValueEx(hKey,
 430                       L"DeleteInstaller",
 431                       0,
 432                       REG_DWORD,
 433                       (LPBYTE)&dwValue,
 434                       sizeof(DWORD)))
 435             return FALSE;
 436
 437     RegCloseKey(hKey);
 438
 439     return TRUE;
 440 }
 441
 442 BOOL
 443 InitProfDlg(HWND hwnd)
 444 {
 445     HKEY hKey;
 446     TCHAR Buf[MAX_PATH];
 447     DWORD dwDisp, dwSize;
 448     DWORD dwValue, dwType = REG_DWORD;
 449
 450     if (RegOpenKey(HKEY_LOCAL_MACHINE,
 451                    TEXT("Software\\ReactOS\\Downloader"),
 452                    &hKey) != ERROR_SUCCESS)
 453     {
 454         if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, L"Software\\ReactOS\\Downloader",
 455                            0, NULL, REG_OPTION_NON_VOLATILE,
 456                            KEY_WRITE, NULL, &hKey, &dwDisp))
 457         {
 458             LoadString(GetModuleHandle(NULL), IDS_NOTCREATE_REGKEY, Buf, sizeof(Buf) / sizeof(TCHAR));
 459             MessageBox(hwnd, (LPCTSTR)Buf, NULL, MB_ICONSTOP);
 460             return FALSE;
 461         }
 462     }
 463
 464     dwSize = MAX_PATH;
 465     if (RegQueryValueEx(hKey,
 466                         L"DownloadFolder",
 467                         NULL,
 468                         NULL,
 469                         (LPBYTE)&Buf,
 470                         &dwSize) == ERROR_SUCCESS)
 471     {
 472         SetDlgItemText(hwnd, IDC_DOWNLOAD_FOLDER_EDIT, Buf);
 473     }
 474     else
 475     {
 476         TCHAR DPath[256];
 477         int i;
 478
 479         if (!GetWindowsDirectory(Buf, sizeof(Buf) / sizeof(TCHAR))) return FALSE;
 480
 481         for (i = 0; i < 4; i++)
 482         {
 483             if (i == 3)
 484             {
 485                 DPath[i] = '\0';
 486                 break;
 487             }
 488             DPath[i] = Buf[i];
 489         }
 490         LoadString(GetModuleHandle(NULL), IDS_DOWNLOAD_FOLDER, Buf, sizeof(Buf) / sizeof(TCHAR));
 491         lstrcat((LPTSTR)DPath, Buf);
 492         if (RegSetValueEx(hKey,
 493                           L"DownloadFolder",
 494                           0,
 495                           REG_SZ,
 496                           (LPBYTE)DPath,
 497                           (DWORD)(sizeof(DPath) / sizeof(TCHAR))))
 498             return FALSE;
 499         else
 500         {
 501             if (GetFileAttributes(DPath) == 0xFFFFFFFF)
 502                 if (!CreateDirectory((LPCTSTR)DPath,NULL)) return FALSE;
 503             SetDlgItemText(hwnd, IDC_DOWNLOAD_FOLDER_EDIT, DPath);
 504         }
 505     }
 506
 507     dwSize = MAX_PATH;
 508     if (RegQueryValueEx(hKey,
 509                         L"UpdateServer",
 510                         NULL,
 511                         NULL,
 512                         (LPBYTE)&Buf,
 513                         &dwSize) == ERROR_SUCCESS)
 514     {
 515         SetDlgItemText(hwnd, IDC_UPDATE_SERVER_EDIT, Buf);
 516     }
 517     else
 518     {
 519         LoadString(GetModuleHandle(NULL), IDS_UPDATE_URL, Buf, sizeof(Buf) / sizeof(TCHAR));
 520         if (RegSetValueEx(hKey,
 521                           L"UpdateServer",
 522                           0,
 523                           REG_SZ,
 524                           (LPBYTE)Buf,
 525                           (DWORD)(sizeof(Buf) / sizeof(TCHAR))))
 526             return FALSE;
 527         else
 528         {
 529             SetDlgItemText(hwnd, IDC_UPDATE_SERVER_EDIT, Buf);
 530         }
 531     }
 532
 533     dwSize = sizeof(DWORD);
 534     if (RegQueryValueEx(hKey,
 535                         L"DeleteInstaller",
 536                         NULL,
 537                         &dwType,
 538                         (LPBYTE)&dwValue,
 539                         &dwSize) == ERROR_SUCCESS)
 540     {
 541         if (dwValue == 0x1)
 542             SendDlgItemMessage(hwnd, IDC_DELINST_FILES_CHECKBOX, BM_SETCHECK, 1, 1);
 543     }
 544     else
 545     {
 546         dwValue = 0x0;
 547         if (RegSetValueEx(hKey,
 548                           L"DeleteInstaller",
 549                           0,
 550                           REG_DWORD,
 551                           (LPBYTE)&dwValue,
 552                           sizeof(DWORD)))
 553             return FALSE;
 554     }
 555
 556     RegCloseKey(hKey);
 557
 558     return TRUE;
 559 }
 560
 561 BOOL
 562 ChooseFolder(HWND hwnd)
 563 {
 564     BROWSEINFO fi;
 565     LPCITEMIDLIST lpItemList;
 566     TCHAR szPath[MAX_PATH],Buf[256];
 567
 568     ZeroMemory(&fi, sizeof(BROWSEINFO));
 569     fi.hwndOwner = hwnd;
 570     LoadString(GetModuleHandle(NULL), IDS_CHOOSE_FOLDER, Buf, sizeof(Buf) / sizeof(TCHAR));
 571     fi.lpszTitle = (LPCTSTR)Buf;
 572     fi.ulFlags = BIF_DONTGOBELOWDOMAIN | BIF_RETURNONLYFSDIRS | BIF_BROWSEFORCOMPUTER | BIF_NEWDIALOGSTYLE;
 573     fi.lpfn = NULL;
 574     fi.lParam = -1;
 575     fi.iImage = 0;
 576
 577     if(!(lpItemList = SHBrowseForFolder(&fi))) return FALSE;
 578     SHGetPathFromIDList(lpItemList, szPath);
 579
 580     if (_tcslen(szPath) == 0) return FALSE;
 581     SetDlgItemText(hwnd, IDC_DOWNLOAD_FOLDER_EDIT, szPath);
 582
 583     return TRUE;
 584 }
 585
 586 INT_PTR CALLBACK
 587 ProfDlgProc(HWND hDlg,
 588             UINT message,
 589             WPARAM wParam,
 590             LPARAM lParam)
 591 {
 592     static HICON hIcon;
 593     UNREFERENCED_PARAMETER(lParam);
 594
 595     switch (message)
 596     {
 597         case WM_INITDIALOG:
 598         {
 599             hIcon = LoadImage(GetModuleHandle(NULL),MAKEINTRESOURCE(IDI_MAIN),IMAGE_ICON,16,16,0);
 600             SendMessage(hDlg,WM_SETICON,ICON_SMALL,(LPARAM)hIcon);
 601             InitProfDlg(hDlg);
 602         }
 603         break;
 604
 605         case WM_COMMAND:
 606         {
 607             switch (wParam)
 608             {
 609                 case IDC_CHOOSE_BUTTON:
 610                     ChooseFolder(hDlg);
 611                 break;
 612                 case IDOK:
 613                 {
 614                     SaveSettings(hDlg);
 615                     DestroyIcon(hIcon);
 616                     EndDialog(hDlg,LOWORD(wParam));
 617                     return TRUE;
 618                 }
 619                 break;
 620                 case IDCANCEL:
 621                 {
 622                     DestroyIcon(hIcon);
 623                     EndDialog(hDlg,LOWORD(wParam));
 624                     return TRUE;
 625                 }
 626                 break;
 627             }
 628         }
 629         break;
 630     }
 631
 632     return FALSE;
 633 }
 634
 635 BOOL IsApplicationInstalled(struct Application* App)
 636 {
 637     WCHAR Uninstaller[200];
 638     if(StrCmpW(App->RegName, L"")) {
 639         if(getUninstaller(App->RegName, Uninstaller)) {
 640             return TRUE;
 641         }
 642     }
 643     return FALSE;
 644 }
 645
 646 struct Application* GetDependency(const WCHAR* Dependency)
 647 {
 648     struct Category* Category = Root.Children;
 649
 650     while (Category->Next)
 651     {
 652         while (Category->Apps)
 653         {
 654             if(StrCmpW(Category->Apps->RegName, Dependency) == 0)
 655                 return Category->Apps;
 656             Category->Apps = Category->Apps->Next;
 657         }
 658         Category = Category->Next;
 659     }
 660     return NULL;
 661 }
 662
 663 LRESULT CALLBACK
 664 WndProc (HWND hwnd, UINT Message, WPARAM wParam, LPARAM lParam)
 665 {
 666     static RECT DescriptionRect;
 667     struct Application* AppToInstall;
 668     WCHAR InstallDep[260];
 669     WCHAR InstallDepBuffer[260];
 670     WCHAR Title[260];
 671
 672     switch (Message)
 673     {
 674         case WM_CREATE:
 675         {
 676             if(!SetupControls(hwnd))
 677                 return -1;
 678             ShowMessage(Strings[IDS_WELCOME_TITLE], Strings[IDS_WELCOME]);
 679         }
 680         break;
 681
 682         case WM_PAINT:
 683         {
 684             PAINTSTRUCT ps;
 685             HDC hdc = BeginPaint(hwnd, &ps);
 686             HDC BackbufferHdc = CreateCompatibleDC(hdc);
 687             HBITMAP BackbufferBmp = CreateCompatibleBitmap(hdc, ps.rcPaint.right, ps.rcPaint.bottom);
 688             SelectObject(BackbufferHdc, BackbufferBmp);
 689
 690             FillRect(BackbufferHdc, &ps.rcPaint, CreateSolidBrush(RGB(235,235,235)));
 691             DrawBitmap(BackbufferHdc, 10, 12, hLogo);
 692             DrawDescription(BackbufferHdc, DescriptionRect);
 693
 694             BitBlt(hdc, 0, 0, ps.rcPaint.right, ps.rcPaint.bottom, BackbufferHdc, 0, 0, SRCCOPY);
 695             DeleteObject(BackbufferBmp);
 696             DeleteDC(BackbufferHdc);
 697             EndPaint(hwnd, &ps);
 698         }
 699         break;
 700
 701         case WM_COMMAND:
 702         {
 703             if(HIWORD(wParam) == BN_CLICKED)
 704             {
 705                 if (lParam == (LPARAM)hProfButton)
 706                 {
 707                     DialogBox(GetModuleHandle(NULL),
 708                               MAKEINTRESOURCE(IDD_PROF),
 709                               hwnd,
 710                               ProfDlgProc);
 711                 }
 712                 if (lParam == (LPARAM)hDownloadButton)
 713                 {
 714                     if(SelectedApplication)
 715                     {
 716                         /* install dependencies */
 717                         if(StrCmpW(SelectedApplication->Depends, L""))
 718                         {
 719                             AppToInstall = SelectedApplication;
 720                             SelectedApplication = GetDependency(SelectedApplication->Depends);
 721                             if (SelectedApplication)
 722                                 if (!IsApplicationInstalled(SelectedApplication))
 723                                 {
 724                                     LoadString(GetModuleHandle(NULL), IDS_INSTALL_DEP, InstallDep, sizeof(InstallDep) / sizeof(WCHAR));
 725                                     LoadString(GetModuleHandle(NULL), IDS_WINDOW_TITLE, Title, sizeof(Title) / sizeof(WCHAR));
 726                                     _snwprintf(InstallDepBuffer, sizeof(InstallDepBuffer) / sizeof(WCHAR), InstallDep, SelectedApplication->Name, AppToInstall->Name, SelectedApplication->Name);
 727                                     if (MessageBox(hwnd, InstallDepBuffer, Title, MB_YESNO | MB_ICONINFORMATION) == IDYES)
 728                                     {
 729                                         DialogBoxW(GetModuleHandle(NULL), MAKEINTRESOURCEW(IDD_DOWNLOAD), 0, DownloadProc);
 730                                     }
 731                                 }
 732                             SelectedApplication = AppToInstall;
 733                         }
 734
 735                         /* download and install the app */
 736                         DialogBoxW(GetModuleHandle(NULL), MAKEINTRESOURCEW(IDD_DOWNLOAD), 0, DownloadProc);
 737
 738                         /* install req. hacks to get it working */
 739                         if(StrCmpW(SelectedApplication->PostInstallAction, L""))
 740                         {
 741                             AppToInstall = SelectedApplication;
 742                             CopyMemory(SelectedApplication->Location, SelectedApplication->PostInstallAction, sizeof(SelectedApplication->Location));
 743                             DialogBoxW(GetModuleHandle(NULL), MAKEINTRESOURCEW(IDD_DOWNLOAD), 0, DownloadProc);
 744                             SelectedApplication = AppToInstall;
 745                         }
 746                     }
 747                     else
 748                         ShowMessage(Strings[IDS_NO_APP_TITLE], Strings[IDS_NO_APP]);
 749                 }
 750                 else if (lParam == (LPARAM)hUninstallButton)
 751                 {
 752                     if(SelectedApplication)
 753                     {
 754                         WCHAR Uninstaller[200];
 755                         if(StrCmpW(SelectedApplication->RegName, L"")) {
 756                             if(getUninstaller(SelectedApplication->RegName, Uninstaller))
 757                                 startUninstaller(Uninstaller);
 758                         }
 759                     }
 760                 }
 761                 else if (lParam == (LPARAM)hUpdateButton)
 762                 {
 763                     ShowMessage(Strings[IDS_UPDATE_TITLE], Strings[IDS_UPDATE]);
 764                 }
 765                 else if (lParam == (LPARAM)hHelpButton)
 766                 {
 767                     ShowMessage(Strings[IDS_HELP_TITLE], Strings[IDS_HELP]);
 768                 }
 769             }
 770         }
 771         break;
 772
 773         case WM_NOTIFY:
 774         {
 775             LPNMHDR data = (LPNMHDR)lParam;
 776             WCHAR Uninstaller[200];
 777
 778             if(data->code == TVN_SELCHANGED)
 779             {
 780                 BOOL bShowUninstaller = FALSE;
 781                 if(data->hwndFrom == hCategories)
 782                 {
 783                     struct Category* Category = (struct Category*) ((LPNMTREEVIEW)lParam)->itemNew.lParam;
 784                     CategoryChoosen (hApps, Category);
 785                 }
 786                 else if(data->hwndFrom == hApps)
 787                 {
 788                     SelectedApplication = (struct Application*) ((LPNMTREEVIEW)lParam)->itemNew.lParam;
 789                     if(SelectedApplication)
 790                     {
 791                         ApplicationText[0]=L'\0';
 792                         if(StrCmpW(SelectedApplication->Version, L"")) {
 793                             StrCatW(ApplicationText, Strings[IDS_VERSION]);
 794                             StrCatW(ApplicationText, SelectedApplication->Version);
 795                             StrCatW(ApplicationText, L"\n");
 796                         }
 797                         if(StrCmpW(SelectedApplication->Licence, L"")) {
 798                             StrCatW(ApplicationText, Strings[IDS_LICENCE]);
 799                             StrCatW(ApplicationText, SelectedApplication->Licence);
 800                             StrCatW(ApplicationText, L"\n");
 801                         }
 802                         if(StrCmpW(SelectedApplication->Maintainer, L"")) {
 803                             StrCatW(ApplicationText, Strings[IDS_MAINTAINER]);
 804                             StrCatW(ApplicationText, SelectedApplication->Maintainer);
 805                             StrCatW(ApplicationText, L"\n");
 806                         }
 807                         if(StrCmpW(SelectedApplication->Licence, L"") || StrCmpW(SelectedApplication->Version, L"") || StrCmpW(SelectedApplication->Maintainer, L""))
 808                             StrCatW(ApplicationText, L"\n");
 809                         StrCatW(ApplicationText, SelectedApplication->Description);
 810                         ShowMessage(SelectedApplication->Name, ApplicationText);
 811                         if(StrCmpW(SelectedApplication->RegName, L"")) {
 812                             if(getUninstaller(SelectedApplication->RegName, Uninstaller)) {
 813                                 bShowUninstaller = TRUE;
 814                             }
 815                         }
 816                     }
 817                 }
 818                 if (bShowUninstaller)
 819                     showUninstaller();
 820                 else
 821                     hideUninstaller();
 822             }
 823         }
 824         break;
 825
 826         case WM_SIZING:
 827         {
 828             LPRECT pRect = (LPRECT)lParam;
 829             if (pRect->right-pRect->left < 520)
 830                 pRect->right = pRect->left + 520;
 831
 832             if (pRect->bottom-pRect->top < 300)
 833                 pRect->bottom = pRect->top + 300;
 834         }
 835         break;
 836
 837         case WM_SIZE:
 838         {
 839             int Split_Hozizontal = (HIWORD(lParam)-(45+60))/2 + 60;
 840             int Split_Vertical = 200;
 841             RECT Rect;
 842
 843             ResizeControl(hCategories, 10, 60, Split_Vertical, HIWORD(lParam)-10);
 844             ResizeControl(hApps, Split_Vertical+5, 60, LOWORD(lParam)-10, Split_Hozizontal);
 845             SetRect(&Rect, Split_Vertical+5, Split_Hozizontal+5, LOWORD(lParam)-10, HIWORD(lParam)-50);
 846             DescriptionRect = Rect;
 847
 848             MoveWindow(hHelpButton, LOWORD(lParam)-50, 10, 40, 40, TRUE);
 849             MoveWindow(hUpdateButton, LOWORD(lParam)-150, 10, 40, 40, TRUE);
 850             MoveWindow(hProfButton, LOWORD(lParam)-100, 10, 40, 40, TRUE);
 851             if(IsWindowVisible(hUninstallButton))
 852                 MoveWindow(hDownloadButton, (Split_Vertical+LOWORD(lParam))/2, HIWORD(lParam)-45, 140, 35, TRUE);
 853             else
 854                 MoveWindow(hDownloadButton, (Split_Vertical+LOWORD(lParam))/2-70, HIWORD(lParam)-45, 140, 35, TRUE);
 855             MoveWindow(hUninstallButton, (Split_Vertical+LOWORD(lParam))/2-140, HIWORD(lParam)-45, 140, 35, TRUE);
 856         }
 857         break;
 858
 859         case WM_DESTROY:
 860         {
 861             DeleteObject(hLogo);
 862             if(Root.Children)
 863                 FreeTree(Root.Children);
 864             PostQuitMessage(0);
 865             return 0;
 866         }
 867         break;
 868     }
 869
 870     return DefWindowProc (hwnd, Message, wParam, lParam);
 871 }
 872
 873 INT WINAPI
 874 wWinMain (HINSTANCE hInstance,
 875           HINSTANCE hPrevInst,
 876           LPTSTR lpCmdLine,
 877           INT nCmdShow)
 878 {
 879     int i;
 880     WNDCLASSEXW WndClass = {0};
 881     MSG msg;
 882
 883     InitCommonControls();
 884
 885     // Load strings
 886     for(i=0; i<STRING_COUNT; i++)
 887         LoadStringW(hInstance, i, Strings[i], MAX_STRING_LENGHT); // if you know a better method please tell me.
 888
 889     // Create the window
 890     WndClass.cbSize            = sizeof(WNDCLASSEX);
 891     WndClass.lpszClassName     = L"Downloader";
 892     WndClass.lpfnWndProc       = WndProc;
 893     WndClass.hInstance         = hInstance;
 894     WndClass.style             = CS_HREDRAW | CS_VREDRAW;
 895     WndClass.hIcon             = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_MAIN));
 896     WndClass.hCursor           = LoadCursor(NULL, IDC_ARROW);
 897
 898     RegisterClassExW(&WndClass);
 899
 900     hwnd = CreateWindowW(L"Downloader",
 901                          Strings[IDS_WINDOW_TITLE],
 902                          WS_OVERLAPPEDWINDOW|WS_CLIPCHILDREN,
 903                          CW_USEDEFAULT,
 904                          CW_USEDEFAULT,
 905                          720, 550,
 906                          NULL, NULL,
 907                          hInstance,
 908                          NULL);
 909
 910     // Show it
 911     ShowWindow(hwnd, SW_SHOW);
 912     UpdateWindow(hwnd);
 913
 914     // Message Loop
 915     while(GetMessage(&msg,NULL,0,0))
 916     {
 917         TranslateMessage(&msg);
 918         DispatchMessage(&msg);
 919     }
 920
 921     return 0;
 922 }