[SHELLFIND] Create PIDLs and add search results to folder

[reactos.git] / dll / win32 / browseui / shellfind / CFindFolder.cpp

diff --git a/dll/win32/browseui/shellfind/CFindFolder.cpp b/dll/win32/browseui/shellfind/CFindFolder.cpp
index c196187..7983ee7 100644 (file)
--- a/dll/win32/browseui/shellfind/CFindFolder.cpp
+++ b/dll/win32/browseui/shellfind/CFindFolder.cpp
@@ -24,6 +24,85 @@ static FolderViewColumns g_ColumnDefs[] =
     {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)
 {
@@ -48,22 +127,23 @@ STDMETHODIMP CFindFolder::GetDefaultColumn(DWORD, ULONG *pSort, ULONG *pDisplay)
 
 STDMETHODIMP CFindFolder::GetDefaultColumnState(UINT iColumn, DWORD *pcsFlags)
 {
-    if (!pcsFlags || iColumn >= _countof(g_ColumnDefs))
+    if (!pcsFlags)
         return E_INVALIDARG;
+    if (iColumn >= _countof(g_ColumnDefs))
+        return m_pisfInner->GetDefaultColumnState(iColumn - _countof(g_ColumnDefs) + 1, pcsFlags);
     *pcsFlags = g_ColumnDefs[iColumn].dwDefaultState;
     return S_OK;
 }
 
 STDMETHODIMP CFindFolder::GetDetailsEx(PCUITEMID_CHILD pidl, const SHCOLUMNID *pscid, VARIANT *pv)
 {
-    UNIMPLEMENTED;
-    return E_NOTIMPL;
+    return m_pisfInner->GetDetailsEx(pidl, pscid, pv);
 }
 
 STDMETHODIMP CFindFolder::GetDetailsOf(PCUITEMID_CHILD pidl, UINT iColumn, SHELLDETAILS *pDetails)
 {
     if (iColumn >= _countof(g_ColumnDefs))
-        return E_FAIL;
+        return m_pisfInner->GetDetailsOf(_ILGetFSPidl(pidl), iColumn - _countof(g_ColumnDefs) + 1, pDetails);
 
     pDetails->cxChar = g_ColumnDefs[iColumn].cxChar;
     pDetails->fmt = g_ColumnDefs[iColumn].fmt;
@@ -71,6 +151,14 @@ STDMETHODIMP CFindFolder::GetDetailsOf(PCUITEMID_CHILD pidl, UINT iColumn, SHELL
     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);
 }
 
@@ -108,8 +196,7 @@ STDMETHODIMP CFindFolder::BindToStorage(PCUIDLIST_RELATIVE pidl, LPBC pbcReserve
 
 STDMETHODIMP CFindFolder::CompareIDs(LPARAM lParam, PCUIDLIST_RELATIVE pidl1, PCUIDLIST_RELATIVE pidl2)
 {
-    UNIMPLEMENTED;
-    return E_NOTIMPL;
+    return m_pisfInner->CompareIDs(lParam, _ILGetFSPidl(pidl1), _ILGetFSPidl(pidl2));
 }
 
 STDMETHODIMP CFindFolder::CreateViewObject(HWND hwndOwner, REFIID riid, LPVOID *ppvOut)
@@ -119,27 +206,53 @@ STDMETHODIMP CFindFolder::CreateViewObject(HWND hwndOwner, REFIID riid, LPVOID *
         SFV_CREATE sfvparams = {};
         sfvparams.cbSize = sizeof(SFV_CREATE);
         sfvparams.pshf = this;
+        sfvparams.psfvcb = this;
         HRESULT hr = SHCreateShellFolderView(&sfvparams, (IShellView **) ppvOut);
-        return hr;
+        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)
 {
-    *rgfInOut = SFGAO_NONENUMERATED;
-    return S_OK;
+    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 E_NOINTERFACE;
+    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 SHSetStrRet(pName, "search result");
+    return m_pisfInner->GetDisplayNameOf(_ILGetFSPidl(pidl), dwFlags, pName);
 }
 
 STDMETHODIMP CFindFolder::SetNameOf(HWND hwndOwner, PCUITEMID_CHILD pidl, LPCOLESTR lpName, DWORD dwFlags,
@@ -148,6 +261,22 @@ STDMETHODIMP CFindFolder::SetNameOf(HWND hwndOwner, PCUITEMID_CHILD pidl, LPCOLE
     UNIMPLEMENTED;
     return E_NOTIMPL;
 }
+
+//// *** 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;
+}
+
 //// *** IPersistFolder2 methods ***
 STDMETHODIMP CFindFolder::GetCurFolder(LPITEMIDLIST *pidl)
 {
@@ -162,7 +291,11 @@ STDMETHODIMP CFindFolder::Initialize(LPCITEMIDLIST pidl)
     if (!m_pidl)
         return E_OUTOFMEMORY;
 
-    return S_OK;
+    return SHELL32_CoCreateInitSF(m_pidl,
+                                  NULL,
+                                  NULL,
+                                  &CLSID_ShellFSFolder,
+                                  IID_PPV_ARG(IShellFolder2, &m_pisfInner));
 }
 
 // *** IPersist methods ***