/*
 * This file is part of the DXX-Rebirth project <http://www.dxx-rebirth.com/>.
 * It is copyright by its individual contributors, as recorded in the
 * project's Git history.  See COPYING.txt at the top level for license
 * terms and a link to the Git history.
 */

/*
 *
 * Functions for accessing arguments.
 *
 */

#include <string>
#include <vector>
#include <stdlib.h>
#include <string.h>
#include <SDL_stdinc.h>
#include "physfsx.h"
#include "args.h"
#include "u_mem.h"
#include "strutil.h"
#include "digi.h"
#include "game.h"
#include "gauges.h"
#include "console.h"
#ifdef USE_UDP
#include "net_udp.h"
#endif

#include "compiler-range_for.h"
#include "partial_range.h"

#define MAX_ARGS 1000
#if defined(DXX_BUILD_DESCENT_I)
#define INI_FILENAME "d1x.ini"
#elif defined(DXX_BUILD_DESCENT_II)
#define INI_FILENAME "d2x.ini"
#endif

typedef std::vector<std::string> Arglist;

class argument_exception
{
public:
	const std::string arg;
	argument_exception(std::string &&a) :
		arg(std::move(a))
	{
	}
};

class missing_parameter : public argument_exception
{
public:
	missing_parameter(std::string &&a) :
		argument_exception(std::move(a))
	{
	}
};

class unhandled_argument : public argument_exception
{
public:
	unhandled_argument(std::string &&a) :
		argument_exception(std::move(a))
	{
	}
};

class conversion_failure : public argument_exception
{
public:
	const std::string value;
	conversion_failure(std::string &&a, std::string &&v) :
		argument_exception(std::move(a)), value(std::move(v))
	{
	}
};

struct Arg GameArg;

static void AppendIniArgs(Arglist &Args)
{
	if (auto f = PHYSFSX_openReadBuffered(INI_FILENAME))
	{
		PHYSFSX_gets_line_t<1024> line;
		while(!PHYSFS_eof(f) && Args.size() < MAX_ARGS && PHYSFSX_fgets(line, f))
		{
			static const char separator[] = " ";
			for(char *token = strtok(line, separator); token != NULL; token = strtok(NULL, separator))
			{
				if (*token == ';')
					break;
				Args.push_back(token);
			}
		}
	}
}

static std::string &&arg_string(Arglist::iterator &pp, Arglist::const_iterator end)
{
	auto arg = pp;
	if (++pp == end)
		throw missing_parameter(std::move(*arg));
	return std::move(*pp);
}

static long arg_integer(Arglist::iterator &pp, Arglist::const_iterator end)
{
	auto arg = pp;
	auto &&value = arg_string(pp, end);
	char *p;
	auto i = strtol(value.c_str(), &p, 10);
	if (*p)
		throw conversion_failure(std::move(*arg), std::move(value));
	return i;
}

static void arg_port_number(Arglist::iterator &pp, Arglist::const_iterator end, uint16_t &out, bool allow_privileged)
{
	auto port = arg_integer(pp, end);
	if (static_cast<uint16_t>(port) == port && (allow_privileged || port >= 1024))
		out = port;
}

