#include <WiFi.h>
#include <WiFiClient.h>
#include <PubSubClient.h>
#include <ArduinoJson.h>
#include <DHTesp.h>
#include <ESP32Servo.h>
#include <FastLED.h>
#include <Adafruit_SSD1306.h>
#include <Wire.h>
#include "RTClib.h"

#define Echo 18
#define Trig 2
#define Merah 14
#define Kuning 12
#define Hijau 13
#define buzzer 26

const char *ssid = "Wokwi-GUEST";
const char *password = "";
const char *mqtt_server = "test.mosquitto.org";
const char *mqtt_topic = "ThinkIOT/Distance";

WiFiClient espClient;
PubSubClient client(espClient);
unsigned long lastMsg = 0;

float jarak = 0;
int nada;

#define LED_PIN 4 // Pin strip LED WS2812
#define NUM_LEDS 16 // Jumlah LED dalam strip

CRGB leds[NUM_LEDS];
Servo myservo; // objek servo

const int DHT_PIN = 15;
DHTesp dht;

RTC_DS3231 rtc; // rtc class

const int RGB_PIN_R = 14; // Pin R untuk lampu RGB
const int RGB_PIN_G = 12; // Pin G untuk lampu RGB
const int RGB_PIN_B = 13; // Pin B untuk lampu RGB
const int RGB_CATHODE = 2; // Pin Cathode untuk lampu RGB

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

char daysOfTheWeek[7][12] = {"Minggu", "Senin", "Selasa", "Rabu", "Kamis", "Jumat", "Sabtu"};

void setup_wifi()
{
  delay(10);
  Serial.println();
  Serial.print("Menghubungkan ke ");
  Serial.println(ssid);

  WiFi.mode(WIFI_STA);
  WiFi.begin(ssid, password);

  while (WiFi.status() != WL_CONNECTED)
  {
    delay(500);
    Serial.print(".");
  }

  randomSeed(micros());

  Serial.println("");
  Serial.println("WiFi terhubung");
  Serial.println("Alamat IP: ");
  Serial.println(WiFi.localIP());
}

void ukur_jarak()
{
  digitalWrite(Trig, LOW);
  delay(10);
  digitalWrite(Trig, HIGH);
  delay(10);
  digitalWrite(Trig, LOW);

  int duration = pulseIn(Echo, HIGH);
  jarak = duration * 0.034 / 2 + 1;

  Serial.print("Jarak: ");
  Serial.print(jarak);
  Serial.println(" cm");
}

void output_jarak()
{
  if (jarak > 400)
  {
    Serial.println("Penuh");
    digitalWrite(Hijau, HIGH);
    client.publish("ThinkIOT/Distance/message", "Terisi Penuh!");
  }
  else if (jarak > 10 && jarak < 400)
  {
    Serial.println("Terisi");
    digitalWrite(Kuning, HIGH);
    client.publish("ThinkIOT/Distance/message", "Sedang Mengisi");
  }
  else if (jarak < 10)
  {
    Serial.println("Kosong");
    digitalWrite(Merah, HIGH);
    client.publish("ThinkIOT/Distance/message", "Kosong!");
  }
}


void pesan()
{
  if (jarak < 10)
  {
    tone(buzzer, nada);
    client.publish(mqtt_topic, "Tangki Kosong!");
  }
  else if (jarak > 10 && jarak < 400)
  {
    tone(buzzer, nada);
    client.publish(mqtt_topic, "Tangki Sedang Diisi");
  }
  else if (jarak > 400)
  {
    tone(buzzer, nada);
    client.publish(mqtt_topic, "Tangki Penuh!");
  }
  else
  {
    noTone(buzzer);
  }
}


