#include <WiFi.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include "ThingsBoard.h"
#include "DHTesp.h"
#include <ESP32Servo.h>

// #define CURRENT_FIRMWARE_TITTLE   "TEST"
// #define CURRENT_FIRMWARE_VERSION  "1.0.0"
#define WIFI_SSID       "Wokwi-GUEST"
#define WIFI_PASSWORD   ""
#define TOKEN               "WAcFK1e4h71AqMUr06H0"
#define THINGSBOARD_SERVER  "thingsboard.cloud"
#define SERIAL_DEBUG_BAUD 115200

#define LED_MODE_INDICATOR 14
#define RELAY 2

// Helper macro to calculate array size
#define COUNT_OF(x) ((sizeof(x)/sizeof(0[x])) / ((size_t)(!(sizeof(x) % sizeof(0[x])))))

int status = WL_IDLE_STATUS;

// Set to true if application is subscribed for the RPC messages.
bool subscribed = false;

// Initial servo value
int servo_value = 0;
int servo_value_percentage = 0;
bool enabled = false;

#define LDR_PIN  34
#define DHT_PIN 15

String kondisi;
String mode;
int ldrValue;
int ldrPercentage;
float temperature;
float humidity;

DHTesp  dhtSensor;
WiFiClient  espClient;
ThingsBoard tb(espClient);
Servo myservo;
Adafruit_SSD1306 display(128, 64, &Wire, -1);

// Processes function for RPC call "setValue"
// RPC_Data is a JSON variant, that can be queried using operator[]
// See https://arduinojson.org/v5/api/jsonvariant/subscript/ for more details
RPC_Response processValueChange(const RPC_Data &data)
{
  Serial.println("Received the set value RPC method");

  // Process data

  servo_value = data;

  Serial.print("value = ");
  Serial.println(servo_value);

  return RPC_Response(NULL, servo_value);
}

// Processes function for RPC call "getValue"
// RPC_Data is a JSON variant, that can be queried using operator[]
// See https://arduinojson.org/v5/api/jsonvariant/subscript/ for more details
RPC_Response processGetValue(const RPC_Data &data)
{
  Serial.println("Received the get value method");

  return RPC_Response(NULL, servo_value);
}

RPC_Response processSetSwitch(const RPC_Data &data)
{
  Serial.println("Received the set value RPC method");

  // Process data

  enabled = data;

  Serial.print("value = ");
  Serial.println(enabled);

  return RPC_Response(NULL, enabled);
}

RPC_Response processGetSwitch(const RPC_Data &data)
{
  Serial.println("Received the get value method");

  return RPC_Response(NULL, enabled);
}

void InitWiFi()
{
  Serial.println("Connecting to AP ...");
  // attempt to connect to WiFi network

  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
  while (WiFi.status() != WL_CONNECTED)
  {
    delay(500);
    Serial.print(".");
  }
  Serial.println("Connected to AP");
}

void reconnect()
{
  // Loop until we're reconnected
  status = WiFi.status();
  if ( status != WL_CONNECTED)
  {
    WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
    while (WiFi.status() != WL_CONNECTED)
    {
      delay(500);
      Serial.print(".");
    }
    Serial.println("Connected to AP");
  }
}

// RPC handlers
RPC_Callback callbacks[] = {
  { "setValue", processValueChange },
  { "getValue", processGetValue },
  { "setSwitchValue", processSetSwitch },
  { "getSwitchValue", processGetSwitch },
};

void setup()
{
  // initialize serial for debugging
  Serial.begin(SERIAL_DEBUG_BAUD);
  Serial.println();
  pinMode(LED_MODE_INDICATOR, OUTPUT);
  pinMode(RELAY, OUTPUT);
  InitWiFi();
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
  myservo.attach(12);
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
}

