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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
|
publishing_mqtt = false
publishing_http = false
watchdog = tmr.create()
chip_id = string.format("%06X", node.chipid())
device_id = "esp8266_" .. chip_id
dofile("config.lua")
if mqtt_host then
mqtt_prefix = "sensor/" .. device_id
mqttclient = mqtt.Client(device_id, 120)
end
print("SDS011 " .. chip_id)
ledpin = 4
gpio.mode(ledpin, gpio.OUTPUT)
gpio.write(ledpin, 0)
sds011 = require("sds011")
poll = tmr.create()
polling = false
function log_restart()
print("Network error " .. wifi.sta.status())
end
function setup_client()
print("Connected")
gpio.write(ledpin, 1)
port = softuart.setup(9600, 2, 1)
port:on("data", 10, uart_callback)
if mqtt_host then
publishing_mqtt = true
mqttclient:publish(mqtt_prefix .. "/state", "online", 0, 1, function(client)
publishing_mqtt = false
port:write(sds011.set_work_period(nil))
end)
else
port:write(sds011.set_work_period(0))
end
end
function connect_mqtt()
print("IP address: " .. wifi.sta.getip())
print("Connecting to MQTT " .. mqtt_host)
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)
if mqtt_host then
wifi.eventmon.register(wifi.eventmon.STA_GOT_IP, connect_mqtt)
else
wifi.eventmon.register(wifi.eventmon.STA_GOT_IP, setup_client)
end
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)
if not sds011.parse_frame(data) 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
gpio.write(ledpin, 0)
if sds011.work_period == 0 and not polling then
polling = true
port:write(sds011.set_report_mode(false))
poll:start()
end
if sds011.work_period > 0 and polling then
polling = false
port:write(sds011.set_report_mode(true))
poll:stop()
end
local json_str = string.format('{"rssi_dbm":%d,"period":"%s"', wifi.sta.getrssi(), work_period)
local influx_str = ""
if sds011.pm2_5i ~= nil then
json_str = string.format('%s,"pm2_5_ugm3":%d.%d,"pm10_ugm3":%d.%d', json_str, sds011.pm2_5i, sds011.pm2_5f, sds011.pm10i, sds011.pm10f)
influx_str = string.format("pm2_5_ugm3=%d.%d,pm10_ugm3=%d.%d", sds011.pm2_5i, sds011.pm2_5f, sds011.pm10i, sds011.pm10f)
end
json_str = json_str .. '}'
if mqtt_host then
if not publishing_mqtt then
watchdog:start(true)
publishing_mqtt = true
mqttclient:publish(mqtt_prefix .. "/data", json_str, 0, 0, function(client)
publishing_mqtt = false
if influx_url and influx_attr and influx_str then
publish_influx(influx_str)
else
gpio.write(ledpin, 1)
collectgarbage()
end
end)
end
elseif influx_url and influx_attr and influx_str then
publish_influx(influx_str)
end
end
function publish_influx(payload)
if not publishing_http then
publishing_http = true
http.post(influx_url, influx_header, "sds011" .. influx_attr .. " " .. payload, function(code, data)
print("POST " .. influx_url .. " sds011" .. influx_attr .. " " .. payload .. " returned " .. code)
publishing_http = false
gpio.write(ledpin, 1)
collectgarbage()
end)
end
end
function query_data()
port:write(sds011.query())
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 + SDS011","name":"SDS011 %s","manufacturer":"derf"}', wifi.sta.getmac(), device_id, chip_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","object_id":"%s_pm2_5","unique_id":"%s_pm2_5","device_class":"pm25","unit_of_measurement":"µg/m³","value_template":"{{value_json.pm2_5_ugm3}}"}', hass_entity_base, device_id, device_id)
local hass_pm10 = string.format('{%s,"name":"PM10","object_id":"%s_pm10","unique_id":"%s_pm10","device_class":"pm10","unit_of_measurement":"µg/m³","value_template":"{{value_json.pm10_ugm3}}"}', hass_entity_base, device_id, device_id)
local hass_rssi = string.format('{%s,"name":"RSSI","object_id":"%s_rssi","unique_id":"%s_rssi","device_class":"signal_strength","unit_of_measurement":"dBm","value_template":"{{value_json.rssi_dbm}}","entity_category":"diagnostic"}', hass_entity_base, device_id, device_id)
local hass_period = string.format('{%s,"name":"Measurement Period","object_id":"%s_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, 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
watchdog:register(20 * 60 * 1000, tmr.ALARM_SEMI, node.restart)
poll:register(10 * 1000, tmr.ALARM_AUTO, query_data)
watchdog:start()
connect_wifi()
|
