[QCKLNCH] Remove qcklnch (#61)
[reactos.git] / base / shell / rshell / CQuickLaunchBand.cpp
1 /*
2 * PROJECT: ReactOS shell extensions
3 * LICENSE: GPL - See COPYING in the top level directory
4 * FILE: dll/shellext/qcklnch/CQuickLaunchBand.cpp
5 * PURPOSE: Quick Launch Toolbar (Taskbar Shell Extension)
6 * PROGRAMMERS: Shriraj Sawant a.k.a SR13 <sr.official@hotmail.com>
7 */
8
9
10 #define WIN32_NO_STATUS
11 #define _INC_WINDOWS
12 #define COM_NO_WINDOWS_H
13
14 #define COBJMACROS
15
16 #include <windef.h>
17 #include <winbase.h>
18 #include <winreg.h>
19 #include <wingdi.h>
20 #include <winnls.h>
21 #include <wincon.h>
22 #include <shellapi.h>
23 #include <shlobj.h>
24 #include <shlobj_undoc.h>
25 #include <shlwapi.h>
26 #include <shlguid_undoc.h>
27 #include <atlbase.h>
28 #include <atlcom.h>
29 #include <atlwin.h>
30
31 #include <undocshell.h>
32 #include <shellutils.h>
33
34 #include "CQuickLaunchBand.h"
35
36 extern "C"
37 HRESULT WINAPI CISFBand_CreateInstance(REFIID riid, void** ppv);
38
39 // {260CB95D-4544-44F6-A079-575BAA60B72F}
40 const GUID CLSID_QuickLaunchBand = { 0x260cb95d, 0x4544, 0x44f6, { 0xa0, 0x79, 0x57, 0x5b, 0xaa, 0x60, 0xb7, 0x2f } };
41
42 // Componenet Category Registration
43 HRESULT RegisterComCat()
44 {
45 CComPtr<ICatRegister> pcr;
46 HRESULT hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(ICatRegister, &pcr));
47 if (SUCCEEDED(hr))
48 {
49 CATID catid = CATID_DeskBand;
50 hr = pcr->RegisterClassImplCategories(CLSID_QuickLaunchBand, 1, &catid);
51 }
52 return hr;
53 }
54
55 HRESULT UnregisterComCat()
56 {
57 CComPtr<ICatRegister> pcr;
58 HRESULT hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(ICatRegister, &pcr));
59 if (SUCCEEDED(hr))
60 {
61 CATID catid = CATID_DeskBand;
62 hr = pcr->UnRegisterClassImplCategories(CLSID_QuickLaunchBand, 1, &catid);
63 }
64 return hr;
65 }
66
67 // Pidl Browser
68 /*++
69 * @name PidlBrowse
70 *
71 * Opens a folder browser dialog,
72 * allowing the user to select a folder for enumeration.
73 *
74 * @param hwnd
75 * A handle to browser dialog window.
76 * @param nCSIDL
77 * A CSIDL representing the root from which the browse folder dialog shows the files and folders.
78 *
79 * @return The PIDL to selected folder.
80 *
81 *--*/
82 LPITEMIDLIST PidlBrowse(HWND hwnd, int nCSIDL)
83 {
84 CComHeapPtr<ITEMIDLIST> pidlRoot;
85
86 WCHAR path[MAX_PATH];
87
88 if (nCSIDL)
89 {
90 SHGetSpecialFolderLocation(hwnd, nCSIDL, &pidlRoot);
91 }
92
93 BROWSEINFO bi = { hwnd, pidlRoot, path, L"Choose a folder", 0, NULL, 0, 0 };
94 LPITEMIDLIST pidlSelected = SHBrowseForFolder(&bi);
95
96 return pidlSelected;
97 }
98
99 // CQuickLaunchBand
100
101 CQuickLaunchBand::CQuickLaunchBand() {}
102
103 CQuickLaunchBand::~CQuickLaunchBand() {}
104
105 /*****************************************************************************/
106 // ATL Construct
107 /*++
108 * @name FinalConstruct
109 *
110 * Creates an instance of CISFBand, and initializes its Shell Folder Band for enumeration.
111 *
112 * @return The error code.
113 *
114 *--*/
115 HRESULT CQuickLaunchBand::FinalConstruct()
116 {
117 HRESULT hr = CISFBand_CreateInstance(IID_PPV_ARG(IUnknown, &m_punkISFB));
118 if (FAILED_UNEXPECTEDLY(hr))
119 return hr;
120
121 CComPtr<IShellFolderBand> pISFB;
122 hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IShellFolderBand, &pISFB));
123 if (FAILED_UNEXPECTEDLY(hr))
124 return hr;
125
126 CComPtr<IShellFolder> pISF;
127 hr = SHGetDesktopFolder(&pISF);
128 if (FAILED_UNEXPECTEDLY(hr))
129 return hr;
130
131 CComHeapPtr<ITEMIDLIST> pidl(PidlBrowse(m_hWndBro, CSIDL_DESKTOP));
132 if (pidl == NULL)
133 return E_FAIL;
134 pISFB->InitializeSFB(pISF, pidl);
135
136 return hr;
137 }
138
139 // IObjectWithSite
140 STDMETHODIMP CQuickLaunchBand::SetSite(IUnknown *pUnkSite)
141 {
142 // Internal CISFBand Calls
143 CComPtr<IObjectWithSite> pIOWS;
144 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IObjectWithSite, &pIOWS));
145 if (FAILED(hr))
146 return hr;
147
148 return pIOWS->SetSite(pUnkSite);
149 }
150
151 STDMETHODIMP CQuickLaunchBand::GetSite(IN REFIID riid, OUT VOID **ppvSite)
152 {
153 CComPtr<IObjectWithSite> pIOWS;
154 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IObjectWithSite, &pIOWS));
155 if (FAILED(hr))
156 return hr;
157
158 return pIOWS->GetSite(riid, ppvSite);
159 }
160
161 /*****************************************************************************/
162 // IDeskBand
163 STDMETHODIMP CQuickLaunchBand::GetWindow(OUT HWND *phwnd)
164 {
165 // Internal CISFBand Calls
166 CComPtr<IDeskBand> pIDB;
167 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IDeskBand, &pIDB));
168 if (FAILED(hr))
169 return hr;
170
171 return pIDB->GetWindow(phwnd);
172 }
173
174 STDMETHODIMP CQuickLaunchBand::ContextSensitiveHelp(IN BOOL fEnterMode)
175 {
176 // Internal CISFBand Calls
177 CComPtr<IDeskBand> pIDB;
178 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IDeskBand, &pIDB));
179 if (FAILED(hr))
180 return hr;
181
182 return pIDB->ContextSensitiveHelp(fEnterMode);
183 }
184
185 STDMETHODIMP CQuickLaunchBand::ShowDW(IN BOOL bShow)
186 {
187 // Internal CISFBand Calls
188 CComPtr<IDeskBand> pIDB;
189 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IDeskBand, &pIDB));
190 if (FAILED(hr))
191 return hr;
192
193 return pIDB->ShowDW(bShow);
194 }
195
196 STDMETHODIMP CQuickLaunchBand::CloseDW(IN DWORD dwReserved)
197 {
198 // Internal CISFBand Calls
199 CComPtr<IDeskBand> pIDB;
200 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IDeskBand, &pIDB));
201 if (FAILED(hr))
202 return hr;
203
204 return pIDB->CloseDW(dwReserved);
205 }
206
207 STDMETHODIMP CQuickLaunchBand::ResizeBorderDW(LPCRECT prcBorder, IUnknown *punkToolbarSite, BOOL fReserved)
208 {
209 // Internal CISFBand Calls
210 CComPtr<IDeskBand> pIDB;
211 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IDeskBand, &pIDB));
212 if (FAILED(hr))
213 return hr;
214
215 return pIDB->ResizeBorderDW(prcBorder, punkToolbarSite, fReserved);
216 }
217
218 STDMETHODIMP CQuickLaunchBand::GetBandInfo(IN DWORD dwBandID, IN DWORD dwViewMode, IN OUT DESKBANDINFO *pdbi)
219 {
220 // Internal CISFBand Calls
221 CComPtr<IDeskBand> pIDB;
222 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IDeskBand, &pIDB));
223 if (FAILED(hr))
224 return hr;
225
226 return pIDB->GetBandInfo(dwBandID, dwViewMode, pdbi);
227 }
228
229 /*****************************************************************************/
230 // IPersistStream
231 STDMETHODIMP CQuickLaunchBand::GetClassID(OUT CLSID *pClassID)
232 {
233 // Internal CISFBand Calls
234 CComPtr<IPersistStream> pIPS;
235 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IPersistStream, &pIPS));
236 if (FAILED(hr))
237 return hr;
238
239 return pIPS->GetClassID(pClassID);
240 }
241
242 STDMETHODIMP CQuickLaunchBand::IsDirty()
243 {
244 // Internal CISFBand Calls
245 CComPtr<IPersistStream> pIPS;
246 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IPersistStream, &pIPS));
247 if (FAILED(hr))
248 return hr;
249
250 return pIPS->IsDirty();
251 }
252
253 STDMETHODIMP CQuickLaunchBand::Load(IN IStream *pStm)
254 {
255 CComPtr<IPersistStream> pIPS;
256 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IPersistStream, &pIPS));
257 if (FAILED(hr))
258 return hr;
259
260 return pIPS->Load(pStm);
261 }
262
263 STDMETHODIMP CQuickLaunchBand::Save(IN IStream *pStm, IN BOOL fClearDirty)
264 {
265 // Internal CISFBand Calls
266 CComPtr<IPersistStream> pIPS;
267 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IPersistStream, &pIPS));
268 if (FAILED(hr))
269 return hr;
270
271 return pIPS->Save(pStm, fClearDirty);
272 }
273
274 STDMETHODIMP CQuickLaunchBand::GetSizeMax(OUT ULARGE_INTEGER *pcbSize)
275 {
276 CComPtr<IPersistStream> pIPS;
277 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IPersistStream, &pIPS));
278 if (FAILED(hr))
279 return hr;
280
281 return pIPS->GetSizeMax(pcbSize);
282 }
283
284
285 /*****************************************************************************/
286 // IWinEventHandler
287 STDMETHODIMP CQuickLaunchBand::ContainsWindow(IN HWND hWnd)
288 {
289 return E_NOTIMPL;
290 }
291
292 STDMETHODIMP CQuickLaunchBand::OnWinEvent(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *theResult)
293 {
294 // Internal CISFBand Calls
295 CComPtr<IWinEventHandler> pWEH;
296 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IWinEventHandler, &pWEH));
297 if (FAILED(hr))
298 return hr;
299
300 return pWEH->OnWinEvent(hWnd, uMsg, wParam, lParam, theResult);
301 }
302
303 STDMETHODIMP CQuickLaunchBand::IsWindowOwner(HWND hWnd)
304 {
305 // Internal CISFBand Calls
306 CComPtr<IWinEventHandler> pWEH;
307 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IWinEventHandler, &pWEH));
308 if (FAILED(hr))
309 return hr;
310
311 return pWEH->IsWindowOwner(hWnd);
312 }
313
314 /*****************************************************************************/
315 // *** IOleCommandTarget methods ***
316 STDMETHODIMP CQuickLaunchBand::QueryStatus(const GUID *pguidCmdGroup, ULONG cCmds, OLECMD prgCmds [], OLECMDTEXT *pCmdText)
317 {
318 // Internal CISFBand Calls
319 CComPtr<IOleCommandTarget> pOCT;
320 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IOleCommandTarget, &pOCT));
321 if (FAILED(hr))
322 return hr;
323
324 return pOCT->QueryStatus(pguidCmdGroup, cCmds, prgCmds, pCmdText);
325 }
326
327 STDMETHODIMP CQuickLaunchBand::Exec(const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANT *pvaIn, VARIANT *pvaOut)
328 {
329 // Internal CISFBand Calls
330 CComPtr<IOleCommandTarget> pOCT;
331 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IOleCommandTarget, &pOCT));
332 if (FAILED(hr))
333 return hr;
334
335 return pOCT->Exec(pguidCmdGroup, nCmdID, nCmdexecopt, pvaIn, pvaOut);
336 }
337
338 /*****************************************************************************/
339 // *** IContextMenu ***
340 STDMETHODIMP CQuickLaunchBand::GetCommandString(UINT_PTR idCmd, UINT uFlags, UINT *pwReserved, LPSTR pszName, UINT cchMax)
341 {
342 // Internal CISFBand Calls
343 CComPtr<IContextMenu> pICM;
344 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IContextMenu, &pICM));
345 if (FAILED(hr))
346 return hr;
347
348 return pICM->GetCommandString(idCmd, uFlags, pwReserved, pszName, cchMax);
349 }
350
351 STDMETHODIMP CQuickLaunchBand::InvokeCommand(LPCMINVOKECOMMANDINFO pici)
352 {
353 // Internal CISFBand Calls
354 CComPtr<IContextMenu> pICM;
355 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IContextMenu, &pICM));
356 if (FAILED(hr))
357 return hr;
358
359 return pICM->InvokeCommand(pici);
360 }
361
362 STDMETHODIMP CQuickLaunchBand::QueryContextMenu(HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
363 {
364 // Internal CISFBand Calls
365 CComPtr<IContextMenu> pICM;
366 HRESULT hr = m_punkISFB->QueryInterface(IID_PPV_ARG(IContextMenu, &pICM));
367 if (FAILED(hr))
368 return hr;
369
370 return pICM->QueryContextMenu(hmenu, indexMenu, idCmdFirst, idCmdLast, uFlags);
371 }