// ReportEvent.h
// This module contains helpers for the WMIReportEvent function.

#pragma once

#include "NCObjApi.h"
#include <wstlallc.h>
#include <map>

class CReportParams
{
public:
    CReportParams(LPCWSTR szName, LPCWSTR szFormat)
    {
        m_szName = szName;
        m_szFormat = szFormat;
    }

    bool operator < (const CReportParams& other) const
    { 
        return m_szName < other.m_szName && m_szFormat < other.m_szFormat;
    }

    bool operator == (const CReportParams& other) const
    {
        return m_szName == other.m_szName && m_szFormat == other.m_szFormat;
    }

    bool IsEquivalent(const CReportParams& other) const
    {
        // The format string is case-senstive due to printf format characters.
        return !_wcsicmp(m_szName, other.m_szName) && 
            !wcscmp(m_szFormat, other.m_szFormat);
    }

protected:
    LPCWSTR m_szName;
    LPCWSTR m_szFormat;
};

class CReportItem
{
public:
    CReportItem( )
        : m_szName( NULL ),
          m_szFormat( NULL ),
          m_hEvent( NULL )
    {
    }

    ~CReportItem()
    {
        if (m_szName)
            free(m_szName);

        if (m_szFormat)
            free(m_szFormat);

        if (m_hEvent)
            WmiDestroyObject(m_hEvent);            
    }

    BOOL Initialize( LPCWSTR szName, LPCWSTR szFormat, HANDLE hEvent )
    {
        if ( szName && szFormat && hEvent )
        {
            m_szName = _wcsdup(szName);
            if ( NULL == m_szName )
            {
                return FALSE;
            }
            
            m_szFormat = _wcsdup(szFormat);
            if ( NULL == m_szFormat )
            {
                return FALSE;
            }
            
            m_hEvent = hEvent;

            return TRUE;
        }

        return FALSE;
    }
    
    HANDLE GetEvent() { return m_hEvent; }
    
protected:
    LPWSTR m_szName;
    LPWSTR m_szFormat;
    HANDLE m_hEvent;
};

class CReportEventMap : protected std::map< CReportParams, 
                                            CReportItem*, 
                                            std::less< CReportParams >, 
                                            wbem_allocator< CReportItem* > >
{
public:
    ~CReportEventMap();

    HANDLE GetEvent(
        HANDLE hConnection, 
        LPCWSTR szName, 
        LPCWSTR szFormat);

    HANDLE CreateEvent(
        HANDLE hConnection, 
        LPCWSTR szName, 
        DWORD dwFlags,
        LPCWSTR szFormat);

protected:
    typedef CReportEventMap::iterator CReportEventMapIterator;
    static CIMTYPE PrintfTypeToCimType(LPCWSTR szType);
};