windows-nt/Source/XPSP1/NT/admin/activec/nodemgr/nodeinit.h

511 lines
15 KiB
C
Raw Permalink Normal View History

2020-09-26 03:20:57 -05:00
//+-------------------------------------------------------------------------
//
// 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_