Method Initialize

PathCopyCopy/src/PathCopyCopyContextMenuExt.cpp:221–281

IShellExtInit::Initialize Called by the shell to initialize our contextual menu extension. We need to use this opportunity to look at the files/folders to act upon. @param p_pFolderPIDL Pointer to ITEMIDLIST representing selected folder. @param p_pDataObject Pointer to data object containing information about selected files/folders. @param p_hKeyFileClass Handle to file class key; unused. @retur

// @return S_OK if successful, otherwise an error code.
//
[[gsl::suppress(c.128)]]
STDMETHODIMP CPathCopyCopyContextMenuExt::Initialize(
    PCIDLIST_ABSOLUTE p_pFolderPIDL,
    IDataObject *p_pDataObject,
    HKEY /*p_hKeyFileClass*/)
{
    HRESULT hRes = S_OK;

    try {
        // Make sure we have a data object.
        if (p_pDataObject != nullptr) {
            // Extract HDROP from data object.
            StStgMedium stgMedium;
            FORMATETC formatEtc = {CF_HDROP, nullptr, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
            if (SUCCEEDED(p_pDataObject->GetData(&formatEtc, &stgMedium))) {
                // Get number of files included in the selection.
                const UINT fileCount = ::DragQueryFileW(
                    static_cast<HDROP>(stgMedium.Get().hGlobal), 0xFFFFFFFF, 0, 0);
                if (fileCount > 0) {
                    // Pre-allocate space in vector to store files.
                    m_vFiles.reserve(fileCount);

                    // Get each file in turn.
                    for (UINT i = 0; i < fileCount; ++i) {
                        const auto bufferSize = ::DragQueryFileW(static_cast<HDROP>(stgMedium.Get().hGlobal), i, nullptr, 0);
                        std::wstring buffer(bufferSize + 1, L'\0');
                        ::DragQueryFileW(static_cast<HDROP>(stgMedium.Get().hGlobal), i, &*buffer.begin(), gsl::narrow<UINT>(buffer.size()));
                        m_vFiles.emplace_back(buffer.c_str());
                    }
                } else {
                    // It's difficult to display a menu item without files to act upon.
                    hRes = E_FAIL;
                }
            }
        } else if (p_pFolderPIDL != nullptr) {
            // No data object, but maybe it's because user clicked on a folder's
            // background. Get the folder path from the ID list.
            std::wstring buffer(MAX_PATH + 1, L'\0');
            if (::SHGetPathFromIDList(p_pFolderPIDL, &*buffer.begin()) != FALSE) {
                m_vFiles.emplace_back(buffer.c_str());
            } else {
                // Nothing like that either, problem.
                hRes = E_FAIL;
            }
        } else {
            hRes = E_POINTER;
        }
    } catch (...) {
        hRes = E_UNEXPECTED;
    }

    if (SUCCEEDED(hRes)) {
        // Check if files and/or folders are selected.
        m_FilesSelected = std::any_of(m_vFiles.cbegin(), m_vFiles.cend(),
                                      [](const auto& path) { return !PCC::PluginUtils::IsDirectory(path); });
        m_FoldersSelected = std::any_of(m_vFiles.cbegin(), m_vFiles.cend(),
                                        [](const auto& path) { return PCC::PluginUtils::IsDirectory(path); });
    }