diff options
Diffstat (limited to 'init.lua')
| -rw-r--r-- | init.lua | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/init.lua b/init.lua new file mode 100644 index 0000000..fe53b86 --- /dev/null +++ b/init.lua @@ -0,0 +1,75 @@ +station_cfg = {} +dofile("config.lua") + +delayed_restart = tmr.create() +chipid = node.chipid() +mqtt_prefix = "sensor/esp8266_" .. chipid +mqttclient = mqtt.Client("esp8266_" .. chipid, 120) + +print("ESP8266 " .. chipid) + +ledpin = 4 +gpio.mode(ledpin, gpio.OUTPUT) +gpio.write(ledpin, 0) + +pm1006 = require("pm1006") + +function log_restart() + print("Network error " .. wifi.sta.status() .. ". Restarting in 20 seconds.") + delayed_restart:start() +end + +function setup_client() + print("Connected") + gpio.write(ledpin, 1) + publishing = true + mqttclient:publish(mqtt_prefix .. "/state", "online", 0, 1, function(client) + publishing = false + end) + port = softuart.setup(9600, nil, 2) + port:on("data", 20, uart_callback) +end + +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("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 pm2_5 = pm1006.parse_frame(data) + if pm2_5 == nil then + print("Invalid PM1006 frame") + return + else + local json_str = string.format('{"pm25_ugm3": %d, "rssi_dbm": %d}', pm2_5, wifi.sta.getrssi()) + 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 +end + +delayed_restart:register(20 * 1000, tmr.ALARM_SINGLE, node.restart) + +connect_wifi() |