d323c3dcae68ae9b4c17c2644f08532e56da181c
[reactos.git] / reactos / dll / win32 / browseui / shellbars / CBandSiteMenu.cpp
1 /*
2 * Band site menu
3 *
4 * Copyright 2007 Hervé Poussineua
5 * Copyright 2009 Andrew Hill
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 */
21
22 #include "shellbars.h"
23 #include <strsafe.h>
24
25 #define IDM_DESKBAND_BEGINCUSTOM 0x10
26 #define IDM_DESKBAND_ENDCUSTOM 0x25
27
28 CBandSiteMenu::CBandSiteMenu():
29 m_menuDsa(NULL),
30 m_hmenu(NULL)
31 {
32 }
33
34 CBandSiteMenu::~CBandSiteMenu()
35 {
36 if (m_hmenu)
37 DestroyMenu(m_hmenu);
38
39 if (m_menuDsa)
40 DSA_Destroy(m_menuDsa);
41
42 m_BandSite = NULL;
43 }
44
45
46 HRESULT CBandSiteMenu::CreateMenuPart()
47 {
48 WCHAR wszBandName[MAX_PATH];
49 WCHAR wszBandGUID[MAX_PATH];
50 WCHAR wRegKey[MAX_PATH];
51 UINT cBands;
52 DWORD dwDataSize;
53 CATID category = CATID_DeskBand;
54 HMENU hmenuToolbars;
55 DWORD dwRead;
56 CComPtr<IEnumGUID> pEnumGUID;
57 HRESULT hr;
58
59 if (m_hmenu)
60 DestroyMenu(m_hmenu);
61
62 if (m_menuDsa)
63 DSA_Destroy(m_menuDsa);
64
65 /* Load the template we will fill in */
66 m_hmenu = LoadMenuW(GetModuleHandleW(L"browseui.dll"), MAKEINTRESOURCEW(IDM_TASKBAR_TOOLBARS));
67 if (!m_hmenu)
68 return HRESULT_FROM_WIN32(GetLastError());
69
70 m_menuDsa = DSA_Create(sizeof(GUID), 5);
71 if (!m_menuDsa)
72 return E_OUTOFMEMORY;
73
74 /* Get the handle of the submenu where the available items will be shown */
75 hmenuToolbars = GetSubMenu(m_hmenu, 0);
76
77 /* Create the category enumerator */
78 hr = SHEnumClassesOfCategories(1, &category, 0, NULL, &pEnumGUID);
79 if (FAILED_UNEXPECTEDLY(hr))
80 return hr;
81
82 /* Enumerate the classes in the CATID_DeskBand category */
83 cBands = 0;
84 do
85 {
86 GUID iter;
87 pEnumGUID->Next(1, &iter, &dwRead);
88 if (!dwRead)
89 continue;
90
91 if (!StringFromGUID2(iter, wszBandGUID, MAX_PATH))
92 continue;
93
94 /* Get the band name */
95 StringCchPrintfW(wRegKey, MAX_PATH, L"CLSID\\%s", wszBandGUID);
96 dwDataSize = MAX_PATH;
97 SHGetValue(HKEY_CLASSES_ROOT, wRegKey, NULL, NULL, wszBandName, &dwDataSize);
98
99 /* Insert it */
100 InsertMenu(hmenuToolbars, cBands, MF_BYPOSITION, DSA_GetItemCount(m_menuDsa), wszBandName);
101 DSA_AppendItem(m_menuDsa, &iter);
102 cBands++;
103 }
104 while (dwRead > 0);
105
106 return S_OK;
107 }
108
109 HRESULT STDMETHODCALLTYPE CBandSiteMenu::SetOwner(IUnknown *pOwner)
110 {
111 TRACE("CBandSiteMenu::SetOwner(%p, %p)\n", this, pOwner);
112
113 /* Cache the menu that will be merged every time QueryContextMenu is called */
114 CreateMenuPart();
115
116 return pOwner->QueryInterface(IID_PPV_ARG(IBandSite, &m_BandSite));
117 }
118
119 HRESULT STDMETHODCALLTYPE CBandSiteMenu::QueryContextMenu(
120 HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
121 {
122 CComPtr<IPersist> pBand;
123 CLSID BandCLSID;
124 DWORD dwBandID;
125 UINT uBand = 0;
126
127 TRACE("CBandSiteMenu::QueryContextMenu(%p, %p, %u, %u, %u, 0x%x)\n", this, hmenu, indexMenu, idCmdFirst, idCmdLast, uFlags);
128
129 /* First Merge the menu with the available bands */
130 Shell_MergeMenus(hmenu, m_hmenu, indexMenu, idCmdFirst, idCmdLast, MM_DONTREMOVESEPS | MM_SUBMENUSHAVEIDS);
131
132 HMENU hmenuToolbars = GetSubMenu(hmenu, indexMenu);
133
134 /* Enumerate all present bands and mark them as checked in the menu */
135 while (SUCCEEDED(m_BandSite->EnumBands(uBand, &dwBandID)))
136 {
137 if (FAILED(m_BandSite->GetBandObject(dwBandID, IID_PPV_ARG(IPersist, &pBand))))
138 continue;
139
140 if (FAILED(pBand->GetClassID(&BandCLSID)))
141 continue;
142
143 /* Try to find the clsid of the band in the dsa */
144 UINT count = DSA_GetItemCount(m_menuDsa);
145 for (UINT i = 0; i < count; i++)
146 {
147 GUID* pdsaGUID = (GUID*)DSA_GetItemPtr(m_menuDsa, i);
148 if (memcmp(pdsaGUID, &BandCLSID, sizeof(GUID)) == 0)
149 {
150 /* The index in the dsa is also the index in the menu */
151 CheckMenuItem(hmenuToolbars, i, MF_CHECKED | MF_BYPOSITION);
152 }
153 }
154
155 uBand++;
156 }
157
158 return S_OK;
159 }
160
161 HRESULT STDMETHODCALLTYPE CBandSiteMenu::InvokeCommand(LPCMINVOKECOMMANDINFO lpici)
162 {
163 /* FIXME: do we need to handle this and how? */
164 if (HIWORD(lpici->lpVerb) != NULL)
165 return E_FAIL;
166
167 /* Get the GUID of the item that was clicked */
168 UINT uID = LOWORD(lpici->lpVerb);
169 GUID *pguidToolbar = (GUID *)DSA_GetItemPtr(m_menuDsa, uID);
170 if (!pguidToolbar)
171 return E_FAIL;
172
173 /* Try to find if a band with a guid is present. If it is remove it and return */
174 CComPtr<IPersist> pBand;
175 CLSID BandCLSID;
176 DWORD dwBandID;
177 UINT uBand = 0;
178 while (SUCCEEDED(m_BandSite->EnumBands(uBand, &dwBandID)))
179 {
180 if (FAILED(m_BandSite->GetBandObject(dwBandID, IID_PPV_ARG(IPersist, &pBand))))
181 continue;
182
183 if (FAILED(pBand->GetClassID(&BandCLSID)))
184 continue;
185
186 if (memcmp(pguidToolbar, &BandCLSID, sizeof(GUID)) == 0)
187 {
188 /* We found it, remove it */
189 m_BandSite->RemoveBand(dwBandID);
190 return S_OK;
191 }
192
193 uBand++;
194 }
195
196 /* It is not present. Add it. */
197 CComPtr<IDeskBand> pDeskBand;
198 HRESULT hRet = CoCreateInstance(*pguidToolbar, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IDeskBand, &pDeskBand));
199 if (FAILED(hRet))
200 return hRet;
201
202 hRet = m_BandSite->AddBand(pDeskBand);
203 if (FAILED_UNEXPECTEDLY(hRet))
204 return hRet;
205
206 return S_OK;
207 }
208
209 HRESULT STDMETHODCALLTYPE CBandSiteMenu::GetCommandString(UINT_PTR idCmd, UINT uType,
210 UINT *pwReserved, LPSTR pszName, UINT cchMax)
211 {
212 FIXME("CBandSiteMenu::GetCommandString is UNIMPLEMENTED (%p, %p, %u, %p, %p, %u)\n", this, idCmd, uType, pwReserved, pszName, cchMax);
213 return E_NOTIMPL;
214 }
215
216 HRESULT STDMETHODCALLTYPE CBandSiteMenu::HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam)
217 {
218 FIXME("CBandSiteMenu::HandleMenuMsg is UNIMPLEMENTED (%p, %u, %p, %p)\n", this, uMsg, wParam, lParam);
219 return E_NOTIMPL;
220 }
221
222 HRESULT STDMETHODCALLTYPE CBandSiteMenu::HandleMenuMsg2(UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *plResult)
223 {
224 FIXME("CBandSiteMenu::HandleMenuMsg2 is UNIMPLEMENTED(%p, %u, %p, %p, %p)\n", this, uMsg, wParam, lParam, plResult);
225 return E_NOTIMPL;
226 }
227
228 extern "C"
229 HRESULT WINAPI CBandSiteMenu_CreateInstance(REFIID riid, void **ppv)
230 {
231 return ShellObjectCreator<CBandSiteMenu>(riid, ppv);
232 }