/*  Bluetooth Mesh */

/*
 * Copyright (c) 2017 Intel Corporation
 * Additional Copyright (c) 2018 Espressif Systems (Shanghai) PTE LTD
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <string.h>
#include <errno.h>
#include <stdbool.h>

#include "sdkconfig.h"
#define BT_DBG_ENABLED IS_ENABLED(CONFIG_BLE_MESH_DEBUG_MODEL)

#include "mesh_types.h"
#include "mesh_util.h"
#include "mesh_trace.h"
#include "health_srv.h"

#include "mesh.h"
#include "adv.h"
#include "net.h"
#include "transport.h"
#include "access.h"
#include "foundation.h"
#include "mesh_common.h"

#define HEALTH_TEST_STANDARD 0x00

/* Maximum message length is 384 in BLE Mesh. Here for health fault status,
 * due to 1 octet opcode and 4 octets TransMIC, 379 octets can be used to
 * store health fault status.
 */
#define HEALTH_FAULT_MAX_LEN    379

/* Health Server context of the primary element */
struct bt_mesh_health_srv *health_srv;

static void health_get_registered(struct bt_mesh_model *mod,
                                  u16_t company_id,
                                  struct net_buf_simple *msg)
{
    struct bt_mesh_health_srv *srv = mod->user_data;
    u8_t *test_id;

    BT_DBG("Company ID 0x%04x", company_id);

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    bt_mesh_model_msg_init(msg, OP_HEALTH_FAULT_STATUS);

    test_id = net_buf_simple_add(msg, 1);
    net_buf_simple_add_le16(msg, company_id);

    if (srv->cb && srv->cb->fault_get_reg) {
        u8_t fault_count = net_buf_simple_tailroom(msg) - 4;
        int err;

        err = srv->cb->fault_get_reg(mod, company_id, test_id,
                                     net_buf_simple_tail(msg),
                                     &fault_count);
        if (err) {
            BT_ERR("%s, Failed to get faults (err %d)", __func__, err);
            *test_id = HEALTH_TEST_STANDARD;
        } else {
            net_buf_simple_add(msg, fault_count);
        }
    } else {
        BT_WARN("No callback for getting faults");
        *test_id = HEALTH_TEST_STANDARD;
    }
}

static size_t health_get_current(struct bt_mesh_model *mod,
                                 struct net_buf_simple *msg)
{
    struct bt_mesh_health_srv *srv = mod->user_data;
    const struct bt_mesh_comp *comp;
    u8_t *test_id, *company_ptr;
    u16_t company_id;
    u8_t fault_count;
    int err;

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return 0;
    }

    bt_mesh_model_msg_init(msg, OP_HEALTH_CURRENT_STATUS);

    test_id = net_buf_simple_add(msg, 1);
    company_ptr = net_buf_simple_add(msg, sizeof(company_id));
    comp = bt_mesh_comp_get();

    if (srv->cb && srv->cb->fault_get_cur) {
        fault_count = net_buf_simple_tailroom(msg);
        err = srv->cb->fault_get_cur(mod, test_id, &company_id,
                                     net_buf_simple_tail(msg),
                                     &fault_count);
        if (err) {
            BT_ERR("%s, Failed to get faults (err %d)", __func__, err);
            sys_put_le16(comp->cid, company_ptr);
            *test_id = HEALTH_TEST_STANDARD;
            fault_count = 0U;
        } else {
            sys_put_le16(company_id, company_ptr);
            net_buf_simple_add(msg, fault_count);
        }
    } else {
        BT_WARN("No callback for getting faults");
        sys_put_le16(comp->cid, company_ptr);
        *test_id = HEALTH_TEST_STANDARD;
        fault_count = 0U;
    }

    return fault_count;
}

static void health_fault_get(struct bt_mesh_model *model,
                             struct bt_mesh_msg_ctx *ctx,
                             struct net_buf_simple *buf)
{
    struct net_buf_simple *sdu = NULL;
    u16_t company_id;

