//%2006////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2000, 2001, 2002 BMC Software; Hewlett-Packard Development
// Company, L.P.; IBM Corp.; The Open Group; Tivoli Systems.
// Copyright (c) 2003 BMC Software; Hewlett-Packard Development Company, L.P.;
// IBM Corp.; EMC Corporation, The Open Group.
// Copyright (c) 2004 BMC Software; Hewlett-Packard Development Company, L.P.;
// IBM Corp.; EMC Corporation; VERITAS Software Corporation; The Open Group.
// Copyright (c) 2005 Hewlett-Packard Development Company, L.P.; IBM Corp.;
// EMC Corporation; VERITAS Software Corporation; The Open Group.
// Copyright (c) 2006 Hewlett-Packard Development Company, L.P.; IBM Corp.;
// EMC Corporation; Symantec Corporation; The Open Group.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// THE ABOVE COPYRIGHT NOTICE AND THIS PERMISSION NOTICE SHALL BE INCLUDED IN
// ALL COPIES OR SUBSTANTIAL PORTIONS OF THE SOFTWARE. THE SOFTWARE IS PROVIDED
// "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
// LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
// ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
//==============================================================================
//
// Author: Barbara Packard (barbara_packard@hp.com)
//
// Modified By: Jair Santos, Hewlett-Packard Company (jair.santos@hp.com)
//              Terry Martin, Hewlett-Packard Company (terry.martin@hp.com)
//
//%/////////////////////////////////////////////////////////////////////////////

// WMIQueryProvider.cpp: implementation of the WMIQueryProvider class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"


#include "WMICollector.h"
#include "WMIBaseProvider.h"
#include "WMIClassProvider.h"
#include "WMIInstanceProvider.h"
#include "WMIAssociatorProvider.h"
#include "WMIQueryProvider.h"

#include "WMIObjectPath.h"
#include "WMIProperty.h"
#include "WMIString.h"
#include "WMIValue.h"
#include "WMIQualifier.h"
#include "WMIQualifierSet.h"
#include "WMIType.h"
#include "WMIException.h"

//////////////////////////////////////////////////////////////////////////////
// WMIQueryProvider::
//
// ///////////////////////////////////////////////////////////////////////////
PEGASUS_NAMESPACE_BEGIN

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

WMIQueryProvider::WMIQueryProvider()
{
	_collector = NULL;
	m_bInitialized = false;

}

WMIQueryProvider::~WMIQueryProvider()
{
	cleanup();

}

/////////////////////////////////////////////////////////////////////////////
// WMIQueryProvider::execQuery
//
// ///////////////////////////////////////////////////////////////////////////
Array<CIMObject> WMIQueryProvider::execQuery(
		const String& nameSpace,
		const String& userName,
		const String& password,
        const String& queryLanguage,
        const String& query,
		const CIMPropertyList& propertyList,
		Boolean includeQualifiers,
		Boolean includeClassOrigin)
{
	HRESULT hr;
	long lCount = 0;
	DWORD dwReturned;
	bool bInst;

	CComPtr<IEnumWbemClassObject>	pObjEnum;
	CComPtr<IWbemClassObject>		pObject;

	Array<CIMObject> objects;
	CIMName className;

	PEG_METHOD_ENTER(TRC_WMIPROVIDER,"WMIQueryProvider::execQuery()");

	setup(nameSpace,userName,password);

	if (!m_bInitialized)
	{
		throw CIMException(CIM_ERR_FAILED);
	}

	// retrieve results
	if (!(_collector->getQueryResult(&pObjEnum, query, queryLanguage)))
	{
		if (pObjEnum)
			pObjEnum.Release();

		throw CIMException(CIM_ERR_FAILED);
	}

	//set proxy security on pObjEnum
	bool bSecurity = _collector->setProxySecurity(pObjEnum);
	
	//get the results and append them to the array
	hr = pObjEnum->Next(WBEM_INFINITE, 1, &pObject, &dwReturned);
	
	if (SUCCEEDED(hr) && (1 == dwReturned))
	{	
		bInst = _collector->isInstance(pObject);	

        CComVariant mVar;
        CComBSTR bs;
        CMyString mstr;  
	    
        HRESULT hRes = pObject->Get(L"__CLASS", 0, &mVar, 0, 0);
        if (SUCCEEDED(hRes)) {
           bs = mVar.bstrVal;
           mstr = bs;
           className = CIMName((LPCTSTR)mstr);
        }
        mVar.Clear();

	}

	
	while (SUCCEEDED(hr) && (1 == dwReturned))
	{
		// collect the information about the current object
		if (bInst)
		{
			//get class from the returned instance 
			//it will avoid "type mismatch" exceptions
			//when deepInheritance is true and instances
			//of subclasses are returned
			CComVariant vTmpClassName;
			String strTmpClassName;
			if (pObject->Get(L"__CLASS", 0, &vTmpClassName, NULL, NULL) == S_OK)
			{
				strTmpClassName = WMIString(vTmpClassName);
			}

			CIMInstance tempInst(strTmpClassName);

			if (_collector->getCIMInstance(pObject, tempInst,
											false, includeQualifiers, includeClassOrigin,
											propertyList, 
											true))  //get key properties
			{
				lCount++;

			    // build the object path
			    CComVariant v;
			    hr = pObject->Get(L"__PATH", 
				                    0,
								    &v,
								    NULL,
								    NULL);

			    WMIObjectPath tempRef(v.bstrVal);
			    tempInst.setPath(tempRef);
			    v.Clear();
				
				objects.append(CIMObject(tempInst));
			}
		}
		else
		{
			// we are collecting a class
			CIMClass cimClass;
            String superClass = _collector->getSuperClass(pObject);
			CIMName objName = className;

			cimClass = CIMClass(objName);

			if (0 != superClass.size())
			{
				CIMName superClassName = superClass;
				cimClass.setSuperClassName(superClassName);
			}

            if (_collector->getCIMClass(pObject, 
				                        cimClass,
				                        false,
				                        includeQualifiers,
				                        includeClassOrigin,
				                        propertyList))
            {
				lCount++;
				objects.append(CIMObject(cimClass));
            }
		}

		if (pObject)
			pObject.Release();

		hr = pObjEnum->Next(WBEM_INFINITE, 1, &pObject, &dwReturned);

	    if (SUCCEEDED(hr) && (1 == dwReturned))
	    {
		    bInst = _collector->isInstance(pObject);
	    }
    }

	if (pObjEnum)
		pObjEnum.Release();

	PEG_TRACE((TRC_WMIPROVIDER, Tracer::LEVEL3,
		"WMIQueryProvider::execQuery() - Result count is %d", lCount)); 

	if (lCount == 0)
	{
		PEG_TRACE((TRC_WMIPROVIDER, Tracer::LEVEL3,
			"WMIQueryProvider::execQuery() - hResult value is %x", hr)); 
	}

	PEG_METHOD_EXIT();

	return objects;
}

PEGASUS_NAMESPACE_END