archive
/
windows-server-2003
mirror of https://github.com/selfrender/Windows-Server-2003
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Leaked source code of windows server 2003
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.
4 Commits
1 Branch
0 Tags
1.5 GiB
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
windows-server-2003/admin/activec/nodemgr/node.cpp

4360 lines
127 KiB
Raw Permalink Blame History

//____________________________________________________________________________
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1996 - 1999
//
// File: Node.cpp
//
// Contents:
//
// Classes:
//
// Functions:
//
// History: 9/16/1996 RaviR Created
//
//____________________________________________________________________________
#include "stdafx.h"
#include "macros.h"
#include "strings.h"
#include "ndmgr.h"
#include "regutil.h"
#include "taskenum.h"
#include "nodemgr.h"
#include "multisel.h"
#include "rsltitem.h"
#include "colwidth.h"
#include "viewpers.h"
#include "tasks.h"
#include "conview.h"
#include "columninfo.h"
#include "util.h" // for CoTaskDupString
#include "mmcprotocol.h"
#include "nodemgrdebug.h"
#include "copypast.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/*+-------------------------------------------------------------------------*
* class CConsoleTaskpadViewExtension
*
*
* PURPOSE: Implements console taskpads as a view extension
*
*+-------------------------------------------------------------------------*/
class CConsoleTaskpadViewExtension
{
public:
/*+-------------------------------------------------------------------------*
*
* ScGetViews
*
* PURPOSE: Adds all console taskpad views to the view extension callback.
*
* PARAMETERS:
* CNode * pNode :
* LPVIEWEXTENSIONCALLBACK pViewExtensionCallback :
*
* RETURNS:
* SC
*
*+-------------------------------------------------------------------------*/
static SC ScGetViews(CNode *pNode, LPVIEWEXTENSIONCALLBACK pViewExtensionCallback)
{
DECLARE_SC(sc, TEXT("CConsoleTaskpadViewExtension::ScGetViews"));
CScopeTree* pScopeTree = CScopeTree::GetScopeTree();
sc = ScCheckPointers(pNode, pViewExtensionCallback, pScopeTree, E_FAIL);
if(sc)
return sc;
// get a filtered list of taskpads that apply to this node.
CConsoleTaskpadFilteredList filteredList;
sc = pScopeTree->GetConsoleTaskpadList()->ScGetTaskpadList(pNode, filteredList);
if(sc)
return sc;
for(CConsoleTaskpadFilteredList::iterator iter = filteredList.begin(); iter!= filteredList.end(); ++iter)
{
CConsoleTaskpad *pConsoleTaskpad = *iter;
sc = ScAddViewForTaskpad(pConsoleTaskpad, pViewExtensionCallback);
if(sc)
return sc;
}
return sc;
}
/*+-------------------------------------------------------------------------*
* ScGetTaskpadViewExtension
*
* Returns S_OK if the given CLSID matches the CLSID of any taskpad view
* extension for the given node, S_FALSE or error otherwise.
*--------------------------------------------------------------------------*/
static SC ScGetViewExtensionTaskpad (CNode* pNode, const CLSID& clsid, CConsoleTaskpad*& pConsoleTaskpad)
{
DECLARE_SC (sc, _T("CConsoleTaskpadViewExtension::ScGetTaskpadViewExtension"));
/*
* initialize output
*/
pConsoleTaskpad = NULL;
/*
* check input
*/
sc = ScCheckPointers (pNode);
if(sc)
return sc;
CScopeTree* pScopeTree = CScopeTree::GetScopeTree();
sc = ScCheckPointers (pScopeTree, E_UNEXPECTED);
if (sc)
return (sc);
// get a filtered list of taskpads that apply to this node.
CConsoleTaskpadFilteredList filteredList;
sc = pScopeTree->GetConsoleTaskpadList()->ScGetTaskpadList(pNode, filteredList);
if(sc)
return sc;
for(CConsoleTaskpadFilteredList::iterator iter = filteredList.begin(); iter!= filteredList.end(); ++iter)
{
CConsoleTaskpad* pTempConsoleTaskpad = *iter;
sc = ScCheckPointers (pTempConsoleTaskpad, E_UNEXPECTED);
if (sc)
return (sc);
/*
* if the CLSID matches the ID of this taskpad, CLSID refers to
* a taskpad view extension
*/
if (clsid == pTempConsoleTaskpad->GetID())
{
pConsoleTaskpad = pTempConsoleTaskpad;
break;
}
}
return (sc);
}
private:
/*+-------------------------------------------------------------------------*
*
* ScAddViewForTaskpad
*
* PURPOSE: Adds a view based on a console taskpad to the view extension
* callback.
*
* PARAMETERS:
* CConsoleTaskpad * pConsoleTaskpad :
* LPVIEWEXTENSIONCALLBACK pViewExtensionCallback :
*
* RETURNS:
* SC
*
*+-------------------------------------------------------------------------*/
static SC ScAddViewForTaskpad(CConsoleTaskpad *pConsoleTaskpad, LPVIEWEXTENSIONCALLBACK pViewExtensionCallback)
{
DECLARE_SC(sc, TEXT("CConsoleTaskpadViewExtension::ScAddViewForTaskpad"));
// validate inputs
sc = ScCheckPointers(pConsoleTaskpad, pViewExtensionCallback);
if(sc)
return sc;
MMC_EXT_VIEW_DATA extViewData = {0};
// get the string form of the taskpad ID.
CCoTaskMemPtr<WCHAR> spszTaskpadID;
sc = StringFromCLSID (pConsoleTaskpad->GetID(), &spszTaskpadID);
if (sc)
return sc;
std::wstring strTaskpad = _W(MMC_PROTOCOL_SCHEMA_NAME) _W(":");
strTaskpad += spszTaskpadID;
extViewData.pszURL = strTaskpad.c_str();
extViewData.bReplacesDefaultView = pConsoleTaskpad->FReplacesDefaultView() ? TRUE : FALSE; // convert from bool to BOOL
extViewData.viewID = pConsoleTaskpad->GetID(); // set the GUID identifier of the view
USES_CONVERSION;
tstring strName = pConsoleTaskpad->GetName();
extViewData.pszViewTitle = T2COLE(strName.data()); // set the title of the view
if(!extViewData.pszViewTitle)
return (sc = E_OUTOFMEMORY).ToHr();
sc = pViewExtensionCallback->AddView(&extViewData);
return sc;
}
};
//############################################################################
//############################################################################
//
// Implementation of class CComponent
//
//############################################################################
//############################################################################
DEBUG_DECLARE_INSTANCE_COUNTER(CComponent);
void CComponent::Construct(CSnapIn * pSnapIn, CComponent* pComponent)
{
ASSERT(pSnapIn);
DEBUG_INCREMENT_INSTANCE_COUNTER(CComponent);
m_spSnapIn = pSnapIn;
m_ComponentID = -1;
m_bIComponentInitialized = false;
if (pComponent)
{
ASSERT(pComponent->m_spIComponent != NULL);
ASSERT(pComponent->m_spIFrame != NULL);
m_spIComponent = pComponent->m_spIComponent;
m_spIFrame = pComponent->m_spIFrame;
m_spIRsltImageList = pComponent->m_spIRsltImageList;
m_ComponentID = pComponent->GetComponentID();
}
}
CComponent::~CComponent()
{
DECLARE_SC(sc, TEXT("CComponent::~CComponent"));
DEBUG_DECREMENT_INSTANCE_COUNTER(CComponent);
if (m_spIFrame)
{
sc = m_spIFrame->SetHeader(NULL);
if (sc)
sc.TraceAndClear();
}
if (m_spIComponent)
{
sc = m_spIComponent->Destroy(NULL);
if (sc)
sc.TraceAndClear();
}
}
HRESULT CComponent::Init(IComponentData* pIComponentData, HMTNODE hMTNode,
HNODE lNode,
COMPONENTID nComponentID, int viewID)
{
DECLARE_SC(sc, TEXT("CComponent::Init"));
ASSERT(hMTNode != 0);
ASSERT(lNode != 0);
sc = ScCheckPointers( pIComponentData, E_POINTER );
if (sc)
return sc.ToHr();
do
{
sc = pIComponentData->CreateComponent(&m_spIComponent);
if (sc)
break;
// recheck the pointers
sc = ScCheckPointers( m_spIComponent, E_UNEXPECTED );
if (sc)
break;
// Create an IFrame for this IComponent
#if _MSC_VER>=1100
sc = m_spIFrame.CreateInstance(CLSID_NodeInit, NULL, MMC_CLSCTX_INPROC);
#else
sc = m_spIFrame.CreateInstance(CLSID_NodeInit, MMC_CLSCTX_INPROC);
#endif
if (sc)
break;
// recheck the pointer
sc = ScCheckPointers( m_spIFrame, E_UNEXPECTED );
if (sc)
break;
Debug_SetNodeInitSnapinName(m_spSnapIn, m_spIFrame.GetInterfacePtr());
// Cache the IComponent in the NodeInit object
sc = m_spIFrame->SetComponent(m_spIComponent);
if (sc)
break;
// recheck the pointer
sc = ScCheckPointers( m_spSnapIn, E_UNEXPECTED );
if (sc)
break;
// Create scope image list
sc = m_spIFrame->CreateScopeImageList(m_spSnapIn->GetSnapInCLSID());
if (sc)
break;
sc = m_spIFrame->SetNode(hMTNode, lNode);
if (sc)
break;
ASSERT(nComponentID == GetComponentID());
sc = m_spIFrame->SetComponentID(nComponentID);
if (sc)
break;
// Result Image list is optional
m_spIRsltImageList = m_spIFrame;
sc = ScCheckPointers( m_spIRsltImageList, E_FAIL );
if (sc)
sc.TraceAndClear();
// Complete IComponent initialization.
// Init m_spIComponent with m_spIFrame.
sc = m_spIComponent->Initialize(m_spIFrame);
if (sc)
break;
CMTNode* const pMTNode = CMTNode::FromHandle (hMTNode);
sc = ScCheckPointers( pMTNode, E_UNEXPECTED );
if (sc)
break;
CMTSnapInNode* const pSnapInNode = pMTNode->GetStaticParent();
sc = ScCheckPointers( pSnapInNode, E_UNEXPECTED );
if (sc)
break;
sc = pSnapInNode->ScInitIComponent(this, viewID);
if (sc)
break;
} while (0);
if (sc)
{
m_spIComponent = NULL;
m_spIFrame = NULL;
m_spIRsltImageList = NULL;
}
return sc.ToHr();
}
inline HRESULT CComponent::Notify(LPDATAOBJECT lpDataObject, MMC_NOTIFY_TYPE event,
LONG_PTR arg, LPARAM param)
{
ASSERT(m_spIComponent != NULL);
if (m_spIComponent == NULL)
return E_FAIL;
HRESULT hr = S_OK;
__try
{
hr = m_spIComponent->Notify(lpDataObject, event, arg, param);
}
__except (EXCEPTION_EXECUTE_HANDLER)
{
hr = E_FAIL;
if (m_spSnapIn)
TraceSnapinException(m_spSnapIn->GetSnapInCLSID(), TEXT("IComponent::Notify"), event);
}
return hr;
}
SC CComponent::ScQueryDispatch(MMC_COOKIE cookie,
DATA_OBJECT_TYPES type,
PPDISPATCH ppSelectedObject)
{
DECLARE_SC(sc, _T("CComponent::ScQueryDispatch"));
sc = ScCheckPointers(m_spIComponent, E_UNEXPECTED);
if (sc)
return sc;
IComponent2Ptr spComponent2 = m_spIComponent;
sc = ScCheckPointers(spComponent2.GetInterfacePtr(), E_NOINTERFACE);
if (sc)
return sc;
sc = spComponent2->QueryDispatch(cookie, type, ppSelectedObject);
return sc;
}
//+-------------------------------------------------------------------
//
// Member: CComponent::ScResetConsoleVerbStates
//
// Synopsis: Reset the verbstates in the CConsoleVerbImpl (the one
// snapin is aware of).
//
// Arguments: None
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CComponent::ScResetConsoleVerbStates ()
{
DECLARE_SC(sc, _T("CComponent::ScResetConsoleVerbStates"));
IFramePrivate* pIFP = GetIFramePrivate();
sc = ScCheckPointers(pIFP, E_UNEXPECTED);
if (sc)
return sc;
IConsoleVerbPtr spConsoleVerb;
sc = pIFP->QueryConsoleVerb(&spConsoleVerb);
if (sc)
return sc;
sc = ScCheckPointers(spConsoleVerb, E_UNEXPECTED);
if (sc)
return sc;
CConsoleVerbImpl* pCVI = dynamic_cast<CConsoleVerbImpl*>(
static_cast<IConsoleVerb*>(spConsoleVerb));
sc = ScCheckPointers(pCVI, E_UNEXPECTED);
if (sc)
return sc;
pCVI->SetDisabledAll();
return (sc);
}
//############################################################################
//############################################################################
//
// Implementation of class CNode
//
//############################################################################
//############################################################################
DEBUG_DECLARE_INSTANCE_COUNTER(CNode);
CNode::CNode (
CMTNode* pMTNode,
CViewData* pViewData,
bool fRootNode) :
m_pMTNode (pMTNode),
m_pViewData (pViewData),
m_hri (0),
m_dwFlags (0),
m_pPrimaryComponent (NULL),
m_bInitComponents (TRUE),
m_fRootNode (fRootNode),
m_fStaticNode (false)
{
CommonConstruct();
}
CNode::CNode (
CMTNode* pMTNode,
CViewData* pViewData,
bool fRootNode,
bool fStaticNode) :
m_pMTNode (pMTNode),
m_pViewData (pViewData),
m_hri (0),
m_dwFlags (0),
m_pPrimaryComponent (NULL),
m_bInitComponents (TRUE),
m_fRootNode (fRootNode),
m_fStaticNode (fStaticNode)
{
CommonConstruct();
}
CNode::CNode(const CNode& other) :
m_pMTNode (other.m_pMTNode),
m_pViewData (other.m_pViewData),
m_hri (other.m_hri),
m_dwFlags (other.m_dwFlags),
m_pPrimaryComponent (other.m_pPrimaryComponent),
m_bInitComponents (other.m_bInitComponents),
m_fRootNode (other.m_fRootNode),
m_fStaticNode (other.m_fStaticNode)
{
CommonConstruct();
}
void CNode::CommonConstruct ()
{
DEBUG_INCREMENT_INSTANCE_COUNTER(CNode);
ASSERT (m_pMTNode != NULL);
m_pMTNode->AddRef();
}
CNode::~CNode()
{
DEBUG_DECREMENT_INSTANCE_COUNTER(CNode);
CDataObjectCleanup::ScUnadviseNode( this );
/*
* if this is a non-static root node, delete the static
* parent node that was created for us in CMTNode::GetNode
*/
if (IsRootNode() && !IsStaticNode())
delete GetStaticParent();
ASSERT (m_pMTNode != NULL);
m_pMTNode->Release();
}
/*+-------------------------------------------------------------------------*
* CNode::FromResultItem
*
* Converts a CResultItem to the CNode it references. This should only
* be called for CResultItems that represent scope items.
*
* This function is out-of-line to eliminate coupling between node.h and
* rsltitem.h.
*--------------------------------------------------------------------------*/
CNode* CNode::FromResultItem (CResultItem* pri)
{
CNode* pNode = NULL;
if (pri != NULL)
{
/*
* only call for scope items
*/
ASSERT (pri->IsScopeItem());
if (pri->IsScopeItem())
pNode = CNode::FromHandle (pri->GetScopeNode());
}
return (pNode);
}
HRESULT
CNode::OnExpand(bool fExpand)
{
HRESULT hr = S_OK;
if (fExpand == FALSE)
{
return (WasExpandedAtLeastOnce() == TRUE) ? S_OK : S_FALSE;
}
if (WasExpandedAtLeastOnce() == TRUE)
return S_FALSE;
CMTNode* pMTNode = GetMTNode();
if (pMTNode->WasExpandedAtLeastOnce() == FALSE)
hr = pMTNode->Expand();
return hr;
}
void CNode::ResetControlbars(BOOL bSelect, SELECTIONINFO* pSelInfo)
{
ASSERT(pSelInfo != NULL);
CViewData* pVD = GetViewData();
ASSERT(pVD != NULL);
if (!pVD)
return;
// Reset controlbars
CControlbarsCache* pCtrlbarsCache =
dynamic_cast<CControlbarsCache*>(GetControlbarsCache());
ASSERT(pCtrlbarsCache != NULL);
if (pCtrlbarsCache != NULL)
{
if (pSelInfo->m_bScope == TRUE)
pCtrlbarsCache->OnScopeSelChange(this, bSelect);
else if (pSelInfo->m_bBackground == FALSE)
pCtrlbarsCache->OnResultSelChange(this, pSelInfo->m_lCookie,
bSelect);
}
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScInitializeVerbs
//
// Synopsis: Selection has changed so initialize the verbs for given
// selection information.
//
// Arguments: [bSelect] - [in] Select or Deselect of an item.
// [pSelInfo] - [in] SELECTIONINFO ptr.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScInitializeVerbs (bool bSelect, SELECTIONINFO* pSelInfo)
{
DECLARE_SC(sc, _T("CNode::ScInitializeVerbs"));
sc = ScCheckPointers(pSelInfo);
if (sc)
return sc;
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pViewData, E_UNEXPECTED);
if (sc)
return sc;
CVerbSet *pVerbSet = dynamic_cast<CVerbSet*>(pViewData->GetVerbSet());
sc = ScCheckPointers(pVerbSet, E_UNEXPECTED);
if (sc)
return sc;
sc = pVerbSet->ScInitialize(this, pSelInfo->m_bScope, bSelect,
pSelInfo->m_bBackground, pSelInfo->m_lCookie);
if (sc)
return sc;
return (sc);
}
HRESULT CNode::GetDispInfo(LV_ITEMW* plvi)
{
DECLARE_SC(sc, TEXT("CNode::GetDispInfo"));
sc = ScCheckPointers(plvi);
if(sc)
return sc.ToHr();
if (plvi->iSubItem == 0)
{
if (plvi->mask & LVIF_IMAGE)
{
plvi->iImage = GetResultImage();
ASSERT (plvi->iImage != -1);
if (plvi->iImage == -1)
plvi->iImage = 0;
}
if (plvi->mask & LVIF_TEXT)
{
tstring strName = GetDisplayName();
if (!strName.empty())
{
USES_CONVERSION;
sc = StringCchCopyW (plvi->pszText, plvi->cchTextMax, T2CW (strName.data()));
if(sc)
return sc.ToHr();
}
else
plvi->pszText[0] = 0;
}
}
else if ((plvi->mask & LVIF_TEXT) && (plvi->cchTextMax > 0))
{
plvi->pszText[0] = 0;
}
return S_OK;
}
//+-------------------------------------------------------------------
//
// Member: ScGetDataObject
//
// Synopsis: Given scope/result and cookie (lParam, if it is result item),
// get the dataobject of the item.
//
// Arguments:
// [bScopePane] - [in] Scope or Result.
// [lResultItemCookie] - [in] If Result pane is selected the item param.
// [bScopeItem] - [out] Is the dataobject returned for scope or result item.
// The scope item can be in result pane.
// [ppDataObject] - [out] The data-object (return val)
// [ppCComponent] - [out] NULL def parameter. The CComponent of the item. In case
// of multiselection, the items may belong to more than one
// component so a NULL is returned.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScGetDataObject(bool bScopePane, LPARAM lResultItemCookie, bool& bScopeItem,
LPDATAOBJECT* ppDataObject, CComponent **ppCComponent /*= NULL*/,
CNode **ppOwnerNode /*= NULL*/)
{
DECLARE_SC(sc, _T("CNode::ScGetDataObject"));
IDataObjectPtr spDataObject;
CComponent *pCC = NULL;
if (ppDataObject == NULL)
return (sc = E_POINTER);
*ppDataObject = NULL; // init
if (ppCComponent)
*ppCComponent = NULL;
if (ppOwnerNode)
*ppOwnerNode = this;
bScopeItem = bScopePane;
// In MMC1.0 when result pane background is selected, for any
// toolbar operation we pass the dataobject of scope selected item.
// The following code is added for this compatibility.
if (lResultItemCookie == LVDATA_BACKGROUND) // =>Result background has focus
{
bScopeItem = TRUE;
}
if (bScopeItem) // => Scope pane has focus
{
CMTNode* pMTNode = GetMTNode();
if (NULL == pMTNode)
return (sc = E_UNEXPECTED);
sc = pMTNode->QueryDataObject(CCT_SCOPE, &spDataObject);
if (sc)
return sc;
pCC = GetPrimaryComponent();
}
else if (lResultItemCookie == LVDATA_CUSTOMOCX) // => Custom OCX has focus
{
*ppDataObject = DOBJ_CUSTOMOCX;
pCC = GetPrimaryComponent();
}
else if (lResultItemCookie == LVDATA_CUSTOMWEB) // => Web has focus
{
*ppDataObject = DOBJ_CUSTOMWEB;
pCC = GetPrimaryComponent();
}
else if (lResultItemCookie == LVDATA_MULTISELECT) // => multi selection
{
// Do not calculate CComponent for multisel dataobject as there are multiple
// items and they can be from different snapins (so different components).
CViewData* pVD = GetViewData();
if (NULL == pVD)
return (sc = E_UNEXPECTED);
CMultiSelection* pMS = pVD->GetMultiSelection();
if (NULL == pMS)
return (sc = E_UNEXPECTED);
sc = pMS->GetMultiSelDataObject(ppDataObject);
if (sc)
return sc;
}
else // result item has focus
{
CViewData* pVD = GetViewData();
if (NULL == pVD)
return (sc = E_UNEXPECTED);
if (! pVD->IsVirtualList())
{
CResultItem* pri = CResultItem::FromHandle (lResultItemCookie);
if (pri != NULL)
{
bScopeItem = pri->IsScopeItem();
lResultItemCookie = pri->GetSnapinData();
if (! bScopeItem)
pCC = GetComponent(pri->GetOwnerID());
}
}
else
pCC = GetPrimaryComponent();
if (bScopeItem)
{
CNode* pNode = CNode::FromHandle((HNODE) lResultItemCookie);
CMTNode* pMTNode = pNode ? pNode->GetMTNode() : NULL;
if (NULL == pMTNode)
return (sc = E_UNEXPECTED);
if (ppOwnerNode)
*ppOwnerNode = pNode;
sc = pMTNode->QueryDataObject(CCT_SCOPE, &spDataObject);
if (sc)
return sc;
pCC = pNode->GetPrimaryComponent();
sc = ScCheckPointers(pCC, E_UNEXPECTED);
if (sc)
return sc;
}
else
{
if (NULL == pCC)
return (sc = E_UNEXPECTED);
sc = pCC->QueryDataObject(lResultItemCookie, CCT_RESULT, &spDataObject);
if (sc)
return sc;
}
}
// if required, get the component of this node
if (ppCComponent)
{
*ppCComponent = pCC;
sc = ScCheckPointers( *ppCComponent, E_UNEXPECTED );
if (sc)
return sc;
}
if (SUCCEEDED(sc.ToHr()) && *ppDataObject == NULL)
*ppDataObject = spDataObject.Detach();
return sc;
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScGetDropTargetDataObject
//
// Synopsis: Given the context get the data object that allows the
// to be drop target (allows paste).
//
// In MMC1.2 the drop target is always scope node. In MMC2.0
// it can be any non-virtual (??) result item. If the snapin
// has RVTI_LIST_OPTIONS_ALLOWPASTE set, then if the item selected
// is in result pane then data object corresponds to result item
// else it is scope item.
//
// Arguments:
// [bScopePane] - Scope or Result.
// [lResultItemCookie] - If Result pane is selected the item param.
// [ppDataObject] - The data-object (return val)
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScGetDropTargetDataObject(bool bScopePane, LPARAM lResultItemCookie, LPDATAOBJECT *ppDataObject)
{
DECLARE_SC(sc, _T("CNode::ScGetDropTargetDataObject"));
sc = ScCheckPointers(ppDataObject);
if (sc)
return sc;
*ppDataObject = NULL;
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pViewData, E_UNEXPECTED);
if (sc)
return sc;
if (pViewData->GetListOptions() & RVTI_LIST_OPTIONS_ALLOWPASTE)
{
bool bScopeItem;
// MMC2.0 use the given context.
sc = ScGetDataObject(bScopePane, lResultItemCookie, bScopeItem, ppDataObject);
if (sc)
return sc;
}
else
{
// MMC1.2 Always scope node.
sc = QueryDataObject(CCT_SCOPE, ppDataObject);
if (sc)
return sc;
}
return (sc);
}
/*+-------------------------------------------------------------------------*
*
* CNode::ScGetPropertyFromINodeProperties
*
* PURPOSE:
*
* PARAMETERS:
* BOOL bForScopeItem :
* LPARAM resultItemParam :
* BSTR bstrPropertyName :
* PBSTR pbstrPropertyValue :
*
* RETURNS:
* SC
*
*+-------------------------------------------------------------------------*/
SC
CNode::ScGetPropertyFromINodeProperties(LPDATAOBJECT pDataObject, BOOL bForScopeItem, LPARAM resultItemParam, BSTR bstrPropertyName, PBSTR pbstrPropertyValue)
{
//DECLARE_SC(sc, TEXT("CNode::ScGetPropertyFromINodeProperties"));
SC sc; // do not use DECLARE_SC here - want to silently ignore errors
sc = ScCheckPointers(pDataObject, bstrPropertyName, pbstrPropertyValue);
if(sc)
return sc;
if(bForScopeItem)
{
// get the MTNode
CMTNode * pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if(sc)
return sc;
// ask MTNode to get it
sc = pMTNode->ScGetPropertyFromINodeProperties(pDataObject, bstrPropertyName, pbstrPropertyValue);
if(sc)
return sc;
// done!
return sc;
}
// for result item.
CComponent *pComponent = GetPrimaryComponent();
sc = ScCheckPointers(pComponent);
if(sc)
{
SC scRet = sc; // returns but does not trace the error
sc.Clear();
return scRet;
}
// get the IComponent and QI for INodeProperties
INodePropertiesPtr spNodeProperties = pComponent->GetIComponent();
// at this point we should have a valid interface if it is supported
sc = ScCheckPointers(spNodeProperties, E_NOINTERFACE);
if(sc)
return sc;
sc = spNodeProperties->GetProperty(pDataObject, bstrPropertyName, pbstrPropertyValue);
return sc;
}
/*+-------------------------------------------------------------------------*
*
* CNode::ScExecuteShellCommand
*
* PURPOSE: Executes a shell command with the specified parameters in the
* specified directory with the correct window size
*
* PARAMETERS:
* BSTR Command :
* BSTR Directory :
* BSTR Parameters :
* BSTR WindowState :
*
* RETURNS:
* SC
*
*+-------------------------------------------------------------------------*/
SC
CNode::ScExecuteShellCommand(BSTR Command, BSTR Directory, BSTR Parameters, BSTR WindowState)
{
DECLARE_SC(sc, TEXT("CNode::ScExecuteShellCommand"));
sc = ScCheckPointers(Command, Directory, Parameters, WindowState);
if(sc)
return sc;
USES_CONVERSION;
CStr strParameters = W2T(Parameters);
CStr strWindowState= W2T(WindowState);
if(strWindowState.GetLength()==0)
strWindowState= XML_ENUM_WINDOW_STATE_RESTORED; // normal
SHELLEXECUTEINFO sei;
ZeroMemory (&sei, sizeof(sei));
sei.cbSize = sizeof(sei);
sei.lpFile = W2T(Command);
sei.lpParameters = strParameters;
sei.lpDirectory = W2T(Directory);
sei.fMask = SEE_MASK_NOCLOSEPROCESS | SEE_MASK_DOENVSUBST;
sei.nShow = (strWindowState == XML_ENUM_WINDOW_STATE_MAXIMIZED) ? SW_SHOWMAXIMIZED :
(strWindowState == XML_ENUM_WINDOW_STATE_MINIMIZED) ? SW_SHOWMINIMIZED :
SW_SHOWNORMAL ;
if (ShellExecuteEx(&sei))
CloseHandle (sei.hProcess);
else
sc = ScFromWin32((GetLastError ()));
return sc;
}
//############################################################################
//############################################################################
//
// Implementation of class COCX
//
//############################################################################
//############################################################################
DEBUG_DECLARE_INSTANCE_COUNTER(COCX);
//############################################################################
//############################################################################
//
// Implementation of class COCXNode
//
//############################################################################
//############################################################################
DEBUG_DECLARE_INSTANCE_COUNTER(COCXNode);
HRESULT CNode::InitComponents()
{
DECLARE_SC(sc, TEXT("CNode::InitComponents"));
if (m_bInitComponents == FALSE)
return S_OK;
HRESULT hr = S_OK;
// Initialize the component.
CMTNode * pMTNode = GetMTNode();
if (pMTNode == NULL)
return (E_UNEXPECTED);
// Ensure the master node is initialized.
if (!pMTNode->IsInitialized())
hr = pMTNode->Init();
if (FAILED(hr))
return hr;
CMTSnapInNode* pMTSnapInNode = pMTNode->GetStaticParent();
if (pMTSnapInNode == NULL)
return (E_UNEXPECTED);
HMTNODE hMTNode = CMTNode::ToHandle(pMTSnapInNode);
HNODE hNode = CNode::ToHandle(GetStaticParent());
CSnapIn* pSnapIn = pMTNode->GetPrimarySnapIn();
if (pSnapIn == NULL)
return (E_UNEXPECTED);
CComponentData* pCCD = pMTSnapInNode->GetComponentData(pSnapIn->GetSnapInCLSID());
if (pCCD == NULL)
return E_FAIL;
if (m_pPrimaryComponent == NULL)
m_pPrimaryComponent = pMTSnapInNode->GetComponent(GetViewID(),
pCCD->GetComponentID(), pSnapIn);
if(m_pPrimaryComponent == NULL)
return E_UNEXPECTED;
ASSERT(m_pPrimaryComponent != NULL);
//
// Init PRIMARY Component.
//
if (!m_pPrimaryComponent->IsInitialized())
{
ASSERT(pCCD->GetComponentID() == m_pPrimaryComponent->GetComponentID());
hr = m_pPrimaryComponent->Init(pCCD->GetIComponentData(), hMTNode, hNode,
pCCD->GetComponentID(), GetViewID());
// Abort if PRIMARY Component fails to init.
if (FAILED(hr))
return hr;
}
m_bInitComponents = FALSE;
//
// Now initalize the extension components. (create them if necessary)
//
// Get the node-type of this node
GUID guidNodeType;
//hr = pCCD->GetNodeType(pMTNode->GetUserParam(), &guidNodeType);
hr = pMTNode->GetNodeType(&guidNodeType);
CHECK_HRESULT(hr);
if (FAILED(hr))
return hr;
LPCLSID pDynExtCLSID;
int cDynExt = pMTNode->GetDynExtCLSID(&pDynExtCLSID);
CExtensionsIterator it;
// TODO: try to use the easier form of it.ScInitialize()
sc = it.ScInitialize(pSnapIn, guidNodeType, g_szNameSpace, pDynExtCLSID, cDynExt);
if(sc)
return S_FALSE;
BOOL fProblem = FALSE;
for (; it.IsEnd() == FALSE; it.Advance())
{
pCCD = pMTSnapInNode->GetComponentData(it.GetCLSID());
if (pCCD == NULL)
continue;
CComponent* pCC = pMTSnapInNode->GetComponent(GetViewID(),
pCCD->GetComponentID(), pCCD->GetSnapIn());
if (pCC->IsInitialized() == TRUE)
continue;
hr = pCC->Init(pCCD->GetIComponentData(), hMTNode, hNode,
pCCD->GetComponentID(), GetViewID());
CHECK_HRESULT(hr);
if (FAILED(hr))
fProblem = TRUE; // Continue even on error.
}
if (fProblem == TRUE)
{
// TODO: Put up an error message.
hr = S_FALSE;
}
return hr;
}
/*+-------------------------------------------------------------------------*
*
* CNode::OnInitOCX
*
* PURPOSE: Sends the MMCN_INITOCX notification when an OCX is created.
*
* PARAMETERS:
* IUnknown* pUnk :
*
* RETURNS:
* HRESULT
*
*+-------------------------------------------------------------------------*/
HRESULT
CNode::OnInitOCX(IUnknown* pUnk)
{
DECLARE_SC(sc, TEXT("CNode::OnInitOCX"));
IDataObjectPtr spdtobj;
sc = QueryDataObject(CCT_SCOPE, &spdtobj);
if(sc)
return sc.ToHr();
CComponent* pCC = GetPrimaryComponent();
sc = ScCheckPointers(pCC);
if(sc)
return sc.ToHr();
sc = pCC->Notify(spdtobj, MMCN_INITOCX, 0, reinterpret_cast<LPARAM>(pUnk));
sc.Clear(); // must ignore errors here - Disk management returns E_UNEXPECTED.!
return sc.ToHr();
}
HRESULT CNode::OnCacheHint(int nStartIndex, int nEndIndex)
{
CComponent* pCC = GetPrimaryComponent();
ASSERT(pCC != NULL);
if (pCC == NULL)
return E_FAIL;
IResultOwnerDataPtr spIResultOwnerData = pCC->GetIComponent();
if (spIResultOwnerData == NULL)
return S_FALSE;
return spIResultOwnerData->CacheHint(nStartIndex, nEndIndex);
}
/***************************************************************************\
*
* METHOD: CNode::ScInitializeViewExtension
*
* PURPOSE: Sets callback representing view extension
*
* PARAMETERS:
* const CLSID& clsid - [in] view extension CLSID
* CViewData *pViewData - [in] view data
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::ScInitializeViewExtension(const CLSID& clsid, CViewData *pViewData)
{
DECLARE_SC(sc, TEXT("CNode::ScInitializeViewExtension"));
sc = ScCheckPointers(pViewData);
if (sc)
return sc;
/*
* Get the CConsoleTaskpad that goes with it this view extension.
* If this is an ordinary (i.e. not taskpad) view extension,
* ScGetViewExtensionTaskpad will set pConsoleTaskpad to NULL.
*/
CConsoleTaskpad* pConsoleTaskpad = NULL;
sc = CConsoleTaskpadViewExtension::ScGetViewExtensionTaskpad (this, clsid, pConsoleTaskpad);
if (sc)
return (sc);
typedef CComObject<CConsoleTaskCallbackImpl> t_ViewExtensionCallbackImpl;
t_ViewExtensionCallbackImpl* pViewExtensionCallbackImpl = NULL;
sc = t_ViewExtensionCallbackImpl::CreateInstance(&pViewExtensionCallbackImpl);
if (sc)
return sc;
// recheck pointer
sc = ScCheckPointers(pViewExtensionCallbackImpl, E_UNEXPECTED);
if (sc)
return sc;
pViewData->m_spTaskCallback = pViewExtensionCallbackImpl; // this addrefs/releases the object.
/*
* If this is a taskpad, initialize the view extension callback as
* a taskpad view extension. Otherwise initialize it as an ordinary
* view extension.
*/
if (pConsoleTaskpad != NULL)
{
sc = pViewExtensionCallbackImpl->ScInitialize (pConsoleTaskpad,
CScopeTree::GetScopeTree(),
this);
}
else
sc = pViewExtensionCallbackImpl->ScInitialize(clsid);
return sc;
}
/*+-------------------------------------------------------------------------*
*
* CNode::ScSetViewExtension
*
* PURPOSE: Forces a display of the given view extension.
*
* PARAMETERS:
* GUID * pGuidViewId : [IN]: The view extension to display.
* bool bUseDefaultTaskpad : [IN}
* bool bSetViewSettingDirty : [IN] (See below notes)
*
* Note:
* The view-extension-ID comes from
*
* 1. Viewsetting if one exists.
* 2. Given by CONUI when user changes tab for different taskpad.
* 3. There are cases in which a new view-extension installed (after the console
* file was created) that will be default. (This will be calculated in this method).
*
* In cases 1 & 3 viewsetting should not be made dirty.
*
* RETURNS:
* SC
*
*+-------------------------------------------------------------------------*/
SC
CNode::ScSetViewExtension(GUID *pGuidViewId, bool bUseDefaultTaskpad, bool bSetViewSettingDirty)
{
DECLARE_SC(sc, TEXT("CNode::ScSetViewExtension"));
CViewData * pViewData = GetViewData();
sc = ScCheckPointers(pGuidViewId, pViewData);
if(sc)
return sc;
// collect the view extensions
CViewExtCollection vecExtensions;
CViewExtInsertIterator itExtensions(vecExtensions, vecExtensions.begin());
sc = ScGetViewExtensions(itExtensions);
if (sc)
sc.Trace_();
if ( bUseDefaultTaskpad )
{
// setup proper taskpad (tab to be selected)
CViewExtCollection::iterator it = vecExtensions.begin();
if (it != vecExtensions.end())
*pGuidViewId = it->viewID; // first one if such exist
else
*pGuidViewId = GUID_NULL; // default
}
else // locate the extension we need to select
{
// see if the extension really exist
CViewExtCollection::iterator it = vecExtensions.begin();
bool bDefaultIsReplaced = false;
while (it != vecExtensions.end() && !IsEqualGUID(*pGuidViewId, it->viewID) )
{
bDefaultIsReplaced = bDefaultIsReplaced || it->bReplacesDefaultView;
++it;
}
// found it?
bool bFound = (it != vecExtensions.end());
// one more chance - we were looking for default and one will be added!
bFound = bFound || ( IsEqualGUID( *pGuidViewId, GUID_NULL ) && !bDefaultIsReplaced );
if ( !bFound )
{
sc = E_FAIL;
}
}
if (sc) // extension missing! need to find the substitute
{
sc.Clear(); // ignore error
// default to first extension or NORMAL view here
CViewExtCollection::iterator it = vecExtensions.begin();
if (it != vecExtensions.end())
*pGuidViewId = it->viewID; // first available
else
*pGuidViewId = GUID_NULL; // "normal" if it's the only choice
}
// set the view extension if one really exist
if (*pGuidViewId != GUID_NULL)
{
sc = ScInitializeViewExtension(*pGuidViewId, GetViewData());
if (sc)
sc.TraceAndClear(); // ignore and proceed
}
else
{
pViewData->m_spTaskCallback = NULL;
}
sc = ScSetTaskpadID(*pGuidViewId, bSetViewSettingDirty);
if (sc)
return sc;
return sc;
}
//+-------------------------------------------------------------------
//
// Member: ScGetResultPane
//
// Synopsis: Get the result pane data from snapin or persisted data.
//
// Arguments: [strResultPane] - Result pane name (If it is OCX/WEB)
// [pViewOptions] - View options.
// [pGuidTaskpadID] - If there is a task-pad the ID.
//
// Returns: SC
//
// History: 04-29-1999 AnandhaG Created
//
//--------------------------------------------------------------------
SC
CNode::ScGetResultPane(CResultViewType &rvt, GUID *pGuidTaskpadID)
{
DECLARE_SC(sc, TEXT("CNode::ScGetResultPane"));
sc = ScCheckPointers(pGuidTaskpadID);
if (sc)
return sc;
CComponent *pComponent = GetPrimaryComponent();
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pComponent, pViewData, E_UNEXPECTED);
if(sc)
return sc;
IComponent* pIComponent = pComponent->GetIComponent();
sc = ScCheckPointers(pIComponent, E_FAIL);
if (sc)
return sc;
// 1. Setup any persisted/default console taskpads or view extensions.
sc = ScSetupTaskpad(pGuidTaskpadID);
if (sc)
return sc;
// 2. Get the persisted CResultViewType information.
sc = ScGetResultViewType(rvt);
if (sc)
return sc;
bool bResultViewDataIsPersisted = (sc == S_OK);
bool bSnapinChangingView = pViewData->IsSnapinChangingView();
CResultViewType rvtOriginal;
CStr strResultPane = _T(""); // init
// 3. If there is persisted result-view-type data then ask the snapin if it
// wants to restore the result-view with this data. If snapin is changing
// its view (by re-selection of node) then dont ask this question.
if (!bSnapinChangingView && bResultViewDataIsPersisted )
{
// 3.a) Ask snapin if it wants to restore the result-view with persisted data.
sc = ScRestoreResultView(rvt);
if (sc)
return sc;
if (S_OK == sc.ToHr()) // snapin accepted the resultviewtype settings so return.
return sc;
// 3.b) Snapin refused the persisted CResultViewType data so...
// cache the data to see if we need to modify the settings
rvtOriginal = rvt;
// Throw away the data as it is not accepted by snapin.
sc = rvt.ScReset();
if (sc)
return sc;
}
// 4. Ask the snapin for result-view-type data.
IComponent2Ptr spIComponent2 = pIComponent;
if(spIComponent2 != NULL)
{
// should be able to move all this to a separate function.
RESULT_VIEW_TYPE_INFO rvti;
ZeroMemory(&rvti, sizeof(rvti));
// the snapin supports IComponent2. Use it to get the result view type.
sc = spIComponent2->GetResultViewType2(GetUserParam(), &rvti);
if(sc)
return sc;
// at this point, we have a valid RESULT_VIEW_TYPE_INFO structure. Initialize the contents into rvt, which zeros out the structure
// and releases all the allocated strings
sc = rvt.ScInitialize(rvti);
if(sc)
return sc;
}
else
{
// the snapin does not support IComponent2. Use IComponent to
// get the result view type from the snapin.
LPOLESTR pszView = NULL;
long lViewOptions = 0;
sc = pIComponent->GetResultViewType(GetUserParam(), &pszView, &lViewOptions);
if(sc)
return sc;
sc = rvt.ScInitialize(pszView, lViewOptions); // this also calls CoTaskMemFree on pszView
if(sc)
return sc;
}
/*
* 5. Persist ResultViewType information only if
* a. Snapin is changing the view OR
* b. Snapin rejected the persisted view setting (we already
* made sure it is not changing the view above) and new view setting
* given is different from original one.
*/
if ( bSnapinChangingView ||
(bResultViewDataIsPersisted && (rvtOriginal != rvt)) )
{
sc = ScSetResultViewType(rvt);
if (sc)
return sc;
}
return sc;
}
//+-------------------------------------------------------------------
//
// Member: ScRestoreResultView
//
// Synopsis: Restore the result pane from given persisted data.
//
// Arguments: [rvt] - CResultViewType data used to restore result pane.
//
// Returns: SC, S_OK if successfully restored.
// S_FALSE if snapin refused to restore.
//
// History: 04-29-1999 AnandhaG Created
//
//--------------------------------------------------------------------
SC CNode::ScRestoreResultView(const CResultViewType& rvt)
{
DECLARE_SC(sc, _T("CNode::ScRestoreResultView"));
CComponent* pComponent = GetPrimaryComponent();
sc = ScCheckPointers(pComponent, E_UNEXPECTED);
if (sc)
return sc;
IComponent2Ptr spIComponent2 = pComponent->GetIComponent();
if( (spIComponent2 != NULL) && (!rvt.IsMMC12LegacyData()))
{
RESULT_VIEW_TYPE_INFO rvti;
ZeroMemory(&rvti, sizeof(rvti));
sc = rvt.ScGetResultViewTypeInfo (rvti);
if (sc)
return sc;
// the snapin supports IComponent2. Use it to get the result view type.
sc = spIComponent2->RestoreResultView(GetUserParam(), &rvti);
if(sc)
{
// If snapin returns error, trace it and translate it to S_FALSE (snapin refused to restore).
TraceSnapinError(TEXT("Snapin returned error from IComponent2::RestoreResultView"), sc);
sc = S_FALSE;
return sc;
}
}
else
{
// the snapin does not support IComponent2. Use IComponent to
// to restore the result view.
LPCOLESTR pszView = NULL;
long lViewOptions = 0;
sc = rvt.ScGetOldTypeViewOptions(&lViewOptions);
if (sc)
return sc;
IDataObjectPtr spdtobj;
sc = QueryDataObject(CCT_SCOPE, &spdtobj);
if (sc)
return sc;
// Notify MMC of persisted view being restored.
MMC_RESTORE_VIEW mrv;
::ZeroMemory(&mrv, sizeof(mrv));
mrv.cookie = GetUserParam();
mrv.dwSize = sizeof(mrv);
mrv.lViewOptions = lViewOptions;
if (rvt.HasOCX())
{
pszView = rvt.GetOCX();
}
else if (rvt.HasWebBrowser())
{
pszView = rvt.GetURL();
}
if (pszView)
{
int cchViewType = wcslen(pszView) + 1;
mrv.pViewType = (LPOLESTR)CoTaskMemAlloc( cchViewType * sizeof(OLECHAR) );
sc = ScCheckPointers(mrv.pViewType, E_OUTOFMEMORY);
if (sc)
return sc;
sc = StringCchCopyW(mrv.pViewType, cchViewType, pszView);
if(sc)
return sc;
pszView = NULL; // Dont want to abuse it later.
}
// If the snapin handles this notification we use the persisted
// data else call GetResultViewType of the snapin.
BOOL bHandledRestoreView = FALSE;
pComponent->Notify(spdtobj, MMCN_RESTORE_VIEW, (LPARAM)&mrv, (LPARAM)&bHandledRestoreView);
CoTaskMemFree(mrv.pViewType);
sc = (bHandledRestoreView) ? S_OK : S_FALSE;
}
return sc;
}
//+-------------------------------------------------------------------
//
// Member: ScRestoreViewMode
//
// Synopsis: If the view mode is persisted restore it.
//
// Arguments: None.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScRestoreViewMode()
{
DECLARE_SC(sc, _T("CNode::ScRestoreViewMode"));
ULONG ulViewMode = 0;
sc = ScGetViewMode(ulViewMode);
if (sc != S_OK) // data not found or some error.
return sc;
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pViewData, E_UNEXPECTED);
if (sc)
return sc;
// tell conui to change the list mode.
CConsoleView* pConsoleView = pViewData->GetConsoleView();
sc = ScCheckPointers(pConsoleView, E_UNEXPECTED);
if (sc)
return sc;
sc = pConsoleView->ScChangeViewMode (ulViewMode);
if (sc)
return sc;
return sc;
}
/*+-------------------------------------------------------------------------*
*
* CNode::ScSetupTaskpad
*
* PURPOSE:
*
* PARAMETERS:
* GUID * pTaskpadID : [OUT]: The guid of the taskpad, else GUID_NULL
*
* RETURNS:
* SC
*
*+-------------------------------------------------------------------------*/
SC
CNode::ScSetupTaskpad(GUID *pGuidTaskpadID)
{
DECLARE_SC(sc, _T("CNode::SetupTaskpad"));
sc = ScCheckPointers(pGuidTaskpadID);
if (sc)
return sc.ToHr();
// this is the case when the document is closed. Do nothing
// VERY IMPORTANT - don't remove. Several functions down the line
// will barf and eventually cause a dialog to be displayed.
if(!m_pViewSettingsPersistor)
return sc = S_FALSE;
*pGuidTaskpadID = GUID_NULL;
// Get the persisted taskpad id if there is one.
sc = ScGetTaskpadID(*pGuidTaskpadID);
if (sc)
return sc;
// restore the taskpad if we've got a ViewSettings object.
// do not use default tab even in case view seting does not have a valid guid
// it only means the "Default" tab needs to be selected
// see bug #97001 - MMC does not persist select CTP when user returns to a node
bool bUseDefaultTaskpad = ( sc == S_FALSE );
// See ScSetViewExtension for parameter meaning.
sc = ScSetViewExtension(pGuidTaskpadID, bUseDefaultTaskpad, /*bSetViewSettingDirty*/ false);
return sc.ToHr();
}
/*+-------------------------------------------------------------------------*
*
* CNode::ShowStandardListView
*
* PURPOSE:
*
* RETURNS:
* HRESULT
*
*+-------------------------------------------------------------------------*/
HRESULT CNode::ShowStandardListView()
{
CComponent* pCC = GetPrimaryComponent();
ASSERT(pCC != NULL);
if (pCC == NULL)
return E_FAIL;
IDataObjectPtr spDataObject = NULL;
HRESULT hr = QueryDataObject(CCT_SCOPE, &spDataObject);
ASSERT(SUCCEEDED(hr));
if (FAILED(hr))
return hr;
IExtendContextMenuPtr spIExtendContextMenu = pCC->GetIComponent();
if(!spIExtendContextMenu.GetInterfacePtr())
return S_FALSE;
hr = spIExtendContextMenu->Command(MMCC_STANDARD_VIEW_SELECT, spDataObject);
return hr;
}
HRESULT
CNode::OnListPad(LONG_PTR arg, LPARAM param)
{
HRESULT hr = S_OK;
IDataObjectPtr spdtobj;
hr = QueryDataObject(CCT_SCOPE, &spdtobj);
ASSERT(SUCCEEDED(hr));
if (SUCCEEDED(hr))
{
CComponent* pCC = GetPrimaryComponent();
ASSERT(pCC != NULL);
hr = pCC->Notify(spdtobj, MMCN_LISTPAD, arg, param);
CHECK_HRESULT(hr);
}
return hr;
}
HRESULT
CNode::OnGetPrimaryTask(IExtendTaskPad **ppExtendTaskPad)
{
HRESULT hr = S_OK;
IExtendTaskPadPtr spExtendTaskPad = GetPrimaryComponent()->GetIComponent();
if (spExtendTaskPad == NULL)
return E_NOINTERFACE;
*ppExtendTaskPad = spExtendTaskPad.Detach();
return hr;
}
IFramePrivate *
CNode::GetIFramePrivate()
{
CComponent* pCC = GetPrimaryComponent();
if (pCC == NULL)
return (NULL);
IFramePrivate* pFramePrivate = pCC->GetIFramePrivate();
ASSERT (pFramePrivate != NULL);
return (pFramePrivate);
}
HRESULT
CNode::GetTaskEnumerator(LPOLESTR pszTaskGroup, IEnumTASK** ppEnumTask)
{
DECLARE_SC(sc, TEXT("CNode::GetTaskEnumerator"));
ASSERT(pszTaskGroup != NULL);
ASSERT(ppEnumTask != NULL);
if (!pszTaskGroup|| !ppEnumTask)
return E_INVALIDARG;
*ppEnumTask = NULL; // init
if (GetPrimaryComponent() == NULL)
{
ASSERT(0 && "UNexpected");
return S_FALSE;
}
CMTNode* pMTNode = GetMTNode();
ASSERT(pMTNode != NULL);
CMTSnapInNode* pMTSnapIn = pMTNode->GetStaticParent();
ASSERT(pMTSnapIn != NULL);
CComponentData* pComponentData = pMTNode->GetPrimaryComponentData();
ASSERT(pComponentData != NULL);
GUID guidNodeType;
HRESULT hr = pMTNode->GetNodeType(&guidNodeType);
ASSERT(SUCCEEDED(hr));
if (FAILED(hr))
return hr;
//
// Add primary task pad.
//
IExtendTaskPadPtr spExtendTaskPad =
GetPrimaryComponent()->GetIComponent();
if (spExtendTaskPad == NULL)
return S_FALSE;
IDataObjectPtr spDataObject;
hr = pMTNode->QueryDataObject(CCT_SCOPE, &spDataObject);
ASSERT(SUCCEEDED(hr));
if (FAILED(hr))
return hr;
IEnumTASKPtr spEnumTASK;
hr = spExtendTaskPad->EnumTasks(spDataObject, pszTaskGroup, &spEnumTASK);
ASSERT(SUCCEEDED(hr));
if (FAILED(hr))
return hr;
CComObject<CTaskEnumerator>* pTaskEnumerator = new CComObject<CTaskEnumerator>;
ASSERT(pTaskEnumerator != NULL);
pTaskEnumerator->AddTaskEnumerator(pComponentData->GetCLSID(), spEnumTASK);
IEnumTASKPtr spEnumTask = pTaskEnumerator;
ASSERT(spEnumTask != NULL);
if (spEnumTask)
*ppEnumTask = spEnumTask.Detach();
//
// Add extension task pads.
//
CArray<GUID,GUID&> DynExtens;
ExtractDynExtensions(spDataObject, DynExtens);
CExtensionsIterator it;
sc = it.ScInitialize(pComponentData->GetSnapIn(), guidNodeType, g_szTask, DynExtens.GetData(), DynExtens.GetSize());
if (sc.IsError() || it.IsEnd() == TRUE)
return S_OK;
for (; it.IsEnd() == FALSE; it.Advance())
{
CComponentData* pCCD = pMTSnapIn->GetComponentData(it.GetCLSID());
if (pCCD == NULL)
{
// See if taskpad extension supports IComponentData. If so, we will add the it to
// the static node's component list and reuse the same instance each time its needed.
IComponentDataPtr spIComponentData;
hr = CreateSnapIn(it.GetCLSID(), &spIComponentData, FALSE);
if (SUCCEEDED(hr))
{
CSnapInPtr spSnapIn;
// If a dynamic extension, we have to get the snap-in ourselves
// otherwise the iterator has it
if (it.IsDynamic())
{
CSnapInsCache* const pCache = theApp.GetSnapInsCache();
ASSERT(pCache != NULL);
SC sc = pCache->ScGetSnapIn(it.GetCLSID(), &spSnapIn);
ASSERT(!sc.IsError());
// On failure, continue with other extensions
if (sc)
continue;
}
else
{
spSnapIn = it.GetSnapIn();
}
ASSERT(spSnapIn != NULL);
pCCD = new CComponentData(spSnapIn);
if (pCCD != NULL)
{
pCCD->SetIComponentData(spIComponentData);
pMTSnapIn->AddComponentDataToArray(pCCD);
}
}
}
// Initialize and load component data if not already done
if (pCCD != NULL && pCCD->IsInitialized() == FALSE)
{
sc = pCCD->Init(CMTNode::ToHandle(pMTSnapIn));
if ( !sc.IsError() )
{
sc = pMTSnapIn->ScInitIComponentData(pCCD);
if (sc)
{
sc.TraceAndClear();
// On failure, continue with other extensions
continue;
}
}
else
{
// if failed to initialize, remove it from the component data array
pMTSnapIn->CompressComponentDataArray();
sc.TraceAndClear();
// On failure, continue with other extensions
continue;
}
}
IExtendTaskPadPtr spExtendTaskPad;
if (pCCD)
{
CComponent* pCC = pMTSnapIn->GetComponent(GetViewID(),
pCCD->GetComponentID(), pCCD->GetSnapIn());
ASSERT(pCC != NULL);
if (pCC)
{
// Ensure the IComponent is initialized.
if (!pCC->IsInitialized())
{
ASSERT(pCCD->GetComponentID() == pCC->GetComponentID());
hr = pCC->Init(pCCD->GetIComponentData(),
CMTNode::ToHandle(pMTSnapIn),
ToHandle(this),
pCCD->GetComponentID(),
GetViewID());
// Abort if PRIMARY Component fails to init.
if (FAILED(hr))
return hr;
}
spExtendTaskPad = pCC->GetIComponent();
}
}
else
{
hr = spExtendTaskPad.CreateInstance(it.GetCLSID(),
#if _MSC_VER >= 1100
NULL,
#endif
MMC_CLSCTX_INPROC);
ASSERT(SUCCEEDED(hr));
if (FAILED(hr))
continue;
}
if (spExtendTaskPad != NULL)
{
IEnumTASKPtr spEnumTASK;
HRESULT hr = spExtendTaskPad->EnumTasks(spDataObject, pszTaskGroup,
&spEnumTASK);
ASSERT(SUCCEEDED(hr));
if (hr == S_OK)
pTaskEnumerator->AddTaskEnumerator(it.GetCLSID(), spEnumTASK);
}
} // end for
// Return S_OK rather than hr because a failed extension shouldn't prevent the
// taskpad from coming up
return S_OK;
}
HRESULT
CNode::GetListPadInfo(IExtendTaskPad* pExtendTaskPad, LPOLESTR szTaskGroup,
MMC_ILISTPAD_INFO* pIListPadInfo)
{
if ((GetPrimaryComponent() == NULL) )
{
ASSERT(0 && "Asking for ListPadInfo on a node that has no snapin");
return S_FALSE;
}
// get primary snapin's IComponentData...
CMTNode* pMTNode = GetMTNode();
ASSERT(pMTNode != NULL);
CComponentData* pComponentData = pMTNode->GetPrimaryComponentData();
ASSERT(pComponentData != NULL);
// ... so we can get CLSID
pIListPadInfo->szClsid = NULL;
HRESULT hr = StringFromCLSID (pComponentData->GetCLSID(), &pIListPadInfo->szClsid);
ASSERT (pIListPadInfo->szClsid != NULL);
if (pIListPadInfo->szClsid == NULL) {
if (hr) return hr;
else return E_FAIL; // just in case.
}
// finally call taskpad extension for info
return pExtendTaskPad->GetListPadInfo (szTaskGroup, (MMC_LISTPAD_INFO*)pIListPadInfo);
}
void
CNode::OnTaskNotify(LONG_PTR arg, LPARAM param)
{
CSnapInNode* pSINode = dynamic_cast<CSnapInNode*>(GetStaticParent());
ASSERT(pSINode != NULL);
IDataObjectPtr spDataObject;
QueryDataObject(CCT_SCOPE, &spDataObject);
IExtendTaskPadPtr spExtendTaskPad;
CComponent* pCC;
LPOLESTR pszClsid = reinterpret_cast<LPOLESTR>(arg);
if (pszClsid[0] == 0)
{
pCC = GetPrimaryComponent();
if (!pCC)
return;
spExtendTaskPad = pCC->GetIComponent();
}
else
{
CLSID clsid;
HRESULT hr = ::CLSIDFromString(pszClsid, &clsid);
ASSERT(SUCCEEDED(hr));
if (FAILED(hr))
return;
// try to get IExtendTaskPad from IComponent, first;
// if that fails, just create using CLSID
pCC = pSINode->GetComponent(const_cast<const CLSID&>(clsid));
if (pCC)
spExtendTaskPad = pCC->GetIComponent();
if (spExtendTaskPad == NULL)
hr = spExtendTaskPad.CreateInstance(clsid,
#if _MSC_VER >= 1100
NULL,
#endif
MMC_CLSCTX_INPROC);
}
ASSERT (spExtendTaskPad != NULL);
if (spExtendTaskPad != NULL)
{
VARIANT** ppvarg = reinterpret_cast<VARIANT**>(param);
spExtendTaskPad->TaskNotify(spDataObject, ppvarg[0], ppvarg[1]);
}
}
HRESULT
CNode::OnScopeSelect(bool bSelect, SELECTIONINFO* pSelInfo)
{
DECLARE_SC (sc, _T("CNode::OnScopeSelect"));
sc = ScCheckPointers(pSelInfo);
if (sc)
return sc.ToHr();
/*
* Bug 178484: reset the sort parameters when scope selection changes
*/
if (bSelect)
{
CComponent *pCC = GetPrimaryComponent();
sc = ScCheckPointers(pCC, E_UNEXPECTED);
if (sc)
return sc.ToHr();
IFramePrivate *pFrame = pCC->GetIFramePrivate();
sc = ScCheckPointers(pFrame, E_UNEXPECTED);
if (sc)
return sc.ToHr();
pFrame->ResetSortParameters();
}
if (bSelect == TRUE && WasExpandedAtLeastOnce() == FALSE)
{
sc = OnExpand(TRUE);
if (sc)
return (sc.ToHr());
}
ASSERT(IsInitialized() == TRUE);
sc = OnSelect(pSelInfo->m_pView, bSelect, pSelInfo->m_bResultPaneIsWeb);
if (sc)
return (sc.ToHr());
return (sc.ToHr());
}
HRESULT CNode::OnActvate(LONG_PTR lActivate)
{
return (DeepNotify (MMCN_ACTIVATE, lActivate, 0));
}
HRESULT CNode::OnMinimize(LONG_PTR fMinimized)
{
return (DeepNotify (MMCN_MINIMIZED, fMinimized, 0));
}
//+-------------------------------------------------------------------
//
// Member: SendShowEvent
//
// Synopsis: Send MMCN_SHOW notification to snapin, persist column
// data if necessary.
//
// Arguments: [bSelect] - TRUE if the node is selected.
//
//--------------------------------------------------------------------
HRESULT CNode::SendShowEvent(BOOL bSelect)
{
DECLARE_SC(sc, _T("CNode::SendShowEvent"));
CComponent* pCC = m_pPrimaryComponent;
ASSERT(pCC != NULL);
// Get the data object for the node and pass it to the primary snap-in
// and all the namespace extensions to the node.
IDataObjectPtr spDataObject;
HRESULT hr = QueryDataObject(CCT_SCOPE, &spDataObject);
if (FAILED(hr))
return hr;
CMTNode* pMTNode = GetMTNode();
IFramePrivatePtr spFrame = pCC->GetIFramePrivate();
sc = ScCheckPointers(spFrame, E_UNEXPECTED);
if (sc)
return sc.ToHr();
IImageListPrivatePtr spImageList;
hr = spFrame->QueryResultImageList(reinterpret_cast<LPIMAGELIST*>(&spImageList));
ASSERT(SUCCEEDED(hr));
ASSERT(spImageList != NULL);
HSCOPEITEM hScopeItem = CMTNode::ToScopeItem(pMTNode);
if (bSelect == TRUE)
{
hr = pCC->Notify(spDataObject, MMCN_ADD_IMAGES,
reinterpret_cast<LPARAM>((LPIMAGELIST)spImageList),
hScopeItem);
CHECK_HRESULT(hr);
//if (FAILED(hr))
// return hr;
}
hr = pCC->Notify(spDataObject, MMCN_SHOW, bSelect, hScopeItem);
CHECK_HRESULT(hr);
if (FAILED(hr))
return hr;
if (bSelect)
{
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pViewData, E_UNEXPECTED);
if (sc)
return sc.ToHr();
if (pViewData->HasList() || pViewData->HasListPad())
{
sc = ScRestoreSortFromPersistedData();
if (sc)
return sc.ToHr();
// Now try to restore the view mode.
sc =ScRestoreViewMode();
if (sc)
return sc.ToHr();
}
}
return hr;
}
HRESULT CNode::DeepNotify(MMC_NOTIFY_TYPE event, LONG_PTR arg, LPARAM param)
{
DECLARE_SC(sc, TEXT("CNode::DeepNotify"));
CComponent* pCC = m_pPrimaryComponent;
ASSERT(pCC != NULL);
if (pCC == NULL)
return E_UNEXPECTED;
// Get the data object for the node and pass it to the primary snap-in
// and all the namespace extensions to the node.
IDataObjectPtr spDataObject;
HRESULT hr = QueryDataObject(CCT_SCOPE, &spDataObject);
if (FAILED(hr))
return hr;
hr = pCC->Notify(spDataObject, event, arg, param);
CHECK_HRESULT(hr);
//if (FAILED(hr))
// return hr;
//
// Notify extensions.
//
CMTNode* pMTNode = GetMTNode();
// Get the node-type of this node
GUID guidNodeType;
hr = pMTNode->GetNodeType(&guidNodeType);
CHECK_HRESULT(hr);
if (FAILED(hr))
return hr;
LPCLSID pDynExtCLSID;
int cDynExt = pMTNode->GetDynExtCLSID(&pDynExtCLSID);
CExtensionsIterator it;
sc = it.ScInitialize(pMTNode->GetPrimarySnapIn(), guidNodeType, g_szNameSpace, pDynExtCLSID, cDynExt);
if (sc)
return S_FALSE;
BOOL fProblem = FALSE;
CSnapInNode* pSINode = GetStaticParent();
for (; it.IsEnd() == FALSE; it.Advance())
{
CComponent* pCC = pSINode->GetComponent(it.GetCLSID());
if (pCC == NULL)
continue;
hr = pCC->Notify(spDataObject, event, arg, param);
CHECK_HRESULT(hr);
// continue even if an error occurs with extension snapins
if (FAILED(hr))
fProblem = TRUE;
}
return (fProblem == TRUE) ? S_FALSE : S_OK;
}
HRESULT CNode::OnSelect(LPUNKNOWN lpView, BOOL bSelect,
BOOL bResultPaneIsWeb)
{
DECLARE_SC(sc, TEXT("CNode::OnSelect"));
#ifdef DBG
if (lpView == NULL)
ASSERT(bSelect == FALSE);
else
ASSERT(bSelect == TRUE);
#endif
sc = ScCheckPointers(m_pPrimaryComponent, E_UNEXPECTED);
if (sc)
{
sc.TraceAndClear();
return sc.ToHr();
}
CComponent* pCC = m_pPrimaryComponent;
sc = ScCheckPointers(pCC, E_UNEXPECTED);
if (sc)
return sc.ToHr();
IFramePrivate *pFrame = pCC->GetIFramePrivate();
sc = ScCheckPointers(pFrame, E_UNEXPECTED);
if (sc)
return sc.ToHr();
// set the correct view in the primary snap-in before it adds items
if (bSelect == TRUE)
pFrame->SetResultView(lpView);
IDataObjectPtr spDataObject;
sc = QueryDataObject(CCT_SCOPE, &spDataObject);
if (sc)
return sc.ToHr();
CMTNode* pMTNode = GetMTNode();
LPARAM hScopeItem = CMTNode::ToScopeItem(pMTNode);
// Send only the MMCN_SHOW message if result pane is the web view.
if (bResultPaneIsWeb)
{
return pCC->Notify(spDataObject, MMCN_SHOW, bSelect, hScopeItem);
}
// Send necessary events like MMCN_ADD_IMAGES and MMCN_SHOW to snapin
sc = SendShowEvent(bSelect);
if (sc)
return sc.ToHr();
// set the correct view in the primary snap-in after it's notified
if (bSelect == FALSE)
pFrame->SetResultView(NULL); //
// Deal with extension ssnap-ins
//
// Get the node-type of this node
GUID guidNodeType;
sc = pMTNode->GetNodeType(&guidNodeType);
if (sc)
return sc.ToHr();
LPCLSID pDynExtCLSID;
int cDynExt = pMTNode->GetDynExtCLSID(&pDynExtCLSID);
CExtensionsIterator it;
sc = it.ScInitialize(pMTNode->GetPrimarySnapIn(), guidNodeType, g_szNameSpace, pDynExtCLSID, cDynExt);
if (sc)
return S_FALSE;
BOOL fProblem = FALSE;
CSnapInNode* pSINode = GetStaticParent();
for (; it.IsEnd() == FALSE; it.Advance())
{
CComponent* pCCExtnSnapin = pSINode->GetComponent(it.GetCLSID());
if (pCCExtnSnapin == NULL)
continue;
IFramePrivate *pFrameExtnSnapin = pCCExtnSnapin->GetIFramePrivate();
sc = ScCheckPointers(pFrameExtnSnapin, E_UNEXPECTED);
if (sc)
{
sc.TraceAndClear();
continue;
}
// set the correct view in the snap-in before it adds items
if (bSelect == FALSE)
{
pFrameExtnSnapin->SetResultView(NULL);
continue;
}
else
{
pFrameExtnSnapin->SetResultView(lpView);
IImageListPrivatePtr spImageList;
sc = pCCExtnSnapin->GetIFramePrivate()->QueryResultImageList(
reinterpret_cast<LPIMAGELIST*>(&spImageList));
if (sc)
{
sc.TraceAndClear();
fProblem = TRUE;
continue;
}
sc = ScCheckPointers(spImageList, E_UNEXPECTED);
if (sc)
{
sc.TraceAndClear();
fProblem = TRUE;
continue;
}
SC scNoTrace = pCCExtnSnapin->Notify(spDataObject, MMCN_ADD_IMAGES,
reinterpret_cast<LPARAM>((LPIMAGELIST)spImageList),
hScopeItem);
if (scNoTrace)
{
TraceSnapinError(TEXT("Snapin returned error from IComponent::Notify MMCN_ADD_IMAGES"), scNoTrace);
}
}
}
return (fProblem == TRUE) ? S_FALSE : S_OK;
}
void CNode::Reset()
{
m_pPrimaryComponent = NULL;
m_bInitComponents = TRUE;
}
HRESULT CNode::GetDispInfoForListItem(LV_ITEMW* plvi)
{
DECLARE_SC(sc, TEXT("CNode::GetDispInfoForListItem"));
ASSERT(plvi != NULL);
RESULTDATAITEM rdi;
ZeroMemory(&rdi, sizeof(rdi));
if (plvi->mask & LVIF_TEXT)
{
ASSERT (!IsBadWritePtr (plvi->pszText, plvi->cchTextMax * sizeof (TCHAR)));
rdi.mask |= RDI_STR;
}
if (plvi->mask & LVIF_IMAGE)
rdi.mask |= RDI_IMAGE;
if (plvi->mask & LVIF_STATE)
rdi.mask |= RDI_STATE;
rdi.nCol = plvi->iSubItem;
CComponent* pCC = NULL;
// if virtual list
if (GetViewData()->IsVirtualList())
{
pCC = GetPrimaryComponent();
ASSERT(pCC != NULL);
// all we can pass is the item index
rdi.nIndex = plvi->iItem;
// no default for virtual lists
rdi.nImage = MMCLV_NOICON;
}
else
{
CResultItem* pri = CResultItem::FromHandle (plvi->lParam);
if (pri != NULL)
{
if (pri->IsScopeItem()) // Folder
{
// convert to real type
CNode* pNodeSubFldr = CNode::FromResultItem (pri);
ASSERT(IsBadReadPtr(pNodeSubFldr, sizeof(CNode)) == FALSE);
if (pNodeSubFldr->IsStaticNode() == TRUE) // Static folders
{
return pNodeSubFldr->GetDispInfo(plvi);
}
else // Enumerated folders
{
// Remap the LParam information.
rdi.lParam = pNodeSubFldr->GetUserParam();
rdi.bScopeItem = TRUE;
pCC = pNodeSubFldr->GetPrimaryComponent();
rdi.nImage = pNodeSubFldr->GetResultImage();
}
}
else // Leaf item
{
// Remap the LParam information.
rdi.nImage = pri->GetImageIndex();
rdi.lParam = pri->GetSnapinData();
pCC = GetPrimaryComponent();
ASSERT(GetComponent(pri->GetOwnerID()) == GetPrimaryComponent());
ASSERT(pCC != NULL);
}
}
}
HRESULT hr = pCC->GetDisplayInfo(&rdi);
if (hr == S_OK)
{
if (rdi.mask & RDI_IMAGE)
{
if (rdi.nImage == MMCLV_NOICON)
{
plvi->iImage = rdi.bScopeItem ? eStockImage_Folder : eStockImage_File;
}
else
{
IImageListPrivate *pIL = pCC->GetIImageListPrivate();
HRESULT hr2 = pIL->MapRsltImage(pCC->GetComponentID(), rdi.nImage,
&(plvi->iImage));
if (FAILED(hr2))
{
Dbg(DEB_USER1, "can't map image provided by snapin. Using default image.\n");
plvi->iImage = rdi.bScopeItem ? eStockImage_Folder : eStockImage_File;
}
}
}
// Move all other info from rdi into lviItem
if (rdi.mask & RDI_STR)
{
if (!IsBadStringPtrW (rdi.str, plvi->cchTextMax))
{
// ignore errors in the next line - if there's not enough space take what we can get.
StringCchCopyW (plvi->pszText, plvi->cchTextMax, rdi.str);
}
else if (plvi->cchTextMax > 0)
plvi->pszText[0] = 0;
}
if (rdi.mask & RDI_STATE)
plvi->state = rdi.nState;
}
return hr;
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScSaveSortData
//
// Synopsis: Save the given sort data for persistence.
//
// Arguments: [nCol] - sort column.
// [dwOptions] - sort options.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScSaveSortData (int nCol, DWORD dwOptions)
{
DECLARE_SC(sc, _T("CNode::ScSaveSortData"));
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pViewData, E_UNEXPECTED);
if (sc)
return sc;
CLSID guidSnapin;
CXMLAutoBinary columnID;
sc = ScGetSnapinAndColumnDataID(guidSnapin, columnID);
if (sc)
return sc;
CXMLBinaryLock sLock(columnID);
SColumnSetID* pColID = NULL;
sc = sLock.ScLock(&pColID);
if (sc)
return sc;
sc = ScCheckPointers(pColID, E_UNEXPECTED);
if (sc)
return sc;
CColumnSortInfo colSortInfo;
colSortInfo.m_nCol = nCol;
colSortInfo.m_dwSortOptions = dwOptions;
colSortInfo.m_lpUserParam = NULL;
sc = pViewData->ScSaveColumnSortData(guidSnapin, *pColID, colSortInfo);
if (sc)
return sc;
// Column data when saved includes the width/order data (column info list) and sort data.
// The width/order data should always be saved regardless of whether sort data is
// persisted or not. So save the width/order data.
CColumnInfoList columnInfoList;
TStringVector strColNames; // unused
// get the current data
sc = ScGetCurrentColumnData( columnInfoList, strColNames );
if (sc)
return sc;
sc = pViewData->ScSaveColumnInfoList(guidSnapin, *pColID, columnInfoList);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: OnColumnClicked
//
// Synopsis: Ask snapin to sort.
//
// Arguments: [nCol] - Column to be sorted.
//
// Note: When column is clicked sort options and user param
// are unknown. So we set them to 0 (zero). In InternalSort
// the sort option is computed.
//
// Returns: HRESULT
//
// History: RaviR Created
// 07-27-1999 AnandhaG renamed OnSort to OnColumnClicked
//--------------------------------------------------------------------
HRESULT CNode::OnColumnClicked(LONG_PTR nCol)
{
CComponent* pComponent = GetPrimaryComponent();
ASSERT(pComponent != NULL);
if (NULL == pComponent)
return E_FAIL;
IResultDataPrivatePtr pResult = pComponent->GetIFramePrivate();
ASSERT(pResult != NULL);
if (NULL == pResult)
return E_FAIL;
HRESULT hr = pResult->InternalSort( nCol, 0, NULL,
TRUE /*column header clicked*/);
if (hr == S_OK)
{
BOOL bAscending = TRUE;
hr = pResult->GetSortDirection(&bAscending);
if (hr == S_OK)
hr = ScSaveSortData(nCol, bAscending ? 0 : RSI_DESCENDING).ToHr();
}
return S_OK;
}
//+-------------------------------------------------------------------
//
// Member: RestoreSort
//
// Synopsis: Sort the list view with persisted data.
// Restore the sort with saved column # and
// sort-options (User param is NULL as this
// is user initiated MMCN_COLUMN_CLICK)*/
//
// Arguments: [nCol] - Column to be sorted.
// [dwSortOptions] - Sortoptions, ascend/descend...
//
// Note: Unlike OnColumnClicked this method wont set columns dirty
// after successful sort.
//
// Returns: HRESULT
//
//--------------------------------------------------------------------
HRESULT CNode::RestoreSort(INT nCol, DWORD dwSortOptions)
{
CComponent* pComponent = GetPrimaryComponent();
ASSERT(pComponent != NULL);
if (NULL == pComponent)
return E_FAIL;
IResultDataPrivatePtr pResult = pComponent->GetIFramePrivate();
ASSERT(pResult != NULL);
if (NULL == pResult)
return E_FAIL;
HRESULT hr = pResult->InternalSort( nCol, dwSortOptions,
NULL /*NULL user param as this is user initiated*/,
FALSE /* Let us not send MMCN_COLUMN_CLICK*/);
return S_OK;
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScRestoreSortFromPersistedData
//
// Synopsis: Get persisted sort data if any and apply it to list-view.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScRestoreSortFromPersistedData ()
{
DECLARE_SC(sc, _T("CNode::ScRestoreSortFromPersistedData"));
CViewData *pViewData = GetViewData();
if (! pViewData->HasList() && ! pViewData->HasListPad() )
return (sc = S_FALSE); // OCX gets MMCN_SHOW which may try to restore sort so this is no failure.
// To get CColumnSetData first get the column-id & snapin guid.
CLSID guidSnapin;
CXMLAutoBinary columnID;
sc = ScGetSnapinAndColumnDataID(guidSnapin, columnID);
if (sc)
return sc;
CXMLBinaryLock sLock(columnID);
SColumnSetID* pColID = NULL;
sc = sLock.ScLock(&pColID);
if (sc)
return sc;
sc = ScCheckPointers(pColID, E_UNEXPECTED);
if (sc)
return sc;
// Get persisted data.
CColumnSetData columnSetData;
BOOL bRet = pViewData->RetrieveColumnData(guidSnapin, *pColID, columnSetData);
if (!bRet)
return (sc = S_FALSE);
CColumnInfoList* pColInfoList = columnSetData.get_ColumnInfoList();
if (!pColInfoList)
return (sc = S_FALSE);
IFramePrivatePtr spFrame = GetIFramePrivate();
sc = ScCheckPointers(spFrame, E_UNEXPECTED);
if (sc)
return sc;
// First check if the number of columns inserted are same as the
// number that is persisted. If not remove the persisted data.
IHeaderCtrlPrivatePtr spHeader = spFrame;
sc = ScCheckPointers(spHeader, E_UNEXPECTED);
if (sc)
return sc;
int cColumns = 0;
sc = spHeader->GetColumnCount(&cColumns);
if (sc)
return sc;
// If the persisted columns and number of columns inserted
// do not match remove the persisted data.
if (pColInfoList->size() != cColumns)
{
pViewData->DeleteColumnData(guidSnapin, *pColID);
return sc;
}
// Set sorting column, order
CColumnSortList* pSortList = columnSetData.get_ColumnSortList();
if (pSortList && ( pSortList->size() > 0))
{
CColumnSortList::iterator itSortInfo = pSortList->begin();
// Restore the sort with saved column # and
// sort-options (User param is NULL as this
// is user initiated MMCN_COLUMN_CLICK)*/
RestoreSort(itSortInfo->getColumn(), itSortInfo->getSortOptions());
}
return (sc);
}
/***************************************************************************\
*
* METHOD: CNode::ScGetCurrentColumnData
*
* PURPOSE: collects current column data to collections passed as args
* [ initially code used to be in OnColumns method ]
*
* PARAMETERS:
* CColumnInfoList& columnInfoList
* TStringVector& strColNames
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::ScGetCurrentColumnData( CColumnInfoList& columnInfoList, TStringVector& strColNames)
{
DECLARE_SC(sc, TEXT("CNode::ScGetCurrentColumnData"));
columnInfoList.clear();
strColNames.clear();
IHeaderCtrlPrivatePtr spHeader = GetIFramePrivate();
sc = ScCheckPointers(spHeader, E_UNEXPECTED);
if (sc)
return sc;
sc = spHeader->GetColumnInfoList(&columnInfoList);
if (sc)
return sc;
int cColumns = columnInfoList.size();
USES_CONVERSION;
for (int i = 0; i < cColumns; i++)
{
CCoTaskMemPtr<OLECHAR> spColumnText;
sc = spHeader->GetColumnText(i, &spColumnText);
if (sc)
return sc;
strColNames.push_back(OLE2T(spColumnText));
}
return sc;
}
/***************************************************************************\
*
* METHOD: CNode::ScSetUpdatedColumnData
*
* PURPOSE: updates column by data specified in collections passed as args
* [ initially code used to be in OnColumns method ]
*
* PARAMETERS:
* CColumnInfoList& oldColumnInfoList - column data befor the change
* CColumnInfoList& newColumnInfoList - updated column data
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::ScSetUpdatedColumnData( CColumnInfoList& oldColumnInfoList, CColumnInfoList& newColumnInfoList)
{
DECLARE_SC(sc, TEXT("CNode::ScSetUpdatedColumnData"));
CColumnInfoList::iterator itColInfo1, itColInfo2;
// Check if there is any change in visible/hidden columns.
// If so send MMCN_COLUMNS_CHANGE notification
for (itColInfo1 = newColumnInfoList.begin(); itColInfo1 != newColumnInfoList.end(); ++itColInfo1)
{
// Get the same column from old list.
itColInfo2 = find_if(oldColumnInfoList.begin(), oldColumnInfoList.end(),
bind2nd( ColPosCompare(), itColInfo1->GetColIndex()) );
if (itColInfo2 == oldColumnInfoList.end())
return sc = E_UNEXPECTED;
// Compare the hidden flag.
if ( itColInfo2->IsColHidden() != itColInfo1->IsColHidden() )
{
// Send MMCN_COLUMNS_CHANGED notification
sc = OnColumnsChange(newColumnInfoList);
if (sc)
return sc;
break; // done anyway
}
}
sc = ScSaveColumnInfoList(newColumnInfoList);
if (sc)
return sc;
IHeaderCtrlPrivatePtr spHeader = GetIFramePrivate();
sc = ScCheckPointers(spHeader, E_UNEXPECTED);
if (sc)
return sc;
sc = spHeader->ModifyColumns(newColumnInfoList);
if (sc)
return sc;
sc = ScRestoreSortFromPersistedData();
if (sc)
return sc;
return sc;
}
//+-------------------------------------------------------------------
//
// Member: OnColumns
//
// Synopsis: Display Columns customization dialog and if necessary
// apply changes made by the user.
//
//--------------------------------------------------------------------
void CNode::OnColumns()
{
DECLARE_SC(sc, TEXT("CNode::OnColumns"));
// first - get the columns
CColumnInfoList columnInfoList;
TStringVector strColNames;
// 1. get the current data
sc = ScGetCurrentColumnData( columnInfoList, strColNames );
if (sc)
return;
// 2. Cache the column data.
CColumnInfoList columnInfoListOld = columnInfoList;
// 3. get the default column settings.
CViewData *pViewData = GetViewData();
IHeaderCtrlPrivatePtr spHeader = GetIFramePrivate();
sc = ScCheckPointers(pViewData, spHeader, E_UNEXPECTED);
if (sc)
return;
CColumnInfoList defaultColumnInfoList;
sc = spHeader->GetDefaultColumnInfoList(defaultColumnInfoList);
if (sc)
return;
// 5. display the dialog
CColumnsDlg dlg(&columnInfoList, &strColNames, defaultColumnInfoList);
INT_PTR nRet = dlg.DoModal();
if (nRet == -1)
{
sc = E_UNEXPECTED;
return;
}
if (nRet == IDOK)
{
// update columns by modified data
sc = ScSetUpdatedColumnData( columnInfoListOld, columnInfoList );
if (sc)
return;
}
// If reset is true then throw away present persisted column data
// and apply the default settings.
if (nRet == IDC_RESTORE_DEFAULT_COLUMNS)
{
// To get CColumnSetData first get the column-id & snapin guid.
CLSID guidSnapin;
CXMLAutoBinary columnID;
sc = ScGetSnapinAndColumnDataID(guidSnapin, columnID);
if (sc)
return;
CXMLBinaryLock sLock(columnID);
SColumnSetID* pColID = NULL;
sc = sLock.ScLock(&pColID);
if (sc)
return;
sc = ScCheckPointers(pColID, E_UNEXPECTED);
if (sc)
return;
pViewData->DeleteColumnData(guidSnapin, *pColID);
sc = spHeader->ModifyColumns(defaultColumnInfoList);
if (sc)
return;
}
}
/***************************************************************************\
*
* METHOD: CNode::ScShowColumn
*
* PURPOSE: shows/hides column. notifies snapin on action
*
* PARAMETERS:
* int iColIndex - index of column to change
* bool bVisible - show/hide flag
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::ScShowColumn(int iColIndex, bool bShow)
{
DECLARE_SC(sc, TEXT("CNode::ScShowColumn"));
// first - get the current column data
CColumnInfoList columnInfoList;
TStringVector strColNames;
sc = ScGetCurrentColumnData( columnInfoList, strColNames );
if (sc)
return sc;
// Save the column data.
CColumnInfoList columnInfoListOld = columnInfoList;
// find the column and change its status
CColumnInfoList::iterator itColInfo = find_if(columnInfoList.begin(), columnInfoList.end(),
bind2nd( ColPosCompare(), iColIndex) );
// check if we did find the column
if (itColInfo == columnInfoList.end())
return sc = E_INVALIDARG; // assume it's not a valid index
// now modify the column status acording to parameters
if (bShow)
{
itColInfo->SetColHidden(false);
// move column to the end
columnInfoList.splice(columnInfoList.end(), columnInfoList, itColInfo);
}
else
{
itColInfo->SetColHidden();
}
// update columns by modified data
sc = ScSetUpdatedColumnData( columnInfoListOld, columnInfoList);
if (sc)
return sc;
return sc;
}
/***************************************************************************\
*
* METHOD: CNode::ScGetSortColumn
*
* PURPOSE: return currently used sort column
*
* PARAMETERS:
* int *piSortCol - sort column index [retval]
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::ScGetSortColumn(int *piSortCol)
{
DECLARE_SC(sc, TEXT("CNode::ScGetSortColumn"));
// parameter check
sc = ScCheckPointers(piSortCol);
if (sc)
return sc;
// retrieve IResultDataPrivate interface
CComponent* pComponent = GetPrimaryComponent();
sc = ScCheckPointers(pComponent, E_UNEXPECTED);
if (sc)
return sc;
IResultDataPrivatePtr pResult = pComponent->GetIFramePrivate();
sc = ScCheckPointers(pResult, E_UNEXPECTED);
if (sc)
return sc;
// forward the call to IResultDataPrivate
sc = pResult->GetSortColumn(piSortCol);
if (sc)
return sc;
return sc;
}
/***************************************************************************\
*
* METHOD: CNode::ScSetSortColumn
*
* PURPOSE: sorts result data by specified column
* [uses private result data interface to implement]
*
* PARAMETERS:
* int iSortCol - index of column to sort by
* bool bAscending - sorting order
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::ScSetSortColumn(int iSortCol, bool bAscending)
{
DECLARE_SC(sc, TEXT("CNode::ScSetSortColumn"));
// retrieve IResultDataPrivate interface
CComponent* pComponent = GetPrimaryComponent();
sc = ScCheckPointers(pComponent, E_UNEXPECTED);
if (sc)
return sc;
IResultDataPrivatePtr pResult = pComponent->GetIFramePrivate();
sc = ScCheckPointers(pResult, E_UNEXPECTED);
if (sc)
return sc;
DWORD dwSortOptions = bAscending ? 0 : RSI_DESCENDING;
// forward the call to IResultDataPrivate
sc = pResult->InternalSort( iSortCol, dwSortOptions, NULL, FALSE );
if (sc)
return sc;
// If sort went thru - save.
if (sc == SC(S_OK))
sc = ScSaveSortData(iSortCol, dwSortOptions);
if (sc)
return sc;
return sc;
}
//+-------------------------------------------------------------------
//
// Member: OnColumnsChange
//
// Synopsis: Send MMCN_COLUMNS_CHANGE notification to snapin.
//
// Arguments: [colInfoList] - Columns data.
//
//--------------------------------------------------------------------
HRESULT CNode::OnColumnsChange(CColumnInfoList& colInfoList)
{
CComponent* pCC = m_pPrimaryComponent;
ASSERT(pCC != NULL);
// Get the data object for the node and pass it to the primary snap-in
// and all the namespace extensions to the node.
IDataObjectPtr spDataObject;
HRESULT hr = QueryDataObject(CCT_SCOPE, &spDataObject);
if (FAILED(hr))
return hr;
int nVisibleColumns = 0;
// Count the number of columns that are visible.
CColumnInfoList::iterator itColInfo;
for (itColInfo = colInfoList.begin(); itColInfo != colInfoList.end();
++itColInfo)
{
if (! itColInfo->IsColHidden())
nVisibleColumns++;
}
int size = sizeof(MMC_VISIBLE_COLUMNS) + nVisibleColumns * sizeof(INT);
HGLOBAL hGlobal = ::GlobalAlloc(GPTR, size);
if (! hGlobal)
return E_OUTOFMEMORY;
MMC_VISIBLE_COLUMNS* pColData = reinterpret_cast<MMC_VISIBLE_COLUMNS*>(hGlobal);
pColData->nVisibleColumns = nVisibleColumns;
// Get the list of visible columns into MMC_VISIBLE_COLUMNS struct.
int i = 0;
for (itColInfo = colInfoList.begin(); itColInfo != colInfoList.end();
++itColInfo)
{
if (! itColInfo->IsColHidden())
pColData->rgVisibleCols[i++] = itColInfo->GetColIndex();
}
LPARAM lParam = reinterpret_cast<LPARAM>(pColData);
hr = pCC->Notify(spDataObject, MMCN_COLUMNS_CHANGED, 0, lParam);
::GlobalFree(hGlobal);
CHECK_HRESULT(hr);
if (FAILED(hr))
return hr;
return hr;
}
//+-------------------------------------------------------------------
//
// Member: ScSaveColumnInfoList
//
// Synopsis: Save the column data in internal data structures.
//
// Arguments: [colInfoList] - Columns data.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScSaveColumnInfoList(CColumnInfoList& columnInfoList)
{
DECLARE_SC(sc, TEXT("CNode::ScSaveColumnInfoList"));
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(pViewData, E_UNEXPECTED);
if (sc)
return sc;
CLSID clsidSnapin;
CXMLAutoBinary columnID;
sc = ScGetSnapinAndColumnDataID(clsidSnapin, columnID);
if (sc)
return sc;
CXMLBinaryLock sLock(columnID);
SColumnSetID* pColID = NULL;
sc = sLock.ScLock(&pColID);
if (sc)
return sc;
sc = ScCheckPointers(pColID, E_UNEXPECTED);
if (sc)
return sc;
sc = pViewData->ScSaveColumnInfoList(clsidSnapin, *pColID, columnInfoList);
if (sc)
return sc;
return sc;
}
//+-------------------------------------------------------------------
//
// Member: ScGetSnapinAndColumnDataID
//
// Synopsis: Returns the snapin guid & column-id in CXMLAutoBinary for this node.
//
// Arguments: [snapinGuid] - [out], snapin guid.
// [columnID] - [out], column-id in CXMLAutoBinary.
//
// Note: Pass in a CXMLAutoBinary object, will return column id in that object.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScGetSnapinAndColumnDataID(GUID& snapinGuid, CXMLAutoBinary& columnID)
{
DECLARE_SC(sc, TEXT("CNode::ScGetSnapinAndColumnDataID"));
// Get Snapin Guid
snapinGuid = GetPrimarySnapInCLSID();
columnID.ScFree(); // clear any data.
IDataObjectPtr spDataObject;
sc = QueryDataObject(CCT_SCOPE, &spDataObject);
if (sc)
return sc;
HGLOBAL hGlobal;
sc = ExtractColumnConfigID(spDataObject, hGlobal);
if (! sc.IsError())
{
int cbSize = GlobalSize(hGlobal);
if (0 == cbSize)
return sc.FromLastError();
columnID.Attach(hGlobal, cbSize);
}
else
{
// Let us use the NodeTypeGUID as the Column Data Identifier
CLSID clsidColID;
sc = GetNodeType(&clsidColID);
if (sc)
return sc;
int cbSize = sizeof(SColumnSetID) + sizeof(CLSID) - 1;
sc = columnID.ScAlloc(cbSize, true);
if (sc)
return sc;
CXMLBinaryLock sLock(columnID);
SColumnSetID* pColID = NULL;
sc = sLock.ScLock(&pColID);
if (sc)
return sc;
sc = ScCheckPointers(pColID, E_UNEXPECTED);
if (sc)
return sc;
pColID->cBytes = sizeof(CLSID);
pColID->dwFlags = 0;
CopyMemory(pColID->id, (BYTE*)&clsidColID, sizeof(pColID->id));
}
return sc;
}
/*+-------------------------------------------------------------------------*
* class CViewExtensionCallback
*
*
* PURPOSE: Implements IViewExtensionCallback
*
*+-------------------------------------------------------------------------*/
class CViewExtensionCallback :
public CComObjectRoot,
public IViewExtensionCallback
{
public:
typedef CViewExtensionCallback ThisClass;
BEGIN_COM_MAP(ThisClass)
COM_INTERFACE_ENTRY(IViewExtensionCallback)
END_COM_MAP()
DECLARE_NOT_AGGREGATABLE(ThisClass)
IMPLEMENTS_SNAPIN_NAME_FOR_DEBUG()
CViewExtensionCallback() : m_pItExt(NULL) {}
SC ScInitialize(CViewExtInsertIterator & itExt)
{
DECLARE_SC(sc, TEXT("CViewExtensionCallback::ScInitialize"));
m_pItExt = &itExt;
return sc;
}
SC ScDeinitialize()
{
DECLARE_SC (sc, _T("CViewExtensionCallback::ScDeinitialize"));
m_pItExt = NULL;
return (sc);
}
public:
STDMETHODIMP AddView(PMMC_EXT_VIEW_DATA pExtViewData) {return ScAddView(pExtViewData).ToHr();}
private:
SC ScAddView(PMMC_EXT_VIEW_DATA pExtViewData)
{
DECLARE_SC(sc, TEXT("CViewExtensionCallback::ScAddView"));
sc = ScCheckPointers(pExtViewData, pExtViewData->pszURL, pExtViewData->pszViewTitle);
if(sc)
return sc; // TODO add snapin error
sc = ScCheckPointers(m_pItExt, E_UNEXPECTED);
if(sc)
return sc; // TODO add snapin error, e.g. "IExtendViewCallback::AddView called outside of IExtendView::GetViews"
/*
* prep the input to IConsoleView::ScAddViewExtension
*/
CViewExtensionData ved;
ved.strURL = pExtViewData->pszURL;
ved.strName = pExtViewData->pszViewTitle;
ved.viewID = pExtViewData->viewID;
ved.bReplacesDefaultView = pExtViewData->bReplacesDefaultView;
/*
* std::basic_string's can't assign from NULL, so we have to check first
*/
if (pExtViewData->pszTooltipText)
ved.strTooltip = pExtViewData->pszTooltipText;
/*
* validate output: URL and title are required, tooltip is optional
*/
if (ved.strURL.empty())
{
TraceSnapinError(TEXT("Invalid parameter to IViewExtensionCallback::AddView (empty URL)"), E_INVALIDARG);
return (sc = E_INVALIDARG);
}
if (ved.strName.empty())
{
TraceSnapinError(TEXT("Invalid parameter to IViewExtensionCallback::AddView (empty title)"), E_INVALIDARG);
return (sc = E_INVALIDARG);
}
/*
* add the extension to the view
*/
*(*m_pItExt)++ = ved;
return sc;
}
private:
CViewExtInsertIterator *m_pItExt;
};
/*+-------------------------------------------------------------------------*
* CNode::ScGetViewExtensions
*
*
*--------------------------------------------------------------------------*/
SC CNode::ScGetViewExtensions (CViewExtInsertIterator itExt)
{
DECLARE_SC (sc, _T("CNode::ScGetViewExtensions"));
IDataObjectPtr spDataObject;
bool bScopeItem ;
sc = ScGetDataObject(/*bScopePane*/ true, NULL /*lResultItemCookie*/, bScopeItem, &spDataObject);
if(sc)
return sc;
CSnapIn* pSnapIn = GetPrimarySnapIn();
sc = ScCheckPointers (pSnapIn, E_FAIL);
if (sc)
return (sc);
CArray<GUID, GUID&> DynExtens;
ExtractDynExtensions(spDataObject, DynExtens);
GUID guidNodeType;
sc = ::ExtractObjectTypeGUID(spDataObject, &guidNodeType);
if(sc)
return sc;
CExtensionsIterator it;
sc = it.ScInitialize(pSnapIn, guidNodeType, g_szView, DynExtens.GetData(), DynExtens.GetSize());
if(sc)
return sc;
typedef CComObject<CViewExtensionCallback> t_ViewExtensionCallback;
t_ViewExtensionCallback *pViewExtensionCallback = NULL;
sc = t_ViewExtensionCallback::CreateInstance(&pViewExtensionCallback);
if(sc)
return sc;
if(NULL == pViewExtensionCallback)
return (sc = E_UNEXPECTED);
sc = pViewExtensionCallback->ScInitialize(itExt);
if(sc)
return sc;
IViewExtensionCallbackPtr spViewExtensionCallback = pViewExtensionCallback;
// add all the console taskpads first
sc = CConsoleTaskpadViewExtension::ScGetViews(this, spViewExtensionCallback);
if(sc)
return sc;
for (; !it.IsEnd(); it.Advance())
{
// any errors in this block should just go on to the next snap-in. Can't let one snap-in
// hose all the others.
/*
* create the extension
*/
IExtendViewPtr spExtendView;
sc = spExtendView.CreateInstance(it.GetCLSID(), NULL, MMC_CLSCTX_INPROC);
if(sc)
{
#ifdef DBG
USES_CONVERSION;
tstring strMsg = _T("Failed to create snapin ");
CCoTaskMemPtr<WCHAR> spszCLSID;
if (SUCCEEDED (StringFromCLSID (it.GetCLSID(), &spszCLSID)))
strMsg += W2T(spszCLSID);
TraceSnapinError(strMsg.data(), sc);
#endif
sc.Clear();
continue;
}
/*
* get the view extension data from the extension
*/
sc = spExtendView->GetViews(spDataObject, spViewExtensionCallback);
if(sc)
{
TraceSnapinError(TEXT("Snapin returned error on call to IExtendView::GetView"), sc);
sc.Clear();
continue;
}
}
/*
* View extensions aren't supposed to hold onto IExtendViewCallback,
* but buggy view extensions might. This will neuter the callback
* so buggy view extensions won't reference stale data.
*/
sc = pViewExtensionCallback->ScDeinitialize();
if (sc)
return (sc);
return (sc);
}
/*******************************************************\
| helper function to avoid too many stack allocations
\*******************************************************/
static std::wstring T2W_ForLoop(const tstring& str)
{
#if defined(_UNICODE)
return str;
#else
USES_CONVERSION;
return A2CW(str.c_str());
#endif
}
/***************************************************************************\
|
| Implementation of subclass CNode::CDataObjectCleanup
| Responsible for data object clenup
|
| Cleanup works by these rules:
|
| 1. Data object created for cut , copy or dragdrop registers every node added to it
| 2. Nodes are registered in the static multimap, mapping node to the data object it belongs to.
| 3. Node destructor checks the map and triggers cleanup for all affected data objects.
| 4. Data Object cleanup is: a) unregistering its nodes,
| b) release contained data objects
| b) entering invalid state (allowing only removal of cut objects to succeed)
| c) revoking itself from clipboard if it is on the clipboard.
| It will not do any of following: a) release references to IComponents as long as is alive
| b) prevent MMCN_CUTORMOVE to be send by invoking RemoveCutItems()
|
\***************************************************************************/
// declare static variable
CNode::CDataObjectCleanup::CMapOfNodes CNode::CDataObjectCleanup::s_mapOfNodes;
/***************************************************************************\
*
* METHOD: CNode::CDataObjectCleanup::ScRegisterNode
*
* PURPOSE: registers node to trigger clipboard clenup on destructor
*
* PARAMETERS:
* CNode *pNode [in] - node to register
* CMMCClipBoardDataObject *pObject [in] - data object to remove from clipboard
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::CDataObjectCleanup::ScRegisterNode(CNode *pNode, CMMCClipBoardDataObject *pObject)
{
DECLARE_SC(sc, TEXT("CNode::CClipboardClenup::ScRegisterNode"));
// parameter check
sc = ScCheckPointers( pNode, pObject );
if (sc)
return sc;
// add to the multimap
s_mapOfNodes.insert( CMapOfNodes::value_type( pNode, pObject ) );
return sc;
}
/***************************************************************************\
*
* METHOD: CNode::CDataObjectCleanup::ScUnadviseDataObject
*
* PURPOSE: Removes nodes-'clenup triggers' kept for the object
*
* PARAMETERS:
* CMMCClipBoardDataObject *pObject [in] object going away
* bool bForceDataObjectCleanup [in] whether need to ask DO to clenup / unregister itself
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::CDataObjectCleanup::ScUnadviseDataObject(CMMCClipBoardDataObject *pObject, bool bForceDataObjectCleanup /*= true*/)
{
DECLARE_SC(sc, TEXT("CNode::CDataObjectCleanup::ScUnadviseDataObject"));
// remove all nodes associated with the data object
CMapOfNodes::iterator it = s_mapOfNodes.begin();
while ( it != s_mapOfNodes.end() )
{
// remove or skip the entry
if ( it->second == pObject )
it = s_mapOfNodes.erase( it );
else
++it;
}
// invalidate data object when required
if ( bForceDataObjectCleanup )
{
sc = pObject->ScInvalidate();
if (sc)
return sc;
}
return sc;
}
/***************************************************************************\
*
* METHOD: CNode::CDataObjectCleanup::ScUnadviseNode
*
* PURPOSE: Does data object claenup triggered by the node
*
* PARAMETERS:
* CNode *pNode [in] - node initiating cleanup
*
* RETURNS:
* SC - result code
*
\***************************************************************************/
SC CNode::CDataObjectCleanup::ScUnadviseNode(CNode *pNode)
{
DECLARE_SC(sc, TEXT("CNode::CClipboardClenup::ScUnadviseNode"));
// parameter check
sc = ScCheckPointers(pNode);
if (sc)
return sc;
// find the node in the map
CMapOfNodes::iterator it;
while ( s_mapOfNodes.end() != ( it = s_mapOfNodes.find(pNode) ) )
{
// one node triggers clenup for whole data object
sc = ScUnadviseDataObject( it->second );
if (sc)
return sc;
}
return sc;
}
//############################################################################
//############################################################################
//
// Implementation of class CSnapInNode
//
//############################################################################
//############################################################################
DEBUG_DECLARE_INSTANCE_COUNTER(CSnapInNode);
CSnapInNode::CSnapInNode(
CMTSnapInNode* pMTNode,
CViewData* pViewData,
bool fRootNode)
: CNode(pMTNode, pViewData, fRootNode, true)
{
m_spData.CreateInstance();
ASSERT(pMTNode != NULL);
DEBUG_INCREMENT_INSTANCE_COUNTER(CSnapInNode);
pMTNode->AddNode(this);
}
CSnapInNode::CSnapInNode(const CSnapInNode& other) :
CNode (other),
m_spData (other.m_spData)
{
DEBUG_INCREMENT_INSTANCE_COUNTER(CSnapInNode);
}
void CSnapInNode::Reset()
{
m_spData->Reset();
ResetFlags();
CNode::Reset();
}
CSnapInNode::~CSnapInNode()
{
DEBUG_DECREMENT_INSTANCE_COUNTER(CSnapInNode);
CMTSnapInNode* pMTSINode = dynamic_cast<CMTSnapInNode*>(GetMTNode());
ASSERT(pMTSINode != NULL);
if (pMTSINode)
pMTSINode->RemoveNode(this);
}
void CSnapInNode::AddComponentToArray(CComponent* pCC)
{
ASSERT((pCC->GetComponentID() >= GetComponentArray().size()) ||
(GetComponentArray().size() > 0) &&
(GetComponentArray()[pCC->GetComponentID()] == NULL));
if (pCC->GetComponentID() >= GetComponentArray().size())
GetComponentArray().resize(pCC->GetComponentID() + 1);
GetComponentArray()[pCC->GetComponentID()] = pCC;
}
CComponent* CSnapInNode::CreateComponent(CSnapIn* pSnapIn, int nID)
{
ASSERT(pSnapIn != NULL);
CComponent* pCC = new CComponent(pSnapIn);
if ( pCC != NULL )
{
pCC->SetComponentID(nID);
AddComponentToArray(pCC);
}
return pCC;
}
//+-------------------------------------------------------------------
//
// Member: CSnapInNode::GetResultImage
//
// Synopsis: Get the image-index in result-image list for this node.
//
// Note: The CSnapInNode member ImageListPrivate is not set all
// the time (if the window is rooted at snapin node), so
// set it temporarily when we need the icon index.
// The member is set while adding sub-folders
// The only other case this will affect is when called for
// the image index for static snapin nodes displayed in result-pane.
// But when static snapin nodes are displayed in result-pane the
// AddSubFolders added it so the imagelist is already set.
//
// Arguments:
//
// Returns: Image index for this item in result-pane.
//
//--------------------------------------------------------------------
UINT CSnapInNode::GetResultImage()
{
IImageListPrivate *pImageList = GetImageList();
if (!pImageList)
{
CComponent *pCC = GetPrimaryComponent();
if (pCC)
pImageList = pCC->GetIImageListPrivate();
}
CMTSnapInNode* pMTSnapInNode = dynamic_cast<CMTSnapInNode*>(GetMTNode());
if (pMTSnapInNode)
return pMTSnapInNode->GetResultImage ((CNode*)this, pImageList);
return CNode::GetResultImage();
}
/*+-------------------------------------------------------------------------*
*
* CSnapInNode::GetControl
*
* PURPOSE: Given the CLSID of the OCX, see if we have stored this
* OCX, if so return the OCXWrapper's IUnknown ptr.
*
* PARAMETERS:
* CLSID clsid: class-id of the OCX.
*
* RETURNS:
* LPUNKNOWN of wrapper OCX.
*
*+-------------------------------------------------------------------------*/
LPUNKNOWN CSnapInNode::GetControl(CLSID& clsid)
{
for (int i=0; i <= GetOCXArray().GetUpperBound(); i++)
{
if (GetOCXArray()[i].IsControlCLSID(clsid) == TRUE)
return GetOCXArray()[i].GetControlUnknown();
}
return NULL;
}
/*+-------------------------------------------------------------------------*
*
* CSnapInNode::SetControl
*
* PURPOSE: Given the CLSID of the OCX and of the wrapper.
*
* PARAMETERS:
* CLSID clsid : of a OCX.
* IUnknown *pUnknown: of OCX wrapper.
*
* RETURNS:
* void
*
*+-------------------------------------------------------------------------*/
void CSnapInNode::SetControl(CLSID& clsid, IUnknown* pUnknown)
{
// check for slot in cache
int iLast = GetOCXArray().GetUpperBound();
for (int i=0; i <= iLast; i++)
{
if (GetOCXArray()[i].IsControlCLSID(clsid) == TRUE)
break;
}
// if not in cache, add one more entry
if (i > iLast)
GetOCXArray().SetSize(i + 1);
GetOCXArray()[i].SetControl(clsid, pUnknown);
}
/*+-------------------------------------------------------------------------*
*
* CSnapInNode::GetControl
*
* PURPOSE: Given the IUnknown of the OCX, see if we have stored this
* OCX, if so return the OCXWrapper's IUnknown ptr.
*
* PARAMETERS:
* IUnknown *pUnkOCX : of a OCX.
*
* RETURNS:
* LPUNKNOWN of wrapper OCX.
*
*+-------------------------------------------------------------------------*/
LPUNKNOWN CSnapInNode::GetControl(LPUNKNOWN pUnkOCX)
{
for (int i=0; i <= GetOCXArray().GetUpperBound(); i++)
{
// Compare IUnknowns.
if (GetOCXArray()[i].IsSameOCXIUnknowns(pUnkOCX) == TRUE)
return GetOCXArray()[i].GetControlUnknown();
}
return NULL;
}
/*+-------------------------------------------------------------------------*
*
* CSnapInNode::SetControl
*
* PURPOSE: Given the IUnknown of the OCX and of the wrapper.
*
* PARAMETERS:
* IUnknown *pUnkOCX : of a OCX.
* IUnknown *pUnknown: of OCX wrapper.
*
* RETURNS:
* void
*
*+-------------------------------------------------------------------------*/
void CSnapInNode::SetControl(LPUNKNOWN pUnkOCX, IUnknown* pUnknown)
{
// check for slot in cache
int iLast = GetOCXArray().GetUpperBound();
for (int i=0; i <= iLast; i++)
{
if (GetOCXArray()[i].IsSameOCXIUnknowns(pUnkOCX) == TRUE)
break; // found the OCX, so replace with given OCXwrapper.
}
// if not in cache, add one more entry
if (i > iLast)
GetOCXArray().SetSize(i + 1);
GetOCXArray()[i].SetControl(pUnkOCX, pUnknown);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScGetConsoleTaskpad
//
// Synopsis: Get the console taskpad identified by given GUID for this node.
//
// Arguments: [guidTaskpad] - [in param]
// [ppTaskpad] - [out param]
//
// Returns: SC, S_FALSE if none exists
//
//--------------------------------------------------------------------
SC CNode::ScGetConsoleTaskpad (const GUID& guidTaskpad, CConsoleTaskpad **ppTaskpad)
{
DECLARE_SC(sc, _T("CNode::ScGetConsoleTaskpad"));
sc = ScCheckPointers(ppTaskpad);
if (sc)
return sc;
*ppTaskpad = NULL;
CScopeTree* pScopeTree = CScopeTree::GetScopeTree();
sc = ScCheckPointers(pScopeTree, E_UNEXPECTED);
if(sc)
return sc;
CConsoleTaskpadList *pConsoleTaskpadList = pScopeTree->GetConsoleTaskpadList();
sc = ScCheckPointers(pConsoleTaskpadList, E_UNEXPECTED);
if (sc)
return sc;
// get a filtered list of taskpads that apply to this node.
CConsoleTaskpadFilteredList filteredList;
sc = pConsoleTaskpadList->ScGetTaskpadList(this, filteredList);
if(sc)
return sc;
for(CConsoleTaskpadFilteredList::iterator iter = filteredList.begin(); iter!= filteredList.end(); ++iter)
{
CConsoleTaskpad *pTaskpad = *iter;
sc = ScCheckPointers(pTaskpad, E_UNEXPECTED);
if (sc)
return sc;
if (pTaskpad->GetID() == guidTaskpad)
{
*ppTaskpad = pTaskpad;
return sc; // found
}
}
return (sc = S_FALSE); // not found
}
/*************************************************************************
*
* There is only one CViewSettingsPersistor object per document.
*
* The object stored as static variable inside CNode as CNode needs
* to access this object frequently.
*
* The Document needs to initialize/save the object by loading/savind
* from/to console file. It calls below ScQueryViewSettingsPersistor.
*
* The object is created with first call to ScQueryViewSettingsPersistor.
* The object is destroyed when DocumentClosed event is received.
*
*************************************************************************/
CComObject<CViewSettingsPersistor>* CNode::m_pViewSettingsPersistor = NULL;
//+-------------------------------------------------------------------
//
// Member: CNode::ScQueryViewSettingsPersistor
//
// Synopsis: Static method to get IPersistStream to load CViewSettingsPersistor
// object from old style console file.
//
// If the CViewSettingsObject is not created then create one.
//
// Arguments: [ppStream] - [out]
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScQueryViewSettingsPersistor (IPersistStream **ppStream)
{
DECLARE_SC(sc, _T("CNode::ScQueryViewSettingsPersistor"));
sc = ScCheckPointers(ppStream);
if (sc)
return sc;
// Create new CViewSettingsPersistor if none exists
if (NULL == m_pViewSettingsPersistor)
{
sc = CComObject<CViewSettingsPersistor>::CreateInstance (&m_pViewSettingsPersistor);
if (sc)
goto ObjectCreationFailed;
sc = ScCheckPointers(m_pViewSettingsPersistor, E_UNEXPECTED);
if (sc)
goto ObjectCreationFailed;
m_pViewSettingsPersistor->AddRef();
}
sc = ScCheckPointers(m_pViewSettingsPersistor, E_UNEXPECTED);
if (sc)
goto ObjectCreationFailed;
*ppStream = static_cast<IPersistStream*>(m_pViewSettingsPersistor);
if (NULL == *ppStream)
return (sc = E_UNEXPECTED);
(*ppStream)->AddRef();
Cleanup:
return (sc);
ObjectCreationFailed:
CStr strMsg;
strMsg.LoadString(GetStringModule(), IDS_ViewSettingCouldNotBePersisted);
::MessageBox(NULL, strMsg, NULL, MB_OK|MB_SYSTEMMODAL);
goto Cleanup;
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScQueryViewSettingsPersistor
//
// Synopsis: Static method to get CXMLObject to load or save
// CViewSettingsPersistor object from XML console file.
//
// If the CViewSettingsObject is not created then create one.
//
// Arguments: [ppXMLObject] - [out]
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScQueryViewSettingsPersistor (CXMLObject **ppXMLObject)
{
DECLARE_SC(sc, _T("CNode::ScQueryViewSettingsPersistor"));
sc = ScCheckPointers(ppXMLObject);
if (sc)
return sc;
// Create new CViewSettingsPersistor if none exists
if (NULL == m_pViewSettingsPersistor) // Create new one
{
sc = CComObject<CViewSettingsPersistor>::CreateInstance (&m_pViewSettingsPersistor);
if (sc)
goto ObjectCreationFailed;
sc = ScCheckPointers(m_pViewSettingsPersistor, E_UNEXPECTED);
if (sc)
goto ObjectCreationFailed;
m_pViewSettingsPersistor->AddRef();
}
sc = ScCheckPointers(m_pViewSettingsPersistor, E_UNEXPECTED);
if (sc)
goto ObjectCreationFailed;
*ppXMLObject = static_cast<CXMLObject*>(m_pViewSettingsPersistor);
if (NULL == *ppXMLObject)
return (sc = E_UNEXPECTED);
Cleanup:
return (sc);
ObjectCreationFailed:
CStr strMsg;
strMsg.LoadString(GetStringModule(), IDS_ViewSettingCouldNotBePersisted);
::MessageBox(NULL, strMsg, NULL, MB_OK|MB_SYSTEMMODAL);
goto Cleanup;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScDeleteViewSettings
//
// Synopsis: Delete the CViewSettings object for given view-id as the
// view is being closed.
//
// Arguments: [nViewID] -
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScDeleteViewSettings (int nViewID)
{
DECLARE_SC(sc, _T("CNode::ScDeleteViewSettings"));
sc = ScCheckPointers(m_pViewSettingsPersistor, E_UNEXPECTED);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScDeleteDataOfView(nViewID);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScOnDocumentClosing
//
// Synopsis: The document is closing, destroy any document related
// objects like CViewSettingsPersistor.
//
// Arguments: None
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScOnDocumentClosing ()
{
DECLARE_SC(sc, _T("CNode::ScOnDocumentClosing"));
if (m_pViewSettingsPersistor)
{
m_pViewSettingsPersistor->Release();
m_pViewSettingsPersistor = NULL;
}
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScSetFavoriteViewSettings
//
// Synopsis: A favorite is selected and it sets viewsettings
// before re-selecting the node so that after re-selection
// the new settings are set for the view.
//
// Arguments: [nViewID] -
// [bookmark] -
// [viewSettings] -
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScSetFavoriteViewSettings (int nViewID, const CBookmark& bookmark,
const CViewSettings& viewSettings)
{
DECLARE_SC(sc, _T("CNode::ScSetFavoriteViewSettings"));
sc = ScCheckPointers(m_pViewSettingsPersistor, E_UNEXPECTED);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScSetFavoriteViewSettings (nViewID, bookmark, viewSettings);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScGetViewMode
//
// Synopsis: Get the viewmode if any persisted for this node.
//
// Arguments: [ulViewMode] - [out]
//
// Returns: SC, S_FALSE if none persisted.
//
//--------------------------------------------------------------------
SC CNode::ScGetViewMode (ULONG& ulViewMode)
{
DECLARE_SC(sc, _T("CNode::ScGetViewMode"));
CMTNode *pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if (sc)
return sc;
CBookmark *pBookmark = pMTNode->GetBookmark();
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(m_pViewSettingsPersistor, pBookmark, pViewData, E_UNEXPECTED);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScGetViewMode (pViewData->GetViewID(),
*pBookmark,
ulViewMode);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScSetViewMode
//
// Synopsis: Set the viewmode in persisted viewsettings.
//
// Arguments: [ulViewMode] - [in]
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScSetViewMode (ULONG ulViewMode)
{
DECLARE_SC(sc, _T("CNode::ScSetViewMode"));
CMTNode *pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if (sc)
return sc;
CBookmark *pBookmark = pMTNode->GetBookmark();
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(m_pViewSettingsPersistor, pBookmark, pViewData, E_UNEXPECTED);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScSetViewMode (pViewData->GetViewID(),
*pBookmark,
ulViewMode);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScGetResultViewType
//
// Synopsis: Get the CResultViewType if any persisted for this node.
//
// Arguments: [rvt] - [out]
//
// Returns: SC, S_FALSE if none persisted.
//
//--------------------------------------------------------------------
SC CNode::ScGetResultViewType (CResultViewType& rvt)
{
DECLARE_SC(sc, _T("CNode::ScGetResultViewType"));
CMTNode *pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if (sc)
return sc;
CBookmark *pBookmark = pMTNode->GetBookmark();
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(m_pViewSettingsPersistor, pBookmark, pViewData, E_UNEXPECTED);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScGetResultViewType (pViewData->GetViewID(),
*pBookmark,
rvt);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScSetResultViewType
//
// Synopsis: Set the CResultViewType in persisted viewsettings.
//
// Arguments: [rvt] - [in]
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScSetResultViewType (const CResultViewType& rvt)
{
DECLARE_SC(sc, _T("CNode::ScSetResultViewType"));
CMTNode *pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if (sc)
return sc;
CBookmark *pBookmark = pMTNode->GetBookmark();
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(m_pViewSettingsPersistor, pBookmark, pViewData, E_UNEXPECTED);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScSetResultViewType (pViewData->GetViewID(),
*pBookmark,
rvt);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScGetTaskpadID
//
// Synopsis: Get the taskpad-id if any persisted for this node.
// First see if there are any node-specific taskpad-id
// else get the node-type-specific setting if one exists.
//
// Arguments: [rvt] - [out]
//
// Returns: SC, S_FALSE if none persisted.
//
//--------------------------------------------------------------------
SC CNode::ScGetTaskpadID (GUID& guidTaskpad)
{
DECLARE_SC(sc, _T("CNode::ScGetTaskpadID"));
CMTNode *pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if (sc)
return sc;
CViewData *pViewData = GetViewData();
CBookmark *pBookmark = pMTNode->GetBookmark();
sc = ScCheckPointers(m_pViewSettingsPersistor, pViewData, pBookmark, E_UNEXPECTED);
if (sc)
return sc;
// 1. Try to get node-specific taskpad-id
sc = m_pViewSettingsPersistor->ScGetTaskpadID (pViewData->GetViewID(),
*pBookmark,
guidTaskpad);
if (sc == S_OK)
return sc;
// 2. Try to get nodetype specific taskpad-id.
GUID guidNodeType;
sc = pMTNode->GetNodeType(&guidNodeType);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScGetTaskpadID(pViewData->GetViewID(),
guidNodeType,
guidTaskpad);
if (sc)
return sc;
return (sc);
}
//+-------------------------------------------------------------------
//
// Member: CNode::ScSetTaskpadID
//
// Synopsis: Set the taskpad-id in persisted viewsettings. Also see if
// the taskpad is node-specific or nodetype-specific and persist
// accordingly.
//
// Arguments: [guidTaskpad] - [in]
// [bSetDirty] - [in], set the console file dirty.
//
// Returns: SC
//
//--------------------------------------------------------------------
SC CNode::ScSetTaskpadID (const GUID& guidTaskpad, bool bSetDirty)
{
DECLARE_SC(sc, _T("CNode::ScSetTaskpadID"));
CMTNode *pMTNode = GetMTNode();
sc = ScCheckPointers(pMTNode, E_UNEXPECTED);
if (sc)
return sc;
CViewData *pViewData = GetViewData();
sc = ScCheckPointers(m_pViewSettingsPersistor, pViewData, E_UNEXPECTED);
if (sc)
return sc;
// Need to know if this task-pad is nodespecific or not.
bool bNodeSpecific = false;
CConsoleTaskpad *pTaskpad = NULL;
sc = ScGetConsoleTaskpad (guidTaskpad, &pTaskpad);
if (sc == S_OK) // S_OK if taskpad exists
{
sc = ScCheckPointers(pTaskpad, E_UNEXPECTED);
if (sc)
return sc;
bNodeSpecific = pTaskpad->IsNodeSpecific();
}
// else it may be viewextension or normal view (which are nodetype-specific).
CBookmark *pBookmark = pMTNode->GetBookmark();
sc = ScCheckPointers(pBookmark, E_UNEXPECTED);
if (sc)
return sc;
if (bNodeSpecific)
{
// Per node taskpad
sc = m_pViewSettingsPersistor->ScSetTaskpadID (pViewData->GetViewID(),
*pBookmark,
guidTaskpad,
bSetDirty);
}
else
{
// Per nodetype taskpad.
GUID guidNodeType;
sc = pMTNode->GetNodeType(&guidNodeType);
if (sc)
return sc;
sc = m_pViewSettingsPersistor->ScSetTaskpadID(pViewData->GetViewID(),
guidNodeType,
*pBookmark,
guidTaskpad,
bSetDirty);
}
if (sc)
return sc;
return (sc);
}