#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <ESP32Servo.h>

#define SDA_PIN 18
#define SCL_PIN 19
#define SOIL_SENSOR_PIN 34
#define LED_RED_PIN 2
#define LED_BLUE_PIN 4
#define SERVO_PIN 15
#define SENSOR_POWER_PIN 21
#define WATER_PUMP_PIN 22
#define MANUAL_CONTROL_PIN 27

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

Servo servo;

bool manualControlActivated = false;

void setup() {
  pinMode(LED_RED_PIN, OUTPUT);
  pinMode(LED_BLUE_PIN, OUTPUT);
  pinMode(SENSOR_POWER_PIN, OUTPUT);
  pinMode(WATER_PUMP_PIN, OUTPUT);
  pinMode(MANUAL_CONTROL_PIN, INPUT_PULLUP);

  digitalWrite(SENSOR_POWER_PIN, LOW);
  
  Wire.begin(SDA_PIN, SCL_PIN);
  Serial.begin(115200);
  Serial.println("Hello, start configuration!");

  servo.attach(SERVO_PIN);
  servo.write(0);

  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }

  display.display();
  delay(2000);
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
}

void loop() {
  int16_t moisture = analogRead(SOIL_SENSOR_PIN);
  String message = "";

  if (moisture < 2165) {
    message = "😟"; // Emoji sedih
    digitalWrite(LED_RED_PIN, LOW);
    digitalWrite(LED_BLUE_PIN, LOW);
    servo.write(0);
    digitalWrite(WATER_PUMP_PIN, LOW);
  } else if (moisture > 3135) {
    message = "😔"; // Emoji khawatir
    digitalWrite(LED_RED_PIN, HIGH);
    digitalWrite(LED_BLUE_PIN, LOW);
    servo.write(0);
    if (manualControlActivated) {
      digitalWrite(WATER_PUMP_PIN, HIGH); // Aktifkan water pump hanya jika manual control diaktifkan
    } else {
      digitalWrite(WATER_PUMP_PIN, LOW);
    }
  } else {
    message = "😊"; // Emoji senang
    digitalWrite(LED_RED_PIN, LOW);
    digitalWrite(LED_BLUE_PIN, HIGH);
    servo.write(0);
    digitalWrite(WATER_PUMP_PIN, LOW);
  }

  // Cek status tombol manual control
  if (digitalRead(MANUAL_CONTROL_PIN) == LOW) {
    manualControlActivated = true;
  } else {
    manualControlActivated = false;
  }

  display.clearDisplay();
  display.setCursor(0,0);
  display.print("Moisture: ");
  display.print(message);
  display.display();
  delay(1000);
}