2 * Drive management UI code
4 * Copyright 2003 Mark Westcott
5 * Copyright 2004 Chris Morgan
6 * Copyright 2003-2004 Mike Hearn
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
26 #define WIN32_LEAN_AND_MEAN
36 #include <wine/debug.h>
41 WINE_DEFAULT_DEBUG_CHANNEL(winecfg);
43 #define BOX_MODE_CD_ASSIGN 1
44 #define BOX_MODE_CD_AUTODETECT 2
45 #define BOX_MODE_NONE 3
46 #define BOX_MODE_NORMAL 4
48 static BOOL advanced = FALSE;
49 static BOOL updating_ui = FALSE;
50 static struct drive* current_drive;
52 static void get_etched_rect(HWND dialog, RECT *rect);
53 static void update_controls(HWND dialog);
55 static DWORD driveui_msgbox (HWND parent, UINT messageId, DWORD flags)
57 WCHAR* caption = load_string (IDS_WINECFG_TITLE);
58 WCHAR* text = load_string (messageId);
59 DWORD result = MessageBoxW (parent, text, caption, flags);
60 HeapFree (GetProcessHeap(), 0, caption);
61 HeapFree (GetProcessHeap(), 0, text);
65 /**** listview helper functions ****/
67 /* clears the item at index in the listview */
68 static void lv_clear_curr_select(HWND dialog, int index)
70 ListView_SetItemState(GetDlgItem(dialog, IDC_LIST_DRIVES), index, 0, LVIS_SELECTED);
73 /* selects the item at index in the listview */
74 static void lv_set_curr_select(HWND dialog, int index)
76 /* no more than one item can be selected in our listview */
77 lv_clear_curr_select(dialog, -1);
78 ListView_SetItemState(GetDlgItem(dialog, IDC_LIST_DRIVES), index, LVIS_SELECTED, LVIS_SELECTED);
81 /* returns the currently selected item in the listview */
82 static int lv_get_curr_select(HWND dialog)
84 return SendDlgItemMessage(dialog, IDC_LIST_DRIVES, LVM_GETNEXTITEM, -1, LVNI_SELECTED);
87 /* sets the item in the listview at item->iIndex */
88 static void lv_set_item(HWND dialog, LVITEM *item)
90 SendDlgItemMessage(dialog, IDC_LIST_DRIVES, LVM_SETITEM, 0, (LPARAM) item);
93 /* sets specified item's text */
94 static void lv_set_item_text(HWND dialog, int item, int subItem, char *text)
97 if (item < 0 || subItem < 0) return;
98 lvItem.mask = LVIF_TEXT;
100 lvItem.iSubItem = subItem;
101 lvItem.pszText = text;
102 lvItem.cchTextMax = lstrlen(lvItem.pszText);
103 lv_set_item(dialog, &lvItem);
106 /* inserts an item into the listview */
107 static void lv_insert_item(HWND dialog, LVITEM *item)
109 SendDlgItemMessage(dialog, IDC_LIST_DRIVES, LVM_INSERTITEM, 0, (LPARAM) item);
112 /* retrieve the item at index item->iIndex */
113 static void lv_get_item(HWND dialog, LVITEM *item)
115 SendDlgItemMessage(dialog, IDC_LIST_DRIVES, LVM_GETITEM, 0, (LPARAM) item);
118 static void set_advanced(HWND dialog)
127 LoadStringW(GetModuleHandle(NULL), IDS_HIDE_ADVANCED, text, 256);
132 LoadStringW(GetModuleHandle(NULL), IDS_SHOW_ADVANCED, text, 256);
135 ShowWindow(GetDlgItem(dialog, IDC_RADIO_AUTODETECT), state);
136 ShowWindow(GetDlgItem(dialog, IDC_RADIO_ASSIGN), state);
137 ShowWindow(GetDlgItem(dialog, IDC_EDIT_LABEL), state);
138 ShowWindow(GetDlgItem(dialog, IDC_EDIT_DEVICE), state);
139 ShowWindow(GetDlgItem(dialog, IDC_STATIC_LABEL), state);
140 ShowWindow(GetDlgItem(dialog, IDC_BUTTON_BROWSE_DEVICE), state);
141 ShowWindow(GetDlgItem(dialog, IDC_EDIT_SERIAL), state);
142 ShowWindow(GetDlgItem(dialog, IDC_STATIC_SERIAL), state);
143 ShowWindow(GetDlgItem(dialog, IDC_LABELSERIAL_STATIC), state);
144 ShowWindow(GetDlgItem(dialog, IDC_COMBO_TYPE), state);
145 ShowWindow(GetDlgItem(dialog, IDC_STATIC_TYPE), state);
147 /* update the button text based on the state */
148 SetWindowTextW(GetDlgItem(dialog, IDC_BUTTON_SHOW_HIDE_ADVANCED), text);
150 /* redraw for the etched line */
151 get_etched_rect(dialog, &rect);
152 InflateRect(&rect, 5, 5);
153 InvalidateRect(dialog, &rect, TRUE);
156 struct drive_typemap {
161 static const struct drive_typemap type_pairs[] = {
162 { DRIVE_UNKNOWN, IDS_DRIVE_UNKNOWN },
163 { DRIVE_FIXED, IDS_DRIVE_FIXED },
164 { DRIVE_REMOTE, IDS_DRIVE_REMOTE },
165 { DRIVE_REMOVABLE, IDS_DRIVE_REMOVABLE },
166 { DRIVE_CDROM, IDS_DRIVE_CDROM }
169 #define DRIVE_TYPE_DEFAULT 0
171 static void enable_labelserial_box(HWND dialog, int mode)
173 WINE_TRACE("mode=%d\n", mode);
177 case BOX_MODE_CD_ASSIGN:
178 enable(IDC_RADIO_ASSIGN);
179 disable(IDC_EDIT_DEVICE);
180 disable(IDC_BUTTON_BROWSE_DEVICE);
181 enable(IDC_EDIT_SERIAL);
182 enable(IDC_EDIT_LABEL);
183 enable(IDC_STATIC_SERIAL);
184 enable(IDC_STATIC_LABEL);
187 case BOX_MODE_CD_AUTODETECT:
188 enable(IDC_RADIO_ASSIGN);
189 enable(IDC_EDIT_DEVICE);
190 enable(IDC_BUTTON_BROWSE_DEVICE);
191 disable(IDC_EDIT_SERIAL);
192 disable(IDC_EDIT_LABEL);
193 disable(IDC_STATIC_SERIAL);
194 disable(IDC_STATIC_LABEL);
198 disable(IDC_RADIO_ASSIGN);
199 disable(IDC_EDIT_DEVICE);
200 disable(IDC_BUTTON_BROWSE_DEVICE);
201 disable(IDC_EDIT_SERIAL);
202 disable(IDC_EDIT_LABEL);
203 disable(IDC_STATIC_SERIAL);
204 disable(IDC_STATIC_LABEL);
207 case BOX_MODE_NORMAL:
208 enable(IDC_RADIO_ASSIGN);
209 disable(IDC_EDIT_DEVICE);
210 disable(IDC_BUTTON_BROWSE_DEVICE);
211 enable(IDC_EDIT_SERIAL);
212 enable(IDC_EDIT_LABEL);
213 enable(IDC_STATIC_SERIAL);
214 enable(IDC_STATIC_LABEL);
219 static int fill_drives_list(HWND dialog)
222 BOOL drivec_present = FALSE;
230 prevsel = lv_get_curr_select(dialog);
232 /* Clear the listbox */
233 SendDlgItemMessage(dialog, IDC_LIST_DRIVES, LVM_DELETEALLITEMS, 0, 0);
235 for(i = 0; i < 26; i++)
240 /* skip over any unused drives */
241 if (!drives[i].in_use)
244 if (drives[i].letter == 'C')
245 drivec_present = TRUE;
251 item.mask = LVIF_TEXT | LVIF_PARAM;
254 item.pszText = letter;
255 item.cchTextMax = lstrlen(item.pszText);
256 item.lParam = (LPARAM) &drives[i];
258 lv_insert_item(dialog, &item);
259 lv_set_item_text(dialog, count, 1, drives[i].unixpath);
264 WINE_TRACE("loaded %d drives\n", count);
266 /* show the warning if there is no Drive C */
268 ShowWindow(GetDlgItem(dialog, IDS_DRIVE_NO_C), SW_NORMAL);
270 ShowWindow(GetDlgItem(dialog, IDS_DRIVE_NO_C), SW_HIDE);
272 lv_set_curr_select(dialog, prevsel == -1 ? 0 : prevsel);
278 static void on_options_click(HWND dialog)
280 if (IsDlgButtonChecked(dialog, IDC_SHOW_DOT_FILES) == BST_CHECKED)
281 set_reg_key(config_key, "", "ShowDotFiles", "Y");
283 set_reg_key(config_key, "", "ShowDotFiles", "N");
285 SendMessage(GetParent(dialog), PSM_CHANGED, 0, 0);
288 static void on_add_click(HWND dialog)
290 /* we should allocate a drive letter automatically. We also need
291 some way to let the user choose the mapping point, for now we
292 will just force them to enter a path automatically, with / being
293 the default. In future we should be able to temporarily map /
294 then invoke the directory chooser dialog. */
296 char new = 'C'; /* we skip A and B, they are historically floppy drives */
297 long mask = ~drive_available_mask(0); /* the mask is now which drives aren't available */
300 while (mask & (1 << (new - 'A')))
305 driveui_msgbox (dialog, IDS_DRIVE_LETTERS_EXCEEDED, MB_OK | MB_ICONEXCLAMATION);
310 WINE_TRACE("allocating drive letter %c\n", new);
315 LoadStringA (GetModuleHandle (NULL), IDS_SYSTEM_DRIVE_LABEL, label,
316 sizeof(label)/sizeof(label[0]));
317 add_drive(new, "../drive_c", label, "", DRIVE_FIXED);
319 else add_drive(new, "/", "", "", DRIVE_UNKNOWN);
321 fill_drives_list(dialog);
323 /* select the newly created drive */
324 mask = ~drive_available_mask(0);
326 for (i = 0; i < 26; i++)
328 if ('A' + i == new) break;
329 if ((1 << i) & mask) c++;
331 lv_set_curr_select(dialog, c);
333 SetFocus(GetDlgItem(dialog, IDC_LIST_DRIVES));
335 update_controls(dialog);
338 static void on_remove_click(HWND dialog)
344 itemIndex = lv_get_curr_select(dialog);
345 if (itemIndex == -1) return; /* no selection */
347 item.mask = LVIF_PARAM;
348 item.iItem = itemIndex;
351 lv_get_item(dialog, &item);
353 drive = (struct drive *) item.lParam;
355 WINE_TRACE("unixpath: %s\n", drive->unixpath);
357 if (drive->letter == 'C')
359 DWORD result = driveui_msgbox (dialog, IDS_CONFIRM_DELETE_C, MB_YESNO | MB_ICONEXCLAMATION);
360 if (result == IDNO) return;
365 fill_drives_list(dialog);
367 itemIndex = itemIndex - 1;
368 if (itemIndex < 0) itemIndex = 0;
369 lv_set_curr_select(dialog, itemIndex); /* previous item */
371 SetFocus(GetDlgItem(dialog, IDC_LIST_DRIVES));
373 update_controls(dialog);
376 static void update_controls(HWND dialog)
383 int i, selection = -1;
388 i = lv_get_curr_select(dialog);
391 /* no selection? let's select something for the user. this will re-enter */
392 lv_set_curr_select(dialog, i);
396 item.mask = LVIF_PARAM;
400 lv_get_item(dialog, &item);
401 current_drive = (struct drive *) item.lParam;
403 WINE_TRACE("Updating sheet for drive %c\n", current_drive->letter);
406 path = current_drive->unixpath;
407 WINE_TRACE("set path control text to '%s'\n", path);
408 set_text(dialog, IDC_EDIT_PATH, path);
411 type = current_drive->type;
412 SendDlgItemMessage(dialog, IDC_COMBO_TYPE, CB_RESETCONTENT, 0, 0);
414 for (i = 0; i < sizeof(type_pairs) / sizeof(struct drive_typemap); i++)
417 LoadStringW (GetModuleHandle (NULL), type_pairs[i].idDesc, driveDesc,
418 sizeof(driveDesc)/sizeof(driveDesc[0]));
419 SendDlgItemMessageW (dialog, IDC_COMBO_TYPE, CB_ADDSTRING, 0, (LPARAM)driveDesc);
421 if (type_pairs[i].sCode == type)
427 if (selection == -1) selection = DRIVE_TYPE_DEFAULT;
428 SendDlgItemMessage(dialog, IDC_COMBO_TYPE, CB_SETCURSEL, selection, 0);
430 EnableWindow( GetDlgItem( dialog, IDC_BUTTON_REMOVE ), (current_drive->letter != 'C') );
431 EnableWindow( GetDlgItem( dialog, IDC_EDIT_PATH ), (current_drive->letter != 'C') );
432 EnableWindow( GetDlgItem( dialog, IDC_BUTTON_BROWSE_PATH ), (current_drive->letter != 'C') );
433 EnableWindow( GetDlgItem( dialog, IDC_COMBO_TYPE ), (current_drive->letter != 'C') );
435 /* removeable media properties */
436 label = current_drive->label;
437 set_text(dialog, IDC_EDIT_LABEL, label);
439 /* set serial edit text */
440 serial = current_drive->serial;
441 set_text(dialog, IDC_EDIT_SERIAL, serial);
443 /* TODO: get the device here to put into the edit box */
444 device = "Not implemented yet";
445 set_text(dialog, IDC_EDIT_DEVICE, device);
448 selection = IDC_RADIO_ASSIGN;
449 if ((type == DRIVE_CDROM) || (type == DRIVE_REMOVABLE))
453 selection = IDC_RADIO_AUTODETECT;
454 enable_labelserial_box(dialog, BOX_MODE_CD_AUTODETECT);
458 selection = IDC_RADIO_ASSIGN;
459 enable_labelserial_box(dialog, BOX_MODE_CD_ASSIGN);
464 enable_labelserial_box(dialog, BOX_MODE_NORMAL);
465 selection = IDC_RADIO_ASSIGN;
468 CheckRadioButton(dialog, IDC_RADIO_AUTODETECT, IDC_RADIO_ASSIGN, selection);
475 static void on_edit_changed(HWND dialog, WORD id)
477 if (updating_ui) return;
479 WINE_TRACE("edit id %d changed\n", id);
487 label = get_text(dialog, id);
488 HeapFree(GetProcessHeap(), 0, current_drive->label);
489 current_drive->label = label ? label : strdupA("");
491 WINE_TRACE("set label to %s\n", current_drive->label);
493 /* enable the apply button */
494 SendMessage(GetParent(dialog), PSM_CHANGED, (WPARAM) dialog, 0);
502 path = get_text(dialog, id);
503 HeapFree(GetProcessHeap(), 0, current_drive->unixpath);
504 current_drive->unixpath = path ? path : strdupA("drive_c");
506 WINE_TRACE("set path to %s\n", current_drive->unixpath);
508 lv_set_item_text(dialog, lv_get_curr_select(dialog), 1,
509 current_drive->unixpath);
511 /* enable the apply button */
512 SendMessage(GetParent(dialog), PSM_CHANGED, (WPARAM) dialog, 0);
516 case IDC_EDIT_SERIAL:
520 serial = get_text(dialog, id);
521 HeapFree(GetProcessHeap(), 0, current_drive->serial);
522 current_drive->serial = serial ? serial : strdupA("");
524 WINE_TRACE("set serial to %s\n", current_drive->serial);
526 /* enable the apply button */
527 SendMessage(GetParent(dialog), PSM_CHANGED, (WPARAM) dialog, 0);
531 case IDC_EDIT_DEVICE:
533 char *device = get_text(dialog, id);
534 /* TODO: handle device if/when it makes sense to do so.... */
535 HeapFree(GetProcessHeap(), 0, device);
541 static void get_etched_rect(HWND dialog, RECT *rect)
543 GetClientRect(dialog, rect);
545 /* these dimensions from the labelserial static in En.rc */
552 /* this just draws a nice line to separate the advanced gui from the n00b gui :) */
553 static void paint(HWND dialog)
557 BeginPaint(dialog, &ps);
563 get_etched_rect(dialog, &rect);
565 DrawEdge(ps.hdc, &rect, EDGE_ETCHED, BF_TOP);
568 EndPaint(dialog, &ps);
571 BOOL browse_for_unix_folder(HWND dialog, WCHAR *pszPath)
573 static WCHAR wszUnixRootDisplayName[] =
574 { ':',':','{','C','C','7','0','2','E','B','2','-','7','D','C','5','-','1','1','D','9','-',
575 'C','6','8','7','-','0','0','0','4','2','3','8','A','0','1','C','D','}', 0 };
576 WCHAR pszChoosePath[FILENAME_MAX];
587 IShellFolder *pDesktop;
588 LPITEMIDLIST pidlUnixRoot, pidlSelectedPath;
591 LoadStringW(GetModuleHandle(NULL), IDS_CHOOSE_PATH, pszChoosePath, FILENAME_MAX);
593 hr = SHGetDesktopFolder(&pDesktop);
594 if (!SUCCEEDED(hr)) return FALSE;
596 hr = IShellFolder_ParseDisplayName(pDesktop, NULL, NULL, wszUnixRootDisplayName, NULL,
597 &pidlUnixRoot, NULL);
598 if (!SUCCEEDED(hr)) {
599 IShellFolder_Release(pDesktop);
603 bi.pidlRoot = pidlUnixRoot;
604 pidlSelectedPath = SHBrowseForFolderW(&bi);
605 SHFree(pidlUnixRoot);
607 if (pidlSelectedPath) {
608 STRRET strSelectedPath;
609 WCHAR *pszSelectedPath;
612 hr = IShellFolder_GetDisplayNameOf(pDesktop, pidlSelectedPath, SHGDN_FORPARSING,
614 IShellFolder_Release(pDesktop);
615 if (!SUCCEEDED(hr)) {
616 SHFree(pidlSelectedPath);
620 hr = StrRetToStrW(&strSelectedPath, pidlSelectedPath, &pszSelectedPath);
621 SHFree(pidlSelectedPath);
622 if (!SUCCEEDED(hr)) return FALSE;
624 lstrcpyW(pszPath, pszSelectedPath);
626 CoTaskMemFree(pszSelectedPath);
632 static void init_listview_columns(HWND dialog)
634 LVCOLUMNW listColumn;
639 GetClientRect(GetDlgItem(dialog, IDC_LIST_DRIVES), &viewRect);
640 width = (viewRect.right - viewRect.left) / 6 - 5;
642 LoadStringW (GetModuleHandle (NULL), IDS_COL_DRIVELETTER, column,
643 sizeof(column)/sizeof(column[0]));
644 listColumn.mask = LVCF_TEXT | LVCF_WIDTH | LVCF_SUBITEM;
645 listColumn.pszText = column;
646 listColumn.cchTextMax = lstrlenW (listColumn.pszText);
647 listColumn.cx = width;
649 SendDlgItemMessageW (dialog, IDC_LIST_DRIVES, LVM_INSERTCOLUMNW, 0, (LPARAM) &listColumn);
651 LoadStringW (GetModuleHandle (NULL), IDS_COL_DRIVEMAPPING, column,
652 sizeof(column)/sizeof(column[0]));
653 listColumn.cx = viewRect.right - viewRect.left - width;
654 listColumn.pszText = column;
655 listColumn.cchTextMax = lstrlenW (listColumn.pszText);
657 SendDlgItemMessageW (dialog, IDC_LIST_DRIVES, LVM_INSERTCOLUMNW, 1, (LPARAM) &listColumn);
660 static void load_drive_options(HWND dialog)
662 if (!strcmp(get_reg_key(config_key, "", "ShowDotFiles", "N"), "Y"))
663 CheckDlgButton(dialog, IDC_SHOW_DOT_FILES, BST_CHECKED);
667 DriveDlgProc (HWND dialog, UINT msg, WPARAM wParam, LPARAM lParam)
675 init_listview_columns(dialog);
677 load_drive_options(dialog);
679 if (!drives[2].in_use)
680 driveui_msgbox (dialog, IDS_NO_DRIVE_C, MB_OK | MB_ICONEXCLAMATION);
682 fill_drives_list(dialog);
683 update_controls(dialog);
684 /* put in non-advanced mode by default */
685 set_advanced(dialog);
689 set_window_title(dialog);
697 switch (HIWORD(wParam))
700 on_edit_changed(dialog, LOWORD(wParam));
704 switch (LOWORD(wParam))
706 case IDC_SHOW_DOT_FILES:
707 on_options_click(dialog);
713 SendMessage(GetParent(dialog), PSM_CHANGED, 0, 0);
717 switch (LOWORD(wParam))
720 if (HIWORD(wParam) != BN_CLICKED) break;
721 on_add_click(dialog);
724 case IDC_BUTTON_REMOVE:
725 if (HIWORD(wParam) != BN_CLICKED) break;
726 on_remove_click(dialog);
729 case IDC_BUTTON_EDIT:
730 if (HIWORD(wParam) != BN_CLICKED) break;
731 item = SendMessage(GetDlgItem(dialog, IDC_LIST_DRIVES), LB_GETCURSEL, 0, 0);
732 drive = (struct drive *) SendMessage(GetDlgItem(dialog, IDC_LIST_DRIVES), LB_GETITEMDATA, item, 0);
735 case IDC_BUTTON_AUTODETECT:
737 fill_drives_list(dialog);
738 SendMessage(GetParent(dialog), PSM_CHANGED, 0, 0);
741 case IDC_BUTTON_SHOW_HIDE_ADVANCED:
742 advanced = !advanced;
743 set_advanced(dialog);
746 case IDC_BUTTON_BROWSE_PATH:
748 WCHAR szTargetPath[FILENAME_MAX];
749 if (browse_for_unix_folder(dialog, szTargetPath))
750 set_textW(dialog, IDC_EDIT_PATH, szTargetPath);
754 case IDC_RADIO_ASSIGN:
758 str = get_text(dialog, IDC_EDIT_LABEL);
759 HeapFree(GetProcessHeap(), 0, current_drive->label);
760 current_drive->label = str ? str : strdupA("");
762 str = get_text(dialog, IDC_EDIT_SERIAL);
763 HeapFree(GetProcessHeap(), 0, current_drive->serial);
764 current_drive->serial = str ? str : strdupA("");
766 /* TODO: we don't have a device at this point */
768 enable_labelserial_box(dialog, BOX_MODE_CD_ASSIGN);
776 int mode = BOX_MODE_NORMAL;
779 if (HIWORD(wParam) != CBN_SELCHANGE) break;
781 selection = SendDlgItemMessage(dialog, IDC_COMBO_TYPE, CB_GETCURSEL, 0, 0);
783 if (selection >= 0 &&
784 (type_pairs[selection].sCode == DRIVE_CDROM ||
785 type_pairs[selection].sCode == DRIVE_REMOVABLE))
787 if (IsDlgButtonChecked(dialog, IDC_RADIO_AUTODETECT))
788 mode = BOX_MODE_CD_AUTODETECT;
790 mode = BOX_MODE_CD_ASSIGN;
793 enable_labelserial_box(dialog, mode);
795 current_drive->type = type_pairs[selection].sCode;
803 switch (((LPNMHDR)lParam)->code)
806 WINE_TRACE("PSN_KILLACTIVE\n");
807 SetWindowLongPtr(dialog, DWLP_MSGRESULT, FALSE);
810 apply_drive_changes();
811 SetWindowLongPtr(dialog, DWLP_MSGRESULT, PSNRET_NOERROR);
815 case LVN_ITEMCHANGED:
817 LPNMLISTVIEW lpnm = (LPNMLISTVIEW)lParam;
818 if (!(lpnm->uOldState & LVIS_SELECTED) &&
819 (lpnm->uNewState & LVIS_SELECTED))
820 update_controls(dialog);