/*

* Copyright (c) 2007-2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

* Commandline handler for imgcheck Tool, responsible to parse the

* commandline options and preserve the data for later use

*

*/

/**

 @file

 @internalComponent

 @released

*/

#include "cmdlinehandler.h"

/**

Constructor initializes the iOptionMap with short and long option names as key and

the value pair says whether the respective option can have value or not.

@internalComponent

@released

*/

CmdLineHandler::CmdLineHandler()

:iDebuggableFlagVal(false),iXmlFileName(GXmlFileName), iNoImage(true), iCommmandFlag(0), iValidations(0), iSuppressions(0) 

{

	iOptionMap[KLongHelpOption] = ENone;

	iOptionMap[KLongAllOption] = ENone;

	iOptionMap[KLongXmlOption] = ENone;

	iOptionMap[KLongOutputOption] = ESingle; //option can have only 1 value

	iOptionMap[KLongQuietOption] = ENone;

	iOptionMap[KLongVerboseOption] = ENone;

	iOptionMap[KLongSuppressOption]= EMultiple; //This value should be updated, while introducing new validations

	iOptionMap[KLongVidValOption]= EMultiple;

	iOptionMap[KShortHelpOption] = ENone;

	iOptionMap[KShortAllOption] = ENone;

	iOptionMap[KShortXmlOption] = ENone;

	iOptionMap[KShortOutputOption] = ESingle; //option can have only 1 value

	iOptionMap[KShortQuietOption] = ENone;

	iOptionMap[KShortVerboseOption] = ENone;

	iOptionMap[KShortSuppressOption] = EMultiple;

	iOptionMap[KShortNoCheck] = ENone;

	iOptionMap[KLongSidAllOption] = ENone;

	iOptionMap[KLongEnableDepCheck] = ENone;

	iOptionMap[KLongEnableSidCheck] = ENone;

	iOptionMap[KLongEnableVidCheck] = ENone;

	iOptionMap[KLongEnableDbgFlagCheck] = EOptional;

	iOptionMap[KLongE32InputOption] = ENone;

	iOptionMap[KLongNoCheck] = ENone;

	iSuppressVal[KSuppressDependency] = EDep;

	iSuppressVal[KSuppressSid] = ESid;

	iSuppressVal[KSuppressVid] = EVid;

	Version();

	Usage();

}

/**

Destructor.

@internalComponent

@released

*/

CmdLineHandler::~CmdLineHandler()

{

	iOptionMap.clear();

	iImageNameList.clear();

	iSuppressVal.clear();

	iVidValList.clear();

}

/**

Function to parse the command line options.

Responsible to

1. Parse the input values.

2. Print the usage note. 

3. Identify the valdations to be carried out.

4. Type of report needs to be generated.

@internalComponent

@released

@param aArgc - argument count

@param aArgv[] - argument values

*/

ReturnType CmdLineHandler::ProcessCommandLine(unsigned int aArgc, char* aArgv[])