void callback(char *topic, byte *payload, unsigned int length)
{
  // Handle MQTT messages received
  payload[length] = '\0'; // Null-terminate the payload
  String message = String((char *)payload);

  if (strcmp(topic, mqtt_topic) == 0)
  {
    Serial.print("Received message from MQTT: ");
    Serial.println(message);

    // Assuming the received message is a numeric value
    jarak = message.toFloat();
    Serial.print("Updated Jarak from MQTT: ");
    Serial.print(jarak);
    Serial.println(" cm");
  }

  String string;
  Serial.print("Pesan diterima [");
  Serial.print(topic);
  Serial.print("] ");
  for (int i = 0; i < length; i++)
  {
    string += ((char)payload[i]);
  }
  Serial.print(string);

  if (strcmp(topic, "/ThinkIOT/temp") == 0)
  {
    // Menangani data suhu di sini
    float temperature = string.toFloat();
    // Lakukan sesuatu dengan data suhu
  }
  else if (strcmp(topic, "/ThinkIOT/hum") == 0)
  {
    // Menangani data kelembaban di sini
    float humidity = string.toFloat();
    // Lakukan sesuatu dengan data kelembaban
  }

  if (strcmp(topic, "/ThinkIOT/Servo-nodered/bro") == 0)
  {
    // Menangani data kontrol servo di sini
    int status = string.toInt();
    int pos = map(status, 1, 100, 0, 180);
    Serial.println(pos);
    myservo.write(pos);
    delay(15);
  }

  if (strcmp(topic, "lights/neopixel") == 0)
  {
    int red, green, blue;
    sscanf(string.c_str(), "%d,%d,%d", &red, &green, &blue);
    Serial.print("Mengatur warna NeoPixel ke (R,G,B): ");
    Serial.print(red);
    Serial.print(",");
    Serial.print(green);
    Serial.print(",");
    Serial.println(blue);
    fill_solid(leds, NUM_LEDS, CRGB(red, green, blue));
    FastLED.show();
  }

  if (strcmp(topic, "lights/rgb") == 0)
  {
    // Menangani data untuk lampu RGB
    int red, green, blue;
    sscanf(string.c_str(), "%d,%d,%d", &red, &green, &blue);
    Serial.print("Mengatur warna lampu RGB ke (R,G,B): ");
    Serial.print(red);
    Serial.print(",");
    Serial.print(green);
    Serial.print(",");
    Serial.println(blue);
    analogWrite(RGB_PIN_R, red);
    analogWrite(RGB_PIN_G, green);
    analogWrite(RGB_PIN_B, blue);
    // Tambahkan fungsi untuk mengatur lampu RGB di sini sesuai dengan kebutuhan perangkat Anda
  }
  // ... (existing callback code)

  if (strcmp(topic, "oled/display") == 0)
  {
    // Menangani pesan untuk OLED display
    updateDisplay(message);
  }
}

void kirim_mqtt()
{
  // Publish Ultrasonic sensor data to the MQTT topic
  client.publish(mqtt_topic, String(jarak).c_str());
}

void updateDisplay(String message)
{
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(WHITE);
  display.setCursor(0, 0);

  // Tampilkan waktu
  displayTime();

  if (message == "")
  {
    // Tampilkan suhu dan kelembaban jika tidak ada pesan MQTT
    displayTime();
    displayTempAndHum();
  }
  else
  {
    // Menampilkan pesan dari MQTT
    display.println(message);
  }

  display.display();
}

void displayTime()
{
  DateTime now = rtc.now();
  display.clearDisplay();

  display.setTextSize(1);                
  display.setCursor(0, 0);
  display.setTextColor(SSD1306_WHITE);
  display.print("");
  display.setCursor(0, 0); // Start at top-left corner 
  display.print(daysOfTheWeek[now.dayOfTheWeek()]);
    display.print(" ");
    display.print(now.day(), DEC);
    display.print('/');
    display.print(now.month(), DEC);
    display.print('/');
    display.print(now.year(), DEC);

  // Kontrol efek kedip tampilan jam
  {
    display.println();
    display.println();
    display.setTextSize(2);
    display.print("  ");
    display.print(now.hour(), DEC);
    display.print(':');
    display.print(now.minute(), DEC);
    display.print(':');
    display.print(now.second(), DEC);
    display.println();
  }
 // Toggle nilai blink

  display.display();
}


