#include "common/bt_target.h"
#include "osi/allocator.h"

#include <string.h>
#include "gatt_int.h"
#include "stack/l2c_api.h"
#include "l2c_int.h"
#include "smp_int.h"

#if (BLE_INCLUDED == TRUE && GATTS_INCLUDED == TRUE)

const char *const gatt_attr_name[] = {
    "primary service",
    "secondary service",
    "included service",
    "characteristic",
};

const char *const gatt_char_desc_name[] = {
    "characteristic extended properties",
    "characteristic user description",
    "client characteristic configuration",
    "server characteristic configuration",
    "characteristic presentation format",
    "characteristic aggregate format",
};

static const char *gatt_get_attr_name(UINT16 uuid)
{
    if (uuid >= GATT_UUID_PRI_SERVICE && uuid <= GATT_UUID_CHAR_DECLARE) {
        return gatt_attr_name[uuid - GATT_UUID_PRI_SERVICE];
    }

    if (uuid >= GATT_UUID_CHAR_EXT_PROP && uuid <= GATT_UUID_CHAR_AGG_FORMAT) {
        return gatt_char_desc_name[uuid - GATT_UUID_CHAR_EXT_PROP];
    }

    return "Unknown Attribute";
}

static void attr_uuid_to_bt_uuid(void *p_attr, tBT_UUID *p_uuid)
{
    tGATT_ATTR16 *p_attr16 = (tGATT_ATTR16 *)p_attr;

    if (p_attr16->uuid_type == GATT_ATTR_UUID_TYPE_16) {
        p_uuid->len = LEN_UUID_16;
        p_uuid->uu.uuid16 = p_attr16->uuid;
    } else if (p_attr16->uuid_type == GATT_ATTR_UUID_TYPE_32) {
        tGATT_ATTR32 *p_attr32 = (tGATT_ATTR32 *)p_attr;
        p_uuid->len = LEN_UUID_32;
        p_uuid->uu.uuid32 = p_attr32->uuid;
    } else if (p_attr16->uuid_type == GATT_ATTR_UUID_TYPE_128) {
        tGATT_ATTR128 *p_attr128 = (tGATT_ATTR128 *)p_attr;
        p_uuid->len = LEN_UUID_128;
        memcpy(p_uuid->uu.uuid128, p_attr128->uuid, LEN_UUID_128);
    }
}

static size_t calculate_database_info_size(void)
{
    UINT8 i;
    tGATT_SVC_DB *p_db;
    tGATT_ATTR16 *p_attr;
    size_t len = 0;

    for (i = 0; i < GATT_MAX_SR_PROFILES; i++) {
        p_db = gatt_cb.sr_reg[i].p_db;
        if (p_db && p_db->p_attr_list) {
            p_attr = (tGATT_ATTR16 *)p_db->p_attr_list;
            while (p_attr) {
                if (p_attr->uuid == GATT_UUID_PRI_SERVICE ||
                    p_attr->uuid == GATT_UUID_SEC_SERVICE) {
                    // Service declaration
                    len += 4 + p_attr->p_value->uuid.len;
                } else if (p_attr->uuid == GATT_UUID_INCLUDE_SERVICE) {
                    // Included service declaration
                    len += 8 + p_attr->p_value->incl_handle.service_type.len;
                } else if (p_attr->uuid == GATT_UUID_CHAR_DECLARE) {
                    tBT_UUID char_uuid = {0};
                    // Characteristic declaration
                    p_attr = (tGATT_ATTR16 *)p_attr->p_next;
                    attr_uuid_to_bt_uuid((void *)p_attr, &char_uuid);
                    // Increment 1 to fetch characteristic uuid from value declaration attribute
                    len += 7 + char_uuid.len;
                } else if (p_attr->uuid == GATT_UUID_CHAR_DESCRIPTION ||
                    p_attr->uuid == GATT_UUID_CHAR_CLIENT_CONFIG ||
                    p_attr->uuid == GATT_UUID_CHAR_SRVR_CONFIG ||
                    p_attr->uuid == GATT_UUID_CHAR_PRESENT_FORMAT ||
                    p_attr->uuid == GATT_UUID_CHAR_AGG_FORMAT) {
                    // Descriptor
                    len += 4;
                } else if (p_attr->uuid == GATT_UUID_CHAR_EXT_PROP) {
                    // Descriptor
                    len += 6;
                }
                p_attr = (tGATT_ATTR16 *) p_attr->p_next;
            }
        }
    }

    return len;
}

