Add Xiaomi Miscale v1 and v2 (#1368)

esphome/components/xiaomi_miscale/sensor.py

@@ -0,0 +1,30 @@
+import esphome.codegen as cg
+import esphome.config_validation as cv
+from esphome.components import sensor, esp32_ble_tracker
+from esphome.const import CONF_MAC_ADDRESS, CONF_ID, CONF_WEIGHT, UNIT_KILOGRAM, \
+    ICON_SCALE_BATHROOM
+
+DEPENDENCIES = ['esp32_ble_tracker']
+
+xiaomi_miscale_ns = cg.esphome_ns.namespace('xiaomi_miscale')
+XiaomiMiscale = xiaomi_miscale_ns.class_('XiaomiMiscale',
+                                         esp32_ble_tracker.ESPBTDeviceListener,
+                                         cg.Component)
+
+CONFIG_SCHEMA = cv.Schema({
+    cv.GenerateID(): cv.declare_id(XiaomiMiscale),
+    cv.Required(CONF_MAC_ADDRESS): cv.mac_address,
+    cv.Optional(CONF_WEIGHT): sensor.sensor_schema(UNIT_KILOGRAM, ICON_SCALE_BATHROOM, 2),
+}).extend(esp32_ble_tracker.ESP_BLE_DEVICE_SCHEMA).extend(cv.COMPONENT_SCHEMA)
+
+
+def to_code(config):
+    var = cg.new_Pvariable(config[CONF_ID])
+    yield cg.register_component(var, config)
+    yield esp32_ble_tracker.register_ble_device(var, config)
+
+    cg.add(var.set_address(config[CONF_MAC_ADDRESS].as_hex))
+
+    if CONF_WEIGHT in config:
+        sens = yield sensor.new_sensor(config[CONF_WEIGHT])
+        cg.add(var.set_weight(sens))

esphome/components/xiaomi_miscale/xiaomi_miscale.cpp

@@ -0,0 +1,101 @@
+#include "xiaomi_miscale.h"
+#include "esphome/core/log.h"
+
+#ifdef ARDUINO_ARCH_ESP32
+
+namespace esphome {
+namespace xiaomi_miscale {
+
+static const char *TAG = "xiaomi_miscale";
+
+void XiaomiMiscale::dump_config() {
+  ESP_LOGCONFIG(TAG, "Xiaomi Miscale");
+  LOG_SENSOR("  ", "Weight", this->weight_);
+}
+
+bool XiaomiMiscale::parse_device(const esp32_ble_tracker::ESPBTDevice &device) {
+  if (device.address_uint64() != this->address_) {
+    ESP_LOGVV(TAG, "parse_device(): unknown MAC address.");
+    return false;
+  }
+  ESP_LOGVV(TAG, "parse_device(): MAC address %s found.", device.address_str().c_str());
+
+  bool success = false;
+  for (auto &service_data : device.get_service_datas()) {
+    auto res = parse_header(service_data);
+    if (!res.has_value()) {
+      continue;
+    }
+    if (!(parse_message(service_data.data, *res))) {
+      continue;
+    }
+    if (!(report_results(res, device.address_str()))) {
+      continue;
+    }
+    if (res->weight.has_value() && this->weight_ != nullptr)
+      this->weight_->publish_state(*res->weight);
+    success = true;
+  }
+
+  return success;
+}
+
+optional<ParseResult> XiaomiMiscale::parse_header(const esp32_ble_tracker::ServiceData &service_data) {
+  ParseResult result;
+  if (!service_data.uuid.contains(0x1D, 0x18)) {
+    ESP_LOGVV(TAG, "parse_header(): no service data UUID magic bytes.");
+    return {};
+  }
+
+  return result;
+}
+
+bool XiaomiMiscale::parse_message(const std::vector<uint8_t> &message, ParseResult &result) {
+  // exemple 1d18 a2 6036 e307 07 11 0f1f11
+  // 1-2 Weight (MISCALE 181D)
+  // 3-4 Years (MISCALE 181D)
+  // 5 month (MISCALE 181D)
+  // 6 day (MISCALE 181D)
+  // 7 hour (MISCALE 181D)
+  // 8 minute (MISCALE 181D)
+  // 9 second (MISCALE 181D)
+
+  const uint8_t *data = message.data();
+  const int data_length = 10;
+
+  if (message.size() != data_length) {
+    ESP_LOGVV(TAG, "parse_message(): payload has wrong size (%d)!", message.size());
+    return false;
+  }
+
+  // weight, 2 bytes, 16-bit  unsigned integer, 1 kg
+  const int16_t weight = uint16_t(data[1]) | (uint16_t(data[2]) << 8);
+  if (data[0] == 0x22 || data[0] == 0xa2)
+    result.weight = weight * 0.01f / 2.0f;  // unit 'kg'
+  else if (data[0] == 0x12 || data[0] == 0xb2)
+    result.weight = weight * 0.01f * 0.6;  // unit 'jin'
+  else if (data[0] == 0x03 || data[0] == 0xb3)
+    result.weight = weight * 0.01f * 0.453592;  // unit 'lbs'
+
+  return true;
+}
+
+bool XiaomiMiscale::report_results(const optional<ParseResult> &result, const std::string &address) {
+  if (!result.has_value()) {
+    ESP_LOGVV(TAG, "report_results(): no results available.");
+    return false;
+  }
+
+  ESP_LOGD(TAG, "Got Xiaomi Miscale (%s):", address.c_str());
+
+  if (result->weight.has_value()) {
+    ESP_LOGD(TAG, "  Weight: %.2fkg", *result->weight);
+  }
+
+  return true;
+}
+
+}  // namespace xiaomi_miscale
+}  // namespace esphome
+
+#endif

esphome/components/xiaomi_miscale/xiaomi_miscale.h

@@ -0,0 +1,37 @@
+#pragma once
+
+#include "esphome/core/component.h"
+#include "esphome/components/sensor/sensor.h"
+#include "esphome/components/esp32_ble_tracker/esp32_ble_tracker.h"
+
+#ifdef ARDUINO_ARCH_ESP32
+
+namespace esphome {
+namespace xiaomi_miscale {
+
+struct ParseResult {
+  optional<float> weight;
+};
+
+class XiaomiMiscale : public Component, public esp32_ble_tracker::ESPBTDeviceListener {
+ public:
+  void set_address(uint64_t address) { address_ = address; };
+
+  bool parse_device(const esp32_ble_tracker::ESPBTDevice &device) override;
+  void dump_config() override;
+  float get_setup_priority() const override { return setup_priority::DATA; }
+  void set_weight(sensor::Sensor *weight) { weight_ = weight; }
+
+ protected:
+  uint64_t address_;
+  sensor::Sensor *weight_{nullptr};
+
+  optional<ParseResult> parse_header(const esp32_ble_tracker::ServiceData &service_data);
+  bool parse_message(const std::vector<uint8_t> &message, ParseResult &result);
+  bool report_results(const optional<ParseResult> &result, const std::string &address);
+};
+
+}  // namespace xiaomi_miscale
+}  // namespace esphome
+
+#endif