Add Number entities (from Home Assistant) (#1971)

Co-authored-by: Oxan van Leeuwen <oxan@oxanvanleeuwen.nl>

esphome/components/mqtt/__init__.py

@@ -91,6 +91,7 @@ MQTTJSONLightComponent = mqtt_ns.class_("MQTTJSONLightComponent", MQTTComponent)
 MQTTSensorComponent = mqtt_ns.class_("MQTTSensorComponent", MQTTComponent)
 MQTTSwitchComponent = mqtt_ns.class_("MQTTSwitchComponent", MQTTComponent)
 MQTTTextSensor = mqtt_ns.class_("MQTTTextSensor", MQTTComponent)
+MQTTNumberComponent = mqtt_ns.class_("MQTTNumberComponent", MQTTComponent)
 
 
 def validate_config(value):

esphome/components/mqtt/mqtt_number.cpp

@@ -0,0 +1,61 @@
+#include "mqtt_number.h"
+#include "esphome/core/log.h"
+
+#ifdef USE_NUMBER
+
+#ifdef USE_DEEP_SLEEP
+#include "esphome/components/deep_sleep/deep_sleep_component.h"
+#endif
+
+namespace esphome {
+namespace mqtt {
+
+static const char *const TAG = "mqtt.number";
+
+using namespace esphome::number;
+
+MQTTNumberComponent::MQTTNumberComponent(Number *number) : MQTTComponent(), number_(number) {}
+
+void MQTTNumberComponent::setup() {
+  this->subscribe(this->get_command_topic_(), [this](const std::string &topic, const std::string &state) {
+    auto val = parse_float(state);
+    if (!val.has_value()) {
+      ESP_LOGE(TAG, "Can't convert '%s' to number!", state.c_str());
+      return;
+    }
+    auto call = this->number_->make_call();
+    call.set_value(*val);
+    call.perform();
+  });
+  this->number_->add_on_state_callback([this](float state) { this->publish_state(state); });
+}
+
+void MQTTNumberComponent::dump_config() {
+  ESP_LOGCONFIG(TAG, "MQTT Number '%s':", this->number_->get_name().c_str());
+  LOG_MQTT_COMPONENT(true, false)
+}
+
+std::string MQTTNumberComponent::component_type() const { return "number"; }
+
+std::string MQTTNumberComponent::friendly_name() const { return this->number_->get_name(); }
+void MQTTNumberComponent::send_discovery(JsonObject &root, mqtt::SendDiscoveryConfig &config) {
+  if (!this->number_->get_icon().empty())
+    root["icon"] = this->number_->get_icon();
+}
+bool MQTTNumberComponent::send_initial_state() {
+  if (this->number_->has_state()) {
+    return this->publish_state(this->number_->state);
+  } else {
+    return true;
+  }
+}
+bool MQTTNumberComponent::is_internal() { return this->number_->is_internal(); }
+bool MQTTNumberComponent::publish_state(float value) {
+  int8_t accuracy = this->number_->get_accuracy_decimals();
+  return this->publish(this->get_state_topic_(), value_accuracy_to_string(value, accuracy));
+}
+
+}  // namespace mqtt
+}  // namespace esphome
+
+#endif

esphome/components/mqtt/mqtt_number.h

@@ -0,0 +1,46 @@
+#pragma once
+
+#include "esphome/core/defines.h"
+
+#ifdef USE_NUMBER
+
+#include "esphome/components/number/number.h"
+#include "mqtt_component.h"
+
+namespace esphome {
+namespace mqtt {
+
+class MQTTNumberComponent : public mqtt::MQTTComponent {
+ public:
+  /** Construct this MQTTNumberComponent instance with the provided friendly_name and number
+   *
+   * @param number The number.
+   */
+  explicit MQTTNumberComponent(number::Number *number);
+
+  // ========== INTERNAL METHODS ==========
+  // (In most use cases you won't need these)
+  /// Override setup.
+  void setup() override;
+  void dump_config() override;
+
+  void send_discovery(JsonObject &root, mqtt::SendDiscoveryConfig &config) override;
+
+  bool send_initial_state() override;
+  bool is_internal() override;
+
+  bool publish_state(float value);
+
+ protected:
+  /// Override for MQTTComponent, returns "number".
+  std::string component_type() const override;
+
+  std::string friendly_name() const override;
+
+  number::Number *number_;
+};
+
+}  // namespace mqtt
+}  // namespace esphome
+
+#endif