2 * PROJECT: ReactOS Search Shell Extension
3 * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
5 * COPYRIGHT: Copyright 2019 Brock Mammen
8 #include "CSearchBar.h"
9 #include <psdk/wingdi.h>
10 #include <commoncontrols.h>
11 #include <../browseui.h>
15 WINE_DEFAULT_DEBUG_CHANNEL(shellfind
);
20 #define UNIMPLEMENTED DbgPrint("%s is UNIMPLEMENTED!\n", __FUNCTION__)
23 CSearchBar::CSearchBar() :
29 CSearchBar::~CSearchBar()
33 LRESULT
CSearchBar::OnInitDialog(UINT uMsg
, WPARAM wParam
, LPARAM lParam
, BOOL
&bHandled
)
35 SetSearchInProgress(FALSE
);
37 HWND hCombobox
= GetDlgItem(IDC_SEARCH_COMBOBOX
);
38 IImageList
*pImageList
;
39 HRESULT hResult
= SHGetImageList(SHIL_SMALL
, IID_PPV_ARG(IImageList
, &pImageList
));
40 SendMessage(hCombobox
, CBEM_SETIMAGELIST
, 0, FAILED_UNEXPECTEDLY(hResult
) ? 0 : reinterpret_cast<LPARAM
>(pImageList
));
42 SendMessage(hCombobox
, CBEM_SETEXTENDEDSTYLE
,
43 CBES_EX_CASESENSITIVE
| CBES_EX_NOSIZELIMIT
, CBES_EX_CASESENSITIVE
| CBES_EX_NOSIZELIMIT
);
44 HWND hEditControl
= reinterpret_cast<HWND
>(SendMessage(hCombobox
, CBEM_GETEDITCONTROL
, 0, 0));
45 hResult
= CAddressEditBox_CreateInstance(IID_PPV_ARG(IAddressEditBox
, &m_AddressEditBox
));
46 if (FAILED_UNEXPECTEDLY(hResult
))
49 CComPtr
<IShellService
> pShellService
;
50 hResult
= m_AddressEditBox
->QueryInterface(IID_PPV_ARG(IShellService
, &pShellService
));
51 if (FAILED_UNEXPECTEDLY(hResult
))
53 hResult
= m_AddressEditBox
->Init(hCombobox
, hEditControl
, 0, m_pSite
);
54 if (FAILED_UNEXPECTEDLY(hResult
))
57 CComPtr
<IDispatch
> pDispatch
;
58 hResult
= m_AddressEditBox
->QueryInterface(IID_PPV_ARG(IDispatch
, &pDispatch
));
59 if (FAILED_UNEXPECTEDLY(hResult
))
61 DISPPARAMS params
= {0};
62 hResult
= pDispatch
->Invoke(DISPID_NAVIGATECOMPLETE2
, GUID_NULL
, 0, DISPATCH_METHOD
, ¶ms
, NULL
, NULL
, NULL
);
64 hResult
= pShellService
->SetOwner(NULL
);
65 if (FAILED_UNEXPECTEDLY(hResult
))
71 HRESULT
CSearchBar::ExecuteCommand(CComPtr
<IContextMenu
>& menu
, UINT nCmd
)
73 CComPtr
<IOleWindow
> pBrowserOleWnd
;
74 CMINVOKECOMMANDINFO cmi
;
78 hr
= IUnknown_QueryService(m_pSite
, SID_SShellBrowser
, IID_PPV_ARG(IOleWindow
, &pBrowserOleWnd
));
79 if (FAILED_UNEXPECTEDLY(hr
))
82 hr
= pBrowserOleWnd
->GetWindow(&browserWnd
);
83 if (FAILED_UNEXPECTEDLY(hr
))
86 ZeroMemory(&cmi
, sizeof(cmi
));
87 cmi
.cbSize
= sizeof(cmi
);
88 cmi
.lpVerb
= MAKEINTRESOURCEA(nCmd
);
89 cmi
.hwnd
= browserWnd
;
90 if (GetKeyState(VK_SHIFT
) & 0x8000)
91 cmi
.fMask
|= CMIC_MASK_SHIFT_DOWN
;
92 if (GetKeyState(VK_CONTROL
) & 0x8000)
93 cmi
.fMask
|= CMIC_MASK_CONTROL_DOWN
;
95 return menu
->InvokeCommand(&cmi
);
99 // *** ATL event handlers ***
100 LRESULT
CSearchBar::OnSetFocus(UINT uMsg
, WPARAM wParam
, LPARAM lParam
, BOOL
&bHandled
)
102 IUnknown_OnFocusChangeIS(m_pSite
, reinterpret_cast<IUnknown
*>(this), TRUE
);
107 HRESULT
CSearchBar::GetSearchResultsFolder(IShellBrowser
**ppShellBrowser
, HWND
*pHwnd
, IShellFolder
**ppShellFolder
)
110 CComPtr
<IShellBrowser
> pShellBrowser
;
112 ppShellBrowser
= &pShellBrowser
;
113 if (!*ppShellBrowser
)
115 hr
= IUnknown_QueryService(m_pSite
, SID_SShellBrowser
, IID_PPV_ARG(IShellBrowser
, ppShellBrowser
));
116 if (FAILED_UNEXPECTEDLY(hr
))
120 CComPtr
<IShellView
> pShellView
;
121 hr
= (*ppShellBrowser
)->QueryActiveShellView(&pShellView
);
122 if (FAILED(hr
) || !pShellView
)
125 CComPtr
<IFolderView
> pFolderView
;
126 hr
= pShellView
->QueryInterface(IID_PPV_ARG(IFolderView
, &pFolderView
));
127 if (FAILED(hr
) || !pFolderView
)
130 CComPtr
<IShellFolder
> pShellFolder
;
132 ppShellFolder
= &pShellFolder
;
133 hr
= pFolderView
->GetFolder(IID_PPV_ARG(IShellFolder
, ppShellFolder
));
134 if (FAILED(hr
) || !pShellFolder
)
138 hr
= IUnknown_GetClassID(*ppShellFolder
, &clsid
);
141 if (clsid
!= CLSID_FindFolder
)
146 hr
= pShellView
->GetWindow(pHwnd
);
147 if (FAILED_UNEXPECTEDLY(hr
))
154 LRESULT
CSearchBar::OnSearchButtonClicked(WORD wNotifyCode
, WORD wID
, HWND hWndCtl
, BOOL
& bHandled
)
156 CComHeapPtr
<SearchStart
> pSearchStart((SearchStart
*)SHAlloc(sizeof(SearchStart
)));
157 GetDlgItemText(IDC_SEARCH_FILENAME
, pSearchStart
->szFileName
, _countof(pSearchStart
->szFileName
));
158 GetDlgItemText(IDC_SEARCH_QUERY
, pSearchStart
->szQuery
, _countof(pSearchStart
->szQuery
));
159 if (!GetAddressEditBoxPath(pSearchStart
->szPath
))
161 ShellMessageBoxW(_AtlBaseModule
.GetResourceInstance(), m_hWnd
, MAKEINTRESOURCEW(IDS_SEARCHINVALID
), MAKEINTRESOURCEW(IDS_SEARCHLABEL
), MB_OK
| MB_ICONERROR
, pSearchStart
->szPath
);
165 CComPtr
<IShellBrowser
> pShellBrowser
;
166 HRESULT hr
= IUnknown_QueryService(m_pSite
, SID_SShellBrowser
, IID_PPV_ARG(IShellBrowser
, &pShellBrowser
));
167 if (FAILED_UNEXPECTEDLY(hr
))
171 if (FAILED(GetSearchResultsFolder(&pShellBrowser
, &hwnd
, NULL
)))
173 // Open a new search results folder
174 WCHAR szShellGuid
[MAX_PATH
];
175 const WCHAR shellGuidPrefix
[] = L
"shell:::";
176 memcpy(szShellGuid
, shellGuidPrefix
, sizeof(shellGuidPrefix
));
177 hr
= StringFromGUID2(CLSID_FindFolder
, szShellGuid
+ _countof(shellGuidPrefix
) - 1,
178 _countof(szShellGuid
) - _countof(shellGuidPrefix
));
179 if (FAILED_UNEXPECTEDLY(hr
))
182 CComHeapPtr
<ITEMIDLIST
> findFolderPidl
;
183 hr
= SHParseDisplayName(szShellGuid
, NULL
, &findFolderPidl
, 0, NULL
);
184 if (FAILED_UNEXPECTEDLY(hr
))
187 hr
= pShellBrowser
->BrowseObject(findFolderPidl
, 0);
188 if (FAILED_UNEXPECTEDLY(hr
))
191 CComPtr
<IShellFolder
> pShellFolder
;
192 hr
= GetSearchResultsFolder(*pShellBrowser
, &hwnd
, &pShellFolder
);
193 if (FAILED_UNEXPECTEDLY(hr
))
196 // Subscribe to search events
198 hr
= AtlAdvise(pShellFolder
, static_cast<IDispatch
*>(this), DIID_DSearchCommandEvents
, &fAdviseCookie
);
199 if (FAILED_UNEXPECTEDLY(hr
))
203 ::PostMessageW(hwnd
, WM_SEARCH_START
, 0, (LPARAM
) pSearchStart
.Detach());
205 SetSearchInProgress(TRUE
);
210 LRESULT
CSearchBar::OnStopButtonClicked(WORD wNotifyCode
, WORD wID
, HWND hWndCtl
, BOOL
& bHandled
)
213 HRESULT hr
= GetSearchResultsFolder(NULL
, &hwnd
, NULL
);
215 ::PostMessageW(hwnd
, WM_SEARCH_STOP
, 0, 0);
220 BOOL
CSearchBar::GetAddressEditBoxPath(WCHAR (&szPath
)[MAX_PATH
])
222 HWND hComboboxEx
= GetDlgItem(IDC_SEARCH_COMBOBOX
);
223 ::GetWindowTextW(hComboboxEx
, szPath
, _countof(szPath
));
224 INT iSelectedIndex
= SendMessageW(hComboboxEx
, CB_GETCURSEL
, 0, 0);
225 if (iSelectedIndex
!= CB_ERR
)
227 WCHAR szItemText
[MAX_PATH
];
228 COMBOBOXEXITEMW item
= {0};
229 item
.mask
= CBEIF_LPARAM
| CBEIF_TEXT
;
230 item
.iItem
= iSelectedIndex
;
231 item
.pszText
= szItemText
;
232 item
.cchTextMax
= _countof(szItemText
);
233 SendMessageW(hComboboxEx
, CBEM_GETITEMW
, 0, (LPARAM
)&item
);
235 if (!wcscmp(szItemText
, szPath
) && SHGetPathFromIDListW((LPCITEMIDLIST
)item
.lParam
, szPath
))
241 DWORD dwAttributes
= GetFileAttributesW(szPath
);
242 return dwAttributes
!= INVALID_FILE_ATTRIBUTES
243 && (dwAttributes
& FILE_ATTRIBUTE_DIRECTORY
);
246 LRESULT
CSearchBar::OnSize(UINT uMsg
, WPARAM wParam
, LPARAM lParam
, BOOL
&bHandled
)
248 INT iWidth
= LOWORD(lParam
);
251 ((CWindow
)GetDlgItem(IDC_SEARCH_LABEL
)).SetWindowPos(NULL
, 0, 0, iWidth
- iPadding
, 40, SWP_NOACTIVATE
| SWP_NOCOPYBITS
| SWP_NOMOVE
| SWP_NOOWNERZORDER
| SWP_NOZORDER
);
253 int inputs
[] = { IDC_SEARCH_FILENAME
, IDC_SEARCH_QUERY
, IDC_SEARCH_COMBOBOX
, IDC_SEARCH_BUTTON
, IDC_SEARCH_STOP_BUTTON
, IDC_PROGRESS_BAR
};
254 HDWP hdwp
= BeginDeferWindowPos(_countof(inputs
));
255 for (SIZE_T i
= 0; i
< _countof(inputs
); i
++)
257 CWindow wnd
= (CWindow
) GetDlgItem(inputs
[i
]);
259 wnd
.GetWindowRect(&rect
);
260 POINT pt
= { rect
.left
, rect
.top
};
262 hdwp
= wnd
.DeferWindowPos(hdwp
,
266 iWidth
- iPadding
* 2,
267 rect
.bottom
- rect
.top
,
268 SWP_NOZORDER
| SWP_NOACTIVATE
);
270 EndDeferWindowPos(hdwp
);
276 // *** IOleWindow methods ***
277 HRESULT STDMETHODCALLTYPE
CSearchBar::GetWindow(HWND
*lphwnd
)
285 HRESULT STDMETHODCALLTYPE
CSearchBar::ContextSensitiveHelp(BOOL fEnterMode
)
292 // *** IDockingWindow methods ***
293 HRESULT STDMETHODCALLTYPE
CSearchBar::CloseDW(DWORD dwReserved
)
295 // We do nothing, we don't have anything to save yet
296 TRACE("CloseDW called\n");
300 HRESULT STDMETHODCALLTYPE
CSearchBar::ResizeBorderDW(const RECT
*prcBorder
, IUnknown
*punkToolbarSite
, BOOL fReserved
)
302 /* Must return E_NOTIMPL according to MSDN */
306 HRESULT STDMETHODCALLTYPE
CSearchBar::ShowDW(BOOL fShow
)
314 // *** IDeskBand methods ***
315 HRESULT STDMETHODCALLTYPE
CSearchBar::GetBandInfo(DWORD dwBandID
, DWORD dwViewMode
, DESKBANDINFO
*pdbi
)
322 if (pdbi
->dwMask
& DBIM_MINSIZE
)
324 pdbi
->ptMinSize
.x
= 200;
325 pdbi
->ptMinSize
.y
= 30;
328 if (pdbi
->dwMask
& DBIM_MAXSIZE
)
330 pdbi
->ptMaxSize
.y
= -1;
333 if (pdbi
->dwMask
& DBIM_INTEGRAL
)
335 pdbi
->ptIntegral
.y
= 1;
338 if (pdbi
->dwMask
& DBIM_ACTUAL
)
340 pdbi
->ptActual
.x
= 200;
341 pdbi
->ptActual
.y
= 30;
344 if (pdbi
->dwMask
& DBIM_TITLE
)
346 if (!LoadStringW(_AtlBaseModule
.GetResourceInstance(), IDS_SEARCHLABEL
, pdbi
->wszTitle
, _countof(pdbi
->wszTitle
)))
347 return HRESULT_FROM_WIN32(GetLastError());
350 if (pdbi
->dwMask
& DBIM_MODEFLAGS
)
352 pdbi
->dwModeFlags
= DBIMF_NORMAL
| DBIMF_VARIABLEHEIGHT
;
355 if (pdbi
->dwMask
& DBIM_BKCOLOR
)
357 pdbi
->dwMask
&= ~DBIM_BKCOLOR
;
363 // *** IObjectWithSite methods ***
364 HRESULT STDMETHODCALLTYPE
CSearchBar::SetSite(IUnknown
*pUnkSite
)
369 if (pUnkSite
== m_pSite
)
372 TRACE("SetSite called \n");
379 if (pUnkSite
!= m_pSite
)
387 hr
= IUnknown_GetWindow(pUnkSite
, &parentWnd
);
390 ERR("Could not get parent's window ! Status: %08lx\n", hr
);
399 SetParent(parentWnd
);
403 CDialogImpl::Create(parentWnd
);
409 HRESULT STDMETHODCALLTYPE
CSearchBar::GetSite(REFIID riid
, void **ppvSite
)
418 // *** IOleCommandTarget methods ***
419 HRESULT STDMETHODCALLTYPE
CSearchBar::QueryStatus(const GUID
*pguidCmdGroup
, ULONG cCmds
, OLECMD prgCmds
[], OLECMDTEXT
*pCmdText
)
425 HRESULT STDMETHODCALLTYPE
CSearchBar::Exec(const GUID
*pguidCmdGroup
, DWORD nCmdID
, DWORD nCmdexecopt
, VARIANT
*pvaIn
, VARIANT
*pvaOut
)
432 // *** IServiceProvider methods ***
433 HRESULT STDMETHODCALLTYPE
CSearchBar::QueryService(REFGUID guidService
, REFIID riid
, void **ppvObject
)
435 /* FIXME: we probably want to handle more services here */
436 return IUnknown_QueryService(m_pSite
, SID_SShellBrowser
, riid
, ppvObject
);
440 // *** IInputObject methods ***
441 HRESULT STDMETHODCALLTYPE
CSearchBar::UIActivateIO(BOOL fActivate
, LPMSG lpMsg
)
448 // TODO: handle message
451 TranslateMessage(lpMsg
);
452 DispatchMessage(lpMsg
);
457 HRESULT STDMETHODCALLTYPE
CSearchBar::HasFocusIO()
462 HRESULT STDMETHODCALLTYPE
CSearchBar::TranslateAcceleratorIO(LPMSG lpMsg
)
464 if (IsDialogMessage(lpMsg
))
467 if ((lpMsg
->hwnd
== m_hWnd
|| IsChild(lpMsg
->hwnd
)))
469 TranslateMessage(lpMsg
);
470 DispatchMessage(lpMsg
);
477 // *** IPersist methods ***
478 HRESULT STDMETHODCALLTYPE
CSearchBar::GetClassID(CLSID
*pClassID
)
482 *pClassID
= CLSID_FileSearchBand
;
487 // *** IPersistStream methods ***
488 HRESULT STDMETHODCALLTYPE
CSearchBar::IsDirty()
494 HRESULT STDMETHODCALLTYPE
CSearchBar::Load(IStream
*pStm
)
500 HRESULT STDMETHODCALLTYPE
CSearchBar::Save(IStream
*pStm
, BOOL fClearDirty
)
506 HRESULT STDMETHODCALLTYPE
CSearchBar::GetSizeMax(ULARGE_INTEGER
*pcbSize
)
508 // TODO: calculate max size
514 // *** IWinEventHandler methods ***
515 HRESULT STDMETHODCALLTYPE
CSearchBar::OnWinEvent(HWND hWnd
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
, LRESULT
*theResult
)
520 HRESULT STDMETHODCALLTYPE
CSearchBar::IsWindowOwner(HWND hWnd
)
522 return (hWnd
== m_hWnd
) ? S_OK
: S_FALSE
;
525 // *** IBandNavigate methods ***
526 HRESULT STDMETHODCALLTYPE
CSearchBar::Select(long paramC
)
532 // *** INamespaceProxy ***
533 HRESULT STDMETHODCALLTYPE
CSearchBar::GetNavigateTarget(long paramC
, long param10
, long param14
)
539 HRESULT STDMETHODCALLTYPE
CSearchBar::Invoke(long paramC
)
545 HRESULT STDMETHODCALLTYPE
CSearchBar::OnSelectionChanged(long paramC
)
551 HRESULT STDMETHODCALLTYPE
CSearchBar::RefreshFlags(long paramC
, long param10
, long param14
)
557 HRESULT STDMETHODCALLTYPE
CSearchBar::CacheItem(long paramC
)
563 // *** IDispatch methods ***
564 HRESULT STDMETHODCALLTYPE
CSearchBar::GetTypeInfoCount(UINT
*pctinfo
)
570 HRESULT STDMETHODCALLTYPE
CSearchBar::GetTypeInfo(UINT iTInfo
, LCID lcid
, ITypeInfo
**ppTInfo
)
576 HRESULT STDMETHODCALLTYPE
CSearchBar::GetIDsOfNames(REFIID riid
, LPOLESTR
*rgszNames
, UINT cNames
, LCID lcid
, DISPID
*rgDispId
)
582 void CSearchBar::SetSearchInProgress(BOOL bInProgress
)
584 ::ShowWindow(GetDlgItem(IDC_SEARCH_BUTTON
), bInProgress
? SW_HIDE
: SW_SHOW
);
585 ::ShowWindow(GetDlgItem(IDC_SEARCH_STOP_BUTTON
), bInProgress
? SW_SHOW
: SW_HIDE
);
586 HWND hProgressBar
= GetDlgItem(IDC_PROGRESS_BAR
);
587 ::ShowWindow(hProgressBar
, bInProgress
? SW_SHOW
: SW_HIDE
);
588 ::PostMessage(hProgressBar
, PBM_SETMARQUEE
, bInProgress
, 0);
591 HRESULT STDMETHODCALLTYPE
CSearchBar::Invoke(DISPID dispIdMember
, REFIID riid
, LCID lcid
, WORD wFlags
, DISPPARAMS
*pDispParams
, VARIANT
*pVarResult
, EXCEPINFO
*pExcepInfo
, UINT
*puArgErr
)
593 if (dispIdMember
== DISPID_SEARCHCOMPLETE
|| dispIdMember
== DISPID_SEARCHABORT
)
595 SetSearchInProgress(FALSE
);