/*
 * SPDX-FileCopyrightText: 2017-2024 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#ifndef _BTC_BLE_MESH_PROV_H_
#define _BTC_BLE_MESH_PROV_H_

#include "btc/btc_manage.h"
#include "mesh/byteorder.h"
#include "mesh/config.h"
#include "mesh/main.h"
#include "fast_prov.h"
#include "esp_ble_mesh_defs.h"

#ifdef __cplusplus
extern "C" {
#endif

typedef enum {
    BTC_BLE_MESH_ACT_MESH_INIT = 0,
    BTC_BLE_MESH_ACT_PROV_ENABLE,
    BTC_BLE_MESH_ACT_PROV_DISABLE,
    BTC_BLE_MESH_ACT_NODE_RESET,
    BTC_BLE_MESH_ACT_SET_OOB_PUB_KEY,
    BTC_BLE_MESH_ACT_INPUT_NUMBER,
    BTC_BLE_MESH_ACT_INPUT_STRING,
    BTC_BLE_MESH_ACT_SET_DEVICE_NAME,
    BTC_BLE_MESH_ACT_PROXY_IDENTITY_ENABLE,
    BTC_BLE_MESH_ACT_PRIVATE_PROXY_IDENTITY_ENABLE,
    BTC_BLE_MESH_ACT_PRIVATE_PROXY_IDENTITY_DISABLE,
    BTC_BLE_MESH_ACT_PROXY_GATT_ENABLE,
    BTC_BLE_MESH_ACT_PROXY_GATT_DISABLE,
    BTC_BLE_MESH_ACT_NODE_ADD_LOCAL_NET_KEY,
    BTC_BLE_MESH_ACT_NODE_ADD_LOCAL_APP_KEY,
    BTC_BLE_MESH_ACT_NODE_BIND_APP_KEY_TO_MODEL,
    BTC_BLE_MESH_ACT_PROVISIONER_READ_OOB_PUB_KEY,
    BTC_BLE_MESH_ACT_PROVISIONER_INPUT_STR,
    BTC_BLE_MESH_ACT_PROVISIONER_INPUT_NUM,
    BTC_BLE_MESH_ACT_PROVISIONER_ENABLE,
    BTC_BLE_MESH_ACT_PROVISIONER_DISABLE,
    BTC_BLE_MESH_ACT_PROVISIONER_DEV_ADD,
    BTC_BLE_MESH_ACT_PROVISIONER_PROV_DEV_WITH_ADDR,
    BTC_BLE_MESH_ACT_PROVISIONER_DEV_DEL,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_DEV_UUID_MATCH,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_PROV_DATA_INFO,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_STATIC_OOB_VAL,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_PRIMARY_ELEM_ADDR,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_NODE_NAME,
    BTC_BLE_MESH_ACT_PROVISIONER_ADD_LOCAL_APP_KEY,
    BTC_BLE_MESH_ACT_PROVISIONER_UPDATE_LOCAL_APP_KEY,
    BTC_BLE_MESH_ACT_PROVISIONER_BIND_LOCAL_MOD_APP,
    BTC_BLE_MESH_ACT_PROVISIONER_ADD_LOCAL_NET_KEY,
    BTC_BLE_MESH_ACT_PROVISIONER_UPDATE_LOCAL_NET_KEY,
    BTC_BLE_MESH_ACT_PROVISIONER_STORE_NODE_COMP_DATA,
    BTC_BLE_MESH_ACT_PROVISIONER_DELETE_NODE_WITH_UUID,
    BTC_BLE_MESH_ACT_PROVISIONER_DELETE_NODE_WITH_ADDR,
    BTC_BLE_MESH_ACT_PROVISIONER_ENABLE_HEARTBEAT_RECV,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_HEARTBEAT_FILTER_TYPE,
    BTC_BLE_MESH_ACT_PROVISIONER_SET_HEARTBEAT_FILTER_INFO,
    BTC_BLE_MESH_ACT_PROVISIONER_DIRECT_ERASE_SETTINGS,
    BTC_BLE_MESH_ACT_PROVISIONER_OPEN_SETTINGS_WITH_INDEX,
    BTC_BLE_MESH_ACT_PROVISIONER_OPEN_SETTINGS_WITH_UID,
    BTC_BLE_MESH_ACT_PROVISIONER_CLOSE_SETTINGS_WITH_INDEX,
    BTC_BLE_MESH_ACT_PROVISIONER_CLOSE_SETTINGS_WITH_UID,
    BTC_BLE_MESH_ACT_PROVISIONER_DELETE_SETTINGS_WITH_INDEX,
    BTC_BLE_MESH_ACT_PROVISIONER_DELETE_SETTINGS_WITH_UID,
    BTC_BLE_MESH_ACT_PROVISIONER_SEND_PROV_RECORDS_GET,
    BTC_BLE_MESH_ACT_PROVISIONER_SEND_PROV_RECORD_REQUEST,
    BTC_BLE_MESH_ACT_PROVISIONER_SEND_PROV_INVITE,
    BTC_BLE_MESH_ACT_PROVISIONER_SEND_LINK_CLOSE,
    BTC_BLE_MESH_ACT_SET_FAST_PROV_INFO,
    BTC_BLE_MESH_ACT_SET_FAST_PROV_ACTION,
    BTC_BLE_MESH_ACT_LPN_ENABLE,
    BTC_BLE_MESH_ACT_LPN_DISABLE,
    BTC_BLE_MESH_ACT_LPN_POLL,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_CONNECT,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_DISCONNECT,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_SET_FILTER_TYPE,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_ADD_FILTER_ADDR,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_REMOVE_FILTER_ADDR,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_DIRECTED_PROXY_SET,
    BTC_BLE_MESH_ACT_PROXY_CLIENT_SEND_SOLIC_PDU,
    BTC_BLE_MESH_ACT_MODEL_SUBSCRIBE_GROUP_ADDR,
    BTC_BLE_MESH_ACT_MODEL_UNSUBSCRIBE_GROUP_ADDR,
    BTC_BLE_MESH_ACT_DEINIT_MESH,
} btc_ble_mesh_prov_act_t;

typedef enum {
    BTC_BLE_MESH_ACT_MODEL_PUBLISH,
    BTC_BLE_MESH_ACT_SERVER_MODEL_SEND,
    BTC_BLE_MESH_ACT_CLIENT_MODEL_SEND,
    BTC_BLE_MESH_ACT_SERVER_MODEL_UPDATE_STATE,
} btc_ble_mesh_model_act_t;

typedef union {
    struct ble_mesh_init_args {
        esp_ble_mesh_prov_t *prov;
        esp_ble_mesh_comp_t *comp;
        SemaphoreHandle_t semaphore;
    } mesh_init;
    struct ble_mesh_node_prov_enable_args {
        esp_ble_mesh_prov_bearer_t bearers;
    } node_prov_enable;
    struct ble_mesh_node_prov_disable_args {
        esp_ble_mesh_prov_bearer_t bearers;
    } node_prov_disable;
    struct ble_mesh_set_oob_pub_key_args {
        uint8_t pub_key_x[32];
        uint8_t pub_key_y[32];
        uint8_t private_key[32];
    } set_oob_pub_key;
    struct ble_mesh_node_input_num_args {
        uint32_t number;
    } input_number;
    struct ble_mesh_node_input_str_args {
        char string[8];
    } input_string;
    struct ble_mesh_set_device_name_args {
        char name[ESP_BLE_MESH_DEVICE_NAME_MAX_LEN + 1];
    } set_device_name;
    struct ble_mesh_node_add_local_net_key_args {
        uint8_t  net_key[16];
        uint16_t net_idx;
    } node_add_local_net_key;
    struct ble_mesh_node_add_local_app_key_args {
        uint8_t  app_key[16];
        uint16_t net_idx;
        uint16_t app_idx;
    } node_add_local_app_key;
    struct ble_mesh_node_bind_local_mod_app_args {
        uint16_t element_addr;
        uint16_t company_id;
        uint16_t model_id;
        uint16_t app_idx;
    } node_local_mod_app_bind;
    struct ble_mesh_provisioner_read_oob_pub_key_args {
        uint8_t link_idx;
        uint8_t pub_key_x[32];
        uint8_t pub_key_y[32];
    } provisioner_read_oob_pub_key;
    struct ble_mesh_provisioner_input_str_args {
        char string[8];
        uint8_t link_idx;
    } provisioner_input_str;
    struct ble_mesh_provisioner_input_num_args {
        uint32_t number;
        uint8_t  link_idx;
    } provisioner_input_num;
    struct ble_mesh_provisioner_enable_args {
        esp_ble_mesh_prov_bearer_t bearers;
    } provisioner_enable;
    struct ble_mesh_provisioner_disable_args {
        esp_ble_mesh_prov_bearer_t bearers;
    } provisioner_disable;
    struct ble_mesh_provisioner_dev_add_args {
        esp_ble_mesh_unprov_dev_add_t add_dev;
        esp_ble_mesh_dev_add_flag_t   flags;
    } provisioner_dev_add;
    struct ble_mesh_provisioner_prov_dev_with_addr_args {
        uint8_t uuid[16];
        esp_ble_mesh_bd_addr_t addr;
        esp_ble_mesh_addr_type_t addr_type;
        esp_ble_mesh_prov_bearer_t bearer;
        uint16_t oob_info;
        uint16_t unicast_addr;
    } provisioner_prov_dev_with_addr;
    struct ble_mesh_provisioner_dev_del_args {
        esp_ble_mesh_device_delete_t del_dev;
    } provisioner_dev_del;
    struct ble_mesh_provisioner_set_dev_uuid_match_args {
        uint8_t offset;
        uint8_t match_len;
        uint8_t match_val[16];
        bool prov_after_match;
    } set_dev_uuid_match;
    struct ble_mesh_provisioner_set_prov_net_idx_args {
        esp_ble_mesh_prov_data_info_t prov_data;
    } set_prov_data_info;
    struct ble_mesh_provisioner_set_static_oob_val_args {
        uint8_t value[16];
        uint8_t length;
    } set_static_oob_val;
    struct ble_mesh_provisioner_set_primary_elem_addr_args {
        uint16_t addr;
    } set_primary_elem_addr;
    struct ble_mesh_provisioner_set_node_name_args {
        uint16_t index;
        char name[ESP_BLE_MESH_NODE_NAME_MAX_LEN + 1];
    } set_node_name;
    struct ble_mesh_provisioner_add_local_app_key_args {
        uint8_t app_key[16];
        uint16_t net_idx;
        uint16_t app_idx;
    } add_local_app_key;
    struct ble_mesh_provisioner_update_local_app_key_args {
        uint8_t app_key[16];
        uint16_t net_idx;
        uint16_t app_idx;
    } update_local_app_key;
    struct ble_mesh_provisioner_bind_local_mod_app_args {
        uint16_t elem_addr;
        uint16_t model_id;
        uint16_t cid;
        uint16_t app_idx;
    } local_mod_app_bind;
    struct ble_mesh_provisioner_add_local_net_key_args {
        uint8_t net_key[16];
        uint16_t net_idx;
    } add_local_net_key;
    struct ble_mesh_provisioner_update_local_net_key_args {
        uint8_t net_key[16];
        uint16_t net_idx;
    } update_local_net_key;
    struct ble_mesh_provisioner_store_node_comp_data_args {
        uint16_t unicast_addr;
        uint16_t length;
        uint8_t *data;
    } store_node_comp_data;
    struct ble_mesh_provisioner_delete_node_with_uuid_args {
        uint8_t uuid[16];
    } delete_node_with_uuid;
    struct ble_mesh_provisioner_delete_node_with_addr_args {
        uint16_t unicast_addr;
    } delete_node_with_addr;
    struct {
        bool enable;
    } enable_heartbeat_recv;
    struct {
        uint8_t type;
    } set_heartbeat_filter_type;
    struct {
        uint8_t  op;
        uint16_t hb_src;
        uint16_t hb_dst;
    } set_heartbeat_filter_info;
    struct {
        uint8_t index;
    } open_settings_with_index;
    struct {
        char uid[ESP_BLE_MESH_SETTINGS_UID_SIZE + 1];
    } open_settings_with_uid;
    struct {
        uint8_t index;
        bool erase;
    } close_settings_with_index;
    struct {
        char uid[ESP_BLE_MESH_SETTINGS_UID_SIZE + 1];
        bool erase;
    } close_settings_with_uid;
    struct {
        uint8_t index;
    } delete_settings_with_index;
    struct {
        char uid[ESP_BLE_MESH_SETTINGS_UID_SIZE + 1];
    } delete_settings_with_uid;
    struct {
        uint16_t link_idx;
    } send_prov_records_get;
    struct {
        uint16_t link_idx;
        uint16_t record_id;
        uint16_t frag_offset;
        uint16_t max_size;
    } send_prov_record_req;
    struct {
        uint16_t link_idx;
    } send_prov_invite;
    struct {
        uint16_t link_idx;
    } send_link_close;
    struct ble_mesh_set_fast_prov_info_args {
        uint16_t unicast_min;
        uint16_t unicast_max;
        uint16_t net_idx;
        uint8_t  flags;
        uint32_t iv_index;
        uint8_t  offset;
        uint8_t  match_len;
        uint8_t  match_val[16];
    } set_fast_prov_info;
    struct ble_mesh_set_fast_prov_action_args {
        uint8_t action;
    } set_fast_prov_action;
    struct ble_mesh_lpn_enable_args {
        /* RFU */
    } lpn_enable;
    struct ble_mesh_lpn_disable_args {
        bool force;
    } lpn_disable;
    struct ble_mesh_lpn_poll_args {
        /* RFU */
    } lpn_poll;
    struct ble_mesh_proxy_client_connect_args {
        uint8_t  addr[6];
        uint8_t  addr_type;
        uint16_t net_idx;
    } proxy_client_connect;
    struct ble_mesh_proxy_client_disconnect_args {
        uint8_t conn_handle;
    } proxy_client_disconnect;
    struct ble_mesh_proxy_client_set_filter_type_args {
        uint8_t  conn_handle;
        uint16_t net_idx;
        uint8_t  filter_type;
    } proxy_client_set_filter_type;
    struct ble_mesh_proxy_client_add_filter_addr_args {
        uint8_t   conn_handle;
        uint16_t  net_idx;
        uint16_t  addr_num;
        uint16_t *addr;
    } proxy_client_add_filter_addr;
    struct ble_mesh_proxy_client_remove_filter_addr_args {
        uint8_t   conn_handle;
        uint16_t  net_idx;
        uint16_t  addr_num;
        uint16_t *addr;
    } proxy_client_remove_filter_addr;
    struct ble_mesh_proxy_client_directed_proxy_set_args {
        uint8_t  conn_handle;
        uint16_t net_idx;
        uint8_t  use_directed;
    } proxy_client_directed_proxy_set;
    struct {
        uint16_t net_idx;
        uint16_t ssrc;
        uint16_t dst;
    } proxy_client_send_solic_pdu;
    struct ble_mesh_model_sub_group_addr_args {
        uint16_t element_addr;
        uint16_t company_id;
        uint16_t model_id;
        uint16_t group_addr;
    } model_sub_group_addr;
    struct ble_mesh_model_unsub_group_addr_args {
        uint16_t element_addr;
        uint16_t company_id;
        uint16_t model_id;
        uint16_t group_addr;
    } model_unsub_group_addr;
    struct ble_mesh_deinit_args {
        esp_ble_mesh_deinit_param_t param;
        SemaphoreHandle_t semaphore;
    } mesh_deinit;
} btc_ble_mesh_prov_args_t;