    company_id = net_buf_simple_pull_le16(buf);

    BT_DBG("company_id 0x%04x", company_id);

    sdu = bt_mesh_alloc_buf(MIN(BLE_MESH_TX_SDU_MAX, HEALTH_FAULT_MAX_LEN));
    if (!sdu) {
        BT_ERR("%s, Failed to allocate memory", __func__);
        return;
    }

    health_get_registered(model, company_id, sdu);

    if (bt_mesh_model_send(model, ctx, sdu, NULL, NULL)) {
        BT_ERR("%s, Unable to send Health Current Status", __func__);
    }

    bt_mesh_free_buf(sdu);
    return;
}

static void health_fault_clear_unrel(struct bt_mesh_model *model,
                                     struct bt_mesh_msg_ctx *ctx,
                                     struct net_buf_simple *buf)
{
    struct bt_mesh_health_srv *srv = model->user_data;
    u16_t company_id;

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    company_id = net_buf_simple_pull_le16(buf);

    BT_DBG("company_id 0x%04x", company_id);

    if (srv->cb && srv->cb->fault_clear) {
        srv->cb->fault_clear(model, company_id);
    }
}

static void health_fault_clear(struct bt_mesh_model *model,
                               struct bt_mesh_msg_ctx *ctx,
                               struct net_buf_simple *buf)
{
    struct bt_mesh_health_srv *srv = model->user_data;
    struct net_buf_simple *sdu = NULL;
    u16_t company_id;

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    company_id = net_buf_simple_pull_le16(buf);

    BT_DBG("company_id 0x%04x", company_id);

    if (srv->cb && srv->cb->fault_clear) {
        srv->cb->fault_clear(model, company_id);
    }

    sdu = bt_mesh_alloc_buf(MIN(BLE_MESH_TX_SDU_MAX, HEALTH_FAULT_MAX_LEN));
    if (!sdu) {
        BT_ERR("%s, Failed to allocate memory", __func__);
        return;
    }

    health_get_registered(model, company_id, sdu);

    if (bt_mesh_model_send(model, ctx, sdu, NULL, NULL)) {
        BT_ERR("%s, Unable to send Health Current Status", __func__);
    }

    bt_mesh_free_buf(sdu);
    return;
}

static void health_fault_test_unrel(struct bt_mesh_model *model,
                                    struct bt_mesh_msg_ctx *ctx,
                                    struct net_buf_simple *buf)
{
    struct bt_mesh_health_srv *srv = model->user_data;
    u16_t company_id;
    u8_t test_id;

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    test_id = net_buf_simple_pull_u8(buf);
    company_id = net_buf_simple_pull_le16(buf);

    BT_DBG("test 0x%02x company 0x%04x", test_id, company_id);

    if (srv->cb && srv->cb->fault_test) {
        srv->cb->fault_test(model, test_id, company_id);
    }
}

static void health_fault_test(struct bt_mesh_model *model,
                              struct bt_mesh_msg_ctx *ctx,
                              struct net_buf_simple *buf)
{
    struct bt_mesh_health_srv *srv = model->user_data;
    struct net_buf_simple *sdu = NULL;
    u16_t company_id;
    u8_t test_id;

    BT_DBG("%s", __func__);

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    test_id = net_buf_simple_pull_u8(buf);
    company_id = net_buf_simple_pull_le16(buf);

    BT_DBG("test 0x%02x company 0x%04x", test_id, company_id);

    if (srv->cb && srv->cb->fault_test) {
        int err;

        err = srv->cb->fault_test(model, test_id, company_id);
        if (err) {
            BT_WARN("Running fault test failed with err %d", err);
            return;
        }
    }

    sdu = bt_mesh_alloc_buf(MIN(BLE_MESH_TX_SDU_MAX, HEALTH_FAULT_MAX_LEN));
    if (!sdu) {
        BT_ERR("%s, Failed to allocate memory", __func__);
        return;
    }

