/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 <assert.h>
#include <string.h>

#include "ble_hs.h"
#include "ble_svc_gap.h"
#include "ble_endian.h"
#include "tuya_hs_port.h"

#define PPCP_ENABLED                                                                                                   \
    (TY_HS_BLE_ROLE_PERIPHERAL) &&                                                                                     \
        ((TY_HS_BLE_SVC_GAP_PPCP_MIN_CONN_INTERVAL) || (TY_HS_BLE_SVC_GAP_PPCP_MAX_CONN_INTERVAL) ||                   \
         (TY_HS_BLE_SVC_GAP_PPCP_SLAVE_LATENCY) || (TY_HS_BLE_SVC_GAP_PPCP_SUPERVISION_TMO))

#define BLE_SVC_GAP_NAME_MAX_LEN (TY_HS_BLE_SVC_GAP_DEVICE_NAME_MAX_LENGTH)

static ble_svc_gap_chr_changed_fn *ble_svc_gap_chr_changed_cb_fn;

static char ble_svc_gap_name[BLE_SVC_GAP_NAME_MAX_LEN + 1] = (TY_HS_BLE_SVC_GAP_DEVICE_NAME);
static uint16_t ble_svc_gap_appearance = (TY_HS_BLE_SVC_GAP_APPEARANCE);

#if TY_HS_BLE_CONNECT
static int ble_svc_gap_access(uint16_t conn_handle, uint16_t attr_handle, struct ble_gatt_access_ctxt *ctxt, void *arg);

static __attribute__((unused)) const struct ble_gatt_svc_def ble_svc_gap_defs[] = {
    {
        /*** Service: GAP. */
        .type = BLE_GATT_SVC_TYPE_PRIMARY,
        .uuid = BLE_UUID16_DECLARE(BLE_SVC_GAP_UUID16),
        .characteristics =
            (struct ble_gatt_chr_def[]){
                {
                    /*** Characteristic: Device Name. */
                    .uuid = BLE_UUID16_DECLARE(BLE_SVC_GAP_CHR_UUID16_DEVICE_NAME),
                    .access_cb = ble_svc_gap_access,
                    .flags = BLE_GATT_CHR_F_READ |
#if (TY_HS_BLE_SVC_GAP_DEVICE_NAME_WRITE_PERM) >= 0
                             BLE_GATT_CHR_F_WRITE | (TY_HS_BLE_SVC_GAP_DEVICE_NAME_WRITE_PERM) |
#endif
                             0,
                },
                {
                    /*** Characteristic: Appearance. */
                    .uuid = BLE_UUID16_DECLARE(BLE_SVC_GAP_CHR_UUID16_APPEARANCE),
                    .access_cb = ble_svc_gap_access,
                    .flags = BLE_GATT_CHR_F_READ |
#if (TY_HS_BLE_SVC_GAP_APPEARANCE_WRITE_PERM) >= 0
                             BLE_GATT_CHR_F_WRITE | (TY_HS_BLE_SVC_GAP_APPEARANCE_WRITE_PERM) |
#endif
                             0,
                },
                {
#if PPCP_ENABLED
                    /*** Characteristic: Peripheral Preferred Connection Parameters. */
                    .uuid = BLE_UUID16_DECLARE(BLE_SVC_GAP_CHR_UUID16_PERIPH_PREF_CONN_PARAMS),
                    .access_cb = ble_svc_gap_access,
                    .flags = BLE_GATT_CHR_F_READ,
                },
                {
#endif
#if (TY_HS_BLE_SVC_GAP_CENTRAL_ADDRESS_RESOLUTION) >= 0
                    /*** Characteristic: Central Address Resolution. */
                    .uuid = BLE_UUID16_DECLARE(BLE_SVC_GAP_CHR_UUID16_CENTRAL_ADDRESS_RESOLUTION),
                    .access_cb = ble_svc_gap_access,
                    .flags = BLE_GATT_CHR_F_READ,
                },
                {
#endif
                    0, /* No more characteristics in this service. */
                }},
    },

    {
        0, /* No more services. */
    },
};

