[BROWSEUI] CISFBand: Misc fixes
[reactos.git] / dll / win32 / browseui / shellbars / CISFBand.cpp
1 /*
2 * PROJECT: ReactOS shell extensions
3 * LICENSE: GPL - See COPYING in the top level directory
4 * FILE: dll/shellext/qcklnch/CISFBand.cpp
5 * PURPOSE: Quick Launch Toolbar (Taskbar Shell Extension)
6 * PROGRAMMERS: Shriraj Sawant a.k.a SR13 <sr.official@hotmail.com>
7 */
8
9 #include "shellbars.h"
10 #include <commoncontrols.h>
11 #include <strsafe.h>
12
13 #define GET_X_LPARAM(lp) ((int)(short)LOWORD(lp))
14 #define GET_Y_LPARAM(lp) ((int)(short)HIWORD(lp))
15
16 // ***Extras***
17 /*++
18 * @name _ILIsDesktop
19 *
20 * Checks whether the given PIDL is of Desktop folder or not.
21 *
22 * @param pidl
23 * PIDL to be checked.
24 *
25 * @return True if PIDL is of Desktop, otherwise false.
26 *
27 *--*/
28 static BOOL _ILIsDesktop(LPCITEMIDLIST pidl)
29 {
30 return (pidl == NULL || pidl->mkid.cb == 0);
31 }
32
33 //*****************************************************************************************
34 // *** CISFBand ***
35
36 CISFBand::CISFBand() :
37 m_BandID(0),
38 m_pidl(NULL),
39 m_textFlag(true),
40 m_iconFlag(true),
41 m_QLaunch(false)
42 {
43 }
44
45 CISFBand::~CISFBand()
46 {
47 CloseDW(0);
48 }
49
50 // Toolbar
51 /*++
52 * @name CreateSimpleToolbar
53 *
54 * Creates a toolbar and fills it up with buttons for enumerated objects.
55 *
56 * @param hWndParent
57 * Handle to the parent window, which receives the appropriate messages from child toolbar.
58 *
59 * @return The error code.
60 *
61 *--*/
62 HRESULT CISFBand::CreateSimpleToolbar(HWND hWndParent)
63 {
64 // Declare and initialize local constants.
65 const DWORD buttonStyles = BTNS_AUTOSIZE;
66
67 // Create the toolbar.
68 m_hWnd = CreateWindowEx(0, TOOLBARCLASSNAME, NULL,
69 WS_CHILD | TBSTYLE_FLAT | TBSTYLE_LIST | CCS_NORESIZE | CCS_NODIVIDER, CW_USEDEFAULT, CW_USEDEFAULT, 0, 0,
70 hWndParent, NULL, 0, NULL);
71 if (m_hWnd == NULL)
72 return E_FAIL;
73
74 // Set the image list.
75 HIMAGELIST* piml;
76 HRESULT hr = SHGetImageList(SHIL_SMALL, IID_IImageList, (void**)&piml);
77 if (FAILED_UNEXPECTEDLY(hr))
78 {
79 DestroyWindow();
80 return hr;
81 }
82 SendMessage(m_hWnd, TB_SETIMAGELIST, 0, (LPARAM)piml);
83
84 // Enumerate objects
85 CComPtr<IEnumIDList> pEndl;
86 LPITEMIDLIST pidl;
87 STRRET stret;
88 hr = m_pISF->EnumObjects(0, SHCONTF_FOLDERS|SHCONTF_NONFOLDERS, &pEndl);
89 if (FAILED_UNEXPECTEDLY(hr))
90 {
91 DestroyWindow();
92 return hr;
93 }
94
95 for (int i=0; pEndl->Next(1, &pidl, NULL) != S_FALSE; i++)
96 {
97 WCHAR sz[MAX_PATH];
98 int index = SHMapPIDLToSystemImageListIndex(m_pISF, pidl, NULL);
99 hr = m_pISF->GetDisplayNameOf(pidl, SHGDN_NORMAL, &stret);
100 if (FAILED_UNEXPECTEDLY(hr))
101 {
102 StringCchCopyW(sz, MAX_PATH, L"<Unknown-Name>");
103 }
104 else
105 StrRetToBuf(&stret, pidl, sz, _countof(sz));
106
107 TBBUTTON tb = { MAKELONG(index, 0), i, TBSTATE_ENABLED, buttonStyles,{ 0 }, (DWORD_PTR)pidl, (INT_PTR)sz };
108 SendMessage(m_hWnd, TB_INSERTBUTTONW, i, (LPARAM)&tb);
109 }
110
111 // Resize the toolbar, and then show it.
112 SendMessage(m_hWnd, TB_AUTOSIZE, 0, 0);
113
114 return hr;
115 }
116
117 /*****************************************************************************/
118
119 // *** IObjectWithSite ***
120 STDMETHODIMP CISFBand::SetSite(IUnknown *pUnkSite)
121 {
122 HRESULT hr;
123 HWND hwndParent;
124
125 TRACE("CISFBand::SetSite(0x%p)\n", pUnkSite);
126
127 hr = IUnknown_GetWindow(pUnkSite, &hwndParent);
128 if (FAILED(hr))
129 {
130 TRACE("Querying site window failed: 0x%x\n", hr);
131 return hr;
132 }
133 m_Site = pUnkSite;
134
135 hr = CreateSimpleToolbar(hwndParent);
136 if (FAILED_UNEXPECTEDLY(hr))
137 return hr;
138
139 return S_OK;
140 }
141
142 STDMETHODIMP CISFBand::GetSite(IN REFIID riid, OUT VOID **ppvSite)
143 {
144 TRACE("CISFBand::GetSite(0x%p,0x%p)\n", riid, ppvSite);
145
146 HRESULT hr;
147 if (m_Site != NULL)
148 {
149 hr = m_Site->QueryInterface(riid, ppvSite);
150 if (FAILED(hr)) return hr;
151 }
152
153 *ppvSite = NULL;
154 return E_FAIL;
155 }
156
157 /*****************************************************************************/
158 // *** IDeskBand ***
159 STDMETHODIMP CISFBand::GetWindow(OUT HWND *phwnd)
160 {
161 if (!m_hWnd)
162 return E_FAIL;
163 if (!phwnd)
164 return E_POINTER;
165 *phwnd = m_hWnd;
166
167 return S_OK;
168 }
169
170 STDMETHODIMP CISFBand::ContextSensitiveHelp(IN BOOL fEnterMode)
171 {
172 /* FIXME: Implement */
173 return E_NOTIMPL;
174 }
175
176 STDMETHODIMP CISFBand::ShowDW(IN BOOL bShow)
177 {
178 if (m_hWnd)
179 {
180 ShowWindow(bShow ? SW_SHOW : SW_HIDE);
181 return S_OK;
182 }
183
184 return E_FAIL;
185 }
186
187 STDMETHODIMP CISFBand::CloseDW(IN DWORD dwReserved)
188 {
189 if (m_hWnd)
190 {
191 ShowWindow(SW_HIDE);
192
193 TBBUTTON tb;
194 for (int i = 0; SendMessage(m_hWnd, TB_GETBUTTON, i, (LPARAM)&tb); i++)
195 {
196 CoTaskMemFree((LPITEMIDLIST)tb.dwData);
197 }
198
199 DestroyWindow();
200 m_hWnd = NULL;
201 return S_OK;
202 }
203
204 return E_FAIL;
205 }
206
207 STDMETHODIMP CISFBand::ResizeBorderDW(LPCRECT prcBorder, IUnknown *punkToolbarSite, BOOL fReserved)
208 {
209 /* No need to implement this method */
210
211 return E_NOTIMPL;
212 }
213
214 STDMETHODIMP CISFBand::GetBandInfo(IN DWORD dwBandID, IN DWORD dwViewMode, IN OUT DESKBANDINFO *pdbi)
215 {
216 TRACE("CTaskBand::GetBandInfo(0x%x,0x%x,0x%p) hWnd=0x%p\n", dwBandID, dwViewMode, pdbi, m_hWnd);
217
218 if (m_hWnd && pdbi)
219 {
220 m_BandID = dwBandID;
221
222 RECT actualRect;
223 POINTL actualSize;
224 POINTL idealSize;
225 POINTL maxSize;
226 POINTL itemSize;
227
228 GetWindowRect(&actualRect);
229 actualSize.x = actualRect.right - actualRect.left;
230 actualSize.y = actualRect.bottom - actualRect.top;
231
232 // Obtain the ideal size, to be used as min and max
233 SendMessageW(m_hWnd, TB_AUTOSIZE, 0, 0);
234 SendMessageW(m_hWnd, TB_GETMAXSIZE, 0, reinterpret_cast<LPARAM>(&maxSize));
235
236 idealSize = maxSize;
237 SendMessageW(m_hWnd, TB_GETIDEALSIZE, FALSE, reinterpret_cast<LPARAM>(&idealSize));
238
239 // Obtain the button size, to be used as the integral size
240 DWORD size = SendMessageW(m_hWnd, TB_GETBUTTONSIZE, 0, 0);
241 itemSize.x = GET_X_LPARAM(size);
242 itemSize.y = GET_Y_LPARAM(size);
243
244 if (pdbi->dwMask & DBIM_MINSIZE)
245 {
246 pdbi->ptMinSize.x = -1;
247 pdbi->ptMinSize.y = idealSize.y;
248 }
249 if (pdbi->dwMask & DBIM_MAXSIZE)
250 {
251 pdbi->ptMaxSize = maxSize;
252 }
253 if (pdbi->dwMask & DBIM_INTEGRAL)
254 {
255 pdbi->ptIntegral = itemSize;
256 }
257 if (pdbi->dwMask & DBIM_ACTUAL)
258 {
259 pdbi->ptActual = actualSize;
260 }
261 if (pdbi->dwMask & DBIM_TITLE)
262 {
263 if (!ILGetDisplayNameEx(NULL, m_pidl, pdbi->wszTitle, ILGDN_INFOLDER))
264 {
265 pdbi->dwMask &= ~DBIM_TITLE;
266 }
267 }
268 if (pdbi->dwMask & DBIM_MODEFLAGS)
269 {
270 pdbi->dwModeFlags = DBIMF_NORMAL | DBIMF_VARIABLEHEIGHT | DBIMF_USECHEVRON | DBIMF_NOMARGINS | DBIMF_BKCOLOR | DBIMF_ADDTOFRONT;
271 }
272 if (pdbi->dwMask & DBIM_BKCOLOR)
273 pdbi->dwMask &= ~DBIM_BKCOLOR;
274
275 return S_OK;
276 }
277
278 return E_FAIL;
279 }
280
281 /*****************************************************************************/
282 // *** IPersistStream ***
283 STDMETHODIMP CISFBand::GetClassID(OUT CLSID *pClassID)
284 {
285 *pClassID = CLSID_ISFBand;
286
287 return S_OK;
288 }
289
290 STDMETHODIMP CISFBand::IsDirty()
291 {
292 /* The object hasn't changed since the last save! */
293
294 return S_FALSE;
295 }
296
297 STDMETHODIMP CISFBand::Load(IN IStream *pStm)
298 {
299 TRACE("CISFBand::Load called\n");
300 /* Nothing to do */
301
302 return S_OK;
303 }
304
305 STDMETHODIMP CISFBand::Save(IN IStream *pStm, IN BOOL fClearDirty)
306 {
307 /* Nothing to do */
308
309 return S_OK;
310 }
311
312 STDMETHODIMP CISFBand::GetSizeMax(OUT ULARGE_INTEGER *pcbSize)
313 {
314 TRACE("CISFBand::GetSizeMax called\n");
315
316 return S_OK;
317 }
318
319 /*****************************************************************************/
320 // *** IWinEventHandler ***
321 STDMETHODIMP CISFBand::ContainsWindow(IN HWND hWnd)
322 {
323 if (hWnd == m_hWnd || IsChild(hWnd))
324 {
325 TRACE("CISFBand::ContainsWindow(0x%p) returns S_OK\n", hWnd);
326 return S_OK;
327 }
328
329 return S_FALSE;
330 }
331
332 STDMETHODIMP CISFBand::OnWinEvent(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *theResult)
333 {
334 switch (uMsg)
335 {
336 case WM_COMMAND:
337 {
338 TBBUTTON tb;
339 bool chk = SendMessage(m_hWnd, TB_GETBUTTON, LOWORD(wParam), (LPARAM)&tb);
340 if (chk)
341 SHInvokeDefaultCommand(m_hWnd, m_pISF, (LPITEMIDLIST)tb.dwData);
342
343 *theResult = TRUE;
344 break;
345 }
346 case WM_NOTIFY:
347 {
348 switch (((LPNMHDR)lParam)->code)
349 {
350 case NM_RCLICK:
351 {
352 HRESULT hr;
353 POINT pt = ((LPNMMOUSE)lParam)->pt;
354 CComPtr<IContextMenu> picm;
355 HMENU fmenu = CreatePopupMenu();
356 TBBUTTON tb;
357
358 bool chk = SendMessage(m_hWnd, TB_GETBUTTON, ((LPNMMOUSE)lParam)->dwItemSpec, (LPARAM)&tb);
359 LPITEMIDLIST pidl = (LPITEMIDLIST)tb.dwData;
360
361 if (chk)
362 {
363 ClientToScreen(&pt);
364 hr = m_pISF->GetUIObjectOf(m_hWnd, 1, &pidl, IID_NULL_PPV_ARG(IContextMenu, &picm));
365 if (FAILED_UNEXPECTEDLY(hr))
366 return hr;
367
368 hr = picm->QueryContextMenu(fmenu, 0, 1, 0x7FFF, CMF_DEFAULTONLY);
369 if (FAILED_UNEXPECTEDLY(hr))
370 return hr;
371
372 int id = TrackPopupMenuEx(fmenu, TPM_LEFTALIGN | TPM_BOTTOMALIGN | TPM_RETURNCMD, pt.x, pt.y, m_hWnd, 0);
373 if (id > 0)
374 {
375 CMINVOKECOMMANDINFOEX info = { 0 };
376 info.cbSize = sizeof(info);
377 info.fMask = CMIC_MASK_PTINVOKE;
378 if (GetKeyState(VK_CONTROL) < 0)
379 {
380 info.fMask |= CMIC_MASK_CONTROL_DOWN;
381 }
382 if (GetKeyState(VK_SHIFT) < 0)
383 {
384 info.fMask |= CMIC_MASK_SHIFT_DOWN;
385 }
386 info.hwnd = m_hWnd;
387 info.lpVerb = MAKEINTRESOURCEA(id - 1);
388 info.nShow = SW_SHOWNORMAL;
389 info.ptInvoke = pt;
390 picm->InvokeCommand((LPCMINVOKECOMMANDINFO)&info);
391 }
392 }
393 DestroyMenu(fmenu);
394
395 *theResult = TRUE;
396 break;
397 }
398 default:
399 *theResult = FALSE;
400 }
401
402 break;
403 }
404 default:
405 *theResult = FALSE;
406 }
407
408 return S_OK;
409 }
410
411 STDMETHODIMP CISFBand::IsWindowOwner(HWND hWnd)
412 {
413 return (hWnd == m_hWnd) ? S_OK : S_FALSE;
414 }
415
416 /*****************************************************************************/
417 // *** IOleCommandTarget methods ***
418 STDMETHODIMP CISFBand::QueryStatus(const GUID *pguidCmdGroup, ULONG cCmds, OLECMD prgCmds [], OLECMDTEXT *pCmdText)
419 {
420 UNIMPLEMENTED;
421
422 return E_NOTIMPL;
423 }
424
425 STDMETHODIMP CISFBand::Exec(const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANT *pvaIn, VARIANT *pvaOut)
426 {
427 if (IsEqualIID(*pguidCmdGroup, IID_IBandSite))
428 {
429 return S_OK;
430 }
431
432 if (IsEqualIID(*pguidCmdGroup, IID_IDeskBand))
433 {
434 return S_OK;
435 }
436
437 UNIMPLEMENTED;
438
439 return E_NOTIMPL;
440 }
441
442 /*****************************************************************************/
443 // *** IShellFolderBand ***
444 STDMETHODIMP CISFBand::GetBandInfoSFB(PBANDINFOSFB pbi)
445 {
446 if (pbi->dwMask == ISFB_MASK_IDLIST)
447 {
448 pbi->pidl = ILClone(m_pidl);
449 if (!pbi->pidl)
450 return E_OUTOFMEMORY;
451 return S_OK;
452 }
453
454 return E_NOTIMPL;
455 }
456
457 STDMETHODIMP CISFBand::InitializeSFB(IShellFolder *psf, PCIDLIST_ABSOLUTE pidl)
458 {
459 HRESULT hr;
460
461 if (!psf && !pidl)
462 return E_INVALIDARG;
463
464 if (psf && pidl)
465 return E_INVALIDARG;
466
467 if (pidl != NULL)
468 {
469 CComPtr<IShellFolder> psfDesktop;
470 hr = SHGetDesktopFolder(&psfDesktop);
471 if (FAILED_UNEXPECTEDLY(hr))
472 return hr;
473
474 if (_ILIsDesktop(pidl))
475 {
476 m_pISF = psfDesktop;
477 }
478 else
479 {
480 hr = psfDesktop->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder, &m_pISF));
481 if (FAILED_UNEXPECTEDLY(hr))
482 return hr;
483 }
484
485 m_pidl = ILClone(pidl);
486 }
487
488 if (psf != NULL)
489 {
490 CComPtr<IPersistFolder2> ppf2;
491 hr = psf->QueryInterface(IID_PPV_ARG(IPersistFolder2, &ppf2));
492 if (FAILED_UNEXPECTEDLY(hr))
493 return hr;
494
495 hr = ppf2->GetCurFolder(&m_pidl);
496 if (FAILED_UNEXPECTEDLY(hr))
497 return hr;
498
499 m_pISF = psf;
500 }
501
502 return S_OK;
503 }
504
505 STDMETHODIMP CISFBand::SetBandInfoSFB( PBANDINFOSFB pbi)
506 {
507 if ((pbi->dwMask & ISFB_MASK_STATE) &&
508 (pbi->dwState & ISFB_STATE_QLINKSMODE) &&
509 (pbi->dwStateMask & ISFB_STATE_QLINKSMODE))
510 {
511 m_QLaunch = true;
512 }
513
514 return E_NOTIMPL;
515 }
516
517 /*****************************************************************************/
518 // *** IContextMenu ***
519 STDMETHODIMP CISFBand::GetCommandString(UINT_PTR idCmd, UINT uFlags, UINT *pwReserved, LPSTR pszName, UINT cchMax)
520 {
521 /*HRESULT hr = E_INVALIDARG;
522
523 if (idCmd == IDM_DISPLAY)
524 {
525 switch (uFlags)
526 {
527 case GCS_HELPTEXTW:
528 // Only useful for pre-Vista versions of Windows that
529 // have a Status bar.
530 hr = StringCchCopyW(reinterpret_cast<PWSTR>(pszName),
531 cchMax,
532 L"Display File Name");
533 break;
534
535 case GCS_VERBW:
536 // GCS_VERBW is an optional feature that enables a caller
537 // to discover the canonical name for the verb that is passed in
538 // through idCommand.
539 hr = StringCchCopyW(reinterpret_cast<PWSTR>(pszName),
540 cchMax,
541 L"DisplayFileName");
542 break;
543 }
544 }
545 return hr; */
546
547 return S_OK;
548 }
549
550 STDMETHODIMP CISFBand::InvokeCommand(LPCMINVOKECOMMANDINFO pici)
551 {
552 if (!HIWORD(pici->lpVerb))
553 {
554 switch (LOWORD(pici->lpVerb))
555 {
556 case IDM_LARGE_ICONS:
557 {
558 m_iconFlag = false;
559
560 HIMAGELIST* piml = (HIMAGELIST*) SendMessage(m_hWnd, TB_GETIMAGELIST, 0, 0);
561 HRESULT hr = SHGetImageList(SHIL_LARGE, IID_IImageList, (void**)&piml);
562 if (FAILED_UNEXPECTEDLY(hr)) return hr;
563 SendMessage(m_hWnd, TB_SETIMAGELIST, 0, (LPARAM)piml);
564 hr = IUnknown_Exec(m_Site, IID_IDeskBand, DBID_BANDINFOCHANGED, 0, NULL, NULL);
565 if (FAILED_UNEXPECTEDLY(hr)) return hr;
566 break;
567 }
568 case IDM_SMALL_ICONS:
569 {
570 m_iconFlag = true;
571
572 HIMAGELIST* piml = (HIMAGELIST*)SendMessage(m_hWnd, TB_GETIMAGELIST, 0, 0);
573 HRESULT hr = SHGetImageList(SHIL_SMALL, IID_IImageList, (void**)&piml);
574 if (FAILED_UNEXPECTEDLY(hr)) return hr;
575 SendMessage(m_hWnd, TB_SETIMAGELIST, 0, (LPARAM)piml);
576 hr = IUnknown_Exec(m_Site, IID_IDeskBand, DBID_BANDINFOCHANGED, 0, NULL, NULL);
577 if (FAILED_UNEXPECTEDLY(hr)) return hr;
578 break;
579 }
580 case IDM_SHOW_TEXT:
581 {
582 if (m_textFlag)
583 {
584 m_textFlag = false;
585 SendMessage(m_hWnd, TB_SETEXTENDEDSTYLE, 0, TBSTYLE_EX_MIXEDBUTTONS);
586 HRESULT hr = IUnknown_Exec(m_Site, IID_IDeskBand, DBID_BANDINFOCHANGED, 0, NULL, NULL);
587 if (FAILED_UNEXPECTEDLY(hr)) return hr;
588 }
589 else
590 {
591 m_textFlag = true;
592 SendMessage(m_hWnd, TB_SETEXTENDEDSTYLE, 0, 0);
593 HRESULT hr = IUnknown_Exec(m_Site, IID_IDeskBand, DBID_BANDINFOCHANGED, 0, NULL, NULL);
594 if (FAILED_UNEXPECTEDLY(hr)) return hr;
595 }
596 break;
597 }
598 default:
599 return E_FAIL;
600 }
601 }
602
603 return S_OK;
604 }
605
606 STDMETHODIMP CISFBand::QueryContextMenu(HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
607 {
608 HMENU qMenu = LoadMenu(GetModuleHandleW(L"browseui.dll"), MAKEINTRESOURCE(IDM_POPUPMENU));
609
610 if(m_textFlag)
611 CheckMenuItem(qMenu, IDM_SHOW_TEXT, MF_CHECKED);
612 else
613 CheckMenuItem(qMenu, IDM_SHOW_TEXT, MF_UNCHECKED);
614
615 if (m_iconFlag)
616 {
617 CheckMenuItem(qMenu, IDM_SMALL_ICONS, MF_CHECKED);
618 CheckMenuItem(qMenu, IDM_LARGE_ICONS, MF_UNCHECKED);
619 }
620 else
621 {
622 CheckMenuItem(qMenu, IDM_LARGE_ICONS, MF_CHECKED);
623 CheckMenuItem(qMenu, IDM_SMALL_ICONS, MF_UNCHECKED);
624 }
625
626 UINT idMax = Shell_MergeMenus(hmenu, GetSubMenu(qMenu, 0), indexMenu, idCmdFirst, idCmdLast, MM_SUBMENUSHAVEIDS);
627 DestroyMenu(qMenu);
628 return MAKE_HRESULT(SEVERITY_SUCCESS, 0, USHORT(idMax - idCmdFirst +1));
629 }
630
631 /*****************************************************************************/
632 // C Constructor
633 extern "C"
634 HRESULT WINAPI RSHELL_CISFBand_CreateInstance(REFIID riid, void** ppv)
635 {
636 return ShellObjectCreator<CISFBand>(riid, ppv);
637 }
638