able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/shell/browseui/legacy/mnfolder.cpp

2990 lines
91 KiB
C++
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
#include "priv.h"
#include "sccls.h"
#include "menuband.h"
#include "itbar.h"
#include "dpastuff.h" // COrderList_*
#include "resource.h"
#include "mnbase.h"
#include "oleacc.h"
#include "apithk.h"
#include "menuisf.h"
#include "mnfolder.h"
#include "icotask.h"
#include "mluisupp.h"
#define PGMP_RECALCSIZE 200
#define SIL_GetType(pidl) (ILIsEmpty(pidl) ? 0 : (pidl)->mkid.abID[0])
//-------------------------------------------------------------------------
//
// Non-Member functions
//
//-------------------------------------------------------------------------
HRESULT IUnknown_RefreshParent(IUnknown* punk, LPCITEMIDLIST pidl, DWORD dwFlags)
{
IShellMenu* psm;
HRESULT hres = IUnknown_QueryService(punk, SID_SMenuBandParent,
IID_IShellMenu, (void **)&psm);
if (SUCCEEDED(hres))
{
LPITEMIDLIST pidlParent = ILClone(pidl);
if (pidlParent)
{
SMDATA smd;
ILRemoveLastID(pidlParent);
smd.dwMask = SMDM_SHELLFOLDER;
smd.pidlFolder = pidlParent;
smd.pidlItem = ILFindLastID(pidl);
hres = psm->InvalidateItem(&smd, dwFlags);
ILFree(pidlParent);
}
psm->Release();
}
return hres;
}
//-------------------------------------------------------------------------
//
// CMenuData class
//
//-------------------------------------------------------------------------
void CMenuData::SetSubMenu(IUnknown* punk)
{
ATOMICRELEASE(_punkSubmenu);
_punkSubmenu = punk;
if (_punkSubmenu)
_punkSubmenu->AddRef();
}
HRESULT CMenuData::GetSubMenu(const GUID* pguidService, REFIID riid, void** ppv)
{
// pguidService is for asking specifically for the Shell Folder portion or the Static portion
if (_punkSubmenu)
{
if (pguidService)
{
return IUnknown_QueryService(_punkSubmenu, *pguidService, riid, ppv);
}
else
return _punkSubmenu->QueryInterface(riid, ppv);
}
else
return E_NOINTERFACE;
}
CMenuData::~CMenuData()
{
ATOMICRELEASE(_punkSubmenu);
}
STDMETHODIMP CMenuSFToolbar::QueryInterface(REFIID riid, void** ppvObj)
{
HRESULT hres = CMenuToolbarBase::QueryInterface(riid, ppvObj);
if (FAILED(hres))
hres = CSFToolbar::QueryInterface(riid, ppvObj);
return hres;
}
//-------------------------------------------------------------------------
//
// CMenuSFToolbar class
//
//-------------------------------------------------------------------------
STDMETHODIMP CMenuSFToolbar::SetSite(IUnknown* punkSite)
{
HRESULT hres = CMenuToolbarBase::SetSite(punkSite);
if (SUCCEEDED(hres))
{
_fMulticolumnMB = BOOLIFY(_pcmb->_dwFlags & SMINIT_MULTICOLUMN);
_fMulticolumn = _fMulticolumnMB;
_fVertical = _fVerticalMB;
if (_fVerticalMB)
_dwStyle |= CCS_VERT;
}
return hres;
}
CMenuSFToolbar::CMenuSFToolbar(CMenuBand* pmb, IShellFolder* psf, LPCITEMIDLIST pidl, HKEY hKey, DWORD dwFlags)
: CMenuToolbarBase(pmb, dwFlags)
{
// Change this to IStream
_hKey = hKey;
// Do we have a place to persist our reorder?
if (_hKey == NULL)
{
// No, then don't allow it.
_fAllowReorder = FALSE;
}
_dwStyle |= TBSTYLE_REGISTERDROP;
_dwStyle &= ~TBSTYLE_TOOLTIPS; // We handle our own tooltips.
_iDefaultIconIndex = -1;
SetShellFolder(psf, pidl);
_AfterLoad();
}
HRESULT CMenuSFToolbar::SetShellFolder(IShellFolder* psf, LPCITEMIDLIST pidl)
{
HRESULT hres = CSFToolbar::SetShellFolder(psf, pidl);
ATOMICRELEASE(_pasf2);
if (psf)
psf->QueryInterface(IID_IAugmentedShellFolder2, (void**)&_pasf2);
return hres;
}
CMenuSFToolbar::~CMenuSFToolbar()
{
ASSERT(_pcmb->_cRef == 0 || _pcmb->_pmtbShellFolder == NULL);
_hwndWorkerWindow = NULL; // This is destroyed by the _pmbState destructor.
// Prevent a double delete which happens in the base class.
ATOMICRELEASE(_pasf2);
if (_hKey)
RegCloseKey(_hKey);
}
void CMenuSFToolbar::v_Close()
{
// We should save this so that we know when a new item is added.
_SaveOrderStream();
CMenuToolbarBase::EmptyToolbar();
_UnregisterToolbar();
if (_hwndPager)
{
DestroyWindow(_hwndPager); // Should Destroy Toolbar.
}
else if (_hwndMB)
{
// In the MultiColumn case, there is no pager so we have to
// manually destroy the Toolbar
DestroyWindow(_hwndMB);
}
_hwndPager = NULL;
_hwndMB = NULL;
_hwndTB = NULL;
}
PIBDATA CMenuSFToolbar::_CreateItemData(PORDERITEM poi)
{
return (PIBDATA)new CMenuData(poi);
}
HRESULT CMenuSFToolbar::_AfterLoad()
{
HRESULT hres = CSFToolbar::_AfterLoad();
if (SUCCEEDED(hres))
_LoadOrderStream();
return hres;
}
HRESULT CMenuSFToolbar::_LoadOrderStream()
{
OrderList_Destroy(&_hdpaOrder);
IStream* pstm;
HRESULT hres = E_FAIL;
if (_hKey)
{
// We use "Menu" for Backwards compatibility with shdoc401 start menu, but having no
// sub key is more correct (Other places use it) so on NT5 we use the new method.
pstm = SHOpenRegStream(_hKey, (_pcmb->_dwFlags & SMINIT_LEGACYMENU) ? TEXT("Menu") : TEXT(""),
TEXT("Order"), STGM_READ);
}
else
{
if (S_FALSE == CallCB(NULL, SMC_GETSFOBJECT, (WPARAM)(GUID*)&IID_IStream, (LPARAM)(void**)&pstm))
pstm = NULL;
}
if (pstm)
{
hres = OrderList_LoadFromStream(pstm, &_hdpaOrder, _psf);
_fHasOrder = FALSE;
_fAllowReorder = TRUE;
// Check to see if we have a persisted order. If we don't have a persisted order,
// then all of the items are -1. If just one of those has a number other than
// -1, then we do have "Order" and should use that instead of alphabetizing.
if (_hdpaOrder)
{
for (int i = 0; !_fHasOrder && i < DPA_GetPtrCount(_hdpaOrder); i++)
{
PORDERITEM poi = (PORDERITEM)DPA_FastGetPtr(_hdpaOrder, i);
if (poi->nOrder != MNFOLDER_NORODER)
_fHasOrder = TRUE;
}
}
pstm->Release();
}
return hres;
}
HRESULT CMenuSFToolbar::_SaveOrderStream()
{
IStream* pstm;
HRESULT hres = E_FAIL;
// Persist the new order out to the registry
// It is reasonable to assume that if we don't have an _hdpa we have
// not filled the toolbar yet. Since we have not filled it, we haven't changed
// the order, so we don't need to persist out that order information.
if(_hdpa)
{
// Always save this information
_FindMinPromotedItems(TRUE);
// Did we load an order stream when we initialized this pane?
if (!_fHasOrder)
{
// No; Then we do not want to persist the order. We will initialize
// all of the order items to -1. This is backward compatible because
// IE 4 will merge alphabetically, but revert to a persited order when saving.
for (int i = 0; i < DPA_GetPtrCount(_hdpa); i++)
{
PORDERITEM poi = (PORDERITEM)DPA_FastGetPtr(_hdpa, i);
poi->nOrder = MNFOLDER_NORODER;
}
}
if (_hKey)
{
pstm = SHOpenRegStream(_hKey, (_pcmb->_dwFlags & SMINIT_LEGACYMENU) ? TEXT("Menu") : TEXT(""),
TEXT("Order"), STGM_CREATE | STGM_WRITE);
}
else
{
if (S_OK != CallCB(NULL, SMC_GETSFOBJECT, (WPARAM)(GUID*)&IID_IStream, (LPARAM)(void**)&pstm))
pstm = NULL;
}
if (pstm)
{
hres = OrderList_SaveToStream(pstm, _hdpaOrder ? _hdpaOrder : _hdpa, _psf);
if (SUCCEEDED(hres))
{
CallCB(NULL, SMC_SETSFOBJECT, (WPARAM)(GUID*)&IID_IStream, (LPARAM)(void**)&pstm);
}
pstm->Release();
}
}
if (SUCCEEDED(hres))
hres = CSFToolbar::_SaveOrderStream();
return hres;
}
void CMenuSFToolbar::_Dropped(int nIndex, BOOL fDroppedOnSource)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
ASSERT(_fDropping);
CSFToolbar::_Dropped(nIndex, fDroppedOnSource);
IEPlaySound(TEXT("MoveMenuItem"), FALSE);
// Set this to false here because it is ugly that we don't behave like a menu right after a drop.
_fEditMode = FALSE;
// Notify the toplevel menuband of the drop in case it was popped open
// because of the drag/drop event.
//
// (There are some functionality/activation problems if we keep the
// menu up after this case. So to avoid those things at this late date,
// we're going to cancel the menu after a timeout.)
IOleCommandTarget * poct;
_pcmb->QueryService(SID_SMenuBandTop, IID_IOleCommandTarget, (LPVOID *)&poct);
if (poct)
{
poct->Exec(&CGID_MenuBand, MBANDCID_ITEMDROPPED, 0, NULL, NULL);
poct->Release();
}
_pcmb->_fDragEntered = FALSE;
}
HMENU CMenuSFToolbar::_GetContextMenu(IContextMenu* pcm, int* pid)
{
*pid += MNIDM_LAST;
HMENU hmenu = CSFToolbar::_GetContextMenu(pcm, pid);
HMENU hmenu2 = LoadMenuPopup_PrivateNoMungeW(MENU_MNFOLDERCONTEXT);
// now find the properties insertion point and
int iCount = GetMenuItemCount(hmenu);
for (int i = 0; i < iCount; i++) {
TCHAR szCommand[40];
UINT id = GetMenuItemID(hmenu, i);
if (IsInRange(id, *pid, 0x7fff )) {
id -= *pid;
ContextMenu_GetCommandStringVerb(pcm, id, szCommand, ARRAYSIZE(szCommand));
if (!lstrcmpi(szCommand, TEXT("properties"))) {
break;
}
}
}
Shell_MergeMenus(hmenu, hmenu2, i, 0, 0x7FFF, 0);
DestroyMenu(hmenu2);
return hmenu;
}
void CMenuSFToolbar::_OnDefaultContextCommand(int idCmd)
{
switch (idCmd)
{
case MNIDM_RESORT:
{
// We used to blow away the order stream and refill, but since we use the order stream
// for calculating the presence of new items, this promoted all of the items were were
// sorting.
HDPA hdpa = _hdpa;
// For some reason we have an _hdpaOrder, so use that for persisting out
// to the registry..
if (_hdpaOrder)
hdpa = _hdpaOrder;
_SortDPA(hdpa);
OrderList_Reorder(hdpa);
_fChangedOrder = TRUE;
// This call knows about _hdpa and _hdpaOrder
_SaveOrderStream();
// MIKESH: this is needed because otherwise FillToolbar will use the current _hdpa
// and nothing gets changed... I think it's because OrderItem_Compare returns failure on some of the pidls
CMenuToolbarBase::EmptyToolbar();
_SetDirty(TRUE);
_LoadOrderStream();
if (_fShow)
{
_FillToolbar();
}
break;
}
}
}
LRESULT CMenuSFToolbar::_OnContextMenu(WPARAM wParam, LPARAM lParam)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
//
// When the NoSetTaskbar restriction is set, this code will disallow
// Context menus. It querys up to the Start menu to ask for permission
// to set.
LRESULT lres = 0;
// No UEM on Context Menus. This avoids the problem where we expand the menubands
// with a context menu present.
_fSuppressUserMonitor = TRUE;
// Allow the selected item to blow away the menus. This is explicitly for the Verbs "Open"
// "Print" and such that launch another process. Inprocess commands are unaffected by this.
MyLockSetForegroundWindow(FALSE);
BOOL fOwnerIsTopmost = (WS_EX_TOPMOST & GetWindowLong(_pcmb->_pmbState->GetSubclassedHWND(), GWL_EXSTYLE));
if (fOwnerIsTopmost)
{
::SetWindowPos(_pcmb->_pmbState->GetSubclassedHWND(), HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
}
KillTimer(_hwndMB, MBTIMER_INFOTIP);
_pcmb->_pmbState->HideTooltip(FALSE);
if (!(_pcmb->_dwFlags & SMINIT_RESTRICT_CONTEXTMENU))
lres = CSFToolbar::_OnContextMenu(wParam, lParam);
if (fOwnerIsTopmost)
{
::SetWindowPos(_pcmb->_pmbState->GetSubclassedHWND(), HWND_TOPMOST, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
IUnknown_QueryServiceExec(SAFECAST(_pcmb, IOleCommandTarget*), SID_SMenuBandTop,
&CGID_MenuBand, MBANDCID_REPOSITION, TRUE, NULL, NULL);
}
// Take the capture back after the context menu
GetMessageFilter()->RetakeCapture();
return lres;
}
HRESULT CMenuSFToolbar::_GetInfo(LPCITEMIDLIST pidl, SMINFO* psminfo)
{
HRESULT hres;
if (psminfo->dwMask & SMIM_TYPE)
{
psminfo->dwType = SMIT_STRING;
}
if (psminfo->dwMask & SMIM_FLAGS)
{
psminfo->dwFlags = SMIF_ICON | SMIF_DROPTARGET;
}
if (psminfo->dwMask & SMIM_ICON)
{
psminfo->dwMask &= ~SMIM_ICON;
psminfo->iIcon = -1;
}
DWORD dwAttr = SFGAO_FOLDER | SFGAO_FILESYSTEM | SFGAO_BROWSABLE;
// Folders that behave like shortcuts should not be considered
// as cascading menu items. Channels are an example.
// HACKHACK: to detect channels, we originally planned to GetUIObject
// IShellLink. But this doesn't work on browser-only b/c it doesn't
// pipe down to the shell extension. So as a hack, we'll key off
// the absence of SFGAO_FILESYSTEM.
// Is this a folder?
// And is it NOT a browseable folder? If it's a Browseable folder, this means that it's a namespace
// such as the Internet Namespace. The Internet name space's shell folder does not return real items, so it
// makes it useless in menus. So, filter it out, and treat it like an item.
hres = _psf->GetAttributesOf(1, &pidl, &dwAttr);
if (SUCCEEDED(hres) &&
IsFlagSet(dwAttr, SFGAO_FOLDER) &&
!IsFlagSet(dwAttr, SFGAO_BROWSABLE))
{
// Since SHIsExpandableFolder is such an expensive call, and we only need
// it for legacy Channels support, only do this call where channels are:
// Favorites menu and Start Menu | Favorites.
if (_dwFlags & SMSET_HASEXPANDABLEFOLDERS)
{
// on integrated install, check to see if the item supports
// is an expandable folder.
if (WhichPlatform() == PLATFORM_INTEGRATED)
{
// Yes; but does it also behave like a shortcut?
if (SHIsExpandableFolder(_psf, pidl))
psminfo->dwFlags |= SMIF_SUBMENU;
}
else if (IsFlagSet(dwAttr, SFGAO_FILESYSTEM))
{
// On browse only, we don't rev the shell, so we rely upon
// the filesystem bit...
// this is a bit of a hack, and then disallows you from
// doing things like having controlpanel be in the menu
// but it's the best we can do.
psminfo->dwFlags |= SMIF_SUBMENU;
}
}
else
{
// We're going to assume that if it's a folder, it really is a folder.
psminfo->dwFlags |= SMIF_SUBMENU;
}
}
CallCB(pidl, SMC_GETSFINFO, 0, (LPARAM)psminfo);
return hres;
}
/*----------------------------------------------------------
Purpose: This function determines the toolbar button style for the
given pidl.
Returns S_OK if pdwMIFFlags is also set (i.e., the object
supported IMenuBandItem to provide more info). S_FALSE if only
*pdwStyle is set.
*/
HRESULT CMenuSFToolbar::_TBStyleForPidl(LPCITEMIDLIST pidl,
DWORD * pdwStyle, DWORD* pdwState, DWORD * pdwMIFFlags, int * piIcon)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
HRESULT hres = S_FALSE;
DWORD dwStyle = TBSTYLE_BUTTON | TBSTYLE_DROPDOWN | TBSTYLE_NOPREFIX;
*pdwState = TBSTATE_ENABLED;
*pdwMIFFlags = 0;
*piIcon = -1;
if (pidl)
{
SMINFO sminfo;
sminfo.dwMask = SMIM_TYPE | SMIM_FLAGS | SMIM_ICON;
if (SUCCEEDED(_GetInfo(pidl, &sminfo)))
{
*pdwMIFFlags = sminfo.dwFlags;
if (sminfo.dwFlags & SMIF_ACCELERATOR)
dwStyle &= ~TBSTYLE_NOPREFIX;
if (sminfo.dwType & SMIT_SEPARATOR)
{
dwStyle &= ~TBSTYLE_BUTTON;
dwStyle |= TBSTYLE_SEP;
}
if (sminfo.dwFlags & SMIF_ICON)
*piIcon = sminfo.iIcon;
if (sminfo.dwFlags & SMIF_DEMOTED &&
!_pcmb->_fExpanded)
{
*pdwState |= TBSTATE_HIDDEN;
_fHasDemotedItems = TRUE;
}
if (sminfo.dwFlags & SMIF_HIDDEN)
*pdwState |= TBSTATE_HIDDEN;
hres = S_OK;
}
}
else
{
// For null pidls ("empty" menuitems), there is no icon.
// SMIF_DROPTTARGET is set so the user can drop into an empty submenu.
*pdwMIFFlags = SMIF_DROPTARGET;
// Return S_OK so the pdwMIFFlags is examined.
hres = S_OK;
}
*pdwStyle = dwStyle;
return hres;
}
BOOL CMenuSFToolbar::_FilterPidl(LPCITEMIDLIST pidl)
{
// BUGBUG raymondc PERF? Can we short-circuit the filter?
return S_OK == CallCB(pidl, SMC_FILTERPIDL, 0, 0);
}
void CMenuSFToolbar::_FillDPA(HDPA hdpa, HDPA hdpaSort, DWORD dwEnumFlags)
{
_fHasSubMenu = FALSE;
CallCB(NULL, SMC_BEGINENUM, (WPARAM)&dwEnumFlags, 0);
CSFToolbar::_FillDPA(hdpa, hdpaSort, dwEnumFlags);
CallCB(NULL, SMC_ENDENUM, 0, 0);
if (0 == DPA_GetPtrCount(hdpa) && _psf)
{
OrderList_Append(hdpa, NULL, -1); // Add a bogus pidl
_fEmpty = TRUE;
_fHasDemotedItems = FALSE;
if (_dwFlags & SMSET_NOEMPTY)
_fDontShowEmpty = TRUE;
}
else
{
_fEmpty = FALSE;
if (_dwFlags & SMSET_NOEMPTY)
_fDontShowEmpty = FALSE;
}
}
void CMenuSFToolbar::_AddChevron()
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
// Does this menu get a chevron button?
if (_fHasDemotedItems && !_pcmb->_fExpanded && _idCmdChevron == -1)
{
// Yes; (we shouldn't get here if the menu is empty)
ASSERT(!_fEmpty);
// Add the chevron to the top or the bottom
if (_dwFlags & SMSET_TOP && _pcmb->_pmtbTop != _pcmb->_pmtbBottom)
_AddOrderItemTB(NULL, 0, NULL); // add to top
else
_AddOrderItemTB(NULL, -1, NULL); // append to bottom
}
}
void CMenuSFToolbar::_RemoveChevron()
{
if (-1 != _idCmdChevron)
{
// Yes; remove the chevron
int iPos = ToolBar_CommandToIndex(_hwndTB, _idCmdChevron);
InlineDeleteButton(iPos);
_idCmdChevron = -1;
}
}
void CMenuSFToolbar::_ToolbarChanged()
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
_pcmb->_fForceButtonUpdate = TRUE;
// We shouldn't change the size of the menubar while we're in the middle
// of a delete. Wait until we're done...
if (!_fPreventToolbarChange && _fShow && !_fEmptyingToolbar)
{
RECT rcOld;
RECT rcNew;
HWND hwndP;
// Resize the MenuBar
hwndP = _hwndPager ? GetParent(_hwndPager): GetParent(_hwndTB);
GetClientRect(hwndP, &rcOld);
_pcmb->ResizeMenuBar();
GetClientRect(hwndP, &rcNew);
// If the rect sizes haven't changed, then we need to re-layout the
// band because the button widths may have changed.
if (EqualRect(&rcOld, &rcNew))
NegotiateSize();
// This pane may have changed sizes. If there is a sub menu, then
// we need to have them reposition themselves
if (_pcmb->_fInSubMenu && _pcmb->_pmtbTracked)
{
_pcmb->_pmtbTracked->PositionSubmenu(-1);
IUnknown_QueryServiceExec(_pcmb->_pmpSubMenu, SID_SMenuBandChild,
&CGID_MenuBand, MBANDCID_REPOSITION, 0, NULL, NULL);
}
}
}
void CMenuSFToolbar::_FillToolbar()
{
// Don't fill the toolbar if we're not dirty or we're emptying the toolbar
// If we try and fill the toolbar while we're emptying we enter a race condition
// where we could AV. This fixes a bug where when dragging a folder into the
// start menu, and cascade a menu, we empty one toolbar, which causes the
// other toolbar to get destroyed, unregister itself, flush the change notify
// queue, causing the original window to empty again... (lamadio) 7.16.98
if (_fDirty && !_fEmptyingToolbar)
{
LPITEMIDLIST pidlItem = NULL;
IShellMenu* psmSubMenu = NULL;
// Populating the menu will take a long time since we're hitting
// the disk. Give the user some feedback if the cursor is
// IDC_ARROW. (If the cursor is something else, then don't
// mess with it.) Note that we have to use (HCURSOR)-1 as a
// sentinel, because it's possible that the current cursor is NULL.
// Prevent _ToolbarChanged from Doing things. (Perf)
_fPreventToolbarChange = TRUE;
// Are we in a sub menu? If we are, then we don't want to blow away
// that menu, so we save it away do the fill and put it back in.
// This is so that if it's displayed we won't collapse it unless absoluley
// necessary.
if (_pcmb->_fInSubMenu)
{
CMenuData* pdata = (CMenuData*)_IDToPibData(_pcmb->_nItemSubMenu);
// This can be null if the fill toolbar was because you right clicked this item and clicked
// delete...
if (pdata)
{
// If we hit this, then _nItemSubMenu is out of sync. Figure out how
// that got set to a non-sub menu item...
ASSERT(pdata->GetFlags() & SMIF_SUBMENU);
// Hold onto this shell menu. We'll try and reuse it...
pdata->GetSubMenu(NULL, IID_IShellMenu, (void**)&psmSubMenu);
pidlItem = ILClone(pdata->GetPidl());
}
}
// Remove the chevron...
_RemoveChevron();
// Reset this bit. Fill toolbar will set it if there are items.
_fHasDemotedItems = FALSE;
CSFToolbar::_FillToolbar();
// If we had a Chevron before we refreshed the toolbar,
// then we need to add it back. To Short circuit the add
// we need to set the id to -1
_idCmdChevron = -1;
_AddChevron();
if (_hwndPager)
SendMessage(_hwndPager, PGMP_RECALCSIZE, (WPARAM) 0, (LPARAM) 0);
_fPreventToolbarChange = FALSE;
// Are we lucky?
if (pidlItem)
{
if (psmSubMenu)
{
// BUGBUG(lamadio): Unify this code with _ReBindToFolder.
// Great, we have all of the information we need. Now see if this pidl wasn't
// one slated for death: Is it now in the toolbar?
TBBUTTONINFO tbinfo = {0};
tbinfo.dwMask = TBIF_COMMAND | TBIF_LPARAM;
LPCITEMIDLIST pcidl = _GetButtonFromPidl(pidlItem, &tbinfo, NULL);
if (pcidl)
{
LPITEMIDLIST pidlFull = NULL;
IShellFolder* psf;
if(_pasf2)
{
LPITEMIDLIST pidlFolder, pidlChild;
_pasf2->UnWrapIDList(pidlItem, 1, NULL, &pidlFolder, &pidlChild, NULL);
pidlFull = ILCombine(pidlFolder, pidlChild);
ILFree(pidlChild);
ILFree(pidlFolder);
}
else
{
pidlFull = ILCombine(_pidl, pcidl);
}
// We need to bind to this pidl because it may have become merged. This will
// spontaniously show both namespaces. Pretty cool huh? Love object reuse...
_psf->BindToObject(pidlItem, NULL, IID_IShellFolder, (void**)&psf);
if (psf)
{
if (pidlFull)
{
// Force the New information into the Sub Menu. This will cause a reenum.
if (SUCCEEDED(psmSubMenu->SetShellFolder(psf, pidlFull, NULL, 0)))
{
// If this Eval fires, then this item was inserted into the
// toolbar with a null pointer, or it was in the process of being
// removed.
CMenuData* pmd = (CMenuData*)tbinfo.lParam;
if (EVAL(pmd))
{
// Make sure to store the Sub menu pointer back in the item it came from.
pmd->SetSubMenu(psmSubMenu);
_pcmb->_nItemSubMenu = tbinfo.idCommand;
}
}
}
psf->Release();
}
ILFree(pidlFull);
}
else
{
_pcmb->_SubMenuOnSelect(MPOS_CANCELLEVEL);
}
psmSubMenu->Release();
}
ILFree(pidlItem);
}
_ToolbarChanged();
}
}
void CMenuSFToolbar::v_OnDeleteButton(LPVOID pData)
{
CMenuData* pmd = (CMenuData*)pData;
ASSERT(pmd && IS_VALID_CODE_PTR(pmd, CMenuData));
if (pmd)
delete pmd;
}
void CMenuSFToolbar::v_OnEmptyToolbar()
{
CMenuToolbarBase::v_OnEmptyToolbar();
OrderList_Destroy(&_hdpa);
_fDirty = TRUE;
_nNextCommandID = 0;
}
void CMenuSFToolbar::_ObtainPIDLName(LPCITEMIDLIST pidl, LPTSTR psz, int cchMax)
{
// We overload this function because a NULL pidl is a place hold for
// (Empty) When there are no items, or the Chevron when there are items.
if (pidl)
{
CSFToolbar::_ObtainPIDLName(pidl, psz, cchMax);
}
else if (_fHasDemotedItems) // Chevron Case.
{
StrCpyN(psz, TEXT(">>"), cchMax);
}
else // Empty Case
{
MLLoadString(IDS_EMPTY, psz, cchMax);
}
}
void CMenuSFToolbar::v_NewItem(LPCITEMIDLIST pidl)
{
// This is called when an item is present in the filesystem
// that is not in the order stream. This occurs when an item is
// created when the menu is not up.
// BUGBUG (lamadio): IntelliMenus: New items are going to have a weird Promotion state
// if there are multiple clients. Each client is going to be the create, and try to increment this.
// We have to syncronize access to this. I'm not sure how to do this.
// New items get promoted.
CallCB(pidl, SMC_NEWITEM, 0, 0);
// Since this is a new item, we want to increment the promoted items
// so that we can do chevron tracking.
_cPromotedItems++;
}
void CMenuSFToolbar::_SetDirty(BOOL fDirty)
{
if (fDirty)
_pcmb->_fForceButtonUpdate = TRUE;
CSFToolbar::_SetDirty(fDirty);
}
void CMenuSFToolbar::_OnFSNotifyAdd(LPCITEMIDLIST pidl)
{
DWORD dwEnumFlags = SHCONTF_FOLDERS | SHCONTF_NONFOLDERS;
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
_RemoveChevron();
CallCB(NULL, SMC_BEGINENUM, (WPARAM)&dwEnumFlags, 0);
CSFToolbar::_OnFSNotifyAdd(pidl);
CallCB(NULL, SMC_ENDENUM, 0, 0);
_AddChevron();
// When we add something to this, we want to promote our parent.
IUnknown_RefreshParent(_pcmb->_punkSite, _pidl, SMINV_PROMOTE);
_SaveOrderStream();
}
UINT ToolBar_GetVisibleCount(HWND hwnd)
{
UINT cVis = 0;
int cItems = ToolBar_ButtonCount(hwnd) - 1;
for (; cItems >= 0; cItems--)
{
TBBUTTONINFO tbinfo;
tbinfo.cbSize = sizeof(tbinfo);
tbinfo.dwMask = TBIF_BYINDEX | TBIF_STATE;
if (ToolBar_GetButtonInfo(hwnd, cItems, &tbinfo))
{
if (!(tbinfo.fsState & TBSTATE_HIDDEN))
{
cVis ++;
}
}
}
return cVis;
}
void CMenuSFToolbar::_OnFSNotifyRemove(LPCITEMIDLIST pidl)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
int i;
_RemoveChevron();
// Check to see if this item is a promoted guy...
LPITEMIDLIST pidlButton = _GetButtonFromPidl(pidl, NULL, &i);
if (pidlButton)
{
int idCmd = GetButtonCmd(_hwndMB, i);
// Is he promoted?
if (!(v_GetFlags(idCmd) & SMIF_DEMOTED))
{
// Yes, then we need to decrement the promoted count because
// we are removing a promoted guy.
_cPromotedItems--;
// We should expand if we go to zero
if (_cPromotedItems == 0)
{
// Demote the parent
IUnknown_RefreshParent(_pcmb->_punkSite, _pidl, SMINV_DEMOTE | SMINV_NEXTSHOW);
Expand(TRUE);
}
}
if (_pcmb->_fInSubMenu && _pcmb->_nItemSubMenu == idCmd)
_pcmb->_SubMenuOnSelect(MPOS_CANCELLEVEL);
}
CSFToolbar::_OnFSNotifyRemove(pidl);
//Oooppsss, we removed the only string. Replace with our "(Empty)"
// handler....
if (0 == DPA_GetPtrCount(_hdpa) && _psf && _fVerticalMB)
{
ASSERT(_fEmpty == FALSE);
// If we are Empty, then we cannot have any demoted items
// NOTE: We can have no demoted items and not be empty, so one does
// not imply the other.
_fHasDemotedItems = FALSE;
_AddPidl(NULL, 0);
_fEmpty = TRUE;
if (_dwFlags & SMSET_NOEMPTY)
_fDontShowEmpty = TRUE;
}
if (_dwFlags & SMSET_COLLAPSEONEMPTY &&
ToolBar_GetVisibleCount(_hwndMB) == 0)
{
// When we don't want to be shown when empty, collapse.
_pcmb->_SiteOnSelect(MPOS_FULLCANCEL);
}
_AddChevron();
}
void CMenuSFToolbar::NegotiateSize()
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
RECT rc;
HWND hwndP;
hwndP = _hwndPager ? GetParent(_hwndPager): GetParent(_hwndTB);
GetClientRect(hwndP, &rc);
_pcmb->OnPosRectChangeDB(&rc);
}
/*----------------------------------------------------------
Purpose: CDelegateDropTarget::DragEnter
Informs Menuband that a drag has entered it's window.
*/
STDMETHODIMP CMenuSFToolbar::DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
_pcmb->_fDragEntered = TRUE;
IOleCommandTarget * poct;
_pcmb->QueryService(SID_SMenuBandTop, IID_IOleCommandTarget, (LPVOID *)&poct);
if (poct)
{
poct->Exec(&CGID_MenuBand, MBANDCID_DRAGENTER, 0, NULL, NULL);
poct->Release();
}
return CSFToolbar::DragEnter(pdtobj, grfKeyState, pt, pdwEffect);
}
/*----------------------------------------------------------
Purpose: CDelegateDropTarget::DragLeave
Informs Menuband that a drag has left it's window.
*/
STDMETHODIMP CMenuSFToolbar::DragLeave(void)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
_pcmb->_fDragEntered = FALSE;
IOleCommandTarget * poct;
_pcmb->QueryService(SID_SMenuBandTop, IID_IOleCommandTarget, (LPVOID *)&poct);
if (poct)
{
poct->Exec(&CGID_MenuBand, MBANDCID_DRAGLEAVE, 0, NULL, NULL);
poct->Release();
}
return CSFToolbar::DragLeave();
}
/*----------------------------------------------------------
Purpose: CDelegateDropTarget::HitTestDDT
Returns the ID to pass to GetObject.
30
*/
HRESULT CMenuSFToolbar::HitTestDDT(UINT nEvent, LPPOINT ppt, DWORD_PTR *pdwId, DWORD *pdwEffect)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
TBINSERTMARK tbim;
DWORD dwFlags = 0;
BOOL fOnButton = FALSE;
// If we're in drag and drop, Take UEM out of the picture
_fSuppressUserMonitor = TRUE;
// Unlike the CISFBand implementation, we always want to insert
// b/t the menu items. So we return a negative index so the
// GetObject method will treat all the drops as if we're dropping
// in b/t the items, even if the cursor is over a menuitem.
switch (nEvent)
{
case HTDDT_ENTER:
// OLE is in its modal drag/drop loop, and it has the capture.
// We shouldn't take the capture back during this time.
if (!(_pcmb->_dwFlags & SMINIT_RESTRICT_DRAGDROP) &&
(S_FALSE == CallCB(NULL, SMC_SFDDRESTRICTED, NULL, NULL)))
{
// Since we've been entered, set the global state as
// having the drag. If at some point the whole menu
// heirarchy does not have the drag inside of it, we want to
// collapse the menu. This is to prevent the hanging menu syndrome.
_pcmb->_pmbState->HasDrag(TRUE);
KillTimer(_hwndMB, MBTIMER_DRAGPOPDOWN);
GetMessageFilter()->PreventCapture(TRUE);
return S_OK;
}
else
return S_FALSE;
case HTDDT_OVER:
BLOCK
{
int iButton;
*pdwEffect = DROPEFFECT_NONE;
POINT pt = *ppt;
ClientToScreen(_hwndTB, &pt);
if (WindowFromPoint(pt) == _hwndPager )
{
iButton = IBHT_PAGER;
}
else
{
// Are we sitting BETWEEN buttons?
if (ToolBar_InsertMarkHitTest(_hwndTB, ppt, &tbim))
{
// Yes.
// Is this on the source button?
if (!(tbim.dwFlags & TBIMHT_BACKGROUND) &&
tbim.iButton == _iDragSource)
{
iButton = IBHT_SOURCE; // Yes; don't drop on the source button
}
else
{
iButton = tbim.iButton;
}
}
// No we're either sitting on a button or the background. Button?
else if (tbim.iButton != -1 && !(tbim.dwFlags & TBIMHT_BACKGROUND))
{
// On a Button. Cool.
iButton = tbim.iButton;
fOnButton = TRUE;
}
// Can this drop target even accept the drop?
int idBtn = GetButtonCmd(_hwndTB, tbim.iButton);
dwFlags = v_GetFlags(idBtn);
if ( _idCmdChevron != idBtn &&
!(dwFlags & (SMIF_DROPTARGET | SMIF_DROPCASCADE)) ||
((_pcmb->_dwFlags & SMINIT_RESTRICT_DRAGDROP) ||
(S_OK == CallCB(NULL, SMC_SFDDRESTRICTED, NULL, NULL))))
{
// No
return E_FAIL;
}
}
*pdwId = iButton;
}
break;
case HTDDT_LEAVE:
// If the dropped occured in this band, then we don't want to collapse the menu
if (!_fHasDrop)
{
// Since we've been left, set the global state. If moving between panes
// then the pane that will be entered will reset this within the timeout period
_pcmb->_pmbState->HasDrag(FALSE);
_SetTimer(MBTIMER_DRAGPOPDOWN);
}
// We can take the capture back anytime now
GetMessageFilter()->PreventCapture(FALSE);
if (!_fVerticalMB)
{
tbim = _tbim;
}
else
{
// Turn off the insertion mark
tbim.iButton = -1;
tbim.dwFlags = 0;
DAD_ShowDragImage(FALSE);
ToolBar_SetInsertMark(_hwndTB, &tbim);
UpdateWindow(_hwndTB);
DAD_ShowDragImage(TRUE);
}
break;
}
// Did the drop target change?
if (tbim.iButton != _tbim.iButton || tbim.dwFlags != _tbim.dwFlags)
{
DAD_ShowDragImage(FALSE);
// Yes
// If we're sitting on a button, highlight it. Otherwise remove the hightlight.
//ToolBar_SetHotItem(_hwndTB, fOnButton? tbim.iButton : -1);
// No.
// We pop open submenus here during drag and drop. But only
// if the button has changed (not the flags). Otherwise we'd
// get flashing submenus as the cursor moves w/in a single item.
if (tbim.iButton != _tbim.iButton)
{
_SetTimer(MBTIMER_DRAGOVER);
BOOL_PTR fOldAnchor = ToolBar_SetAnchorHighlight(_hwndTB, FALSE);
ToolBar_SetHotItem(_hwndTB, -1);
_pcmb->_SiteOnSelect(MPOS_CHILDTRACKING);
ToolBar_SetAnchorHighlight(_hwndTB, fOldAnchor);
}
// for now I don't want to rely on non-filesystem IShellFolder
// implementations to call our OnChange method when a drop occurs,
// so don't even show the insert mark.
// We do not want to display the Insert mark if we do not allow reorder.
if ((_fFSNotify || _iDragSource >= 0) && (dwFlags & SMIF_DROPTARGET) && _fAllowReorder)
{
ToolBar_SetInsertMark(_hwndTB, &tbim);
}
if (ppt)
_tbim = tbim;
UpdateWindow(_hwndTB);
DAD_ShowDragImage(TRUE);
}
if (!_fVerticalMB && HTDDT_LEAVE == nEvent)
{
// Cursor leaving menuband, reset
_tbim.iButton = -1;
_iDragSource = -1;
}
return S_OK;
}
/*----------------------------------------------------------
Purpose: CDelegateDropTarget::GetObjectDDT
*/
HRESULT CMenuSFToolbar::GetObjectDDT(DWORD_PTR dwId, REFIID riid, LPVOID * ppvObj)
{
HRESULT hres = E_NOINTERFACE;
int nID = (int)dwId;
*ppvObj = NULL;
if (nID == IBHT_PAGER)
{
SendMessage(_hwndPager, PGM_GETDROPTARGET, 0, (LPARAM)ppvObj);
}
// Is the target the source?
else if (IBHT_SOURCE != nID && IBHT_BACKGROUND != nID)
{
// No; does the shellfolder support IDropTarget?
// We want to pass the subclassed HWND, because all we want the parent of the context menus to be
// the Subclassed window. This is so we don't loose focus and collapse.
hres = _psf->CreateViewObject(_pcmb->_pmbState->GetWorkerWindow(_hwndMB), riid, ppvObj);
}
if (*ppvObj)
hres = S_OK;
//TraceMsg(TF_BAND, "ISFBand::GetObject(%d) returns %x", dwId, hres);
return hres;
}
/*----------------------------------------------------------
Purpose: CDelegateDropTarget::OnDropDDT
Returns: S_OK if the drop was handled. Otherwise S_FALSE.
*/
HRESULT CMenuSFToolbar::OnDropDDT(IDropTarget *pdt, IDataObject *pdtobj, DWORD * pgrfKeyState, POINTL pt, DWORD *pdwEffect)
{
// Since the modal drag-drop loop released the capture, take it
// back so we behave properly.
KillTimer(_hwndMB, MBTIMER_DRAGPOPDOWN);
HRESULT hres = S_FALSE;
// We need to say that the last drag leave is really the drop.
_fHasDrop = TRUE;
_idCmdDragging = -1;
MyLockSetForegroundWindow(TRUE);
// Only send an hwnd to the callback if the drop source is external
if (!(_pcmb->_dwFlags & SMINIT_RESTRICT_DRAGDROP) &&
(S_FALSE == CallCB(NULL, SMC_SFDDRESTRICTED, (WPARAM)pdtobj,
(LPARAM)(_iDragSource < 0 ? GetHWNDForUIObject() : NULL)) ))
{
_RemoveChevron();
hres = CSFToolbar::OnDropDDT(pdt, pdtobj, pgrfKeyState, pt, pdwEffect);
_AddChevron();
}
return hres;
}
PIBDATA CMenuSFToolbar::_AddOrderItemTB(PORDERITEM poi, int index, TBBUTTON* ptbb)
{
PIBDATA pibd = CSFToolbar::_AddOrderItemTB(poi, index, ptbb);
if (pibd)
{
if (pibd->GetFlags() & SMIF_SUBMENU)
{
_fHasSubMenu = TRUE;
}
// A null pidl means one of two things:
// 1) a menu item called "(Empty)", or
// 2) a chevron button
// Are we adding the chevron button?
if (!pibd->GetPidl() && _fHasDemotedItems)
{
// Yes; the chevron is either the first item in the toolbar or the last item.
int iPos = (index == 0) ? 0 : ToolBar_ButtonCount(_hwndTB) - 1;
_idCmdChevron = GetButtonCmd(_hwndTB, iPos);
}
}
return pibd;
}
BOOL CMenuSFToolbar::_AddPidl(LPITEMIDLIST pidl, int index)
{
BOOL bRet;
// Is this item being added to an empty menu?
if (_fEmpty)
{
// Yes; remove the empty menu item
InlineDeleteButton(0);
DPA_DeletePtr(_hdpa, 0);
_fEmpty = FALSE;
if (_dwFlags & SMSET_NOEMPTY)
_fDontShowEmpty = FALSE;
bRet = CSFToolbar::_AddPidl(pidl, index);
// Failed to add new item?
if (!bRet)
{
// Yes; add the empty menu item back
OrderList_Append(_hdpa, NULL, -1); // Add a bogus pidl
_fEmpty = TRUE;
_fHasDemotedItems = FALSE;
if (_dwFlags & SMSET_NOEMPTY)
_fDontShowEmpty = TRUE;
}
}
else
bRet = CSFToolbar::_AddPidl(pidl, index);
return bRet;
}
BOOL CMenuSFToolbar::_ReBindToFolder(LPCITEMIDLIST pidl)
{
// We may be able to share this code with the code in _FillToolbar, but the difference is,
// in Fill Toolbar, the Toolbar Button does not have a Sub Menu. We reinitialize one we save away,
// and force it back into the child button. Here, we have the luxury of having the Sub Menu still
// in the toolbar button. I may be able to extract common code into a separate function. Left
// as an exercise to the reader.
// Need special Handling for this. We need to free the sub menu and
// rebind to it ifit's up.
BOOL fBound = FALSE;
TBBUTTONINFO tbinfo = {0};
tbinfo.dwMask = TBIF_COMMAND | TBIF_LPARAM;
LPCITEMIDLIST pidlItem = _GetButtonFromPidl(ILFindLastID(pidl), &tbinfo, NULL);
if (pidlItem)
{
CMenuData* pmd = (CMenuData*)tbinfo.lParam;
if (EVAL(pmd))
{
IShellFolderBand* psfb;
// We have the Toolbar button into, we should see if it has a sub menu associated with it.
if (SUCCEEDED(pmd->GetSubMenu(&SID_MenuShellFolder, IID_IShellFolderBand, (void**)&psfb)))
{
// It does. Then reuse!
LPITEMIDLIST pidlFull = NULL;
IShellFolder* psf = NULL;
if(_pasf2)
{
LPITEMIDLIST pidlFolder, pidlChild;
// Remember: Folder pidls must be unwrapped.
_pasf2->UnWrapIDList(pidlItem, 1, NULL, &pidlFolder, &pidlChild, NULL);
pidlFull = ILCombine(pidlFolder, pidlChild);
ILFree(pidlChild);
ILFree(pidlFolder);
}
else
{
// Not a wrapped guy, Sweet!
pidlFull = ILCombine(_pidl, pidlItem);
}
_psf->BindToObject(pidlItem, NULL, IID_IShellFolder, (void**)&psf);
if (psf)
{
if (pidlFull)
{
fBound = SUCCEEDED(psfb->InitializeSFB(psf, pidlFull));
if (fBound)
{
_pcmb->_nItemSubMenu = tbinfo.idCommand;
}
}
psf->Release();
}
ILFree(pidlFull);
psfb->Release();
}
}
}
return fBound;
}
HRESULT CMenuSFToolbar::OnTranslatedChange(LONG lEvent, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
{
HRESULT hres = CSFToolbar::OnTranslatedChange(lEvent, pidl1, pidl2);
// Deal with SubMenus:
if (SUCCEEDED(hres))
{
switch(lEvent)
{
case SHCNE_RENAMEFOLDER:
if (_IsChildID(pidl2, TRUE))
{
_ReBindToFolder(pidl2);
}
break;
case SHCNE_RMDIR:
if (_IsChildID(pidl1, TRUE))
{
_ReBindToFolder(pidl1);
}
break;
case SHCNE_EXTENDED_EVENT:
{
SHChangeDWORDAsIDList UNALIGNED * pdwidl = (SHChangeDWORDAsIDList UNALIGNED *)pidl1;
if (pidl2 && _IsChildID(pidl2, TRUE))
{
if (!SHChangeMenuWasSentByMe(this, pidl1))
{
DWORD dwFlags = SMINV_NOCALLBACK; // So that we don't doubly increment
SMDATA smd = {0};
smd.dwMask = SMDM_SHELLFOLDER;
smd.pidlFolder = _pidl;
smd.pidlItem = ILFindLastID(pidl2);
// Syncronize Promotion state.
if (pdwidl->dwItem1 == SHCNEE_PROMOTEDITEM)
{
dwFlags |= SMINV_PROMOTE;
}
else if (pdwidl->dwItem1 == SHCNEE_DEMOTEDITEM)
{
dwFlags |= SMINV_DEMOTE;
}
// Are we actually doing something?
if (SMINV_NOCALLBACK != dwFlags)
{
v_InvalidateItem(&smd, dwFlags);
}
}
}
}
break;
default:
break;
}
}
return hres;
}
/*----------------------------------------------------------
Purpose: IShellChangeNotify::OnChange
*/
HRESULT CMenuSFToolbar::OnChange(LONG lEvent, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
{
HRESULT hres = E_FAIL;
// If we're in the middle of being destroyed, don't process this.
if (!_hwndMB)
return S_OK;
_pcmb->_pmbState->PushChangeNotify();
SMCSHCHANGENOTIFYSTRUCT shns;
shns.lEvent = lEvent;
shns.pidl1 = pidl1;
shns.pidl2 = pidl2;
CallCB(NULL, SMC_SHCHANGENOTIFY, NULL, (LPARAM)&shns); // Ignore return value. Notify only.
// Since we may be removing the selected item, we want the selection to move to the next item
int iHot = ToolBar_GetHotItem(_hwndMB);
hres = CSFToolbar::OnChange(lEvent, pidl1, pidl2);
// BUGBUG(lamadio): Investigate moving this into the OnTranslateChange handler. Saves us
// from having to wrap them again.... This could be a big perf win....
// Is this a child of this toolbar is some shape or form?
// 1) The changing pidl is a child of this pane.
// 2) What the pidl is changing to is in this pane (For renames)
// 3) Updatedirs. Recursive change notifies must forward update dirs all the way down the chain.
// 4) EXTENDED events with a pidl2 == NULL. This means Reorder all your items.
if (_IsChildID(pidl1, FALSE) ||
(pidl2 && _IsChildID(pidl2, FALSE)) ||
lEvent == SHCNE_UPDATEDIR ||
(lEvent == SHCNE_EXTENDED_EVENT &&
pidl2 == NULL))
{
// We need to forward this down then.
HRESULT hresInner = _pcmb->ForwardChangeNotify(lEvent, pidl1, pidl2);
// Did either of us handle this change?
if (SUCCEEDED(hresInner) || SUCCEEDED(hres))
{
hres = S_OK;
}
else if (lEvent != SHCNE_EXTENDED_EVENT) // Don't bother with extended events...
{
// Ok so neither of us handled this?
// Must be the SHChangeNotifyCollapsing code that collapses
// the Directory Create and item create into a single item create.
// We need to force an update dir on ourselves so that we get this change.
hres = CSFToolbar::OnChange(SHCNE_UPDATEDIR, pidl1, pidl2);
}
}
// Set the hot item back, wrapping if necessary.
if (ToolBar_GetHotItem(_hwndMB) != iHot)
SetHotItem(1, iHot, -1, 0);
_pcmb->_pmbState->PopChangeNotify();
return hres;
}
void CMenuSFToolbar::_OnDragBegin(int iItem, DWORD dwPreferedEffect)
{
// During drag and drop, allow dialogs to collapse menu.
MyLockSetForegroundWindow(FALSE);
CSFToolbar::_OnDragBegin(iItem, DROPEFFECT_MOVE);
if (_fEditMode)
SetTimer(_hwndTB, MBTIMER_ENDEDIT, MBTIMER_ENDEDITTIME, 0);
}
void CMenuSFToolbar::v_SendMenuNotification(UINT idCmd, BOOL fClear)
{
if (fClear)
{
// If we're clearing, tell the browser
PostMessage(_pcmb->_pmbState->GetSubclassedHWND(), WM_MENUSELECT,
MAKEWPARAM(0, -1), NULL);
}
else
{
PIBDATA pibdata = _IDToPibData(idCmd);
LPCITEMIDLIST pidl;
// Only send notifications for non submenu items
if (EVAL(pibdata) && (pidl = pibdata->GetPidl()))
{
CallCB(pidl, SMC_SFSELECTITEM, 0, 0);
// Don't free Pidl
}
}
}
LRESULT CMenuSFToolbar::_OnGetObject(NMOBJECTNOTIFY* pnmon)
{
pnmon->hResult = QueryInterface(*pnmon->piid, &pnmon->pObject);
return 1;
}
LRESULT CMenuSFToolbar::_OnNotify(LPNMHDR pnm)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
LRESULT lres = 0;
// These are notifies we handle even when disengaged from the message hook.
switch (pnm->code)
{
case TBN_DELETINGBUTTON:
if (_fEmptyingToolbar)
return 0;
else
goto DoDefault;
break;
case TBN_GETDISPINFOA:
case TBN_GETDISPINFOW:
case NM_CUSTOMDRAW:
goto DoDefault;
}
// Pager notifications MUST be forwarded even when the message hook is disengaged.
if((pnm->code <= PGN_FIRST) && (pnm->code >= PGN_LAST))
{
goto DoNotify;
}
// Is the Global Message filter Disengaged? This will happen when the Subclassed window
// looses activation to a dialog box of some kind.
if (lres == 0 && !GetMessageFilter()->IsEngaged())
{
// Yes; We've lost activation so we don't want to track like a normal menu...
// For hot item change, return 1 so that the toolbar does not change the hot item.
if (pnm->code == TBN_HOTITEMCHANGE && _pcmb->_fMenuMode)
return 1;
// For all other items, don't do anything....
return 0;
}
DoNotify:
switch (pnm->code)
{
case PGN_SCROLL:
KillTimer(_hwndMB, MBTIMER_DRAGPOPDOWN);
if (_pcmb->_fInSubMenu)
_pcmb->_SubMenuOnSelect(MPOS_CANCELLEVEL);
_fSuppressUserMonitor = TRUE;
break;
case TBN_GETOBJECT:
lres = _OnGetObject((NMOBJECTNOTIFY*)pnm);
break;
case TBN_DRAGOUT:
{
TBNOTIFY *ptbn = (TBNOTIFY*)pnm;
if (!_fEmpty && ptbn->iItem != _idCmdChevron &&
!(_pcmb->_dwFlags & SMINIT_RESTRICT_DRAGDROP) &&
(S_FALSE == CallCB(NULL, SMC_SFDDRESTRICTED, NULL, NULL)))
{
// We're now in edit mode
_fEditMode = TRUE;
_idCmdDragging = ptbn->iItem;
_MarkItem(ptbn->iItem);
lres = 1; // Allow the drag to occur
goto DoDefault;
}
else
lres = 0; // Do not allow the drag out.
}
break;
default:
DoDefault:
lres = CMenuToolbarBase::_OnNotify(pnm);
if (lres == 0)
{
lres = CSFToolbar::_OnNotify(pnm);
}
break;
}
return lres;
}
void CMenuSFToolbar::CreateToolbar(HWND hwndParent)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
CSFToolbar::_CreateToolbar(hwndParent);
if (_hwndPager)
{
SHSetWindowBits(_hwndPager, GWL_STYLE, PGS_DRAGNDROP, PGS_DRAGNDROP);
SHSetWindowBits(_hwndPager, GWL_STYLE, PGS_AUTOSCROLL, PGS_AUTOSCROLL);
SHSetWindowBits(_hwndPager, GWL_STYLE, PGS_HORZ|PGS_VERT,
_fVertical ? PGS_VERT : PGS_HORZ);
}
_hwndMB = _hwndTB;
CMenuToolbarBase::CreateToolbar(hwndParent);
// By "Registering optimized" means that someone else is going to pass the change to us,
// we don't need to register for it. This is for the disjoint Fast Items | Programs menu case.
// We still need top level change notify registration for Favorites, Documents, Printers and Control
// Panel (Depending on their visibility)
//
if (_pcmb->_uId == MNFOLDER_IS_PARENT ||
(_dwFlags & SMSET_DONTREGISTERCHANGENOTIFY))
_fRegisterChangeNotify = FALSE;
// This is a good as spot as any to do this:
_RegisterToolbar();
}
HKEY CMenuSFToolbar::_GetKey(LPCITEMIDLIST pidl)
{
HKEY hMenuKey;
DWORD dwDisp;
TCHAR szDisplay[MAX_PATH];
if (!_hKey)
return NULL;
_ObtainPIDLName(pidl, szDisplay, ARRAYSIZE(szDisplay));
RegCreateKeyEx(_hKey, szDisplay, NULL, NULL,
REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE,
NULL, &hMenuKey, &dwDisp);
TraceMsg(TF_MENUBAND, "%d is setting %s\'s Key to %d", _hKey, szDisplay, hMenuKey);
return hMenuKey;
}
//***
// NOTES
// idtCmd is currently always -1. we'll need other values when we're
// called from CallCB. however we can't do that until we move idtCmd
// 'down' into CallCB.
HRESULT CMenuSFToolbar::v_GetState(int idtCmd, LPSMDATA psmd)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
HRESULT hres = E_FAIL;
CMenuData* pdata;
LPITEMIDLIST pidl = NULL;
psmd->dwMask = SMDM_SHELLFOLDER;
if (idtCmd == -1)
idtCmd = GetButtonCmd(_hwndTB, ToolBar_GetHotItem(_hwndTB));
pdata = (CMenuData*)_IDToPibData(idtCmd);
if (EVAL(pdata))
{
pidl = pdata->GetPidl();
ASSERT(IS_VALID_PIDL(pidl));
}
if (pidl)
{
if( _pasf2 && S_OK == _pasf2->UnWrapIDList(pidl, 1, &psmd->psf, &psmd->pidlFolder, &psmd->pidlItem, NULL))
{
/*NOTHING*/
;
}
else
{
// Then it must be a straight ShellFolder.
psmd->psf = _psf;
if (EVAL(psmd->psf))
psmd->psf->AddRef();
psmd->pidlFolder = ILClone(_pidl);
psmd->pidlItem = ILClone(ILFindLastID(pidl));
}
psmd->uIdParent = _pcmb->_uId;
psmd->punk = SAFECAST(_pcmb, IShellMenu*);
psmd->punk->AddRef();
hres = S_OK;
}
return hres;
}
HRESULT CMenuSFToolbar::CallCB(LPCITEMIDLIST pidl, DWORD dwMsg, WPARAM wParam, LPARAM lParam)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
if (!_pcmb->_psmcb)
return S_FALSE;
SMDATA smd;
HRESULT hres = S_FALSE;
BOOL fDestroy = FALSE;
// todo: call v_GetState (but need idCmd for pidl)
smd.dwMask = SMDM_SHELLFOLDER;
if (pidl)
{
// We used to unwrap the pidl here in the case of AUGMISF, but why? In the Callback, we only
// needed the Full pidl for Executing and for Darwin. The unwrap is an expensive call that in
// the majority case wasn't even used. Put it on the client to unwrap it. Start Menu is the
// only user of Augmented shell folders anyway....
smd.psf = _psf;
smd.pidlFolder = _pidl;
smd.pidlItem = (LPITEMIDLIST)pidl;
}
else
{
// Null pidl means tell the callback about me...
smd.pidlItem = ILClone(ILFindLastID(_pidl));
smd.pidlFolder = ILClone(_pidl);
ILRemoveLastID(smd.pidlFolder);
smd.psf = NULL; // Incase bind fails.
IEBindToObject(smd.pidlFolder, &smd.psf);
fDestroy = TRUE;
}
smd.uIdParent = _pcmb->_uId;
smd.uIdAncestor = _pcmb->_uIdAncestor;
smd.punk = SAFECAST(_pcmb, IShellMenu*);
smd.pvUserData = _pcmb->_pvUserData;
hres = _pcmb->_psmcb->CallbackSM(&smd, dwMsg, wParam, lParam);
if (fDestroy)
{
ATOMICRELEASE(smd.psf);
ILFree(smd.pidlFolder);
ILFree(smd.pidlItem);
}
return hres;
}
HRESULT CMenuSFToolbar::v_CallCBItem(int idtCmd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
HRESULT hres = E_FAIL;
CMenuData* pdata = (CMenuData*)_IDToPibData(idtCmd);
ASSERT(pdata);
ASSERT(pdata->GetPidl() == NULL || IS_VALID_PIDL(pdata->GetPidl()));
if (pdata)
hres = CallCB(pdata->GetPidl(), uMsg, wParam, lParam);
return hres;
}
HRESULT CMenuSFToolbar::v_GetSubMenu(int idCmd, const GUID* pguidService, REFIID riid, void** ppvObj)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
CMenuData* pdata = (CMenuData*)_IDToPibData(idCmd);
HRESULT hres = E_FAIL;
ASSERT(IS_VALID_WRITE_PTR(ppvObj, void*));
*ppvObj = NULL;
ASSERT(pdata);
if (pdata && pdata->GetFlags() & SMIF_SUBMENU)
{
hres = pdata->GetSubMenu(pguidService, riid, (void**)ppvObj);
if ( FAILED(hres) && IsEqualGUID(riid, IID_IShellMenu))
{
hres = CallCB(pdata->GetPidl(), SMC_GETSFOBJECT, (WPARAM)&riid, (LPARAM)ppvObj);
if (SUCCEEDED(hres))
{
BOOL fCache = TRUE;
if (S_OK != hres)
{
hres = E_FAIL;
IShellMenu* psm = (IShellMenu*) new CMenuBand();
if (psm)
{
IShellFolder* psf = NULL;
LPITEMIDLIST pidlItem = pdata->GetPidl();
LPITEMIDLIST pidlFolder = _pidl;
BOOL fDestroy = FALSE;
IShellMenuCallback* psmcb;
// Ask the callback if they want to supply a different callback
// object for this sub menu. If they do, then use what they
// pass back NOTE: If they pass back S_OK, it's perfectly Ok,
// for them to pass back a NULL psmcb. This means, I don't want
// my child to have a callback. Use the default.
// If they don't handle it, then use their pointer.
if (S_FALSE == CallCB(pdata->GetPidl(), SMC_GETSFOBJECT,
(WPARAM)&IID_IShellMenuCallback, (LPARAM)&psmcb))
{
psmcb = _pcmb->_psmcb;
if (psmcb)
psmcb->AddRef();
}
// This has to be before the unwrap because it does name resolution through
// the Augmented ISF.
HKEY hMenuKey = _GetKey(pidlItem);
if (_pasf2)
{
if (S_OK == _pasf2->UnWrapIDList(pdata->GetPidl(), 1, &psf, &pidlFolder, &pidlItem, NULL))
{
psf->Release(); // I don't need this
psf = NULL;
fDestroy = TRUE;
}
_pasf2->BindToObject(pdata->GetPidl(), NULL, IID_IShellFolder, (LPVOID*)&psf);
}
// Inherit the flags from the parent...
DWORD dwFlags = SMINIT_VERTICAL |
(_pcmb->_dwFlags & (SMINIT_RESTRICT_CONTEXTMENU |
SMINIT_RESTRICT_DRAGDROP |
SMINIT_MULTICOLUMN));
LPITEMIDLIST pidlFull = ILCombine(pidlFolder, pidlItem);
if (psf == NULL)
{
hres = _psf->BindToObject(pidlItem, NULL, IID_IShellFolder, (void**)&psf);
}
DWORD dwAttrib = SFGAO_FILESYSTEM;
LPCITEMIDLIST pidlWrappedItem = pdata->GetPidl();
// _psf can be an augmented shell folder. Use the wrapped item....
_psf->GetAttributesOf(1, (LPCITEMIDLIST*)&pidlWrappedItem, &dwAttrib);
// We need to sniff the pidl to see if it's a junction
// The format of a Filesytem pidl is
// WORD cb;
// BYTE bFlags;
// if (pidl->bFlags & Namespace Junction)
// ...Or...
// It's not in the filesystem.
if (SIL_GetType(pidlItem) & 0x80 || !(dwAttrib & SFGAO_FILESYSTEM))
{
// We're not going to persist anything
RegCloseKey(hMenuKey);
hMenuKey = NULL;
psmcb = NULL; // We're not going to pass a callback. NOTE: We don't need to release this
dwFlags &= ~SMINIT_MULTICOLUMN; // No multi on FShortcut...
fCache = FALSE;
}
UINT uIdAncestor = _pcmb->_uIdAncestor;
if (uIdAncestor == ANCESTORDEFAULT)
uIdAncestor = idCmd;
psm->Initialize(psmcb, MNFOLDER_IS_PARENT, uIdAncestor, dwFlags);
if (psf)
{
psm->SetShellFolder(psf, pidlFull, hMenuKey,
_dwFlags & (SMSET_HASEXPANDABLEFOLDERS | SMSET_USEBKICONEXTRACTION));
hres = psm->QueryInterface(riid, ppvObj);
psf->Release();
}
ILFree(pidlFull);
psm->Release();
if (psmcb)
psmcb->Release();
if (fDestroy)
{
ILFree(pidlFolder);
ILFree(pidlItem);
}
}
}
if (*ppvObj)
{
if (fCache)
{
pdata->SetSubMenu((IUnknown*)*ppvObj);
}
VARIANT Var;
Var.vt = VT_UNKNOWN;
Var.byref = SAFECAST(_pcmb->_pmbm, IUnknown*);
// Set the CMenuBandMetrics into the new menuband
IUnknown_Exec((IUnknown*)*ppvObj, &CGID_MenuBand, MBANDCID_SETFONTS, 0, &Var, NULL);
// Set the CMenuBandState into the new menuband
Var.vt = VT_INT_PTR;
Var.byref = _pcmb->_pmbState;
IUnknown_Exec((IUnknown*)*ppvObj, &CGID_MenuBand, MBANDCID_SETSTATEOBJECT, 0, &Var, NULL);
}
}
}
}
return hres;
}
DWORD CMenuSFToolbar::v_GetFlags(int idCmd)
{
CMenuData* pdata = (CMenuData*)_IDToPibData(idCmd);
// Toolbar is allowed to pass a bad command in the case of background erase
if (pdata)
return pdata->GetFlags();
else
return 0;
// BUGBUG (lamadio): Should I query each time? For like Volitile items?
// SMINFO SMINFO = {SMIM_FLAGS};
// CallCB(pdata->GetPidl(), SMC_GETFSINFO, (WPARAM)idCmd, (LPARAM)&SMINFO);
}
// This is to tell all other clients that we updated the promotion state of something.
void CMenuSFToolbar::BroadcastIntelliMenuState(LPCITEMIDLIST pidlItem, BOOL fPromoted)
{
LPITEMIDLIST pidlFolder;
LPITEMIDLIST pidlItemUnwrapped;
LPITEMIDLIST pidlFull;
if( _pasf2 && S_OK == _pasf2->UnWrapIDList(pidlItem, 1, NULL, &pidlFolder, &pidlItemUnwrapped, NULL))
{
pidlFull = ILCombine(pidlFolder, pidlItemUnwrapped);
ILFree(pidlFolder);
ILFree(pidlItemUnwrapped);
}
else
{
pidlFull = ILCombine(_pidl, pidlItem);
}
SHSendChangeMenuNotify(this,
fPromoted ? SHCNEE_PROMOTEDITEM : SHCNEE_DEMOTEDITEM,
0, pidlFull);
ILFree(pidlFull);
}
HRESULT CMenuSFToolbar::v_ExecItem(int idCmd)
{
CMenuData* pdata = (CMenuData*)_IDToPibData(idCmd);
HRESULT hres = E_FAIL;
if (pdata && !_fEmpty && idCmd != _idCmdChevron)
{
// STRESS: pdata was becomming 0x8 for some reason after the InvokeDefault.
// I assume that this call was causing a flush, which frees our list of pidls.
// So, I'm cloning it. I also changed the order, so that we'll just fire the
// UEM event.
LPITEMIDLIST pidl = ILClone(pdata->GetPidl());
if (pidl)
{
ASSERT(IS_VALID_PIDL(pidl));
SMDATA smd;
smd.dwMask = SMDM_SHELLFOLDER;
smd.pidlFolder = _pidl;
smd.pidlItem = pidl;
v_InvalidateItem(&smd, SMINV_PROMOTE | SMINV_FORCE);
hres = CallCB(pidl, SMC_SFEXEC, 0, 0);
// Did the Callback handle this execute for us?
if (hres == S_FALSE)
{
// No, Ok, do it ourselves.
hres = SHInvokeDefaultCommand(_hwndTB, _psf, pidl);
}
ILFree(pidl);
}
}
return hres;
}
HRESULT CMenuSFToolbar::v_GetInfoTip(int idCmd, LPTSTR psz, UINT cch)
{
CMenuData* pdata = (CMenuData*)_IDToPibData(idCmd);
HRESULT hres = E_FAIL;
if (_fEmpty || !pdata)
return hres;
hres = CallCB(pdata->GetPidl(), SMC_GETSFINFOTIP, (WPARAM)psz, (LPARAM)cch);
if (S_FALSE == hres)
{
hres = E_FAIL;
if (GetInfoTip(_psf, pdata->GetPidl(), psz, cch))
{
hres = NOERROR;
}
}
return hres;
}
void CMenuSFToolbar::v_ForwardMouseMessage(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
RECT rc;
POINT pt;
HWND hwndFwd;
// These are in screen coords
pt.x = GET_X_LPARAM(lParam);
pt.y = GET_Y_LPARAM(lParam);
hwndFwd = _hwndPager ? _hwndPager : _hwndTB;
GetWindowRect(hwndFwd, &rc);
if (PtInRect(&rc, pt))
{
MapWindowPoints(NULL, hwndFwd, &pt, 1);
HWND hwnd = ChildWindowFromPoint(hwndFwd, pt);
if (hwnd)
{
MapWindowPoints(hwndFwd, hwnd, &pt, 1);
}
else
{
hwnd = hwndFwd;
}
SendMessage(hwnd, uMsg, wParam, MAKELONG(pt.x, pt.y));
}
}
HRESULT CMenuSFToolbar::OnWinEvent(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *plres)
{
switch(uMsg)
{
case WM_SYSCOLORCHANGE:
if (_hwndPager)
Pager_SetBkColor(_hwndPager, GetSysColor(COLOR_MENU));
// Change the color, so that we can see it.
ToolBar_SetInsertMarkColor(_hwndMB, GetSysColor(COLOR_MENUTEXT));
break;
}
HRESULT hres = CMenuToolbarBase::OnWinEvent(hwnd, uMsg, wParam, lParam, plres);
if (hres != S_OK)
hres = CSFToolbar::OnWinEvent(hwnd, uMsg, wParam, lParam, plres);
return hres;
}
BOOL CMenuSFToolbar::v_UpdateIconSize(UINT uIconSize, BOOL fUpdateButtons)
{
if (uIconSize == -1)
uIconSize = _uIconSize;
_uIconSizeMB = uIconSize;
return _UpdateIconSize(uIconSize, fUpdateButtons);
}
HRESULT CMenuSFToolbar::GetShellFolder(LPITEMIDLIST* ppidl, REFIID riid, void** ppvObj)
{
HRESULT hres = E_FAIL;
*ppvObj = NULL;
if (_psf)
{
hres = _psf->QueryInterface(riid, ppvObj);
}
if(SUCCEEDED(hres) && ppidl)
{
*ppidl = ILClone(_pidl);
if (! *ppidl)
{
(*(IUnknown**)ppvObj)->Release();
hres = E_FAIL;
}
}
return hres;
}
LRESULT CMenuSFToolbar::_OnTimer(WPARAM wParam)
{
switch(wParam)
{
case MBTIMER_ENDEDIT:
KillTimer(_hwndTB, wParam);
_fEditMode = FALSE;
break;
case MBTIMER_CLICKUNHANDLE:
KillTimer(_hwndTB, wParam);
_fClickHandled = FALSE;
break;
default:
return CMenuToolbarBase::_OnTimer(wParam);
}
return 1;
}
LRESULT CMenuSFToolbar::_OnDropDown(LPNMTOOLBAR pnmtb)
{
if (GetAsyncKeyState(VK_LBUTTON) < 0 && _fEditMode)
{
// Are we in edit mode?
if (_fEditMode)
{
// Yes, mark the item as the item that is subject to moving
_MarkItem(pnmtb->iItem);
}
return TBDDRET_TREATPRESSED;
}
return CMenuToolbarBase::_OnDropDown(pnmtb);
}
/*----------------------------------------------------------
Purpose: In the context of a menuband, marking means putting
a black rectangle around the item currently being dragged.
*/
void CMenuSFToolbar::_MarkItem(int idCmd)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
// Un-highlight the previously moved button
if (0 <= _pcmb->_nItemMove)
{
// Should item move be a member of SFToolbar?
ToolBar_MarkButton(_hwndTB, _pcmb->_nItemMove, FALSE);
_pcmb->_nItemMove = -1;
}
if (_fEditMode)
{
_pcmb->_nItemMove = idCmd;
ToolBar_MarkButton(_hwndTB, _pcmb->_nItemMove, TRUE);
}
}
STDMETHODIMP CMenuSFToolbar::IsWindowOwner(HWND hwnd)
{
if (_hwndTB == hwnd || _hwndPager == hwnd || HWND_BROADCAST == hwnd)
{
return S_OK;
}
else
{
return S_FALSE;
}
}
void CMenuSFToolbar::SetWindowPos(LPSIZE psize, LPRECT prc, DWORD dwFlags)
{
if (!_hwndPager)
{
CMenuToolbarBase::SetWindowPos(psize, prc, dwFlags);
return;
}
DWORD rectWidth = RECTWIDTH(*prc);
TraceMsg(TF_MENUBAND, "CMSFTB::SetWindowPos %d - (%d,%d,%d,%d)", psize?psize->cx:0,
prc->left, prc->top, prc->right, prc->bottom);
ShowWindow(_hwndPager, SW_SHOW);
::SetWindowPos(_hwndPager, NULL, prc->left, prc->top,
rectWidth, RECTHEIGHT(*prc), SWP_NOZORDER | SWP_NOACTIVATE | dwFlags);
if (psize)
SendMessage(_hwndTB, TB_SETBUTTONWIDTH, 0, MAKELONG(psize->cx, psize->cx));
SendMessage(_hwndPager, PGMP_RECALCSIZE, 0L, 0L);
}
void CMenuSFToolbar::SetParent(HWND hwndParent)
{
int nCmdShow = SW_SHOW;
if (hwndParent)
{
if (!_hwndTB)
CreateToolbar(hwndParent);
else
{
// make sure width is set correctly . . .
SendMessage(_hwndTB, TB_SETBUTTONWIDTH, 0, MAKELONG(_cxMin, _cxMax));
}
}
else
{
// As an optimization, we implement "disowning" ourselves
// as just moving ourselves offscreen. The previous parent
// still owns us. The parent is invariably the menusite.
RECT rc = {-1,-1,-1,-1};
SetWindowPos(NULL, &rc, 0);
nCmdShow = SW_HIDE;
}
HWND hwnd = _hwndPager ? _hwndPager: _hwndTB;
if (IsWindow(hwnd)) // JANK : Fix for bug #98253
{
::SetParent(hwnd, hwndParent);
SendMessage(hwnd, TB_SETPARENT, (WPARAM)hwndParent, NULL);
ShowWindow(hwnd, nCmdShow);
}
}
void CMenuSFToolbar::Expand(BOOL fExpand)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
TBBUTTON tbb;
DAD_ShowDragImage(FALSE);
// Since we're not sure if the Chevron is going to be visible we should remove it here
// Later we'll add it back in if it's needed.
_RemoveChevron();
// Loop through and apply the fExpand
int iNumButtons = ToolBar_ButtonCount(_hwndTB);
// We reset these when iterating.
_cPromotedItems = 0;
_fHasDemotedItems = FALSE;
int iHotItem = ToolBar_GetHotItem(_hwndMB);
// SendMessage(_hwndMB, WM_SETREDRAW, FALSE, 0);
for (int i = 0; i < iNumButtons; i++)
{
if (!ToolBar_GetButton(_hwndMB, i, &tbb))
continue;
CMenuData* pmd = (CMenuData*)tbb.dwData;
// Get the toolbar state. Toolbar can set things like
// TBSTATE_WRAP that we would go nuke.
DWORD dwState = tbb.fsState;
DWORD dwFlags = pmd ? pmd->GetFlags() : 0;
if (dwFlags & SMIF_DEMOTED)
{
// Are we expanding?
if (fExpand)
{
//Yes; Enable the button and remove the hidden state
dwState |= TBSTATE_ENABLED;
dwState &= ~TBSTATE_HIDDEN;
}
else
{
//No; Remove the Enabled state and hide the button
dwState |= TBSTATE_HIDDEN;
dwState &= ~TBSTATE_ENABLED;
}
_fHasDemotedItems = TRUE;
}
else if (dwFlags & SMIF_HIDDEN)
{
dwState |= TBSTATE_HIDDEN;
dwState &= ~TBSTATE_ENABLED;
}
else if (tbb.idCommand != _idCmdChevron)
{
dwState |= TBSTATE_ENABLED;
dwState &= ~TBSTATE_HIDDEN;
_cPromotedItems++;
}
// If the state has changed, then set it into the toolbar.
if (dwState != tbb.fsState)
ToolBar_SetState(_hwndTB, tbb.idCommand, dwState);
}
// _fExpand means "Draw as Expanded". We do not want to
// draw expanded when we have no demoted items.
_pcmb->_fExpanded = _fHasDemotedItems? fExpand : FALSE;
if (fExpand)
{
if (_pcmb->_pmbState)
{
_pcmb->_pmbState->SetExpand(TRUE);
_pcmb->_pmbState->HideTooltip(TRUE);
}
}
else
{
_AddChevron();
}
// Have the menubar think about changing its height
IUnknown_QueryServiceExec(_pcmb->_punkSite, SID_SMenuPopup, &CGID_MENUDESKBAR,
MBCID_SETEXPAND, _fHasDemotedItems?(int)_pcmb->_pmbState->GetExpand():FALSE, NULL, NULL);
// SendMessage(_hwndMB, WM_SETREDRAW, TRUE, 0);
_ToolbarChanged();
ToolBar_SetHotItem(_hwndMB, iHotItem);
if (_hwndPager)
UpdateWindow(_hwndPager);
UpdateWindow(_hwndTB);
// DAD_ShowDragImage(TRUE);
}
void CMenuSFToolbar::GetSize(SIZE* psize)
{
CMenuToolbarBase::GetSize(psize);
if (_fEmpty && _fDontShowEmpty)
{
psize->cy = 0;
TraceMsg(TF_MENUBAND, "CMSFT::GetSize (%d, %d)", psize->cx, psize->cy);
}
}
void CMenuSFToolbar::_RefreshInfo()
{
int cButton = ToolBar_ButtonCount(_hwndMB);
for (int iButton = 0; iButton < cButton; iButton++)
{
int idCmd = GetButtonCmd(_hwndTB, iButton);
if (idCmd != _idCmdChevron)
{
// Get the information from that button.
CMenuData* pmd = (CMenuData*)_IDToPibData(idCmd);
if (pmd)
{
SMINFO sminfo;
sminfo.dwMask = SMIM_FLAGS;
if (SUCCEEDED(_GetInfo(pmd->GetPidl(), &sminfo)))
{
pmd->SetFlags(sminfo.dwFlags);
}
}
}
}
}
void CMenuSFToolbar::_FindMinPromotedItems(BOOL fSetOrderStream)
{
// We need to iterate through the buttons and set the Promoted flag.
int cButton = ToolBar_ButtonCount(_hwndMB);
for (int iButton = 0; iButton < cButton; iButton++)
{
int idCmd = GetButtonCmd(_hwndTB, iButton);
if (idCmd != _idCmdChevron)
{
// Get the information from that button.
CMenuData* pmd = (CMenuData*)_IDToPibData(idCmd);
if (pmd)
{
PORDERITEM poi = pmd->GetOrderItem();
if (fSetOrderStream)
{
DWORD dwFlags = pmd->GetFlags();
OrderItem_SetFlags(poi, dwFlags);
}
else // Query the order stream
{
DWORD dwFlags = OrderItem_GetFlags(poi);
DWORD dwOldFlags = pmd->GetFlags();
// When reading the flags from the registry, we only care about the demote flag.
if (dwFlags & SMIF_DEMOTED)
{
dwOldFlags |= SMIF_DEMOTED;
}
else if (!(dwOldFlags & SMIF_SUBMENU)) // Don't promote sub menus.
{
// Force a promote
CallCB(pmd->GetPidl(), SMC_PROMOTE, 0, 0);
dwOldFlags &= ~SMIF_DEMOTED;
}
pmd->SetFlags(dwOldFlags);
}
}
}
}
}
void CMenuSFToolbar::v_Show(BOOL fShow, BOOL fForceUpdate)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
CMenuToolbarBase::v_Show(fShow, fForceUpdate);
if (fShow)
{
BOOL fDirty = _fDirty;
_fClickHandled = FALSE;
_RegisterToolbar();
_FillToolbar();
_pcmb->SetTracked(NULL); // Since hot item is NULL
ToolBar_SetHotItem(_hwndTB, -1);
if (_fEmpty && (_dwFlags & SMSET_NOEMPTY))
{
_fDontShowEmpty = TRUE;
}
else if (_fRefreshInfo && !fDirty) // Do we need to refresh our information?
{
// Yes;
_RefreshInfo();
}
// HACKHACK (lamadio) : There is a sizing issue, where the sizing between the
// toolbars gets preemted by a resize of the menubar before the size calculation completes.
// So:
// ShowDW - Asks each toolbar to calc it's width
// CMenuSFToolbar::v_Show - Does a _FillToolbar. Since (in this senario) an item
// Has been added, it calls _ToolbarChanged
// _ToolbarChanged - This says to the menubar, I've changed sizes, recalc.
// ResizeMenuBar - In the depths, it eventually calls OnPosRectChanged, which asks each
// Toolbar what it's size is. Since the menu portion has not calculated it yet,
// It has the old size which is has the old size of the sftoolbar. So everything
// Gets reset to that size.
//
// We only want to Call expand if we are dirty or the expand state has changed. We
// call for the Dirty case, because Expand does some neat stuff in calculating the
// number of promoted items. If the state has changed, we want to reflect that.
BOOL fExpand = _pcmb->_pmbState ? _pcmb->_pmbState->GetExpand() : FALSE;
if ((BOOL)_pcmb->_fExpanded != fExpand || fDirty || _fRefreshInfo)
{
fForceUpdate = TRUE;
Expand(fExpand);
}
// Only do this in the beginning.
if (_fFirstTime)
{
CallCB(NULL, SMC_GETMINPROMOTED, 0, (LPARAM)&_cMinPromotedItems);
if (_cPromotedItems < _cMinPromotedItems)
{
_FindMinPromotedItems(FALSE);
Expand(fExpand);
}
}
// Have the menubar think about changing its height
// we need to do this here because the menubar may have changed it's
// expand state independant of the pane.
IUnknown_QueryServiceExec(_pcmb->_punkSite, SID_SMenuPopup, &CGID_MENUDESKBAR,
MBCID_SETEXPAND, (int)_pcmb->_fExpanded, NULL, NULL);
// If we're dirty, have our parent consider promoting itself if there
// are promoted items in the menu, or demoting itself if there arn't.
// Don't worry, the parent won't do anything if it's already in that state.
if ( fDirty )
{
IUnknown_RefreshParent(_pcmb->_punkSite, _pidl,
((_cPromotedItems == 0)? SMINV_DEMOTE : SMINV_PROMOTE) | SMINV_NEXTSHOW);
}
// If it is empty, we want to auto expand.
// We have to do this before the update buttons, so that the size is calculate correctly.
if (_cPromotedItems == 0 && !_pcmb->_fExpanded)
Expand(TRUE);
if (fForceUpdate)
_UpdateButtons();
if (_fHasDemotedItems)
{
if (S_OK == CallCB(NULL, SMC_DISPLAYCHEVRONTIP, 0, 0))
{
_FlashChevron();
}
}
_fFirstTime = FALSE;
_fRefreshInfo = FALSE;
}
else
{
KillTimer(_hwndMB, MBTIMER_UEMTIMEOUT);
}
_fShowMB = _fShow = fShow;
// Reset these so we don't have stale information for the next drag drop cycle. NT #287914 (lamadio) 3.22.99
_tbim.iButton = -1;
_tbim.dwFlags = 0;
_idCmdDragging = -1;
// n.b. for !fShow, we don't kill the tracked site chain. we
// count on this in startmnu.cpp!CStartMenuCallback::_OnExecItem,
// where we walk up the chain to find all hit 'nodes'. if we need
// to change this we could fire a 'pre-exec' event.
}
void CMenuSFToolbar::v_UpdateButtons(BOOL fNegotiateSize)
{
CSFToolbar::_UpdateButtons();
if (_hwndTB && fNegotiateSize && _fVerticalMB)
NegotiateSize();
}
// this method invalidates a single item in the toolbar
HRESULT CMenuSFToolbar::v_InvalidateItem(LPSMDATA psmd, DWORD dwFlags)
{
ASSERT(_pcmb); // if you hit this assert, you haven't initialized yet.. call SetSite first
// Default to not not handling this event.
HRESULT hres = S_FALSE;
if (NULL == psmd)
{
if (dwFlags & SMINV_REFRESH)
{
// bugbug: Needs to be optimized
_Refresh();
hres = S_OK;
}
}
// CMenuSFToolbar only handles ShellFolder items.
// Is this a shell folder?
else if (psmd->dwMask & SMDM_SHELLFOLDER)
{
// Yes;
int i;
LPITEMIDLIST pidlButton = NULL;
SMINFO sminfo;
sminfo.dwMask = SMIM_FLAGS;
// Since this pidl is comming from an outside source,
// we may need to translate it to a wrapped pidl.
// Do we have a pidl Translator?
if (_ptscn)
{
// Yes;
LPITEMIDLIST pidlTranslated;
LPITEMIDLIST pidlDummy = NULL;
LPITEMIDLIST pidlToTranslate = ILCombine(psmd->pidlFolder, psmd->pidlItem);
if (pidlToTranslate)
{
LONG lEvent = 0, lEvent2;
LPITEMIDLIST pidlDummy1, pidlDummy2;
if (SUCCEEDED(_ptscn->TranslateIDs(&lEvent, pidlToTranslate, NULL, &pidlTranslated, &pidlDummy,
&lEvent2, &pidlDummy1, &pidlDummy2)))
{
// Get the button in the toolbar that corresponds to this pidl.
pidlButton = _GetButtonFromPidl(ILFindLastID(pidlTranslated), NULL, &i);
// if pidl does not get translated TranslateIDs returns the same pidl passed
// to the function
if (pidlTranslated != pidlToTranslate)
ILFree(pidlTranslated);
// Don't need to delete pidlDummy because it's not set.
ASSERT(pidlDummy == NULL);
ASSERT(pidlDummy1 == NULL);
ASSERT(pidlDummy2 == NULL);
}
ILFree(pidlToTranslate);
}
}
// Did we come from a non-augmented shell folder, or
// did the caller pass a wrapped pidl?
if (!pidlButton)
{
// Seems like it, we'll try to find the pidl they passed in
// Get the button in the toolbar that corresponds to this pidl.
pidlButton = _GetButtonFromPidl(psmd->pidlItem, NULL, &i);
}
// Did we find this pidl in the toolbar?
if (pidlButton)
{
int idCmd = GetButtonCmd(_hwndTB, i);
// Yes, Get the information from that button.
CMenuData* pmd = (CMenuData*)_IDToPibData(idCmd);
if (pmd)
{
BOOL fRefresh = FALSE;
DWORD dwFlagsUp = dwFlags;
DWORD dwOldItemFlags = pmd->GetFlags();
DWORD dwNewItemFlags = dwOldItemFlags;
if ((dwFlags & SMINV_DEMOTE) &&
(!(dwOldItemFlags & SMIF_DEMOTED) || dwFlags & SMINV_FORCE))
{
if (!(dwFlags & SMINV_NOCALLBACK))
{
CallCB(pidlButton, SMC_DEMOTE, 0, 0);
BroadcastIntelliMenuState(pidlButton, FALSE);
}
dwNewItemFlags |= SMIF_DEMOTED;
dwFlagsUp |= SMINV_DEMOTE;
}
else if ((dwFlags & SMINV_PROMOTE) &&
((dwOldItemFlags & SMIF_DEMOTED) || dwFlags & SMINV_FORCE))
{
if (!(dwFlags & SMINV_NOCALLBACK))
{
CallCB(pidlButton, SMC_PROMOTE, 0, 0);
BroadcastIntelliMenuState(pidlButton, TRUE);
}
dwNewItemFlags &= ~SMIF_DEMOTED;
dwFlagsUp |= SMINV_PROMOTE;
}
// Was it promoted and now Demoted or
// Was it demoted and now promoted
if ((dwNewItemFlags & SMIF_DEMOTED) ^
(dwOldItemFlags & SMIF_DEMOTED))
{
fRefresh = TRUE;
if (dwNewItemFlags & SMIF_DEMOTED)
{
// Yes; Then decrement the Promoted count
_cPromotedItems--;
// If we're decementing, then we not have a demoted item.
_fHasDemotedItems = TRUE;
// Have we dropped off the face of the earth?
if (_cPromotedItems == 0)
{
dwFlagsUp |= SMINV_DEMOTE;
Expand(TRUE);
}
else
{
fRefresh = FALSE;
}
}
else
{
int cButtons = ToolBar_ButtonCount(_hwndMB);
_cPromotedItems++;
if (cButtons == _cPromotedItems)
{
// if the button count is the number of promoted items,
// then we can't have any demoted items
// then we need to reset the _fHasDemotedItems flag so that
// we don't get a chevron and stuff...
_fHasDemotedItems = FALSE;
}
dwFlagsUp |= SMINV_PROMOTE;
fRefresh = TRUE;
}
}
if (fRefresh || dwFlags & SMINV_FORCE)
IUnknown_RefreshParent(_pcmb->_punkSite, _pidl, dwFlagsUp);
if (dwOldItemFlags != dwNewItemFlags || dwFlags & SMINV_FORCE)
{
if (dwFlags & SMINV_NEXTSHOW || !_fShow)
{
_fRefreshInfo = TRUE;
}
else
{
// Since we updated the flags, set them into the cache
pmd->SetFlags(dwNewItemFlags);
// Based on the new flags, do we enable?
DWORD dwState = ToolBar_GetState(_hwndTB, idCmd);
dwState |= TBSTATE_ENABLED;
if (dwNewItemFlags & SMIF_DEMOTED &&
!_pcmb->_fExpanded)
{
// No; We're not expanded and this is a demoted item
dwState |= TBSTATE_HIDDEN;
dwState &= ~TBSTATE_ENABLED;
_fHasDemotedItems = TRUE;
// Just in case the chevron is not there, we should
// try and add it. This call will never add more than 1
_AddChevron();
}
else if (!_fHasDemotedItems)
{
_RemoveChevron();
}
// Adjust the state of the button in the toolbar.
ToolBar_SetState(_hwndTB, idCmd, dwState);
_ToolbarChanged();
}
}
}
}
// We handled this one.
hres = S_OK;
}
return hres;
}
LRESULT CMenuSFToolbar::_DefWindowProc(HWND hwnd, UINT uMessage, WPARAM wParam, LPARAM lParam)
{
switch (uMessage)
{
case WM_GETOBJECT:
// Yet another poor design choice on the part of the accessibility team.
// Typically, if you do not answer a WM_* you return 0. They choose 0 as their success
// code.
return _DefWindowProcMB(hwnd, uMessage, wParam, lParam);
break;
}
return CSFToolbar::_DefWindowProc(hwnd, uMessage, wParam, lParam);
}
void CMenuSFToolbar::_SetFontMetrics()
{
CMenuToolbarBase::_SetFontMetrics();
if (_hwndPager && _pcmb->_pmbm)
Pager_SetBkColor(_hwndPager, _pcmb->_pmbm->_clrBackground);
}
int CMenuSFToolbar::_GetBitmap(int iCommandID, PIBDATA pibdata, BOOL fUseCache)
{
int iIcon = -1;
// If we don't have a pibdata, or we can't get an icon return.
if (!pibdata || pibdata->GetNoIcon())
return -1;
if (_dwFlags & SMSET_USEBKICONEXTRACTION)
{
LPITEMIDLIST pidlItem = pibdata->GetPidl();
// If the caller is using background icon extraction, we need them to provide a
// default icon that we are going to display until we get the real one. This is
// specifically to make favorites fast.
if (_iDefaultIconIndex == -1)
{
TCHAR szIconPath [MAX_PATH];
if (S_OK == CallCB(NULL, SMC_DEFAULTICON, (WPARAM)szIconPath, (LPARAM)&iIcon))
{
_iDefaultIconIndex = Shell_GetCachedImageIndex(szIconPath, iIcon, 0);
}
}
iIcon = _iDefaultIconIndex;
DWORD dwAttrib = 0;
if (pidlItem && SUCCEEDED(_psf->GetAttributesOf(1, (LPCITEMIDLIST*)&pidlItem, &dwAttrib)))
{
if (dwAttrib & SFGAO_FOLDER)
iIcon = II_FOLDER;
}
IShellTaskScheduler* pScheduler = _pcmb->_pmbState->GetScheduler();
if (pScheduler)
{
IShellFolder* psf = NULL;
LPITEMIDLIST pidlFolder = _pidl;
LPITEMIDLIST pidlItemUnwrapped;
// Since this can be an augmented shell folder, we should do the correct thing so that
// the icon extraction with the full pidl takes place correctly.
if( _pasf2 &&
S_OK == _pasf2->UnWrapIDList(pidlItem, 1, NULL, &pidlFolder, &pidlItemUnwrapped, NULL))
{
pidlItem = ILCombine(pidlFolder, pidlItemUnwrapped);
ILFree(pidlFolder);
ILFree(pidlItemUnwrapped);
}
else
{
psf = _psf;
}
// AddIconTask takes ownership of the pidl when psf is NULL and will free it.
HRESULT hres = AddIconTask(pScheduler, psf, pidlFolder, pidlItem,
s_IconCallback, (LPVOID)_hwndTB, iCommandID, NULL);
pScheduler->Release();
if (FAILED(hres))
{
// If that call failed for some reason, default to the shell32 impl.
goto DoSyncMap;
}
}
else
goto DoSyncMap;
}
else
{
DoSyncMap:
iIcon = CSFToolbar::_GetBitmap(iCommandID, pibdata, fUseCache);
}
return iIcon;
}
void CMenuSFToolbar::s_IconCallback(LPVOID pvData, UINT uId, UINT iIconIndex)
{
HWND hwnd = (HWND)pvData;
if (hwnd && IsWindow(hwnd))
{
DAD_ShowDragImage(FALSE);
SendMessage(hwnd, TB_CHANGEBITMAP, uId, iIconIndex);
DAD_ShowDragImage(TRUE);
}
}
HWND CMenuSFToolbar::GetHWNDForUIObject()
{
HWND hwnd = _pcmb->_pmbState->GetWorkerWindow(_hwndMB);
if (hwnd)
::SetWindowPos(hwnd, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE);
return hwnd;
}
HWND CMenuSFToolbar::CreateWorkerWindow()
{
return GetHWNDForUIObject();
}
Reference in a new issue Copy permalink