9d0380354b762915a529ad8938afc6ee9ec29099
[reactos.git] / reactos / dll / win32 / browseui / addresseditbox.cpp
1 /*
2 * ReactOS Explorer
3 *
4 * Copyright 2009 Andrew Hill <ash77 at domain reactos.org>
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21 /*
22 This class handles the combo box of the address band.
23 */
24
25 #include "precomp.h"
26
27 /*
28 TODO:
29 Handle listbox dropdown message and fill contents
30 Add drag and drop of icon in edit box
31 Handle change notifies to update appropriately
32 Fix so selection in combo listbox navigates
33 */
34
35 CAddressEditBox::CAddressEditBox() :
36 fCombobox(NULL, this, 1),
37 fEditWindow(NULL, this, 1),
38 fSite(NULL)
39 {
40 }
41
42 CAddressEditBox::~CAddressEditBox()
43 {
44 }
45
46 HRESULT STDMETHODCALLTYPE CAddressEditBox::SetOwner(IUnknown *)
47 {
48 // connect to browser connection point
49 return 0;
50 }
51
52 HRESULT STDMETHODCALLTYPE CAddressEditBox::FileSysChange(long param8, long paramC)
53 {
54 return E_NOTIMPL;
55 }
56
57 HRESULT STDMETHODCALLTYPE CAddressEditBox::Refresh(long param8)
58 {
59 return E_NOTIMPL;
60 }
61
62 HRESULT STDMETHODCALLTYPE CAddressEditBox::Init(HWND comboboxEx, HWND editControl, long param14, IUnknown *param18)
63 {
64 CComPtr<IBrowserService> browserService;
65
66 fCombobox.SubclassWindow(comboboxEx);
67 fEditWindow.SubclassWindow(editControl);
68 fSite = param18;
69
70 SHAutoComplete(fEditWindow.m_hWnd, SHACF_FILESYSTEM | SHACF_URLALL | SHACF_USETAB);
71
72 // take advice to watch events
73 HRESULT hResult = IUnknown_QueryService(param18, SID_SShellBrowser, IID_PPV_ARG(IBrowserService, &browserService));
74 if (SUCCEEDED(hResult))
75 {
76 hResult = AtlAdvise(browserService, static_cast<IDispatch *>(this), DIID_DWebBrowserEvents, &fAdviseCookie);
77 }
78
79 return hResult;
80 }
81
82 HRESULT STDMETHODCALLTYPE CAddressEditBox::SetCurrentDir(long paramC)
83 {
84 return E_NOTIMPL;
85 }
86
87 HRESULT STDMETHODCALLTYPE CAddressEditBox::ParseNow(long paramC)
88 {
89 ULONG eaten;
90 ULONG attributes;
91 HRESULT hr;
92 HWND topLevelWindow;
93 PIDLIST_ABSOLUTE pidlCurrent= NULL;
94 PIDLIST_RELATIVE pidlRelative = NULL;
95 CComPtr<IShellFolder> psfCurrent;
96
97 CComPtr<IBrowserService> pbs;
98 hr = IUnknown_QueryService(fSite, SID_SShellBrowser, IID_PPV_ARG(IBrowserService, &pbs));
99 if (FAILED_UNEXPECTEDLY(hr))
100 return hr;
101
102 hr = IUnknown_GetWindow(pbs, &topLevelWindow);
103 if (FAILED_UNEXPECTEDLY(hr))
104 return hr;
105
106 /* Get the path to browse and expand it if needed */
107 LPWSTR input;
108 int inputLength = fCombobox.GetWindowTextLength() + 2;
109
110 input = new WCHAR[inputLength];
111 fCombobox.GetWindowText(input, inputLength);
112
113 LPWSTR address;
114 int addressLength = ExpandEnvironmentStrings(input, NULL, 0);
115
116 if (addressLength <= 0)
117 {
118 address = input;
119 }
120 else
121 {
122 addressLength += 2;
123 address = new WCHAR[addressLength];
124 if (!ExpandEnvironmentStrings(input, address, addressLength))
125 {
126 delete[] address;
127 address = input;
128 }
129 }
130
131 /* Try to parse a relative path and if it fails, try to browse an absolute path */
132 CComPtr<IShellFolder> psfDesktop;
133 hr = SHGetDesktopFolder(&psfDesktop);
134 if (FAILED_UNEXPECTEDLY(hr))
135 goto cleanup;
136
137 hr = pbs->GetPidl(&pidlCurrent);
138 if (FAILED_UNEXPECTEDLY(hr))
139 goto cleanup;
140
141 hr = psfDesktop->BindToObject(pidlCurrent, NULL, IID_PPV_ARG(IShellFolder, &psfCurrent));
142 if (FAILED_UNEXPECTEDLY(hr))
143 goto cleanup;
144
145 hr = psfCurrent->ParseDisplayName(topLevelWindow, NULL, address, &eaten, &pidlRelative, &attributes);
146 if (SUCCEEDED(hr))
147 {
148 pidlLastParsed = ILCombine(pidlCurrent, pidlRelative);
149 ILFree(pidlRelative);
150 goto cleanup;
151 }
152
153 /* We couldn't parse a relative path, attempt to parse an absolute path */
154 hr = psfDesktop->ParseDisplayName(topLevelWindow, NULL, address, &eaten, &pidlLastParsed, &attributes);
155
156 cleanup:
157 if (pidlCurrent)
158 ILFree(pidlCurrent);
159 if (address != input)
160 delete [] address;
161 delete [] input;
162
163 return hr;
164 }
165
166 HRESULT STDMETHODCALLTYPE CAddressEditBox::Execute(long paramC)
167 {
168 HRESULT hr;
169
170 /*
171 * Parse the path is it wasn't parsed
172 */
173 if (!pidlLastParsed)
174 ParseNow(0);
175
176 if (!pidlLastParsed)
177 return E_FAIL;
178
179 /*
180 * Get the IShellBrowser and IBrowserService interfaces of the shell browser
181 */
182 CComPtr<IShellBrowser> pisb;
183 hr = IUnknown_QueryService(fSite, SID_SShellBrowser, IID_PPV_ARG(IShellBrowser, &pisb));
184 if (FAILED(hr))
185 return hr;
186
187 CComPtr<IBrowserService> pbs;
188 pisb->QueryInterface(IID_PPV_ARG(IBrowserService, &pbs));
189 if (FAILED(hr))
190 return hr;
191
192 /*
193 * Get the current pidl of the shellbrowser and check if it is the same with the parsed one
194 */
195 PIDLIST_ABSOLUTE pidl;
196 hr = pbs->GetPidl(&pidl);
197 if (FAILED(hr))
198 return hr;
199
200 CComPtr<IShellFolder> psf;
201 hr = SHGetDesktopFolder(&psf);
202 if (FAILED(hr))
203 return hr;
204
205 hr = psf->CompareIDs(0, pidl, pidlLastParsed);
206
207 SHFree(pidl);
208 if (hr == 0)
209 return S_OK;
210
211 /*
212 * Attempt to browse to the parsed pidl
213 */
214 hr = pisb->BrowseObject(pidlLastParsed, 0);
215 if (SUCCEEDED(hr))
216 return hr;
217
218 /*
219 * Browsing to the pidl failed so it's not a folder. So invoke its defaule command.
220 */
221 HWND topLevelWindow;
222 hr = IUnknown_GetWindow(pisb, &topLevelWindow);
223 if (FAILED(hr))
224 return hr;
225
226 LPCITEMIDLIST pidlChild;
227 CComPtr<IShellFolder> sf;
228 hr = SHBindToParent(pidlLastParsed, IID_PPV_ARG(IShellFolder, &sf), &pidlChild);
229 if (FAILED(hr))
230 return hr;
231
232 hr = SHInvokeDefaultCommand(topLevelWindow, sf, pidlChild);
233 if (FAILED(hr))
234 return hr;
235
236 return hr;
237 }
238
239 HRESULT STDMETHODCALLTYPE CAddressEditBox::Save(long paramC)
240 {
241 return E_NOTIMPL;
242 }
243
244 HRESULT STDMETHODCALLTYPE CAddressEditBox::OnWinEvent(
245 HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *theResult)
246 {
247 LPNMHDR hdr;
248
249 *theResult = 0;
250
251 switch (uMsg)
252 {
253 case WM_NOTIFY:
254 hdr = (LPNMHDR) lParam;
255 if (hdr->code == CBEN_ENDEDIT)
256 {
257 NMCBEENDEDITW *endEdit = (NMCBEENDEDITW*) lParam;
258 if (endEdit->iWhy == CBENF_RETURN)
259 {
260 Execute(0);
261 }
262 else if (endEdit->iWhy == CBENF_ESCAPE)
263 {
264 /* Reset the contents of the combo box */
265 }
266 }
267 break;
268 }
269 return S_OK;
270 }
271
272 HRESULT STDMETHODCALLTYPE CAddressEditBox::IsWindowOwner(HWND hWnd)
273 {
274 if (fCombobox.m_hWnd == hWnd)
275 return S_OK;
276 if (fEditWindow.m_hWnd == hWnd)
277 return S_OK;
278 return S_FALSE;
279 }
280
281 HRESULT STDMETHODCALLTYPE CAddressEditBox::QueryStatus(
282 const GUID *pguidCmdGroup, ULONG cCmds, OLECMD prgCmds[ ], OLECMDTEXT *pCmdText)
283 {
284 return E_NOTIMPL;
285 }
286
287 HRESULT STDMETHODCALLTYPE CAddressEditBox::Exec(const GUID *pguidCmdGroup, DWORD nCmdID,
288 DWORD nCmdexecopt, VARIANT *pvaIn, VARIANT *pvaOut)
289 {
290 return E_NOTIMPL;
291 }
292
293 HRESULT STDMETHODCALLTYPE CAddressEditBox::GetTypeInfoCount(UINT *pctinfo)
294 {
295 return E_NOTIMPL;
296 }
297
298 HRESULT STDMETHODCALLTYPE CAddressEditBox::GetTypeInfo(UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
299 {
300 return E_NOTIMPL;
301 }
302
303 HRESULT STDMETHODCALLTYPE CAddressEditBox::GetIDsOfNames(
304 REFIID riid, LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
305 {
306 return E_NOTIMPL;
307 }
308
309 HRESULT STDMETHODCALLTYPE CAddressEditBox::Invoke(DISPID dispIdMember, REFIID riid, LCID lcid,
310 WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
311 {
312 CComPtr<IBrowserService> isb;
313 CComPtr<IShellFolder> sf;
314 HRESULT hr;
315 INT indexClosed, indexOpen, itemExists, oldIndex;
316 DWORD result;
317 COMBOBOXEXITEMW item;
318 PIDLIST_ABSOLUTE absolutePIDL;
319 LPCITEMIDLIST pidlChild;
320 LPITEMIDLIST pidlPrevious;
321 STRRET ret;
322 WCHAR buf[4096];
323
324 if (pDispParams == NULL)
325 return E_INVALIDARG;
326
327 switch (dispIdMember)
328 {
329 case DISPID_NAVIGATECOMPLETE2:
330 case DISPID_DOCUMENTCOMPLETE:
331 pidlLastParsed = NULL;
332
333 oldIndex = fCombobox.SendMessage(CB_GETCURSEL, 0, 0);
334
335 itemExists = FALSE;
336 pidlPrevious = NULL;
337
338 ZeroMemory(&item, sizeof(item));
339 item.mask = CBEIF_LPARAM;
340 item.iItem = 0;
341 if (fCombobox.SendMessage(CBEM_GETITEM, 0, reinterpret_cast<LPARAM>(&item)))
342 {
343 pidlPrevious = reinterpret_cast<LPITEMIDLIST>(item.lParam);
344 if (pidlPrevious)
345 itemExists = TRUE;
346 }
347
348 hr = IUnknown_QueryService(fSite, SID_STopLevelBrowser, IID_PPV_ARG(IBrowserService, &isb));
349 if (FAILED_UNEXPECTEDLY(hr))
350 return hr;
351
352 hr = isb->GetPidl(&absolutePIDL);
353 if (FAILED_UNEXPECTEDLY(hr))
354 return hr;
355
356 hr = SHBindToParent(absolutePIDL, IID_PPV_ARG(IShellFolder, &sf), &pidlChild);
357 if (FAILED_UNEXPECTEDLY(hr))
358 return hr;
359
360 hr = sf->GetDisplayNameOf(pidlChild, SHGDN_FORADDRESSBAR | SHGDN_FORPARSING, &ret);
361 if (FAILED_UNEXPECTEDLY(hr))
362 return hr;
363
364 hr = StrRetToBufW(&ret, pidlChild, buf, 4095);
365 if (FAILED_UNEXPECTEDLY(hr))
366 return hr;
367
368 indexClosed = SHMapPIDLToSystemImageListIndex(sf, pidlChild, &indexOpen);
369
370 item.mask = CBEIF_IMAGE | CBEIF_SELECTEDIMAGE | CBEIF_TEXT | CBEIF_LPARAM;
371 item.iItem = 0;
372 item.iImage = indexClosed;
373 item.iSelectedImage = indexOpen;
374 item.pszText = buf;
375 item.lParam = reinterpret_cast<LPARAM>(absolutePIDL);
376
377 if (itemExists)
378 {
379 result = fCombobox.SendMessage(CBEM_SETITEM, 0, reinterpret_cast<LPARAM>(&item));
380 oldIndex = 0;
381
382 if (result)
383 {
384 ILFree(pidlPrevious);
385 }
386 }
387 else
388 {
389 oldIndex = fCombobox.SendMessage(CBEM_INSERTITEM, 0, reinterpret_cast<LPARAM>(&item));
390
391 if (oldIndex < 0)
392 DbgPrint("ERROR %d\n", GetLastError());
393 }
394
395 fCombobox.SendMessage(CB_SETCURSEL, -1, 0);
396 fCombobox.SendMessage(CB_SETCURSEL, oldIndex, 0);
397
398 //fAddressEditBox->SetCurrentDir(index);
399 }
400 return S_OK;
401 }
402
403 HRESULT STDMETHODCALLTYPE CAddressEditBox::GetClassID(CLSID *pClassID)
404 {
405 if (pClassID == NULL)
406 return E_POINTER;
407 *pClassID = CLSID_AddressEditBox;
408 return S_OK;
409 }
410
411 HRESULT STDMETHODCALLTYPE CAddressEditBox::IsDirty()
412 {
413 return E_NOTIMPL;
414 }
415
416 HRESULT STDMETHODCALLTYPE CAddressEditBox::Load(IStream *pStm)
417 {
418 return E_NOTIMPL;
419 }
420
421 HRESULT STDMETHODCALLTYPE CAddressEditBox::Save(IStream *pStm, BOOL fClearDirty)
422 {
423 return E_NOTIMPL;
424 }
425
426 HRESULT STDMETHODCALLTYPE CAddressEditBox::GetSizeMax(ULARGE_INTEGER *pcbSize)
427 {
428 return E_NOTIMPL;
429 }