#include "global.h"
#include "ub_env.h"


int Verbose;
extern char compiled[], compiledby[], compilehost[];


static void
Usage()
{
	printf(	" -= U-Boot-Env =- Tool for u-boot-env nand map management\n"
		"=== %s === %s@%s ===\n\n"
		"  Syntax: ube [options] [set_value]\n"
		"\n"
		"\t-g <name>\tSet parameter to value\n"
		"\t-s <name>\tGet parameter value\n"
		"\t-q\t\tQuiet mode\n"
		"\t-v\t\tVerbose ...\n"
		"\n", compiled, compiledby, compilehost);
}

int
main(int argc, char **argv)
{
	char ch, mode = 0, szName[STRSIZ] = { 0 }, szVal[STRSIZ] = { 0 };
	const char *str;
	int ret;

	while ((ch = getopt(argc, argv, "hvqg:s:")) != -1)
		switch (ch) {
			case 'g':
				mode |= 1;
				strlcpy(szName, optarg, sizeof szName);
				break;
			case 's':
				mode |= 2;
				strlcpy(szName, optarg, sizeof szName);
				break;
			case 'v':
				Verbose++;
				break;
			case 'q':
				mode |= 0x80;
				break;
			case 'h':
			default:
				Usage();
				return 1;
		}
	argc -= optind;
	argv += optind;
	if ((mode & 0x7f) == 2 && argc)
		strlcpy(szVal, *argv, sizeof szVal);

	VERB(1) printf("u-boot-env: mode=0x%hhx name=%s value=%s\n", mode, szName, szVal);

	if (!(mode & 0x7f)) {
		VERB(2) printf("u-boot-env: list variables\n");
	}

	if ((mode & 0x7f) & 1) {
		VERB(2) printf("u-boot-env: get variable %s\n", szName);
		str = ub_getenv(szName);
		if (!str) {
			printf("Error:: Variable %s not found!\n", szName);
			return 2;
		} else if (mode & 0x80)
			printf("%s\n", str);
		else
			printf("Variable %s=%s\n", szName, str);
	}

	if ((mode & 0x7f) & 2) {
		VERB(2) printf("u-boot-env: set variable %s\n", szName);
		ret = ub_setenv(szName, argc ? szVal : NULL);
		if (ret) {
			printf("Error:: Writing variable %s!\n", szName);
			return 3;
		}
		if (!(mode & 0x80))
			printf("Done\n");
	}

	return 0;
}