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


// SOL9

// 2009/12/21 Modified to use SOL::ReportWriter class to save a query result.
// 2009/12/22 Modified to show an error name for an exception caught by catch caluse.
// 2012/03/08 Updated.

#pragma once

#include <sol/Locale.h>
#include <sol/Exception.h>
#include <sol/Thread.h>
#include <sol/com/ApartmentThreadedModel.h>
#include <sol/com/MultiThreadedModel.h>
#include <sol/wmi/SWbemQueryProperties.h>
#include <sol/wmi/SWbemQueryPropertiesLoader.h>

#include <sol/wmi/SWbemLocator.h>
#include <sol/wmi/SWbemServices.h>
#include <sol/wmi/SWbemObjectSet.h>

//2003/12/22
#include <sol/LocalDateTime.h>
#include <sol/wmi/WbemErrors.h>


namespace SOL {

/**
 * Thread for SWbemServices::ExecQuery method call
 */
class SWbemQueryApplet: public Thread {

private:
    SWbemQueryProperties properties;


public:    
    /**
     * Constructor
     */
    SWbemQueryApplet()
    {
        SWbemQueryPropertiesLoader loader;
        loader.load(properties);
    }


public:
    //Redefine this method in your own sublcass to inherit this class.

    virtual void display(SWbemObjectSet& objectSet)
    {
        long count = objectSet.getCount();
        
        _bstr_t reportFile = properties.getReportFile();

        FileWriter writer(reportFile);

        _bstr_t query = properties.getQuery();

        writer.writeln(L"[SWbemQuery result for \"%s\"]", (const wchar_t*)query);

        LocalDateTime ldt;
        StringT<wchar_t> now;
        writer.writeln(L"(Generated on %s)", ldt.nowToSeconds(now));

        for (long i = 0; i<count; i++) {
            try {
                SWbemObject object = objectSet.itemIndex(i);

                _bstr_t text = object.getObjectText(0);
                //printf("\n\nObjectText No:%d %S\n", i, (const wchar_t*)text);
                
                writer.writeln(L"\r\n\r\nObjectText No:%d %s", i, (const wchar_t*)text);
                            
            } catch (...) {

            }     
        }
    }

public:
    SWbemQueryProperties& getQueryProperties()
    {
        return properties;
    }

public:
    /**
     * Thread procedure to watch an instance creation of Win32_Process.
     */
    virtual void run()
    {
        ApartmentThreadedModel apartmentThreaded;

        //MultiThreadedModel   multiThreaded;

        try {
    
            SWbemLocator locator;

            _bstr_t server = properties.getServer();
            _bstr_t nspace = properties.getNamespace();
            _bstr_t query  = properties.getQuery();

            SWbemServices services = locator.connectServer((BSTR)server, (BSTR)nspace);
                

            SWbemObjectSet objectSet = services.execQuery((BSTR)query);

            display(objectSet);

        } catch (HRESULT hr) {
            WbemErrors errors;
            printf("SWbemQueryApplet,Exception:HRESULT=%08x %s", hr, errors.getName(hr));
        } catch (...) {
            printf("SWbemQueryApplet,Unknown Exception\n");
        }
    }

};

}