#ifndef __FindReplaceDialogWithMessageFilter_h__
#define __FindReplaceDialogWithMessageFilter_h__

#pragma once

#ifndef __cplusplus
	#error ATL requires C++ compilation (use a .cpp suffix)
#endif

#ifndef __ATLAPP_H__
	#error FindReplaceDialogWithMessageFilter.h requires atlapp.h to be included first
#endif

#ifndef __ATLWIN_H__
	#error FindReplaceDialogWithMessageFilter.h requires atlwin.h to be included first
#endif

class CFindReplaceDialogWithMessageFilter :
	public CFindReplaceDialogImpl<CFindReplaceDialogWithMessageFilter>,
	public CMessageFilter
{
protected:
	CMessageLoop* m_messageLoop;

public:
	CFindReplaceDialogWithMessageFilter(CMessageLoop* messageLoop) :
		m_messageLoop(messageLoop)
	{
	}

public:
	BOOL PreTranslateMessage(MSG* pMsg)
	{
		HWND hWndFocus = ::GetFocus();
		if((m_hWnd == hWndFocus) || this->IsChild(hWndFocus))
			return this->IsDialogMessage(pMsg);

		return FALSE;
	}

	virtual void OnFinalMessage(HWND /*hWnd*/)
	{
		delete this;
	}

	BEGIN_MSG_MAP(CFindReplaceDialogWithMessageFilter)
		MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
		MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
	END_MSG_MAP()

	LRESULT OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
	{
		if(m_messageLoop)
			m_messageLoop->AddMessageFilter(this);

		bHandled = FALSE;
		return 0;
	}

	LRESULT OnDestroy(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& bHandled)
	{
		if(m_messageLoop)
			m_messageLoop->RemoveMessageFilter(this);

		bHandled = FALSE;
		return 0;
	}
};

#endif //__FindReplaceDialogWithMessageFilter_h__