#include <stdlib.h>
#include <string.h>

#include "../shared/bin.h"
#include "../shared/errcatch.h"
#include "../shared/serial_parse.h"
#include "hypervisor.h"
#include "io.h"
#include "mode_dirc.h"
#include "mode_grid.h"
#include "modes.h"
#include "orangutan_shim.h"
#include "sercomm.h"

w2_s_bin *g_w2_sercomm_buffer[W2_SERCOMM_BUFFER_SIZE] = {0};
uint8_t g_w2_sercomm_index							  = 0;
uint8_t g_w2_sercomm_offset							  = 0;
uint8_t g_w2_sercomm_buffer_full					  = 0;

char g_w2_serial_buffer[W2_SERIAL_READ_BUFFER_SIZE] = {0};
uint8_t g_w2_serial_buffer_index					= 0;
uint8_t g_w2_serial_buffer_head						= 0;

void w2_sercomm_main() {
#ifdef W2_SIM
	simprintfunc("w2_sercomm_main", "");
#endif
	// read and parse data
	while (serial_get_received_bytes() != g_w2_serial_buffer_index) {
		if (!w2_serial_parse(g_w2_serial_buffer[g_w2_serial_buffer_index]))
			w2_errcatch_throw(W2_E_WARN_SERIAL_NOISY);
		g_w2_serial_buffer_index = (g_w2_serial_buffer_index + 1) % W2_SERIAL_READ_BUFFER_SIZE;
	}

	// check time-out
	if (!g_w2_ping_received && w2_hypervisor_time_end(W2_TIMER_PING) > W2_PING_TIMEOUT) {
		g_w2_ping_timeout = true;
		g_w2_connected	  = false;
		w2_errcatch_throw(W2_E_WARN_PING_TIMEOUT);
	}
	// send ping every W2_TIMER_PING ms
	if ((g_w2_ping_received && w2_hypervisor_time_end(W2_TIMER_PING) > W2_PING_FREQUENCY) ||
		g_w2_ping_timeout) {
		g_w2_ping_timeout  = false;
		g_w2_ping_received = false;
		g_w2_ping_id	   = (uint8_t)rand();

		W2_CREATE_MSG_BIN(w2_s_cmd_ping_tx, msg, bin);
		msg->opcode = W2_CMD_PING | W2_CMDDIR_TX;
		msg->id		= g_w2_ping_id;

		w2_sercomm_append_msg(bin);
		free(bin);

		w2_hypervisor_time_start(W2_TIMER_PING);
	}

	// send data
	while (g_w2_sercomm_offset != g_w2_sercomm_index) {
		w2_s_bin *data = g_w2_sercomm_buffer[g_w2_sercomm_offset];
#ifdef W2_SIM
		if (DBG_ENABLE_SERIAL) w2_sim_print_serial(data);
#endif
		serial_send_blocking("\xff", 1);
		for (uint8_t i = 0; i < data->bytes; i++) {
			uint8_t byte = data->data[i];
			byte == 0xff ? serial_send_blocking("\xff\xff", 2)
						 : serial_send_blocking((char *)&byte, 1);
		}
		g_w2_sercomm_offset = (g_w2_sercomm_offset + 1) % W2_SERCOMM_BUFFER_SIZE;
	}
}

void w2_sercomm_append_msg(w2_s_bin *data) {
#ifdef W2_SIM
	simprintfunc("w2_sercomm_append_msg", "");
#endif
	uint8_t next_index		 = (g_w2_sercomm_index + 1) % W2_SERCOMM_BUFFER_SIZE;
	g_w2_sercomm_buffer_full = next_index == g_w2_sercomm_offset;
	free(g_w2_sercomm_buffer[g_w2_sercomm_index]);
	g_w2_sercomm_buffer[g_w2_sercomm_index] = w2_bin_s_alloc(data->bytes, data->data);
	if (g_w2_sercomm_buffer_full) return;
	g_w2_sercomm_index = next_index;
}

#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Warray-bounds"

#include <stdlib.h>
#include <string.h>

void w2_cmd_ping_tx(w2_s_bin *data) {
	g_w2_ping_ms	   = w2_hypervisor_time_end(W2_TIMER_PING);
	g_w2_ping_received = true;
	g_w2_ping_timeout  = false;
	g_w2_connected	   = true;
}

void w2_cmd_ping_rx(w2_s_bin *data) { w2_sercomm_append_msg(data); }