static int ble_svc_gap_device_name_read_access(struct ble_gatt_access_ctxt *ctxt)
{
    int rc;

    rc = os_mbuf_append(ctxt->om, ble_svc_gap_name, strlen(ble_svc_gap_name));

    return rc == 0 ? 0 : BLE_ATT_ERR_INSUFFICIENT_RES;
}

static int ble_svc_gap_device_name_write_access(struct ble_gatt_access_ctxt *ctxt)
{
#if (TY_HS_BLE_SVC_GAP_DEVICE_NAME_WRITE_PERM) < 0
    TUYA_HS_ASSERT(0);
    return 0;
#else
    uint16_t om_len;
    int rc;

    om_len = OS_MBUF_PKTLEN(ctxt->om);
    if (om_len > BLE_SVC_GAP_NAME_MAX_LEN) {
        return BLE_ATT_ERR_INVALID_ATTR_VALUE_LEN;
    }

    rc = ble_hs_mbuf_to_flat(ctxt->om, ble_svc_gap_name, om_len, NULL);
    if (rc != 0) {
        return BLE_ATT_ERR_UNLIKELY;
    }

    ble_svc_gap_name[om_len] = '\0';

    if (ble_svc_gap_chr_changed_cb_fn) {
        ble_svc_gap_chr_changed_cb_fn(BLE_SVC_GAP_CHR_UUID16_DEVICE_NAME);
    }

    return rc;
#endif
}

static int ble_svc_gap_appearance_read_access(struct ble_gatt_access_ctxt *ctxt)
{
    uint16_t appearance = htole16(ble_svc_gap_appearance);
    int rc;

    rc = os_mbuf_append(ctxt->om, &appearance, sizeof(appearance));

    return rc == 0 ? 0 : BLE_ATT_ERR_INSUFFICIENT_RES;
}

static int ble_svc_gap_appearance_write_access(struct ble_gatt_access_ctxt *ctxt)
{
#if (TY_HS_BLE_SVC_GAP_APPEARANCE_WRITE_PERM) < 0
    TUYA_HS_ASSERT(0);
    if (ble_svc_gap_chr_changed_cb_fn) {
        ble_svc_gap_chr_changed_cb_fn = ble_svc_gap_chr_changed_cb_fn; // avoid warning
    }
    return 0;
#else
    uint16_t om_len;
    int rc;

    om_len = OS_MBUF_PKTLEN(ctxt->om);
    if (om_len != sizeof(ble_svc_gap_appearance)) {
        return BLE_ATT_ERR_INVALID_ATTR_VALUE_LEN;
    }

    rc = ble_hs_mbuf_to_flat(ctxt->om, &ble_svc_gap_appearance, om_len, NULL);
    if (rc != 0) {
        return BLE_ATT_ERR_UNLIKELY;
    }

    ble_svc_gap_appearance = le16toh(ble_svc_gap_appearance);

    if (ble_svc_gap_chr_changed_cb_fn) {
        ble_svc_gap_chr_changed_cb_fn(BLE_SVC_GAP_CHR_UUID16_APPEARANCE);
    }

    return rc;
#endif
}

