mirror of
https://github.com/reactos/reactos.git
synced 2024-12-29 02:25:17 +00:00
460a01b167
+ Add a missing break CORE-18646
1578 lines
46 KiB
C++
1578 lines
46 KiB
C++
/*
|
|
* ReactOS Explorer
|
|
*
|
|
* Copyright 2016 Sylvain Deverre <deverre dot sylv at gmail dot com>
|
|
* Copyright 2020 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
#include <commoncontrols.h>
|
|
#include <undocshell.h>
|
|
|
|
#if 1
|
|
#undef UNIMPLEMENTED
|
|
|
|
#define UNIMPLEMENTED DbgPrint("%s is UNIMPLEMENTED!\n", __FUNCTION__)
|
|
#endif
|
|
|
|
/*
|
|
* TODO:
|
|
* - Monitor correctly "external" shell interrupts (seems like we need to register/deregister them for each folder)
|
|
* - find and fix what cause explorer crashes sometimes (seems to be explorer that does more releases than addref)
|
|
* - TESTING
|
|
*/
|
|
|
|
typedef struct _PIDLDATA
|
|
{
|
|
BYTE type;
|
|
BYTE data[1];
|
|
} PIDLDATA, *LPPIDLDATA;
|
|
|
|
#define PT_GUID 0x1F
|
|
#define PT_SHELLEXT 0x2E
|
|
#define PT_YAGUID 0x70
|
|
|
|
static BOOL _ILIsSpecialFolder (LPCITEMIDLIST pidl)
|
|
{
|
|
LPPIDLDATA lpPData = (LPPIDLDATA)&pidl->mkid.abID;
|
|
|
|
return (pidl &&
|
|
((lpPData && (PT_GUID == lpPData->type || PT_SHELLEXT== lpPData->type ||
|
|
PT_YAGUID == lpPData->type)) || (pidl && pidl->mkid.cb == 0x00)));
|
|
}
|
|
|
|
HRESULT GetDisplayName(LPCITEMIDLIST pidlDirectory,TCHAR *szDisplayName,UINT cchMax,DWORD uFlags)
|
|
{
|
|
IShellFolder *pShellFolder = NULL;
|
|
LPCITEMIDLIST pidlRelative = NULL;
|
|
STRRET str;
|
|
HRESULT hr;
|
|
|
|
if (pidlDirectory == NULL || szDisplayName == NULL)
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
hr = SHBindToParent(pidlDirectory, IID_PPV_ARG(IShellFolder, &pShellFolder), &pidlRelative);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pShellFolder->GetDisplayNameOf(pidlRelative,uFlags,&str);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = StrRetToBuf(&str,pidlDirectory,szDisplayName,cchMax);
|
|
}
|
|
pShellFolder->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
/*
|
|
This is a Windows hack, because shell event messages in Windows gives an
|
|
ill-formed PIDL stripped from useful data that parses incorrectly with SHGetFileInfo.
|
|
So we need to re-enumerate subfolders until we find one with the same name.
|
|
*/
|
|
HRESULT _ReparsePIDL(LPITEMIDLIST buggyPidl, LPITEMIDLIST *cleanPidl)
|
|
{
|
|
HRESULT hr;
|
|
CComPtr<IShellFolder> folder;
|
|
CComPtr<IPersistFolder2> persist;
|
|
CComPtr<IEnumIDList> pEnumIDList;
|
|
LPITEMIDLIST childPidl;
|
|
LPITEMIDLIST correctChild;
|
|
LPITEMIDLIST correctParent;
|
|
ULONG fetched;
|
|
DWORD EnumFlags;
|
|
|
|
|
|
EnumFlags = SHCONTF_FOLDERS | SHCONTF_INCLUDEHIDDEN;
|
|
hr = SHBindToParent(buggyPidl, IID_PPV_ARG(IShellFolder, &folder), (LPCITEMIDLIST*)&childPidl);
|
|
*cleanPidl = NULL;
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Can't bind to parent folder\n");
|
|
return hr;
|
|
}
|
|
hr = folder->QueryInterface(IID_PPV_ARG(IPersistFolder2, &persist));
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("PIDL doesn't belong to the shell namespace, aborting\n");
|
|
return hr;
|
|
}
|
|
|
|
hr = persist->GetCurFolder(&correctParent);
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Unable to get current folder\n");
|
|
return hr;
|
|
}
|
|
|
|
hr = folder->EnumObjects(NULL,EnumFlags,&pEnumIDList);
|
|
// avoid broken IShellFolder implementations that return null pointer with success
|
|
if (!SUCCEEDED(hr) || !pEnumIDList)
|
|
{
|
|
ERR("Can't enum the folder !\n");
|
|
return hr;
|
|
}
|
|
|
|
while(SUCCEEDED(pEnumIDList->Next(1, &correctChild, &fetched)) && correctChild && fetched)
|
|
{
|
|
if (!folder->CompareIDs(0, childPidl, correctChild))
|
|
{
|
|
*cleanPidl = ILCombine(correctParent, correctChild);
|
|
ILFree(correctChild);
|
|
goto Cleanup;
|
|
}
|
|
ILFree(correctChild);
|
|
}
|
|
Cleanup:
|
|
ILFree(correctParent);
|
|
return hr;
|
|
}
|
|
|
|
CExplorerBand::CExplorerBand()
|
|
: m_pSite(NULL)
|
|
, m_fVisible(FALSE)
|
|
, m_bNavigating(FALSE)
|
|
, m_dwBandID(0)
|
|
, m_isEditing(FALSE)
|
|
, m_pidlCurrent(NULL)
|
|
{
|
|
}
|
|
|
|
CExplorerBand::~CExplorerBand()
|
|
{
|
|
if (m_pidlCurrent)
|
|
{
|
|
ILFree(m_pidlCurrent);
|
|
}
|
|
}
|
|
|
|
void CExplorerBand::InitializeExplorerBand()
|
|
{
|
|
// Init the treeview here
|
|
HRESULT hr;
|
|
LPITEMIDLIST pidl;
|
|
CComPtr<IWebBrowser2> browserService;
|
|
SHChangeNotifyEntry shcne;
|
|
|
|
hr = SHGetDesktopFolder(&m_pDesktop);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return;
|
|
|
|
hr = SHGetFolderLocation(m_hWnd, CSIDL_DESKTOP, NULL, 0, &pidl);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return;
|
|
|
|
IImageList * piml;
|
|
hr = SHGetImageList(SHIL_SMALL, IID_PPV_ARG(IImageList, &piml));
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return;
|
|
|
|
TreeView_SetImageList(m_hWnd, (HIMAGELIST)piml, TVSIL_NORMAL);
|
|
|
|
// Insert the root node
|
|
m_hRoot = InsertItem(0, m_pDesktop, pidl, pidl, FALSE);
|
|
if (!m_hRoot)
|
|
{
|
|
ERR("Failed to create root item\n");
|
|
return;
|
|
}
|
|
|
|
NodeInfo* pNodeInfo = GetNodeInfo(m_hRoot);
|
|
|
|
// Insert child nodes
|
|
InsertSubitems(m_hRoot, pNodeInfo);
|
|
TreeView_Expand(m_hWnd, m_hRoot, TVE_EXPAND);
|
|
|
|
// Navigate to current folder position
|
|
NavigateToCurrentFolder();
|
|
|
|
// Register shell notification
|
|
shcne.pidl = pidl;
|
|
shcne.fRecursive = TRUE;
|
|
m_shellRegID = SHChangeNotifyRegister(
|
|
m_hWnd,
|
|
SHCNRF_ShellLevel | SHCNRF_InterruptLevel | SHCNRF_RecursiveInterrupt,
|
|
SHCNE_DISKEVENTS | SHCNE_RENAMEFOLDER | SHCNE_RMDIR | SHCNE_MKDIR,
|
|
WM_USER_SHELLEVENT,
|
|
1,
|
|
&shcne);
|
|
if (!m_shellRegID)
|
|
{
|
|
ERR("Something went wrong, error %08x\n", GetLastError());
|
|
}
|
|
// Register browser connection endpoint
|
|
hr = IUnknown_QueryService(m_pSite, SID_SWebBrowserApp, IID_PPV_ARG(IWebBrowser2, &browserService));
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return;
|
|
|
|
hr = AtlAdvise(browserService, dynamic_cast<IDispatch*>(this), DIID_DWebBrowserEvents, &m_adviseCookie);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return;
|
|
|
|
ILFree(pidl);
|
|
}
|
|
|
|
void CExplorerBand::DestroyExplorerBand()
|
|
{
|
|
HRESULT hr;
|
|
CComPtr <IWebBrowser2> browserService;
|
|
|
|
TRACE("Cleaning up explorer band ...\n");
|
|
|
|
hr = IUnknown_QueryService(m_pSite, SID_SWebBrowserApp, IID_PPV_ARG(IWebBrowser2, &browserService));
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return;
|
|
|
|
hr = AtlUnadvise(browserService, DIID_DWebBrowserEvents, m_adviseCookie);
|
|
/* Remove all items of the treeview */
|
|
RevokeDragDrop(m_hWnd);
|
|
TreeView_DeleteAllItems(m_hWnd);
|
|
m_pDesktop = NULL;
|
|
m_hRoot = NULL;
|
|
TRACE("Cleanup done !\n");
|
|
}
|
|
|
|
CExplorerBand::NodeInfo* CExplorerBand::GetNodeInfo(HTREEITEM hItem)
|
|
{
|
|
TVITEM tvItem;
|
|
|
|
tvItem.mask = TVIF_PARAM;
|
|
tvItem.hItem = hItem;
|
|
|
|
if (!TreeView_GetItem(m_hWnd, &tvItem))
|
|
return 0;
|
|
|
|
return reinterpret_cast<NodeInfo*>(tvItem.lParam);
|
|
}
|
|
|
|
HRESULT CExplorerBand::ExecuteCommand(CComPtr<IContextMenu>& menu, UINT nCmd)
|
|
{
|
|
CComPtr<IOleWindow> pBrowserOleWnd;
|
|
CMINVOKECOMMANDINFO cmi;
|
|
HWND browserWnd;
|
|
HRESULT hr;
|
|
|
|
hr = IUnknown_QueryService(m_pSite, SID_SShellBrowser, IID_PPV_ARG(IOleWindow, &pBrowserOleWnd));
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return hr;
|
|
|
|
hr = pBrowserOleWnd->GetWindow(&browserWnd);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return hr;
|
|
|
|
ZeroMemory(&cmi, sizeof(cmi));
|
|
cmi.cbSize = sizeof(cmi);
|
|
cmi.lpVerb = MAKEINTRESOURCEA(nCmd);
|
|
cmi.hwnd = browserWnd;
|
|
if (GetKeyState(VK_SHIFT) & 0x8000)
|
|
cmi.fMask |= CMIC_MASK_SHIFT_DOWN;
|
|
if (GetKeyState(VK_CONTROL) & 0x8000)
|
|
cmi.fMask |= CMIC_MASK_CONTROL_DOWN;
|
|
|
|
return menu->InvokeCommand(&cmi);
|
|
}
|
|
|
|
HRESULT CExplorerBand::UpdateBrowser(LPITEMIDLIST pidlGoto)
|
|
{
|
|
CComPtr<IShellBrowser> pBrowserService;
|
|
HRESULT hr;
|
|
|
|
hr = IUnknown_QueryService(m_pSite, SID_STopLevelBrowser, IID_PPV_ARG(IShellBrowser, &pBrowserService));
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return hr;
|
|
|
|
hr = pBrowserService->BrowseObject(pidlGoto, SBSP_SAMEBROWSER | SBSP_ABSOLUTE);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return hr;
|
|
|
|
if (m_pidlCurrent)
|
|
{
|
|
ILFree(m_pidlCurrent);
|
|
m_pidlCurrent = ILClone(pidlGoto);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
// *** notifications handling ***
|
|
BOOL CExplorerBand::OnTreeItemExpanding(LPNMTREEVIEW pnmtv)
|
|
{
|
|
NodeInfo *pNodeInfo;
|
|
|
|
if (pnmtv->action == TVE_COLLAPSE) {
|
|
if (pnmtv->itemNew.hItem == m_hRoot)
|
|
{
|
|
// Prenvent root from collapsing
|
|
pnmtv->itemNew.mask |= TVIF_STATE;
|
|
pnmtv->itemNew.stateMask |= TVIS_EXPANDED;
|
|
pnmtv->itemNew.state &= ~TVIS_EXPANDED;
|
|
pnmtv->action = TVE_EXPAND;
|
|
return TRUE;
|
|
}
|
|
}
|
|
if (pnmtv->action == TVE_EXPAND) {
|
|
// Grab our directory PIDL
|
|
pNodeInfo = GetNodeInfo(pnmtv->itemNew.hItem);
|
|
// We have it, let's try
|
|
if (pNodeInfo && !pNodeInfo->expanded)
|
|
if (!InsertSubitems(pnmtv->itemNew.hItem, pNodeInfo)) {
|
|
// remove subitem "+" since we failed to add subitems
|
|
TV_ITEM tvItem;
|
|
|
|
tvItem.mask = TVIF_CHILDREN;
|
|
tvItem.hItem = pnmtv->itemNew.hItem;
|
|
tvItem.cChildren = 0;
|
|
|
|
TreeView_SetItem(m_hWnd, &tvItem);
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL CExplorerBand::OnTreeItemDeleted(LPNMTREEVIEW pnmtv)
|
|
{
|
|
/* Destroy memory associated to our node */
|
|
NodeInfo* ptr = GetNodeInfo(pnmtv->itemNew.hItem);
|
|
if (ptr)
|
|
{
|
|
ILFree(ptr->relativePidl);
|
|
ILFree(ptr->absolutePidl);
|
|
delete ptr;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
void CExplorerBand::OnSelectionChanged(LPNMTREEVIEW pnmtv)
|
|
{
|
|
NodeInfo* pNodeInfo = GetNodeInfo(pnmtv->itemNew.hItem);
|
|
|
|
/* Prevents navigation if selection is initiated inside the band */
|
|
if (m_bNavigating)
|
|
return;
|
|
|
|
UpdateBrowser(pNodeInfo->absolutePidl);
|
|
|
|
SetFocus();
|
|
// Expand the node
|
|
//TreeView_Expand(m_hWnd, pnmtv->itemNew.hItem, TVE_EXPAND);
|
|
}
|
|
|
|
void CExplorerBand::OnTreeItemDragging(LPNMTREEVIEW pnmtv, BOOL isRightClick)
|
|
{
|
|
CComPtr<IShellFolder> pSrcFolder;
|
|
CComPtr<IDataObject> pObj;
|
|
LPCITEMIDLIST pLast;
|
|
HRESULT hr;
|
|
DWORD dwEffect;
|
|
DWORD dwEffect2;
|
|
|
|
dwEffect = DROPEFFECT_COPY | DROPEFFECT_MOVE;
|
|
if (!pnmtv->itemNew.lParam)
|
|
return;
|
|
NodeInfo* pNodeInfo = GetNodeInfo(pnmtv->itemNew.hItem);
|
|
hr = SHBindToParent(pNodeInfo->absolutePidl, IID_PPV_ARG(IShellFolder, &pSrcFolder), &pLast);
|
|
if (!SUCCEEDED(hr))
|
|
return;
|
|
hr = pSrcFolder->GetUIObjectOf(m_hWnd, 1, &pLast, IID_IDataObject, 0, reinterpret_cast<void**>(&pObj));
|
|
if (!SUCCEEDED(hr))
|
|
return;
|
|
DoDragDrop(pObj, this, dwEffect, &dwEffect2);
|
|
return;
|
|
}
|
|
|
|
|
|
// *** ATL event handlers ***
|
|
LRESULT CExplorerBand::OnContextMenu(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled)
|
|
{
|
|
HTREEITEM item;
|
|
NodeInfo *info;
|
|
HMENU treeMenu;
|
|
WORD x;
|
|
WORD y;
|
|
CComPtr<IShellFolder> pFolder;
|
|
CComPtr<IContextMenu> contextMenu;
|
|
HRESULT hr;
|
|
UINT uCommand;
|
|
LPITEMIDLIST pidlChild;
|
|
|
|
treeMenu = NULL;
|
|
item = TreeView_GetSelection(m_hWnd);
|
|
bHandled = TRUE;
|
|
if (!item)
|
|
{
|
|
goto Cleanup;
|
|
}
|
|
|
|
x = LOWORD(lParam);
|
|
y = HIWORD(lParam);
|
|
if (x == -1 && y == -1)
|
|
{
|
|
// TODO: grab position of tree item and position it correctly
|
|
}
|
|
|
|
info = GetNodeInfo(item);
|
|
if (!info)
|
|
{
|
|
ERR("No node data, something has gone wrong !\n");
|
|
goto Cleanup;
|
|
}
|
|
hr = SHBindToParent(info->absolutePidl, IID_PPV_ARG(IShellFolder, &pFolder),
|
|
(LPCITEMIDLIST*)&pidlChild);
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Can't bind to folder!\n");
|
|
goto Cleanup;
|
|
}
|
|
hr = pFolder->GetUIObjectOf(m_hWnd, 1, (LPCITEMIDLIST*)&pidlChild, IID_IContextMenu,
|
|
NULL, reinterpret_cast<void**>(&contextMenu));
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Can't get IContextMenu interface\n");
|
|
goto Cleanup;
|
|
}
|
|
|
|
IUnknown_SetSite(contextMenu, (IDeskBand *)this);
|
|
|
|
treeMenu = CreatePopupMenu();
|
|
hr = contextMenu->QueryContextMenu(treeMenu, 0, FCIDM_SHVIEWFIRST, FCIDM_SHVIEWLAST,
|
|
CMF_EXPLORE);
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
WARN("Can't get context menu for item\n");
|
|
DestroyMenu(treeMenu);
|
|
goto Cleanup;
|
|
}
|
|
uCommand = TrackPopupMenu(treeMenu, TPM_LEFTALIGN | TPM_RETURNCMD | TPM_LEFTBUTTON | TPM_RIGHTBUTTON,
|
|
x, y, 0, m_hWnd, NULL);
|
|
|
|
ExecuteCommand(contextMenu, uCommand);
|
|
|
|
Cleanup:
|
|
if (contextMenu)
|
|
IUnknown_SetSite(contextMenu, NULL);
|
|
if (treeMenu)
|
|
DestroyMenu(treeMenu);
|
|
m_bNavigating = TRUE;
|
|
TreeView_SelectItem(m_hWnd, m_oldSelected);
|
|
m_bNavigating = FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
LRESULT CExplorerBand::ContextMenuHack(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled)
|
|
{
|
|
bHandled = FALSE;
|
|
if (uMsg == WM_RBUTTONDOWN)
|
|
{
|
|
TVHITTESTINFO info;
|
|
info.pt.x = LOWORD(lParam);
|
|
info.pt.y = HIWORD(lParam);
|
|
info.flags = TVHT_ONITEM;
|
|
info.hItem = NULL;
|
|
|
|
// Save the current location
|
|
m_oldSelected = TreeView_GetSelection(m_hWnd);
|
|
|
|
// Move to the item selected by the treeview (don't change right pane)
|
|
TreeView_HitTest(m_hWnd, &info);
|
|
m_bNavigating = TRUE;
|
|
TreeView_SelectItem(m_hWnd, info.hItem);
|
|
m_bNavigating = FALSE;
|
|
}
|
|
return FALSE; /* let the wndproc process the message */
|
|
}
|
|
|
|
LRESULT CExplorerBand::OnShellEvent(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled)
|
|
{
|
|
LPITEMIDLIST *dest;
|
|
LPITEMIDLIST clean;
|
|
HTREEITEM pItem;
|
|
|
|
dest = (LPITEMIDLIST*)wParam;
|
|
/* TODO: handle shell notifications */
|
|
switch(lParam & ~SHCNE_INTERRUPT)
|
|
{
|
|
case SHCNE_MKDIR:
|
|
if (!SUCCEEDED(_ReparsePIDL(dest[0], &clean)))
|
|
{
|
|
ERR("Can't reparse PIDL to a valid one\n");
|
|
return FALSE;
|
|
}
|
|
NavigateToPIDL(clean, &pItem, FALSE, TRUE, FALSE);
|
|
ILFree(clean);
|
|
break;
|
|
case SHCNE_RMDIR:
|
|
DeleteItem(dest[0]);
|
|
break;
|
|
case SHCNE_RENAMEFOLDER:
|
|
if (!SUCCEEDED(_ReparsePIDL(dest[1], &clean)))
|
|
{
|
|
ERR("Can't reparse PIDL to a valid one\n");
|
|
return FALSE;
|
|
}
|
|
if (NavigateToPIDL(dest[0], &pItem, FALSE, FALSE, FALSE))
|
|
RenameItem(pItem, clean);
|
|
ILFree(clean);
|
|
break;
|
|
case SHCNE_UPDATEDIR:
|
|
// We don't take care of this message
|
|
TRACE("Directory updated\n");
|
|
break;
|
|
default:
|
|
TRACE("Unhandled message\n");
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
LRESULT CExplorerBand::OnSetFocus(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled)
|
|
{
|
|
m_bFocused = TRUE;
|
|
IUnknown_OnFocusChangeIS(m_pSite, reinterpret_cast<IUnknown*>(this), TRUE);
|
|
bHandled = FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
LRESULT CExplorerBand::OnKillFocus(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled)
|
|
{
|
|
IUnknown_OnFocusChangeIS(m_pSite, reinterpret_cast<IUnknown*>(this), FALSE);
|
|
bHandled = FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
// *** Helper functions ***
|
|
HTREEITEM CExplorerBand::InsertItem(HTREEITEM hParent, IShellFolder *psfParent, LPITEMIDLIST pElt, LPITEMIDLIST pEltRelative, BOOL bSort)
|
|
{
|
|
TV_INSERTSTRUCT tvInsert;
|
|
HTREEITEM htiCreated;
|
|
|
|
/* Get the attributes of the node */
|
|
SFGAOF attrs = SFGAO_STREAM | SFGAO_HASSUBFOLDER;
|
|
HRESULT hr = psfParent->GetAttributesOf(1, &pEltRelative, &attrs);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return NULL;
|
|
|
|
/* Ignore streams */
|
|
if ((attrs & SFGAO_STREAM))
|
|
{
|
|
TRACE("Ignoring stream\n");
|
|
return NULL;
|
|
}
|
|
|
|
/* Get the name of the node */
|
|
WCHAR wszDisplayName[MAX_PATH];
|
|
STRRET strret;
|
|
hr = psfParent->GetDisplayNameOf(pEltRelative, SHGDN_INFOLDER, &strret);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return NULL;
|
|
|
|
hr = StrRetToBufW(&strret, pEltRelative, wszDisplayName, MAX_PATH);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return NULL;
|
|
|
|
/* Get the icon of the node */
|
|
INT iIcon = SHMapPIDLToSystemImageListIndex(psfParent, pEltRelative, NULL);
|
|
|
|
NodeInfo* pChildInfo = new NodeInfo;
|
|
if (!pChildInfo)
|
|
{
|
|
ERR("Failed to allocate NodeInfo\n");
|
|
return FALSE;
|
|
}
|
|
|
|
// Store our node info
|
|
pChildInfo->absolutePidl = ILClone(pElt);
|
|
pChildInfo->relativePidl = ILClone(pEltRelative);
|
|
pChildInfo->expanded = FALSE;
|
|
|
|
// Set up our treeview template
|
|
tvInsert.hParent = hParent;
|
|
tvInsert.hInsertAfter = TVI_LAST;
|
|
tvInsert.item.mask = TVIF_PARAM | TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_CHILDREN;
|
|
tvInsert.item.cchTextMax = MAX_PATH;
|
|
tvInsert.item.pszText = wszDisplayName;
|
|
tvInsert.item.iImage = tvInsert.item.iSelectedImage = iIcon;
|
|
tvInsert.item.cChildren = (attrs & SFGAO_HASSUBFOLDER) ? 1 : 0;
|
|
tvInsert.item.lParam = (LPARAM)pChildInfo;
|
|
|
|
htiCreated = TreeView_InsertItem(m_hWnd, &tvInsert);
|
|
|
|
if (bSort)
|
|
{
|
|
TVSORTCB sortCallback;
|
|
sortCallback.hParent = hParent;
|
|
sortCallback.lpfnCompare = CompareTreeItems;
|
|
sortCallback.lParam = (LPARAM)this;
|
|
SendMessage(TVM_SORTCHILDRENCB, 0, (LPARAM)&sortCallback);
|
|
}
|
|
|
|
return htiCreated;
|
|
}
|
|
|
|
/* This is the slow version of the above method */
|
|
HTREEITEM CExplorerBand::InsertItem(HTREEITEM hParent, LPITEMIDLIST pElt, LPITEMIDLIST pEltRelative, BOOL bSort)
|
|
{
|
|
CComPtr<IShellFolder> psfFolder;
|
|
HRESULT hr = SHBindToParent(pElt, IID_PPV_ARG(IShellFolder, &psfFolder), NULL);
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
return NULL;
|
|
|
|
return InsertItem(hParent, psfFolder, pElt, pEltRelative, bSort);
|
|
}
|
|
|
|
BOOL CExplorerBand::InsertSubitems(HTREEITEM hItem, NodeInfo *pNodeInfo)
|
|
{
|
|
CComPtr<IEnumIDList> pEnumIDList;
|
|
LPITEMIDLIST pidlSub;
|
|
LPITEMIDLIST entry;
|
|
SHCONTF EnumFlags;
|
|
HRESULT hr;
|
|
ULONG fetched;
|
|
ULONG uItemCount;
|
|
CComPtr<IShellFolder> pFolder;
|
|
TVSORTCB sortCallback;
|
|
|
|
entry = pNodeInfo->absolutePidl;
|
|
fetched = 1;
|
|
uItemCount = 0;
|
|
EnumFlags = SHCONTF_FOLDERS;
|
|
|
|
hr = SHGetFolderLocation(m_hWnd, CSIDL_DESKTOP, NULL, 0, &pidlSub);
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Can't get desktop PIDL !\n");
|
|
return FALSE;
|
|
}
|
|
|
|
if (!m_pDesktop->CompareIDs(NULL, pidlSub, entry))
|
|
{
|
|
// We are the desktop, so use pDesktop as pFolder
|
|
pFolder = m_pDesktop;
|
|
}
|
|
else
|
|
{
|
|
// Get an IShellFolder of our pidl
|
|
hr = m_pDesktop->BindToObject(entry, NULL, IID_PPV_ARG(IShellFolder, &pFolder));
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ILFree(pidlSub);
|
|
ERR("Can't bind folder to desktop !\n");
|
|
return FALSE;
|
|
}
|
|
}
|
|
ILFree(pidlSub);
|
|
|
|
// TODO: handle hidden folders according to settings !
|
|
EnumFlags |= SHCONTF_INCLUDEHIDDEN;
|
|
|
|
// Enum through objects
|
|
hr = pFolder->EnumObjects(NULL,EnumFlags,&pEnumIDList);
|
|
|
|
// avoid broken IShellFolder implementations that return null pointer with success
|
|
if (!SUCCEEDED(hr) || !pEnumIDList)
|
|
{
|
|
ERR("Can't enum the folder !\n");
|
|
return FALSE;
|
|
}
|
|
|
|
/* Don't redraw while we add stuff into the tree */
|
|
SendMessage(WM_SETREDRAW, FALSE, 0);
|
|
while(SUCCEEDED(pEnumIDList->Next(1, &pidlSub, &fetched)) && pidlSub && fetched)
|
|
{
|
|
LPITEMIDLIST pidlSubComplete;
|
|
pidlSubComplete = ILCombine(entry, pidlSub);
|
|
|
|
if (InsertItem(hItem, pFolder, pidlSubComplete, pidlSub, FALSE))
|
|
uItemCount++;
|
|
ILFree(pidlSubComplete);
|
|
ILFree(pidlSub);
|
|
}
|
|
pNodeInfo->expanded = TRUE;
|
|
/* Let's do sorting */
|
|
sortCallback.hParent = hItem;
|
|
sortCallback.lpfnCompare = CompareTreeItems;
|
|
sortCallback.lParam = (LPARAM)this;
|
|
SendMessage(TVM_SORTCHILDRENCB, 0, (LPARAM)&sortCallback);
|
|
|
|
/* Now we can redraw */
|
|
SendMessage(WM_SETREDRAW, TRUE, 0);
|
|
|
|
return (uItemCount > 0) ? TRUE : FALSE;
|
|
}
|
|
|
|
/**
|
|
* Navigate to a given PIDL in the treeview, and return matching tree item handle
|
|
* - dest: The absolute PIDL we should navigate in the treeview
|
|
* - item: Handle of the tree item matching the PIDL
|
|
* - bExpand: expand collapsed nodes in order to find the right element
|
|
* - bInsert: insert the element at the right place if we don't find it
|
|
* - bSelect: select the item after we found it
|
|
*/
|
|
BOOL CExplorerBand::NavigateToPIDL(LPITEMIDLIST dest, HTREEITEM *item, BOOL bExpand, BOOL bInsert,
|
|
BOOL bSelect)
|
|
{
|
|
HTREEITEM current;
|
|
HTREEITEM tmp;
|
|
HTREEITEM parent;
|
|
NodeInfo *nodeData;
|
|
LPITEMIDLIST relativeChild;
|
|
TVITEM tvItem;
|
|
|
|
if (!item)
|
|
return FALSE;
|
|
|
|
current = m_hRoot;
|
|
parent = NULL;
|
|
while (TRUE)
|
|
{
|
|
nodeData = GetNodeInfo(current);
|
|
if (!nodeData)
|
|
{
|
|
ERR("Something has gone wrong, no data associated to node !\n");
|
|
*item = NULL;
|
|
return FALSE;
|
|
}
|
|
// If we found our node, give it back
|
|
if (!m_pDesktop->CompareIDs(0, nodeData->absolutePidl, dest))
|
|
{
|
|
if (bSelect)
|
|
TreeView_SelectItem(m_hWnd, current);
|
|
*item = current;
|
|
return TRUE;
|
|
}
|
|
|
|
// Check if we are a parent of the requested item
|
|
relativeChild = ILFindChild(nodeData->absolutePidl, dest);
|
|
if (relativeChild != 0)
|
|
{
|
|
// Notify treeview we have children
|
|
tvItem.mask = TVIF_CHILDREN;
|
|
tvItem.hItem = current;
|
|
tvItem.cChildren = 1;
|
|
TreeView_SetItem(m_hWnd, &tvItem);
|
|
|
|
// If we can expand and the node isn't expanded yet, do it
|
|
if (bExpand)
|
|
{
|
|
if (!nodeData->expanded)
|
|
InsertSubitems(current, nodeData);
|
|
TreeView_Expand(m_hWnd, current, TVE_EXPAND);
|
|
}
|
|
|
|
// Try to get a child
|
|
tmp = TreeView_GetChild(m_hWnd, current);
|
|
if (tmp)
|
|
{
|
|
// We have a child, let's continue with it
|
|
parent = current;
|
|
current = tmp;
|
|
continue;
|
|
}
|
|
|
|
if (bInsert && nodeData->expanded)
|
|
{
|
|
// Happens when we have to create a subchild inside a child
|
|
current = InsertItem(current, dest, relativeChild, TRUE);
|
|
}
|
|
// We end up here, without any children, so we found nothing
|
|
// Tell the parent node it has children
|
|
ZeroMemory(&tvItem, sizeof(tvItem));
|
|
*item = NULL;
|
|
return FALSE;
|
|
}
|
|
|
|
// Find sibling
|
|
tmp = TreeView_GetNextSibling(m_hWnd, current);
|
|
if (tmp)
|
|
{
|
|
current = tmp;
|
|
continue;
|
|
}
|
|
if (bInsert)
|
|
{
|
|
current = InsertItem(parent, dest, ILFindLastID(dest), TRUE);
|
|
*item = current;
|
|
return TRUE;
|
|
}
|
|
*item = NULL;
|
|
return FALSE;
|
|
}
|
|
UNREACHABLE;
|
|
}
|
|
|
|
BOOL CExplorerBand::NavigateToCurrentFolder()
|
|
{
|
|
LPITEMIDLIST explorerPidl;
|
|
CComPtr<IBrowserService> pBrowserService;
|
|
HRESULT hr;
|
|
HTREEITEM dummy;
|
|
BOOL result;
|
|
explorerPidl = NULL;
|
|
|
|
hr = IUnknown_QueryService(m_pSite, SID_STopLevelBrowser, IID_PPV_ARG(IBrowserService, &pBrowserService));
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Can't get IBrowserService !\n");
|
|
return FALSE;
|
|
}
|
|
|
|
hr = pBrowserService->GetPidl(&explorerPidl);
|
|
if (!SUCCEEDED(hr) || !explorerPidl)
|
|
{
|
|
ERR("Unable to get browser PIDL !\n");
|
|
return FALSE;
|
|
}
|
|
m_bNavigating = TRUE;
|
|
/* find PIDL into our explorer */
|
|
result = NavigateToPIDL(explorerPidl, &dummy, TRUE, FALSE, TRUE);
|
|
m_bNavigating = FALSE;
|
|
return result;
|
|
}
|
|
|
|
BOOL CExplorerBand::DeleteItem(LPITEMIDLIST idl)
|
|
{
|
|
HTREEITEM toDelete;
|
|
TVITEM tvItem;
|
|
HTREEITEM parentNode;
|
|
|
|
if (!NavigateToPIDL(idl, &toDelete, FALSE, FALSE, FALSE))
|
|
return FALSE;
|
|
|
|
// TODO: check that the treeview item is really deleted
|
|
|
|
parentNode = TreeView_GetParent(m_hWnd, toDelete);
|
|
// Navigate to parent when deleting child item
|
|
if (!m_pDesktop->CompareIDs(0, idl, m_pidlCurrent))
|
|
{
|
|
TreeView_SelectItem(m_hWnd, parentNode);
|
|
}
|
|
|
|
// Remove the child item
|
|
TreeView_DeleteItem(m_hWnd, toDelete);
|
|
// Probe parent to see if it has children
|
|
if (!TreeView_GetChild(m_hWnd, parentNode))
|
|
{
|
|
// Decrement parent's child count
|
|
ZeroMemory(&tvItem, sizeof(tvItem));
|
|
tvItem.mask = TVIF_CHILDREN;
|
|
tvItem.hItem = parentNode;
|
|
tvItem.cChildren = 0;
|
|
TreeView_SetItem(m_hWnd, &tvItem);
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CExplorerBand::RenameItem(HTREEITEM toRename, LPITEMIDLIST newPidl)
|
|
{
|
|
WCHAR wszDisplayName[MAX_PATH];
|
|
TVITEM itemInfo;
|
|
LPCITEMIDLIST relPidl;
|
|
NodeInfo *treeInfo;
|
|
TVSORTCB sortCallback;
|
|
HTREEITEM child;
|
|
|
|
ZeroMemory(&itemInfo, sizeof(itemInfo));
|
|
itemInfo.mask = TVIF_PARAM;
|
|
itemInfo.hItem = toRename;
|
|
|
|
// Change PIDL associated to the item
|
|
relPidl = ILFindLastID(newPidl);
|
|
TreeView_GetItem(m_hWnd, &itemInfo);
|
|
if (!itemInfo.lParam)
|
|
{
|
|
ERR("Unable to fetch lParam\n");
|
|
return FALSE;
|
|
}
|
|
SendMessage(WM_SETREDRAW, FALSE, 0);
|
|
treeInfo = (NodeInfo*)itemInfo.lParam;
|
|
ILFree(treeInfo->absolutePidl);
|
|
ILFree(treeInfo->relativePidl);
|
|
treeInfo->absolutePidl = ILClone(newPidl);
|
|
treeInfo->relativePidl = ILClone(relPidl);
|
|
|
|
// Change the display name
|
|
GetDisplayName(newPidl, wszDisplayName, MAX_PATH, SHGDN_INFOLDER);
|
|
ZeroMemory(&itemInfo, sizeof(itemInfo));
|
|
itemInfo.hItem = toRename;
|
|
itemInfo.mask = TVIF_TEXT;
|
|
itemInfo.pszText = wszDisplayName;
|
|
TreeView_SetItem(m_hWnd, &itemInfo);
|
|
|
|
if((child = TreeView_GetChild(m_hWnd, toRename)) != NULL)
|
|
{
|
|
RefreshTreePidl(child, newPidl);
|
|
}
|
|
|
|
// Sorting
|
|
sortCallback.hParent = TreeView_GetParent(m_hWnd, toRename);
|
|
sortCallback.lpfnCompare = CompareTreeItems;
|
|
sortCallback.lParam = (LPARAM)this;
|
|
SendMessage(TVM_SORTCHILDRENCB, 0, (LPARAM)&sortCallback);
|
|
SendMessage(WM_SETREDRAW, TRUE, 0);
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL CExplorerBand::RefreshTreePidl(HTREEITEM tree, LPITEMIDLIST pidlParent)
|
|
{
|
|
HTREEITEM tmp;
|
|
NodeInfo *pInfo;
|
|
|
|
// Update our node data
|
|
pInfo = GetNodeInfo(tree);
|
|
if (!pInfo)
|
|
{
|
|
WARN("No tree info !\n");
|
|
return FALSE;
|
|
}
|
|
ILFree(pInfo->absolutePidl);
|
|
pInfo->absolutePidl = ILCombine(pidlParent, pInfo->relativePidl);
|
|
if (!pInfo->absolutePidl)
|
|
{
|
|
WARN("PIDL allocation failed\n");
|
|
return FALSE;
|
|
}
|
|
// Recursively update children
|
|
if ((tmp = TreeView_GetChild(m_hWnd, tree)) != NULL)
|
|
{
|
|
RefreshTreePidl(tmp, pInfo->absolutePidl);
|
|
}
|
|
|
|
tmp = TreeView_GetNextSibling(m_hWnd, tree);
|
|
while(tmp != NULL)
|
|
{
|
|
pInfo = GetNodeInfo(tmp);
|
|
if(!pInfo)
|
|
{
|
|
WARN("No tree info !\n");
|
|
continue;
|
|
}
|
|
ILFree(pInfo->absolutePidl);
|
|
pInfo->absolutePidl = ILCombine(pidlParent, pInfo->relativePidl);
|
|
tmp = TreeView_GetNextSibling(m_hWnd, tmp);
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
// *** Tree item sorting callback ***
|
|
int CALLBACK CExplorerBand::CompareTreeItems(LPARAM p1, LPARAM p2, LPARAM p3)
|
|
{
|
|
/*
|
|
* We first sort drive letters (Path root), then PIDLs and then regular folder
|
|
* display name.
|
|
* This is not how Windows sorts item, but it gives decent results.
|
|
*/
|
|
NodeInfo *info1;
|
|
NodeInfo *info2;
|
|
CExplorerBand *pThis;
|
|
WCHAR wszFolder1[MAX_PATH];
|
|
WCHAR wszFolder2[MAX_PATH];
|
|
|
|
info1 = (NodeInfo*)p1;
|
|
info2 = (NodeInfo*)p2;
|
|
pThis = (CExplorerBand*)p3;
|
|
|
|
GetDisplayName(info1->absolutePidl, wszFolder1, MAX_PATH, SHGDN_FORPARSING);
|
|
GetDisplayName(info2->absolutePidl, wszFolder2, MAX_PATH, SHGDN_FORPARSING);
|
|
if (PathIsRoot(wszFolder1) && PathIsRoot(wszFolder2))
|
|
{
|
|
return lstrcmpiW(wszFolder1,wszFolder2);
|
|
}
|
|
if (PathIsRoot(wszFolder1) && !PathIsRoot(wszFolder2))
|
|
{
|
|
return -1;
|
|
}
|
|
if (!PathIsRoot(wszFolder1) && PathIsRoot(wszFolder2))
|
|
{
|
|
return 1;
|
|
}
|
|
// Now, we compare non-root folders, grab display name
|
|
GetDisplayName(info1->absolutePidl, wszFolder1, MAX_PATH, SHGDN_INFOLDER);
|
|
GetDisplayName(info2->absolutePidl, wszFolder2, MAX_PATH, SHGDN_INFOLDER);
|
|
|
|
if (_ILIsSpecialFolder(info1->relativePidl) && !_ILIsSpecialFolder(info2->relativePidl))
|
|
{
|
|
return -1;
|
|
}
|
|
if (!_ILIsSpecialFolder(info1->relativePidl) && _ILIsSpecialFolder(info2->relativePidl))
|
|
{
|
|
return 1;
|
|
}
|
|
if (_ILIsSpecialFolder(info1->relativePidl) && !_ILIsSpecialFolder(info2->relativePidl))
|
|
{
|
|
HRESULT hr;
|
|
hr = pThis->m_pDesktop->CompareIDs(0, info1->absolutePidl, info2->absolutePidl);
|
|
if (!hr) return 0;
|
|
return (hr > 0) ? -1 : 1;
|
|
}
|
|
return StrCmpLogicalW(wszFolder1, wszFolder2);
|
|
}
|
|
|
|
// *** IOleWindow methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetWindow(HWND *lphwnd)
|
|
{
|
|
if (!lphwnd)
|
|
return E_INVALIDARG;
|
|
*lphwnd = m_hWnd;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::ContextSensitiveHelp(BOOL fEnterMode)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
|
|
// *** IDockingWindow methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::CloseDW(DWORD dwReserved)
|
|
{
|
|
// We do nothing, we don't have anything to save yet
|
|
TRACE("CloseDW called\n");
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::ResizeBorderDW(const RECT *prcBorder, IUnknown *punkToolbarSite, BOOL fReserved)
|
|
{
|
|
/* Must return E_NOTIMPL according to MSDN */
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::ShowDW(BOOL fShow)
|
|
{
|
|
m_fVisible = fShow;
|
|
ShowWindow(fShow);
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
// *** IDeskBand methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetBandInfo(DWORD dwBandID, DWORD dwViewMode, DESKBANDINFO *pdbi)
|
|
{
|
|
if (!pdbi)
|
|
{
|
|
return E_INVALIDARG;
|
|
}
|
|
this->m_dwBandID = dwBandID;
|
|
|
|
if (pdbi->dwMask & DBIM_MINSIZE)
|
|
{
|
|
pdbi->ptMinSize.x = 200;
|
|
pdbi->ptMinSize.y = 30;
|
|
}
|
|
|
|
if (pdbi->dwMask & DBIM_MAXSIZE)
|
|
{
|
|
pdbi->ptMaxSize.y = -1;
|
|
}
|
|
|
|
if (pdbi->dwMask & DBIM_INTEGRAL)
|
|
{
|
|
pdbi->ptIntegral.y = 1;
|
|
}
|
|
|
|
if (pdbi->dwMask & DBIM_ACTUAL)
|
|
{
|
|
pdbi->ptActual.x = 200;
|
|
pdbi->ptActual.y = 30;
|
|
}
|
|
|
|
if (pdbi->dwMask & DBIM_TITLE)
|
|
{
|
|
if (!LoadStringW(_AtlBaseModule.GetResourceInstance(), IDS_FOLDERSLABEL, pdbi->wszTitle, _countof(pdbi->wszTitle)))
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
|
|
if (pdbi->dwMask & DBIM_MODEFLAGS)
|
|
{
|
|
pdbi->dwModeFlags = DBIMF_NORMAL | DBIMF_VARIABLEHEIGHT;
|
|
}
|
|
|
|
if (pdbi->dwMask & DBIM_BKCOLOR)
|
|
{
|
|
pdbi->dwMask &= ~DBIM_BKCOLOR;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
// *** IObjectWithSite methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::SetSite(IUnknown *pUnkSite)
|
|
{
|
|
HRESULT hr;
|
|
HWND parentWnd;
|
|
|
|
if (pUnkSite == m_pSite)
|
|
return S_OK;
|
|
|
|
TRACE("SetSite called \n");
|
|
if (!pUnkSite)
|
|
{
|
|
DestroyExplorerBand();
|
|
DestroyWindow();
|
|
m_hWnd = NULL;
|
|
}
|
|
|
|
if (pUnkSite != m_pSite)
|
|
{
|
|
m_pSite = NULL;
|
|
}
|
|
|
|
if(!pUnkSite)
|
|
return S_OK;
|
|
|
|
hr = IUnknown_GetWindow(pUnkSite, &parentWnd);
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
ERR("Could not get parent's window ! Status: %08lx\n", hr);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
m_pSite = pUnkSite;
|
|
|
|
if (m_hWnd)
|
|
{
|
|
// Change its parent
|
|
SetParent(parentWnd);
|
|
}
|
|
else
|
|
{
|
|
HWND wnd = CreateWindow(WC_TREEVIEW, NULL,
|
|
WS_CHILD | WS_CLIPCHILDREN | WS_CLIPSIBLINGS | TVS_HASLINES | TVS_HASBUTTONS | TVS_SHOWSELALWAYS | TVS_EDITLABELS /* | TVS_SINGLEEXPAND*/ , // remove TVS_SINGLEEXPAND for now since it has strange behaviour
|
|
0, 0, 0, 0, parentWnd, NULL, _AtlBaseModule.GetModuleInstance(), NULL);
|
|
|
|
// Subclass the window
|
|
SubclassWindow(wnd);
|
|
|
|
// Initialize our treeview now
|
|
InitializeExplorerBand();
|
|
RegisterDragDrop(m_hWnd, dynamic_cast<IDropTarget*>(this));
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetSite(REFIID riid, void **ppvSite)
|
|
{
|
|
if (!ppvSite)
|
|
return E_POINTER;
|
|
*ppvSite = m_pSite;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
// *** IOleCommandTarget methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::QueryStatus(const GUID *pguidCmdGroup, ULONG cCmds, OLECMD prgCmds [], OLECMDTEXT *pCmdText)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Exec(const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANT *pvaIn, VARIANT *pvaOut)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
|
|
// *** IServiceProvider methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::QueryService(REFGUID guidService, REFIID riid, void **ppvObject)
|
|
{
|
|
/* FIXME: we probably want to handle more services here */
|
|
return IUnknown_QueryService(m_pSite, SID_SShellBrowser, riid, ppvObject);
|
|
}
|
|
|
|
|
|
// *** IInputObject methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::UIActivateIO(BOOL fActivate, LPMSG lpMsg)
|
|
{
|
|
if (fActivate)
|
|
{
|
|
//SetFocus();
|
|
SetActiveWindow();
|
|
}
|
|
// TODO: handle message
|
|
if(lpMsg)
|
|
{
|
|
TranslateMessage(lpMsg);
|
|
DispatchMessage(lpMsg);
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::HasFocusIO()
|
|
{
|
|
return m_bFocused ? S_OK : S_FALSE;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::TranslateAcceleratorIO(LPMSG lpMsg)
|
|
{
|
|
if (lpMsg->hwnd == m_hWnd ||
|
|
(m_isEditing && IsChild(lpMsg->hwnd)))
|
|
{
|
|
TranslateMessage(lpMsg);
|
|
DispatchMessage(lpMsg);
|
|
return S_OK;
|
|
}
|
|
|
|
return S_FALSE;
|
|
}
|
|
|
|
// *** IPersist methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetClassID(CLSID *pClassID)
|
|
{
|
|
if (!pClassID)
|
|
return E_POINTER;
|
|
memcpy(pClassID, &CLSID_ExplorerBand, sizeof(CLSID));
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
// *** IPersistStream methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::IsDirty()
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Load(IStream *pStm)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Save(IStream *pStm, BOOL fClearDirty)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetSizeMax(ULARGE_INTEGER *pcbSize)
|
|
{
|
|
// TODO: calculate max size
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
|
|
// *** IWinEventHandler methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::OnWinEvent(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT *theResult)
|
|
{
|
|
BOOL bHandled;
|
|
LRESULT result;
|
|
|
|
if (uMsg == WM_NOTIFY)
|
|
{
|
|
NMHDR *pNotifyHeader = (NMHDR*)lParam;
|
|
switch (pNotifyHeader->code)
|
|
{
|
|
case TVN_ITEMEXPANDING:
|
|
result = OnTreeItemExpanding((LPNMTREEVIEW)lParam);
|
|
if (theResult)
|
|
*theResult = result;
|
|
break;
|
|
case TVN_SELCHANGED:
|
|
OnSelectionChanged((LPNMTREEVIEW)lParam);
|
|
break;
|
|
case TVN_DELETEITEM:
|
|
OnTreeItemDeleted((LPNMTREEVIEW)lParam);
|
|
break;
|
|
case NM_RCLICK:
|
|
OnContextMenu(WM_CONTEXTMENU, (WPARAM)m_hWnd, GetMessagePos(), bHandled);
|
|
if (theResult)
|
|
*theResult = 1;
|
|
break;
|
|
case TVN_BEGINDRAG:
|
|
case TVN_BEGINRDRAG:
|
|
OnTreeItemDragging((LPNMTREEVIEW)lParam, pNotifyHeader->code == TVN_BEGINRDRAG);
|
|
break;
|
|
case TVN_BEGINLABELEDITW:
|
|
{
|
|
// TODO: put this in a function ? (mostly copypasta from CDefView)
|
|
DWORD dwAttr = SFGAO_CANRENAME;
|
|
LPNMTVDISPINFO dispInfo = (LPNMTVDISPINFO)lParam;
|
|
CComPtr<IShellFolder> pParent;
|
|
LPCITEMIDLIST pChild;
|
|
HRESULT hr;
|
|
|
|
if (theResult)
|
|
*theResult = 1;
|
|
NodeInfo *info = GetNodeInfo(dispInfo->item.hItem);
|
|
if (!info)
|
|
return E_FAIL;
|
|
hr = SHBindToParent(info->absolutePidl, IID_PPV_ARG(IShellFolder, &pParent), &pChild);
|
|
if (!SUCCEEDED(hr) || !pParent.p)
|
|
return E_FAIL;
|
|
|
|
hr = pParent->GetAttributesOf(1, &pChild, &dwAttr);
|
|
if (SUCCEEDED(hr) && (dwAttr & SFGAO_CANRENAME))
|
|
{
|
|
if (theResult)
|
|
*theResult = 0;
|
|
m_isEditing = TRUE;
|
|
}
|
|
return S_OK;
|
|
}
|
|
case TVN_ENDLABELEDITW:
|
|
{
|
|
LPNMTVDISPINFO dispInfo = (LPNMTVDISPINFO)lParam;
|
|
NodeInfo *info = GetNodeInfo(dispInfo->item.hItem);
|
|
HRESULT hr;
|
|
|
|
m_isEditing = FALSE;
|
|
if (theResult)
|
|
*theResult = 0;
|
|
if (dispInfo->item.pszText)
|
|
{
|
|
LPITEMIDLIST pidlNew;
|
|
CComPtr<IShellFolder> pParent;
|
|
LPCITEMIDLIST pidlChild;
|
|
|
|
hr = SHBindToParent(info->absolutePidl, IID_PPV_ARG(IShellFolder, &pParent), &pidlChild);
|
|
if (!SUCCEEDED(hr) || !pParent.p)
|
|
return E_FAIL;
|
|
|
|
hr = pParent->SetNameOf(0, pidlChild, dispInfo->item.pszText, SHGDN_INFOLDER, &pidlNew);
|
|
if(SUCCEEDED(hr) && pidlNew)
|
|
{
|
|
CComPtr<IPersistFolder2> pPersist;
|
|
LPITEMIDLIST pidlParent, pidlNewAbs;
|
|
|
|
hr = pParent->QueryInterface(IID_PPV_ARG(IPersistFolder2, &pPersist));
|
|
if(!SUCCEEDED(hr))
|
|
return E_FAIL;
|
|
|
|
hr = pPersist->GetCurFolder(&pidlParent);
|
|
if(!SUCCEEDED(hr))
|
|
return E_FAIL;
|
|
pidlNewAbs = ILCombine(pidlParent, pidlNew);
|
|
|
|
// Navigate to our new location
|
|
UpdateBrowser(pidlNewAbs);
|
|
|
|
ILFree(pidlParent);
|
|
ILFree(pidlNewAbs);
|
|
ILFree(pidlNew);
|
|
if (theResult)
|
|
*theResult = 1;
|
|
}
|
|
return S_OK;
|
|
}
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::IsWindowOwner(HWND hWnd)
|
|
{
|
|
return (hWnd == m_hWnd) ? S_OK : S_FALSE;
|
|
}
|
|
|
|
// *** IBandNavigate methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Select(long paramC)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
// *** INamespaceProxy ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetNavigateTarget(long paramC, long param10, long param14)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Invoke(long paramC)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::OnSelectionChanged(long paramC)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::RefreshFlags(long paramC, long param10, long param14)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::CacheItem(long paramC)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
// *** IDispatch methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetTypeInfoCount(UINT *pctinfo)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetTypeInfo(UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GetIDsOfNames(REFIID riid, LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Invoke(DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
switch (dispIdMember)
|
|
{
|
|
case DISPID_DOWNLOADCOMPLETE:
|
|
case DISPID_NAVIGATECOMPLETE2:
|
|
TRACE("DISPID_NAVIGATECOMPLETE2 received\n");
|
|
NavigateToCurrentFolder();
|
|
return S_OK;
|
|
}
|
|
TRACE("Unknown dispid requested: %08x\n", dispIdMember);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
// *** IDropTarget methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::DragEnter(IDataObject *pObj, DWORD glfKeyState, POINTL pt, DWORD *pdwEffect)
|
|
{
|
|
ERR("Entering drag\n");
|
|
m_pCurObject = pObj;
|
|
m_oldSelected = TreeView_GetSelection(m_hWnd);
|
|
return DragOver(glfKeyState, pt, pdwEffect);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::DragOver(DWORD glfKeyState, POINTL pt, DWORD *pdwEffect)
|
|
{
|
|
TVHITTESTINFO info;
|
|
CComPtr<IShellFolder> pShellFldr;
|
|
NodeInfo *nodeInfo;
|
|
//LPCITEMIDLIST pChild;
|
|
HRESULT hr;
|
|
|
|
info.pt.x = pt.x;
|
|
info.pt.y = pt.y;
|
|
info.flags = TVHT_ONITEM;
|
|
info.hItem = NULL;
|
|
ScreenToClient(&info.pt);
|
|
|
|
// Move to the item selected by the treeview (don't change right pane)
|
|
TreeView_HitTest(m_hWnd, &info);
|
|
|
|
if (info.hItem)
|
|
{
|
|
m_bNavigating = TRUE;
|
|
TreeView_SelectItem(m_hWnd, info.hItem);
|
|
m_bNavigating = FALSE;
|
|
// Delegate to shell folder
|
|
if (m_pDropTarget && info.hItem != m_childTargetNode)
|
|
{
|
|
m_pDropTarget = NULL;
|
|
}
|
|
if (info.hItem != m_childTargetNode)
|
|
{
|
|
nodeInfo = GetNodeInfo(info.hItem);
|
|
if (!nodeInfo)
|
|
return E_FAIL;
|
|
#if 0
|
|
hr = SHBindToParent(nodeInfo->absolutePidl, IID_PPV_ARG(IShellFolder, &pShellFldr), &pChild);
|
|
if (!SUCCEEDED(hr))
|
|
return E_FAIL;
|
|
hr = pShellFldr->GetUIObjectOf(m_hWnd, 1, &pChild, IID_IDropTarget, NULL, reinterpret_cast<void**>(&pDropTarget));
|
|
if (!SUCCEEDED(hr))
|
|
return E_FAIL;
|
|
#endif
|
|
if(_ILIsDesktop(nodeInfo->absolutePidl))
|
|
pShellFldr = m_pDesktop;
|
|
else
|
|
{
|
|
hr = m_pDesktop->BindToObject(nodeInfo->absolutePidl, 0, IID_PPV_ARG(IShellFolder, &pShellFldr));
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
/* Don't allow dnd since we couldn't get our folder object */
|
|
ERR("Can't bind to folder object\n");
|
|
*pdwEffect = DROPEFFECT_NONE;
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
hr = pShellFldr->CreateViewObject(m_hWnd, IID_PPV_ARG(IDropTarget, &m_pDropTarget));
|
|
if (!SUCCEEDED(hr))
|
|
{
|
|
/* Don't allow dnd since we couldn't get our drop target */
|
|
ERR("Can't get drop target for folder object\n");
|
|
*pdwEffect = DROPEFFECT_NONE;
|
|
return E_FAIL;
|
|
}
|
|
hr = m_pDropTarget->DragEnter(m_pCurObject, glfKeyState, pt, pdwEffect);
|
|
m_childTargetNode = info.hItem;
|
|
}
|
|
if (m_pDropTarget)
|
|
{
|
|
hr = m_pDropTarget->DragOver(glfKeyState, pt, pdwEffect);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
m_childTargetNode = NULL;
|
|
m_pDropTarget = NULL;
|
|
*pdwEffect = DROPEFFECT_NONE;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::DragLeave()
|
|
{
|
|
m_bNavigating = TRUE;
|
|
TreeView_SelectItem(m_hWnd, m_oldSelected);
|
|
m_bNavigating = FALSE;
|
|
m_childTargetNode = NULL;
|
|
if (m_pCurObject)
|
|
{
|
|
m_pCurObject = NULL;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::Drop(IDataObject *pObj, DWORD glfKeyState, POINTL pt, DWORD *pdwEffect)
|
|
{
|
|
if (!m_pDropTarget)
|
|
return E_FAIL;
|
|
m_pDropTarget->Drop(pObj, glfKeyState, pt, pdwEffect);
|
|
DragLeave();
|
|
return S_OK;
|
|
}
|
|
|
|
// *** IDropSource methods ***
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::QueryContinueDrag(BOOL fEscapePressed, DWORD grfKeyState)
|
|
{
|
|
if (fEscapePressed)
|
|
return DRAGDROP_S_CANCEL;
|
|
if ((grfKeyState & MK_LBUTTON) || (grfKeyState & MK_RBUTTON))
|
|
return S_OK;
|
|
return DRAGDROP_S_DROP;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE CExplorerBand::GiveFeedback(DWORD dwEffect)
|
|
{
|
|
return DRAGDROP_S_USEDEFAULTCURSORS;
|
|
}
|