void displayTempAndHum()
{
  TempAndHumidity data = dht.getTempAndHumidity();
  String temp = String(data.temperature, 2);
  String hum = String(data.humidity, 1);

  // Menampilkan data suhu & kelembapan
  display.setTextSize(1);
  display.setCursor(0, 40);
  display.print("Temp: ");
  display.print(temp);
  display.print(" C");
  display.setCursor(0, 50);
  display.print("Humidity: ");
  display.print(hum);
  display.print(" %");
}

void reconnect()
{
  while (!client.connected())
  {
    Serial.print("Mencoba koneksi MQTT...");
    String clientId = "ESP32Client-";
    clientId += String(random(0xffff), HEX);
    if (client.connect(clientId.c_str()))
    {
      Serial.println("Terhubung");
      client.subscribe("/ThinkIOT/temp");
      client.subscribe("/ThinkIOT/hum");
      client.subscribe("/ThinkIOT/Servo-nodered/bro");
      client.subscribe("lights/neopixel");
      client.subscribe("lights/rgb");
      client.subscribe("oled/display");
    }
    else
    {
      Serial.print("gagal, rc=");
      Serial.print(client.state());
      Serial.println(" coba lagi dalam 5 detik");
      delay(5000);
    }
  }
}

void setup()
{
  pinMode(2, OUTPUT);
  Serial.begin(115200);
  setup_wifi();
  client.setServer(mqtt_server, 1883);
  client.setCallback(callback);
  dht.setup(DHT_PIN, DHTesp::DHT22);
  myservo.attach(5);

  // Setup strip LED WS2812
  FastLED.addLeds<WS2812, LED_PIN, GRB>(leds, NUM_LEDS);

  // Setup OLED display
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();

  // RTC setup
  if (!rtc.begin())
  {
    Serial.println("Couldn't find RTC");
    while (1)
      ;
  }

  if (rtc.lostPower())
  {
    Serial.println("RTC lost power, lets set the time!");
    rtc.adjust(DateTime(F(__DATE__), F(__TIME__))); // set time from computer time
  }

  // Setup pins for RGB LED
  pinMode(RGB_PIN_R, OUTPUT);
  pinMode(RGB_PIN_G, OUTPUT);
  pinMode(RGB_PIN_B, OUTPUT);
  pinMode(RGB_CATHODE, OUTPUT);
  digitalWrite(RGB_CATHODE, HIGH); // For common anode, change to LOW for common cathode
}

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

  unsigned long now = millis();
  if (now - lastMsg > 2000)
  {
    lastMsg = now;
    TempAndHumidity data = dht.getTempAndHumidity();

    String temp = String(data.temperature, 2);
    client.publish("/ThinkIOT/temp", temp.c_str());

    String hum = String(data.humidity, 1);
    client.publish("/ThinkIOT/hum", hum.c_str());

    DateTime now = rtc.now();
    String timeStr = String(now.day()) + "/" + String(now.month()) + "/" + String(now.year()) + " " +
                     String(now.hour()) + ":" + String(now.minute()) + ":" + String(now.second());
    client.publish("/ThinkIOT/rtc", timeStr.c_str());

    Serial.print("Suhu: ");
    Serial.println(temp);
    Serial.print("Kelembaban: ");
    Serial.println(hum);
    Serial.print("Waktu RTC: ");
    Serial.println(timeStr);

  // Panggil fungsi updateDisplay() untuk memperbarui tampilan layar OLED
    updateDisplay("");
  }
  // Panggil fungsi displayTime() untuk memperbarui tampilan layar OLED dengan waktu saat ini
    displayTime();
    displayTempAndHum();

  // Additional code for Ultrasonic sensor
    ukur_jarak();
    output_jarak();
    //pesan();
    kirim_mqtt();
    
}