// NightSecApp.cpp: implementation of the CNightSecApp class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "NightSecApp.h"

#include <regsvr.h>
#include <appmisc.h>
#include <path.h>
#include "../common/defines.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CComponentArray g_aComponents;

#define NS_REGISTER_FAILED _T("Night Security was unable to register the %s component.\n\n%s\nIf Night Security doesn't seem to be working properly please reinstall.")
const DWORD dwTimeOut = 5000; // time for EXE to be idle before shutting down

CNightSecApp::CNightSecApp(const string& sAppName, const string& sMtxName)
	: m_sAppName(sAppName), m_oneInstance(sMtxName), 
	  m_settings(HKEY_CURRENT_USER, NS_REG_MAIN)
{

}

CNightSecApp::~CNightSecApp()
{

}

bool CNightSecApp::InitInstance()
{
	LPCTSTR szCmdLine = GetCommandLine();

	// If we've been asked to close then exit here
	if(_tcsstr(szCmdLine, _T("close")) != NULL)
		return false;

	// Load up all Components
	if(!g_aComponents.LoadComponents())
	{
		RegisterDlls();
		g_aComponents.LoadComponents();
	}

	return true;
}

bool CNightSecApp::ExitInstance() 
{
	g_aComponents.UnloadComponents();
	return false;
}

HRESULT CNightSecApp::RegisterDlls(bool bUI /*= true*/)
{
	file_iterator iterFiles(GetProgramFolder(_Module.m_hInst), 
							_T("*.dll"), file_iterator::full_paths);

	file_iterator end;

	HRESULT hRet = S_OK;

	for( ; iterFiles != end; iterFiles++)
	{
		HRESULT hr = RegisterDLL(iterFiles->cFileName);

		if(SUCCEEDED(hr) || !bUI)
			continue;

		string sError;
		file_path fileDll(iterFiles->cFileName);
		sError.format(NS_REGISTER_FAILED, (LPCTSTR)fileDll.file(), (LPCTSTR)FormatHR(hr));

		MessageBox(NULL, sError, m_sAppName, MB_OK | MB_ICONEXCLAMATION);
	}

	return hRet;
}

// Passed to CreateThread to monitor the shutdown event
/*static DWORD WINAPI MonitorProc(void* pv)
{
    CNightSecApp* p = (CNightSecApp*)pv;
    p->MonitorShutdown();
    return 0;
}*/

/*LONG CNightSecApp::Unlock()
{
    LONG l = CComModule::Unlock();
    if (l == 0)
    {
        bActivity = true;
        SetEvent(hEventShutdown); // tell monitor that we transitioned to zero
    }
    return l;
}
*/
//Monitors the shutdown event
/*void CNightSecApp::MonitorShutdown()
{
    while (1)
    {
        WaitForSingleObject(hEventShutdown, INFINITE);
        DWORD dwWait=0;
        do
        {
            bActivity = false;
            dwWait = WaitForSingleObject(hEventShutdown, dwTimeOut);
        } while (dwWait == WAIT_OBJECT_0);
        // timed out
        if (!bActivity && m_nLockCnt == 0) // if no activity let's really bail
        {
#if _WIN32_WINNT >= 0x0400 & defined(_ATL_FREE_THREADED)
            CoSuspendClassObjects();
            if (!bActivity && m_nLockCnt == 0)
#endif
                break;
        }
    }
    CloseHandle(hEventShutdown);
    PostThreadMessage(dwThreadID, WM_QUIT, 0, 0);
}

bool CNightSecApp::StartMonitor()
{
    hEventShutdown = CreateEvent(NULL, false, false, NULL);
    if (hEventShutdown == NULL)
        return false;
    DWORD dwThreadID;
    HANDLE h = CreateThread(NULL, 0, MonitorProc, this, 0, &dwThreadID);
    return (h != NULL);
}
*/