mirror of
https://github.com/reactos/reactos.git
synced 2024-11-05 22:26:39 +00:00
a938d19714
Reduce code. And correctly do the filename check. CORE-18909
111 lines
3.8 KiB
C++
111 lines
3.8 KiB
C++
/*
|
|
* PROJECT: ReactOS shell32
|
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
|
* PURPOSE: CopyTo and MoveTo implementation
|
|
* COPYRIGHT: Copyright 2020-2023 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
|
|
*/
|
|
#pragma once
|
|
|
|
class CCopyMoveToMenu :
|
|
public CComObjectRootEx<CComMultiThreadModelNoCS>,
|
|
public IContextMenu2,
|
|
public IObjectWithSite,
|
|
public IShellExtInit
|
|
{
|
|
protected:
|
|
UINT m_idCmdFirst, m_idCmdLast, m_idCmdAction;
|
|
CComPtr<IDataObject> m_pDataObject;
|
|
CComPtr<IUnknown> m_pSite;
|
|
|
|
HRESULT DoRealFileOp(LPCMINVOKECOMMANDINFO lpici, PCUIDLIST_ABSOLUTE pidl);
|
|
HRESULT DoAction(LPCMINVOKECOMMANDINFO lpici);
|
|
|
|
public:
|
|
CComHeapPtr<ITEMIDLIST> m_pidlFolder;
|
|
WNDPROC m_fnOldWndProc;
|
|
BOOL m_bIgnoreTextBoxChange;
|
|
|
|
CCopyMoveToMenu();
|
|
|
|
virtual UINT GetCaptionStringID() const = 0;
|
|
virtual UINT GetButtonStringID() const = 0;
|
|
virtual UINT GetActionTitleStringID() const = 0;
|
|
virtual UINT GetFileOp() const = 0;
|
|
virtual LPCSTR GetVerb() const = 0;
|
|
|
|
// IContextMenu
|
|
STDMETHODIMP GetCommandString(UINT_PTR idCommand, UINT uFlags, UINT *lpReserved, LPSTR lpszName, UINT uMaxNameLen) override;
|
|
STDMETHODIMP InvokeCommand(LPCMINVOKECOMMANDINFO lpcmi) override;
|
|
|
|
// IContextMenu2
|
|
STDMETHODIMP HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam) override;
|
|
|
|
// IShellExtInit
|
|
STDMETHODIMP Initialize(PCIDLIST_ABSOLUTE pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID) override;
|
|
|
|
// IObjectWithSite
|
|
STDMETHODIMP SetSite(IUnknown *pUnkSite) override;
|
|
STDMETHODIMP GetSite(REFIID riid, void **ppvSite) override;
|
|
};
|
|
|
|
class CCopyToMenu
|
|
: public CComCoClass<CCopyToMenu, &CLSID_CopyToMenu>
|
|
, public CCopyMoveToMenu
|
|
{
|
|
public:
|
|
CComHeapPtr<ITEMIDLIST> m_pidlFolder;
|
|
WNDPROC m_fnOldWndProc;
|
|
BOOL m_bIgnoreTextBoxChange;
|
|
|
|
CCopyToMenu() { }
|
|
|
|
// IContextMenu
|
|
STDMETHODIMP QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags) override;
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_COPYTOMENU)
|
|
DECLARE_NOT_AGGREGATABLE(CCopyToMenu)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CCopyToMenu)
|
|
COM_INTERFACE_ENTRY_IID(IID_IContextMenu2, IContextMenu2)
|
|
COM_INTERFACE_ENTRY_IID(IID_IContextMenu, IContextMenu)
|
|
COM_INTERFACE_ENTRY_IID(IID_IShellExtInit, IShellExtInit)
|
|
COM_INTERFACE_ENTRY_IID(IID_IObjectWithSite, IObjectWithSite)
|
|
END_COM_MAP()
|
|
|
|
UINT GetCaptionStringID() const override { return IDS_COPYITEMS; }
|
|
UINT GetButtonStringID() const override { return IDS_COPYBUTTON; }
|
|
UINT GetActionTitleStringID() const override { return IDS_COPYTOTITLE; }
|
|
UINT GetFileOp() const override { return FO_COPY; }
|
|
LPCSTR GetVerb() const override { return "copyto"; }
|
|
};
|
|
|
|
class CMoveToMenu
|
|
: public CComCoClass<CMoveToMenu, &CLSID_MoveToMenu>
|
|
, public CCopyMoveToMenu
|
|
{
|
|
public:
|
|
CMoveToMenu() { }
|
|
|
|
// IContextMenu
|
|
STDMETHODIMP QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags) override;
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_MOVETOMENU)
|
|
DECLARE_NOT_AGGREGATABLE(CMoveToMenu)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CMoveToMenu)
|
|
COM_INTERFACE_ENTRY_IID(IID_IContextMenu2, IContextMenu2)
|
|
COM_INTERFACE_ENTRY_IID(IID_IContextMenu, IContextMenu)
|
|
COM_INTERFACE_ENTRY_IID(IID_IShellExtInit, IShellExtInit)
|
|
COM_INTERFACE_ENTRY_IID(IID_IObjectWithSite, IObjectWithSite)
|
|
END_COM_MAP()
|
|
|
|
UINT GetCaptionStringID() const override { return IDS_MOVEITEMS; }
|
|
UINT GetButtonStringID() const override { return IDS_MOVEBUTTON; }
|
|
UINT GetActionTitleStringID() const override { return IDS_MOVETOTITLE; }
|
|
UINT GetFileOp() const override { return FO_MOVE; }
|
|
LPCSTR GetVerb() const override { return "moveto"; }
|
|
};
|