c++ - 如何从命令行C++ VSPackage获取DTE对象

我有一个仅命令行的C++ VSPackage。在我的pkgdef文件中设置以下自动加载注册表项，并不会从命令行运行devenv时强制加载VSPackage。

[$RootKey$\AutoLoadPackages\{ADFC4E64-0397-11D1-9F4E-00A0C911004F}]
"{75726504-cacb-4781-b384-63815a289e0a}"=dword:00000000
@="UICONTEXT_NoSolution"

但是，如果添加命令行参数，则可以根据需要加载VSPackage:

[$RootKey$\AppCommandLine]
"vsp"="{75726504-cacb-4781-b384-63815a289e0a}"

[$RootKey$\AppCommandLine\vsp]
"Arguments"="0"
"DemandLoad"=dword:00000001
"Package"="{75726504-cacb-4781-b384-63815a289e0a}"
"HelpString"="#102"

问题是，当我尝试在OnAfterOpenSolution事件中获取DTE对象时，它失败。当我在GUI模式下运行devenv时，相同的代码有效。

有没有一种更好的方法可以让我的命令行vspackage在启动时加载，还是有一种方法可以使用我上面使用的方法来获取DTE对象？

这是我用来获取DTE的代码:

CComPtr<EnvDTE::_DTE> SinkSolutionEvents::GetDTE(CString program)
{
    CString rotEntry;
    rotEntry.Format(L"!VisualStudio.DTE.14.0:%d", program, GetCurrentProcessId());

    IRunningObjectTable *rot = NULL;
    GetRunningObjectTable(0, &rot);
    IEnumMoniker *enumMoniker;
    rot->EnumRunning(&enumMoniker);
    enumMoniker->Reset();
    ULONG fetched = 0;
    IMoniker *moniker = NULL;
    while (enumMoniker->Next(1, &moniker, &fetched) == 0)
    {
        IBindCtx *bindCtx = NULL;
        CreateBindCtx(0, &bindCtx);
        LPOLESTR pwszName;
        HRESULT hr = moniker->GetDisplayName(bindCtx, NULL, &pwszName);
        CString displayName;
        if (SUCCEEDED(hr)) {
            displayName = pwszName;
            CoTaskMemFree(pwszName);
            if (displayName == rotEntry)
            {
                CComPtr<IUnknown> punk;
                rot->GetObject(moniker, &punk);
                CComPtr<EnvDTE::_DTE> dte;
                dte = punk;
                return dte;
            }
        }
    }
    return NULL;
}

最佳答案

我找到了问题的答案。您可以重写IVsPackageImpl::SetSite()并在指定的IServiceProvider上使用QueryService来访问EnvDTE。

HRESULT STDMETHODCALLTYPE MyVSPackage::SetSite( __RPC__in_opt IServiceProvider *pSP)
{
    CComPtr<EnvDTE::_DTE> m_pDTE = NULL;
    HRESULT hr = pSP->QueryService(SID_SDTE, IID__DTE, (void **)&m_pDTE);
    return IVsPackageImpl::SetSite(pSP);
}