static void ReadCmdArgs(Arglist &Args)
{
	GameArg.SysMaxFPS = MAXIMUM_FPS;
#if defined(DXX_BUILD_DESCENT_II)
	GameArg.SndDigiSampleRate = SAMPLE_RATE_22K;
#endif
#ifdef USE_UDP
	GameArg.MplUdpHostAddr = UDP_MANUAL_ADDR_DEFAULT;
#ifdef USE_TRACKER
	GameArg.MplTrackerAddr = TRACKER_ADDR_DEFAULT;
	GameArg.MplTrackerPort = TRACKER_PORT_DEFAULT;
#endif
#endif
	GameArg.DbgVerbose = CON_NORMAL;
	GameArg.DbgBpp 			= 32;
#ifdef OGL
	GameArg.DbgGlIntensity4Ok 	= 1;
	GameArg.DbgGlLuminance4Alpha4Ok = 1;
	GameArg.DbgGlRGBA2Ok 		= 1;
	GameArg.DbgGlReadPixelsOk 	= 1;
	GameArg.DbgGlGetTexLevelParamOk = 1;
#endif
	for (Arglist::iterator pp = Args.begin(), end = Args.end(); pp != end; ++pp)
	{
		const char *p = pp->c_str();
	// System Options

		if (!d_stricmp(p, "-help") || !d_stricmp(p, "-h") || !d_stricmp(p, "-?") || !d_stricmp(p, "?"))
			GameArg.SysShowCmdHelp = 1;
		else if (!d_stricmp(p, "-nonicefps"))
			GameArg.SysNoNiceFPS = 1;
		else if (!d_stricmp(p, "-maxfps"))
			GameArg.SysMaxFPS = arg_integer(pp, end);
		else if (!d_stricmp(p, "-hogdir"))
			GameArg.SysHogDir = arg_string(pp, end);
		else if (!d_stricmp(p, "-nohogdir"))
			GameArg.SysNoHogDir = 1;
		else if (!d_stricmp(p, "-use_players_dir"))
			GameArg.SysUsePlayersDir 	= 1;
		else if (!d_stricmp(p, "-lowmem"))
			GameArg.SysLowMem 		= 1;
		else if (!d_stricmp(p, "-pilot"))
			GameArg.SysPilot = arg_string(pp, end);
		else if (!d_stricmp(p, "-record-demo-format"))
			GameArg.SysRecordDemoNameTemplate = arg_string(pp, end);
		else if (!d_stricmp(p, "-auto-record-demo"))
			GameArg.SysAutoRecordDemo = 1;
		else if (!d_stricmp(p, "-window"))
			GameArg.SysWindow 		= 1;
		else if (!d_stricmp(p, "-noborders"))
			GameArg.SysNoBorders 		= 1;
#if defined(DXX_BUILD_DESCENT_I)
		else if (!d_stricmp(p, "-notitles"))
			GameArg.SysNoTitles 		= 1;
#elif defined(DXX_BUILD_DESCENT_II)
		else if (!d_stricmp(p, "-nomovies"))
			GameArg.SysNoMovies 		= 1;
#endif
		else if (!d_stricmp(p, "-autodemo"))
			GameArg.SysAutoDemo 		= 1;

	// Control Options

		else if (!d_stricmp(p, "-nocursor"))
			GameArg.CtlNoCursor 		= 1;
		else if (!d_stricmp(p, "-nomouse"))
			GameArg.CtlNoMouse 		= 1;
		else if (!d_stricmp(p, "-nojoystick"))
			GameArg.CtlNoJoystick 		= 1;
		else if (!d_stricmp(p, "-nostickykeys"))
			GameArg.CtlNoStickyKeys		= 1;

	// Sound Options

		else if (!d_stricmp(p, "-nosound"))
			GameArg.SndNoSound 		= 1;
		else if (!d_stricmp(p, "-nomusic"))
			GameArg.SndNoMusic 		= 1;
#if defined(DXX_BUILD_DESCENT_II)
		else if (!d_stricmp(p, "-sound11k"))
			GameArg.SndDigiSampleRate 		= SAMPLE_RATE_11K;
#endif
		else if (!d_stricmp(p, "-nosdlmixer"))
		{
#ifdef USE_SDLMIXER
			GameArg.SndDisableSdlMixer = true;
#endif
		}

	// Graphics Options

		else if (!d_stricmp(p, "-lowresfont"))
			GameArg.GfxSkipHiresFNT	= 1;
#if defined(DXX_BUILD_DESCENT_II)
		else if (!d_stricmp(p, "-lowresgraphics"))
			GameArg.GfxSkipHiresGFX	= 1;
		else if (!d_stricmp(p, "-lowresmovies"))
			GameArg.GfxSkipHiresMovie 		= 1;
#endif
#ifdef OGL
	// OpenGL Options

		else if (!d_stricmp(p, "-gl_fixedfont"))
			GameArg.OglFixedFont 		= 1;
#endif

	// Multiplayer Options

#ifdef USE_UDP
		else if (!d_stricmp(p, "-udp_hostaddr"))
			GameArg.MplUdpHostAddr = arg_string(pp, end);
		else if (!d_stricmp(p, "-udp_hostport"))
			/* Peers use -udp_myport to change, so peer cannot set a
			 * privileged port.
			 */
			arg_port_number(pp, end, GameArg.MplUdpHostPort, false);
		else if (!d_stricmp(p, "-udp_myport"))
		{
			arg_port_number(pp, end, GameArg.MplUdpMyPort, false);
		}
		else if (!d_stricmp(p, "-no-tracker"))
		{
			/* Always recognized.  No-op if tracker support compiled
			 * out. */
#ifdef USE_TRACKER
			GameArg.MplTrackerAddr = nullptr;
#endif
		}
#ifdef USE_TRACKER
		else if (!d_stricmp(p, "-tracker_hostaddr"))
		{
			GameArg.MplTrackerAddr = arg_string(pp, end);
		}
		else if (!d_stricmp(p, "-tracker_hostport"))
			arg_port_number(pp, end, GameArg.MplTrackerPort, true);
#endif
#endif

#if defined(DXX_BUILD_DESCENT_I)
		else if (!d_stricmp(p, "-nobm"))
			GameArg.EdiNoBm 		= 1;
#elif defined(DXX_BUILD_DESCENT_II)
#ifdef EDITOR
	// Editor Options

		else if (!d_stricmp(p, "-autoload"))
			GameArg.EdiAutoLoad = arg_string(pp, end);
		else if (!d_stricmp(p, "-macdata"))
			GameArg.EdiMacData 		= 1;
		else if (!d_stricmp(p, "-hoarddata"))
			GameArg.EdiSaveHoardData 	= 1;
#endif
#endif

	// Debug Options

		else if (!d_stricmp(p, "-debug"))
			GameArg.DbgVerbose 	= CON_DEBUG;
		else if (!d_stricmp(p, "-verbose"))
			GameArg.DbgVerbose 	= CON_VERBOSE;

		else if (!d_stricmp(p, "-no-grab"))
			GameArg.DbgForbidConsoleGrab = 1;
		else if (!d_stricmp(p, "-safelog"))
			GameArg.DbgSafelog 		= 1;
		else if (!d_stricmp(p, "-norun"))
			GameArg.DbgNoRun 		= 1;
		else if (!d_stricmp(p, "-renderstats"))
			GameArg.DbgRenderStats 		= 1;
		else if (!d_stricmp(p, "-text"))
			GameArg.DbgAltTex = arg_string(pp, end);
		else if (!d_stricmp(p, "-tmap"))
			GameArg.DbgTexMap = arg_string(pp, end);
		else if (!d_stricmp(p, "-showmeminfo"))
			GameArg.DbgShowMemInfo 		= 1;
		else if (!d_stricmp(p, "-nodoublebuffer"))
			GameArg.DbgNoDoubleBuffer 	= 1;
		else if (!d_stricmp(p, "-bigpig"))
			GameArg.DbgNoCompressPigBitmap 		= 1;
		else if (!d_stricmp(p, "-16bpp"))
			GameArg.DbgBpp 		= 16;

#ifdef OGL
		else if (!d_stricmp(p, "-gl_oldtexmerge"))
			GameArg.DbgUseOldTextureMerge 		= 1;
		else if (!d_stricmp(p, "-gl_intensity4_ok"))
			GameArg.DbgGlIntensity4Ok 		                       = arg_integer(pp, end);
		else if (!d_stricmp(p, "-gl_luminance4_alpha4_ok"))
			GameArg.DbgGlLuminance4Alpha4Ok                        = arg_integer(pp, end);
		else if (!d_stricmp(p, "-gl_rgba2_ok"))
			GameArg.DbgGlRGBA2Ok 			                       = arg_integer(pp, end);
		else if (!d_stricmp(p, "-gl_readpixels_ok"))
			GameArg.DbgGlReadPixelsOk 		                       = arg_integer(pp, end);
		else if (!d_stricmp(p, "-gl_gettexlevelparam_ok"))
			GameArg.DbgGlGetTexLevelParamOk                        = arg_integer(pp, end);
#else
		else if (!d_stricmp(p, "-hwsurface"))
			GameArg.DbgSdlHWSurface = 1;
		else if (!d_stricmp(p, "-asyncblit"))
			GameArg.DbgSdlASyncBlit = 1;
#endif
		else
			throw unhandled_argument(std::move(*pp));
	}

	if (GameArg.SysMaxFPS < MINIMUM_FPS)
		GameArg.SysMaxFPS = MINIMUM_FPS;
	else if (GameArg.SysMaxFPS > MAXIMUM_FPS)
		GameArg.SysMaxFPS = MAXIMUM_FPS;

	static char sdl_disable_lock_keys[] = "SDL_DISABLE_LOCK_KEYS=0";
	if (GameArg.CtlNoStickyKeys) // Must happen before SDL_Init!
		sdl_disable_lock_keys[sizeof(sdl_disable_lock_keys) - 1] = '1';
	SDL_putenv(sdl_disable_lock_keys);
}

bool InitArgs( int argc,char **argv )
{
	Arglist Args;
	Args.reserve(argc);
	range_for (auto &i, unchecked_partial_range(argv, 1u, static_cast<unsigned>(argc)))
		Args.push_back(i);

	AppendIniArgs(Args);
	try {
		ReadCmdArgs(Args);
		return true;
	} catch(const missing_parameter& e) {
		Warning("Missing parameter for argument \"%s\"", e.arg.c_str());
	} catch(const unhandled_argument& e) {
		Warning("Unhandled argument \"%s\"", e.arg.c_str());
	} catch(const conversion_failure& e) {
		Warning("Failed to convert argument \"%s\" parameter \"%s\"", e.arg.c_str(), e.value.c_str());
	}
	return false;
}