void w2_cmd_mode_rx(w2_s_bin *data) {
	W2_CAST_BIN(w2_s_cmd_mode_rx, data, req);
	if (req->mode == W2_M_SCAL) {
		w2_modes_call(req->mode);
	} else {
		w2_modes_swap(req->mode);
	}
}

void w2_cmd_sped_rx(w2_s_bin *data) { return; }

void w2_cmd_dirc_rx(w2_s_bin *data) {
	W2_CAST_BIN(w2_s_cmd_dirc_rx, data, req);

	g_w2_mode_dirc_motor_l = w2_bin_ntoh16(req->left);
	g_w2_mode_dirc_motor_r = w2_bin_ntoh16(req->right);
}

void w2_cmd_cord_rx(w2_s_bin *data) { return; }

#include <stdio.h>
void w2_cmd_bomd_rx(w2_s_bin *data) {
	W2_CAST_BIN(w2_s_cmd_bomd_rx, data, req);

	req->position = w2_bin_hton32(req->position);

	char buf[32];
	clear();
	// sprintf(buf, "%lu, %lu", req->position % W2_MAP_DEFAULT_WIDTH, req->position / W2_MAP_DEFAULT_WIDTH);
	sprintf(buf, "%lu", req->position);
	print(buf);
	g_w2_order[g_w2_order_index].x = req->position % W2_MAP_DEFAULT_WIDTH;
	g_w2_order[g_w2_order_index].y = req->position / W2_MAP_DEFAULT_WIDTH;
	g_w2_order_index++;
}

void w2_cmd_sres_rx(w2_s_bin *data) {
	W2_CAST_BIN(w2_s_cmd_sres_rx, data, req);

	switch (req->type) {
		case W2_CMD_SRES_RX_TYPE_REINITGS: {
			// TODO: soft-reset
			break;
		}
		case W2_CMD_SRES_RX_TYPE_PREVMODE: {
			w2_modes_call(W2_M_PREV);
			break;
		}
		default: {
			w2_errcatch_throw(W2_E_WARN_SERIAL_NOISY);
		}
	}
}

void w2_cmd_mcfg_rx(w2_s_bin *data) { return; }

void w2_cmd_info_rx(w2_s_bin *data) {
	W2_CREATE_MSG_BIN(w2_s_cmd_info_tx, res_msg, res_bin);
	res_msg->opcode = W2_CMD_INFO | W2_CMDDIR_TX;
	strncpy((char *)res_msg->build_str, W2_BUILD_STR, sizeof(res_msg->build_str));
	res_msg->errcatch_ms	 = (uint8_t)g_w2_hypervisor_ema_errcatch_qs / 1e3;
	res_msg->io_ms			 = (uint8_t)g_w2_hypervisor_ema_io_qs / 1e3;
	res_msg->sercomm_ms		 = (uint8_t)g_w2_hypervisor_ema_sercomm_qs / 1e3;
	res_msg->mode_ms		 = (uint8_t)g_w2_hypervisor_ema_mode_qs / 1e3;
	res_msg->uptime_s		 = w2_bin_hton32((uint32_t)(g_w2_hypervisor_uptime_qs / 1e6));
	res_msg->mode			 = g_w2_mode_history[g_w2_mode_history_index];
	res_msg->battery_mv		 = w2_bin_hton16(g_w2_io_battery_mv);
	res_msg->battery_percent = g_w2_io_battery_percentage;

	w2_sercomm_append_msg(res_bin);
	free(res_bin);
}

void w2_cmd_disp_rx(w2_s_bin *data) { return; }

void w2_cmd_play_rx(w2_s_bin *data) { return; }

void w2_cmd_cled_rx(w2_s_bin *data) { return; }

void w2_cmd_tarq_rx(w2_s_bin *data) {
	W2_CAST_BIN(w2_s_cmd_tarq_rx, data, req);
	g_w2_target_area = req->target_area;
}

#pragma GCC diagnostic pop

void w2_cmd_expt_tx(w2_s_bin *data) {}
void w2_cmd_mode_tx(w2_s_bin *data) {}
void w2_cmd_cord_tx(w2_s_bin *data) {}
void w2_cmd_bomd_tx(w2_s_bin *data) {}
void w2_cmd_sens_tx(w2_s_bin *data) {}
void w2_cmd_info_tx(w2_s_bin *data) {}