For a more battle tested solution, I recommend trying ESPHome - https://esphome.io/ - it supports MQTT as well as many other protocols.