#include "tcardemulator.h"
#include "sap_app.h"
#include <dlog.h>
#include <unistd.h>
#include <stdlib.h>
#include <nfc.h>
#include <sap.h>
#include <glib.h>

#define HELLO_ACCESSORY_PROFILE_ID "/com/vsmcartcard"
#define HELLO_ACCESSORY_CHANNELID 104

typedef struct priv {
	sap_agent_h agent;
	sap_socket_h socket;
	sap_peer_agent_h peer_agent;
	nfc_se_h nfc_handle;
} priv_s;

typedef struct message_queue {
	void *message;
	unsigned int message_len;
	struct message_queue *next;
} message_queue_s;

static gboolean agent_created = FALSE;
static gboolean agent_connected = FALSE;

static priv_s priv_data = { 0 };
static message_queue_s *m_queue = NULL;

void on_peer_agent_updated(sap_peer_agent_h peer_agent,
		sap_peer_agent_status_e peer_status,
		sap_peer_agent_found_result_e result,
		void *user_data)
{
	switch (result) {
		case SAP_PEER_AGENT_FOUND_RESULT_DEVICE_NOT_CONNECTED:
			dlog_print(DLOG_INFO, LOG_TAG, "device is not connected");
			break;

		case SAP_PEER_AGENT_FOUND_RESULT_FOUND:
			if (peer_status == SAP_PEER_AGENT_STATUS_AVAILABLE) {
				priv_data.peer_agent = peer_agent;
				dlog_print(DLOG_INFO, LOG_TAG, "Find Peer Success!!");
				request_service_connection();
			} else {
				dlog_print(DLOG_INFO, LOG_TAG, "peer agent removed");
				sap_peer_agent_destroy(peer_agent);
			}
			break;

		case SAP_PEER_AGENT_FOUND_RESULT_SERVICE_NOT_FOUND:
			dlog_print(DLOG_INFO, LOG_TAG, "service not found");
			break;

		case SAP_PEER_AGENT_FOUND_RESULT_TIMEDOUT:
			dlog_print(DLOG_INFO, LOG_TAG, "peer agent find timed out");
			break;

		case SAP_PEER_AGENT_FOUND_RESULT_INTERNAL_ERROR:
			dlog_print(DLOG_INFO, LOG_TAG, "peer agent find search failed");
			break;
	}
}


static void on_service_connection_terminated(sap_peer_agent_h peer_agent,
		sap_socket_h socket,
		sap_service_connection_terminated_reason_e result,
		void *user_data)
{
	switch (result) {
		case SAP_CONNECTION_TERMINATED_REASON_PEER_DISCONNECTED:
			dlog_print(DLOG_INFO, LOG_TAG, "disconnected because peer lost");
			break;

		case SAP_CONNECTION_TERMINATED_REASON_DEVICE_DETACHED:
			dlog_print(DLOG_INFO, LOG_TAG, "disconnected because device is detached");
			break;

		default:
			// fall through
		case SAP_CONNECTION_TERMINATED_REASON_UNKNOWN:
			dlog_print(DLOG_INFO, LOG_TAG, "disconnected because of unknown reason");
			break;
	}

	sap_socket_destroy(priv_data.socket);
	priv_data.socket = NULL;
	agent_connected = FALSE;
}


static void on_data_recieved(sap_socket_h socket,
		unsigned short int channel_id,
		unsigned int payload_length,
		void *buffer,
		void *user_data)
{
	dlog_print(DLOG_INFO, LOG_TAG, "received data: %p, len:%d", buffer, payload_length);
	//unsigned char *p = realloc(rapdu, payload_length);
//	if (p) {
//		rapdu = p;
//		rapdu_length = payload_length;
//	} else {
//		dlog_print(DLOG_ERROR, LOG_TAG, "not enough for storing data");
//		if (rapdu && rapdu_length >= 2) {
//			dlog_print(DLOG_ERROR, LOG_TAG,
//				   "not enough memory, returning 0x6D00 as R-APDU");
//			rapdu[0] = 0x6D;
//			rapdu[1] = 0x00;
//			rapdu_length = 2;
//		} else {
//			dlog_print(DLOG_ERROR, LOG_TAG,
//				   	"not enough memory, setting R-APDU to nothing");
//			rapdu_length = 0;
//		}
//	}

//	rapdu_received = TRUE;

	send_apdu_response(priv_data.nfc_handle, buffer, payload_length);
}

message_queue_s *read_message() {
	if (m_queue != NULL) {
		message_queue_s *message = m_queue;
		m_queue = message->next;
		return message;
	}
	return NULL;
}


