2019-07-19 20:23:20 +00:00
|
|
|
/*
|
|
|
|
* PROJECT: ReactOS Search Shell Extension
|
|
|
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
|
|
|
* PURPOSE: Search results folder
|
|
|
|
* COPYRIGHT: Copyright 2019 Brock Mammen
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "CFindFolder.h"
|
|
|
|
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(shellfind);
|
|
|
|
|
|
|
|
struct FolderViewColumns
|
|
|
|
{
|
|
|
|
LPCWSTR wzColumnName;
|
|
|
|
DWORD dwDefaultState;
|
|
|
|
int fmt;
|
|
|
|
int cxChar;
|
|
|
|
};
|
|
|
|
|
|
|
|
static FolderViewColumns g_ColumnDefs[] =
|
|
|
|
{
|
|
|
|
{L"Name", SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_LEFT, 30},
|
|
|
|
{L"In Folder", SHCOLSTATE_TYPE_STR | SHCOLSTATE_ONBYDEFAULT, LVCFMT_LEFT, 30},
|
|
|
|
{L"Relevance", SHCOLSTATE_TYPE_STR, LVCFMT_LEFT, 0}
|
|
|
|
};
|
|
|
|
|
2019-07-20 00:27:31 +00:00
|
|
|
static LPITEMIDLIST _ILCreate(LPCWSTR lpszPath, LPCITEMIDLIST lpcFindDataPidl)
|
|
|
|
{
|
|
|
|
int pathLen = (wcslen(lpszPath) + 1) * sizeof(WCHAR);
|
|
|
|
int cbData = sizeof(WORD) + pathLen + lpcFindDataPidl->mkid.cb;
|
|
|
|
LPITEMIDLIST pidl = (LPITEMIDLIST) SHAlloc(cbData + sizeof(WORD));
|
|
|
|
if (!pidl)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
LPBYTE p = (LPBYTE) pidl;
|
|
|
|
*((WORD *) p) = cbData;
|
|
|
|
p += sizeof(WORD);
|
|
|
|
|
|
|
|
memcpy(p, lpszPath, pathLen);
|
|
|
|
p += pathLen;
|
|
|
|
|
|
|
|
memcpy(p, lpcFindDataPidl, lpcFindDataPidl->mkid.cb);
|
|
|
|
p += lpcFindDataPidl->mkid.cb;
|
|
|
|
|
|
|
|
*((WORD *) p) = 0;
|
|
|
|
|
|
|
|
return pidl;
|
|
|
|
}
|
|
|
|
|
|
|
|
static LPCWSTR _ILGetPath(LPCITEMIDLIST pidl)
|
|
|
|
{
|
|
|
|
if (!pidl || !pidl->mkid.cb)
|
|
|
|
return NULL;
|
|
|
|
return (LPCWSTR) pidl->mkid.abID;
|
|
|
|
}
|
|
|
|
|
|
|
|
static LPCITEMIDLIST _ILGetFSPidl(LPCITEMIDLIST pidl)
|
|
|
|
{
|
|
|
|
if (!pidl || !pidl->mkid.cb)
|
|
|
|
return pidl;
|
|
|
|
return (LPCITEMIDLIST) ((LPBYTE) pidl->mkid.abID
|
|
|
|
+ ((wcslen((LPCWSTR) pidl->mkid.abID) + 1) * sizeof(WCHAR)));
|
|
|
|
}
|
|
|
|
|
|
|
|
LRESULT CFindFolder::AddItem(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL &bHandled)
|
|
|
|
{
|
|
|
|
if (!lParam)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
HRESULT hr;
|
|
|
|
LPWSTR path = (LPWSTR) lParam;
|
|
|
|
|
|
|
|
CComPtr<IShellFolder> pShellFolder;
|
|
|
|
hr = SHGetDesktopFolder(&pShellFolder);
|
|
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
|
|
{
|
|
|
|
LocalFree(path);
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
|
|
|
|
LPITEMIDLIST lpFSPidl;
|
|
|
|
DWORD pchEaten;
|
|
|
|
hr = pShellFolder->ParseDisplayName(NULL, NULL, path, &pchEaten, &lpFSPidl, NULL);
|
|
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
|
|
{
|
|
|
|
LocalFree(path);
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
|
|
|
|
LPITEMIDLIST lpLastFSPidl = ILFindLastID(lpFSPidl);
|
|
|
|
LPITEMIDLIST lpSearchPidl = _ILCreate(path, lpLastFSPidl);
|
|
|
|
ILFree(lpFSPidl);
|
|
|
|
LocalFree(path);
|
|
|
|
if (!lpSearchPidl)
|
|
|
|
{
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
}
|
|
|
|
|
|
|
|
UINT uItemIndex;
|
|
|
|
hr = m_shellFolderView->AddObject(lpSearchPidl, &uItemIndex);
|
|
|
|
ILFree(lpSearchPidl);
|
|
|
|
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
|
2019-07-19 20:23:20 +00:00
|
|
|
// *** IShellFolder2 methods ***
|
|
|
|
STDMETHODIMP CFindFolder::GetDefaultSearchGUID(GUID *pguid)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::EnumSearches(IEnumExtraSearch **ppenum)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetDefaultColumn(DWORD, ULONG *pSort, ULONG *pDisplay)
|
|
|
|
{
|
|
|
|
if (pSort)
|
|
|
|
*pSort = 0;
|
|
|
|
if (pDisplay)
|
|
|
|
*pDisplay = 0;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetDefaultColumnState(UINT iColumn, DWORD *pcsFlags)
|
|
|
|
{
|
2019-07-20 00:02:06 +00:00
|
|
|
if (!pcsFlags)
|
2019-07-19 20:23:20 +00:00
|
|
|
return E_INVALIDARG;
|
2019-07-20 00:02:06 +00:00
|
|
|
if (iColumn >= _countof(g_ColumnDefs))
|
|
|
|
return m_pisfInner->GetDefaultColumnState(iColumn - _countof(g_ColumnDefs) + 1, pcsFlags);
|
2019-07-19 20:23:20 +00:00
|
|
|
*pcsFlags = g_ColumnDefs[iColumn].dwDefaultState;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetDetailsEx(PCUITEMID_CHILD pidl, const SHCOLUMNID *pscid, VARIANT *pv)
|
|
|
|
{
|
2019-07-20 00:02:06 +00:00
|
|
|
return m_pisfInner->GetDetailsEx(pidl, pscid, pv);
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetDetailsOf(PCUITEMID_CHILD pidl, UINT iColumn, SHELLDETAILS *pDetails)
|
|
|
|
{
|
|
|
|
if (iColumn >= _countof(g_ColumnDefs))
|
2019-07-20 00:27:31 +00:00
|
|
|
return m_pisfInner->GetDetailsOf(_ILGetFSPidl(pidl), iColumn - _countof(g_ColumnDefs) + 1, pDetails);
|
2019-07-19 20:23:20 +00:00
|
|
|
|
|
|
|
pDetails->cxChar = g_ColumnDefs[iColumn].cxChar;
|
|
|
|
pDetails->fmt = g_ColumnDefs[iColumn].fmt;
|
|
|
|
|
|
|
|
if (!pidl)
|
|
|
|
return SHSetStrRet(&pDetails->str, g_ColumnDefs[iColumn].wzColumnName);
|
|
|
|
|
2019-07-20 00:27:31 +00:00
|
|
|
if (iColumn == 1)
|
|
|
|
{
|
|
|
|
WCHAR path[MAX_PATH];
|
|
|
|
wcscpy(path, _ILGetPath(pidl));
|
|
|
|
PathRemoveFileSpecW(path);
|
|
|
|
return SHSetStrRet(&pDetails->str, path);
|
|
|
|
}
|
|
|
|
|
2019-07-19 20:23:20 +00:00
|
|
|
return GetDisplayNameOf(pidl, SHGDN_NORMAL, &pDetails->str);
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::MapColumnToSCID(UINT iColumn, SHCOLUMNID *pscid)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
|
|
|
// *** IShellFolder methods ***
|
|
|
|
STDMETHODIMP CFindFolder::ParseDisplayName(HWND hwndOwner, LPBC pbc, LPOLESTR lpszDisplayName, ULONG *pchEaten,
|
|
|
|
PIDLIST_RELATIVE *ppidl, ULONG *pdwAttributes)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::EnumObjects(HWND hwndOwner, DWORD dwFlags, LPENUMIDLIST *ppEnumIDList)
|
|
|
|
{
|
|
|
|
*ppEnumIDList = NULL;
|
|
|
|
return S_FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::BindToObject(PCUIDLIST_RELATIVE pidl, LPBC pbcReserved, REFIID riid, LPVOID *ppvOut)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::BindToStorage(PCUIDLIST_RELATIVE pidl, LPBC pbcReserved, REFIID riid, LPVOID *ppvOut)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::CompareIDs(LPARAM lParam, PCUIDLIST_RELATIVE pidl1, PCUIDLIST_RELATIVE pidl2)
|
|
|
|
{
|
2019-07-20 00:27:31 +00:00
|
|
|
return m_pisfInner->CompareIDs(lParam, _ILGetFSPidl(pidl1), _ILGetFSPidl(pidl2));
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::CreateViewObject(HWND hwndOwner, REFIID riid, LPVOID *ppvOut)
|
|
|
|
{
|
|
|
|
if (riid == IID_IShellView)
|
|
|
|
{
|
|
|
|
SFV_CREATE sfvparams = {};
|
|
|
|
sfvparams.cbSize = sizeof(SFV_CREATE);
|
|
|
|
sfvparams.pshf = this;
|
2019-07-20 00:15:38 +00:00
|
|
|
sfvparams.psfvcb = this;
|
2019-07-20 00:27:31 +00:00
|
|
|
HRESULT hr = SHCreateShellFolderView(&sfvparams, (IShellView **) ppvOut);
|
|
|
|
if (FAILED_UNEXPECTEDLY(hr))
|
|
|
|
{
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
|
|
|
|
return ((IShellView * ) * ppvOut)->QueryInterface(IID_PPV_ARG(IShellFolderView, &m_shellFolderView));
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetAttributesOf(UINT cidl, PCUITEMID_CHILD_ARRAY apidl, DWORD *rgfInOut)
|
|
|
|
{
|
2019-07-20 00:27:31 +00:00
|
|
|
CComHeapPtr<PCITEMID_CHILD> aFSPidl;
|
|
|
|
aFSPidl.Allocate(cidl);
|
|
|
|
for (UINT i = 0; i < cidl; i++)
|
|
|
|
{
|
|
|
|
aFSPidl[i] = _ILGetFSPidl(apidl[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
return m_pisfInner->GetAttributesOf(cidl, aFSPidl, rgfInOut);
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetUIObjectOf(HWND hwndOwner, UINT cidl, PCUITEMID_CHILD_ARRAY apidl, REFIID riid,
|
|
|
|
UINT *prgfInOut, LPVOID *ppvOut)
|
|
|
|
{
|
2019-07-20 00:27:31 +00:00
|
|
|
if (cidl <= 0)
|
|
|
|
{
|
|
|
|
return m_pisfInner->GetUIObjectOf(hwndOwner, cidl, apidl, riid, prgfInOut, ppvOut);
|
|
|
|
}
|
|
|
|
|
|
|
|
PCITEMID_CHILD *aFSPidl = new PCITEMID_CHILD[cidl];
|
|
|
|
for (UINT i = 0; i < cidl; i++)
|
|
|
|
{
|
|
|
|
aFSPidl[i] = _ILGetFSPidl(apidl[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
HRESULT hr = m_pisfInner->GetUIObjectOf(hwndOwner, cidl, aFSPidl, riid, prgfInOut, ppvOut);
|
|
|
|
delete[] aFSPidl;
|
|
|
|
|
|
|
|
return hr;
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::GetDisplayNameOf(PCUITEMID_CHILD pidl, DWORD dwFlags, LPSTRRET pName)
|
|
|
|
{
|
2019-07-20 00:27:31 +00:00
|
|
|
return m_pisfInner->GetDisplayNameOf(_ILGetFSPidl(pidl), dwFlags, pName);
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
STDMETHODIMP CFindFolder::SetNameOf(HWND hwndOwner, PCUITEMID_CHILD pidl, LPCOLESTR lpName, DWORD dwFlags,
|
|
|
|
PITEMID_CHILD *pPidlOut)
|
|
|
|
{
|
|
|
|
UNIMPLEMENTED;
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
2019-07-20 00:15:38 +00:00
|
|
|
|
|
|
|
//// *** IShellFolderViewCB method ***
|
|
|
|
STDMETHODIMP CFindFolder::MessageSFVCB(UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
|
|
{
|
|
|
|
switch (uMsg)
|
|
|
|
{
|
|
|
|
case SFVM_DEFVIEWMODE:
|
|
|
|
{
|
|
|
|
FOLDERVIEWMODE *pViewMode = (FOLDERVIEWMODE *) lParam;
|
|
|
|
*pViewMode = FVM_DETAILS;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return E_NOTIMPL;
|
|
|
|
}
|
|
|
|
|
2019-07-19 20:23:20 +00:00
|
|
|
//// *** IPersistFolder2 methods ***
|
|
|
|
STDMETHODIMP CFindFolder::GetCurFolder(LPITEMIDLIST *pidl)
|
|
|
|
{
|
|
|
|
*pidl = ILClone(m_pidl);
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
// *** IPersistFolder methods ***
|
|
|
|
STDMETHODIMP CFindFolder::Initialize(LPCITEMIDLIST pidl)
|
|
|
|
{
|
|
|
|
m_pidl = ILClone(pidl);
|
|
|
|
if (!m_pidl)
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
|
2019-07-20 00:02:06 +00:00
|
|
|
return SHELL32_CoCreateInitSF(m_pidl,
|
|
|
|
NULL,
|
|
|
|
NULL,
|
|
|
|
&CLSID_ShellFSFolder,
|
|
|
|
IID_PPV_ARG(IShellFolder2, &m_pisfInner));
|
2019-07-19 20:23:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// *** IPersist methods ***
|
|
|
|
STDMETHODIMP CFindFolder::GetClassID(CLSID *pClassId)
|
|
|
|
{
|
|
|
|
if (pClassId == NULL)
|
|
|
|
return E_INVALIDARG;
|
|
|
|
memcpy(pClassId, &CLSID_FindFolder, sizeof(CLSID));
|
|
|
|
return S_OK;
|
|
|
|
}
|