From a07247be87a77b6cc2f90c71b371bbf60766ef26 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <derf@finalrewind.org>
Date: Mon, 18 Apr 2022 18:11:25 +0200
Subject: initial commit

---
 init.lua | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 init.lua

(limited to 'init.lua')

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()
-- 
cgit v1.2.3