CCopyToMoveToMenu.h

Go to the documentation of this file.

/*
 * PROJECT:     shell32
 * LICENSE:     GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
 * PURPOSE:     CopyTo and MoveTo implementation
 * COPYRIGHT:   Copyright 2020 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
 */
#pragma once
 
 
class CCopyToMenu :
    public CComCoClass<CCopyToMenu, &CLSID_CopyToMenu>,
    public CComObjectRootEx<CComMultiThreadModelNoCS>,
    public IContextMenu2,
    public IObjectWithSite,
    public IShellExtInit
{
protected:
    UINT m_idCmdFirst, m_idCmdLast, m_idCmdCopyTo;
    CComPtr<IDataObject> m_pDataObject;
    CComPtr<IUnknown> m_pSite;
 
    HRESULT DoCopyToFolder(LPCMINVOKECOMMANDINFO lpici);
    HRESULT DoRealCopy(LPCMINVOKECOMMANDINFO lpici, PCUIDLIST_ABSOLUTE pidl);
    CStringW DoGetFileTitle();
 
public:
    CComHeapPtr<ITEMIDLIST> m_pidlFolder;
    WNDPROC m_fnOldWndProc;
    BOOL m_bIgnoreTextBoxChange;
 
    CCopyToMenu();
    ~CCopyToMenu();
 
    // IContextMenu
    virtual HRESULT WINAPI QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags);
    virtual HRESULT WINAPI InvokeCommand(LPCMINVOKECOMMANDINFO lpcmi);
    virtual HRESULT WINAPI GetCommandString(UINT_PTR idCommand, UINT uFlags, UINT *lpReserved, LPSTR lpszName, UINT uMaxNameLen);
 
    // IContextMenu2
    virtual HRESULT WINAPI HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam);
 
    // IShellExtInit
    virtual HRESULT WINAPI Initialize(PCIDLIST_ABSOLUTE pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID);
 
    // IObjectWithSite
    virtual HRESULT WINAPI SetSite(IUnknown *pUnkSite);
    virtual HRESULT WINAPI GetSite(REFIID riid, void **ppvSite);
 
    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()
};
 
class CMoveToMenu :
    public CComCoClass<CMoveToMenu, &CLSID_MoveToMenu>,
    public CComObjectRootEx<CComMultiThreadModelNoCS>,
    public IContextMenu2,
    public IObjectWithSite,
    public IShellExtInit
{
protected:
    UINT m_idCmdFirst, m_idCmdLast, m_idCmdMoveTo;
    CComPtr<IDataObject> m_pDataObject;
    CComPtr<IUnknown> m_pSite;
 
    HRESULT DoMoveToFolder(LPCMINVOKECOMMANDINFO lpici);
    HRESULT DoRealMove(LPCMINVOKECOMMANDINFO lpici, PCUIDLIST_ABSOLUTE pidl);
    CStringW DoGetFileTitle();
 
public:
    CComHeapPtr<ITEMIDLIST> m_pidlFolder;
    WNDPROC m_fnOldWndProc;
    BOOL m_bIgnoreTextBoxChange;
 
    CMoveToMenu();
    ~CMoveToMenu();
 
    // IContextMenu
    virtual HRESULT WINAPI QueryContextMenu(HMENU hMenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags);
    virtual HRESULT WINAPI InvokeCommand(LPCMINVOKECOMMANDINFO lpcmi);
    virtual HRESULT WINAPI GetCommandString(UINT_PTR idCommand, UINT uFlags, UINT *lpReserved, LPSTR lpszName, UINT uMaxNameLen);
 
    // IContextMenu2
    virtual HRESULT WINAPI HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam);
 
    // IShellExtInit
    virtual HRESULT WINAPI Initialize(PCIDLIST_ABSOLUTE pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID);
 
    // IObjectWithSite
    virtual HRESULT WINAPI SetSite(IUnknown *pUnkSite);
    virtual HRESULT WINAPI GetSite(REFIID riid, void **ppvSite);
 
    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()
};