// WMIFilterManager.cpp : Implementation of CWMIFilterManager
#include "stdafx.h"
#include <wbemidl.h>
#include "resource.h"
#include "defines.h"
#include "ntdmutils.h"
#include "SchemaManager.h"
#include "WMIFilterManager.h"
#include "WMIFilterMgrDlg.h"

extern CWMIFilterManagerDlg * g_pFilterManagerDlg;

/////////////////////////////////////////////////////////////////////////////
// CWMIFilterManager

CWMIFilterManager::CWMIFilterManager()
{
	m_hWnd = NULL;
}

//---------------------------------------------------------------------------

CWMIFilterManager::~CWMIFilterManager()
{
}

//---------------------------------------------------------------------------

STDMETHODIMP CWMIFilterManager::ConnectToWMI()
{
	HRESULT hr;
	CComPtr<IWbemLocator>pIWbemLocator;

	NTDM_BEGIN_METHOD()

	m_pIWbemServices = NULL;

	// create the webm locator
	NTDM_ERR_MSG_IF_FAIL(CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER,
			IID_IWbemLocator, (LPVOID *) &pIWbemLocator));

	NTDM_ERR_MSG_IF_FAIL(pIWbemLocator->ConnectServer(	_T("root\\policy"),
													NULL,
													NULL,
													NULL,
													0,
													NULL,
													NULL,
													&m_pIWbemServices));

	NTDM_ERR_MSG_IF_FAIL(CoSetProxyBlanket(m_pIWbemServices, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL,
        RPC_C_AUTHN_LEVEL_CONNECT, RPC_C_IMP_LEVEL_IMPERSONATE  , 
        NULL, EOAC_NONE));

	
	
	NTDM_END_METHOD()

	// cleanup

	return hr;
}

//---------------------------------------------------------------------------

STDMETHODIMP CWMIFilterManager::RunManager(HWND hwndParent, BSTR bstrDomain, VARIANT *vSelection)
{
	HRESULT hr;

	g_pFilterManagerDlg = NULL;

	NTDM_BEGIN_METHOD()

	m_hWnd = hwndParent;

	NTDM_ERR_IF_FAIL(ConnectToWMI());

	g_pFilterManagerDlg = new CWMIFilterManagerDlg(this, false, bstrDomain);
	NTDM_ERRID_IF_NULL(g_pFilterManagerDlg, E_OUTOFMEMORY);

	if(IDOK == DialogBox(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDD_WMI_FILTER_MANAGER), (HWND)hwndParent, WMIFilterManagerDlgProc))
	{
		if(g_pFilterManagerDlg->m_pIWbemClassObject)
		{
			VariantClear(vSelection);

			V_VT(vSelection) = VT_UNKNOWN;
			NTDM_ERR_MSG_IF_FAIL(g_pFilterManagerDlg->m_pIWbemClassObject->QueryInterface(IID_IUnknown, (void **)&(V_UNKNOWN(vSelection))));
		}
		else
			hr = S_FALSE;
	}
	else
	{
		hr = S_FALSE;
	}
	
	NTDM_END_METHOD()

	// cleanup
	NTDM_DELETE_OBJECT(g_pFilterManagerDlg);

	return hr;
}

//---------------------------------------------------------------------------

STDMETHODIMP CWMIFilterManager::RunBrowser(HWND hwndParent, BSTR bstrDomain, VARIANT *vSelection)
{
	HRESULT hr;

	g_pFilterManagerDlg = NULL;

	NTDM_BEGIN_METHOD()

	m_hWnd = hwndParent;

	NTDM_ERR_IF_FAIL(ConnectToWMI());
	
	//brijeshb: wrap with macro for bug 228812 prefix bug. 	
	NTDM_ERR_IF_NULL((g_pFilterManagerDlg = new CWMIFilterManagerDlg(this, true, bstrDomain)));
	if(IDOK == DialogBox(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDD_WMI_FILTER_BROWSER), (HWND)hwndParent, WMIFilterManagerDlgProc))
	{
		if(g_pFilterManagerDlg->m_pIWbemClassObject)
		{
			VariantClear(vSelection);

			V_VT(vSelection) = VT_UNKNOWN;
			NTDM_ERR_MSG_IF_FAIL(g_pFilterManagerDlg->m_pIWbemClassObject->QueryInterface(IID_IUnknown, (void **)&(V_UNKNOWN(vSelection))));
		}
		else
			hr = S_FALSE;
	}
	else
	{
		hr = S_FALSE;
	}
	
	
	NTDM_END_METHOD()

	// cleanup
	NTDM_DELETE_OBJECT(g_pFilterManagerDlg);

	return hr;
}

//--------------------------------------------------------------------------

STDMETHODIMP CWMIFilterManager::SetMultiSelection(VARIANT_BOOL vbValue)
{
	// TODO: Add your implementation code here

	return S_OK;
}