#include <WiFi.h>
#include "ThingsBoard.h"

#define COUNT_OF(x) ((sizeof(x)/sizeof(0[x])) / ((size_t)(!(sizeof(x) % sizeof(0[x])))))

#define CURRENT_FIRMWARE_TITLE    "TEST"
#define CURRENT_FIRMWARE_VERSION  "1.0.0"
#define TOKEN               "rUnsN9A3eWv1tEYiTZid"
#define THINGSBOARD_SERVER  "thingsboard.cloud"

#define WIFI_SSID           "Wokwi-GUEST"
#define WIFI_PASSWORD       ""
#define SERIAL_DEBUG_BAUD    115200

WiFiClient espClient;
ThingsBoard tb(espClient);
int status = WL_IDLE_STATUS;

uint8_t leds_control[] = {14, 13, 12};

#define pompa 12
int SW_State=0;

int quant = 20;
int led_delay = 1000;
int led_passed = 0;
bool subscribed = false;
int current_led = 0;

RPC_Response processDelayChange(const RPC_Data &data)
{
  Serial.println("Received the set delay RPC method");
  // Process data
  led_delay = data;

  Serial.print("Set new delay: ");
  Serial.println(led_delay);

  return String(led_delay);
}
RPC_Response processGetDelay(const RPC_Data &data)
{
  Serial.println("Received the get value method");
  return String(led_delay);
}

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

  int pin = data["pin"];
  bool enabled = data["enabled"];

  if (pin < COUNT_OF(leds_control)) {
    Serial.print("Setting LED ");
    Serial.print(pin);
    Serial.print(" to state ");
    Serial.println(enabled);

    digitalWrite(leds_control[pin], enabled);
  }

  return String("{\"" + String(pin) + "\": " + String(enabled?"true":"false") + "}");
  //return RPC_Response(data["pin"], (bool)data["enabled"]);
}

RPC_Response processGetGpioState(const RPC_Data &data)
{
  Serial.println("Received the get GPIO RPC method");
  String respStr = "{";
  
  for (size_t i = 0; i < COUNT_OF(leds_control); ++i) {
    int pin = leds_control[i];
    Serial.print("Getting LED ");
    Serial.print(pin);
    Serial.print(" state ");
    bool ledState = digitalRead(pin);
    Serial.println(ledState);

    respStr += String("\"" + String(i) + "\": " + String(ledState?"true":"false") + ", ");
  }  
  respStr = respStr.substring(0, respStr.length() - 2);
  respStr += "}";
  return respStr;
}

// RPC handlers
RPC_Callback callbacks[] = {
  { "setValue",         processDelayChange },
  { "getValue",         processGetDelay },  
  { "setGpioStatus",    processSetGpioState },
  { "getGpioStatus",    processGetGpioState },
};

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

void reconnect()
{
  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");
  }
}

void setup()
{
  Serial.begin(115200);
  Serial.begin(SERIAL_DEBUG_BAUD);
  InitWiFi();
  pinMode(pompa, OUTPUT);
  for (size_t i = 0; i < COUNT_OF(leds_control); ++i) {
    pinMode(leds_control[i], OUTPUT);
  }
}

void loop()
{
  delay(quant);

  led_passed += quant;

  if (WiFi.status() != WL_CONNECTED)
  {
    reconnect();
  }

  if (!tb.connected())
  {
    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;
    }
  }

  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;
  }  

  int kelembaban = analogRead(33);
  kelembaban = map(kelembaban, 0, 4095, 0, 100);

  Serial.println("Sending data...");
  tb.sendTelemetryFloat("Kelembaban", kelembaban);

  if (kelembaban < 60)
  {
    Serial.print("Tanah Kering,");
    Serial.print(" >> Kelembaban Tanah: ");
    Serial.print(kelembaban);
    Serial.println(" %");
    Serial.println("Pompa ON");
    digitalWrite(pompa, HIGH);
    tb.sendTelemetryString("Kondisi Tanah", "KERING");
    tb.sendTelemetryString("Status Pompa", "ON");
  }
  else if (kelembaban >= 60 && kelembaban <= 80 )
  {
    Serial.print("Tanah Ideal,");
    Serial.print(" >> Kelembaban Tanah: ");
    Serial.print(kelembaban);
    Serial.println(" %");
    Serial.println("Pompa OFF");
    digitalWrite(pompa, LOW);
    tb.sendTelemetryString("Kondisi Tanah", "IDEAL");
    tb.sendTelemetryString("Status Pompa", "OFF");
  }

  else 
  {
    Serial.print("Tanah Basah,");
    Serial.print(" >> Kelembaban Tanah: ");
    Serial.print(kelembaban);
    Serial.println(" %");
    Serial.println("Pompa OFF");
    digitalWrite(pompa, LOW);
    tb.sendTelemetryString("Kondisi Tanah", "BASAH");
    tb.sendTelemetryString("Status Pompa", "OFF");
  }
  delay(2000);

  tb.loop();
  
}