From 65a1fe9cfd7c283acf4697dda3e11d4b8df30f02 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Sat, 6 Jan 2024 21:47:30 +0100 Subject: initial commit --- .gitmodules | 6 +++ ext/mh-z19 | 1 + ext/ssd1306 | 1 + src/framebuffer.lua | 1 + src/init.lua | 136 ++++++++++++++++++++++++++++++++++++++++++++++++++++ src/mh-z19.lua | 1 + src/ssd1306.lua | 1 + src/terminus16.lua | 1 + 8 files changed, 148 insertions(+) create mode 100644 .gitmodules create mode 160000 ext/mh-z19 create mode 160000 ext/ssd1306 create mode 120000 src/framebuffer.lua create mode 100644 src/init.lua create mode 120000 src/mh-z19.lua create mode 120000 src/ssd1306.lua create mode 120000 src/terminus16.lua diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..04df129 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,6 @@ +[submodule "ext/ssd1306"] + path = ext/ssd1306 + url = https://github.com/derf/esp8266-nodemcu-ssd1306.git +[submodule "ext/mh-z19"] + path = ext/mh-z19 + url = https://github.com/derf/esp8266-nodemcu-mh-z19.git diff --git a/ext/mh-z19 b/ext/mh-z19 new file mode 160000 index 0000000..4b0115c --- /dev/null +++ b/ext/mh-z19 @@ -0,0 +1 @@ +Subproject commit 4b0115c6f5d80d416ac4e84e679a4790f88ed6f6 diff --git a/ext/ssd1306 b/ext/ssd1306 new file mode 160000 index 0000000..3ac749b --- /dev/null +++ b/ext/ssd1306 @@ -0,0 +1 @@ +Subproject commit 3ac749b83cf218ab151ae1ca72badd41eaf0cc49 diff --git a/src/framebuffer.lua b/src/framebuffer.lua new file mode 120000 index 0000000..ba71daf --- /dev/null +++ b/src/framebuffer.lua @@ -0,0 +1 @@ +../ext/ssd1306/framebuffer.lua \ No newline at end of file diff --git a/src/init.lua b/src/init.lua new file mode 100644 index 0000000..f31aa78 --- /dev/null +++ b/src/init.lua @@ -0,0 +1,136 @@ +chip_id = string.format("%06X", node.chipid()) +device_id = "esp8266_" .. chip_id +mqtt_prefix = "sensor/" .. device_id +mqttclient = mqtt.Client(device_id, 120) + +dofile("config.lua") + +i2c.setup(0, 5, 6, i2c.SLOW) +ssd1306 = require("ssd1306") +fn = require("terminus16") +fb = require("framebuffer") +mh_z19 = require("mh-z19") + +ledpin = 4 +gpio.mode(ledpin, gpio.OUTPUT) +gpio.write(ledpin, 0) + +ssd1306.init(128, 32) +ssd1306.contrast(128) +fb.init(128, 32) + +no_wifi_count = 0 +publish_count = 0 +publishing_mqtt = false + +function connect_wifi() + print("Connecting to ESSID " .. station_cfg.ssid) + wifi.eventmon.register(wifi.eventmon.STA_GOT_IP, wifi_connected) + wifi.eventmon.register(wifi.eventmon.STA_DHCP_TIMEOUT, wifi_err) + wifi.eventmon.register(wifi.eventmon.STA_DISCONNECTED, wifi_err) + wifi.setmode(wifi.STATION, false) + wifi.sta.config(station_cfg) + wifi.sta.connect() +end + +function init_mhz19() + port = softuart.setup(9600, 1, 2) + port:on("data", 9, uart_callback) + poll = tmr.create() + poll:register(5 * 1000, tmr.ALARM_AUTO, query_mhz19) + poll:start() + query_mhz19() + gpio.write(ledpin, 1) +end + +function query_mhz19() + port:write(mh_z19.c_query) +end + +function uart_callback(data) + fb.init(128, 32) + + if not mh_z19.parse_frame(data) then + fb.print(fn, "MH-Z19 error") + ssd1306.show(fb.buf) + return + end + + local line1 = string.format("%8d ppm\n", mh_z19.co2) + local line2 = string.format("%8d c\n", mh_z19.temp) + + fb.print(fn, line1) + fb.print(fn, line2) + + if have_wifi then + fb.y = 16 + fb.x = 100 + fb.print(fn, string.format("%d", wifi.sta.getrssi())) + else + if no_wifi_count == 5 then + wifi.setmode(wifi.NULLMODE, false) + end + if no_wifi_count < 24 then + no_wifi_count = no_wifi_count + 1 + else + no_wifi_count = 0 + connect_wifi() + end + end + ssd1306.show(fb.buf) + fb.init(128, 32) + publish_count = publish_count + 1 + if have_wifi and publish_count >= 4 and not publishing_mqtt then + publish_count = 0 + publishing_mqtt = true + gpio.write(ledpin, 0) + local json_str = string.format('{"rssi_dbm":%d,"co2_ppm":%d,"temperature_celsius":%d}', wifi.sta.getrssi(), mh_z19.co2, mh_z19.temp) + mqttclient:publish(mqtt_prefix .. "/data", json_str, 0, 0, function(client) + publishing_mqtt = false + gpio.write(ledpin, 1) + collectgarbage() + end) + else + collectgarbage() + end +end + +function wifi_connected() + print("IP address: " .. wifi.sta.getip()) + have_wifi = true + no_wifi_count = 0 + print("Connecting to MQTT " .. mqtt_host) + mqttclient:on("connect", hass_register) + mqttclient:on("offline", wifi_err) + mqttclient:lwt(mqtt_prefix .. "/state", "offline", 0, 1) + mqttclient:connect(mqtt_host) +end + +function wifi_err() + have_wifi = false +end + +function hass_register() + local hass_device = string.format('{"connections":[["mac","%s"]],"identifiers":["%s"],"model":"ESP8266 + MH-Z19","name":"MH-Z19 %s","manufacturer":"derf"}', wifi.sta.getmac(), device_id, chip_id) + local hass_entity_base = string.format('"device":%s,"state_topic":"%s/data","expire_after":120', hass_device, mqtt_prefix) + local hass_co2 = string.format('{%s,"name":"CO₂","object_id":"%s_co2","unique_id":"%s_co2","device_class":"carbon_dioxide","unit_of_measurement":"ppm","value_template":"{{value_json.co2_ppm}}"}', hass_entity_base, device_id, device_id) + local hass_temp = string.format('{%s,"name":"Temperature","object_id":"%s_temp","unique_id":"%s_temp","device_class":"temperature","unit_of_measurement":"°c","value_template":"{{value_json.temperature_celsius}}","entity_category":"diagnostic"}', 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) + + publishing_mqtt = true + mqttclient:publish("homeassistant/sensor/" .. device_id .. "/co2/config", hass_co2, 0, 1, function(client) + mqttclient:publish("homeassistant/sensor/" .. device_id .. "/temperature/config", hass_temp, 0, 1, function(client) + mqttclient:publish("homeassistant/sensor/" .. device_id .. "/rssi/config", hass_rssi, 0, 1, function(client) + mqttclient:publish(mqtt_prefix .. "/state", "online", 0, 1, function(client) + publishing_mqtt = false + print("Registered with Home Assistant") + collectgarbage() + end) + end) + end) + end) +end + +print("WiFi MAC: " .. wifi.sta.getmac()) +init_mhz19() +connect_wifi() diff --git a/src/mh-z19.lua b/src/mh-z19.lua new file mode 120000 index 0000000..bc0e445 --- /dev/null +++ b/src/mh-z19.lua @@ -0,0 +1 @@ +../ext/mh-z19/mh-z19.lua \ No newline at end of file diff --git a/src/ssd1306.lua b/src/ssd1306.lua new file mode 120000 index 0000000..ed7ee52 --- /dev/null +++ b/src/ssd1306.lua @@ -0,0 +1 @@ +../ext/ssd1306/ssd1306.lua \ No newline at end of file diff --git a/src/terminus16.lua b/src/terminus16.lua new file mode 120000 index 0000000..1ee383e --- /dev/null +++ b/src/terminus16.lua @@ -0,0 +1 @@ +../ext/ssd1306/terminus16.lua \ No newline at end of file -- cgit v1.2.3