// CmdLineApp.cpp: implementation of the CCmdLineApp class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "CmdLineApp.h"
#include "CmdLineDlg.h"

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

CCmdLineApp::CCmdLineApp()
	: CNightSecApp(_T("Secure Shutdown"))
{ }

CCmdLineApp::~CCmdLineApp()
{ }

bool CCmdLineApp::InitInstance()
{
	if(!CNightSecApp::InitInstance())
		return false;

	// Get Parameters
	bool bParams = false;
	bool bShowHelp = true;
	LPSTR szChar;
	int nIndex = -1;
	
	// Loop through command line arguments
	for (int i = 1; i < __argc; i++)
	{
		szChar = __targv[i];
		_tcslwr(szChar);

		// Check if it's a param
		// Note we can use - as a param because we don't
		// accept file names (otherwise - can be in file name)
		if (szChar[0] == '/' || szChar[0] == '-')
		{
			// Accepts multiple arguments in one param
			// eg: -rti
			
			// First skip over first token then go through
			// other chars one by one

			for(szChar++; szChar[0] != '\0'; szChar++)
			{
				// Check if we have a match
				if((nIndex = FindParam(szChar[0])) > -1) 
				{

					// If so then call it
					g_aComponents[nIndex]->DoShutdown(NULL);

					// Set flag so we don't show dialog later on
					bShowHelp = false;

				}

			}
		} 
			
	}

	// Return if no dialog
	if(!bShowHelp)
		return false;

	// Otherwise...
	CCmdLineDlg dlg;
	dlg.DoModal();

	// Since the dialog has been closed, return FALSE so that we exit the
	//  application, rather than start the application's message pump.
	return false;
}

//////////////////////////////////////////////////////////////////
// Checks Component Array for any components with Param passed

UINT CCmdLineApp::FindParam(TCHAR chParam)
{
	string sTemp;

	// Loop through'm all
	for(int nCnt = 0; nCnt < g_aComponents.size(); nCnt++) 
	{
		// Only Windows Components can be called from cmd line
		if(g_aComponents[nCnt]->GetType() == COMPONENT_WIN)
		{
			// Check for match
			sTemp = g_aComponents[nCnt]->GetInfoStr(nsCmdLine, _T(""));
			sTemp.make_lower();

			if(!sTemp.empty() && sTemp[0] == chParam)
				return nCnt;
		}
	}

	return -1;

}