1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
station_cfg = {}
dofile("config.lua")
delayed_restart = tmr.create()
chip_id = node.chipid()
device_id = "esp8266_" .. chip_id
mqtt_prefix = "sensor/" .. device_id
mqttclient = mqtt.Client(device_id, 120)
print("ESP8266 " .. chip_id)
ledpin = 4
gpio.mode(ledpin, gpio.OUTPUT)
gpio.write(ledpin, 0)
sds011 = require("sds011")
function log_restart()
print("Network error " .. wifi.sta.status() .. ". Restarting in 20 seconds.")
delayed_restart:start()
end
function setup_client()
gpio.write(ledpin, 1)
publishing = true
mqttclient:publish(mqtt_prefix .. "/state", "online", 0, 1, function(client)
publishing = false
end)
port = softuart.setup(9600, 2, 1)
port:on("data", 10, uart_callback)
end
function connect_mqtt()
print("IP address: " .. wifi.sta.getip())
print("Connecting to MQTT " .. mqtt_host)
delayed_restart:stop()
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)
end
function connect_wifi()
print("WiFi MAC: " .. wifi.sta.getmac())
print("Connecting to ESSID " .. station_cfg.ssid)
wifi.eventmon.register(wifi.eventmon.STA_GOT_IP, connect_mqtt)
wifi.eventmon.register(wifi.eventmon.STA_DHCP_TIMEOUT, log_restart)
wifi.eventmon.register(wifi.eventmon.STA_DISCONNECTED, log_restart)
wifi.setmode(wifi.STATION)
wifi.sta.config(station_cfg)
wifi.sta.connect()
end
function uart_callback(data)
local pm25i, pm25f, pm10i, pm10f = sds011.parse_frame(data)
if pm25i == nil then
print("Invalid or data-less SDS011 frame")
return
end
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)
mqttclient:publish(mqtt_prefix .. "/data", json_str, 0, 0, function(client)
publishing = false
gpio.write(ledpin, 1)
collectgarbage()
end)
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:wifi","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","icon":"mdi:clock-outline","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()
|
