/*
 * Copyright (C) 2017 XRADIO TECHNOLOGY CO., LTD. All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *    1. Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *    2. Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the
 *       distribution.
 *    3. Neither the name of XRADIO TECHNOLOGY CO., LTD. nor the names of
 *       its contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "common/cmd/cmd_util.h"
#include "common/cmd/cmd.h"
#include "net/wlan/wlan.h"

#if PRJCONF_NET_EN

#define COMMAND_IPERF       1
#define COMMAND_PING        1
#define COMMAND_HTTPC       0
#define COMMAND_TLS         0
#define COMMAND_HTTPD       0
#define COMMAND_MQTT        0
#define COMMAND_NOPOLL      0
#define COMMAND_SNTP        0
#define COMMAND_DHCPD       0
#define COMMAND_BRROADCAST  0
#define COMMAND_ARP         0
#define COMMAND_WLAN        1

extern char sta_ssid[100];
extern char sta_psk[100];
extern int clear_bss_in_flash(void);

enum cmd_status cmd_wlan_fc_exec(char *cmd)
{
	int ret;

	cmd_write_respond(CMD_STATUS_OK, "OK");

	if (cmd_strncmp(cmd, "config ", 7) == 0) {
		char *argv[3];
		uint32_t flag;
		int argc;

		argc = cmd_parse_argv(cmd + 7, argv, cmd_nitems(argv));
		if (argc == 0) {
			ret = -2;
			goto out;
		}
		cmd_memset(sta_ssid, 0, 100);
		cmd_memset(sta_psk, 0, 100);
		cmd_memcpy(sta_ssid, (uint8_t *)argv[0], cmd_strlen(argv[0]));
		cmd_memcpy(sta_psk, (uint8_t *)argv[1], cmd_strlen(argv[1]));

		if (argc == 3 && (cmd_strcmp(argv[2], "WPA3") == 0)) {
			flag = WLAN_STA_CONF_FLAG_WPA3;
		} else {
			/* for the fastest speed, WPA2 is used by default */
			flag = 0;
		}
		ret = wlan_sta_config((uint8_t *)argv[0], cmd_strlen(argv[0]),
		                      (uint8_t *)argv[1], flag);
	} else if (cmd_strcmp(cmd, "enable") == 0) {
		ret = wlan_sta_enable();
	} else if (cmd_strcmp(cmd, "disable") == 0) {
		ret = wlan_sta_disable();
	} else if (cmd_strcmp(cmd, "clear_bss") == 0) {
		ret = clear_bss_in_flash();
	} else {
		CMD_ERR("unknown cmd '%s'\n", cmd);
		return CMD_STATUS_ACKED;
	}

out:
	if (ret == -2) {
		CMD_ERR("cmd '%s' invalid arg\n", cmd);
		return CMD_STATUS_ACKED;
	} else if (ret == -1) {
		CMD_ERR("cmd '%s' exec failed\n", cmd);
		return CMD_STATUS_ACKED;
	}

	return CMD_STATUS_ACKED;
}

/*
 * net commands
 */
static const struct cmd_data g_net_cmds[] = {
	{ "mode",        cmd_wlan_mode_exec },
#ifdef CONFIG_WLAN_AP
	{ "ap",          cmd_wlan_ap_exec },
#endif
#ifdef CONFIG_WLAN_STA
	{ "sta",         cmd_wlan_sta_exec },
	{ "fc",          cmd_wlan_fc_exec },
#endif
	{ "ifconfig",    cmd_ifconfig_exec },
	{ "smartconfig", cmd_smart_config_exec },
	{ "airkiss",     cmd_airkiss_exec },
	{ "smartlink",   cmd_smartlink_exec },

#if COMMAND_IPERF
	{ "iperf",       cmd_iperf_exec },
#endif

#if COMMAND_PING
	{ "ping",        cmd_ping_exec },
#endif

#if COMMAND_HTTPC
	{ "httpc",       cmd_httpc_exec },
#endif

#if COMMAND_TLS
	{ "tls",         cmd_tls_exec },
#endif

#if COMMAND_HTTPD
	{ "httpd",       cmd_httpd_exec },
#endif

#if COMMAND_SNTP
	{ "sntp",        cmd_sntp_exec },
#endif

#if COMMAND_NOPOLL
	{ "nopoll",      cmd_nopoll_exec },
#endif

#if COMMAND_MQTT
	{ "mqtt",        cmd_mqtt_exec },
#endif

#if COMMAND_DHCPD
	{ "dhcpd",       cmd_dhcpd_exec },
#endif

#if COMMAND_BRROADCAST
	{ "broadcast",   cmd_broadcast_exec },
#endif

#if COMMAND_ARP
	{ "arp",         cmd_arp_exec },
#endif

#if COMMAND_WLAN
	{ "wlan",        cmd_wlan_exec },
#endif
};

static enum cmd_status cmd_net_exec(char *cmd)
{
	return cmd_exec(cmd, g_net_cmds, cmd_nitems(g_net_cmds));
}

#endif /* PRJCONF_NET_EN */

/*
 * driver commands
 */
static const struct cmd_data g_drv_cmds[] = {
};

static enum cmd_status cmd_drv_exec(char *cmd)
{
	return cmd_exec(cmd, g_drv_cmds, cmd_nitems(g_drv_cmds));
}

/*
 * main commands
 */
static const struct cmd_data g_main_cmds[] = {
#if PRJCONF_NET_EN
	{ "net",     cmd_net_exec },
	{ "rf",      cmd_rf_exec },
#endif
	{ "drv",     cmd_drv_exec },
	{ "echo",    cmd_echo_exec },
	{ "mem",     cmd_mem_exec },
	{ "heap",    cmd_heap_exec },
	{ "thread",  cmd_thread_exec },
	{ "upgrade", cmd_upgrade_exec },
	{ "reboot",  cmd_reboot_exec },
#ifdef CONFIG_OTA
	{ "ota",     cmd_ota_exec },
	{ "etf",     cmd_etf_exec },
#endif
	{ "pm",      cmd_pm_exec },
	{ "efpg",    cmd_efpg_exec },
	{ "flash",   cmd_flash_exec },
	{ "gpio",    cmd_gpio_exec },
#if PRJCONF_NET_EN
	{ "lmac",    cmd_lmac_exec },
#endif
	{ "sysinfo", cmd_sysinfo_exec },
};

void main_cmd_exec(char *cmd)
{
	cmd_main_exec(cmd, g_main_cmds, cmd_nitems(g_main_cmds));
}