/*
 * SWbemMethodSet.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/SWbemMethod.h>

namespace SOL {

class SWbemMethodSet :public SWbemBaseObjectSet {

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

    }

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

public:
    SWbemMethodSet& operator=(ISWbemMethodSet* pDisp)
    { 
        set(pDisp);
        return *this;
    }
    
public:
    IUnknown* get__NewEnum()
    {
        IUnknown* pUnk = NULL;
        HRESULT hr = E_FAIL;
        ISWbemMethodSet* set = get();

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

public:
    ISWbemMethod* item( 
            __in _bstr_t strName,
           __in long iFlags)
    {
        ISWbemMethod* pObjWbemMethod = NULL;
        HRESULT hr = E_FAIL;
        ISWbemMethodSet* set = get();

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

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

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


/*public:
    void display()
    {

    }
*/

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

        writer.writeln(L"<MethodSet>");
        
        writer.writeln(L"<Count>%d</Count>", c);

        try {
            EnumVariant enumVar = ComObjectSet::getNewEnum();
            IDispatch* pDisp = NULL;
            while (enumVar.next(1, &pDisp) ==1) {
                    
                try {                
                    SWbemMethod method = (ISWbemMethod*)pDisp;
                    method.write(writer);
                } catch (...) {
                    break;
                }
            }

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

    }
};


}