/*
 * Copyright (c) 2017 Rockchip, Inc. All Rights Reserved.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *	 http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

#include <string.h>
#include <pthread.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <math.h>
#include <sys/select.h>
#include <linux/input.h>
#include <linux/rtc.h>

#include "bt_test.h"
#include "rk_ble_app.h"
#include "rk_wifi_test.h"

static void rkwifibt_test_bluetooth();
static void rkwifibt_test_wifi_config();
#ifdef BLUEZ_USE
static void rkwifibt_test_bluetooth_1s2();
#endif

typedef struct {
	const char *cmd;
	const char *desc;
	void (*action)(void);
} menu_command_t;

static menu_command_t menu_command_table[] = {
	{"bluetooth", "show bluetooth test cmd menu", rkwifibt_test_bluetooth},
	{"wificonfig", "show wifi config test cmd menu", rkwifibt_test_wifi_config},
};

static void show_bt_cmd();
static void show_wifi_config_cmd();

typedef struct {
	const char *cmd;
	void (*action)(void *data);
} command_t;

typedef struct {
	const char *cmd;
	void (*action)(char *data);
} command_bt_t;

static command_t wifi_config_command_table[] = {
	{"", NULL},
	{"ble_wifi_config_start", rk_ble_wifi_init},
	{"ble_wifi_config_stop", rk_ble_wifi_deinit},
	{"softap_wifi_config_start", rk_wifi_softap_start},
	{"softap_wifi_config_stop", rk_wifi_softap_stop},
	{"wifi_open", rk_wifi_open},
	{"wifi_onoff_test", rk_wifi_openoff_test},
	{"wifi_close", rk_wifi_close},
	{"wifi_connect", rk_wifi_connect},
	{"wifi_ping", rk_wifi_ping},
	{"wifi_scan", rk_wifi_scan},
	{"wifi_getSavedInfo", rk_wifi_getSavedInfo},
	{"rk_wifi_getConnectionInfo", rk_wifi_getConnectionInfo},
	{"wifi_connect_with_ssid", rk_wifi_connect_with_ssid},
	{"wifi_cancel", rk_wifi_cancel},
	{"wifi_forget_with_ssid", rk_wifi_forget_with_ssid},
	{"wifi_connect1", rk_wifi_connect1},
	{"rk_wifi_disconnect", rk_wifi_disconnect},
};

static command_bt_t bt_command_table[] = {
	{"", NULL},
	{"bt_server_open", bt_test_bluetooth_init},
	{"bt_onoff_test", bt_test_bluetooth_onff_init},
	{"bt_test_set_class", bt_test_set_class},
	{"bt_test_get_device_name", bt_test_get_device_name},
	{"bt_test_get_device_addr", bt_test_get_device_addr},
	{"bt_test_set_device_name", bt_test_set_device_name},
	{"bt_test_enable_reconnect", bt_test_enable_reconnect},
	{"bt_test_disable_reconnect", bt_test_disable_reconnect},
	{"bt_test_start_discovery", bt_test_start_discovery},
	{"bt_test_start_discovery_le", bt_test_start_discovery_le},
	{"bt_test_start_discovery_bredr", bt_test_start_discovery_bredr},
	{"bt_test_start_discovery_pan", bt_test_start_discovery_pan},
	{"bt_test_cancel_discovery", bt_test_cancel_discovery},
	{"bt_test_is_discovering", bt_test_is_discovering},
	{"bt_test_display_devices", bt_test_display_devices},
	{"bt_test_read_remote_device_name", bt_test_read_remote_device_name},
	{"bt_test_get_scaned_devices", bt_test_get_scaned_devices},
	{"bt_test_display_paired_devices", bt_test_display_paired_devices},
	{"bt_test_get_paired_devices", bt_test_get_paired_devices},
	{"bt_test_free_paired_devices", bt_test_free_paired_devices},
	{"bt_test_pair_by_addr", bt_test_pair_by_addr},
	{"bt_test_unpair_by_addr", bt_test_unpair_by_addr},
	{"bt_test_get_connected_properties", bt_test_get_connected_properties},
	{"bt_test_source_auto_start", bt_test_source_auto_start},
	{"bt_test_source_connect_status", bt_test_source_connect_status},
	{"bt_test_source_auto_stop", bt_test_source_auto_stop},
	{"bt_test_source_open", bt_test_source_open},
	{"bt_test_source_close", bt_test_source_close},
	{"bt_test_source_connect_by_addr", bt_test_source_connect_by_addr},
	{"bt_test_source_disconnect", bt_test_source_disconnect},
	{"bt_test_source_disconnect_by_addr", bt_test_source_disconnect_by_addr},
	{"bt_test_source_remove_by_addr", bt_test_source_remove_by_addr},
	{"bt_test_sink_open", bt_test_sink_open},
	{"bt_test_sink_visibility00", bt_test_sink_visibility00},
	{"bt_test_sink_visibility01", bt_test_sink_visibility01},
	{"bt_test_sink_visibility10", bt_test_sink_visibility10},
	{"bt_test_sink_visibility11", bt_test_sink_visibility11},
	{"bt_test_ble_visibility00", bt_test_ble_visibility00},
	{"bt_test_ble_visibility11", bt_test_ble_visibility11},
	{"bt_test_sink_status", bt_test_sink_status},
	{"bt_test_sink_music_play", bt_test_sink_music_play},
	{"bt_test_sink_music_pause", bt_test_sink_music_pause},
	{"bt_test_sink_music_next", bt_test_sink_music_next},
	{"bt_test_sink_music_previous", bt_test_sink_music_previous},
	{"bt_test_sink_music_stop", bt_test_sink_music_stop},
	{"bt_test_sink_set_volume", bt_test_sink_set_volume},
	{"bt_test_sink_connect_by_addr", bt_test_sink_connect_by_addr},
	{"bt_test_sink_disconnect_by_addr", bt_test_sink_disconnect_by_addr},
	{"bt_test_sink_get_play_status", bt_test_sink_get_play_status},
	{"bt_test_sink_get_poschange", bt_test_sink_get_poschange},
	{"bt_test_sink_disconnect", bt_test_sink_disconnect},
	{"bt_test_sink_close", bt_test_sink_close},
	{"bt_test_ble_start", bt_test_ble_start},
	{"bt_test_ble_set_address", bt_test_ble_set_address},
	{"bt_test_ble_set_adv_interval", bt_test_ble_set_adv_interval},
	{"bt_test_ble_write", bt_test_ble_write},
	{"bt_test_ble_disconnect", bt_test_ble_disconnect},
	{"bt_test_ble_stop", bt_test_ble_stop},
	{"bt_test_ble_get_status", bt_test_ble_get_status},
	{"bt_test_ble_client_open", bt_test_ble_client_open},
	{"bt_test_ble_client_close", bt_test_ble_client_close},
	{"bt_test_ble_client_connect", bt_test_ble_client_connect},
	{"bt_test_ble_client_disconnect", bt_test_ble_client_disconnect},
	{"bt_test_ble_client_get_status", bt_test_ble_client_get_status},
	{"bt_test_ble_client_get_service_info", bt_test_ble_client_get_service_info},
	{"bt_test_ble_client_read", bt_test_ble_client_read},
	{"bt_test_ble_client_write", bt_test_ble_client_write},
	{"bt_test_ble_client_is_notify", bt_test_ble_client_is_notify},
	{"bt_test_ble_client_notify_on", bt_test_ble_client_notify_on},
	{"bt_test_ble_client_notify_off", bt_test_ble_client_notify_off},
	{"bt_test_ble_client_indicate_on", bt_test_ble_client_indicate_on},
	{"bt_test_ble_client_indicate_off", bt_test_ble_client_indicate_off},
	{"bt_test_ble_client_get_eir_data", bt_test_ble_client_get_eir_data},
	{"bt_test_spp_open", bt_test_spp_open},
	{"bt_test_spp_write", bt_test_spp_write},
	{"bt_test_spp_close", bt_test_spp_close},
	{"bt_test_spp_status", bt_test_spp_status},
	{"bt_test_spp_listen", bt_test_spp_listen},
	{"bt_test_spp_connect", bt_test_spp_connect},
	{"bt_test_spp_disconnect", bt_test_spp_disconnect},
	{"bt_test_start_discovery_spp", bt_test_start_discovery_spp},
	{"bt_test_hfp_sink_open", bt_test_hfp_sink_open},
	{"bt_test_hfp_hp_open", bt_test_hfp_hp_open},
	{"bt_test_hfp_hp_accept", bt_test_hfp_hp_accept},
	{"bt_test_hfp_hp_hungup", bt_test_hfp_hp_hungup},
	{"bt_test_hfp_hp_redail", bt_test_hfp_hp_redial},
	{"bt_test_hfp_hp_dial_number", bt_test_hfp_hp_dial_number},
	{"bt_test_hfp_hp_report_battery", bt_test_hfp_hp_report_battery},
	{"bt_test_hfp_hp_set_volume", bt_test_hfp_hp_set_volume},
	{"bt_test_hfp_hp_close", bt_test_hfp_hp_close},
	{"bt_test_hfp_hp_disconnect", bt_test_hfp_hp_disconnect},
	{"bt_test_obex_init", bt_test_obex_init},
	{"bt_test_obex_pbap_init", bt_test_obex_pbap_init},
	{"bt_test_obex_pbap_connect", bt_test_obex_pbap_connect},
	{"bt_test_obex_pbap_get_pb_vcf", bt_test_obex_pbap_get_pb_vcf},
	{"bt_test_obex_pbap_get_ich_vcf", bt_test_obex_pbap_get_ich_vcf},
	{"bt_test_obex_pbap_get_och_vcf", bt_test_obex_pbap_get_och_vcf},
	{"bt_test_obex_pbap_get_mch_vcf", bt_test_obex_pbap_get_mch_vcf},
	{"bt_test_obex_pbap_get_spd_vcf", bt_test_obex_pbap_get_spd_vcf},
	{"bt_test_obex_pbap_get_fav_vcf", bt_test_obex_pbap_get_fav_vcf},
	{"bt_test_obex_pbap_disconnect", bt_test_obex_pbap_disconnect},
	{"bt_test_obex_pbap_deinit", bt_test_obex_pbap_deinit},
	{"bt_test_obex_deinit", bt_test_obex_deinit},
	{"bt_test_pan_init", bt_test_pan_init},
	{"bt_test_pan_deinit", bt_test_pan_deinit},
	{"bt_test_pan_connect", bt_test_pan_connect},
	{"bt_test_pan_disconnect", bt_test_pan_disconnect},
	{"bt_test_get_eir_data", bt_test_get_eir_data},
	{"bt_server_close", bt_test_bluetooth_deinit},
};

static void show_wifi_config_cmd() {
	unsigned int i;
	for (i = 1; i < sizeof(wifi_config_command_table) / sizeof(wifi_config_command_table[0]); i++) {
		printf("%02d.  %s \n", i, wifi_config_command_table[i].cmd);
	}
}

static void show_bt_cmd() {
	unsigned int i;
	for (i = 1; i < sizeof(bt_command_table) / sizeof(bt_command_table[0]); i++) {
		printf("%02d.  %s \n", i, bt_command_table[i].cmd);
	}
}

static void show_help(char *bin_name) {
	unsigned int i;
	printf("%s [Usage]:\n", bin_name);
	for (i = 0; i < sizeof(menu_command_table)/sizeof(menu_command_t); i++)
		printf("\t\"%s %s\":%s.\n", bin_name, menu_command_table[i].cmd, menu_command_table[i].desc);
}

static void rkwifibt_test_wifi_config()
{
	int i, item_cnt;
	char *input_start;
	char cmdBuf[64] = {0};
	char szBuf[64] = {0};
	char szBuf_space[64] = {0};

	item_cnt = sizeof(wifi_config_command_table) / sizeof(command_t);
	show_wifi_config_cmd();

	while(true) {
		printf("Please input number or help to run: ");

		memset(szBuf, 0, sizeof(szBuf));
		if (fgets(szBuf_space, 64, stdin) == NULL)
			continue;

		if (!strncmp("help", szBuf_space, 4) || !strncmp("h", szBuf_space, 1))
			show_wifi_config_cmd();

		//printf("szBuf_space: %s:len\n", szBuf_space, sizeof(szBuf_space));
		strncpy(szBuf, szBuf_space, strlen(szBuf_space) - 1);

		input_start = strstr(szBuf, "input");
		if(input_start == NULL) {
			i = atoi(szBuf);
			if ((i >= 1) && (i < item_cnt))
				wifi_config_command_table[i].action(NULL);
		} else {
			memset(cmdBuf, 0, sizeof(cmdBuf));
			strncpy(cmdBuf, szBuf, strlen(szBuf) - strlen(input_start) - 1);
			i = atoi(cmdBuf);
			printf("%s: i = %d\n", __func__, i);
			if ((i >= 1) && (i < item_cnt))
				wifi_config_command_table[i].action(input_start + strlen("input") + 1);
		}
	}

	return;
}

static void rkwifibt_test_bluetooth()
{
	int i, item_cnt;
	char *input_start;
	char cmdBuf[64] = {0};
	char szBuf[64] = {0};
	char szBuf_space[64] = {0};

	item_cnt = sizeof(bt_command_table) / sizeof(command_bt_t);
	show_bt_cmd();

	while(true) {
		memset(szBuf, 0, sizeof(szBuf));
		printf("Please input number or help to run: ");

		if (fgets(szBuf_space, 64, stdin) == NULL)
			continue;

		if (!strncmp("help", szBuf_space, 4) || !strncmp("h", szBuf_space, 1))
			show_bt_cmd();

		///printf("szBuf_space: %s:len\n", szBuf_space, sizeof(szBuf_space));
		strncpy(szBuf, szBuf_space, strlen(szBuf_space) - 1);

		input_start = strstr(szBuf, "input");
		if(input_start == NULL) {
			i = atoi(szBuf);
			//printf("%s: select %d\n", __func__, i);
			if ((i >= 1) && (i < item_cnt))
				bt_command_table[i].action(NULL);
		} else {
			memset(cmdBuf, 0, sizeof(cmdBuf));
			strncpy(cmdBuf, szBuf, strlen(szBuf) - strlen(input_start) - 1);
			//printf("%s: cmdBuf = %s\n", __func__, cmdBuf);
			i = atoi(cmdBuf);
			printf("%s: i = %d\n", __func__, i);
			if ((i >= 1) && (i < item_cnt))
				bt_command_table[i].action(input_start + strlen("input") + 1);
		}
	}

	return;
}

int main(int argc, char *argv[])
{
	int i, item_cnt;
	char version[64] = {0};

	//RK_read_version(version, 64);
	item_cnt = sizeof(menu_command_table) / sizeof(menu_command_t);

	if (argc < 2) {
		printf("ERROR:invalid argument.\n");
		show_help(argv[0]);
		return -1;
	}

	if ((!strncmp(argv[1], "-h", 2)) || (!strncmp(argv[1], "help", 4))) {
		show_help(argv[0]);
		return 0;
	}

	for (i = 0; i < item_cnt; i++) {
		if (!strncmp(argv[1], menu_command_table[i].cmd, strlen(menu_command_table[i].cmd)))
			break;
	}

	if (i >= item_cnt) {
		printf("ERROR:invalid menu cmd.\n");
		show_help(argv[0]);
		return -1;
	}

	menu_command_table[i].action();

	while(true)
		sleep(1);

	return 0;
}