Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-09-15 15:19:37 +00:00
Code Issues Packages Projects Releases Wiki Activity
reactos/dll/win32/shell32/brfolder.cpp
Katayama Hirofumi MZ 4696996ee2
[SHELL32] brfolder.cpp: Remove pEnumIL member (#6946) 
Refactoring. Simplify item insertion code.
JIRA issue: CORE-17340
- Remove pEnumIL member from BrItemData structure.
- Add BrFolder_GetChildrenEnum helper function.
- Remove IEnumIDList* argument from BrFolder_InsertItem and
  BrFolder_Expand functions.
- Add BrFolder_OnOK function and use it.
2024-05-28 08:00:12 +09:00

1152 lines
34 KiB
C++
Raw Blame History

/*
* PROJECT: ReactOS shell32
* LICENSE: LGPL-2.1-or-later (https://spdx.org/licenses/LGPL-2.1-or-later)
* PURPOSE: SHBrowseForFolderA/W functions
* COPYRIGHT: Copyright 1999 Juergen Schmied
* Copyright 2024 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
*/
// FIXME: Many flags unimplemented
#include "precomp.h"
#include <ui/layout.h> // Resizable window
WINE_DEFAULT_DEBUG_CHANNEL(shell);
#define SHV_CHANGE_NOTIFY (WM_USER + 0x1111)
struct BrFolder
{
LPBROWSEINFOW lpBrowseInfo;
HWND hWnd;
HWND hwndTreeView;
PIDLIST_ABSOLUTE pidlRet;
LAYOUT_DATA* layout; // Filled by LayoutInit, used by LayoutUpdate
SIZE szMin;
ULONG hChangeNotify; // Change notification handle
};
struct BrItemData
{
CComPtr<IShellFolder> lpsfParent; // IShellFolder of the parent
CComHeapPtr<ITEMIDLIST_RELATIVE> pidlChild; // PIDL relative to parent
CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlFull; // Fully qualified PIDL
};
static const LAYOUT_INFO g_layout_info[] =
{
{ IDC_BROWSE_FOR_FOLDER_TITLE, BF_TOP | BF_LEFT | BF_RIGHT },
{ IDC_BROWSE_FOR_FOLDER_STATUS, BF_TOP | BF_LEFT | BF_RIGHT },
{ IDC_BROWSE_FOR_FOLDER_TREEVIEW, BF_TOP | BF_BOTTOM | BF_LEFT | BF_RIGHT },
{ IDC_BROWSE_FOR_FOLDER_FOLDER_TEXT, BF_TOP | BF_LEFT | BF_RIGHT },
{ IDC_BROWSE_FOR_FOLDER_NEW_FOLDER, BF_BOTTOM | BF_LEFT },
{ IDOK, BF_BOTTOM | BF_RIGHT },
{ IDCANCEL, BF_BOTTOM | BF_RIGHT },
};
#define SUPPORTED_FLAGS (BIF_STATUSTEXT | BIF_BROWSEFORCOMPUTER | BIF_RETURNFSANCESTORS | \
BIF_RETURNONLYFSDIRS | BIF_NONEWFOLDERBUTTON | BIF_NEWDIALOGSTYLE | \
BIF_BROWSEINCLUDEFILES)
static HTREEITEM
BrFolder_InsertItem(
_Inout_ BrFolder *info,
_Inout_ IShellFolder *lpsf,
_In_ PCUITEMID_CHILD pidlChild,
_In_ PCIDLIST_ABSOLUTE pidlParent,
_In_ HTREEITEM hParent);
static inline DWORD
BrowseFlagsToSHCONTF(UINT ulFlags)
{
return SHCONTF_FOLDERS | ((ulFlags & BIF_BROWSEINCLUDEFILES) ? SHCONTF_NONFOLDERS : 0);
}
static void
BrFolder_Callback(LPBROWSEINFOW lpBrowseInfo, HWND hWnd, UINT uMsg, LPARAM lParam)
{
if (!lpBrowseInfo->lpfn)
return;
lpBrowseInfo->lpfn(hWnd, uMsg, lParam, lpBrowseInfo->lParam);
}
static BrItemData *
BrFolder_GetItemData(BrFolder *info, HTREEITEM hItem)
{
TVITEMW item = { TVIF_HANDLE | TVIF_PARAM };
item.hItem = hItem;
if (!TreeView_GetItem(info->hwndTreeView, &item))
{
ERR("TreeView_GetItem failed\n");
return NULL;
}
return (BrItemData *)item.lParam;
}
static HRESULT
BrFolder_GetChildrenEnum(
_In_ BrFolder *info,
_In_ BrItemData *pItemData,
_Outptr_opt_ IEnumIDList **ppEnum)
{
if (!pItemData)
return E_FAIL;
CComPtr<IEnumIDList> pEnumIL;
PCUITEMID_CHILD pidlRef = pItemData->pidlChild;
ULONG attrs = SFGAO_HASSUBFOLDER | SFGAO_FOLDER;
IShellFolder *lpsf = pItemData->lpsfParent;
HRESULT hr = lpsf->GetAttributesOf(1, &pidlRef, &attrs);
if (FAILED_UNEXPECTEDLY(hr) || !(attrs & SFGAO_FOLDER))
return E_FAIL;
CComPtr<IShellFolder> psfChild;
if (_ILIsDesktop(pItemData->pidlFull))
hr = SHGetDesktopFolder(&psfChild);
else
hr = lpsf->BindToObject(pidlRef, NULL, IID_PPV_ARG(IShellFolder, &psfChild));
if (FAILED_UNEXPECTEDLY(hr))
return E_FAIL;
DWORD flags = BrowseFlagsToSHCONTF(info->lpBrowseInfo->ulFlags);
hr = psfChild->EnumObjects(info->hWnd, flags, &pEnumIL);
if (hr == S_OK)
{
if ((pEnumIL->Skip(1) != S_OK) || FAILED(pEnumIL->Reset()))
pEnumIL = NULL;
}
if (!pEnumIL || !(attrs & SFGAO_HASSUBFOLDER))
return E_FAIL; // No children
if (ppEnum)
*ppEnum = pEnumIL.Detach();
return S_OK; // There are children
}
/******************************************************************************
* BrFolder_InitTreeView [Internal]
*
* Called from WM_INITDIALOG handler.
*/
static void
BrFolder_InitTreeView(BrFolder *info)
{
HIMAGELIST hImageList;
HRESULT hr;
CComPtr<IShellFolder> lpsfParent;
CComPtr<IEnumIDList> pEnumChildren;
HTREEITEM hItem;
Shell_GetImageLists(NULL, &hImageList);
if (hImageList)
TreeView_SetImageList(info->hwndTreeView, hImageList, 0);
/* We want to call BrFolder_InsertItem down the code, in order to insert
* the root item of the treeview. Due to BrFolder_InsertItem's signature,
* we need the following to do this:
*
* + An ITEMIDLIST corresponding to _the parent_ of root.
* + An ITEMIDLIST, which is a relative path from root's parent to root
* (containing a single SHITEMID).
* + An IShellFolder interface pointer of root's parent folder.
*
* If root is 'Desktop', then root's parent is also 'Desktop'.
*/
PCIDLIST_ABSOLUTE pidlRoot = info->lpBrowseInfo->pidlRoot;
CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlParent(ILClone(pidlRoot));
ILRemoveLastID(pidlParent);
CComHeapPtr<ITEMIDLIST_RELATIVE> pidlChild(ILClone(ILFindLastID(pidlRoot)));
if (_ILIsDesktop(pidlParent))
{
hr = SHGetDesktopFolder(&lpsfParent);
if (FAILED_UNEXPECTEDLY(hr))
return;
}
else
{
CComPtr<IShellFolder> lpsfDesktop;
hr = SHGetDesktopFolder(&lpsfDesktop);
if (FAILED_UNEXPECTEDLY(hr))
return;
hr = lpsfDesktop->BindToObject(pidlParent, NULL, IID_PPV_ARG(IShellFolder, &lpsfParent));
if (FAILED_UNEXPECTEDLY(hr))
return;
}
TreeView_DeleteItem(info->hwndTreeView, TVI_ROOT);
hItem = BrFolder_InsertItem(info, lpsfParent, pidlChild, pidlParent, TVI_ROOT);
TreeView_Expand(info->hwndTreeView, hItem, TVE_EXPAND);
}
static INT
BrFolder_GetIcon(PCIDLIST_ABSOLUTE pidl, UINT uFlags)
{
SHFILEINFOW sfi;
SHGetFileInfoW((LPCWSTR)pidl, 0, &sfi, sizeof(sfi), uFlags);
return sfi.iIcon;
}
static void
BrFolder_GetIconPair(PCIDLIST_ABSOLUTE pidl, LPTVITEMW pItem)
{
DWORD flags;
CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlDesktop;
if (!pidl)
{
pidlDesktop.Attach(_ILCreateDesktop());
pidl = pidlDesktop;
}
flags = SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON;
pItem->iImage = BrFolder_GetIcon(pidl, flags);
flags = SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_OPENICON;
pItem->iSelectedImage = BrFolder_GetIcon(pidl, flags);
}
/******************************************************************************
* BrFolder_GetName [Internal]
*
* Query a shell folder for the display name of one of its children
*
* PARAMS
* lpsf [I] IShellFolder interface of the folder to be queried.
* pidlChild [I] ITEMIDLIST of the child, relative to parent
* dwFlags [I] as in IShellFolder::GetDisplayNameOf
* lpFriendlyName [O] The desired display name in unicode
*
* RETURNS
* Success: TRUE
* Failure: FALSE
*/
static BOOL
BrFolder_GetName(
IShellFolder *lpsf,
PCIDLIST_RELATIVE pidlChild,
DWORD dwFlags,
LPWSTR lpFriendlyName)
{
BOOL bSuccess = FALSE;
STRRET str;
TRACE("%p %p %x %p\n", lpsf, pidlChild, dwFlags, lpFriendlyName);
if (!FAILED_UNEXPECTEDLY(lpsf->GetDisplayNameOf(pidlChild, dwFlags, &str)))
bSuccess = StrRetToStrNW(lpFriendlyName, MAX_PATH, &str, pidlChild);
TRACE("-- %s\n", debugstr_w(lpFriendlyName));
return bSuccess;
}
/******************************************************************************
* BrFolder_InsertItem [Internal]
*
* PARAMS
* info [I] data for the dialog
* lpsf [I] IShellFolder interface of the item's parent shell folder
* pidlChild [I] ITEMIDLIST of the child to insert, relative to parent
* pidlParent [I] ITEMIDLIST of the parent shell folder
* hParent [I] The treeview-item that represents the parent shell folder
*
* RETURNS
* Success: Handle to the created and inserted treeview-item
* Failure: NULL
*/
static HTREEITEM
BrFolder_InsertItem(
_Inout_ BrFolder *info,
_Inout_ IShellFolder *lpsf,
_In_ PCUITEMID_CHILD pidlChild,
_In_ PCIDLIST_ABSOLUTE pidlParent,
_In_ HTREEITEM hParent)
{
WCHAR szName[MAX_PATH];
if (!BrFolder_GetName(lpsf, pidlChild, SHGDN_NORMAL, szName))
return NULL;
BrItemData *pItemData = new BrItemData();
TVITEMW item = { TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM | TVIF_CHILDREN };
item.pszText = szName;
item.cchTextMax = _countof(szName);
item.lParam = (LPARAM)pItemData;
PIDLIST_ABSOLUTE pidlFull =
(pidlParent ? ILCombine(pidlParent, pidlChild) : ILClone(pidlChild));
BrFolder_GetIconPair(pidlFull, &item);
pItemData->lpsfParent = lpsf;
pItemData->pidlChild.Attach(ILClone(pidlChild));
pItemData->pidlFull.Attach(pidlFull);
if (BrFolder_GetChildrenEnum(info, pItemData, NULL) == S_OK)
item.cChildren = 1;
TVINSERTSTRUCTW tvins = { hParent };
tvins.item = item;
return TreeView_InsertItem(info->hwndTreeView, &tvins);
}
/******************************************************************************
* BrFolder_Expand [Internal]
*
* For each child (given by pEnum) of the parent shell folder, which is given by
* lpsf and whose PIDL is pidl, insert a treeview-item right under hParent
*
* PARAMS
* info [I] data for the dialog
* lpsf [I] IShellFolder interface of the parent shell folder
* pidl [I] ITEMIDLIST of the parent shell folder
* hParent [I] The treeview item that represents the parent shell folder
* pEnum [I] An iterator for the children of the parent shell folder
*/
static void
BrFolder_Expand(
_In_ BrFolder *info,
_In_ IShellFolder *lpsf,
_In_ PCIDLIST_ABSOLUTE pidlFull,
_In_ HTREEITEM hParent)
{
TRACE("%p %p %p\n", lpsf, pidlFull, hParent);
// No IEnumIDList -> No children
BrItemData *pItemData = BrFolder_GetItemData(info, hParent);
CComPtr<IEnumIDList> pEnum;
HRESULT hr = BrFolder_GetChildrenEnum(info, pItemData, &pEnum);
if (FAILED(hr))
return;
SetCapture(info->hWnd);
SetCursor(LoadCursorW(NULL, (LPWSTR)IDC_WAIT));
CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlTemp;
ULONG ulFetched;
while (S_OK == pEnum->Next(1, &pidlTemp, &ulFetched))
{
if (!BrFolder_InsertItem(info, lpsf, pidlTemp, pidlFull, hParent))
break;
pidlTemp.Free(); // Finally, free the pidl that the shell gave us...
}
ReleaseCapture();
SetCursor(LoadCursorW(NULL, (LPWSTR)IDC_ARROW));
}
static inline BOOL
PIDLIsType(LPCITEMIDLIST pidl, PIDLTYPE type)
{
LPPIDLDATA data = _ILGetDataPointer(pidl);
if (!data)
return FALSE;
return (data->type == type);
}
static void
BrFolder_CheckValidSelection(BrFolder *info, BrItemData *pItemData)
{
LPBROWSEINFOW lpBrowseInfo = info->lpBrowseInfo;
PCIDLIST_RELATIVE pidlChild = pItemData->pidlChild;
DWORD dwAttributes;
HRESULT hr;
BOOL bEnabled = TRUE;
if ((lpBrowseInfo->ulFlags & BIF_BROWSEFORCOMPUTER) && !PIDLIsType(pidlChild, PT_COMP))
bEnabled = FALSE;
if (lpBrowseInfo->ulFlags & BIF_RETURNFSANCESTORS)
{
dwAttributes = SFGAO_FILESYSANCESTOR | SFGAO_FILESYSTEM;
hr = pItemData->lpsfParent->GetAttributesOf(1, &pidlChild, &dwAttributes);
if (FAILED(hr) || !(dwAttributes & (SFGAO_FILESYSANCESTOR | SFGAO_FILESYSTEM)))
bEnabled = FALSE;
}
dwAttributes = SFGAO_FOLDER | SFGAO_FILESYSTEM;
hr = pItemData->lpsfParent->GetAttributesOf(1, &pidlChild, &dwAttributes);
if (FAILED_UNEXPECTEDLY(hr) ||
((dwAttributes & (SFGAO_FOLDER | SFGAO_FILESYSTEM)) != (SFGAO_FOLDER | SFGAO_FILESYSTEM)))
{
if (lpBrowseInfo->ulFlags & BIF_RETURNONLYFSDIRS)
bEnabled = FALSE;
EnableWindow(GetDlgItem(info->hWnd, IDC_BROWSE_FOR_FOLDER_NEW_FOLDER), FALSE);
}
else
{
EnableWindow(GetDlgItem(info->hWnd, IDC_BROWSE_FOR_FOLDER_NEW_FOLDER), TRUE);
}
SendMessageW(info->hWnd, BFFM_ENABLEOK, 0, bEnabled);
}
static LRESULT
BrFolder_Treeview_Delete(BrFolder *info, NMTREEVIEWW *pnmtv)
{
BrItemData *pItemData = (BrItemData *)pnmtv->itemOld.lParam;
TRACE("TVN_DELETEITEMA/W %p\n", pItemData);
delete pItemData;
return 0;
}
static LRESULT
BrFolder_Treeview_Expand(BrFolder *info, NMTREEVIEWW *pnmtv)
{
BrItemData *pItemData = (BrItemData *)pnmtv->itemNew.lParam;
TRACE("TVN_ITEMEXPANDINGA/W\n");
if ((pnmtv->itemNew.state & TVIS_EXPANDEDONCE))
return 0;
HRESULT hr = S_OK;
CComPtr<IShellFolder> lpsf2;
if (!_ILIsEmpty(pItemData->pidlChild))
{
hr = pItemData->lpsfParent->BindToObject(pItemData->pidlChild, NULL,
IID_PPV_ARG(IShellFolder, &lpsf2));
}
else
{
lpsf2 = pItemData->lpsfParent;
}
HTREEITEM hItem = pnmtv->itemNew.hItem;
if (!FAILED_UNEXPECTEDLY(hr))
BrFolder_Expand(info, lpsf2, pItemData->pidlFull, hItem);
// My Computer is already sorted and trying to do a simple text
// sort will only mess things up
if (!_ILIsMyComputer(pItemData->pidlChild))
TreeView_SortChildren(info->hwndTreeView, hItem, FALSE);
return 0;
}
static HRESULT
BrFolder_Treeview_Changed(BrFolder *info, NMTREEVIEWW *pnmtv)
{
BrItemData *pItemData = (BrItemData *)pnmtv->itemNew.lParam;
ILFree(info->pidlRet);
info->pidlRet = ILClone(pItemData->pidlFull);
WCHAR szName[MAX_PATH];
if (BrFolder_GetName(pItemData->lpsfParent, pItemData->pidlChild, SHGDN_NORMAL, szName))
SetDlgItemTextW(info->hWnd, IDC_BROWSE_FOR_FOLDER_FOLDER_TEXT, szName);
BrFolder_Callback(info->lpBrowseInfo, info->hWnd, BFFM_SELCHANGED, (LPARAM)info->pidlRet);
BrFolder_CheckValidSelection(info, pItemData);
return S_OK;
}
static LRESULT
BrFolder_Treeview_Rename(BrFolder *info, NMTVDISPINFOW *pnmtv)
{
WCHAR old_path[MAX_PATH], new_path[MAX_PATH];
NMTREEVIEWW nmtv;
TVITEMW item;
if (!pnmtv->item.pszText)
return 0;
item.hItem = TreeView_GetSelection(info->hwndTreeView);
BrItemData *item_data = BrFolder_GetItemData(info, item.hItem);
SHGetPathFromIDListW(item_data->pidlFull, old_path);
lstrcpynW(new_path, old_path, _countof(new_path));
PathRemoveFileSpecW(new_path);
PathAppendW(new_path, pnmtv->item.pszText);
if (!MoveFileW(old_path, new_path))
return 0;
item_data->pidlFull.Free();
item_data->pidlChild.Free();
item_data->pidlFull.Attach(SHSimpleIDListFromPathW(new_path));
item_data->lpsfParent->ParseDisplayName(NULL, NULL, pnmtv->item.pszText, NULL,
&item_data->pidlChild, NULL);
item.mask = TVIF_HANDLE | TVIF_TEXT;
item.pszText = pnmtv->item.pszText;
TreeView_SetItem(info->hwndTreeView, &item);
nmtv.itemNew.lParam = item.lParam;
BrFolder_Treeview_Changed(info, &nmtv);
return 0;
}
static HRESULT
BrFolder_Rename(BrFolder *info, HTREEITEM hItem)
{
TreeView_SelectItem(info->hwndTreeView, hItem);
TreeView_EditLabel(info->hwndTreeView, hItem);
return S_OK;
}
static void
BrFolder_Delete(BrFolder *info, HTREEITEM hItem)
{
SHFILEOPSTRUCTW fileop = { info->hwndTreeView };
WCHAR szzFrom[MAX_PATH + 1];
// Get item_data
BrItemData *item_data = BrFolder_GetItemData(info, hItem);
// Get the path
if (!SHGetPathFromIDListW(item_data->pidlFull, szzFrom))
{
ERR("SHGetPathFromIDListW failed\n");
return;
}
szzFrom[lstrlenW(szzFrom) + 1] = UNICODE_NULL; // Double NULL-terminated
fileop.pFrom = szzFrom;
// Delete folder
fileop.fFlags = FOF_ALLOWUNDO;
fileop.wFunc = FO_DELETE;
SHFileOperationW(&fileop);
}
static LRESULT
BrFolder_Treeview_Keydown(BrFolder *info, LPNMTVKEYDOWN keydown)
{
// Old dialog doesn't support those advanced features
if (!(info->lpBrowseInfo->ulFlags & BIF_USENEWUI))
return 0;
HTREEITEM hItem = TreeView_GetSelection(info->hwndTreeView);
switch (keydown->wVKey)
{
case VK_F2:
BrFolder_Rename(info, hItem);
break;
case VK_DELETE:
BrFolder_Delete(info, hItem);
break;
}
return 0;
}
static LRESULT
BrFolder_OnNotify(BrFolder *info, UINT CtlID, LPNMHDR lpnmh)
{
NMTREEVIEWW *pnmtv = (NMTREEVIEWW *)lpnmh;
TRACE("%p %x %p msg=%x\n", info, CtlID, lpnmh, pnmtv->hdr.code);
if (pnmtv->hdr.idFrom != IDC_BROWSE_FOR_FOLDER_TREEVIEW)
return 0;
switch (pnmtv->hdr.code)
{
case TVN_DELETEITEMA:
case TVN_DELETEITEMW:
return BrFolder_Treeview_Delete(info, pnmtv);
case TVN_ITEMEXPANDINGA:
case TVN_ITEMEXPANDINGW:
return BrFolder_Treeview_Expand(info, pnmtv);
case TVN_SELCHANGEDA:
case TVN_SELCHANGEDW:
return BrFolder_Treeview_Changed(info, pnmtv);
case TVN_ENDLABELEDITA:
case TVN_ENDLABELEDITW:
return BrFolder_Treeview_Rename(info, (LPNMTVDISPINFOW)pnmtv);
case TVN_KEYDOWN:
return BrFolder_Treeview_Keydown(info, (LPNMTVKEYDOWN)pnmtv);
default:
WARN("unhandled (%d)\n", pnmtv->hdr.code);
break;
}
return 0;
}
static BOOL
BrFolder_OnInitDialog(HWND hWnd, BrFolder *info)
{
CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlDesktop;
SHChangeNotifyEntry ntreg;
LPBROWSEINFOW lpBrowseInfo = info->lpBrowseInfo;
info->hWnd = hWnd;
SetPropW(hWnd, L"__WINE_BRSFOLDERDLG_INFO", info);
if (lpBrowseInfo->ulFlags & BIF_NEWDIALOGSTYLE)
FIXME("flags BIF_NEWDIALOGSTYLE partially implemented\n");
if (lpBrowseInfo->ulFlags & ~SUPPORTED_FLAGS)
FIXME("flags %x not implemented\n", (lpBrowseInfo->ulFlags & ~SUPPORTED_FLAGS));
info->layout = NULL;
if (lpBrowseInfo->ulFlags & BIF_USENEWUI)
{
RECT rcWnd;
// Resize the treeview if there's not editbox
if ((lpBrowseInfo->ulFlags & BIF_NEWDIALOGSTYLE) &&
!(lpBrowseInfo->ulFlags & BIF_EDITBOX))
{
RECT rcEdit, rcTreeView;
GetWindowRect(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_FOLDER_TEXT), &rcEdit);
GetWindowRect(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_TREEVIEW), &rcTreeView);
LONG cy = rcTreeView.top - rcEdit.top;
MapWindowPoints(NULL, hWnd, (LPPOINT)&rcTreeView, sizeof(RECT) / sizeof(POINT));
rcTreeView.top -= cy;
MoveWindow(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_TREEVIEW),
rcTreeView.left, rcTreeView.top,
rcTreeView.right - rcTreeView.left,
rcTreeView.bottom - rcTreeView.top, TRUE);
}
if (lpBrowseInfo->ulFlags & BIF_NEWDIALOGSTYLE)
info->layout = LayoutInit(hWnd, g_layout_info, _countof(g_layout_info));
// TODO: Windows allows shrinking the windows a bit
GetWindowRect(hWnd, &rcWnd);
info->szMin.cx = rcWnd.right - rcWnd.left;
info->szMin.cy = rcWnd.bottom - rcWnd.top;
}
if (lpBrowseInfo->lpszTitle)
SetDlgItemTextW(hWnd, IDC_BROWSE_FOR_FOLDER_TITLE, lpBrowseInfo->lpszTitle);
else
ShowWindow(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_TITLE), SW_HIDE);
if (!(lpBrowseInfo->ulFlags & BIF_STATUSTEXT) || (lpBrowseInfo->ulFlags & BIF_USENEWUI))
ShowWindow(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_STATUS), SW_HIDE);
// Hide "Make New Folder" Button?
if ((lpBrowseInfo->ulFlags & BIF_NONEWFOLDERBUTTON) ||
!(lpBrowseInfo->ulFlags & BIF_NEWDIALOGSTYLE))
{
ShowWindow(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_NEW_FOLDER), SW_HIDE);
}
// Hide the editbox?
if (!(lpBrowseInfo->ulFlags & BIF_EDITBOX))
{
ShowWindow(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_FOLDER), SW_HIDE);
ShowWindow(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_FOLDER_TEXT), SW_HIDE);
}
info->hwndTreeView = GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_TREEVIEW);
if (info->hwndTreeView)
BrFolder_InitTreeView(info);
else
ERR("treeview control missing!\n");
// Register for change notifications
SHGetFolderLocation(NULL, CSIDL_DESKTOP, NULL, 0, &pidlDesktop);
ntreg.pidl = pidlDesktop;
ntreg.fRecursive = TRUE;
info->hChangeNotify = SHChangeNotifyRegister(hWnd,
SHCNRF_ShellLevel | SHCNRF_NewDelivery,
SHCNE_ALLEVENTS,
SHV_CHANGE_NOTIFY, 1, &ntreg);
BrFolder_Callback(info->lpBrowseInfo, hWnd, BFFM_INITIALIZED, 0);
SHAutoComplete(GetDlgItem(hWnd, IDC_BROWSE_FOR_FOLDER_FOLDER_TEXT),
(SHACF_FILESYS_ONLY | SHACF_URLHISTORY | SHACF_FILESYSTEM));
return TRUE;
}
static HRESULT
BrFolder_NewFolder(BrFolder *info)
{
CComPtr<IShellFolder> desktop, cur;
WCHAR wszNewFolder[25], path[MAX_PATH], name[MAX_PATH];
HRESULT hr = SHGetDesktopFolder(&desktop);
if (FAILED_UNEXPECTEDLY(hr))
return hr;
if (info->pidlRet)
{
hr = desktop->BindToObject(info->pidlRet, NULL, IID_PPV_ARG(IShellFolder, &cur));
if (FAILED_UNEXPECTEDLY(hr))
return hr;
hr = SHGetPathFromIDListW(info->pidlRet, path);
}
else
{
cur = desktop;
hr = SHGetFolderPathW(NULL, CSIDL_DESKTOPDIRECTORY, NULL, SHGFP_TYPE_CURRENT, path);
}
if (FAILED_UNEXPECTEDLY(hr))
return hr;
hr = E_FAIL;
if (!LoadStringW(shell32_hInstance, IDS_NEWFOLDER, wszNewFolder, _countof(wszNewFolder)))
return hr;
if (!PathYetAnotherMakeUniqueName(name, path, NULL, wszNewFolder))
return hr;
INT len = lstrlenW(path);
if (len < MAX_PATH && name[len] == L'\\')
len++;
if (!CreateDirectoryW(name, NULL))
return hr;
// Update parent of newly created directory
HTREEITEM hParent = TreeView_GetSelection(info->hwndTreeView);
if (!hParent)
return hr;
TreeView_Expand(info->hwndTreeView, hParent, TVE_EXPAND);
TVITEMW item = { TVIF_PARAM | TVIF_STATE };
item.hItem = hParent;
TreeView_GetItem(info->hwndTreeView, &item);
BrItemData *item_data = (BrItemData *)item.lParam;
if (!item_data)
return hr;
// Update treeview
if (!(item.state & TVIS_EXPANDEDONCE))
{
item.mask = TVIF_STATE;
item.state = item.stateMask = TVIS_EXPANDEDONCE;
TreeView_SetItem(info->hwndTreeView, &item);
}
CComHeapPtr<ITEMIDLIST_RELATIVE> pidlNew;
hr = cur->ParseDisplayName(NULL, NULL, name + len, NULL, &pidlNew, NULL);
if (FAILED_UNEXPECTEDLY(hr))
return hr;
HTREEITEM hAdded = BrFolder_InsertItem(info, cur, pidlNew, item_data->pidlFull, hParent);
TreeView_SortChildren(info->hwndTreeView, hParent, FALSE);
return BrFolder_Rename(info, hAdded);
}
static void
BrFolder_OnOK(BrFolder *info)
{
// Get the text
WCHAR szPath[MAX_PATH];
GetDlgItemTextW(info->hWnd, IDC_BROWSE_FOR_FOLDER_FOLDER_TEXT, szPath, _countof(szPath));
StrTrimW(szPath, L" \t");
// The original pidl is owned by the treeview and will be free'd.
if (!PathIsRelativeW(szPath) && PathIsDirectoryW(szPath))
info->pidlRet = ILCreateFromPathW(szPath); // It's valid path
else
info->pidlRet = ILClone(info->pidlRet);
if (!info->pidlRet) // A null pidl would mean a cancel
info->pidlRet = _ILCreateDesktop();
pdump(info->pidlRet);
LPBROWSEINFOW lpBrowseInfo = info->lpBrowseInfo;
if (!lpBrowseInfo->pszDisplayName)
return;
SHFILEINFOW fileInfo = { NULL };
lpBrowseInfo->pszDisplayName[0] = UNICODE_NULL;
if (SHGetFileInfoW((LPCWSTR)info->pidlRet, 0, &fileInfo, sizeof(fileInfo),
SHGFI_PIDL | SHGFI_DISPLAYNAME))
{
lstrcpynW(lpBrowseInfo->pszDisplayName, fileInfo.szDisplayName, MAX_PATH);
}
}
static void
BrFolder_OnCommand(BrFolder *info, UINT id)
{
switch (id)
{
case IDOK:
{
BrFolder_OnOK(info);
EndDialog(info->hWnd, IDOK);
break;
}
case IDCANCEL:
{
EndDialog(info->hWnd, IDCANCEL);
break;
}
case IDC_BROWSE_FOR_FOLDER_NEW_FOLDER:
{
BrFolder_NewFolder(info);
break;
}
}
}
static BOOL
BrFolder_OnSetExpandedPidl(BrFolder *info, LPITEMIDLIST pidlSelection, HTREEITEM *phItem)
{
if (_ILIsDesktop(pidlSelection))
{
if (phItem)
*phItem = TVI_ROOT;
return TRUE;
}
// Move pidlCurrent behind the SHITEMIDs in pidlSelection, which are the root of
// the sub-tree currently displayed.
PCIDLIST_ABSOLUTE pidlRoot = info->lpBrowseInfo->pidlRoot;
LPITEMIDLIST pidlCurrent = pidlSelection;
while (!_ILIsEmpty(pidlRoot) && _ILIsEqualSimple(pidlRoot, pidlCurrent))
{
pidlRoot = ILGetNext(pidlRoot);
pidlCurrent = ILGetNext(pidlCurrent);
}
// The given ID List is not part of the SHBrowseForFolder's current sub-tree.
if (!_ILIsEmpty(pidlRoot))
{
if (phItem)
*phItem = NULL;
return FALSE;
}
// Initialize item to point to the first child of the root folder.
TVITEMEXW item = { TVIF_PARAM };
item.hItem = TreeView_GetRoot(info->hwndTreeView);
if (item.hItem)
item.hItem = TreeView_GetChild(info->hwndTreeView, item.hItem);
// Walk the tree along the nodes corresponding to the remaining ITEMIDLIST
while (item.hItem && !_ILIsEmpty(pidlCurrent))
{
TreeView_GetItem(info->hwndTreeView, &item);
BrItemData *pItemData = (BrItemData *)item.lParam;
if (_ILIsEqualSimple(pItemData->pidlChild, pidlCurrent))
{
pidlCurrent = ILGetNext(pidlCurrent);
if (!_ILIsEmpty(pidlCurrent))
{
// Only expand current node and move on to its first child,
// if we didn't already reach the last SHITEMID
TreeView_Expand(info->hwndTreeView, item.hItem, TVE_EXPAND);
item.hItem = TreeView_GetChild(info->hwndTreeView, item.hItem);
}
}
else
{
item.hItem = TreeView_GetNextSibling(info->hwndTreeView, item.hItem);
}
}
if (phItem)
*phItem = item.hItem;
return (_ILIsEmpty(pidlCurrent) && item.hItem);
}
static BOOL
BrFolder_OnSetExpandedString(BrFolder *info, LPWSTR pszString, HTREEITEM *phItem)
{
CComPtr<IShellFolder> psfDesktop;
HRESULT hr = SHGetDesktopFolder(&psfDesktop);
if (FAILED_UNEXPECTEDLY(hr))
return FALSE;
CComHeapPtr<ITEMIDLIST_ABSOLUTE> pidlSelection;
hr = psfDesktop->ParseDisplayName(NULL, NULL, pszString, NULL, &pidlSelection, NULL);
if (FAILED_UNEXPECTEDLY(hr))
return FALSE;
return BrFolder_OnSetExpandedPidl(info, pidlSelection, phItem);
}
static BOOL
BrFolder_OnSetSelectionPidl(BrFolder *info, LPITEMIDLIST pidlSelection)
{
if (!pidlSelection)
return FALSE;
HTREEITEM hItem;
BOOL ret = BrFolder_OnSetExpandedPidl(info, pidlSelection, &hItem);
if (ret)
TreeView_SelectItem(info->hwndTreeView, hItem);
return ret;
}
static BOOL
BrFolder_OnSetSelectionW(BrFolder *info, LPWSTR pszSelection)
{
if (!pszSelection)
return FALSE;
HTREEITEM hItem;
BOOL ret = BrFolder_OnSetExpandedString(info, pszSelection, &hItem);
if (ret)
TreeView_SelectItem(info->hwndTreeView, hItem);
return ret;
}
static BOOL
BrFolder_OnSetSelectionA(BrFolder *info, LPSTR pszSelectionA)
{
if (!pszSelectionA)
return FALSE;
CComHeapPtr<WCHAR> pszSelectionW;
__SHCloneStrAtoW(&pszSelectionW, pszSelectionA);
if (!pszSelectionW)
return FALSE;
return BrFolder_OnSetSelectionW(info, pszSelectionW);
}
static void
BrFolder_OnDestroy(BrFolder *info)
{
if (info->layout)
{
LayoutDestroy(info->layout);
info->layout = NULL;
}
SHChangeNotifyDeregister(info->hChangeNotify);
}
// Find a treeview node by recursively walking the treeview
static HTREEITEM
BrFolder_FindItemByPidl(BrFolder *info, PCIDLIST_ABSOLUTE pidlFull, HTREEITEM hItem)
{
BrItemData *item_data = BrFolder_GetItemData(info, hItem);
if (ILIsEqual(item_data->pidlFull, pidlFull))
return hItem;
for (hItem = TreeView_GetChild(info->hwndTreeView, hItem); hItem;
hItem = TreeView_GetNextSibling(info->hwndTreeView, hItem))
{
HTREEITEM newItem = BrFolder_FindItemByPidl(info, pidlFull, hItem);
if (newItem)
return newItem;
}
return NULL;
}
static void
BrFolder_OnChangeEx(
_Inout_ BrFolder *info,
_In_ PCIDLIST_ABSOLUTE pidl0,
_In_ PCIDLIST_ABSOLUTE pidl1,
_In_ LONG event)
{
TRACE("(%p)->(%p, %p, 0x%lX)\n", info, pidl0, pidl1, event);
switch (event)
{
case SHCNE_DRIVEADD:
case SHCNE_MKDIR:
case SHCNE_CREATE:
{
// FIXME
break;
}
case SHCNE_DRIVEREMOVED:
case SHCNE_RMDIR:
case SHCNE_DELETE:
{
HTREEITEM hRoot = TreeView_GetRoot(info->hwndTreeView);
for (;;)
{
HTREEITEM hItem = BrFolder_FindItemByPidl(info, pidl0, hRoot);
if (!hItem)
break;
TreeView_DeleteItem(info->hwndTreeView, hItem);
}
break;
}
case SHCNE_RENAMEFOLDER:
case SHCNE_RENAMEITEM:
case SHCNE_UPDATEDIR:
case SHCNE_UPDATEITEM:
{
// FIXME
break;
}
}
}
// SHV_CHANGE_NOTIFY
static void
BrFolder_OnChange(BrFolder *info, WPARAM wParam, LPARAM lParam)
{
// We use SHCNRF_NewDelivery method
HANDLE hChange = (HANDLE)wParam;
DWORD dwProcID = (DWORD)lParam;
PIDLIST_ABSOLUTE *ppidl = NULL;
LONG event;
HANDLE hLock = SHChangeNotification_Lock(hChange, dwProcID, &ppidl, &event);
if (hLock == NULL)
{
ERR("hLock == NULL\n");
return;
}
BrFolder_OnChangeEx(info, ppidl[0], ppidl[1], (event & ~SHCNE_INTERRUPT));
SHChangeNotification_Unlock(hLock);
}
/*************************************************************************
* BrFolderDlgProc32 (not an exported API function)
*/
static INT_PTR CALLBACK
BrFolderDlgProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
if (uMsg == WM_INITDIALOG)
return BrFolder_OnInitDialog(hWnd, (BrFolder *)lParam);
BrFolder *info = (BrFolder *)GetPropW(hWnd, L"__WINE_BRSFOLDERDLG_INFO");
if (!info)
return 0;
switch (uMsg)
{
case WM_NOTIFY:
return BrFolder_OnNotify(info, (UINT)wParam, (LPNMHDR)lParam);
case WM_COMMAND:
BrFolder_OnCommand(info, wParam);
break;
case WM_GETMINMAXINFO:
((LPMINMAXINFO)lParam)->ptMinTrackSize.x = info->szMin.cx;
((LPMINMAXINFO)lParam)->ptMinTrackSize.y = info->szMin.cy;
break;
case WM_SIZE:
if (info->layout) // New style dialogs
LayoutUpdate(hWnd, info->layout, g_layout_info, _countof(g_layout_info));
break;
case BFFM_SETSTATUSTEXTA:
SetDlgItemTextA(hWnd, IDC_BROWSE_FOR_FOLDER_STATUS, (LPSTR)lParam);
break;
case BFFM_SETSTATUSTEXTW:
SetDlgItemTextW(hWnd, IDC_BROWSE_FOR_FOLDER_STATUS, (LPWSTR)lParam);
break;
case BFFM_ENABLEOK:
EnableWindow(GetDlgItem(hWnd, IDOK), lParam != 0);
break;
case BFFM_SETOKTEXT: // Unicode only
SetDlgItemTextW(hWnd, IDOK, (LPWSTR)lParam);
break;
case BFFM_SETSELECTIONA:
if (wParam) // String
return BrFolder_OnSetSelectionA(info, (LPSTR)lParam);
else // PIDL
return BrFolder_OnSetSelectionPidl(info, (LPITEMIDLIST)lParam);
case BFFM_SETSELECTIONW:
if (wParam) // String
return BrFolder_OnSetSelectionW(info, (LPWSTR)lParam);
else // PIDL
return BrFolder_OnSetSelectionPidl(info, (LPITEMIDLIST)lParam);
case BFFM_SETEXPANDED: // Unicode only
if (wParam) // String
return BrFolder_OnSetExpandedString(info, (LPWSTR)lParam, NULL);
else // PIDL
return BrFolder_OnSetExpandedPidl(info, (LPITEMIDLIST)lParam, NULL);
case SHV_CHANGE_NOTIFY:
BrFolder_OnChange(info, wParam, lParam);
break;
case WM_DESTROY:
BrFolder_OnDestroy(info);
break;
}
return 0;
}
/*************************************************************************
* SHBrowseForFolderA [SHELL32.@]
* SHBrowseForFolder [SHELL32.@]
*/
EXTERN_C
LPITEMIDLIST WINAPI
SHBrowseForFolderA(LPBROWSEINFOA lpbi)
{
BROWSEINFOW bi;
bi.hwndOwner = lpbi->hwndOwner;
bi.pidlRoot = lpbi->pidlRoot;
WCHAR szName[MAX_PATH];
bi.pszDisplayName = (lpbi->pszDisplayName ? szName : NULL);
CComHeapPtr<WCHAR> pszTitle;
if (lpbi->lpszTitle)
__SHCloneStrAtoW(&pszTitle, lpbi->lpszTitle);
bi.lpszTitle = pszTitle;
bi.ulFlags = lpbi->ulFlags;
bi.lpfn = lpbi->lpfn;
bi.lParam = lpbi->lParam;
bi.iImage = lpbi->iImage;
PIDLIST_ABSOLUTE pidl = SHBrowseForFolderW(&bi);
if (bi.pszDisplayName)
SHUnicodeToAnsi(bi.pszDisplayName, lpbi->pszDisplayName, MAX_PATH);
lpbi->iImage = bi.iImage;
return pidl;
}
/*************************************************************************
* SHBrowseForFolderW [SHELL32.@]
*/
EXTERN_C
LPITEMIDLIST WINAPI
SHBrowseForFolderW(LPBROWSEINFOW lpbi)
{
TRACE("%p\n", lpbi);
BrFolder info = { lpbi };
HRESULT hr = OleInitialize(NULL);
INT id = ((lpbi->ulFlags & BIF_USENEWUI) ? IDD_BROWSE_FOR_FOLDER_NEW : IDD_BROWSE_FOR_FOLDER);
INT_PTR ret = DialogBoxParamW(shell32_hInstance, MAKEINTRESOURCEW(id), lpbi->hwndOwner,
BrFolderDlgProc, (LPARAM)&info);
if (SUCCEEDED(hr))
OleUninitialize();
if (ret != IDOK)
{
ILFree(info.pidlRet);
return NULL;
}
return info.pidlRet;
}
Reference in a new issue View git blame Copy permalink