static int ble_svc_gap_access(uint16_t conn_handle, uint16_t attr_handle, struct ble_gatt_access_ctxt *ctxt, void *arg)
{
    uint16_t uuid16;
#if (TY_HS_BLE_SVC_GAP_CENTRAL_ADDRESS_RESOLUTION) >= 0
    uint8_t central_ar = (TY_HS_BLE_SVC_GAP_CENTRAL_ADDRESS_RESOLUTION);
#endif
#if PPCP_ENABLED
    uint16_t ppcp[4] = {
        htole16((TY_HS_BLE_SVC_GAP_PPCP_MIN_CONN_INTERVAL)), htole16((TY_HS_BLE_SVC_GAP_PPCP_MAX_CONN_INTERVAL)),
        htole16((TY_HS_BLE_SVC_GAP_PPCP_SLAVE_LATENCY)), htole16((TY_HS_BLE_SVC_GAP_PPCP_SUPERVISION_TMO))};
#endif
    int rc;

    uuid16 = ble_uuid_u16(ctxt->chr->uuid);
    TUYA_HS_ASSERT(uuid16 != 0);

    switch (uuid16) {
    case BLE_SVC_GAP_CHR_UUID16_DEVICE_NAME:
        if (ctxt->op == BLE_GATT_ACCESS_OP_READ_CHR) {
            rc = ble_svc_gap_device_name_read_access(ctxt);
        } else if (ctxt->op == BLE_GATT_ACCESS_OP_WRITE_CHR) {
            rc = ble_svc_gap_device_name_write_access(ctxt);
        } else {
            TUYA_HS_ASSERT(0);
            rc = BLE_ATT_ERR_UNLIKELY;
        }
        return rc;

    case BLE_SVC_GAP_CHR_UUID16_APPEARANCE:
        if (ctxt->op == BLE_GATT_ACCESS_OP_READ_CHR) {
            rc = ble_svc_gap_appearance_read_access(ctxt);
        } else if (ctxt->op == BLE_GATT_ACCESS_OP_WRITE_CHR) {
            rc = ble_svc_gap_appearance_write_access(ctxt);
        } else {
            TUYA_HS_ASSERT(0);
            rc = BLE_ATT_ERR_UNLIKELY;
        }
        return rc;

#if PPCP_ENABLED
    case BLE_SVC_GAP_CHR_UUID16_PERIPH_PREF_CONN_PARAMS:
        TUYA_HS_ASSERT(ctxt->op == BLE_GATT_ACCESS_OP_READ_CHR);
        rc = os_mbuf_append(ctxt->om, &ppcp, sizeof(ppcp));
        return rc == 0 ? 0 : BLE_ATT_ERR_INSUFFICIENT_RES;
#endif

#if (TY_HS_BLE_SVC_GAP_CENTRAL_ADDRESS_RESOLUTION) >= 0
    case BLE_SVC_GAP_CHR_UUID16_CENTRAL_ADDRESS_RESOLUTION:
        TUYA_HS_ASSERT(ctxt->op == BLE_GATT_ACCESS_OP_READ_CHR);
        rc = os_mbuf_append(ctxt->om, &central_ar, sizeof(central_ar));
        return rc == 0 ? 0 : BLE_ATT_ERR_INSUFFICIENT_RES;
#endif

    default:
        TUYA_HS_ASSERT(0);
        return BLE_ATT_ERR_UNLIKELY;
    }
}
#endif

const char *ble_svc_gap_device_name(void)
{
    return ble_svc_gap_name;
}

int ble_svc_gap_device_name_set(const char *name)
{
    int len;

    len = strlen(name);
    if (len > BLE_SVC_GAP_NAME_MAX_LEN) {
        return BLE_HS_EINVAL;
    }

    memcpy(ble_svc_gap_name, name, len);
    ble_svc_gap_name[len] = '\0';

    return 0;
}

uint16_t ble_svc_gap_device_appearance(void)
{
    return ble_svc_gap_appearance;
}

int ble_svc_gap_device_appearance_set(uint16_t appearance)
{
    ble_svc_gap_appearance = appearance;

    return 0;
}

void ble_svc_gap_set_chr_changed_cb(ble_svc_gap_chr_changed_fn *cb)
{
    ble_svc_gap_chr_changed_cb_fn = cb;
}

void ble_svc_gap_init(void)
{
#if 0 // clost tuyaos default config
#if TY_HS_BLE_CONNECT
    int rc;
#endif

    /* Ensure this function only gets called by sysinit. */
    BLE_ASSERT_ACTIVE();


    rc = ble_gatts_count_cfg(ble_svc_gap_defs);
    BLE_PANIC_ASSERT(rc == 0);

    rc = ble_gatts_add_svcs(ble_svc_gap_defs);
    BLE_PANIC_ASSERT(rc == 0);
#endif
}