4 * Copyright 2014 David Quintana
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.
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.
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 #include <shlwapi_undoc.h>
25 WINE_DEFAULT_DEBUG_CHANNEL(menusite
);
28 public CComObjectRootEx
<CComMultiThreadModelNoCS
>,
29 public CWindowImpl
<CMenuSite
, CWindow
, CControlWinTraits
>,
31 public IDeskBarClient
,
32 public IOleCommandTarget
,
34 public IInputObjectSite
,
35 public IWinEventHandler
,
36 public IServiceProvider
39 CComPtr
<IUnknown
> m_DeskBarSite
;
40 CComPtr
<IUnknown
> m_BandObject
;
41 CComPtr
<IDeskBand
> m_DeskBand
;
42 CComPtr
<IWinEventHandler
> m_WinEventHandler
;
49 DECLARE_WND_CLASS_EX(_T("MenuSite"), 0, COLOR_MENU
)
51 DECLARE_NOT_AGGREGATABLE(CMenuSite
)
52 DECLARE_PROTECT_FINAL_CONSTRUCT()
53 BEGIN_COM_MAP(CMenuSite
)
54 COM_INTERFACE_ENTRY_IID(IID_IBandSite
, IBandSite
)
55 COM_INTERFACE_ENTRY_IID(IID_IDeskBarClient
, IDeskBarClient
)
56 COM_INTERFACE_ENTRY_IID(IID_IOleWindow
, IOleWindow
)
57 COM_INTERFACE_ENTRY_IID(IID_IOleCommandTarget
, IOleCommandTarget
)
58 COM_INTERFACE_ENTRY_IID(IID_IInputObject
, IInputObject
)
59 COM_INTERFACE_ENTRY_IID(IID_IInputObjectSite
, IInputObjectSite
)
60 COM_INTERFACE_ENTRY_IID(IID_IWinEventHandler
, IWinEventHandler
)
61 COM_INTERFACE_ENTRY_IID(IID_IServiceProvider
, IServiceProvider
)
65 virtual HRESULT STDMETHODCALLTYPE
AddBand(IUnknown
* punk
);
66 virtual HRESULT STDMETHODCALLTYPE
EnumBands(UINT uBand
, DWORD
* pdwBandID
);
67 virtual HRESULT STDMETHODCALLTYPE
QueryBand(DWORD dwBandID
, IDeskBand
**ppstb
, DWORD
*pdwState
, LPWSTR pszName
, int cchName
);
68 virtual HRESULT STDMETHODCALLTYPE
GetBandObject(DWORD dwBandID
, REFIID riid
, VOID
**ppv
);
71 virtual HRESULT STDMETHODCALLTYPE
SetDeskBarSite(IUnknown
*punkSite
);
72 virtual HRESULT STDMETHODCALLTYPE
GetSize(DWORD dwWhich
, LPRECT prc
);
73 virtual HRESULT STDMETHODCALLTYPE
UIActivateDBC(DWORD dwState
);
76 virtual HRESULT STDMETHODCALLTYPE
GetWindow(HWND
*phwnd
);
79 virtual HRESULT STDMETHODCALLTYPE
QueryStatus(const GUID
* pguidCmdGroup
, ULONG cCmds
, OLECMD prgCmds
[], OLECMDTEXT
*pCmdText
);
80 virtual HRESULT STDMETHODCALLTYPE
Exec(const GUID
* pguidCmdGroup
, DWORD nCmdID
, DWORD nCmdexecopt
, VARIANT
*pvaIn
, VARIANT
*pvaOut
);
83 virtual HRESULT STDMETHODCALLTYPE
UIActivateIO(BOOL fActivate
, LPMSG lpMsg
);
84 virtual HRESULT STDMETHODCALLTYPE
HasFocusIO();
85 virtual HRESULT STDMETHODCALLTYPE
TranslateAcceleratorIO(LPMSG lpMsg
);
88 virtual HRESULT STDMETHODCALLTYPE
OnFocusChangeIS(IUnknown
*punkObj
, BOOL fSetFocus
);
91 virtual HRESULT STDMETHODCALLTYPE
IsWindowOwner(HWND hWnd
);
92 virtual HRESULT STDMETHODCALLTYPE
OnWinEvent(HWND hWnd
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
, LRESULT
*theResult
);
95 virtual HRESULT STDMETHODCALLTYPE
QueryService(REFGUID guidService
, REFIID riid
, void **ppvObject
);
98 // Using custom message map instead
99 virtual BOOL
ProcessWindowMessage(HWND hWnd
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
, LRESULT
&lResult
, DWORD mapId
= 0);
102 virtual HRESULT STDMETHODCALLTYPE
ContextSensitiveHelp(BOOL fEnterMode
);
103 virtual HRESULT STDMETHODCALLTYPE
GetBandSiteInfo(BANDSITEINFO
*pbsinfo
);
104 virtual HRESULT STDMETHODCALLTYPE
RemoveBand(DWORD dwBandID
);
105 virtual HRESULT STDMETHODCALLTYPE
SetBandSiteInfo(const BANDSITEINFO
*pbsinfo
);
106 virtual HRESULT STDMETHODCALLTYPE
SetBandState(DWORD dwBandID
, DWORD dwMask
, DWORD dwState
);
107 virtual HRESULT STDMETHODCALLTYPE
SetModeDBC(DWORD dwMode
);
110 IUnknown
* ToIUnknown() { return static_cast<IDeskBarClient
*>(this); }
114 HRESULT
CMenuSite_Constructor(REFIID riid
, LPVOID
*ppv
)
118 CMenuSite
* site
= new CComObject
<CMenuSite
>();
121 return E_OUTOFMEMORY
;
123 HRESULT hr
= site
->QueryInterface(riid
, ppv
);
131 CMenuSite::CMenuSite() :
135 m_WinEventHandler(NULL
),
140 HRESULT STDMETHODCALLTYPE
CMenuSite::ContextSensitiveHelp(BOOL fEnterMode
)
145 HRESULT STDMETHODCALLTYPE
CMenuSite::GetBandSiteInfo(BANDSITEINFO
*pbsinfo
)
150 HRESULT STDMETHODCALLTYPE
CMenuSite::RemoveBand(DWORD dwBandID
)
155 HRESULT STDMETHODCALLTYPE
CMenuSite::SetBandSiteInfo(const BANDSITEINFO
*pbsinfo
)
160 HRESULT STDMETHODCALLTYPE
CMenuSite::SetBandState(DWORD dwBandID
, DWORD dwMask
, DWORD dwState
)
165 HRESULT STDMETHODCALLTYPE
CMenuSite::SetModeDBC(DWORD dwMode
)
170 HRESULT STDMETHODCALLTYPE
CMenuSite::TranslateAcceleratorIO(LPMSG lpMsg
)
175 HRESULT STDMETHODCALLTYPE
CMenuSite::HasFocusIO()
180 HRESULT STDMETHODCALLTYPE
CMenuSite::OnFocusChangeIS(IUnknown
*punkObj
, BOOL fSetFocus
)
185 HRESULT STDMETHODCALLTYPE
CMenuSite::AddBand(IUnknown
* punk
)
189 #define TO_HRESULT(x) ((HRESULT)(S_OK+(x)))
191 if (SHIsSameObject(punk
, m_BandObject
))
192 return TO_HRESULT(0);
196 hr
= IUnknown_SetSite(m_BandObject
, NULL
);
203 m_WinEventHandler
= NULL
;
207 return TO_HRESULT(0);
209 hr
= punk
->QueryInterface(IID_PPV_ARG(IDeskBand
, &m_DeskBand
));
213 hr
= punk
->QueryInterface(IID_PPV_ARG(IWinEventHandler
, &m_WinEventHandler
));
217 hr
= IUnknown_SetSite(punk
, this->ToIUnknown());
221 hr
= IUnknown_GetWindow(punk
, &m_hWndBand
);
229 return TO_HRESULT(0);
232 HRESULT STDMETHODCALLTYPE
CMenuSite::EnumBands(UINT uBand
, DWORD
* pdwBandID
)
242 HRESULT STDMETHODCALLTYPE
CMenuSite::Exec(const GUID
* pguidCmdGroup
, DWORD nCmdID
, DWORD nCmdexecopt
, VARIANT
*pvaIn
, VARIANT
*pvaOut
)
244 return IUnknown_Exec(m_DeskBarSite
, *pguidCmdGroup
, nCmdID
, nCmdexecopt
, pvaIn
, pvaOut
);
247 HRESULT STDMETHODCALLTYPE
CMenuSite::GetBandObject(DWORD dwBandID
, REFIID riid
, VOID
**ppv
)
249 if (dwBandID
!= 0 || m_BandObject
== NULL
)
252 return E_NOINTERFACE
;
255 return m_BandObject
->QueryInterface(riid
, ppv
);
258 HRESULT STDMETHODCALLTYPE
CMenuSite::GetSize(DWORD dwWhich
, LPRECT prc
)
260 memset(prc
, 0, sizeof(*prc
));
265 if (m_DeskBand
== NULL
)
268 DESKBANDINFO info
= { 0 };
269 info
.dwMask
= DBIM_MAXSIZE
;
271 m_DeskBand
->GetBandInfo(0, 0, &info
);
273 prc
->right
= info
.ptMaxSize
.x
;
274 prc
->bottom
= info
.ptMaxSize
.y
;
279 HRESULT STDMETHODCALLTYPE
CMenuSite::GetWindow(HWND
*phwnd
)
289 HRESULT STDMETHODCALLTYPE
CMenuSite::IsWindowOwner(HWND hWnd
)
294 if (!m_WinEventHandler
)
297 return m_WinEventHandler
->IsWindowOwner(hWnd
);
300 HRESULT STDMETHODCALLTYPE
CMenuSite::OnWinEvent(HWND hWnd
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
, LRESULT
*theResult
)
302 if (!m_WinEventHandler
)
305 return m_WinEventHandler
->OnWinEvent(hWnd
, uMsg
, wParam
, lParam
, theResult
);
308 HRESULT STDMETHODCALLTYPE
CMenuSite::QueryBand(DWORD dwBandID
, IDeskBand
**ppstb
, DWORD
*pdwState
, LPWSTR pszName
, int cchName
)
316 return E_NOINTERFACE
;
319 HRESULT hr
= m_BandObject
->QueryInterface(IID_PPV_ARG(IDeskBand
, ppstb
));
329 HRESULT STDMETHODCALLTYPE
CMenuSite::QueryService(REFGUID guidService
, REFIID riid
, void **ppvObject
)
333 if (IsEqualGUID(guidService
, SID_SMenuBandBottom
) ||
334 IsEqualGUID(guidService
, SID_SMenuBandBottomSelected
) ||
335 IsEqualGUID(guidService
, SID_SMenuBandChild
))
337 if (m_BandObject
== NULL
)
340 return IUnknown_QueryService(m_BandObject
, guidService
, riid
, ppvObject
);
346 return IUnknown_QueryService(m_DeskBarSite
, guidService
, riid
, ppvObject
);
349 HRESULT STDMETHODCALLTYPE
CMenuSite::QueryStatus(const GUID
* pguidCmdGroup
, ULONG cCmds
, OLECMD prgCmds
[], OLECMDTEXT
*pCmdText
)
354 return IUnknown_QueryStatus(m_DeskBarSite
, *pguidCmdGroup
, cCmds
, prgCmds
, pCmdText
);
357 HRESULT STDMETHODCALLTYPE
CMenuSite::SetDeskBarSite(IUnknown
*punkSite
)
361 CComPtr
<IUnknown
> protectThis(this->ToIUnknown());
367 m_DeskBarSite
= NULL
;
369 hr
= IUnknown_GetWindow(punkSite
, &hWndSite
);
371 if (FAILED(hr
) || !hWndSite
)
376 Create(hWndSite
, NULL
, L
"MenuSite");
379 m_DeskBarSite
= punkSite
;
386 m_DeskBand
->CloseDW(0);
389 hr
= IUnknown_SetSite(m_BandObject
, NULL
);
393 m_WinEventHandler
= NULL
;
396 m_DeskBarSite
= NULL
;
401 HRESULT STDMETHODCALLTYPE
CMenuSite::UIActivateDBC(DWORD dwState
)
406 return m_DeskBand
->ShowDW(dwState
!= 0);
409 HRESULT STDMETHODCALLTYPE
CMenuSite::UIActivateIO(BOOL fActivate
, LPMSG lpMsg
)
414 return IUnknown_UIActivateIO(m_BandObject
, fActivate
, lpMsg
);
417 BOOL
CMenuSite::ProcessWindowMessage(HWND hWnd
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
, LRESULT
&lResult
, DWORD mapId
)
419 HWND hWndTarget
= NULL
;
420 CComPtr
<IUnknown
> protectThis(this->ToIUnknown());
427 CComPtr
<IMenuPopup
> pMenuPopup
;
428 if (SUCCEEDED(m_BandObject
->QueryInterface(IID_PPV_ARG(IMenuPopup
, &pMenuPopup
))))
431 GetClientRect(&Rect
);
432 Rect
.right
= Rect
.right
;
433 pMenuPopup
->OnPosRectChangeDB(&Rect
);
440 hWndTarget
= reinterpret_cast<LPNMHDR
>(lParam
)->hwndFrom
;
443 hWndTarget
= (HWND
) lParam
;
449 if (hWndTarget
&& m_WinEventHandler
&&
450 m_WinEventHandler
->IsWindowOwner(hWndTarget
) == S_OK
)
452 if (SUCCEEDED(m_WinEventHandler
->OnWinEvent(hWndTarget
, uMsg
, wParam
, lParam
, &lResult
)))