|
|
#include "stdafx.h"
#include "BackupSnap.h"
#include "Backup.h"
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
//
// CBackupSnapNode
//
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
static const GUID CBackupSnapGUID_NODETYPE = { 0x3fa55cce, 0x7d0a, 0x4b5c, { 0x90, 0xe9, 0x6e, 0x8c, 0xcb, 0x77, 0x5b, 0x23 } };
const GUID* CBackupSnapNode::m_NODETYPE = &CBackupSnapGUID_NODETYPE;const OLECHAR* CBackupSnapNode::m_SZNODETYPE = OLESTR("3FA55CCE-7D0A-4b5c-90E9-6E8CCB775B23");const OLECHAR* CBackupSnapNode::m_SZDISPLAY_NAME = OLESTR("");const CLSID* CBackupSnapNode::m_SNAPIN_CLASSID = &CLSID_BackupSnap;
CBackupSnapNode::CBackupSnapNode(){ // Initialize Scope Data Information
memset(&m_scopeDataItem, 0, sizeof(m_scopeDataItem)); m_scopeDataItem.mask = SDI_STR | SDI_IMAGE | SDI_OPENIMAGE | SDI_PARAM; m_scopeDataItem.displayname = L""; m_scopeDataItem.nImage = 0; m_scopeDataItem.nOpenImage = 0; m_scopeDataItem.lParam = (LPARAM) this;
// Initialize Result Data Information
memset(&m_resultDataItem, 0, sizeof(m_resultDataItem)); m_resultDataItem.mask = RDI_STR | RDI_IMAGE | RDI_PARAM; m_resultDataItem.str = L""; m_resultDataItem.nImage = 0; m_resultDataItem.lParam = (LPARAM) this;
// Load the Snap-In Display Name
tstring strTemp = StrLoadString( IDS_SnapinName ); m_bstrDisplayName = strTemp.c_str();
if( m_bstrDisplayName.m_str ) { m_SZDISPLAY_NAME = m_bstrDisplayName.m_str; }}
HRESULT CBackupSnapNode::GetScopePaneInfo(SCOPEDATAITEM *pScopeDataItem){ if( !pScopeDataItem ) return E_POINTER;
if( pScopeDataItem->mask & SDI_STR ) pScopeDataItem->displayname = m_bstrDisplayName; if( pScopeDataItem->mask & SDI_IMAGE ) pScopeDataItem->nImage = m_scopeDataItem.nImage; if( pScopeDataItem->mask & SDI_OPENIMAGE ) pScopeDataItem->nOpenImage = m_scopeDataItem.nOpenImage; if( pScopeDataItem->mask & SDI_PARAM ) pScopeDataItem->lParam = m_scopeDataItem.lParam; if( pScopeDataItem->mask & SDI_STATE ) pScopeDataItem->nState = m_scopeDataItem.nState; return S_OK;}
HRESULT CBackupSnapNode::GetResultPaneInfo(RESULTDATAITEM *pResultDataItem){ if( !pResultDataItem ) return E_POINTER;
if( pResultDataItem->bScopeItem ) { if( pResultDataItem->mask & RDI_STR ) { pResultDataItem->str = L""; } if( pResultDataItem->mask & RDI_IMAGE ) { pResultDataItem->nImage = m_scopeDataItem.nImage; } if( pResultDataItem->mask & RDI_PARAM ) { pResultDataItem->lParam = m_scopeDataItem.lParam; }
return S_OK; } if( pResultDataItem->mask & RDI_STR ) { pResultDataItem->str = L""; } if( pResultDataItem->mask & RDI_IMAGE ) { pResultDataItem->nImage = m_resultDataItem.nImage; } if( pResultDataItem->mask & RDI_PARAM ) { pResultDataItem->lParam = m_resultDataItem.lParam; } if( pResultDataItem->mask & RDI_INDEX ) { pResultDataItem->nIndex = m_resultDataItem.nIndex; }
return S_OK;}
HRESULT CBackupSnapNode::GetClassID(CLSID* pID){ if( !pID ) return E_POINTER;
pID = (CLSID*)m_SNAPIN_CLASSID;
return S_OK;}HRESULT CBackupSnapNode::Notify( MMC_NOTIFY_TYPE event, LPARAM arg, LPARAM param, IComponentData* pComponentData, IComponent* pComponent, DATA_OBJECT_TYPES type){ HRESULT hr = S_FALSE;
_ASSERT( pComponentData || pComponent );
CComPtr<IConsole> spConsole = NULL; if( pComponentData ) { spConsole = ((CBackupSnapData*)pComponentData)->m_spConsole; } else if ( pComponent ) { spConsole = ((CBackupSnapComponent*)pComponent)->m_spConsole; } if( !spConsole ) return E_INVALIDARG;
switch( event ) { case MMCN_SHOW: { hr = S_OK; // configure the ocx message in the result pane
IMessageView* pIMessageView = NULL; LPUNKNOWN pIUnk = NULL; hr = spConsole->QueryResultView(&pIUnk);
if( SUCCEEDED(hr) ) { hr = pIUnk->QueryInterface(_uuidof(IMessageView), reinterpret_cast<void**>(&pIMessageView)); }
if( SUCCEEDED(hr) ) { hr = pIMessageView->SetIcon(Icon_Information); } if( SUCCEEDED(hr) ) { tstring strTitle = StrLoadString( IDS_SnapinName ); hr = pIMessageView->SetTitleText( strTitle.c_str() ); } if( SUCCEEDED(hr) ) { tstring strMessage = StrLoadString( IDS_WARNING ); hr = pIMessageView->SetBodyText( strMessage.c_str() ); }
if( pIMessageView ) { pIMessageView->Release(); pIMessageView = NULL; }
break; } case MMCN_EXPAND: { hr = S_OK; break; } case MMCN_ADD_IMAGES: { IImageList* pImageList = (IImageList*)arg; if( !pImageList ) return E_INVALIDARG; hr = LoadImages(pImageList); break; } case MMCN_SELECT: { // if selecting node
if( HIWORD(arg) ) { hr = S_OK;
// get the verb interface and enable rename
CComPtr<IConsoleVerb> spConsVerb; if( spConsole->QueryConsoleVerb(&spConsVerb) == S_OK ) { hr = spConsVerb->SetVerbState(MMC_VERB_RENAME, ENABLED, FALSE); if( FAILED(hr) ) return hr;
spConsVerb->SetVerbState(MMC_VERB_RENAME, HIDDEN, TRUE); if( FAILED(hr) ) return hr; } }
break; }
case MMCN_CONTEXTHELP: { hr = S_OK; TCHAR szWindowsDir[MAX_PATH+1] = {0}; tstring strHelpFile = _T(""); tstring strHelpFileName = StrLoadString(IDS_HELPFILE); tstring strHelpTopicName = StrLoadString(IDS_HELPTOPIC);
if( strHelpFileName.empty() || strHelpTopicName.empty() ) { return E_FAIL; } // Build path to %systemroot%\help
UINT nSize = GetSystemWindowsDirectory( szWindowsDir, MAX_PATH ); if( nSize == 0 || nSize > MAX_PATH ) { return E_FAIL; } strHelpFile = szWindowsDir; // D:\windows
strHelpFile += _T("\\Help\\"); // \help
strHelpFile += strHelpFileName; // \filename.chm
strHelpFile += _T("::/"); // ::/
strHelpFile += strHelpTopicName; // index.htm
// Show the Help topic
CComQIPtr<IDisplayHelp> spHelp = spConsole; if( !spHelp ) return E_NOINTERFACE;
hr = spHelp->ShowTopic( (LPTSTR)strHelpFile.c_str() ); break; }
}// switch
return hr;}
////////////////////////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////////////////////////
//
// CBackupSnapData
//
//////////////////////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////////////////////////
HRESULT CBackupSnapData::Initialize(LPUNKNOWN pUnknown){ if( !pUnknown ) return E_POINTER;
HRESULT hr = IComponentDataImpl<CBackupSnapData, CBackupSnapComponent >::Initialize(pUnknown); if( FAILED(hr) ) return hr;
CComPtr<IImageList> spImageList;
if( m_spConsole->QueryScopeImageList(&spImageList) != S_OK ) { ATLTRACE(_T("IConsole::QueryScopeImageList failed\n")); return E_UNEXPECTED; }
hr = LoadImages(spImageList); return hr;}
HRESULT WINAPI CBackupSnapData::UpdateRegistry(BOOL bRegister){ // Load snap-in name
tstring strSnapinName = StrLoadString(IDS_SnapinName);
// Specify the substitution parameters for IRegistrar.
_ATL_REGMAP_ENTRY rgEntries[] = { {TEXT("SNAPIN_NAME"), strSnapinName.c_str()}, {NULL, NULL}, };
// Register the component data object
return _Module.UpdateRegistryFromResource(IDR_BackupSNAP, bRegister, rgEntries);}
HRESULT CBackupSnapData::GetHelpTopic(LPOLESTR* ppszHelpFile){ if( !ppszHelpFile ) return E_POINTER;
USES_CONVERSION; *ppszHelpFile = NULL;
tstring strHelpFileName = StrLoadString(IDS_HELPFILE); if( strHelpFileName.empty() ) return E_FAIL; // Build path to %systemroot%\help
TCHAR szWindowsDir[MAX_PATH+1] = {0}; UINT nSize = GetSystemWindowsDirectory( szWindowsDir, MAX_PATH ); if( (nSize == 0) || (nSize > MAX_PATH) ) return E_FAIL;
tstring strHelpFile = szWindowsDir; // D:\windows
strHelpFile += _T("\\Help\\"); // \help
strHelpFile += strHelpFileName; // \filename.chm
// Show the Help topic
// Form file path in allocated buffer
int nLen = strHelpFile.length() + 1;
*ppszHelpFile = (LPOLESTR)CoTaskMemAlloc(nLen * sizeof(OLECHAR)); if( !*ppszHelpFile ) return E_OUTOFMEMORY;
// Copy into allocated buffer
ocscpy( *ppszHelpFile, T2OLE((LPTSTR)strHelpFile.c_str()) );
return S_OK;}
HRESULT CBackupSnapData::GetLinkedTopics(LPOLESTR* ppszLinkedFiles){ if( !ppszLinkedFiles ) return E_POINTER;
// no linked files
*ppszLinkedFiles = NULL;
return S_FALSE;}
////////////////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////////////////////
//
// CBackupSnapComponent
//
//////////////////////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////////////////////
CBackupSnapComponent::CBackupSnapComponent(){ }
HRESULT CBackupSnapComponent::GetClassID(CLSID* pID){ return m_pComponentData->GetClassID(pID);}
HRESULT CBackupSnapComponent::GetResultViewType( MMC_COOKIE cookie, LPOLESTR* ppViewType, long* pViewOptions ){ if( !ppViewType ) return E_POINTER;
// show standard MMC OCX with message in the result pane
HRESULT hr = StringFromCLSID(CLSID_MessageView, ppViewType); return hr;}
|
