/*
 * (C) Copyright 2022 Rockchip Electronics Co., Ltd.
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <malloc.h>

/*
 * Script file example:
 **************************
 * > # script file start
 * > echo "hello world"
 * > % script file end
 **************************
 */
#define SCRIPT_FILE_MAX_SIZE	(12 * 1024)
#define SCRIPT_FILE_COMMENT	'#'
#define SCRIPT_FILE_END		'%'
#define IS_COMMENT(x)		(SCRIPT_FILE_COMMENT == (x))
#define IS_FILE_END(x)		(SCRIPT_FILE_END == (x))
#define IS_LINE_END(x)		('\r' == (x) || '\n' == (x))
#define MAX_LINE_SIZE		8000

static char *script_next_line(char **line_buf_ptr)
{
	char *line_buf = (*line_buf_ptr);
	char *next_line;
	int i = 0;

	/* strip '\r', '\n' and comment */
	while (1) {
		/* strip '\r' & '\n' */
		if (IS_LINE_END(line_buf[0])) {
			line_buf++;
		/* strip comment */
		} else if (IS_COMMENT(line_buf[0])) {
			for (i = 0; !IS_LINE_END(line_buf[0]) && i <= MAX_LINE_SIZE; i++)
				line_buf++;

			if (i > MAX_LINE_SIZE) {
				line_buf[0] = SCRIPT_FILE_END;
				printf("Error: max line length is %d!!!\n", MAX_LINE_SIZE);
				break;
			}
		} else {
			break;
		}
	}

	/* get next line */
	if (IS_FILE_END(line_buf[0])) {
		next_line = NULL;
	} else {
		next_line = line_buf;
		for (i = 0; !IS_LINE_END(line_buf[0]) && i <= MAX_LINE_SIZE; i++)
			line_buf++;

		if (i > MAX_LINE_SIZE) {
			next_line = NULL;
			printf("Error: max line length is %d!!!\n", MAX_LINE_SIZE);
		} else {
			line_buf[0] = '\0';
			*line_buf_ptr = line_buf + 1;
		}
	}

	return next_line;
}

static int do_script(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	char *next_line, *script, *buf;
	ulong addr;
	int ret = CMD_RET_SUCCESS;

	if (argc != 2 || !argv[1])
		return CMD_RET_USAGE;

	addr = simple_strtoul(argv[1], NULL, 16);
	if (!addr)
		return CMD_RET_USAGE;

	buf = calloc(SCRIPT_FILE_MAX_SIZE, 1);
	if (!buf)
		return CMD_RET_FAILURE;

	script = buf;
	memcpy(buf, (char *)addr, SCRIPT_FILE_MAX_SIZE);
	while ((next_line = script_next_line(&script)) != NULL) {
		printf("\n$ %s\n", next_line);
		ret = run_command(next_line, 0);
		if (ret)
			break;	/* fail */
	}
	free(buf);

	return ret;
}

static int do_sd_update(cmd_tbl_t *cmdtp, int flag,
			int argc, char * const argv[])
{
	char cmd[128];
	char *buf;
	int ret;

	buf = memalign(ARCH_DMA_MINALIGN, SCRIPT_FILE_MAX_SIZE * 2);
	if (!buf)
		return CMD_RET_FAILURE;

	snprintf(cmd, 128, "fatload mmc 1 0x%08lx sd_update.txt", (ulong)buf);
	ret = run_command(cmd, 0);
	if (!ret) {
		snprintf(cmd, 128, "script 0x%08lx", (ulong)buf);
		ret = run_command(cmd, 0);
	}
	free(buf);

	return ret;
}

static int do_usb_update(cmd_tbl_t *cmdtp, int flag,
			 int argc, char * const argv[])
{
	char cmd[128];
	char *buf;
	int ret;

	buf = memalign(ARCH_DMA_MINALIGN, SCRIPT_FILE_MAX_SIZE * 2);
	if (!buf)
		return CMD_RET_FAILURE;

	snprintf(cmd, 128, "usb reset");
	ret = run_command(cmd, 0);
	if (!ret) {
		snprintf(cmd, 128, "fatload usb 0 0x%08lx usb_update.txt", (ulong)buf);
		ret = run_command(cmd, 0);
	}
	if (!ret) {
		snprintf(cmd, 128, "script 0x%08lx", (ulong)buf);
		ret = run_command(cmd, 0);
	}
	free(buf);

	return ret;
}

static int do_tftp_update(cmd_tbl_t *cmdtp, int flag,
			  int argc, char * const argv[])
{
	char cmd[128];
	char *buf;
	int dhcp = 0;
	int ret;

	if ((argc > 1) && !strcmp(argv[1], "-d"))
		dhcp = 1;

	buf = memalign(ARCH_DMA_MINALIGN, SCRIPT_FILE_MAX_SIZE * 2);
	if (!buf)
		return CMD_RET_FAILURE;

	if (dhcp)
		run_command("dhcp", 0);

	snprintf(cmd, 128, "tftp 0x%08lx tftp_update.txt", (ulong)buf);
	ret = run_command(cmd, 0);
	if (!ret) {
		snprintf(cmd, 128, "script 0x%08lx", (ulong)buf);
		ret = run_command(cmd, 0);
	}
	free(buf);

	return ret;
}

U_BOOT_CMD(
	script, 2, 1, do_script,
	"Run a script", "[file addr]"
);

U_BOOT_CMD(
	sd_update, 1, 1, do_sd_update,
	"sdcard auto upgrade", ""
);

U_BOOT_CMD(
	usb_update, 1, 1, do_usb_update,
	"usb auto upgrade", ""
);

U_BOOT_CMD(
	tftp_update, 2, 1, do_tftp_update,
	"tftp auto upgrade", "[-d]"
);