103 lines
3.6 KiB
C++
103 lines
3.6 KiB
C++
#include "homeassistant_number.h"
|
|
|
|
#include "esphome/components/api/api_pb2.h"
|
|
#include "esphome/components/api/api_server.h"
|
|
#include "esphome/core/log.h"
|
|
|
|
namespace esphome {
|
|
namespace homeassistant {
|
|
|
|
static const char *const TAG = "homeassistant.number";
|
|
|
|
void HomeassistantNumber::state_changed_(const std::string &state) {
|
|
auto number_value = parse_number<float>(state);
|
|
if (!number_value.has_value()) {
|
|
ESP_LOGW(TAG, "'%s': Can't convert '%s' to number!", this->entity_id_.c_str(), state.c_str());
|
|
this->publish_state(NAN);
|
|
return;
|
|
}
|
|
if (this->state == number_value.value()) {
|
|
return;
|
|
}
|
|
ESP_LOGD(TAG, "'%s': Got state %s", this->entity_id_.c_str(), state.c_str());
|
|
this->publish_state(number_value.value());
|
|
}
|
|
|
|
void HomeassistantNumber::min_retrieved_(const std::string &min) {
|
|
auto min_value = parse_number<float>(min);
|
|
if (!min_value.has_value()) {
|
|
ESP_LOGE(TAG, "'%s': Can't convert 'min' value '%s' to number!", this->entity_id_.c_str(), min.c_str());
|
|
return;
|
|
}
|
|
ESP_LOGD(TAG, "'%s': Min retrieved: %s", get_name().c_str(), min.c_str());
|
|
this->traits.set_min_value(min_value.value());
|
|
}
|
|
|
|
void HomeassistantNumber::max_retrieved_(const std::string &max) {
|
|
auto max_value = parse_number<float>(max);
|
|
if (!max_value.has_value()) {
|
|
ESP_LOGE(TAG, "'%s': Can't convert 'max' value '%s' to number!", this->entity_id_.c_str(), max.c_str());
|
|
return;
|
|
}
|
|
ESP_LOGD(TAG, "'%s': Max retrieved: %s", get_name().c_str(), max.c_str());
|
|
this->traits.set_max_value(max_value.value());
|
|
}
|
|
|
|
void HomeassistantNumber::step_retrieved_(const std::string &step) {
|
|
auto step_value = parse_number<float>(step);
|
|
if (!step_value.has_value()) {
|
|
ESP_LOGE(TAG, "'%s': Can't convert 'step' value '%s' to number!", this->entity_id_.c_str(), step.c_str());
|
|
return;
|
|
}
|
|
ESP_LOGD(TAG, "'%s': Step Retrieved %s", get_name().c_str(), step.c_str());
|
|
this->traits.set_step(step_value.value());
|
|
}
|
|
|
|
void HomeassistantNumber::setup() {
|
|
api::global_api_server->subscribe_home_assistant_state(
|
|
this->entity_id_, nullopt, std::bind(&HomeassistantNumber::state_changed_, this, std::placeholders::_1));
|
|
|
|
api::global_api_server->get_home_assistant_state(
|
|
this->entity_id_, optional<std::string>("min"),
|
|
std::bind(&HomeassistantNumber::min_retrieved_, this, std::placeholders::_1));
|
|
api::global_api_server->get_home_assistant_state(
|
|
this->entity_id_, optional<std::string>("max"),
|
|
std::bind(&HomeassistantNumber::max_retrieved_, this, std::placeholders::_1));
|
|
api::global_api_server->get_home_assistant_state(
|
|
this->entity_id_, optional<std::string>("step"),
|
|
std::bind(&HomeassistantNumber::step_retrieved_, this, std::placeholders::_1));
|
|
}
|
|
|
|
void HomeassistantNumber::dump_config() {
|
|
LOG_NUMBER("", "Homeassistant Number", this);
|
|
ESP_LOGCONFIG(TAG, " Entity ID: '%s'", this->entity_id_.c_str());
|
|
}
|
|
|
|
float HomeassistantNumber::get_setup_priority() const { return setup_priority::AFTER_CONNECTION; }
|
|
|
|
void HomeassistantNumber::control(float value) {
|
|
if (!api::global_api_server->is_connected()) {
|
|
ESP_LOGE(TAG, "No clients connected to API server");
|
|
return;
|
|
}
|
|
|
|
this->publish_state(value);
|
|
|
|
api::HomeassistantServiceResponse resp;
|
|
resp.service = "number.set_value";
|
|
|
|
api::HomeassistantServiceMap entity_id;
|
|
entity_id.key = "entity_id";
|
|
entity_id.value = this->entity_id_;
|
|
resp.data.push_back(entity_id);
|
|
|
|
api::HomeassistantServiceMap entity_value;
|
|
entity_value.key = "value";
|
|
entity_value.value = to_string(value);
|
|
resp.data.push_back(entity_value);
|
|
|
|
api::global_api_server->send_homeassistant_service_call(resp);
|
|
}
|
|
|
|
} // namespace homeassistant
|
|
} // namespace esphome
|