// NightSecWizard.h: interface for the CNightSecWizard class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_NIGHTSECWIZARD_H__E7FC7C13_8F86_11D3_BFBC_0020182B97FC__INCLUDED_)
#define AFX_NIGHTSECWIZARD_H__E7FC7C13_8F86_11D3_BFBC_0020182B97FC__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "Checklist.h"
#include "../common/Interfaces.h"
#include "../common/proppageholder.h"

#include "contexthelp.h"

#define countof(a) (sizeof(a) / sizeof(a[0]))

#include <map>
using std::multimap;
typedef multimap < CComponentHolder*, CPropPageHolder* > PropertyPageMap;

#include <atlprsht.h>

class ATL_NO_VTABLE CNightSecWizard : 
	public CComObjectRootEx<CComMultiThreadModel>,
	public IDispatchImpl<INightSecSiteInfo, &IID_INightSecSiteInfo, &LIBID_CHECKLISTLib>,
	public CPropertySheet,
	public CContextHelp<CNightSecWizard>
{

// Construction
public:
	CNightSecWizard() 
		: CPropertySheet(_T("Night Security Checklist"), NULL, 0) 
	{ 
		SetWizardMode();
		SetHelp(true);
	}

	virtual ~CNightSecWizard();

// For Base Classes
protected:
	HRESULT LoadPropertyPages(CComponentHolder* pComponent);
	IUnknown* GetIUnknown(bool bAddRef = false);
	void SetSite(IUnknown* pSite = NULL);
	void ClearSite();

// COM Interface support
DECLARE_PROTECT_FINAL_CONSTRUCT()

BEGIN_COM_MAP(CNightSecWizard)
	COM_INTERFACE_ENTRY(IDispatch)
	COM_INTERFACE_ENTRY(INightSecSiteInfo)
END_COM_MAP()

BEGIN_MSG_MAP(CNightSecWizard)
	MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
	CHAIN_MSG_MAP(CContextHelp<CNightSecWizard>)
	CHAIN_MSG_MAP(CPropertySheet)
END_MSG_MAP()

BEGIN_HELP_MAP(NS_HELP_FILE)
	HELP_ID(ID_WIZBACK, 502)
	HELP_ID(ID_WIZNEXT, 501)
	HELP_ID(ID_WIZFINISH, 503)
END_HELP_MAP

	LRESULT OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);

// Data
public:
	PropertyPageMap m_mapPages;
};


class CAdvancedWizardT 	: public CNightSecWizard
{
public:
	CAdvancedWizardT()
		{  }
	virtual ~CAdvancedWizardT() {}

// Message Map
BEGIN_MSG_MAP(CAdvancedWizardT)
	MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
	MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
	CHAIN_MSG_MAP(CNightSecWizard)
END_MSG_MAP()

// Message Handlers
protected:
	LRESULT OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
	LRESULT OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);

// INightSecSiteInfo
public:
	STDMETHOD(get_Info)(/*[in]*/ NightSecSiteInfo nsItem, /*[out, retval]*/ VARIANT* pvVal);

};

typedef CComObjectMember<CAdvancedWizardT> CAdvancedWizard;



#include "ExtraPage.h"

#include <vector>
using std::vector;
typedef vector<CExtraPage*> ExtraPageArray;


class CNormalWizardT : public CNightSecWizard
{
public:
	CNormalWizardT()
	{ 
		m_bSiteSet = false;
	}
	virtual ~CNormalWizardT();

// Message Map
BEGIN_MSG_MAP(CNormalWizardT)
	MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
	MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
	CHAIN_MSG_MAP(CNightSecWizard)
END_MSG_MAP()

// Message Handlers
protected:
	LRESULT OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
	LRESULT OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
	
// Operations
public:
	void AddExtraPages();

// INightSecSiteInfo
public:
	STDMETHOD(get_Info)(/*[in]*/ NightSecSiteInfo nsItem, /*[out, retval]*/ VARIANT* pvVal);

protected:
	ExtraPageArray m_aPages;
	bool m_bSiteSet;
};

typedef CComObjectMember<CNormalWizardT> CNormalWizard;

#endif // !defined(AFX_NIGHTSECWIZARD_H__E7FC7C13_8F86_11D3_BFBC_0020182B97FC__INCLUDED_)