Flipdot-display/src/esp32/net/home_assistant.cpp

#include "net/home_assistant.h"

#include <ArduinoHA.h>
#include <WiFi.h>
#include <string.h>

#include "config.h"
#include "credentials.h"
#include "system/boot_target.h"

static WiFiClient client;
static HADevice device(HA_DEVICE_ID, sizeof(HA_DEVICE_ID));
static HAMqtt mqtt(client, device);

// Unique ids: do not change, Home Assistant's entities are linked to them
static HASwitch clockSwitch("FlipDotDisplayClock");
static HASwitch powerSwitch("FlipDotDisplayPower");
static HASensor firmwareSensor("FlipDotDisplayFirmware");
static HAButton recoveryButton("FlipDotDisplayRecovery");

static char firmware_status[64] = "";
static bool firmware_status_sent = false;

static DisplayController *displayController = nullptr;
static bool enabled = false;

// The switches are not optimistic: the new state must be reported back to Home Assistant
static void onClockCommand(bool state, HASwitch *sender)
{
	displayController->setClockEnabled(state);
	sender->setState(state);
}

static void onPowerCommand(bool state, HASwitch *sender)
{
	displayController->setPowerEnabled(state);
	sender->setState(state);
}

void homeAssistantBegin(DisplayController &controller, const HomeAssistantSettings &settings)
{
	displayController = &controller;
	enabled = settings.enabled;
	if (!enabled)
	{
		Serial.println("Home Assistant disabled in the settings");
		return;
	}

	device.setName(HA_DEVICE_NAME);
	device.setSoftwareVersion(HA_SOFTWARE_VERSION);

	clockSwitch.setName("FlipDot Display Clock");
	clockSwitch.setIcon("mdi:lightbulb");
	clockSwitch.setCurrentState(controller.clockEnabled());
	clockSwitch.onCommand(onClockCommand);

	powerSwitch.setName("FlipDot Display Power");
	powerSwitch.setIcon("mdi:lightbulb");
	powerSwitch.setCurrentState(controller.powerEnabled());
	powerSwitch.onCommand(onPowerCommand);

	recoveryButton.setName("FlipDot Display Recovery");
	recoveryButton.setIcon("mdi:lifebuoy");
	recoveryButton.onCommand([](HAButton *) { rebootToRecovery(RecoveryReason::Manual); });

	firmwareSensor.setName("FlipDot Display Firmware");
	firmwareSensor.setIcon("mdi:chip");
	// The sensor doesn't resend its value on reconnect by itself
	mqtt.onConnected([]() { firmware_status_sent = false; });

	// settings must stay alive: the MQTT library keeps the pointers
	if (settings.host[0])
		mqtt.begin(settings.host, settings.port, settings.user[0] ? settings.user : nullptr,
				   settings.password[0] ? settings.password : nullptr);
	else
		mqtt.begin(BROKER_ADDR, settings.port, MQTT_USER, MQTT_PASSWORD); // secrets.h
}

bool homeAssistantConnected()
{
	return enabled && mqtt.isConnected();
}

void homeAssistantLoop()
{
	if (!enabled)
		return;
	mqtt.loop();
	if (!firmware_status_sent && firmware_status[0] && mqtt.isConnected())
		firmware_status_sent = firmwareSensor.setValue(firmware_status);
}

void homeAssistantSetFirmwareStatus(const char *status)
{
	if (strcmp(status, firmware_status) == 0)
		return;
	strncpy(firmware_status, status, sizeof(firmware_status) - 1);
	firmware_status_sent = false;
}