typedef union {
    struct ble_mesh_model_publish_args {
        esp_ble_mesh_model_t *model;
        uint8_t device_role __attribute__((deprecated));
    } model_publish;
    struct ble_mesh_model_send_args {
        esp_ble_mesh_model_t *model;
        esp_ble_mesh_msg_ctx_t *ctx;
        uint32_t opcode;
        bool need_rsp;
        uint16_t length;
        uint8_t *data;
        uint8_t device_role __attribute__((deprecated));
        int32_t msg_timeout;
    } model_send;
    struct ble_mesh_server_model_update_state_args {
        esp_ble_mesh_model_t *model;
        esp_ble_mesh_server_state_type_t type;
        esp_ble_mesh_server_state_value_t *value;
    } model_update_state;
} btc_ble_mesh_model_args_t;

void btc_ble_mesh_prov_arg_deep_copy(btc_msg_t *msg, void *p_dest, void *p_src);

void btc_ble_mesh_prov_arg_deep_free(btc_msg_t *msg);

void btc_ble_mesh_model_arg_deep_copy(btc_msg_t *msg, void *p_dest, void *p_src);

void btc_ble_mesh_model_arg_deep_free(btc_msg_t *msg);

const uint8_t *btc_ble_mesh_node_get_local_net_key(uint16_t net_idx);

