/*
 * SWbemPropertySet.h 
 * Copyright (c) 2012 Antillia.com TOSHIYUKI ARAI. ALL RIGHTS RESERVED. 
 */


// 2012/03/08 Updated\.

#pragma once

#include <sol/wmi/SWbemBaseObjectSet.h>
#include <sol/wmi/SWbemProperty.h>


namespace SOL {

class SWbemPropertySet : public SWbemBaseObjectSet {

public:
    SWbemPropertySet(IDispatch* pDisp=NULL)
    :SWbemBaseObjectSet(pDisp)
    {

    }

public:
    ISWbemPropertySet* get()
    {
        return (ISWbemPropertySet*)getIDispatch();
    }

public:
    SWbemPropertySet& operator=(ISWbemPropertySet* pDisp)
    { 
        set(pDisp);
        return *this;
    }
 

public:
    IUnknown* get__NewEnum()
    {
        IUnknown* pUnk = NULL;
        HRESULT hr = E_FAIL;
        ISWbemPropertySet* set = get();

        if (FAILED(hr = set->get__NewEnum( 
            &pUnk))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return pUnk;
    }
        

public:
    ISWbemProperty* item( 
            __in _bstr_t strName,
           __in long iFlags)
    {
        HRESULT hr = E_FAIL;
        ISWbemPropertySet* set = get();
        ISWbemProperty *objWbemProperty = NULL;

        if (FAILED(hr = set->Item( 
                (BSTR)strName,
                iFlags,
                &objWbemProperty))) {

            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return objWbemProperty;
    }

public:
    long getCount()
    {
        HRESULT hr = E_FAIL;
        ISWbemPropertySet* set = get();
        long iCount = 0;
        if (FAILED(hr = set->get_Count( 
            &iCount))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return iCount;
    }
        

public:
    ISWbemProperty* add( 
            __in _bstr_t strName,
            __in WbemCimtypeEnum iCIMType,
           __in bool bIsArray,
           __in long iFlags)
    {
        ISWbemProperty* pObjWbemProperty = NULL;
        HRESULT hr = E_FAIL;
        ISWbemPropertySet* set = get();

        if (FAILED(hr = set->Add( 
                (BSTR)strName,
                iCIMType,
                toVariantBool(bIsArray),
                iFlags,
                &pObjWbemProperty))) {

            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return pObjWbemProperty;
    }
        

public:
    HRESULT remove( 
            __in _bstr_t strName,
           __in long iFlags = 0)
    {
        HRESULT hr = E_FAIL;
        ISWbemPropertySet* set = get();

        if (FAILED(hr = set->Remove( 
                (BSTR)strName,
                iFlags))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return hr;
    }

public:
    void write(Writer& writer)
    {
        long c = getCount();

        writer.writeln(L"<PropertySet>");
        
        try {
            EnumVariant enumVar = ComObjectSet::getNewEnum();
            IDispatch* pDisp = NULL;
            while (enumVar.next(1, &pDisp) == 1) {
        
                try {
                    SWbemProperty prop = (ISWbemProperty*)pDisp;
                    prop.write(writer);
                } catch (...) {
                    break;
                }
            }

        } catch (...) {
            
        }
        writer.writeln(L"</PropertySet>");

    }
};

}