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

#include <freertos/FreeRTOS.h>
#include <freertos/semphr.h>
#include <freertos/task.h>
#include <freertos/queue.h>
#include <esp_log.h>

#include "eos.h"
#include "msgq.h"
#include "net.h"
#include "cell.h"

#define CELL_SIZE_QUEUE     2

static const char *TAG = "EOS CELL";

static EOSBufQ cell_buf_q;
static unsigned char *cell_bufq_array[CELL_SIZE_QUEUE];

static SemaphoreHandle_t mutex;
static QueueHandle_t cell_queue;

static void _cell_handler(unsigned char _mtype, unsigned char *buffer, uint16_t buf_len) {
    uint8_t mtype;

    mtype = buffer[0];
    switch (mtype & EOS_CELL_MTYPE_MASK) {
        case EOS_CELL_MTYPE_DEV: {
            switch (mtype & ~EOS_CELL_MTYPE_MASK) {
                case EOS_CELL_MTYPE_STATUS: {
                    size_t rv;

                    rv = eos_modem_get_status(buffer + 1);
                    eos_net_reply(EOS_NET_MTYPE_CELL, buffer, rv + 1);
                    break;
                }

                case EOS_CELL_MTYPE_RESET: {
                    int rv;

                    rv = eos_modem_take(1000);
                    if (rv) {
                        ESP_LOGE(TAG, "Reset modem failed: %d", rv);
                        break;
                    }
                    eos_modem_reset();
                    eos_modem_give();
                    break;
                }

                case EOS_CELL_MTYPE_UART_TAKE: {
                    int rv;

                    if (eos_modem_get_mode() == EOS_CELL_UART_MODE_ATCMD) {
                        rv = eos_modem_set_mode(EOS_CELL_UART_MODE_RELAY);
                        if (rv) ESP_LOGE(TAG, "Set RELAY mode failed: %d", rv);
                    }
                    break;
                }

                case EOS_CELL_MTYPE_UART_GIVE: {
                    int rv;

                    if (eos_modem_get_mode() == EOS_CELL_UART_MODE_RELAY) {
                        rv = eos_modem_take(1000);
                        if (!rv) {
                            eos_modem_atinit();
                            eos_modem_give();
                        }
                        rv = eos_modem_set_mode(EOS_CELL_UART_MODE_ATCMD);
                        if (rv) ESP_LOGE(TAG, "Set ATCMD mode failed: %d", rv);
                    }
                    break;
                }

                case EOS_CELL_MTYPE_UART_DATA: {
                    if (eos_modem_get_mode() == EOS_CELL_UART_MODE_RELAY) {
                        eos_modem_write(buffer + 1, buf_len - 1);
                    }
                    break;
                }
            }
            break;
        }

        case EOS_CELL_MTYPE_VOICE: {
            eos_cell_voice_handler(mtype & ~EOS_CELL_MTYPE_MASK, buffer, buf_len);
            break;
        }

        case EOS_CELL_MTYPE_SMS: {
            eos_cell_sms_handler(mtype & ~EOS_CELL_MTYPE_MASK, buffer, buf_len);
            break;
        }

        case EOS_CELL_MTYPE_USSD: {
            eos_cell_ussd_handler(mtype & ~EOS_CELL_MTYPE_MASK, buffer, buf_len);
            break;
        }

        case EOS_CELL_MTYPE_PDP: {
            eos_cell_pdp_handler(mtype & ~EOS_CELL_MTYPE_MASK, buffer, buf_len);
            break;
        }
    }
}

static void cell_handler_task(void *pvParameters) {
    EOSMsgItem mi;

    while (1) {
        if (xQueueReceive(cell_queue, &mi, portMAX_DELAY)) {
            _cell_handler(mi.type, mi.buffer, mi.len);
            xSemaphoreTake(mutex, portMAX_DELAY);
            eos_bufq_push(&cell_buf_q, mi.buffer);
            xSemaphoreGive(mutex);
        }
    }
    vTaskDelete(NULL);
}

static void cell_handler(unsigned char type, unsigned char *buffer, uint16_t buf_len) {
    EOSMsgItem mi;
    unsigned char *_buffer;

    if (buf_len < 1) return;

    if (type & EOS_NET_MTYPE_FLAG_REPL) {
        _cell_handler(type, buffer, buf_len);
        return;
    }

    xSemaphoreTake(mutex, portMAX_DELAY);
    _buffer = eos_bufq_pop(&cell_buf_q);
    xSemaphoreGive(mutex);

    if (_buffer == NULL) {
        ESP_LOGE(TAG, "Cell message NOT handled: %2x", buffer[0]);
        return;
    }

    memcpy(_buffer, buffer, buf_len);
    mi.type = type;
    mi.buffer = _buffer;
    mi.len = buf_len;
    xQueueSend(cell_queue, &mi, portMAX_DELAY);
}

void eos_cell_init(void) {
    int i;

    eos_bufq_init(&cell_buf_q, cell_bufq_array, CELL_SIZE_QUEUE);
    for (i=0; i<CELL_SIZE_QUEUE; i++) {
        unsigned char *buffer;

        buffer = malloc(EOS_NET_SIZE_BUF);
        assert(buffer != NULL);
        eos_bufq_push(&cell_buf_q, buffer);
    }

    mutex = xSemaphoreCreateBinary();
    assert(mutex != NULL);
    xSemaphoreGive(mutex);

    cell_queue = xQueueCreate(CELL_SIZE_QUEUE, sizeof(EOSMsgItem));
    assert(cell_queue != NULL);

    eos_net_set_handler(EOS_NET_MTYPE_CELL, cell_handler);

    ESP_LOGI(TAG, "INIT");
}

void eos_cell_run(void) {
    BaseType_t rv;

    rv = xTaskCreate(cell_handler_task, "cell_handler", EOS_TASK_SSIZE_CELL, NULL, EOS_TASK_PRIORITY_CELL, NULL);
    assert(rv == pdPASS);

    ESP_LOGI(TAG, "RUN");
}