static void fill_database_info(UINT8 *p_data)
{
    UINT8 i;
    tGATT_SVC_DB *p_db;
    tGATT_ATTR16 *p_attr;

    for (i = 0; i < GATT_MAX_SR_PROFILES; i++) {
        p_db = gatt_cb.sr_reg[i].p_db;
        if (p_db && p_db->p_attr_list) {
            p_attr = (tGATT_ATTR16 *)p_db->p_attr_list;
            while (p_attr) {
                if (p_attr->uuid == GATT_UUID_PRI_SERVICE ||
                    p_attr->uuid == GATT_UUID_SEC_SERVICE) {
                    // Service declaration
                    UINT16_TO_STREAM(p_data, p_attr->handle);
                    UINT16_TO_STREAM(p_data, p_attr->uuid);
                    gatt_build_uuid_to_stream(&p_data, p_attr->p_value->uuid);
                } else if (p_attr->uuid == GATT_UUID_INCLUDE_SERVICE) {
                    // Included service declaration
                    UINT16_TO_STREAM(p_data, p_attr->handle);
                    UINT16_TO_STREAM(p_data, GATT_UUID_INCLUDE_SERVICE);
                    UINT16_TO_STREAM(p_data, p_attr->p_value->incl_handle.s_handle);
                    UINT16_TO_STREAM(p_data, p_attr->p_value->incl_handle.e_handle);
                    gatt_build_uuid_to_stream(&p_data, p_attr->p_value->incl_handle.service_type);
                } else if (p_attr->uuid == GATT_UUID_CHAR_DECLARE) {
                    tBT_UUID char_uuid = {0};
                    // Characteristic declaration
                    UINT16_TO_STREAM(p_data, p_attr->handle);
                    UINT16_TO_STREAM(p_data, GATT_UUID_CHAR_DECLARE);
                    UINT8_TO_STREAM(p_data, p_attr->p_value->char_decl.property);
                    UINT16_TO_STREAM(p_data, p_attr->p_value->char_decl.char_val_handle);
                    p_attr = (tGATT_ATTR16 *)p_attr->p_next;
                    attr_uuid_to_bt_uuid((void *)p_attr, &char_uuid);
                    // Increment 1 to fetch characteristic uuid from value declaration attribute
                    gatt_build_uuid_to_stream(&p_data, char_uuid);
                } else if (p_attr->uuid == GATT_UUID_CHAR_DESCRIPTION ||
                    p_attr->uuid == GATT_UUID_CHAR_CLIENT_CONFIG ||
                    p_attr->uuid == GATT_UUID_CHAR_SRVR_CONFIG ||
                    p_attr->uuid == GATT_UUID_CHAR_PRESENT_FORMAT ||
                    p_attr->uuid == GATT_UUID_CHAR_AGG_FORMAT) {
                    // Descriptor
                    UINT16_TO_STREAM(p_data, p_attr->handle);
                    UINT16_TO_STREAM(p_data, p_attr->uuid);
                } else if (p_attr->uuid == GATT_UUID_CHAR_EXT_PROP) {
                    // Descriptor
                    UINT16_TO_STREAM(p_data, p_attr->handle);
                    UINT16_TO_STREAM(p_data, p_attr->uuid);
                    // TODO: process extended properties descriptor
                    if (p_attr->p_value->attr_val.attr_len == 2) {
                        memcpy(p_data, p_attr->p_value->attr_val.attr_val, 2);
                    } else {
                        UINT16_TO_STREAM(p_data, 0x0000);
                    }
                }
                p_attr = (tGATT_ATTR16 *) p_attr->p_next;
            }
        }
    }
}

