/*
 * SWbemQualifierSet.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/SWbemQualifier.h>

namespace SOL {

class SWbemQualifierSet : public SWbemBaseObjectSet {

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

    }

public:
    ISWbemQualifierSet* get()
    {
        return (ISWbemQualifierSet*)getIDispatch();
    }
 
public:
    SWbemQualifierSet& operator=(ISWbemQualifierSet* pDisp)
    { 
        set(pDisp);
        return *this;
    }

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

        if (FAILED(hr = set->get__NewEnum( 
                &pUnk))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return pUnk;
    }
        
public:
    ISWbemQualifier* item( 
            __in _bstr_t name,
           __in long iFlags)
    {
        ISWbemQualifier* pObjWbemQualifier = NULL;
        HRESULT hr = E_FAIL;
        ISWbemQualifierSet* set = get();

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

public:
    long getCount()
    {
        HRESULT hr = E_FAIL;
        ISWbemQualifierSet* set = get();
        long iCount = 0;

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

        
public:
    ISWbemQualifier* add( 
            __in _bstr_t strName,
            __in VARIANT *varVal,
           __in bool bPropagatesToSubclass,
           __in bool bPropagatesToInstance,
           __in bool bIsOverridable,
           __in long iFlags)
    {
        ISWbemQualifier* pObjWbemQualifier = NULL;
        HRESULT hr = E_FAIL;
        ISWbemQualifierSet* set = get();

        if (FAILED(hr = set->Add( 
                (BSTR)strName,
                varVal,
                toVariantBool(bPropagatesToSubclass),
                 toVariantBool(bPropagatesToInstance),
                toVariantBool(bIsOverridable),
                iFlags,
                &pObjWbemQualifier))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return pObjWbemQualifier;
    }
        
public:
    HRESULT remove( 
            __in _bstr_t strName,
           __in long iFlags = 0)        
    {
        HRESULT hr = E_FAIL;
        ISWbemQualifierSet* 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.write(L"<QualifierSet>\n");
        try {
            IDispatch* pDisp = NULL;
            EnumVariant enumVar = ComObjectSet::getNewEnum();
             
            while (enumVar.next(1, &pDisp) ==1) {
                    
                try {                
                    SWbemQualifier qualifier = (ISWbemQualifier*)pDisp;
                    qualifier.write(writer);

                } catch (...) {
                    break;
                }
            }
        } catch (...) {
            
        }
        writer.write(L"</QualifierSet>\n");

    }
};

}