// #include "ha/esp_zigbee_ha_standard.h" #include "zigbee.h" #include "esp_zigbee_core.h" #include "light_driver.h" #include #include #include static const char* TAG = "zigbee"; static char manufacturer[] = "\x05""Drath"; static char model[] = "\x15""Dimmable Zigbee Light"; static const uint8_t endpoint_no = 1; static const uint16_t transition_on_off_ms = 500; static const uint16_t transition_level_ms = 100; // looks like it gets lvl updates from HA every 100ms 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"); } esp_err_t zigbee_utility_add_ep_basic_manufacturer_info(esp_zb_ep_list_t *ep_list, const uint8_t endpoint_id, const zcl_basic_manufacturer_info_t *info) { esp_zb_cluster_list_t *cluster_list = NULL; esp_zb_attribute_list_t *basic_cluster = NULL; cluster_list = esp_zb_ep_list_get_ep(ep_list, endpoint_id); ESP_RETURN_ON_FALSE(cluster_list, ESP_ERR_INVALID_ARG, TAG, "Failed to find endpoint id: %d in list: %p", endpoint_id, ep_list); basic_cluster = esp_zb_cluster_list_get_cluster(cluster_list, ESP_ZB_ZCL_CLUSTER_ID_BASIC, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); ESP_RETURN_ON_FALSE(basic_cluster, ESP_ERR_INVALID_ARG, TAG, "Failed to find basic cluster in endpoint: %d", endpoint_id); ESP_RETURN_ON_FALSE((info && info->manufacturer_name), ESP_ERR_INVALID_ARG, TAG, "Invalid manufacturer name"); ESP_ERROR_CHECK(esp_zb_basic_cluster_add_attr(basic_cluster, ESP_ZB_ZCL_ATTR_BASIC_MANUFACTURER_NAME_ID, info->manufacturer_name)); ESP_RETURN_ON_FALSE((info && info->model_identifier), ESP_ERR_INVALID_ARG, TAG, "Invalid model identifier"); ESP_ERROR_CHECK(esp_zb_basic_cluster_add_attr(basic_cluster, ESP_ZB_ZCL_ATTR_BASIC_MODEL_IDENTIFIER_ID, info->model_identifier)); return ESP_OK; } static esp_err_t zigbee_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); if (message->info.dst_endpoint == endpoint_no) { switch (message->info.cluster) { case ESP_ZB_ZCL_CLUSTER_ID_IDENTIFY: ESP_LOGI(TAG, "Identify requested"); break; 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, transition_on_off_ms); } 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(light_level, transition_level_ms); ESP_LOGI(TAG, "Light level changes to %d", light_level); } break; default: ESP_LOGW(TAG, "Unhandled zigbee message, data: cluster(0x%x), attribute(0x%x) ", message->info.cluster, message->attribute.id); } } return ESP_OK; } static esp_err_t zigbee_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 = zigbee_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; } void zigbee_clusterfuck_init(esp_zb_cluster_list_t *cluster_list) { 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_ota_cluster_cfg_t ota_cfg = { // 0 // }; 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_attribute_list_t *ota_cluster = esp_zb_ota_cluster_create(&ota_cfg); esp_zb_cluster_list_add_basic_cluster(cluster_list, basic_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); esp_zb_cluster_list_add_identify_cluster(cluster_list, identify_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); esp_zb_cluster_list_add_groups_cluster(cluster_list, groups_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); esp_zb_cluster_list_add_scenes_cluster(cluster_list, scenes_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); esp_zb_cluster_list_add_on_off_cluster(cluster_list, on_off_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); esp_zb_cluster_list_add_level_cluster(cluster_list, level_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); // esp_zb_cluster_list_add_ota_cluster(cluster_list, ota_cluster, ESP_ZB_ZCL_CLUSTER_SERVER_ROLE); // disabling for now, not ready. maybe one day } void zigbee_endpoint_init(esp_zb_ep_list_t *ep_list, esp_zb_cluster_list_t *cluster_list) { const zcl_basic_manufacturer_info_t mfg_info = { .manufacturer_name = manufacturer, .model_identifier = model, }; const esp_zb_endpoint_config_t ep_cfg = { .endpoint = endpoint_no, .app_device_id = 0x0101, .app_profile_id = 0x0104, .app_device_version = 0, }; esp_zb_ep_list_add_ep(ep_list, cluster_list, ep_cfg); zigbee_utility_add_ep_basic_manufacturer_info(ep_list, ep_cfg.endpoint, &mfg_info); } /* appearently you cant rename this? */ void esp_zb_app_signal_handler(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, steering anyway"); esp_zb_bdb_start_top_level_commissioning(ESP_ZB_BDB_MODE_NETWORK_STEERING); } } 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; } } void zigbee_task(void* pvParameters) { esp_zb_platform_config_t zb_platform_config = { .radio_config = { .radio_mode = ZB_RADIO_MODE_NATIVE, }, .host_config = { .host_connection_mode = ZB_HOST_CONNECTION_MODE_NONE, } }; esp_zb_cfg_t zb_nwk_cfg = { .esp_zb_role = ESP_ZB_DEVICE_TYPE_ROUTER, .install_code_policy = false, .nwk_cfg.zczr_cfg = { .max_children = 10, }, }; ESP_ERROR_CHECK(esp_zb_platform_config(&zb_platform_config)); esp_zb_init(&zb_nwk_cfg); esp_zb_cluster_list_t *cluster_list = esp_zb_zcl_cluster_list_create(); esp_zb_ep_list_t *endpoint_list = esp_zb_ep_list_create(); zigbee_clusterfuck_init(cluster_list); zigbee_endpoint_init(endpoint_list, cluster_list); esp_zb_device_register(endpoint_list); esp_zb_core_action_handler_register(zigbee_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 }