#include <WiFi.h>
#include <PubSubClient.h>
#include <ArduinoJson.h>

const char *ssid = "Wokwi-GUEST";
const char *password = "";

const char *mqttServer = "demo.thingsboard.io";
const int mqttPort = 1883;
const char *accessToken = "aImx9OwfslDJw1QsbRYj";  // Thay đổi YOUR_ACCESS_TOKEN

WiFiClient espClient;
PubSubClient client(espClient);

unsigned int attr_counter = 0;
String tb_payload;  //payload in message which is send to TB
char data_buffer[100];  //memory buffer to store payload data, using in AT command
StaticJsonDocument<400> jsonBuffer;

unsigned long timming;

void tb_callback(char* topic, byte* payload, unsigned int length){
  Serial.println(F("On TB message"));

  char* ctrl_json = new char[length + 1];
  strncpy (ctrl_json, (char*)payload, length);
  ctrl_json[length] = '\0';
      
  //debug
  //topic: "v1/devices/me/rpc/request/$request_id"  
  Serial.print(F("Topic: ")); Serial.println(topic); Serial.print(F("Message: ")); Serial.println(ctrl_json);
                        
  DeserializationError error = deserializeJson(jsonBuffer, ctrl_json);
  if(error){
    Serial.println(F("parseObject() failed"));
    return;
  }            
  
  /**/
  if(jsonBuffer.containsKey ("Counter")) {
    int val = jsonBuffer["Counter"].as<int>();
    Serial.println("Attribute \"Counter\" change to " + String(val));
   }
  /**/
}

void setup() {
  Serial.begin(115200);

  // Kết nối WiFi
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
    Serial.println("Connecting to WiFi...");
  }
  Serial.println("Connected to WiFi");

  // Kết nối với máy chủ thông qua giao thức MQTT
  client.setServer(mqttServer, mqttPort);
  client.setCallback(tb_callback);  //đăng ký hàm callback để xử lý dữ liệu từ TB

  attr_counter = 0;
  timming = millis();
}

void loop() {
  if (!client.connected()) {
    reconnect();
  }
  else client.loop();
}

//hàm kiểm tra và thực hiện kết nối lại với máy chủ TB
void reconnect() {
  // kiểm tra kết nối với TB, nếu kết nối bị ngắt sẽ thực hiện kết nối lại với máy chủ
  while (!client.connected()) {
    Serial.print("Attempting MQTT connection...");
    
    // Attempt to connect
    if (client.connect("ESP32Client", accessToken, NULL)) {
      Serial.println("connected");
      delay(1000);

      Serial.println("\tsubscribe attribute on TB!");
      client.subscribe("v1/devices/me/attributes");
    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      
      // Wait 5 seconds before retrying
      delay(5000);
    }
  }
}