{

	if(aArgc < 2)

	{

		std::cout << PrintVersion().c_str() << std::endl;

		std::cout << PrintUsage().c_str() << std::endl;

		return EQuit;

	}

	ArgumentList argumentList(&aArgv[0], aArgv + aArgc);

	int argCount = argumentList.size();

	 iInputCommand = KToolName;

	for( int i = 1; i < argCount; i++ ) //Skip tool name

	{

		String name = argumentList.at(i);

		iInputCommand += " ";

		iInputCommand += name;

		int longOptionFlag = 0;

		if(IsOption(name, longOptionFlag))

		{

			String optionName;

			bool optionValue = false;

			StringList optionValueList;

			ParseOption(name, optionName, optionValueList, optionValue);

			char shortOption = KNull;

			if(Validate(ReaderUtil::ToLower(optionName), optionValue, optionValueList.size()))

			{

				if(longOptionFlag)

				{

					shortOption = optionName.at(2);

				}

				else

				{

					shortOption = optionName.at(1);

				}

			}

			switch(shortOption)

			{

				case 'q':

					iCommmandFlag |= QuietMode;

					break;

				case 'a':

					iCommmandFlag |= KAll;

					break;

				case 'x':

					iCommmandFlag |= KXmlReport;

					break;

				case 'o':

					iXmlFileName.assign(optionValueList.front());

					NormaliseName();

					break;

				case 's':

					if((optionName == KShortSuppressOption) || (optionName == KLongSuppressOption))

					{

						String value;

						while(optionValueList.size() > 0)

						{

							value = optionValueList.front();

							if(iSuppressVal[value])

							{

								if(iValidations > 0) //Is any check enabled?

								{

									if(iValidations & iSuppressVal[value])

									{

										iValidations ^= iSuppressVal[value]; //Consider only 3 LSB's

									}

								}

								else //Is this valid value?

								{

									iSuppressions |= iSuppressVal[value];

								}

							}

							else

							{

								throw ExceptionReporter(UNKNOWNSUPPRESSVAL,(char*)(optionValueList.front().c_str()));

							}

							optionValueList.pop_front();

						}

					}

					else if(optionName == KLongEnableSidCheck)

					{

						iValidations |= KMarkEnable;

						iValidations |= ESid;

					}

					else if(optionName == KLongSidAllOption)

					{

						iCommmandFlag |= KSidAll;

					}

					break;

				case 'd':

					if(optionName == KLongEnableDbgFlagCheck)

					{

						iValidations |= KMarkEnable;

						iValidations |= EDbg;

						if(optionValueList.size() > 0)

						{

							if(optionValueList.front() == String("true"))

							{

								iDebuggableFlagVal = true;

							}

							else if (optionValueList.front() == String("false"))

							{

								iDebuggableFlagVal = false; 

							}

							else

							{

								throw ExceptionReporter(UNKNOWNDBGVALUE);

							}

						}

					}

					else if (optionName == KLongEnableDepCheck)

					{

						iValidations |= KMarkEnable;

						iValidations |= EDep;

					}

					break;

				case 'e':

					if (optionName == KLongE32InputOption)

					{

						iCommmandFlag |= KE32Input;

					}

					break;

				case 'v':

					if(optionName == KLongVidValOption)

					{

						StringListToUnIntList(optionValueList, iVidValList);

					}

					else if(optionName == KLongEnableVidCheck)

					{

						iValidations |= KMarkEnable;

						iValidations |= EVid;

					}

					else

					{

						iCommmandFlag |= KVerbose;

						/**Initialize ExceptionImplementation class with verbose mode flag

						to print all status information to standard output*/

						ExceptionImplementation::Instance(iCommmandFlag);

					}

					break;

				case 'n':

						iCommmandFlag |= KNoCheck;

					break;

				case 'h':

					std::cout << PrintVersion().c_str() << std::endl;

					std::cout << PrintUsage().c_str() << std::endl;

					return EQuit; //Don't proceed further

			}

		}

		else

		{

			if(!AlreadyReceived(name))

			{

				iImageNameList.push_back(name);

			}

			else

			{

				ExceptionReporter(IMAGENAMEALREADYRECEIVED, (char*)name.c_str()).Report();

			}

			iNoImage = false;

		}

	} //While loop ends here

	if((iCommmandFlag || iValidations || iSuppressions) && iNoImage)

	{

		PrintVersion();

		PrintUsage();

	}

	//Always log the version information into log file

	ExceptionImplementation::Instance(iCommmandFlag)->Log(iVersion);

	ValidateArguments();

	ValidateE32NoCheckArguments();

	if(iCommmandFlag & KE32Input)

	{

		ValidateImageNameList();

	}

	return ESuccess;

}

/**

Function identify whether the passed string is an option or not.

@internalComponent

@released

@param aName - a string received as part of command line

@param aLongOptionFlag - this flag is set if the option is long else

it is assumed as short option.

@return - returns true or false

*/

bool CmdLineHandler::IsOption(const String& aName, int& aLongOptionFlag)

{

	unsigned int prefixCount = 0;

	while(aName.at(prefixCount) == KShortOptionPrefix)

	{

		if(aName.length() == ++prefixCount)

		{

			throw ExceptionReporter(UNKNOWNOPTION, (char*)aName.c_str());

		}

	}

	switch(prefixCount)

	{

		case 0: //argument can be an image

			return false;

		case 1: // '-'

			return true;

		case 2: // '--'

			aLongOptionFlag = 1;

			return true;

		default:

			throw ExceptionReporter(UNKNOWNPREFIX, (char*)aName.c_str());

	}

}

/**

Function to do syntax validation on the received option.

1. Identifies whether the received option is valid or not.

2. Identifies whether the option can have vaue or not.

3. Throws an error if no value received for an option which should have value.

4. Throws an error if more number of values received.

5. Throws an error if an unwanted value received.

6. Throws an error if the option is not a valid one.

@internalComponent

@released

@param aOption - a string received as part of command line.

@param aOptionValue - Whether option value received or not.

@param aNoOfVal - Number of values received for this option.

@return - returns true if it is a valid option

*/

bool CmdLineHandler::Validate(const String& aOption, bool aOptionValue, unsigned int aNoOfVal)

