{L"Relevance", SHCOLSTATE_TYPE_STR, LVCFMT_LEFT, 0}
};
+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;
+}
+
// *** IShellFolder2 methods ***
STDMETHODIMP CFindFolder::GetDefaultSearchGUID(GUID *pguid)
{
STDMETHODIMP CFindFolder::GetDetailsOf(PCUITEMID_CHILD pidl, UINT iColumn, SHELLDETAILS *pDetails)
{
if (iColumn >= _countof(g_ColumnDefs))
- return m_pisfInner->GetDetailsOf(pidl, iColumn - _countof(g_ColumnDefs) + 1, pDetails);
+ return m_pisfInner->GetDetailsOf(_ILGetFSPidl(pidl), iColumn - _countof(g_ColumnDefs) + 1, pDetails);
pDetails->cxChar = g_ColumnDefs[iColumn].cxChar;
pDetails->fmt = g_ColumnDefs[iColumn].fmt;
if (!pidl)
return SHSetStrRet(&pDetails->str, g_ColumnDefs[iColumn].wzColumnName);
+ if (iColumn == 1)
+ {
+ WCHAR path[MAX_PATH];
+ wcscpy(path, _ILGetPath(pidl));
+ PathRemoveFileSpecW(path);
+ return SHSetStrRet(&pDetails->str, path);
+ }
+
return GetDisplayNameOf(pidl, SHGDN_NORMAL, &pDetails->str);
}
STDMETHODIMP CFindFolder::CompareIDs(LPARAM lParam, PCUIDLIST_RELATIVE pidl1, PCUIDLIST_RELATIVE pidl2)
{
- return m_pisfInner->CompareIDs(lParam, pidl1, pidl2);
+ return m_pisfInner->CompareIDs(lParam, _ILGetFSPidl(pidl1), _ILGetFSPidl(pidl2));
}
STDMETHODIMP CFindFolder::CreateViewObject(HWND hwndOwner, REFIID riid, LPVOID *ppvOut)
sfvparams.cbSize = sizeof(SFV_CREATE);
sfvparams.pshf = this;
sfvparams.psfvcb = this;
- return SHCreateShellFolderView(&sfvparams, (IShellView **) ppvOut);
+ HRESULT hr = SHCreateShellFolderView(&sfvparams, (IShellView **) ppvOut);
+ if (FAILED_UNEXPECTEDLY(hr))
+ {
+ return hr;
+ }
+
+ return ((IShellView * ) * ppvOut)->QueryInterface(IID_PPV_ARG(IShellFolderView, &m_shellFolderView));
}
return E_NOINTERFACE;
}
STDMETHODIMP CFindFolder::GetAttributesOf(UINT cidl, PCUITEMID_CHILD_ARRAY apidl, DWORD *rgfInOut)
{
- return m_pisfInner->GetAttributesOf(cidl, apidl, rgfInOut);
+ 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);
}
STDMETHODIMP CFindFolder::GetUIObjectOf(HWND hwndOwner, UINT cidl, PCUITEMID_CHILD_ARRAY apidl, REFIID riid,
UINT *prgfInOut, LPVOID *ppvOut)
{
- return m_pisfInner->GetUIObjectOf(hwndOwner, cidl, apidl, riid, prgfInOut, ppvOut);
+ 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;
}
STDMETHODIMP CFindFolder::GetDisplayNameOf(PCUITEMID_CHILD pidl, DWORD dwFlags, LPSTRRET pName)
{
- return m_pisfInner->GetDisplayNameOf(pidl, dwFlags, pName);
+ return m_pisfInner->GetDisplayNameOf(_ILGetFSPidl(pidl), dwFlags, pName);
}
STDMETHODIMP CFindFolder::SetNameOf(HWND hwndOwner, PCUITEMID_CHILD pidl, LPCOLESTR lpName, DWORD dwFlags,