From 14c0657c42169d1953f3a7947a51a59c346616b9 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Tue, 19 Apr 2022 17:46:19 +0200 Subject: add homeassistant integration, set measurement period via MQTT --- init.lua | 52 ++++++++++++++++++++++++++++++++++++++++++++++------ 1 file changed, 46 insertions(+), 6 deletions(-) (limited to 'init.lua') diff --git a/init.lua b/init.lua index e9abef9..2f51e0e 100644 --- a/init.lua +++ b/init.lua @@ -2,11 +2,13 @@ station_cfg = {} dofile("config.lua") delayed_restart = tmr.create() -chipid = node.chipid() -mqtt_prefix = "sensor/esp8266_" .. chipid -mqttclient = mqtt.Client("esp8266_" .. chipid, 120) +chip_id = node.chipid() +device_id = "esp8266_" .. chip_id +mqtt_prefix = "sensor/" .. device_id +mqttclient = mqtt.Client(device_id, 120) -print("ESP8266 " .. chipid) + +print("ESP8266 " .. chip_id) ledpin = 4 gpio.mode(ledpin, gpio.OUTPUT) @@ -33,7 +35,8 @@ function connect_mqtt() print("IP address: " .. wifi.sta.getip()) print("Connecting to MQTT " .. mqtt_host) delayed_restart:stop() - mqttclient:on("connect", setup_client) + mqttclient:on("connect", hass_register) + mqttclient:on("message", hass_config) mqttclient:on("offline", log_restart) mqttclient:lwt(mqtt_prefix .. "/state", "offline", 0, 1) mqttclient:connect(mqtt_host) @@ -56,7 +59,11 @@ function uart_callback(data) print("Invalid or data-less SDS011 frame") return end - local json_str = string.format('{"pm25_ugm3": %d.%d, "pm10_ugm3": %d.%d, "rssi_dbm": %d}', pm25i, pm25f, pm10i, pm10f, wifi.sta.getrssi()) + local work_period = "continuous" + if sds011.work_period > 0 then + work_period = string.format("%d min", sds011.work_period) + end + local json_str = string.format('{"pm2_5_ugm3": %d.%d, "pm10_ugm3": %d.%d, "rssi_dbm": %d, "period": "%s"}', pm25i, pm25f, pm10i, pm10f, wifi.sta.getrssi(), work_period) if not publishing then publishing = true gpio.write(ledpin, 0) @@ -68,6 +75,39 @@ function uart_callback(data) end end +function hass_config(client, topic, message) + if topic == "config/" .. device_id .. "/set/work_period" then + local work_period = 0 + local _, _, minutes = string.find(message, "([0-9]+) min") + if minutes ~= nil then + work_period = tonumber(minutes) + end + port:write(sds011.set_work_period(work_period)) + end +end + +function hass_register() + local hass_device = string.format('{"connections":[["mac","%s"]],"identifiers":["%s"],"model":"ESP8266","name":"ESP8266 SDS011","manufacturer":"DIY"}', wifi.sta.getmac(), device_id) + local hass_entity_base = string.format('"device":%s,"state_topic":"%s/data","expire_after":1800', hass_device, mqtt_prefix) + local hass_pm2_5 = string.format('{%s,"name":"PM2.5","unique_id":"%s_pm2_5","icon":"mdi:air-filter","unit_of_measurement":"µg/m³","value_template":"{{value_json.pm2_5_ugm3}}"}', hass_entity_base, device_id) + local hass_pm10 = string.format('{%s,"name":"PM10","unique_id":"%s_pm10","icon":"mdi:air-filter","unit_of_measurement":"µg/m³","value_template":"{{value_json.pm10_ugm3}}"}', hass_entity_base, device_id) + local hass_rssi = string.format('{%s,"name":"RSSI","unique_id":"%s_rssi","icon":"mdi:signal","unit_of_measurement":"dBm","value_template":"{{value_json.rssi_dbm}}","entity_category":"diagnostic"}', hass_entity_base, device_id) + local hass_period = string.format('{%s,"name":"Measurement Period","unique_id":"%s_period","command_topic":"config/%s/set/work_period","options":["continuous","1 min","2 min","3 min","4 min","5 min","6 min","7 min","8 min","9 min","10 min"],"value_template":"{{value_json.period}}","entity_category":"config"}', hass_entity_base, device_id, device_id) + + mqttclient:publish("homeassistant/sensor/" .. device_id .. "/pm2_5/config", hass_pm2_5, 0, 1, function(client) + mqttclient:publish("homeassistant/sensor/" .. device_id .. "/pm10/config", hass_pm10, 0, 1, function(client) + mqttclient:publish("homeassistant/sensor/" .. device_id .. "/rssi/config", hass_rssi, 0, 1, function(client) + mqttclient:publish("homeassistant/select/" .. device_id .. "/work_period/config", hass_period, 0, 1, function(client) + client:subscribe("config/" .. device_id .. "/set/work_period", 0, function(client) + collectgarbage() + setup_client() + end) + end) + end) + end) + end) +end + delayed_restart:register(20 * 1000, tmr.ALARM_SINGLE, node.restart) connect_wifi() -- cgit v1.2.3