tGATT_STATUS gatts_calculate_datebase_hash(BT_OCTET16 hash)
{
    UINT8 tmp;
    UINT16 i;
    UINT16 j;
    size_t len;
    UINT8 *data_buf = NULL;

    len = calculate_database_info_size();

    data_buf = (UINT8 *)osi_malloc(len);
    if (data_buf == NULL) {
        GATT_TRACE_ERROR ("%s failed to allocate buffer (%u)\n", __func__, len);
        return GATT_NO_RESOURCES;
    }

    fill_database_info(data_buf);

    // reverse database info
    for (i = 0, j = len-1; i < j; i++, j--) {
        tmp = data_buf[i];
        data_buf[i] = data_buf[j];
        data_buf[j] = tmp;
    }

#if SMP_INCLUDED == TRUE
    BT_OCTET16 key = {0};
    aes_cipher_msg_auth_code(key, data_buf, len, 16, hash);
    //ESP_LOG_BUFFER_HEX("db hash", hash, BT_OCTET16_LEN);
#endif

    osi_free(data_buf);
    return GATT_SUCCESS;
}

void gatts_show_local_database(void)
{
    UINT8 i;
    tGATT_SVC_DB *p_db;
    tGATT_ATTR16 *p_attr;

    printf("\n================= GATTS DATABASE DUMP START =================\n");
    for (i = 0; i < GATT_MAX_SR_PROFILES; i++) {
        p_db = gatt_cb.sr_reg[i].p_db;
        if (p_db && p_db->p_attr_list) {
            p_attr = (tGATT_ATTR16 *)p_db->p_attr_list;
            while (p_attr) {
                switch (p_attr->uuid) {
                case GATT_UUID_PRI_SERVICE:
                case GATT_UUID_SEC_SERVICE:
                    // Service declaration
                    printf("%s\n", gatt_get_attr_name(p_attr->uuid));
                    printf("\tuuid %s\n", gatt_uuid_to_str(&p_attr->p_value->uuid));
                    printf("\thandle %d\n", p_attr->handle);
                    printf("\tend_handle %d\n",p_db->end_handle-1);
                    break;
                case GATT_UUID_INCLUDE_SERVICE:
                    // Included service declaration
                    printf("%s\n", gatt_get_attr_name(p_attr->uuid));
                    printf("\tuuid %s\t", gatt_uuid_to_str(&p_attr->p_value->incl_handle.service_type));
                    printf("\thandle %d\n", p_attr->p_value->incl_handle.s_handle);
                    printf("\tend_handle %d\n", p_attr->p_value->incl_handle.e_handle);
                    break;
                case GATT_UUID_CHAR_DECLARE: {
                    tBT_UUID char_uuid;
                    tGATT_ATTR16 *p_char_val;
                    p_char_val = (tGATT_ATTR16 *)p_attr->p_next;
                    attr_uuid_to_bt_uuid((void *)p_char_val, &char_uuid);

                    printf("%s\n", gatt_get_attr_name(p_attr->uuid));
                    printf("\tuuid %s\n", gatt_uuid_to_str(&char_uuid));
                    printf("\tdef_handle %d\n", p_attr->handle);
                    printf("\tval_handle %d\n", p_attr->p_value->char_decl.char_val_handle);
                    printf("\tperm 0x%04x, prop 0x%02x\n", p_char_val->permission, p_attr->p_value->char_decl.property);
                    break;
                }
                case GATT_UUID_CHAR_EXT_PROP:
                case GATT_UUID_CHAR_DESCRIPTION:
                case GATT_UUID_CHAR_CLIENT_CONFIG:
                case GATT_UUID_CHAR_SRVR_CONFIG:
                case GATT_UUID_CHAR_PRESENT_FORMAT:
                case GATT_UUID_CHAR_AGG_FORMAT:
                    printf("%s\n", gatt_get_attr_name(p_attr->uuid));
                    printf("\thandle %d\n", p_attr->handle);
                    break;
                }
                p_attr = (tGATT_ATTR16 *) p_attr->p_next;
            }
        }
    }
    printf("================= GATTS DATABASE DUMP END =================\n");
}
#endif  /* BLE_INCLUDED == TRUE && GATTS_INCLUDED == TRUE */