const uint8_t *btc_ble_mesh_node_get_local_app_key(uint16_t app_idx);

esp_ble_mesh_node_t *btc_ble_mesh_provisioner_get_node_with_uuid(const uint8_t uuid[16]);

esp_ble_mesh_node_t *btc_ble_mesh_provisioner_get_node_with_addr(uint16_t unicast_addr);

esp_ble_mesh_node_t *btc_ble_mesh_provisioner_get_node_with_name(const char *name);

uint16_t btc_ble_mesh_provisioner_get_prov_node_count(void);

const esp_ble_mesh_node_t **btc_ble_mesh_provisioner_get_node_table_entry(void);

int btc_ble_mesh_client_model_init(esp_ble_mesh_model_t *model);

int btc_ble_mesh_client_model_deinit(esp_ble_mesh_model_t *model);

int32_t btc_ble_mesh_model_pub_period_get(esp_ble_mesh_model_t *mod);

uint16_t btc_ble_mesh_get_primary_addr(void);

uint16_t *btc_ble_mesh_model_find_group(esp_ble_mesh_model_t *mod, uint16_t addr);

esp_ble_mesh_elem_t *btc_ble_mesh_elem_find(uint16_t addr);

uint8_t btc_ble_mesh_elem_count(void);

esp_ble_mesh_model_t *btc_ble_mesh_model_find_vnd(const esp_ble_mesh_elem_t *elem,
                                                  uint16_t company, uint16_t id);

esp_ble_mesh_model_t *btc_ble_mesh_model_find(const esp_ble_mesh_elem_t *elem, uint16_t id);

const esp_ble_mesh_comp_t *btc_ble_mesh_comp_get(void);

const char *btc_ble_mesh_provisioner_get_settings_uid(uint8_t index);

uint8_t btc_ble_mesh_provisioner_get_settings_index(const char *uid);

uint8_t btc_ble_mesh_provisioner_get_free_settings_count(void);

void btc_ble_mesh_model_call_handler(btc_msg_t *msg);
void btc_ble_mesh_model_cb_handler(btc_msg_t *msg);

void btc_ble_mesh_prov_call_handler(btc_msg_t *msg);
void btc_ble_mesh_prov_cb_handler(btc_msg_t *msg);

#if CONFIG_BLE_MESH_DF_SRV
int btc_ble_mesh_enable_directed_forwarding(uint16_t net_idx, bool directed_forwarding,
                                            bool directed_forwarding_relay);
#endif

#ifdef __cplusplus
}
#endif

#endif /* _BTC_BLE_MESH_PROV_H_ */
