#include <WiFi.h>
#include <MQTT.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define SCREEN_WIDTH 128 // OLED width, in pixels
#define SCREEN_HEIGHT 64 // OLED height, in pixels

// Create an OLED display object connected to I2C
Adafruit_SSD1306 oled(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

WiFiClient net;
MQTTClient Client;

const char ssid[]="Wokwi-GUEST";
const char pass[]="";

const int pinRed   = 2;
const int pinGreen = 4;
const int pinBlue  = 17;
const int ledMerah = 14; 

void setup() {
  pinMode(pinRed, OUTPUT);
  pinMode(pinGreen, OUTPUT);
  pinMode(pinBlue, OUTPUT);
  pinMode(ledMerah, OUTPUT);
  Serial.begin(9600);

   // Initialize OLED display with I2C address 0x3C
 if (!oled.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("Failed to start SSD1306 OLED"));
    while (1);
 }

 delay(2000);         // Wait two seconds for initializing
 oled.clearDisplay(); // Clear display
 oled.setTextSize(1);         // Set text size
 oled.setTextColor(WHITE);    // Set text color
 oled.setCursor(12, 30);       // Set position to display (x,y)
 oled.println("Welcome to MQTT"); // Set text
 oled.display();              // Display on OLED

  WiFi.begin(ssid,pass);
  Client.begin("broker.emqx.io", net);
  Client.onMessage(subscribe);
  reconnect();
  // digitalWrite(ledMerah,1);
}

void loop() {
  Client.loop();
  if(!Client.connected()){
    reconnect();
  }
  // put your main code here, to run repeatedly:
  delay(10); // this speeds up the simulation
}

void rgb(bool red, bool green, bool blue){
  digitalWrite(pinRed, red);
  digitalWrite(pinGreen, green);
  digitalWrite(pinBlue, blue);
}

void reconnect(){
  rgb(1,0,0); //merah
  // Serial.println("merah");
  while (WiFi.status() != WL_CONNECTED){
    delay(500);
  }
 rgb(0,1,0);
 while (!Client.connect("Client-connnect-01")){
  delay(500);
 }
 rgb(0,0,1); //biru
 Client.subscribe("belajar/#", 1);
}

void subscribe(String &topic, String &data){

  if(topic == "belajar/led"){
    data.trim();
    if(data=="ON"){
      digitalWrite(ledMerah,1);
    } 
    else if(data=="OFF"){
      digitalWrite(ledMerah,0);
    }
  }
//   if (topic == "belajar/led") {
//   data.trim(); // Hapus spasi di awal/akhir data
//   Serial.print("Data = ");
//   Serial.println(data);
//   if (data == "ON") {
//     Serial.println("Turning LED ON");
//     digitalWrite(ledMerah, HIGH);
//   } else if (data == "OFF") {
//     Serial.println("Turning LED OFF");
//     digitalWrite(ledMerah, LOW);
//   }
// }
}