ShellCOMUtils.hpp

浏览该文件的文档.

 
 
#pragma once
 
#if defined(_WIN32) && !defined(SWIG)
 
#include "AstGlobal.h"
 
#include <Windows.h>
#include <shellapi.h>
#include <shlobj.h>
#include <string>
 
AST_NAMESPACE_BEGIN
 
inline void aShellPumpMessages(unsigned long waitMs)
{
    unsigned long start = GetTickCount();
 
    while ((GetTickCount() - start) < waitMs)
    {
        MSG msg;
        while (PeekMessageW(&msg, nullptr, 0, 0, PM_REMOVE))
        {
            TranslateMessage(&msg);
            DispatchMessageW(&msg);
        }
        unsigned long elapsed = GetTickCount() - start;
        DWORD remain = (elapsed < waitMs) ? (DWORD)(waitMs - elapsed) : 0;
        if (remain > 50) remain = 50;   // 分段等待，避免错过队列中的消息
        MsgWaitForMultipleObjectsEx(0, nullptr, remain, QS_ALLINPUT,
                                    MWMO_INPUTAVAILABLE);
    }
}
 
inline std::wstring aShellItemRealName(FolderItem* pItem)
{
    if (!pItem) return std::wstring();
 
    BSTR bstrPath = nullptr;
    if (SUCCEEDED(pItem->get_Path(&bstrPath)) && bstrPath)
    {
        std::wstring path(bstrPath, SysStringLen(bstrPath));
        SysFreeString(bstrPath);
        size_t pos = path.find_last_of(L"\\/");
        if (pos != std::wstring::npos)
        {
            if (pos + 1 < path.size()) return path.substr(pos + 1);
        }
        else if (!path.empty())
        {
            return path;   // 没有分隔符，Path 本身就是名字
        }
    }
 
    // 退路：Path 拿不到时退回显示名称（少数命名空间未实现 Path）
    BSTR bstrName = nullptr;
    if (SUCCEEDED(pItem->get_Name(&bstrName)) && bstrName)
    {
        std::wstring name(bstrName, SysStringLen(bstrName));
        SysFreeString(bstrName);
        return name;
    }
    return std::wstring();
}
 
 
//           这里把 UI 全部关掉，完成与否只由 aShellWaitForItem 轮询判定
constexpr long kShellComCopyFlags = FOF_NO_UI;
 
constexpr unsigned long kShellComWaitTimeoutMs = 60000;
 
inline bool aShellWaitForItem(Folder* pFolder, const std::wstring& itemName, unsigned long timeoutMs = kShellComWaitTimeoutMs)
{
    unsigned long start = GetTickCount();
 
    while ((GetTickCount() - start) < timeoutMs)
    {
        FolderItem* pCheck = nullptr;
        BSTR bstrName = SysAllocString(itemName.c_str());
        if (!bstrName) return false;
        HRESULT hr = pFolder->ParseName(bstrName, &pCheck);
        SysFreeString(bstrName);
 
        if (SUCCEEDED(hr) && pCheck != nullptr)
        {
            pCheck->Release();
            return true;
        }
        aShellPumpMessages(100);
    }
    return false;
}
 
AST_NAMESPACE_END
 
#endif // _WIN32