// Передатчик
#include <WiFi.h>
#include <esp_now.h>
#include "esp_mac.h"

#define BUTTON_PIN      4      // GPIO для кнопки
#define LED_PIN         5      // GPIO для светодиода
#define CONFIRM_TIMEOUT   10000   // 10 сек для проверки связи
#define   SEND_INTERVAL   5000   // 5 секунд
#define   BLINK_INTERVAL   1000   // 1 секунда

// MAC приёмника
uint8_t receiverMac[] = {0xB6, 0xBB, 0xBB, 0xBB, 0xBB, 0xBB};

typedef struct
{
   bool buttonState;
} ButtonMessage;

typedef struct
{
   bool ack;
} AckMessage;

unsigned long lastSendTime = 0;
unsigned long lastConfirmation = 0;
unsigned long lastBlink = 0;
bool ledState = LOW;

void OnDataSent (const uint8_t *mac, esp_now_send_status_t status)
{
   // Логирование статуса
}

void OnDataRecv (const uint8_t *mac, const uint8_t *data, int len)
{
   if (len == sizeof (AckMessage))
   {
      AckMessage *ack = (AckMessage*)data;

      if (ack->ack)
         lastConfirmation = millis();
   }
}

void setup()
{
   // Установка кастомного MAC для передатчика
   uint8_t newMac[] = {0xB6, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA};
   esp_base_mac_addr_set (newMac);

   pinMode (BUTTON_PIN, INPUT_PULLUP);
   pinMode (LED_PIN, OUTPUT);

   WiFi.mode (WIFI_STA);

   if (esp_now_init() != ESP_OK) return;

   esp_now_register_send_cb (OnDataSent);
   esp_now_register_recv_cb (OnDataRecv);

   esp_now_peer_info_t peerInfo;
   memcpy (peerInfo.peer_addr, receiverMac, 6);
   peerInfo.channel = 1;
   peerInfo.encrypt = false;
   esp_now_add_peer (&peerInfo);
}

void loop()
{
   unsigned long now = millis();

   // Отправка состояния каждые 5 секунд
   if (now - lastSendTime >= SEND_INTERVAL)
   {
      ButtonMessage msg;
      msg.buttonState = digitalRead (BUTTON_PIN) == HIGH;
      esp_now_send (receiverMac, (uint8_t*)&msg, sizeof (msg));
      lastSendTime = now;
   }

   // Управление светодиодом
   bool isConnected = (now - lastConfirmation) < CONFIRM_TIMEOUT;

   if (isConnected)
      digitalWrite (LED_PIN, HIGH);

   else
   {
      // Мигание раз в секунду
      if (now - lastBlink >= BLINK_INTERVAL)
      {
         ledState = !ledState;
         digitalWrite (LED_PIN, ledState);
         lastBlink = now;
      }
   }

   delay (10);
}

// Приёмник
#include <ESP8266WiFi.h>
#include <espnow.h>
//#include "esp_mac.h"

#define   RELAY_PIN   0      // GPIO0 для реле
#define TIMEOUT      30000   // Таймаут 30 секунд
#define   OFF         1

// MAC передатчика
uint8_t transmitterMac[] = {0xB6, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA};

typedef struct
{
   bool buttonState;
} ButtonMessage;

typedef struct
{
   bool ack;
} AckMessage;

unsigned long lastMsgTime = 0;

void OnDataRecv (uint8_t *mac, uint8_t *data, uint8_t len)
{
   if (len == sizeof (ButtonMessage))
   {
      ButtonMessage *msg = (ButtonMessage*)data;

      digitalWrite (RELAY_PIN, msg->buttonState ? HIGH : LOW);

      AckMessage ack;
      ack.ack = true;
      esp_now_send (mac, (uint8_t*)&ack, sizeof (ack));

      lastMsgTime = millis();
   }
}

void setup()
{
   // Установка кастомного MAC для приёмника
   uint8_t newMac[] = {0xB6, 0xBB, 0xBB, 0xBB, 0xBB, 0xBB};
   //esp_base_mac_addr_set (newMac);
   pinMode (RELAY_PIN, OUTPUT);
   digitalWrite (RELAY_PIN, OFF);

   WiFi.mode (WIFI_STA);
   wifi_set_macaddr (STATION_IF, &newMac[0]);

   if (esp_now_init() != 0) return;

   esp_now_set_self_role (ESP_NOW_ROLE_SLAVE);
   esp_now_register_recv_cb (OnDataRecv);
   esp_now_add_peer (transmitterMac, ESP_NOW_ROLE_CONTROLLER, 1, NULL, 0);
}

void loop()
{
   unsigned long now = millis();

   if (now - lastMsgTime > TIMEOUT) digitalWrite (RELAY_PIN, OFF);

   delay (100);
}