gboolean add_message(void *message, unsigned int message_len) {
	message_queue_s *message_pointer = m_queue;
	message_queue_s *prev = message_pointer;
	while (message_pointer != NULL) {
		prev = message_pointer;
		message_pointer = message_pointer->next;
	}
	message_pointer = malloc(sizeof (message_queue_s));
	if (message_pointer != NULL) {
		if (m_queue == NULL) {
			m_queue = message_pointer;
		} else {
			prev->next = message_pointer;
		}
		message_pointer->message = message;
		message_pointer->message_len = message_len;
		message_pointer->next = NULL;
		return TRUE;
	}
	return FALSE;
}

gboolean send_data(nfc_se_h nfc_handle, void *message, unsigned int message_len) {
	gboolean result;
	if (agent_connected) {
		priv_data.nfc_handle = nfc_handle;
		result = sap_socket_send_data(priv_data.socket,
				HELLO_ACCESSORY_CHANNELID, message_len, message);

		if (result != SAP_RESULT_SUCCESS) {

		}
	} else {
		result = add_message(message, message_len);
	}
	return result;
}

static void on_service_connection_created(sap_peer_agent_h peer_agent,
		sap_socket_h socket,
		sap_service_connection_result_e result,
		void *user_data)
{
	switch (result) {
		case SAP_CONNECTION_SUCCESS:
			dlog_print(DLOG_INFO, LOG_TAG, "peer agent connection is successful, pa :%u", peer_agent);
			sap_peer_agent_set_service_connection_terminated_cb(priv_data.peer_agent,
					on_service_connection_terminated,
					NULL);

			sap_socket_set_data_received_cb(socket, on_data_recieved, peer_agent);
			priv_data.socket = socket;
			agent_connected = TRUE;
			// read messages in queue and send them
			message_queue_s *message_queue = read_message();
			while (message_queue != NULL) {
				send_data(priv_data.nfc_handle, message_queue->message, message_queue->message_len);
				message_queue = read_message();
			}
			break;

		case SAP_CONNECTION_ALREADY_EXIST:
			dlog_print(DLOG_INFO, LOG_TAG, "connection is already exist");
			priv_data.socket = socket;
			break;

		case SAP_CONNECTION_FAILURE_DEVICE_UNREACHABLE:
			dlog_print(DLOG_INFO, LOG_TAG, "device is not unreachable");
			break;

		case SAP_CONNECTION_FAILURE_INVALID_PEERAGENT:
			dlog_print(DLOG_INFO, LOG_TAG, "invalid peer agent");
			break;

		case SAP_CONNECTION_FAILURE_NETWORK:
			dlog_print(DLOG_INFO, LOG_TAG, "network failure");
			break;

		case SAP_CONNECTION_FAILURE_PEERAGENT_NO_RESPONSE:
			dlog_print(DLOG_INFO, LOG_TAG, "peer agent is no response");
			break;

		case SAP_CONNECTION_FAILURE_PEERAGENT_REJECTED:
			dlog_print(DLOG_INFO, LOG_TAG, "peer agent is rejected");
			break;

		case SAP_CONNECTION_FAILURE_UNKNOWN:
			dlog_print(DLOG_INFO, LOG_TAG, "unknown error");
			break;
	}
}

static gboolean _create_service_connection(gpointer user_data)
{
	struct priv *priv = NULL;
	sap_result_e result = SAP_RESULT_FAILURE;

	priv = (struct priv *)user_data;
	result = sap_agent_request_service_connection(priv->agent,
			priv->peer_agent,
			on_service_connection_created,
			NULL);

	if (result == SAP_RESULT_SUCCESS) {
		dlog_print(DLOG_DEBUG, LOG_TAG, "req service conn call succeeded");
	} else {
		dlog_print(DLOG_ERROR, LOG_TAG, "req service conn call is failed (%d)", result);
	}

	return FALSE;
}

gboolean request_service_connection(void)
{
	g_idle_add(_create_service_connection, &priv_data);

	dlog_print(DLOG_DEBUG, LOG_TAG, "request_service_connection call over");
	return TRUE;
}

static gboolean _terminate_service_connection(gpointer user_data)
{
	struct priv *priv = NULL;
	sap_result_e result = SAP_RESULT_FAILURE;

	priv = (struct priv *)user_data;

	if (priv->socket)
		result = sap_peer_agent_terminate_service_connection(priv->peer_agent);
	else {
		return FALSE;
	}

	if (result == SAP_RESULT_SUCCESS) {
		dlog_print(DLOG_DEBUG, LOG_TAG, "terminate service conn call succeeded");

		priv->socket = NULL;
		agent_connected = FALSE;
	} else {
		dlog_print(DLOG_ERROR, LOG_TAG, "terminate service conn call is failed (%d)", result);
	}

	return FALSE;
}

gboolean terminate_service_connection(void)
{
	g_idle_add(_terminate_service_connection, &priv_data);

	return TRUE;
}

