CmdLine.h

///----------------------------------------------------------------------------------------------------
/// Copyright (c) Raidcore.GG - All rights reserved.
///
/// Name         :  CmdLine.h
/// Description  :  Contains functions for the commandline.
/// Authors      :  K. Bieniek
///----------------------------------------------------------------------------------------------------

#pragma once

#include <shellapi.h>
#include <string>
#include <vector>
#include <windows.h>

#include "Strings.h"

///----------------------------------------------------------------------------------------------------
/// CmdLine Namespace
///----------------------------------------------------------------------------------------------------
namespace CmdLine
{
	///----------------------------------------------------------------------------------------------------
	/// Parse:
	/// 	Returns all the arguments.
	///----------------------------------------------------------------------------------------------------
	inline std::vector<std::string> Parse()
	{
		static bool s_Parsed = false;
		static std::vector<std::string> s_Arguments{};

		if (s_Parsed)
		{
			return s_Arguments;
		}

		std::vector<std::string> args; // extra vector to return value arguments merged

		int argc;
		LPWSTR* argv = CommandLineToArgvW(GetCommandLineW(), &argc);

		for (int i = 0; i < argc; i++)
		{
			std::wstring paramW = argv[i];
			std::string token = String::ToString(paramW);

			s_Arguments.push_back(token);

			if (String::StartsWith(token, "-"))
			{
				args.push_back(token);
			}
			else
			{
				if (args.size() >= 1)
				{
					args[args.size() - 1].append(" " + token);
				}
				else
				{
					args.push_back(token);
				}
			}
		}

		s_Parsed = true;

		return args;
	}

	///----------------------------------------------------------------------------------------------------
	/// HasArgument:
	/// 	Returns true if the commandline contains the given argument.
	///----------------------------------------------------------------------------------------------------
	inline bool HasArgument(const std::string& aArgument)
	{
		std::vector<std::string> args = Parse();

		std::string arg = String::ToLower(aArgument);

		for (const std::string& str : args)
		{
			std::string cmp = String::ToLower(str);

			if (arg == cmp) { return true; }
		}

		return false;
	}

	///----------------------------------------------------------------------------------------------------
	/// GetArgumentValue:
	/// 	Returns the value following the given argument.
	///----------------------------------------------------------------------------------------------------
	inline std::string GetArgumentValue(const std::string& aArgument)
	{
		std::vector<std::string> args = Parse();

		std::string arg = String::ToLower(aArgument);

		for (size_t i = 0; i < args.size(); i++)
		{
			std::string cmp = String::ToLower(args[i]);

			if (arg == cmp)
			{
				if (i < args.size() - 1)
				{
					const std::string& argVal = args[i + 1];

					return argVal;
				}

				break;
			}
		}

		return "";
	}
}