#ifndef ZIGBEE_MODE_ZCZR
#error "Zigbee Router mode required"
#endif

#include "Zigbee.h"

#define ENDPOINT_ID 1

ZigbeeThermostat thermostat(ENDPOINT_ID);

/* ---------------- ZCL ATTRIBUTE WRITE HANDLER ---------------- */
extern "C" void zb_attribute_set_handler(
  esp_zb_zcl_addr_t src_addr,
  uint8_t endpoint,
  uint16_t cluster,
  uint16_t attribute,
  uint8_t *data,
  uint16_t length
) {
  (void)src_addr;

  if (endpoint != ENDPOINT_ID) return;

  if (cluster == 0x0201) {  // Thermostat cluster
    Serial.println(">>> HA → Thermostat Write <<<");

    // SystemMode
    if (attribute == 0x001C && length == 1) {
      Serial.printf("SystemMode: %u\n", data[0]);
      // 0=Off, 3=Cool, 4=Heat
    }

    // Occupied Heating Setpoint
    if (attribute == 0x0012 && length == 2) {
      int16_t raw = data[0] | (data[1] << 8);
      float temp = raw / 100.0f;
      Serial.printf("Set Temperature: %.2f °C\n", temp);
    }

    // Unoccupied Heating Setpoint (HA also sends this)
    if (attribute == 0x0011 && length == 2) {
      int16_t raw = data[0] | (data[1] << 8);
      float temp = raw / 100.0f;
      Serial.printf("Unoccupied Setpoint: %.2f °C\n", temp);
    }
  }
}
/* ------------------------------------------------------------- */

void setup() {
  delay(1000);
  Serial.begin(115200);
  Serial.println("Zigbee Thermostat (IR Controller)");

  thermostat.setManufacturerAndModel("Espressif", "ZB-IR-AC");

  // Add endpoint
  Zigbee.addEndpoint(&thermostat);

  // Join existing coordinator (Home Assistant)
  if (!Zigbee.begin(ZIGBEE_ROUTER)) {
    Serial.println("Zigbee start failed");
    ESP.restart();
  }

  Serial.print("Joining Zigbee network");
  while (!Zigbee.connected()) {
    Serial.print(".");
    delay(500);
  }

  Serial.println("\nConnected to Home Assistant");
}

void loop() {
  // Nothing required here
  delay(1000);
}