void loop()
{
  delay(1000);
  if (WiFi.status() != WL_CONNECTED)
  {
    reconnect();
    return;
  }

  if (!tb.connected())
  {
    // Connect to the ThingsBoard
    Serial.print("Connecting to: ");
    Serial.print(THINGSBOARD_SERVER);
    Serial.print(" with token ");
    Serial.println(TOKEN);
    if (!tb.connect(THINGSBOARD_SERVER, TOKEN))
    {
      Serial.println("Failed to connect");
      return;
    }
  }

  // Subscribe for RPC, if needed
  if (!subscribed) {
    Serial.println("Subscribing for RPC...");

    // Perform a subscription. All consequent data processing will happen in
    // callbacks as denoted by callbacks[] array.
    if (!tb.RPC_Subscribe(callbacks, COUNT_OF(callbacks))) {
      Serial.println("Failed to subscribe for RPC");
      return;
    }

    Serial.println("Subscribe done");
    subscribed = true;
  }

  Serial.println("Sending data...");

  // Uploads new telemetry to ThingsBoard using MQTT.
  // See https://thingsboard.io/docs/reference/mqtt-api/#telemetry-upload-api
  // for more details

  lightIntensity();
  TempAndHumidity  data = dhtSensor.getTempAndHumidity();
  temperature = data.temperature;
  humidity = data.humidity;
  servo_value_percentage = map(servo_value, 0, 180, 0, 100);
  tb.sendTelemetryFloat("temperature", temperature);
  tb.sendTelemetryFloat("humidity", humidity);
  tb.sendTelemetryInt("light percentage", ldrPercentage);
  tb.sendTelemetryInt("bukaan atap", servo_value_percentage);
  tb.sendTelemetryString("kondisi cuaca", kondisi.c_str());
  Serial.print("Suhu : ");
  Serial.print(temperature);
  Serial.print(" Kelembaban : ");
  Serial.println(humidity);
  Serial.print("Bukaan Servo : ");
  Serial.println(servo_value);

  updateDisplay();

  switch (enabled){
    case 0:
      // Serial.println("mode otomatis");
      digitalWrite(LED_MODE_INDICATOR, enabled);
      controlServo();
      mode = "Otomatis";
    break;

    case 1:
      // Serial.println("masuk mode manual");
      digitalWrite(LED_MODE_INDICATOR, enabled);
      myservo.write(servo_value);
      mode = "Manual";
      // delay(1000);
    break;
  }

  if(humidity < 80){
    digitalWrite(RELAY, HIGH);
  }
  else{
    digitalWrite(RELAY, LOW);
  } 

  tb.loop();
}

void controlServo(){
  if (ldrPercentage > 80)
      {
        if (temperature > 26){
          servo_value =0;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature > 24 && temperature <= 26){
          servo_value = 22;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature <= 24){
          servo_value =45; 
          myservo.write(servo_value); 
          delay(15);
        }
        kondisi = "Sangat Terik";      
      }
  else if (ldrPercentage > 60 && ldrPercentage <= 80)
  {
          if (temperature > 26){
          servo_value = 45;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature > 24 && temperature <= 26){
          servo_value = 67;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature <= 24){
          servo_value = 90;
          myservo.write(servo_value); 
          delay(15);
        }
        kondisi = "Terik";
  }
  else if (ldrPercentage > 40 && ldrPercentage <= 60)
  {
        if (temperature > 26){
          servo_value = 90;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature > 24 && temperature <= 26){
          servo_value = 112;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature <= 24){
          servo_value = 135;
          myservo.write(servo_value); 
          delay(15);
        }
        kondisi = "Cerah";
  }
  else if (ldrPercentage > 20 && ldrPercentage <= 40)
  {
        if (temperature > 26){
          servo_value = 135;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature > 24 && temperature <= 26){
          servo_value = 157;
          myservo.write(servo_value); 
          delay(15);
        }
        else if(temperature <= 24){
          servo_value = 180;
          myservo.write(servo_value); 
          delay(15);
        }
        kondisi = "Mendung";
  }
  else {
        servo_value = 180;
        myservo.write(servo_value); 
        delay(15);
        kondisi = "Gelap";
  }

}

void lightIntensity(){
  ldrValue = analogRead(LDR_PIN);
  ldrPercentage = map(ldrValue, 4063, 32, 0, 100);
  Serial.print("LDR Value = ");
  Serial.println(ldrPercentage);
  if (ldrPercentage < 20 ) {
    Serial.print("Light Condition");
    Serial.println(" => Dark");
  } else if (ldrPercentage < 40) {
    Serial.print("Light Condition");
    Serial.println(" => Dim");
  } else if (ldrPercentage < 60) {
    Serial.print("Light Condition");
    Serial.println(" => Light");
  } else if (ldrPercentage < 80) {
    Serial.print("Light Condition");
    Serial.println(" => Bright");
  } else {
    Serial.print("Light Condition");
    Serial.println(" => Very Bright");
  }
}

void updateDisplay() {
  display.clearDisplay();
  display.setFont();
  display.setTextColor(1);
  
  display.setCursor(0, 10);
  display.print("Temperature(C) = ");
  display.setCursor(100, 10);
  display.print(temperature);

  display.setCursor(0, 20);
  display.print("Humidity(%) = ");
  display.setCursor(80, 20);
  display.print(humidity);

  display.setCursor(0, 30);
  display.print("Light(%) = ");
  display.setCursor(65, 30);
  display.print(ldrPercentage);

  display.setCursor(0, 40);
  display.print("Cuaca ");
  display.setCursor(35, 40);
  display.print(kondisi);

  display.setCursor(0, 50);
  display.print("Mode ");
  display.setCursor(30, 50);
  display.print(mode);

  display.display();
}