    health_get_registered(model, company_id, sdu);

    if (bt_mesh_model_send(model, ctx, sdu, NULL, NULL)) {
        BT_ERR("%s, Unable to send Health Current Status", __func__);
    }

    bt_mesh_free_buf(sdu);
    return;
}

static void send_attention_status(struct bt_mesh_model *model,
                                  struct bt_mesh_msg_ctx *ctx)
{
    /* Needed size: opcode (2 bytes) + msg + MIC */
    NET_BUF_SIMPLE_DEFINE(msg, 2 + 1 + 4);
    struct bt_mesh_health_srv *srv = model->user_data;
    u8_t time;

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    time = k_delayed_work_remaining_get(&srv->attn_timer) / 1000;
    BT_DBG("%u second%s", time, (time == 1U) ? "" : "s");

    bt_mesh_model_msg_init(&msg, OP_ATTENTION_STATUS);

    net_buf_simple_add_u8(&msg, time);

    if (bt_mesh_model_send(model, ctx, &msg, NULL, NULL)) {
        BT_ERR("%s, Unable to send Health Attention Status", __func__);
    }
}

static void attention_get(struct bt_mesh_model *model,
                          struct bt_mesh_msg_ctx *ctx,
                          struct net_buf_simple *buf)
{
    BT_DBG("%s", __func__);

    send_attention_status(model, ctx);
}

static void attention_set_unrel(struct bt_mesh_model *model,
                                struct bt_mesh_msg_ctx *ctx,
                                struct net_buf_simple *buf)
{
    u8_t time;

    time = net_buf_simple_pull_u8(buf);

    BT_DBG("%u second%s", time, (time == 1U) ? "" : "s");

    bt_mesh_attention(model, time);
}

static void attention_set(struct bt_mesh_model *model,
                          struct bt_mesh_msg_ctx *ctx,
                          struct net_buf_simple *buf)
{
    BT_DBG("%s", __func__);

    attention_set_unrel(model, ctx, buf);

    send_attention_status(model, ctx);
}

static void send_health_period_status(struct bt_mesh_model *model,
                                      struct bt_mesh_msg_ctx *ctx)
{
    /* Needed size: opcode (2 bytes) + msg + MIC */
    NET_BUF_SIMPLE_DEFINE(msg, 2 + 1 + 4);

    bt_mesh_model_msg_init(&msg, OP_HEALTH_PERIOD_STATUS);

    net_buf_simple_add_u8(&msg, model->pub->period_div);

    if (bt_mesh_model_send(model, ctx, &msg, NULL, NULL)) {
        BT_ERR("%s, Unable to send Health Period Status", __func__);
    }
}

static void health_period_get(struct bt_mesh_model *model,
                              struct bt_mesh_msg_ctx *ctx,
                              struct net_buf_simple *buf)
{
    BT_DBG("%s", __func__);

    send_health_period_status(model, ctx);
}

static void health_period_set_unrel(struct bt_mesh_model *model,
                                    struct bt_mesh_msg_ctx *ctx,
                                    struct net_buf_simple *buf)
{
    u8_t period;

    period = net_buf_simple_pull_u8(buf);
    if (period > 15) {
        BT_WARN("%s, Prohibited period value %u", __func__, period);
        return;
    }

    BT_DBG("period %u", period);

    model->pub->period_div = period;
}

static void health_period_set(struct bt_mesh_model *model,
                              struct bt_mesh_msg_ctx *ctx,
                              struct net_buf_simple *buf)
{
    BT_DBG("%s", __func__);

    health_period_set_unrel(model, ctx, buf);

    send_health_period_status(model, ctx);
}

