#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 get_device_attributes(){
  client.subscribe("v1/devices/me/attributes/response/+");
  //Serial.println("Request to get attributes value");
  
  // Prepare a JSON payload string 
  tb_payload = "{\"clientKeys\":\"Counter\",\"sharedKeys\":\"key1\"}"; 
  //Serial.println(tb_payload);
  tb_payload.toCharArray( data_buffer, sizeof(data_buffer) );
  client.publish( "v1/devices/me/attributes/request/0", data_buffer );
}

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 ("attr_counter")) {
    int val = jsonBuffer["attr_counter"].as<int>();
    Serial.println("Attribute \"Key1\" 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();

  //gửi yêu cầu xác định giá trị của thuộc tính trên Thingsboard
  if(millis() - timming > 10000){
    timming = millis();
    get_device_attributes();
  }  
}

//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");
    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      
      // Wait 5 seconds before retrying
      delay(5000);
    }
  }
}