1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
|
#include "main.h"
#include "./folderbrowseex.h"
#include "../nu/CGlobalAtom.h"
#include <shobjidl.h>
#include "../replicant/nu/AutoWide.h"
static CGlobalAtom CLSPROP(L"FBEXDLG");
#ifdef _WIN64
#define LONGPTR_CAST LONG_PTR
#else
#define LONGPTR_CAST LONG
#endif
#define PATHTYPE_PIDL FALSE
#define PATHTYPE_STRING TRUE
BOOL CALLBACK browseEnumProc(HWND hwnd, LPARAM lParam)
{
wchar_t cl[32] = {0};
GetClassName(hwnd, cl, ARRAYSIZE(cl));
if (!lstrcmpi(cl, WC_TREEVIEW))
{
PostMessage(hwnd, TVM_ENSUREVISIBLE, 0, (LPARAM)TreeView_GetSelection(hwnd));
return FALSE;
}
return TRUE;
}
static int WINAPI BrowseCallback_Helper(HWND hwnd, UINT uMsg, LPARAM lParam, LPARAM lpData)
{
FolderBrowseEx *lpfbex = reinterpret_cast<FolderBrowseEx*>(lpData);
if (!lpfbex) return 0;
switch (uMsg)
{
case BFFM_INITIALIZED:
lpfbex->hwnd = hwnd;
if(SetPropW(hwnd, CLSPROP, (void*)lpData))
{
lpfbex->oldProc = (LONG_PTR) ((IsWindowUnicode(hwnd)) ? SetWindowLongPtrW(hwnd, DWLP_DLGPROC, (LONGPTR_CAST)WindowProc_Helper) : SetWindowLongPtrA(hwnd, DWLP_DLGPROC, (LONGPTR_CAST)WindowProc_Helper));
if (NULL == lpfbex->oldProc) RemovePropW(hwnd, CLSPROP);
}
// this is not nice but it fixes the selection not working correctly on all OSes
EnumChildWindows(hwnd, browseEnumProc, 0);
break;
}
return lpfbex->BrowseCallback(uMsg, lParam);
}
static LRESULT WINAPI WindowProc_Helper(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
FolderBrowseEx *lpfbex = static_cast<FolderBrowseEx*>(GetPropW(hwnd, CLSPROP));
if (!lpfbex) return 0;
switch(uMsg)
{
case WM_NCDESTROY:
lpfbex->DialogProc(uMsg, wParam, lParam);
if (IsWindowUnicode(hwnd)) SetWindowLongPtrW(hwnd, DWLP_DLGPROC, (LONGPTR_CAST)lpfbex->oldProc);
else SetWindowLongPtrA(hwnd, DWLP_DLGPROC, (LONGPTR_CAST)lpfbex->oldProc);
RemovePropW(hwnd, CLSPROP);
lpfbex->oldProc = NULL;
lpfbex->hwnd = NULL;
return 0;
}
return lpfbex->DialogProc(uMsg, wParam, lParam);
}
static void Initialize(FolderBrowseEx *lpfbx, LPCITEMIDLIST pidlRoot, UINT ulFlags, LPCWSTR lpszCaption, LPCWSTR lpszTitle)
{
lpfbx->pidlRoot = pidlRoot;
lpfbx->ulFlags = ulFlags;
lpfbx->lpszCaption = NULL;
lpfbx->lpszTitle = NULL;
lpfbx->image = -1;
lpfbx->pidl = NULL;
lpfbx->hwnd = NULL;
lpfbx->oldProc = NULL;
CoInitialize(NULL);
lpfbx->pathExpanded.empty = TRUE;
lpfbx->pathSelection.empty = TRUE;
if (lpszCaption) lpfbx->SetCaption(lpszCaption);
if (lpszTitle) lpfbx->SetTitle(lpszTitle);
}
FolderBrowseEx::FolderBrowseEx(LPCITEMIDLIST pidlRoot, UINT ulFlags, LPCWSTR lpszCaption, LPCWSTR lpszTitle)
{
Initialize(this, pidlRoot, ulFlags, lpszCaption, lpszTitle);
}
FolderBrowseEx::FolderBrowseEx(UINT ulFlags, LPCWSTR lpszCaption, LPCWSTR lpszTitle)
{
Initialize(this, NULL, ulFlags, lpszCaption, lpszTitle);
}
FolderBrowseEx::FolderBrowseEx(UINT ulFlags, LPCWSTR lpszTitle)
{
Initialize(this, NULL, ulFlags, L"", lpszTitle);
}
FolderBrowseEx::FolderBrowseEx(void)
{
Initialize(this, NULL, 0, L"", L"");
}
FolderBrowseEx::~FolderBrowseEx(void)
{
if (pidl) CoTaskMemFree(pidl);
if (lpszTitle) free(lpszTitle);
if (lpszCaption) free(lpszCaption);
CoUninitialize();
}
HRESULT FolderBrowseEx::ParseDisplayName(LPCWSTR lpszPath, IBindCtx *pbc, LPITEMIDLIST *ppidl, SFGAOF sfgaoIn, SFGAOF *psfgaoOut)
{
IShellFolder *isf = NULL;
HRESULT result;
SFGAOF attrib;
if (NOERROR != (result = SHGetDesktopFolder(&isf))) return result;
attrib = sfgaoIn;
result = isf->ParseDisplayName(NULL, pbc, (LPWSTR)lpszPath, NULL, ppidl, &attrib);
isf->Release();
if (S_OK != result) *ppidl = NULL;
else if (psfgaoOut) *psfgaoOut = attrib;
return result;
}
void FolderBrowseEx::SetExpanded(LPCITEMIDLIST pidlExpand)
{
pathExpanded.empty = FALSE;
pathExpanded.type = PATHTYPE_PIDL;
pathExpanded.value = (void*)pidlExpand;
if (hwnd) SendMessage(BFFM_SETEXPANDED, pathExpanded.type, (LPARAM)pathExpanded.value);
}
void FolderBrowseEx::SetExpanded(LPCWSTR lpszExpand)
{
pathExpanded.empty = FALSE;
pathExpanded.type = PATHTYPE_STRING;
pathExpanded.value = (void*)lpszExpand;
if (hwnd) SendMessage(BFFM_SETEXPANDED, pathExpanded.type, (LPARAM)pathExpanded.value);
}
void FolderBrowseEx::SetSelection(LPCITEMIDLIST pidlSelect)
{
pathSelection.empty = FALSE;
pathSelection.type = PATHTYPE_PIDL;
pathSelection.value = (void*)pidlSelect;
if (hwnd) SendMessage(BFFM_SETSELECTIONW, pathSelection.type, (LPARAM)pathSelection.value);
}
void FolderBrowseEx::SetSelection(LPCWSTR lpszSelect)
{
pathSelection.empty = FALSE;
pathSelection.type = PATHTYPE_STRING;
pathSelection.value = (void*)lpszSelect;
if (hwnd) SendMessage(BFFM_SETSELECTIONW, pathSelection.type, (LPARAM)pathSelection.value);
}
void FolderBrowseEx::SetCaption(LPCWSTR lpszCaption)
{
if (this->lpszCaption) free(this->lpszCaption);
this->lpszCaption = _wcsdup((lpszCaption) ? lpszCaption : L"");
if (hwnd) SetWindowText(lpszCaption);
}
void FolderBrowseEx::SetTitle(LPCWSTR lpszTitle)
{
if (this->lpszTitle) free(this->lpszTitle);
this->lpszTitle = _wcsdup((lpszTitle) ? lpszTitle : L"");
}
LPITEMIDLIST FolderBrowseEx::Browse(HWND hwndOwner)
{
BROWSEINFOW bi = {0};
bi.hwndOwner = hwndOwner;
bi.pidlRoot = pidlRoot;
bi.pszDisplayName = pszDisplayName;
bi.lpszTitle = lpszTitle;
bi.ulFlags = ulFlags;
bi.lpfn = BrowseCallback_Helper;
bi.lParam = (LPARAM)this;
pidl = SHBrowseForFolderW(&bi);
if (pidl) OnSelectionDone(pidl);
image = (pidl) ? bi.iImage : -1;
return pidl;
}
INT FolderBrowseEx::BrowseCallback(UINT uMsg, LPARAM lParam)
{
switch(uMsg)
{
case BFFM_INITIALIZED: OnInitialized(); break;
case BFFM_IUNKNOWN: OnIUnknown((IUnknown*)lParam); break;
case BFFM_SELCHANGED: OnSelectionChanged((LPCITEMIDLIST)lParam); break;
case BFFM_VALIDATEFAILEDA: return OnValidateFailed(AutoWide((LPCSTR)lParam));
case BFFM_VALIDATEFAILEDW: return OnValidateFailed((LPCWSTR)lParam);
}
return 0;
}
void FolderBrowseEx::OnInitialized(void)
{
if (!pathSelection.empty) SendMessage(BFFM_SETSELECTIONW, pathSelection.type, (LPARAM)pathSelection.value);
if (!pathExpanded.empty) SendMessage(BFFM_SETEXPANDED, pathExpanded.type, (LPARAM)pathExpanded.value);
SetWindowText(lpszCaption);
}
void FolderBrowseEx::OnSelectionChanged(LPCITEMIDLIST pidl)
{
}
INT_PTR FolderBrowseEx::DialogProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
return CallWindowProc(uMsg, wParam, lParam);
}
INT_PTR FolderBrowseEx::CallWindowProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
return (IsWindowUnicode(hwnd)) ? ::CallWindowProcW((WNDPROC)oldProc, hwnd, uMsg, wParam, lParam)
: ::CallWindowProcA((WNDPROC)oldProc, hwnd, uMsg, wParam, lParam);
}
void FolderBrowseEx::SetDialogResult(LRESULT result)
{
SetWindowLongPtr(hwnd, DWLP_MSGRESULT, (LONG)(LONG_PTR)result);
}
|
