// ProgressDlg.cpp : Implementation of CProgressDlg
#include "stdafx.h"
#include "ProgressDlg.h"

/////////////////////////////////////////////////////////////////////////////
// CProgressDlg

CActionProgressDlg::CActionProgressDlg(IUnknown* pSite, UINT nIDD) 
	: CDialogImplEx(nIDD)
{
	m_pUnk = pSite;
	m_bCancel = false;
}

CActionProgressDlg::~CActionProgressDlg() 
{
	if(IsWindow())
		DestroyWindow();

	// Release any pointers we're holding on to
	if(m_pUnk)
		m_pUnk->Release();
	if(m_pErrorLog)
		m_pErrorLog->Release();
}

LRESULT CActionProgressDlg::OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
	// Used later
	m_ctlProgress = GetDlgItem(IDC_PROGRESS);

	CenterWindow(GetDesktopWindow());

	// This sets up the Event Mechanism
	if(m_pUnk != NULL)
		DispEventAdvise(m_pUnk);	// Pointer from IObjectWithSite

	return 1;  // Let the system set the focus
}

LRESULT CActionProgressDlg::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{

	// Release any pointers we're holding on to
	if(m_pUnk)
	{
		// Drop the Event Stuff
		DispEventUnadvise(m_pUnk);	// Pointer from IObjectWithSite

		m_pUnk->Release();
	}
	if(m_pErrorLog)
		m_pErrorLog->Release();

	return 0;
}

// Update Message from Action Engine
LRESULT CActionProgressDlg::OnUpdateFile(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
	NS_UPDATE_DATA* pData = (NS_UPDATE_DATA*)lParam;
	
	// If there 0 in the total then clear
	if(pData->lTot == 0)
	{
		m_ctlProgress.SetRange(0, 1);
		m_ctlProgress.SetPos(0);
	}
	else
	{
		// If total != -1 then change range
		if(pData->lTot > 0)
			m_ctlProgress.SetRange(0, pData->lTot);
		
		// If Current changed (!= -1) then set state
		if(pData->lCur >= 0)
			m_ctlProgress.SetPos(pData->lCur);
	}

	// If there's a message display it
	if(pData->szMessage1)
		::SetWindowText(GetDlgItem(IDC_MESSAGE1), pData->szMessage1);
	if(pData->szMessage2)
		::SetWindowText(GetDlgItem(IDC_MESSAGE2), pData->szMessage2);

	// If m_bCancel is set then this cancels the CActionEngine
	return m_bCancel ? 0 : 1;
}

// Called in response to the button being pressed, window being closed, or
// event being fired
LRESULT CActionProgressDlg::OnCancel(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
	// This gets checked when the Engine sends and Update (OnUpdateFile)
	m_bCancel = true;

	// We're working on cancelling so reflect that in UI
	::SetWindowText(GetDlgItem(IDCANCEL), _T("Cancelling..."));
	::EnableWindow(GetDlgItem(IDCANCEL), FALSE);
	return 0;
}

// Called when event Fired
STDMETHODIMP CActionProgressDlg::OnCancelEv()
{ 
	// Just call normal cancel event
	BOOL bHandled = TRUE;
	OnCancel(0, 0, NULL, bHandled); 

	return S_OK; 
};

// Error Message from CActionEngine
LRESULT CActionProgressDlg::OnError(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
	// See if we can get an INightSecErrorLog pointer that IObjectWithSite
	// gave us
	if(m_pErrorLog == NULL)
		m_pErrorLog = m_pUnk;

	if(m_pErrorLog == NULL)
		return 0;

	// Create a new NightSecError object
	CComObject<CActionErrorImpl>* pError = NULL;
	HRESULT hr = CComObject<CActionErrorImpl>::CreateInstance(&pError);

	if(FAILED(hr))
		return 0;

	// Set info on the Error Object
	NS_ERROR_DATA* pData = (NS_ERROR_DATA*)lParam;
	pError->SetError(pData->szDesc, pData->hRes);
	pError->Attach((CAction*)pData->lParam);

	// Send it to the INightSecErrorLog
	m_pErrorLog->AddError(pError->GetUnknown());

	return 0;
}

// Starts the AVI file playing from resources
bool CActionProgressDlg::PlayAmination(UINT nID)
{
	ASSERT(IsWindow());
	ASSERT(GetDlgItem(IDC_ANIMATION));

	// Dialog box should have AutoPlay enabled
	return ((BOOL)::SendMessage(GetDlgItem(IDC_ANIMATION), ACM_OPEN, 0, (LPARAM)MAKEINTRESOURCE(nID))) ? true : false;
}