/* Copyright (C) 2015 Wildfire Games.
 * This file is part of 0 A.D.
 *
 * 0 A.D. is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * 0 A.D. is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with 0 A.D.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "precompiled.h"
#include "CmdLineArgs.h"

#include "lib/sysdep/sysdep.h"

CmdLineArgs::CmdLineArgs(int argc, const char* argv[])
{
	if (argc >= 1)
	{
		std::string arg0(argv[0]);
		// avoid OsPath complaining about mixing both types of separators,
		// which happens when running in the VC2010 debugger
		std::replace(arg0.begin(), arg0.end(), '/', SYS_DIR_SEP);
		m_Arg0 = arg0;
	}

	for (int i = 1; i < argc; ++i)
	{
		// Only accept arguments that start with '-'
		if (argv[i][0] != '-')
			continue;

		CStr name, value;

		// Check for "-arg=value"
		const char* eq = strchr(argv[i], '=');
		if (eq)
		{
			name = CStr(argv[i]+1, eq-argv[i]-1);
			value = CStr(eq+1);
		}
		else
		{
			name = CStr(argv[i]+1);
		}

		m_Args.emplace_back(std::move(name), std::move(value));
	}
}

bool CmdLineArgs::Has(const char* name) const
{
	return m_Args.end() != find_if(m_Args.begin(), m_Args.end(),
		[&name](const std::pair<CStr, CStr>& a) { return a.first == name; });
}

CStr CmdLineArgs::Get(const char* name) const
{
	ArgsT::const_iterator it = find_if(m_Args.begin(), m_Args.end(),
		[&name](const std::pair<CStr, CStr>& a) { return a.first == name; });
	if (it != m_Args.end())
		return it->second;
	else
		return "";
}

std::vector<CStr> CmdLineArgs::GetMultiple(const char* name) const
{
	std::vector<CStr> values;
	ArgsT::const_iterator it = m_Args.begin();
	while (1)
	{
		it = find_if(it, m_Args.end(),
			[&name](const std::pair<CStr, CStr>& a) { return a.first == name; });
		if (it == m_Args.end())
			break;
		values.push_back(it->second);
		++it; // start searching from the next one in the next iteration
	}
	return values;
}

OsPath CmdLineArgs::GetArg0() const
{
	return m_Arg0;
}
