setupapi: Enable Browse action on SetupPromptForDisk dialog.
[wine] / dlls / setupapi / dialog.c
1 /*
2  * SetupAPI dialog functions
3  *
4  * Copyright 2009 Ricardo Filipe
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdarg.h>
22
23 #include "wine/debug.h"
24 #include "windef.h"
25 #include "winbase.h"
26 #include "wingdi.h"
27 #include "winuser.h"
28 #include "winreg.h"
29 #include "commdlg.h"
30 #include "setupapi.h"
31 #include "winnls.h"
32 #include "setupapi_private.h"
33
34 #include "wine/unicode.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(setupapi);
37
38 struct promptdisk_params {
39     PCWSTR DialogTitle;
40     PCWSTR DiskName;
41     PCWSTR PathToSource;
42     PCWSTR FileSought;
43     PCWSTR TagFile;
44     DWORD DiskPromptStyle;
45     PWSTR PathBuffer;
46     DWORD PathBufferSize;
47     PDWORD PathRequiredSize;
48 };
49
50 /* initiates the fields of the SetupPromptForDisk dialog according to the parameters
51 */
52 static void promptdisk_init(HWND hwnd, struct promptdisk_params *params)
53 {
54     WCHAR format[256];
55     WCHAR message[256];
56
57     SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)params);
58
59     if(params->DialogTitle)
60         SetWindowTextW(hwnd, params->DialogTitle);
61     if(params->PathToSource)
62         SetDlgItemTextW(hwnd, IDC_PATH, params->PathToSource);
63
64     if(!(params->DiskPromptStyle & IDF_OEMDISK))
65     {
66         LoadStringW(SETUPAPI_hInstance, IDS_PROMPTDISK, format,
67             sizeof(format)/sizeof(format[0]));
68
69         if(params->DiskName)
70             snprintfW(message, sizeof(message)/sizeof(message[0]), format,
71                 params->FileSought, params->DiskName);
72         else
73         {
74             WCHAR unknown[256];
75             LoadStringW(SETUPAPI_hInstance, IDS_UNKNOWN, unknown,
76                 sizeof(unknown)/sizeof(unknown[0]));
77             snprintfW(message, sizeof(message)/sizeof(message[0]), format,
78                 params->FileSought, unknown);
79         }
80         SetDlgItemTextW(hwnd, IDC_FILENEEDED, message);
81
82         LoadStringW(SETUPAPI_hInstance, IDS_INFO, message,
83             sizeof(message)/sizeof(message[0]));
84         SetDlgItemTextW(hwnd, IDC_INFO, message);
85         LoadStringW(SETUPAPI_hInstance, IDS_COPYFROM, message,
86             sizeof(message)/sizeof(message[0]));
87         SetDlgItemTextW(hwnd, IDC_COPYFROM, message);
88     }
89     if(params->DiskPromptStyle & IDF_NOBROWSE)
90         ShowWindow(GetDlgItem(hwnd, IDC_RUNDLG_BROWSE), SW_HIDE);
91 }
92
93 /* When the user clicks the browse button in SetupPromptForDisk dialog
94  * it copies the path of the selected file to the dialog path field
95  */
96 static void promptdisk_browse(HWND hwnd, struct promptdisk_params *params)
97 {
98     OPENFILENAMEW ofn;
99     ZeroMemory(&ofn, sizeof(ofn));
100
101     ofn.lStructSize = sizeof(ofn);
102     ofn.Flags = OFN_HIDEREADONLY | OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST;
103     ofn.hwndOwner = hwnd;
104     ofn.nMaxFile = MAX_PATH;
105     ofn.lpstrFile = HeapAlloc(GetProcessHeap(), 0, MAX_PATH*sizeof(WCHAR));
106     strcpyW(ofn.lpstrFile, params->FileSought);
107
108     if(GetOpenFileNameW(&ofn))
109     {
110         WCHAR* last_slash = strrchrW(ofn.lpstrFile, '\\');
111         if (last_slash) *last_slash = 0;
112         SetDlgItemTextW(hwnd, IDC_PATH, ofn.lpstrFile);
113     }
114     HeapFree(GetProcessHeap(), 0, ofn.lpstrFile);
115 }
116
117 /* Handles the messages sent to the SetupPromptForDisk dialog
118 */
119 static INT_PTR CALLBACK promptdisk_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
120 {
121     switch(msg)
122     {
123         case WM_INITDIALOG:
124             promptdisk_init(hwnd, (struct promptdisk_params *)lParam);
125             return TRUE;
126         case WM_COMMAND:
127             switch(wParam)
128             {
129                 case IDCANCEL:
130                     EndDialog(hwnd, DPROMPT_CANCEL);
131                     return TRUE;
132                 case IDC_RUNDLG_BROWSE:
133                 {
134                     struct promptdisk_params *params =
135                         (struct promptdisk_params *)GetWindowLongPtrW(hwnd, DWLP_USER);
136                     promptdisk_browse(hwnd, params);
137                     return TRUE;
138                 }
139             }
140     }
141     return FALSE;
142 }
143
144 /***********************************************************************
145  *      SetupPromptForDiskW (SETUPAPI.@)
146  */
147 UINT WINAPI SetupPromptForDiskW(HWND hwndParent, PCWSTR DialogTitle, PCWSTR DiskName,
148         PCWSTR PathToSource, PCWSTR FileSought, PCWSTR TagFile, DWORD DiskPromptStyle,
149         PWSTR PathBuffer, DWORD PathBufferSize, PDWORD PathRequiredSize)
150 {
151     struct promptdisk_params params;
152     UINT ret;
153
154     TRACE("%p, %s, %s, %s, %s, %s, 0x%08x, %p, %d, %p\n", hwndParent, debugstr_w(DialogTitle),
155           debugstr_w(DiskName), debugstr_w(PathToSource), debugstr_w(FileSought),
156           debugstr_w(TagFile), DiskPromptStyle, PathBuffer, PathBufferSize,
157           PathRequiredSize);
158
159     if(!FileSought)
160     {
161         SetLastError(ERROR_INVALID_PARAMETER);
162         return DPROMPT_CANCEL;
163     }
164     params.DialogTitle = DialogTitle;
165     params.DiskName = DiskName;
166     params.PathToSource = PathToSource;
167     params.FileSought = FileSought;
168     params.TagFile = TagFile;
169     params.DiskPromptStyle = DiskPromptStyle;
170     params.PathBuffer = PathBuffer;
171     params.PathBufferSize = PathBufferSize;
172     params.PathRequiredSize = PathRequiredSize;
173
174     ret = DialogBoxParamW(SETUPAPI_hInstance, MAKEINTRESOURCEW(IDPROMPTFORDISK),
175         hwndParent, promptdisk_proc, (LPARAM)&params);
176
177     if(ret == DPROMPT_CANCEL)
178         SetLastError(ERROR_CANCELLED);
179     return ret;
180 }