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

// WIFI config
const char* ssid = "Wokwi-GUEST";
const char* password =  "";
WiFiClient wifiClient;

// MQTT client config
const char* mqttServer = "broker.hivemq.com";
const int mqttPort = 1883;
const char* mqttUser = "";
const char* mqttPassword = "";
const char* topic_pub = "fah/Lab/counter";
PubSubClient mqttClient(wifiClient);

// some variables buffer
String publishMessage;
String inputText = ""; // ตัวแปรสำหรับเก็บข้อความที่รับจาก Serial

void setup_wifi() {
  // Start by connecting to a WiFi network
  Serial.begin(115200);
  Serial.println();
  Serial.print("Connecting to ");
  Serial.println(ssid);
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println();
  Serial.print("WiFi connected on ");
  Serial.print("IP address: ");
  Serial.print(WiFi.localIP());
  Serial.print(" MAC address: ");
  Serial.println(WiFi.macAddress());
}

void reconnectMQTT() {
  // Loop until we're reconnected
  char mqttinfo[80];
  char clientId[100] = "\0";
  snprintf (mqttinfo, 75, "Attempting MQTT connection at %s:%d (%s/%s)...", mqttServer, mqttPort, mqttUser, mqttPassword);
  while (!mqttClient.connected()) {
    Serial.println(mqttinfo);
    sprintf(clientId, "ESP32Client-%04X", random(0xffff));
    if (mqttClient.connect(clientId)) {
      Serial.println("...MQTT Broker connected...");
    } else {
      Serial.print("failed, rc=");
      Serial.print(mqttClient.state());
      Serial.println(" try again in 5 seconds");
      delay(5000);
    }
  }
}

void setup() {
  Serial.begin(115200);
  setup_wifi();
  mqttClient.setServer(mqttServer, mqttPort);
}

void loop() {
  if (!mqttClient.connected()) {
    reconnectMQTT(); // Reconnect to MQTT if disconnected
  }
  mqttClient.loop();

  // เช็คว่า Serial Monitor มีข้อมูลเข้าไหม
  if (Serial.available() > 0) {
    inputText = Serial.readStringUntil('\n');  // รับข้อความจาก Serial Monitor จนกว่าจะเจอ '\n'
    inputText.trim();  // ลบช่องว่างข้างหน้าและข้างหลัง

    Serial.print("Received: ");
    Serial.println(inputText);  // แสดงข้อความที่ได้รับจาก Serial
  }

  // prepare the message/payload
  publishMessage = inputText;  // ใช้ข้อความที่ได้รับจาก Serial

  // send message to MQTT topic
  mqttClient.publish(topic_pub, publishMessage.c_str());

  // debug
  Serial.println(publishMessage);  // print value to serial monitor

  delay(2000); // this speeds up the simulation, 2 seconds delay
}