static gboolean _find_peer_agent(gpointer user_data)
{
	struct priv *priv = NULL;
	sap_result_e result = SAP_RESULT_FAILURE;

	priv = (struct priv *)user_data;

	result = sap_agent_find_peer_agent(priv->agent, on_peer_agent_updated, NULL);

	if (result == SAP_RESULT_SUCCESS) {
		dlog_print(DLOG_DEBUG, LOG_TAG, "find peer call succeeded");
	} else {
		dlog_print(DLOG_ERROR, LOG_TAG, "find peer call failed (%d)", result);
	}

	return FALSE;
}

gboolean find_peers()
{
	g_idle_add(_find_peer_agent, &priv_data);
	dlog_print(DLOG_DEBUG, LOG_TAG, "find peer called");
	return TRUE;
}

GSList* request_installed_aids() {
	GSList* aid_list = g_slist_append(NULL, "A000000397425446590201");
	return aid_list;
}

static void on_agent_initialized(sap_agent_h agent,
		sap_agent_initialized_result_e result,
		void *user_data)
{
	switch (result) {
		case SAP_AGENT_INITIALIZED_RESULT_SUCCESS:
			dlog_print(DLOG_INFO, LOG_TAG, "agent is initialized");

			priv_data.agent = agent;
			agent_created = TRUE;
			break;

		case SAP_AGENT_INITIALIZED_RESULT_DUPLICATED:
			dlog_print(DLOG_INFO, LOG_TAG, "duplicate registration");
			break;

		case SAP_AGENT_INITIALIZED_RESULT_INVALID_ARGUMENTS:
			dlog_print(DLOG_INFO, LOG_TAG, "invalid arguments");
			break;

		case SAP_AGENT_INITIALIZED_RESULT_INTERNAL_ERROR:
			dlog_print(DLOG_INFO, LOG_TAG, "internal sap error");
			break;

		default:
			dlog_print(DLOG_INFO, LOG_TAG, "unknown status (%d)", result);
			break;
	}
}

static void on_device_status_changed(sap_device_status_e status, sap_transport_type_e transport_type,
		void *user_data)
{
	switch (transport_type) {
		case SAP_TRANSPORT_TYPE_BT:
			dlog_print(DLOG_INFO, LOG_TAG, "connectivity type(%d): bt", transport_type);
			break;

		case SAP_TRANSPORT_TYPE_BLE:
			dlog_print(DLOG_INFO, LOG_TAG, "connectivity type(%d): ble", transport_type);
			break;

		case SAP_TRANSPORT_TYPE_TCP:
			dlog_print(DLOG_INFO, LOG_TAG, "connectivity type(%d): tcp/ip", transport_type);
			break;

		case SAP_TRANSPORT_TYPE_USB:
			dlog_print(DLOG_INFO, LOG_TAG, "connectivity type(%d): usb", transport_type);
			break;

		case SAP_TRANSPORT_TYPE_MOBILE:
			dlog_print(DLOG_INFO, LOG_TAG, "connectivity type(%d): mobile", transport_type);
			break;

		default:
			dlog_print(DLOG_INFO, LOG_TAG, "unknown connectivity type (%d)", transport_type);
			break;
	}

	switch (status) {
		case SAP_DEVICE_STATUS_DETACHED:

			if (priv_data.peer_agent) {
				sap_socket_destroy(priv_data.socket);
				priv_data.socket = NULL;
				sap_peer_agent_destroy(priv_data.peer_agent);
				priv_data.peer_agent = NULL;
			}
			break;

		case SAP_DEVICE_STATUS_ATTACHED:
			if (agent_created) {
				g_idle_add(_find_peer_agent, &priv_data);
			}
			break;

		default:
			dlog_print(DLOG_INFO, LOG_TAG, "unknown status (%d)", status);
			break;
	}
}

gboolean agent_initialize()
{
	int result = 0, i = 1;
	gboolean agent_initialized = TRUE;

	do {
		if (i > 100) {
			agent_initialized = FALSE;
			break;
		}
		result = sap_agent_initialize(priv_data.agent,
				HELLO_ACCESSORY_PROFILE_ID, SAP_AGENT_ROLE_CONSUMER,
				on_agent_initialized, NULL);
		dlog_print(DLOG_DEBUG, LOG_TAG, "sap_agent_initialize >>> %d", result);
		i++;
	} while (result != SAP_RESULT_SUCCESS);

	return agent_initialized;
}

void initialize_sap()
{
	sap_agent_h agent = NULL;

	sap_agent_create(&agent);

	if (agent == NULL)
		dlog_print(DLOG_ERROR, LOG_TAG, "ERROR in creating agent");
	else
		dlog_print(DLOG_INFO, LOG_TAG, "SUCCESSFULLY create sap agent");

	priv_data.agent = agent;

	sap_set_device_status_changed_cb(on_device_status_changed, NULL);

	agent_initialize();
}