392 lines
11 KiB
C++
392 lines
11 KiB
C++
|
#include "stdafx.h"
|
||
|
|
||
|
#ifdef FEATURE_STARTPAGE
|
||
|
|
||
|
#include "pidlbutton.h"
|
||
|
|
||
|
#include <shellp.h>
|
||
|
|
||
|
namespace DirectUI
|
||
|
{
|
||
|
|
||
|
int PIDLButton::s_nImageSize = 0;
|
||
|
|
||
|
// Construction / Destruction
|
||
|
//
|
||
|
PIDLButton::~PIDLButton()
|
||
|
{
|
||
|
ILFree(_pidl);
|
||
|
}
|
||
|
|
||
|
|
||
|
HRESULT PIDLButton::Create(LPITEMIDLIST pidl, UINT nActive, OUT Element** ppElement)
|
||
|
{
|
||
|
*ppElement = NULL;
|
||
|
|
||
|
PIDLButton* pb = HNew <PIDLButton>();
|
||
|
if (!pb)
|
||
|
return E_OUTOFMEMORY;
|
||
|
|
||
|
HRESULT hr = pb->Initialize(pidl, nActive);
|
||
|
if (FAILED(hr))
|
||
|
{
|
||
|
pb->Destroy();
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
*ppElement = pb;
|
||
|
|
||
|
return S_OK;
|
||
|
}
|
||
|
|
||
|
HIMAGELIST GetSysImageList(UINT flags)
|
||
|
{
|
||
|
static HIMAGELIST s_imgList;
|
||
|
static HIMAGELIST s_imgListSmall;
|
||
|
if (!s_imgList)
|
||
|
{
|
||
|
Shell_GetImageLists(&s_imgList, &s_imgListSmall);
|
||
|
}
|
||
|
|
||
|
return (flags & SHGFI_SMALLICON) ? s_imgListSmall : s_imgList;
|
||
|
}
|
||
|
|
||
|
HRESULT PIDLButton::Initialize(LPITEMIDLIST pidl, UINT nActive)
|
||
|
{
|
||
|
_pidl = pidl;
|
||
|
|
||
|
Layout* pbl = NULL;
|
||
|
Element* peLabel = NULL;
|
||
|
Element* peImage = NULL;
|
||
|
Value* pvChildren;
|
||
|
ElementList* pel;
|
||
|
|
||
|
HRESULT hr = Button::Initialize(nActive);
|
||
|
if (FAILED(hr))
|
||
|
goto CleanUp;
|
||
|
|
||
|
// Create children
|
||
|
hr = Element::Create(0, &peLabel);
|
||
|
if (FAILED(hr))
|
||
|
goto CleanUp;
|
||
|
|
||
|
hr = Element::Create(0, &peImage);
|
||
|
if (FAILED(hr))
|
||
|
goto CleanUp;
|
||
|
|
||
|
// Create layout
|
||
|
hr = BorderLayout::Create(&pbl);
|
||
|
if (FAILED(hr))
|
||
|
goto CleanUp;
|
||
|
|
||
|
// Initialize
|
||
|
SetLayout(pbl);
|
||
|
|
||
|
// Setup children
|
||
|
peImage->SetLayoutPos(BLP_Left);
|
||
|
peImage->SetID(L"Image");
|
||
|
peLabel->SetLayoutPos(BLP_Client);
|
||
|
peLabel->SetID(L"Label");
|
||
|
|
||
|
Add(peImage);
|
||
|
Add(peLabel);
|
||
|
|
||
|
pel = GetChildren(&pvChildren);
|
||
|
|
||
|
// Initialize Shell's ImageList
|
||
|
GetSysImageList(0);
|
||
|
|
||
|
if (_pidl && pel && (pel->GetSize() >= 2))
|
||
|
{
|
||
|
IShellFolder *psf;
|
||
|
LPCITEMIDLIST pidlItem;
|
||
|
if (SUCCEEDED(SHBindToFolderIDListParent(NULL, _pidl, IID_PPV_ARG(IShellFolder, &psf), &pidlItem)))
|
||
|
{
|
||
|
int iSysIndex = SHMapPIDLToSystemImageListIndex(psf, pidlItem, NULL);
|
||
|
|
||
|
if (iSysIndex != -1)
|
||
|
{
|
||
|
HICON hi = ImageList_GetIcon(GetSysImageList(s_nImageSize), iSysIndex, 0);
|
||
|
Value *pvalIcon = Value::CreateGraphic(hi);
|
||
|
if (pvalIcon)
|
||
|
{
|
||
|
pel->GetItem(0)->SetValue(ContentProp, PI_Local, pvalIcon);
|
||
|
pvalIcon->Release();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
DestroyIcon(hi);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
TCHAR szOut[MAX_PATH];
|
||
|
if (SUCCEEDED(DisplayNameOf(psf, pidlItem, SHGDN_NORMAL, szOut, ARRAYSIZE(szOut))))
|
||
|
{
|
||
|
pel->GetItem(1)->SetContentString(szOut);
|
||
|
}
|
||
|
psf->Release();
|
||
|
}
|
||
|
}
|
||
|
pvChildren->Release();
|
||
|
|
||
|
return hr;
|
||
|
|
||
|
CleanUp:
|
||
|
|
||
|
if (pbl)
|
||
|
pbl->Destroy();
|
||
|
|
||
|
if (peImage)
|
||
|
peImage->Destroy();
|
||
|
|
||
|
if (peLabel)
|
||
|
peLabel->Destroy();
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT PIDLButton::InvokePidl()
|
||
|
{
|
||
|
HRESULT hr = E_FAIL;
|
||
|
IShellFolder *psf;
|
||
|
LPCITEMIDLIST pidlShort;
|
||
|
// Multi-level child pidl
|
||
|
if (SUCCEEDED(SHBindToFolderIDListParent(NULL, _pidl, IID_PPV_ARG(IShellFolder, &psf), &pidlShort)))
|
||
|
{
|
||
|
hr = SHInvokeDefaultCommand(GetHWND(), psf, pidlShort);
|
||
|
psf->Release();
|
||
|
}
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT PIDLButton::OnContextMenu(POINT *ppt)
|
||
|
{
|
||
|
HRESULT hr = E_FAIL;
|
||
|
IShellFolder *psf;
|
||
|
LPCITEMIDLIST pidlShort;
|
||
|
|
||
|
if (!GetHWND())
|
||
|
return hr;
|
||
|
|
||
|
if (ppt->x == -1) // Keyboard context menu
|
||
|
{
|
||
|
Value *pv;
|
||
|
const SIZE *psize = GetExtent(&pv);
|
||
|
ppt->x = psize->cx/2;
|
||
|
ppt->y = psize->cy/2;
|
||
|
pv->Release();
|
||
|
}
|
||
|
|
||
|
POINT pt;
|
||
|
GetRoot()->MapElementPoint(this, ppt, &pt);
|
||
|
|
||
|
ClientToScreen(GetHWND(), &pt);
|
||
|
|
||
|
|
||
|
// Multi-level child pidl
|
||
|
if (SUCCEEDED(SHBindToFolderIDListParent(NULL, _pidl, IID_PPV_ARG(IShellFolder, &psf), &pidlShort)))
|
||
|
{
|
||
|
IContextMenu *pcm;
|
||
|
|
||
|
if (SUCCEEDED(hr = psf->GetUIObjectOf(GetHWND(), 1, &pidlShort, IID_IContextMenu, NULL, (void **)&pcm)))
|
||
|
{
|
||
|
HMENU hmenu = ::CreatePopupMenu();
|
||
|
|
||
|
if (hmenu)
|
||
|
{
|
||
|
UINT uFlags = CMF_NORMAL;
|
||
|
if (GetKeyState(VK_SHIFT) < 0)
|
||
|
{
|
||
|
uFlags |= CMF_EXTENDEDVERBS;
|
||
|
}
|
||
|
|
||
|
#if 0 // REVIEW fabriced
|
||
|
if (_dwFlags & HOSTF_CANRENAME)
|
||
|
{
|
||
|
uFlags |= CMF_CANRENAME;
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
pcm->QueryContextMenu(hmenu, 0, IDM_QCM_MIN, IDM_QCM_MAX, uFlags);
|
||
|
|
||
|
// Remove "Create shortcut" from context menu because it creates
|
||
|
// the shortcut on the desktop, which the user can't see...
|
||
|
ContextMenu_DeleteCommandByName(pcm, hmenu, IDM_QCM_MIN, TEXT("link"));
|
||
|
|
||
|
// Remove "Cut" from context menu because we don't want objects
|
||
|
// to be deleted.
|
||
|
ContextMenu_DeleteCommandByName(pcm, hmenu, IDM_QCM_MIN, TEXT("cut"));
|
||
|
|
||
|
// Change "Delete" to "Remove from this list".
|
||
|
// If client doesn't support "delete" then nuke it outright.
|
||
|
ContextMenu_DeleteCommandByName(pcm, hmenu, IDM_QCM_MIN, TEXT("delete"));
|
||
|
|
||
|
#if 0 // REVIEW fabriced
|
||
|
if (_dwFlags & HOSTF_CANDELETE)
|
||
|
{
|
||
|
if (LoadString(_Module.GetResourceInstance(), IDS_SFTHOST_REMOVEFROMLIST, szBuf, ARRAYSIZE(szBuf)))
|
||
|
{
|
||
|
ModifyMenu(hmenu, iposDelete, MF_BYPOSITION | MF_STRING, GetMenuItemID(hmenu, iposDelete), szBuf);
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
#endif
|
||
|
_SHPrettyMenu(hmenu);
|
||
|
|
||
|
ASSERT(_pcm2Pop == NULL); // Shouldn't be recursing
|
||
|
pcm->QueryInterface(IID_PPV_ARG(IContextMenu2, &_pcm2Pop));
|
||
|
|
||
|
ASSERT(_pcm3Pop == NULL); // Shouldn't be recursing
|
||
|
pcm->QueryInterface(IID_PPV_ARG(IContextMenu3, &_pcm3Pop));
|
||
|
|
||
|
GetHWNDHost()->WndProc(GetHWND(), PBM_SETMENUFORWARD, 0, (LPARAM)this);
|
||
|
|
||
|
int idCmd = TrackPopupMenuEx(hmenu,
|
||
|
TPM_RETURNCMD | TPM_RIGHTBUTTON | TPM_LEFTALIGN,
|
||
|
pt.x, pt.y, GetHWND(), NULL);
|
||
|
|
||
|
if (idCmd >= IDM_QCM_MIN && idCmd < IDM_QCM_MAX)
|
||
|
{
|
||
|
idCmd -= IDM_QCM_MIN;
|
||
|
|
||
|
CMINVOKECOMMANDINFOEX ici = {
|
||
|
sizeof(ici), // cbSize
|
||
|
CMIC_MASK_ASYNCOK, // fMask
|
||
|
GetHWND(), // hwnd
|
||
|
(LPCSTR)IntToPtr(idCmd),// lpVerb
|
||
|
NULL, // lpParameters
|
||
|
NULL, // lpDirectory
|
||
|
SW_SHOWDEFAULT, // nShow
|
||
|
0, // dwHotKey
|
||
|
0, // hIcon
|
||
|
NULL, // lpTitle
|
||
|
(LPCWSTR)IntToPtr(idCmd),// lpVerbW
|
||
|
NULL, // lpParametersW
|
||
|
NULL, // lpDirectoryW
|
||
|
NULL, // lpTitleW
|
||
|
{ pt.x, pt.y }, // ptInvoke
|
||
|
};
|
||
|
|
||
|
#if 0 // REVIEW fabriced
|
||
|
if (_dwFlags & (HOSTF_CANDELETE | HOSTF_CANRENAME))
|
||
|
{
|
||
|
ContextMenu_GetCommandStringVerb(pcm, idCmd, szBuf, ARRAYSIZE(szBuf));
|
||
|
}
|
||
|
|
||
|
if ((_dwFlags & HOSTF_CANDELETE) &&
|
||
|
StrCmpI(szBuf, TEXT("delete")) == 0)
|
||
|
{
|
||
|
ContextMenuDeleteItem(pitem, pcm, &ici);
|
||
|
SMNMCOMMANDINVOKED ci;
|
||
|
ListView_GetItemRect(_hwndList, iItem, &ci.rcItem, LVIR_BOUNDS);
|
||
|
MapWindowRect(_hwndList, NULL, &ci.rcItem);
|
||
|
_SendNotify(_hwnd, SMN_COMMANDINVOKED, &ci);
|
||
|
}
|
||
|
else if ((_dwFlags & HOSTF_CANRENAME) &&
|
||
|
StrCmpI(szBuf, TEXT("rename")) == 0)
|
||
|
{
|
||
|
ListView_EditLabel(_hwndList, iItem);
|
||
|
}
|
||
|
else
|
||
|
#endif
|
||
|
{
|
||
|
pcm->InvokeCommand(reinterpret_cast<LPCMINVOKECOMMANDINFO>(&ici));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
DestroyMenu(hmenu);
|
||
|
|
||
|
GetHWNDHost()->WndProc(GetHWND(), PBM_SETMENUFORWARD, 0, (LPARAM)NULL);
|
||
|
|
||
|
ATOMICRELEASE(_pcm2Pop);
|
||
|
ATOMICRELEASE(_pcm3Pop);
|
||
|
}
|
||
|
pcm->Release();
|
||
|
}
|
||
|
|
||
|
psf->Release();
|
||
|
}
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
LRESULT PIDLButton::OnMenuMessage(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
||
|
{
|
||
|
LRESULT lres;
|
||
|
if (_pcm3Pop && SUCCEEDED(_pcm3Pop->HandleMenuMsg2(uMsg, wParam, lParam, &lres)))
|
||
|
{
|
||
|
return lres;
|
||
|
}
|
||
|
|
||
|
if (_pcm2Pop && SUCCEEDED(_pcm2Pop->HandleMenuMsg(uMsg, wParam, lParam)))
|
||
|
{
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////
|
||
|
// System events
|
||
|
|
||
|
void PIDLButton::OnPropertyChanged(PropertyInfo* ppi, int iIndex, Value* pvOld, Value* pvNew)
|
||
|
{
|
||
|
// Do default processing
|
||
|
Button::OnPropertyChanged(ppi, iIndex, pvOld, pvNew);
|
||
|
}
|
||
|
|
||
|
// Pointer is only guaranteed good for the lifetime of the call
|
||
|
void PIDLButton::OnEvent(Event *pEvent)
|
||
|
{
|
||
|
if ((pEvent->peTarget == this) && (pEvent->uidType == Button::Click))
|
||
|
{
|
||
|
// execute pidl
|
||
|
InvokePidl();
|
||
|
pEvent->fHandled = true;
|
||
|
return;
|
||
|
}
|
||
|
else if ((pEvent->peTarget == this) && (pEvent->uidType == Button::Context))
|
||
|
{
|
||
|
ButtonContextEvent *peButton = reinterpret_cast<ButtonContextEvent *>(pEvent);
|
||
|
OnContextMenu(&peButton->pt);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
|
||
|
Button::OnEvent(pEvent);
|
||
|
}
|
||
|
|
||
|
void PIDLButton::OnInput(InputEvent* pie)
|
||
|
{
|
||
|
// Fabrice, I will be putting right button click support for context menu here -- I'll have it for you Monday
|
||
|
Button::OnInput(pie);
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////
|
||
|
// Property definitions
|
||
|
|
||
|
/** Property template (replace !!!), also update private PropertyInfo* parray and class header (element.h)
|
||
|
// !!! property
|
||
|
static int vv!!![] = { DUIV_INT, -1 }; StaticValue(svDefault!!!, DUIV_INT, 0);
|
||
|
static PropertyInfo imp!!!Prop = { L"!!!", PF_Normal, 0, vv!!!, (Value*)&svDefault!!! };
|
||
|
PropertyInfo* Element::!!!Prop = &imp!!!Prop;
|
||
|
**/
|
||
|
|
||
|
////////////////////////////////////////////////////////
|
||
|
// ClassInfo (must appear after property definitions)
|
||
|
|
||
|
// Class properties
|
||
|
|
||
|
// Define class info with type and base type, set static class pointer
|
||
|
IClassInfo* PIDLButton::Class = NULL;
|
||
|
|
||
|
HRESULT PIDLButton::Register()
|
||
|
{
|
||
|
return ClassInfo<PIDLButton,Button>::Register(L"PIDLButton", NULL, 0);
|
||
|
}
|
||
|
|
||
|
|
||
|
} // namespace DirectUI
|
||
|
|
||
|
#endif // FEATURE_STARTPAGE
|