const struct bt_mesh_model_op bt_mesh_health_srv_op[] = {
    { OP_HEALTH_FAULT_GET,         2,   health_fault_get },
    { OP_HEALTH_FAULT_CLEAR,       2,   health_fault_clear },
    { OP_HEALTH_FAULT_CLEAR_UNREL, 2,   health_fault_clear_unrel },
    { OP_HEALTH_FAULT_TEST,        3,   health_fault_test },
    { OP_HEALTH_FAULT_TEST_UNREL,  3,   health_fault_test_unrel },
    { OP_HEALTH_PERIOD_GET,        0,   health_period_get },
    { OP_HEALTH_PERIOD_SET,        1,   health_period_set },
    { OP_HEALTH_PERIOD_SET_UNREL,  1,   health_period_set_unrel },
    { OP_ATTENTION_GET,            0,   attention_get },
    { OP_ATTENTION_SET,            1,   attention_set },
    { OP_ATTENTION_SET_UNREL,      1,   attention_set_unrel },
    BLE_MESH_MODEL_OP_END,
};

static int health_pub_update(struct bt_mesh_model *mod)
{
    struct bt_mesh_model_pub *pub = mod->pub;
    size_t count;

    BT_DBG("%s", __func__);

    count = health_get_current(mod, pub->msg);
    if (count) {
        pub->fast_period = 1U;
    } else {
        pub->fast_period = 0U;
    }

    return 0;
}

int bt_mesh_fault_update(struct bt_mesh_elem *elem)
{
    struct bt_mesh_model *mod;

    mod = bt_mesh_model_find(elem, BLE_MESH_MODEL_ID_HEALTH_SRV);
    if (!mod) {
        BT_ERR("%s, Health Server does not exist", __func__);
        return -EINVAL;
    }

    if (!mod->pub) {
        BT_ERR("%s, Health Server has no publication support", __func__);
        return -EIO;
    }

    /* Let periodic publishing, if enabled, take care of sending the
     * Health Current Status.
     */
    if (bt_mesh_model_pub_period_get(mod)) {
        return 0;
    }

    health_pub_update(mod);

    return bt_mesh_model_publish(mod);
}

static void attention_off(struct k_work *work)
{
    struct bt_mesh_health_srv *srv = CONTAINER_OF(work,
                                     struct bt_mesh_health_srv,
                                     attn_timer.work);
    BT_DBG("%s", __func__);

    if (!srv) {
        BT_ERR("%s, No Health Server context provided", __func__);
        return;
    }

    if (srv->cb && srv->cb->attn_off) {
        srv->cb->attn_off(srv->model);
    }
}

int bt_mesh_health_srv_init(struct bt_mesh_model *model, bool primary)
{
    struct bt_mesh_health_srv *srv = model->user_data;

    if (!srv) {
        if (!primary) {
            return 0;
        }

        BT_ERR("%s, No Health Server context provided", __func__);
        return -EINVAL;
    }

    if (!model->pub) {
        BT_ERR("%s, Health Server has no publication support", __func__);
        return -EINVAL;
    }

    model->pub->update = health_pub_update;

    k_delayed_work_init(&srv->attn_timer, attention_off);

    srv->model = model;

    if (primary) {
        health_srv = srv;
    }

    return 0;
}

void bt_mesh_attention(struct bt_mesh_model *model, u8_t time)
{
    struct bt_mesh_health_srv *srv;

    if (!model) {
        srv = health_srv;
        if (!srv) {
            BT_WARN("%s, No Health Server context provided", __func__);
            return;
        }

        model = srv->model;
    } else {
        srv = model->user_data;
        if (!srv) {
            BT_WARN("%s, No Health Server context provided", __func__);
            return;
        }
    }

    if (time) {
        if (srv->cb && srv->cb->attn_on) {
            srv->cb->attn_on(model);
        }

        k_delayed_work_submit(&srv->attn_timer, time * 1000U);
    } else {
        k_delayed_work_cancel(&srv->attn_timer);

        if (srv->cb && srv->cb->attn_off) {
            srv->cb->attn_off(model);
        }
    }
}