#include <ultra64.h>
#include "constants.h"
#include "game/data/data_000000.h"
#include "game/data/data_0083d0.h"
#include "game/data/data_00e460.h"
#include "game/data/data_0160b0.h"
#include "game/data/data_01a3a0.h"
#include "game/data/data_020df0.h"
#include "game/data/data_02a0e0.h"
#include "game/data/data_02da90.h"
#include "gvars/gvars.h"
#include "lib/args.h"
#include "lib/lib_2fa00.h"
#include "types.h"

s32 argc = 1;

char *argv[17] = {
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
	NULL,
};

/**
 * Split the given string into words and populate the argv array.
 */
char *argParseString(char *str)
{
	argv[0] = "";
	argc = 1;

	while (*str != '\0') {
		while (*str == ' ') {
			*str = '\0';
			str++;
		}

		argv[argc] = str;
		argc++;

		while (*str > ' ') {
			*str++;
		}
	}

	return str;
}

void argSetString(char *string)
{
	strcpy(g_ArgBuffer, string);
	argParseString((char *) g_ArgBuffer);
}

bool argsParseDebugArgs(void)
{
	u32 devaddr;
	u32 stack;
	s32 i;
	bool ret = false;

	devaddr = 0x1ffff00;

	if (rmonIsDisabled()) {
		g_ArgBuffer[0] = 0;
	} else {
		for (i = 0; i < 30; i++) {
			osPiReadIo(devaddr, &g_ArgBuffer[i]);
			devaddr += 4;
		}
	}

	argParseString((char *) g_ArgBuffer);

	// I'm guessing the -d stands for debug. If set at boot, the main thread
	// stops itself immediately after creating the rmon thread.
	if (argFindByPrefix(1, "-d")) {
		ret = true;
	}

	if (argFindByPrefix(1, "-s")) {
		g_Silent = true;
	}

	if (argFindByPrefix(1, "-j")) {
		g_LanguageId = LANGUAGE_JAPANESE;
	}

	return ret;
}

/**
 * Find a program argument based on its prefix and occurrence of that prefix,
 * and return a pointer to whatever comes after the prefix.
 *
 * Typically called with arguments such as 1, "-hard". In this example, if the
 * argument "-hard2" exists then it would return a pointer to "2".
 */
char *argFindByPrefix(s32 occurrence, char *str)
{
	s32 len = strlen(str);
	s32 i;

	for (i = 1; i < argc; i++) {
		if (strncmp(str, argv[i], len) == 0) {
			occurrence--;

			if (occurrence == 0) {
				return &argv[i][len];
			}
		}
	}

	return NULL;
}

void argGetLevel(s32 *stagenum)
{
	char *ptr = argFindByPrefix(1, "-level_");

	if (ptr) {
		*stagenum = ptr[0] * 10 + ptr[1] - 528;
	}
}