#include <Wire.h>
#include <LiquidCrystal_I2C.h>
#include <DHT.h>

#include<Firebase_ESP_Client.h>
#include "addons/TokenHelper.h"
#include "addons/RTDBHelper.h"

#define WIFI_SSID "Wokwi-GUEST"
#define WIFI_PASSWORD ""
#define API_KEY "AIzaSyD0l38qPbt1hKArNnIaDNqC3ZWrnP4p1vg"
#define DATABASE_URL "https://iot-app-leafy-1-default-rtdb.asia-southeast1.firebasedatabase.app/"

#define DHTPIN 4
#define DHTTYPE DHT22
#define LEDPIN 2

FirebaseData fbdo;
FirebaseAuth auth;
FirebaseConfig config;

unsigned long sendDataPrevMillis = 0;
bool signupOK =  false;
float soilMoisture = 0;
float temperature = 0;
float humidity = 0;
bool siram = false;

DHT dht(DHTPIN, DHTTYPE);
LiquidCrystal_I2C lcd(0x27, 16, 2);

// Fungsi untuk transformasi nilai kelembaban tanah
int transformSoilMoisture(int rawValue) {
  int minValue = 1680;
  int maxValue = 3620;
  return map(rawValue, minValue, maxValue, 0, 100);
}

void setup()
{
  Wire.begin(23, 22);
  Serial.begin(9600);
  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
  Serial.print("Connecting to Wi-Fi");
  while (WiFi.status() != WL_CONNECTED) {
    Serial.print(".");
    delay(300);
  }

  Serial.println();
  Serial.print("Connected with IP: ");
  Serial.println(WiFi.localIP());
  Serial.println();

  config.api_key = API_KEY;
  config.database_url = DATABASE_URL;
  if (Firebase.signUp (&config, &auth, "", "")) {
    Serial.println("signUp OK");
    signupOK = true;
  } else {
    Serial.printf("%s\n", config.signer.signupError.message.c_str());
  }

  config.token_status_callback = tokenStatusCallback;
  Firebase.begin(&config, &auth);
  Firebase.reconnectWiFi(true);

  lcd.init();
  lcd.backlight();
  pinMode(LEDPIN, OUTPUT);
  dht.begin();
}

void loop()
{
  if (Firebase.ready() && signupOK && (millis() - sendDataPrevMillis > 5000 || sendDataPrevMillis == 0)) {
    sendDataPrevMillis = millis();
    // ----Store data

    int16_t soilMoisture = analogRead(34);
    int soilMoisturePercent = transformSoilMoisture(soilMoisture);
    String soilMsg = soilMoisturePercent > 70 ? "WET" : soilMoisturePercent < 30 ? "DRY" : "OK";

    soilMoisture = soilMoisturePercent;
    humidity = dht.readHumidity();
    temperature = dht.readTemperature();

    if (Firebase.RTDB.setFloat(&fbdo, "sensor/soilMoisture", soilMoisture)) {
      Serial.println();
      Serial.print(soilMoisture);
      Serial.print ("successfully saved to: " + fbdo.dataPath());
      Serial.println(" (" + fbdo.dataType() + ")");
    } else {
      Serial.println("FAILED:" + fbdo.errorReason ());
    }

    if (Firebase.RTDB.setFloat(&fbdo, "sensor/humidity", humidity)) {
      Serial.println();
      Serial.print(humidity);
      Serial.print ("successfully saved to: " + fbdo.dataPath());
      Serial.println(" (" + fbdo.dataType() + ")");
    } else {
      Serial.println("FAILED:" + fbdo.errorReason ());
    }

    if (Firebase.RTDB.setFloat(&fbdo, "sensor/temperature", temperature)) {
      Serial.println();
      Serial.print(temperature);
      Serial.print ("successfully saved to: " + fbdo.dataPath());
      Serial.println(" (" + fbdo.dataType() + ")");
    } else {
      Serial.println("FAILED:" + fbdo.errorReason ());
    }


    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Soil: ");
    lcd.print(soilMoisturePercent);
    lcd.print("% ");
    lcd.print(soilMsg);
    lcd.setCursor(0, 1);
    lcd.print("T: ");
    lcd.print(temperature);
    lcd.print("C ");
    lcd.print("H: ");
    lcd.print(humidity);
    lcd.print("%");

    if (Firebase.RTDB.getBool(&fbdo, "control/pumpStatus")) {
      bool pumpStatus = fbdo.boolData();
      digitalWrite(LEDPIN, pumpStatus ? HIGH : LOW);
      Serial.print("LED Status: ");
      Serial.println(pumpStatus ? "ON" : "OFF");
    } else {
      Serial.println("FAILED to read LED status: " + fbdo.errorReason());
    }

    // bool newpumpStatus = (soilMoisturePercent < 30);
    // if (Firebase.RTDB.setBool(&fbdo, "control/pumpStatus", newpumpStatus)) {
    //   Serial.println(newpumpStatus ? "LED ON status saved to Firebase." : "LED OFF status saved to Firebase.");
    // } else {
    //   Serial.println("FAILED to write LED status to Firebase: " + fbdo.errorReason());
    // }

    // digitalWrite(LEDPIN, soilMoisturePercent < 30 ? HIGH : LOW);
  }
}