// SPDX-License-Identifier: (GPL-2.0+ OR MIT)
/*
 * Copyright (c) 2019 Amlogic, Inc. All rights reserved.
 */

#include <asm/amlogic/arch/pwr_ctrl.h>
#include <command.h>
#include <vsprintf.h>

#ifndef PM_MAX
#define PM_MAX	0
static char *domain_name[PM_MAX];
#else
extern char* domain_name[];
#endif
extern unsigned long pwr_ctrl_status_psci_smc(unsigned int power_domain);

static int do_powershow(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	int i;
	int ret=0;

	if (argc > 1) {
		return CMD_RET_USAGE;
	}

	if (PM_MAX  == 0) {
		printf("Don't support this feature now!\n");
		return ret;
	}

	for (i = 0; i < PM_MAX; i++) {
		/*
		* The features of each board are different, ignore cov weak cryptor report.
		*/
		/* coverity[overrun-local] */
		if (strcmp("", domain_name[i]) != 0)
			printf("%s[%d]:		%lx\n", domain_name[i], i,
				pwr_ctrl_status_psci_smc(i));
	}

	return ret;
}

U_BOOT_CMD(
	powershow, 1, 1, do_powershow,
	"show the power domain status , 0: on; 1: off",
	"\n arg[0]: cmd\n"
);

static int do_powerset(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	unsigned int domain_id;
	bool domain_status;
	int ret=0;

	if (argc <= 1) {
		printf("plese input power set args: domain ID, domain status!\n");
		return CMD_RET_USAGE;
	}

	domain_id = simple_strtoul(argv[1], NULL, 10);
	domain_status = simple_strtoul(argv[2], NULL, 10);

	pwr_ctrl_psci_smc(domain_id, !domain_status);

	return ret;
}

U_BOOT_CMD(
	powerset, 3, 1, do_powerset,
	"power on/off a certain power domain",
	"\n arg[0]: cmd\n"
	"arg[1]: power domain ID \n"
	"arg[2]: power status to set, 0: on, 1: off\n"
);