mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
510 lines
15 KiB
510 lines
15 KiB
//+-------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
//
|
|
// Copyright (C) Microsoft Corporation, 1999 - 1999
|
|
//
|
|
// File: nodeinit.h
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
// NodeInit.h : Declaration of the CNodeInitObject
|
|
|
|
#ifndef _NODEINIT_H_
|
|
#define _NODEINIT_H_
|
|
|
|
class CMenuItem;
|
|
class SnapinStruct;
|
|
class CCommandSink;
|
|
class CImageIndexMap;
|
|
class CMTNode;
|
|
class CNode;
|
|
class CSnapIn;
|
|
class CConsoleFrame;
|
|
class CConsoleView;
|
|
class CConsoleStatusBar;
|
|
class CContextMenu;
|
|
class CMenuItem;
|
|
|
|
interface IExtendContextMenu;
|
|
|
|
typedef CList<CMenuItem*, CMenuItem*> MenuItemList;
|
|
typedef CList<SnapinStruct*, SnapinStruct*> SnapinStructList;
|
|
|
|
typedef long MENU_OWNER_ID;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// NodeMgr
|
|
|
|
#include <pshpack8.h>
|
|
|
|
class CNodeInitObject :
|
|
public IFramePrivate,
|
|
public IHeaderCtrlPrivate,
|
|
public IContextMenuProvider,
|
|
public IResultDataPrivate,
|
|
public IScopeDataPrivate,
|
|
public IImageListPrivate,
|
|
public ISupportErrorInfo,
|
|
public IDisplayHelp,
|
|
public IStringTable,
|
|
public CPropertySheetProvider,
|
|
public CColumnData,
|
|
public CComObjectRoot,
|
|
public CComCoClass<CNodeInitObject, &CLSID_NodeInit>
|
|
{
|
|
// Constructor/Destructor
|
|
public:
|
|
CNodeInitObject();
|
|
~CNodeInitObject();
|
|
|
|
friend CColumnData;
|
|
|
|
BEGIN_COM_MAP(CNodeInitObject)
|
|
COM_INTERFACE_ENTRY(IFramePrivate)
|
|
COM_INTERFACE_ENTRY(IConsole)
|
|
COM_INTERFACE_ENTRY(IConsole2)
|
|
COM_INTERFACE_ENTRY(IConsole3)
|
|
COM_INTERFACE_ENTRY(IHeaderCtrl)
|
|
COM_INTERFACE_ENTRY(IHeaderCtrl2)
|
|
COM_INTERFACE_ENTRY(IHeaderCtrlPrivate)
|
|
COM_INTERFACE_ENTRY(IContextMenuProvider)
|
|
// THis interface used to be exposed on this class, and removing it may
|
|
// expose compatibility problems [vivekj]
|
|
// COM_INTERFACE_ENTRY(IContextMenuCallback)
|
|
COM_INTERFACE_ENTRY(ISupportErrorInfo)
|
|
COM_INTERFACE_ENTRY(IResultData)
|
|
COM_INTERFACE_ENTRY(IResultData2)
|
|
COM_INTERFACE_ENTRY(IResultDataPrivate)
|
|
COM_INTERFACE_ENTRY(IConsoleNameSpace)
|
|
COM_INTERFACE_ENTRY(IConsoleNameSpace2)
|
|
COM_INTERFACE_ENTRY(IScopeDataPrivate)
|
|
COM_INTERFACE_ENTRY(IImageList)
|
|
COM_INTERFACE_ENTRY(IImageListPrivate)
|
|
COM_INTERFACE_ENTRY(IPropertySheetProviderPrivate)
|
|
COM_INTERFACE_ENTRY(IPropertySheetProvider)
|
|
COM_INTERFACE_ENTRY(IPropertySheetCallback)
|
|
COM_INTERFACE_ENTRY(IPropertySheetNotify)
|
|
COM_INTERFACE_ENTRY(IDisplayHelp)
|
|
COM_INTERFACE_ENTRY(IStringTable)
|
|
COM_INTERFACE_ENTRY_FUNC(IID_IColumnData, 0, ColumnInterfaceFunc)
|
|
END_COM_MAP()
|
|
// Use DECLARE_NOT_AGGREGATABLE(CNodeInitObject) if you don't want your object
|
|
// to support aggregation
|
|
DECLARE_AGGREGATABLE(CNodeInitObject)
|
|
|
|
DECLARE_MMC_OBJECT_REGISTRATION (
|
|
g_szMmcndmgrDll, // implementing DLL
|
|
CLSID_NodeInit, // CLSID
|
|
_T("NodeInit 1.0 Object"), // class name
|
|
_T("NODEMGR.NodeInitObject.1"), // ProgID
|
|
_T("NODEMGR.NodeInitObject")) // version-independent ProgID
|
|
|
|
IMPLEMENTS_SNAPIN_NAME_FOR_DEBUG()
|
|
|
|
public:
|
|
// ISupportsErrorInfo
|
|
STDMETHOD(InterfaceSupportsErrorInfo)(REFIID riid);
|
|
|
|
public:
|
|
#ifdef DBG
|
|
int dbg_cRef;
|
|
ULONG InternalAddRef()
|
|
{
|
|
++dbg_cRef;
|
|
return CComObjectRoot::InternalAddRef();
|
|
}
|
|
ULONG InternalRelease()
|
|
{
|
|
--dbg_cRef;
|
|
return CComObjectRoot::InternalRelease();
|
|
}
|
|
int dbg_InstID;
|
|
#endif // DBG
|
|
|
|
// IFramePrivate
|
|
public:
|
|
STDMETHOD(CreateScopeImageList)(REFCLSID refClsidSnapIn);
|
|
STDMETHOD(SetResultView)(LPUNKNOWN pUnknown);
|
|
STDMETHOD(IsResultViewSet)(BOOL* pbIsLVSet);
|
|
STDMETHOD(SetTaskPadList)(LPUNKNOWN pUnknown);
|
|
STDMETHOD(GetComponentID)(COMPONENTID* lpComponentID);
|
|
STDMETHOD(SetComponentID)(COMPONENTID id);
|
|
STDMETHOD(SetNode)(HMTNODE hMTNode, HNODE hNode = NULL);
|
|
STDMETHOD(QueryScopeTree)(IScopeTree** ppScopeTree);
|
|
STDMETHOD(SetScopeTree)(IScopeTree* pScopeTree);
|
|
STDMETHOD(SetComponent)(LPCOMPONENT lpComponent);
|
|
STDMETHOD(SetUsedByExtension)(BOOL bExtension);
|
|
STDMETHOD(IsUsedByExtension)(void);
|
|
STDMETHOD(GetMainWindow)(HWND* phwnd);
|
|
|
|
HWND GetMainWindow()
|
|
{
|
|
ASSERT(m_spScopeTree != NULL);
|
|
if (m_spScopeTree == NULL)
|
|
return NULL;
|
|
|
|
CScopeTree* const pst = dynamic_cast<CScopeTree*>(&*m_spScopeTree);
|
|
ASSERT(pst != NULL);
|
|
return (pst != NULL && m_spScopeTree != NULL) ?
|
|
pst->GetMainWindow() : NULL;
|
|
}
|
|
|
|
CMTNode* GetMTNode() { return m_pMTNode;}
|
|
|
|
STDMETHOD(InitViewData)(LONG_PTR lViewData);
|
|
STDMETHOD(CleanupViewData)(LONG_PTR lViewData);
|
|
STDMETHOD(ResetSortParameters)();
|
|
|
|
STDMETHOD(ReleaseCachedOleObjects)();
|
|
|
|
// IConsole2
|
|
public:
|
|
STDMETHOD(Expand)(HSCOPEITEM hItem, BOOL bExpand);
|
|
STDMETHOD(IsTaskpadViewPreferred)();
|
|
STDMETHOD(SetStatusText)(LPOLESTR pszStatusText);
|
|
|
|
// IConsole3
|
|
public:
|
|
STDMETHOD(RenameScopeItem)(HSCOPEITEM hScopeItem);
|
|
|
|
protected:
|
|
static IScopeTreePtr m_spScopeTree;
|
|
IConsoleVerbPtr m_spConsoleVerb;
|
|
CMTNode* m_pMTNode;
|
|
CNode* m_pNode;
|
|
BOOL m_bExtension;
|
|
|
|
// IFrame
|
|
public:
|
|
STDMETHOD(QueryResultView)(LPUNKNOWN* ppIUnknown);
|
|
STDMETHOD(SetHeader)(IHeaderCtrl* pHeader);
|
|
STDMETHOD(SetToolbar)(IToolbar* pToolbar);
|
|
STDMETHOD(QueryScopeImageList)(LPIMAGELIST *ppImageList);
|
|
STDMETHOD(QueryResultImageList)(LPIMAGELIST *ppImageList);
|
|
STDMETHOD(MessageBox)(LPCWSTR lpszText, LPCWSTR lpszTitle, UINT fuStyle, int* piRetval);
|
|
STDMETHOD(UpdateAllViews)(LPDATAOBJECT lpDataObject, LPARAM data, LONG_PTR hint);
|
|
STDMETHOD(QueryConsoleVerb)(LPCONSOLEVERB* ppConsoleVerb);
|
|
STDMETHOD(SelectScopeItem)(HSCOPEITEM hScopeItem);
|
|
STDMETHOD(NewWindow)(HSCOPEITEM hScopeItem, unsigned long lOptions);
|
|
|
|
// IFrames members
|
|
protected:
|
|
LPUNKNOWN m_pLVImage;
|
|
LPUNKNOWN m_pTVImage;
|
|
LPUNKNOWN m_pToolbar;
|
|
LPIMAGELISTPRIVATE m_pImageListPriv;
|
|
COMPONENTID m_componentID;
|
|
IComponentPtr m_spComponent;
|
|
|
|
IUnknownPtr m_spResultViewUnk; // IUnknown for the result view
|
|
|
|
public:
|
|
CConsoleFrame* GetConsoleFrame() const;
|
|
CConsoleView* GetConsoleView(bool fDefaultToActive = true) const;
|
|
CConsoleStatusBar* GetStatusBar (bool fDefaultToActive = true) const;
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IHeaderCtrl interface
|
|
|
|
protected:
|
|
STDMETHOD(InsertColumn)(int nCol, LPCWSTR lpszTitle, int nFormat, int nWidth);
|
|
STDMETHOD(DeleteColumn)(int nCol);
|
|
STDMETHOD(SetColumnWidth)(int nCol, int nWidth);
|
|
STDMETHOD(GetColumnText)(int nCol, LPWSTR* pText);
|
|
STDMETHOD(SetColumnText)(int nCol, LPCWSTR title);
|
|
STDMETHOD(GetColumnWidth)(int nCol, int* pWidth);
|
|
|
|
// IHeaderCtrl2 interface
|
|
STDMETHOD(SetChangeTimeOut)(unsigned long uTimeout);
|
|
STDMETHOD(SetColumnFilter)(UINT nColumn, DWORD dwType, MMC_FILTERDATA* pFilterData);
|
|
STDMETHOD(GetColumnFilter)(UINT nColumn, LPDWORD pType, MMC_FILTERDATA* pFilterData);
|
|
|
|
// IHeaderCtrlPrivate interface.
|
|
STDMETHOD(GetColumnCount)(INT* pnCol);
|
|
STDMETHOD(GetColumnInfoList)(/*[out]*/ CColumnInfoList *pColumnsList);
|
|
STDMETHOD(ModifyColumns)(/*[in]*/ const CColumnInfoList& columnsList);
|
|
STDMETHOD(GetDefaultColumnInfoList)(/*[out]*/ CColumnInfoList& columnsList);
|
|
|
|
private:
|
|
CCLVSortParams m_sortParams;
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IDisplayHelp interface
|
|
|
|
protected:
|
|
STDMETHOD(ShowTopic)(LPOLESTR pszHelpTopic);
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IStringTable interface
|
|
|
|
protected:
|
|
STDMETHOD(AddString) (LPCOLESTR pszAdd, MMC_STRING_ID* pID);
|
|
STDMETHOD(GetString) (MMC_STRING_ID id, ULONG cchBuffer, LPOLESTR lpBuffer, ULONG* pcchOut);
|
|
STDMETHOD(GetStringLength) (MMC_STRING_ID id, ULONG* pcchString);
|
|
STDMETHOD(DeleteString) (MMC_STRING_ID id);
|
|
STDMETHOD(DeleteAllStrings) ();
|
|
STDMETHOD(FindString) (LPCOLESTR pszFind, MMC_STRING_ID* pID);
|
|
STDMETHOD(Enumerate) (IEnumString** ppEnum);
|
|
|
|
HRESULT GetSnapinCLSID (CLSID& pclsid) const;
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IResultDataPrivate interface
|
|
|
|
protected:
|
|
IMMCListViewPtr m_spListViewPrivate;
|
|
|
|
STDMETHOD(Arrange)(long style);
|
|
STDMETHOD(InsertItem)(LPRESULTDATAITEM item);
|
|
STDMETHOD(DeleteItem)(HRESULTITEM itemID, int nCol);
|
|
STDMETHOD(FindItemByLParam)(LPARAM lParam, HRESULTITEM *pItemID);
|
|
STDMETHOD(DeleteAllRsltItems)();
|
|
STDMETHOD(SetItem)(LPRESULTDATAITEM item);
|
|
STDMETHOD(GetItem)(LPRESULTDATAITEM item);
|
|
STDMETHOD(ModifyItemState)(int nIndex, HRESULTITEM ItemID, UINT uAdd, UINT uRemove);
|
|
STDMETHOD(ModifyViewStyle)(MMC_RESULT_VIEW_STYLE add, MMC_RESULT_VIEW_STYLE remove);
|
|
STDMETHOD(GetNextItem)(LPRESULTDATAITEM item);
|
|
STDMETHOD(SetViewMode)(LONG nViewMode);
|
|
STDMETHOD(GetViewMode)(LONG* nViewMode);
|
|
STDMETHOD(ResetResultData)();
|
|
STDMETHOD(GetListStyle)(long * pStyle);
|
|
STDMETHOD(SetListStyle)(long Style);
|
|
STDMETHOD(UpdateItem)(HRESULTITEM itemID);
|
|
STDMETHOD(Sort)(int nCol, DWORD dwSortOptions, LPARAM lUserParam);
|
|
STDMETHOD(InternalSort)(INT nCol, DWORD dwSortOptions, LPARAM lUserParam, BOOL bColumnClicked);
|
|
STDMETHOD(SetDescBarText)(LPOLESTR DescText);
|
|
STDMETHOD(SetItemCount)(int nItemCount, DWORD dwOptions);
|
|
STDMETHOD(SetLoadMode)(BOOL bState);
|
|
STDMETHOD(GetSortColumn)(INT* pnCol);
|
|
STDMETHOD(GetSortDirection)(BOOL* pbAscending);
|
|
|
|
// IResultData2
|
|
STDMETHOD(RenameResultItem)(HRESULTITEM itemID);
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IScopeData interface
|
|
|
|
protected:
|
|
// IConsoleNameSpace methods
|
|
STDMETHOD(InsertItem)(LPSCOPEDATAITEM item);
|
|
STDMETHOD(DeleteItem)(HSCOPEITEM hItem, long fDeleteThis);
|
|
STDMETHOD(SetItem)(LPSCOPEDATAITEM item);
|
|
STDMETHOD(GetItem)(LPSCOPEDATAITEM item);
|
|
STDMETHOD(GetChildItem)(HSCOPEITEM item, HSCOPEITEM* pItemChild,
|
|
MMC_COOKIE* pCookie);
|
|
STDMETHOD(GetNextItem)(HSCOPEITEM item, HSCOPEITEM* pItemNext,
|
|
MMC_COOKIE* pCookie);
|
|
STDMETHOD(GetParentItem)(HSCOPEITEM item, HSCOPEITEM* pItemParent,
|
|
MMC_COOKIE* pCookie);
|
|
|
|
// IConsoleNameSpace2 method(s)
|
|
STDMETHOD(Expand)(HSCOPEITEM hItem);
|
|
STDMETHOD(AddExtension)(HSCOPEITEM hItem, LPCLSID lpclsid);
|
|
|
|
private:
|
|
|
|
enum EGetItem
|
|
{
|
|
egiParent = 1,
|
|
egiChild = 2,
|
|
egiNext = 3
|
|
};
|
|
|
|
HRESULT GetRelativeItem(EGetItem egi, HSCOPEITEM item, HSCOPEITEM* pItem,
|
|
MMC_COOKIE* pCookie);
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IContextMenuCallback interface
|
|
|
|
public:
|
|
STDMETHOD(AddItem) ( CONTEXTMENUITEM* pItem );
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IContextMenuProvider interface
|
|
|
|
public:
|
|
STDMETHOD(EmptyMenuList) ();
|
|
STDMETHOD(AddThirdPartyExtensionItems) (
|
|
IDataObject* piDataObject );
|
|
STDMETHOD(AddPrimaryExtensionItems) (
|
|
IUnknown* piCallback,
|
|
IDataObject* piDataObject );
|
|
STDMETHOD(ShowContextMenu) (HWND hwndParent,
|
|
LONG xPos,
|
|
LONG yPos,
|
|
LONG* plSelected);
|
|
|
|
private:
|
|
ContextMenuPtr m_spContextMenu;
|
|
CContextMenu * GetContextMenu();
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IImageListPrivate interface
|
|
public:
|
|
STDMETHOD(ImageListSetIcon)(PLONG_PTR pIcon, LONG nLoc);
|
|
STDMETHOD(ImageListSetStrip)(PLONG_PTR pBMapSm, PLONG_PTR pBMapLg,
|
|
LONG nStartLoc, COLORREF cMask);
|
|
|
|
STDMETHOD(MapRsltImage)(COMPONENTID id, int nSnapinIndex, int *pnConsoleIndex);
|
|
STDMETHOD(UnmapRsltImage)(COMPONENTID id, int nConsoleIndex, int *pnSnapinIndex);
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// IToobar interface
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// Node Manager internal members
|
|
private:
|
|
void Construct();
|
|
void Destruct();
|
|
|
|
HRESULT CheckArgument(VARIANT* pArg);
|
|
|
|
SC ScIsVirtualList(bool& bVirtual);
|
|
public:
|
|
HRESULT static GetSnapInAndNodeType(LPDATAOBJECT pDataObject,
|
|
CSnapIn** ppSnapIn, GUID* pguidObjectType);
|
|
|
|
IComponent* GetComponent() { return m_spComponent;}
|
|
};
|
|
|
|
#include <poppack.h>
|
|
|
|
inline STDMETHODIMP CNodeInitObject::GetComponentID(COMPONENTID* lpComponentID)
|
|
{
|
|
MMC_TRY
|
|
|
|
ASSERT(m_componentID != -1); // The component ID has not been set yet!!!
|
|
ASSERT(lpComponentID);
|
|
|
|
*lpComponentID = m_componentID;
|
|
|
|
return S_OK;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
inline STDMETHODIMP CNodeInitObject::SetComponentID(COMPONENTID id)
|
|
{
|
|
MMC_TRY
|
|
|
|
/* for dynamic icon, we need to change this value temporarily
|
|
if (m_componentID != -1)
|
|
{
|
|
ASSERT(FALSE); // ID already has been set!!!
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ASSERT(id != -1);
|
|
*/
|
|
m_componentID = id;
|
|
return S_OK;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
inline STDMETHODIMP CNodeInitObject::SetNode(HMTNODE hMTNode, HNODE hNode)
|
|
{
|
|
MMC_TRY
|
|
|
|
m_pMTNode = CMTNode::FromHandle(hMTNode);
|
|
m_pNode = CNode::FromHandle(hNode);
|
|
return S_OK;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
inline STDMETHODIMP CNodeInitObject::SetComponent(LPCOMPONENT lpComponent)
|
|
{
|
|
MMC_TRY
|
|
|
|
if (lpComponent == NULL)
|
|
{
|
|
ASSERT(FALSE);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
m_spComponent = lpComponent;
|
|
|
|
return S_OK;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
inline STDMETHODIMP CNodeInitObject::SetUsedByExtension(BOOL bExtension)
|
|
{
|
|
MMC_TRY
|
|
|
|
m_bExtension = bExtension;
|
|
return S_OK;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
inline STDMETHODIMP CNodeInitObject::IsUsedByExtension(void)
|
|
{
|
|
MMC_TRY
|
|
|
|
return m_bExtension ? S_OK : S_FALSE;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
inline SC CNodeInitObject::ScIsVirtualList(bool& bVirtual)
|
|
{
|
|
DECLARE_SC(sc, TEXT("CNodeInitObject::ScIsVirtualList"));
|
|
sc = ScCheckPointers(m_spListViewPrivate, E_UNEXPECTED);
|
|
if (sc)
|
|
return sc;
|
|
|
|
long lStyle = m_spListViewPrivate->GetListStyle();
|
|
|
|
bVirtual = (lStyle & LVS_OWNERDATA);
|
|
|
|
return sc;
|
|
}
|
|
|
|
|
|
inline STDMETHODIMP CNodeInitObject::GetMainWindow(HWND* phwnd)
|
|
{
|
|
MMC_TRY
|
|
|
|
if (phwnd == NULL)
|
|
return E_POINTER;
|
|
*phwnd = GetMainWindow();
|
|
ASSERT(*phwnd != NULL);
|
|
return (*phwnd != NULL) ? S_OK : E_UNEXPECTED;
|
|
|
|
MMC_CATCH
|
|
}
|
|
|
|
// Used for getting snapin name for debug info.
|
|
|
|
inline void Debug_SetNodeInitSnapinName(CSnapInPtr pSnapIn, IFramePrivate* pIFrame)
|
|
{
|
|
#ifdef DBG
|
|
CNodeInitObject* pNodeInitObj = dynamic_cast<CNodeInitObject*>(pIFrame);
|
|
|
|
if ((pSnapIn != NULL) && (pNodeInitObj != NULL))
|
|
{
|
|
WTL::CString strSnapInName;
|
|
SC sc = pSnapIn->ScGetSnapInName(strSnapInName);
|
|
if (sc)
|
|
return;
|
|
|
|
if (!strSnapInName.IsEmpty())
|
|
{
|
|
pNodeInitObj->SetSnapinName(strSnapInName);
|
|
CColumnData* pColumnData = dynamic_cast<CColumnData*>(pNodeInitObj);
|
|
if (pColumnData)
|
|
pColumnData->SetSnapinName(strSnapInName);
|
|
}
|
|
};
|
|
#endif
|
|
}
|
|
|
|
|
|
#endif // _NODEINIT_H_
|