227 lines
10 KiB
C
227 lines
10 KiB
C
#include "esp_zb_light.h"
|
|
#include "esp_check.h"
|
|
#include "esp_log.h"
|
|
#include "nvs_flash.h"
|
|
#include "freertos/FreeRTOS.h"
|
|
#include "freertos/task.h"
|
|
#include "ha/esp_zigbee_ha_standard.h"
|
|
|
|
#if !defined CONFIG_ZB_ZCZR
|
|
#error Define ZB_ZCZR in idf.py menuconfig to compile light (Router) source code.
|
|
#endif
|
|
|
|
static const char* TAG = "ESP_ZB_COLOR_DIMM_LIGHT";
|
|
/********************* Define functions **************************/
|
|
static esp_err_t deferred_driver_init(void) {
|
|
static bool is_inited = false;
|
|
if (!is_inited) {
|
|
// light_driver_init(LIGHT_DEFAULT_OFF);
|
|
is_inited = true;
|
|
}
|
|
return is_inited ? ESP_OK : ESP_FAIL;
|
|
}
|
|
|
|
static void bdb_start_top_level_commissioning_cb(const uint8_t mode_mask) {
|
|
ESP_RETURN_ON_FALSE(esp_zb_bdb_start_top_level_commissioning(mode_mask) == ESP_OK, , TAG, "Failed to start Zigbee commissioning");
|
|
}
|
|
|
|
void esp_zb_app_signal_handler(const esp_zb_app_signal_t* signal_struct) {
|
|
uint32_t* p_sg_p = signal_struct->p_app_signal;
|
|
const esp_err_t err_status = signal_struct->esp_err_status;
|
|
const esp_zb_app_signal_type_t sig_type = *p_sg_p;
|
|
|
|
switch (sig_type) {
|
|
|
|
case ESP_ZB_ZDO_SIGNAL_SKIP_STARTUP:
|
|
ESP_LOGI(TAG, "Initialize Zigbee stack");
|
|
esp_zb_bdb_start_top_level_commissioning(ESP_ZB_BDB_MODE_INITIALIZATION);
|
|
break;
|
|
|
|
case ESP_ZB_BDB_SIGNAL_DEVICE_FIRST_START:
|
|
case ESP_ZB_BDB_SIGNAL_DEVICE_REBOOT:
|
|
if (err_status == ESP_OK) {
|
|
ESP_LOGI(TAG, "Deferred driver initialization %s", deferred_driver_init() ? "failed" : "successful");
|
|
ESP_LOGI(TAG, "Device started up in%s factory-reset mode", esp_zb_bdb_is_factory_new() ? "" : " non");
|
|
if (esp_zb_bdb_is_factory_new()) {
|
|
ESP_LOGI(TAG, "Start network steering");
|
|
esp_zb_bdb_start_top_level_commissioning(ESP_ZB_BDB_MODE_NETWORK_STEERING);
|
|
}
|
|
else { ESP_LOGI(TAG, "Device rebooted"); }
|
|
}
|
|
else {
|
|
ESP_LOGW(TAG, "%s failed with status: %s, retrying", esp_zb_zdo_signal_to_string(sig_type),
|
|
esp_err_to_name(err_status));
|
|
esp_zb_scheduler_alarm((esp_zb_callback_t)bdb_start_top_level_commissioning_cb,
|
|
ESP_ZB_BDB_MODE_INITIALIZATION, 1000);
|
|
}
|
|
break;
|
|
|
|
case ESP_ZB_BDB_SIGNAL_STEERING:
|
|
if (err_status == ESP_OK) {
|
|
esp_zb_ieee_addr_t extended_pan_id;
|
|
esp_zb_get_extended_pan_id(extended_pan_id);
|
|
ESP_LOGI(TAG, "Joined network successfully (Extended PAN ID: %02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x, PAN ID: 0x%04hx, Channel:%d, Short Address: 0x%04hx)",
|
|
extended_pan_id[7], extended_pan_id[6], extended_pan_id[5], extended_pan_id[4],
|
|
extended_pan_id[3], extended_pan_id[2], extended_pan_id[1], extended_pan_id[0],
|
|
esp_zb_get_pan_id(), esp_zb_get_current_channel(), esp_zb_get_short_address());
|
|
}
|
|
else {
|
|
ESP_LOGI(TAG, "Network steering was not successful (status: %s)", esp_err_to_name(err_status));
|
|
esp_zb_scheduler_alarm((esp_zb_callback_t)bdb_start_top_level_commissioning_cb, ESP_ZB_BDB_MODE_NETWORK_STEERING, 1000);
|
|
}
|
|
break;
|
|
|
|
case ESP_ZB_NWK_SIGNAL_PERMIT_JOIN_STATUS:
|
|
if (err_status == ESP_OK) {
|
|
if (*(uint8_t*)esp_zb_app_signal_get_params(p_sg_p)) {
|
|
ESP_LOGI(TAG, "Network(0x%04hx) is open for %d seconds", esp_zb_get_pan_id(), *(uint8_t *)esp_zb_app_signal_get_params(p_sg_p));
|
|
}
|
|
else { ESP_LOGW(TAG, "Network(0x%04hx) closed, devices joining not allowed.", esp_zb_get_pan_id()); }
|
|
}
|
|
break;
|
|
|
|
default:
|
|
ESP_LOGI(TAG, "ZDO signal: %s (0x%x), status: %s", esp_zb_zdo_signal_to_string(sig_type), sig_type, esp_err_to_name(err_status));
|
|
break;
|
|
}
|
|
}
|
|
|
|
static esp_err_t zb_attribute_handler(const esp_zb_zcl_set_attr_value_message_t* message) {
|
|
ESP_RETURN_ON_FALSE(message, ESP_FAIL, TAG, "Empty message");
|
|
ESP_RETURN_ON_FALSE(message->info.status == ESP_ZB_ZCL_STATUS_SUCCESS, ESP_ERR_INVALID_ARG, TAG, "Received message: error status(%d)", message->info.status);
|
|
ESP_LOGI(TAG, "Received message: endpoint(%d), cluster(0x%x), attribute(0x%x), data size(%d)", message->info.dst_endpoint, message->info.cluster, message->attribute.id, message->attribute.data.size);
|
|
|
|
if (message->info.dst_endpoint == HA_COLOR_DIMMABLE_LIGHT_ENDPOINT) {
|
|
switch (message->info.cluster) {
|
|
|
|
case ESP_ZB_ZCL_CLUSTER_ID_ON_OFF:
|
|
if (message->attribute.id == ESP_ZB_ZCL_ATTR_ON_OFF_ON_OFF_ID && message->attribute.data.type == ESP_ZB_ZCL_ATTR_TYPE_BOOL) {
|
|
bool light_state = false;
|
|
light_state = message->attribute.data.value ? *(bool*)message->attribute.data.value : light_state;
|
|
ESP_LOGI(TAG, "Light sets to %s", light_state ? "On" : "Off");
|
|
// light_driver_set_power(light_state);
|
|
}
|
|
else { ESP_LOGW(TAG, "On/Off cluster data: attribute(0x%x), type(0x%x)", message->attribute.id, message->attribute.data.type); }
|
|
break;
|
|
|
|
case ESP_ZB_ZCL_CLUSTER_ID_COLOR_CONTROL:
|
|
ESP_LOGE(TAG, "we dont fuck with no color control in here");
|
|
break;
|
|
|
|
case ESP_ZB_ZCL_CLUSTER_ID_LEVEL_CONTROL:
|
|
if (message->attribute.id == ESP_ZB_ZCL_ATTR_LEVEL_CONTROL_CURRENT_LEVEL_ID && message->attribute.data.type == ESP_ZB_ZCL_ATTR_TYPE_U8) {
|
|
uint8_t light_level = 0;
|
|
light_level = message->attribute.data.value ? *(uint8_t*)message->attribute.data.value : light_level;
|
|
// light_driver_set_level((uint8_t)light_level);
|
|
ESP_LOGI(TAG, "Light level changes to %d", light_level);
|
|
}
|
|
else { ESP_LOGW(TAG, "Level Control cluster data: attribute(0x%x), type(0x%x)", message->attribute.id, message->attribute.data.type); }
|
|
break;
|
|
|
|
default:
|
|
ESP_LOGI(TAG, "Message data: cluster(0x%x), attribute(0x%x) ", message->info.cluster, message->attribute.id);
|
|
}
|
|
}
|
|
return ESP_OK;
|
|
}
|
|
|
|
static esp_err_t zb_action_handler(esp_zb_core_action_callback_id_t callback_id, const void* message) {
|
|
esp_err_t ret = ESP_OK;
|
|
switch (callback_id) {
|
|
|
|
case ESP_ZB_CORE_SET_ATTR_VALUE_CB_ID:
|
|
ret = zb_attribute_handler((esp_zb_zcl_set_attr_value_message_t*)message);
|
|
break;
|
|
|
|
default:
|
|
ESP_LOGW(TAG, "Receive Zigbee action(0x%x) callback", callback_id);
|
|
break;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static void esp_zb_task(void* pvParameters) {
|
|
/* initialize Zigbee stack */
|
|
esp_zb_cfg_t zb_nwk_cfg = ESP_ZB_ZR_CONFIG();
|
|
esp_zb_init(&zb_nwk_cfg);
|
|
|
|
const zcl_basic_manufacturer_info_t info = {
|
|
.manufacturer_name = ESP_MANUFACTURER_NAME,
|
|
.model_identifier = ESP_MODEL_IDENTIFIER,
|
|
};
|
|
|
|
esp_zb_cluster_list_t *cluster_list = esp_zb_zcl_cluster_list_create();
|
|
|
|
esp_zb_basic_cluster_cfg_t basic_cfg = {
|
|
.zcl_version = ESP_ZB_ZCL_BASIC_ZCL_VERSION_DEFAULT_VALUE,
|
|
.power_source = ESP_ZB_ZCL_BASIC_POWER_SOURCE_DC_SOURCE,
|
|
};
|
|
|
|
esp_zb_identify_cluster_cfg_t identify_cfg = {
|
|
.identify_time = ESP_ZB_ZCL_IDENTIFY_IDENTIFY_TIME_DEFAULT_VALUE
|
|
};
|
|
|
|
esp_zb_groups_cluster_cfg_t group_cfg = {
|
|
.groups_name_support_id = ESP_ZB_ZCL_GROUPS_NAME_SUPPORT_DEFAULT_VALUE,
|
|
};
|
|
|
|
esp_zb_scenes_cluster_cfg_t scenes_cfg = {
|
|
.scenes_count = ESP_ZB_ZCL_SCENES_SCENE_COUNT_DEFAULT_VALUE,
|
|
.current_scene = ESP_ZB_ZCL_SCENES_CURRENT_SCENE_DEFAULT_VALUE,
|
|
.current_group = ESP_ZB_ZCL_SCENES_CURRENT_GROUP_DEFAULT_VALUE,
|
|
.scene_valid = ESP_ZB_ZCL_SCENES_SCENE_VALID_DEFAULT_VALUE,
|
|
.name_support = ESP_ZB_ZCL_SCENES_NAME_SUPPORT_DEFAULT_VALUE,
|
|
};
|
|
|
|
esp_zb_on_off_cluster_cfg_t on_off_cfg = {
|
|
.on_off = ESP_ZB_ZCL_ON_OFF_ON_OFF_DEFAULT_VALUE
|
|
};
|
|
|
|
esp_zb_level_cluster_cfg_t level_cfg = {
|
|
.current_level = ESP_ZB_ZCL_LEVEL_CONTROL_CURRENT_LEVEL_DEFAULT_VALUE,
|
|
};
|
|
|
|
esp_zb_attribute_list_t *basic_cluster = esp_zb_basic_cluster_create(&basic_cfg);
|
|
esp_zb_attribute_list_t *identify_cluster = esp_zb_identify_cluster_create(&identify_cfg);
|
|
esp_zb_attribute_list_t *groups_cluster = esp_zb_groups_cluster_create(&group_cfg);
|
|
esp_zb_attribute_list_t *scenes_cluster = esp_zb_scenes_cluster_create(&scenes_cfg);
|
|
esp_zb_attribute_list_t *on_off_cluster = esp_zb_on_off_cluster_create(&on_off_cfg);
|
|
esp_zb_attribute_list_t *level_cluster = esp_zb_level_cluster_create(&level_cfg);
|
|
|
|
esp_zb_cluster_list_add_basic_cluster(cluster_list, basic_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE);
|
|
esp_zb_cluster_list_add_basic_cluster(cluster_list, identify_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE);
|
|
esp_zb_cluster_list_add_basic_cluster(cluster_list, groups_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE);
|
|
esp_zb_cluster_list_add_basic_cluster(cluster_list, scenes_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE);
|
|
esp_zb_cluster_list_add_basic_cluster(cluster_list, on_off_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE);
|
|
esp_zb_cluster_list_add_basic_cluster(cluster_list, level_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE);
|
|
|
|
esp_zb_ep_list_t *ep_list = esp_zb_ep_list_create();
|
|
|
|
const esp_zb_endpoint_config_t ep_cfg = {
|
|
.endpoint = HA_COLOR_DIMMABLE_LIGHT_ENDPOINT,
|
|
.app_device_id = 0x0104,
|
|
.app_profile_id = 0x0101,
|
|
.app_device_version = 0,
|
|
};
|
|
|
|
esp_zb_ep_list_add_ep(ep_list, cluster_list, ep_cfg);
|
|
|
|
|
|
esp_zcl_utility_add_ep_basic_manufacturer_info(ep_list, HA_COLOR_DIMMABLE_LIGHT_ENDPOINT, &info);
|
|
esp_zb_device_register(ep_list);
|
|
esp_zb_core_action_handler_register(zb_action_handler);
|
|
esp_zb_set_primary_network_channel_set(ESP_ZB_TRANSCEIVER_ALL_CHANNELS_MASK);
|
|
ESP_ERROR_CHECK(esp_zb_start(false));
|
|
esp_zb_stack_main_loop(); // blocking, holds the task here
|
|
}
|
|
|
|
void app_main(void) {
|
|
esp_zb_platform_config_t config = {
|
|
.radio_config = ESP_ZB_DEFAULT_RADIO_CONFIG(),
|
|
.host_config = ESP_ZB_DEFAULT_HOST_CONFIG(),
|
|
};
|
|
ESP_ERROR_CHECK(nvs_flash_init());
|
|
ESP_ERROR_CHECK(esp_zb_platform_config(&config));
|
|
xTaskCreate(esp_zb_task, "Zigbee_main", 4096, NULL, 5, NULL);
|
|
}
|