{

	if(iOptionMap.find(aOption) != iOptionMap.end())

	{

		if(iOptionMap[aOption]) //Option can have value?

		{

			if((aNoOfVal == ENone) && (iOptionMap[aOption] != EOptional)) //No values received?

			{

				throw ExceptionReporter(VALUEEXPECTED, (char*)aOption.c_str());

			}

			if((iOptionMap[aOption] == ESingle) && (ESingle < aNoOfVal)) //Received values are more than expected

			{

				throw ExceptionReporter(UNEXPECTEDNUMBEROFVALUE,(char*)aOption.c_str());

  			}

		}

		else

		{

			if(aOptionValue) //Is option value received? Any character after the option considered as value.

			{

				throw ExceptionReporter(VALUENOTEXPECTED, (char*)aOption.c_str());

			}

		}

		return true;

	}

	throw ExceptionReporter(UNKNOWNOPTION, (char*)aOption.c_str());

}

/**

Function to split the option name and option values.

1. Ignore's the '=' symbol which is following the option. But this is an error, if that

option does not expecting any value.

2. Parses the value received with options.

@internalComponent

@released

@param aFullName - Option with its value

@param aOptionName - Option name put into this parameter

@param aOptionValues - Option values put into this parameter

@param aOptionValue - Set this flag if any value received with the option.

*/

void CmdLineHandler::ParseOption(const String& aFullName, String& aOptionName, StringList& aOptionValues, bool& aOptionValue)

{

	unsigned int optionEndLocation = aFullName.find("=");

	if(optionEndLocation != String::npos)

	{

		aOptionValue = true;

		aOptionName = aFullName.substr(0, optionEndLocation++);

		if(aFullName.length() == optionEndLocation)

		{

			throw ExceptionReporter(VALUEEXPECTED, (char*)aOptionName.c_str());

		}

		String sub = aFullName.substr(optionEndLocation);

		char* optionValues = (char*)sub.c_str();

		//Get all the values; use (,) as delimiter

		char* value = strtok(optionValues,",");

		while(value != KNull)

		{

			String str(value);

			aOptionValues.push_back(ReaderUtil::ToLower(str));

			value = strtok(KNull,",");

		}

		return;

	}

	aOptionName = aFullName;

}

/**

Function to initialize the usage.

@internalComponent

@released

*/

void CmdLineHandler::Usage(void)

{

    iUsage.assign("imgcheck [options] <img1> [<img2 .. imgN>] \n"

		"imgcheck --e32input [options] (<file> | <directory>) \n"

        "\n"

        "options: \n"

		"  -a, --all,             Report all executable's status\n"

	    "  -q, --quiet,           Command line display off\n"

	    "  -x, --xml,             Generate XML report\n"

        "  -o=xxx, --output=xxx   Override default XML file name\n"

		"  -v, --verbose,         Verbose mode output\n"

		"  -h, --help,            Display this message\n"

		"  -s=val1[,val2][...], --suppress=val1[,val2][...] \n"

		"                         Suppress one or more check,\n"

		"                         Possible values are dep, sid and vid\n"

        "  --vidlist=val1[,val2][...] \n"

		"                         One or more VID value(s) \n"

        "  --dep                  Enable dependency check\n"

        "  --vid                  Enable VID check\n"

        "  --sid                  Enable SID check, only EXEs are considered by default\n"

		"  --sidall               Include DLL also into SID check\n"

		"  --dbg[=val]            Enable Debug flag check,\n"

		"                         Optionally over ride the default value 'false'\n"

		"  --e32input             Switches the tool to filesystem mode\n"

		"  -n, --nocheck          Don't report any check(s) status\n");

}

/**

Function to return the usage.

@internalComponent

@released

*/

const String& CmdLineHandler::PrintUsage(void) const

{

	return iUsage;

}

/**

Function to prepare the version information.

@internalComponent

@released

*/

void CmdLineHandler::Version(void)

{

	iVersion.append(gToolDesc);

	iVersion.append(gMajorVersion);

	iVersion.append(gMinorVersion);

	iVersion.append(gMaintenanceVersion);

	iVersion.append(gCopyright);

}

/**

Function to return the version information.

@internalComponent

@released

*/

const String& CmdLineHandler::PrintVersion(void) const

{

	return iVersion;

}

/**

Function to return the image name one by one.

@internalComponent

@released

@return - returns image name

*/

String CmdLineHandler::NextImageName(void)

{

	String imageName = iImageNameList.front();

	iImageNameList.pop_front();

	return imageName;

}

/**

Function to return the iCommmandFlag.

@internalComponent

@released

@return - returns iCommmandFlag value.

*/

const unsigned int CmdLineHandler::ReportFlag(void) const

{

	return iCommmandFlag;

}

/**

Function to return the iXmlFileName.

@internalComponent

@released

@return - returns iXmlFileName value.

*/

const String& CmdLineHandler::XmlReportName(void) const

{

	return iXmlFileName;

}

/**

Function to append the XML extension to the received XML name.

@internalComponent

@released

*/

void CmdLineHandler::NormaliseName(void)

{

	if (iXmlFileName.find(KXmlExtension) == String::npos)

	{

		iXmlFileName.append(KXmlExtension);

	}

}