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


// 2009/12/22 Moddified base class to be SWbemBaseObjectSet not SWbemBaseObject
// 2012/03/08 Updated.

#pragma once

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

namespace SOL {

class SWbemPrivilegeSet : public SWbemBaseObjectSet {

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

    }

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

public:
    SWbemPrivilegeSet& operator=(ISWbemPrivilegeSet* pDisp)
    { 
        set(pDisp);
        return *this;
    }

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

        if (FAILED(hr = set -> get__NewEnum(&pUnk))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return pUnk;
    }        
        
public:
    ISWbemPrivilege* item( 
            __in WbemPrivilegeEnum iPrivilege)
    {
        ISWbemPrivilege* pObjWbemPrivilege = NULL;
        HRESULT hr = E_FAIL;
        ISWbemPrivilegeSet* set = get();

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

        if (FAILED(hr = set -> get_Count( 
            &iCount))) {

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

public:
    ISWbemPrivilege* add( 
            __in WbemPrivilegeEnum iPrivilege,
           __in bool bIsEnabled)
    {
        ISWbemPrivilege* pObjWbemPrivilege = NULL;
        HRESULT hr = E_FAIL;
        ISWbemPrivilegeSet* set = get();

        if (FAILED(hr = set -> Add( 
            iPrivilege,
            toVariantBool(bIsEnabled),
            &pObjWbemPrivilege))) {

            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return pObjWbemPrivilege;
    }        
        
public:
    HRESULT remove( 
            __in WbemPrivilegeEnum iPrivilege)
    {
        HRESULT hr = E_FAIL;
        ISWbemPrivilegeSet* set = get();

        if (FAILED(hr = set -> Remove( 
            iPrivilege))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return hr;
    }        
        
public:
    HRESULT deleteAll()
    {
        HRESULT hr = E_FAIL;
        ISWbemPrivilegeSet* set = get();

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

    //2009/12/20
public:
     ISWbemPrivilege* addAsString( 
            __in _bstr_t strPrivilege,
           __in bool bIsEnabled)
    {
        HRESULT hr = E_FAIL;
        ISWbemPrivilegeSet* set = get();
        ISWbemPrivilege *objWbemPrivilege = NULL;

        if (FAILED(hr = set -> AddAsString( 
                (BSTR)strPrivilege,
                toVariantBool(bIsEnabled),
                &objWbemPrivilege))) {
            throw Exception(E_FAIL, "%s: %s", "E_FAIL",__FUNCTION__);
        }
        return objWbemPrivilege;
    }        

public:
    void write(Writer& writer)
    {
        writer.writeln(L"<PrivilegeSet>");

        try {
            EnumVariant enumVar = ComObjectSet::getNewEnum();
            IDispatch* pDisp = NULL;
    
            while ((pDisp = enumVar.next()) !=NULL ) {
                try {
                    SWbemPrivilege privilege = pDisp;
                    privilege.write(writer);
                } catch (...) {

                }
            }
        } catch (...) {

        }
        writer.writeln(L"</PrivilegeSet>");

    }

};

}