/**
 * Basic Write Example code for InfluxDBClient library for Arduino
 * Data can be immediately seen in a InfluxDB UI: wifi_status measurement
 * Enter WiFi and InfluxDB parameters below
 *
 * Measures signal level of the actually connected WiFi network
 * This example supports only InfluxDB running from unsecure (http://...)
 * For secure (https://...) or Influx Cloud 2 use SecureWrite example
 **/

#include <Arduino.h>
#include <WiFiMulti.h>
#include <InfluxDbClient.h>
#include <InfluxDbCloud.h>
#include <DHT.h>
#include <DHT_U.h>

// Project Macros
#define WIFI_SSID             "Wokwi-GUEST"
#define WIFI_PASSWORD         ""
#define INFLUXDB_URL          "https://eu-central-1-1.aws.cloud2.influxdata.com/"
#define INFLUXDB_TOKEN        "srSQ3YtBUduTFS12EJeeQq16mbzWlMXWRweU1PGF9a2oQhVXhk_FN2787uYLjDoa5dfFuPQSLZ9Ryp9pEXwzNw=="
#define INFLUXDB_ORG          "9e452ef9c0ca8ebe"
#define INFLUXDB_BUCKET       "mdp_iot"

// Set timezone string according to https://www.gnu.org/software/libc/manual/html_node/TZ-Variable.html
// Examples:
//  Pacific Time: "PST8PDT"
//  Eastern: "EST5EDT"
//  Japanesse: "JST-9"
//  Central Europe: "CET-1CEST,M3.5.0,M10.5.0/3"
#define TZ_INFO "CET-1CEST,M3.5.0,M10.5.0/3"

#define DEVICE                          "ESP32"
#define NPM                             "2125250012"

// DHT Sensor Configuration Macros
#define DHTPIN                          (22)
#define DHTTYPE                         (DHT22)

#define DHT11_REFRESH_TIME              (2000u)
#define INFLUXDB_SEND_TIME              (10000u)

#define ADC1_PIN              (36)   // ADC pin 1
#define ADC2_PIN              (39)   // ADC pin 2
#define ADC3_PIN              (34)   // ADC pin 3
#define ADC4_PIN              (35)   // ADC pin 4

#define BTN1_PIN              (25)   // Button pin 1
#define BTN2_PIN              (26)   // Button pin 2
#define BTN3_PIN              (27)   // Button pin 3
#define BTN4_PIN              (14)   // Button pin 4

#define DATA_TRANSMIT_INTERVAL (500u)

// Priavate Variables

// InfluxDB client instance with preconfigured InfluxCloud Certificate
InfluxDBClient client(INFLUXDB_URL, INFLUXDB_ORG, INFLUXDB_BUCKET, INFLUXDB_TOKEN, InfluxDbCloud2CACert);

// Data point
Point sensor("Sensor_Data");

WiFiMulti wifiMulti;

static uint8_t dht11_temperature = 0;
static uint8_t dht11_humidity = 0u;
DHT dht(DHTPIN, DHTTYPE);

// Task Time related Variables
static uint32_t dht_refresh_timestamp = 0u;
static uint32_t influxdb_send_timestamp = 0u;

// Private Function Prototypes
static void System_Init( void );
static void WiFi_Setup( void );
static void DHT11_TaskInit( void );
static void DHT11_TaskMng( void );
static void InfluxDB_TaskInit( void );
static void InfluxDB_TaskMng( void );

static uint32_t data_transmit_timestamp = 0u;

void setup()
{
  System_Init();
  WiFi_Setup();
  DHT11_TaskInit();
  InfluxDB_TaskInit();
  dht.begin();
}

void loop()
{
  uint32_t now = millis();
  DHT11_TaskMng();
  InfluxDB_TaskMng();
  // ADC_And_Button_TaskMng();

  if (now - data_transmit_timestamp >= DATA_TRANSMIT_INTERVAL)
  {
    data_transmit_timestamp = now; // Perbarui timestamp
    ADC_And_Button_TaskMng();      // Panggil fungsi untuk membaca data ADC dan tombol
  }
}

// Private Function Definitions
static void System_Init( void )
{
  Serial.begin( 115200 );
  // TODO: XS
  pinMode(ADC1_PIN, INPUT);
  pinMode(ADC2_PIN, INPUT);
  pinMode(ADC3_PIN, INPUT);
  pinMode(ADC4_PIN, INPUT);
  pinMode(BTN1_PIN, INPUT_PULLUP);
  pinMode(BTN2_PIN, INPUT_PULLUP);
  pinMode(BTN3_PIN, INPUT_PULLUP);
  pinMode(BTN4_PIN, INPUT_PULLUP);
}

static void WiFi_Setup( void )
{
  // Connect WiFi
  Serial.println("Connecting to WiFi");
  WiFi.mode(WIFI_STA);
  wifiMulti.addAP(WIFI_SSID, WIFI_PASSWORD);
  while (wifiMulti.run() != WL_CONNECTED)
  {
    Serial.print(".");
    delay(500);
  }
  Serial.println();
}

static void DHT11_TaskInit( void )
{
  dht.begin();
  // delay(2000);
  dht_refresh_timestamp = millis();
}

static void DHT11_TaskMng( void )
{
  uint32_t now = millis();
  float temperature, humidity;
  if( now - dht_refresh_timestamp >= DHT11_REFRESH_TIME )
  {
    dht_refresh_timestamp = now;
    // Reading temperature or humidity takes about 250 milliseconds!
    // Sensor readings may also be up to 2 seconds 'old' (its a very slow sensor)
    humidity = dht.readHumidity();
    // Read temperature as Celsius (the default)
    temperature = dht.readTemperature();
    
    // Check if any reads failed and exit early (to try again).
    if (isnan(humidity) || isnan(temperature) ) 
    {
      Serial.println(F("Failed to read from DHT sensor!"));
    }
    else
    {
      Serial.print(F("Humidity: "));
      Serial.print(humidity);
      Serial.print(F("%  Temperature: "));
      Serial.print(temperature);
      Serial.println(F("°C "));
      // sensor.addField("temperature", temperature);
      // sensor.addField("humidity", humidity);
      // store this in the global variables
      dht11_humidity = (uint8_t)humidity;
      dht11_temperature = (uint8_t)temperature;
    }
  }
}

static void InfluxDB_TaskInit( void )
{
  // Add constant tags - only once
  sensor.addTag("device", DEVICE);
  sensor.addTag("SSID", WIFI_SSID);
  sensor.addTag("NPM", NPM);

  // Accurate time is necessary for certificate validation & writing in batches
  // For the fastest time sync find NTP servers in your area: 
  // https://www.pool.ntp.org/zone/
  // Syncing progress and the time will be printed to Serial.
  timeSync(TZ_INFO, "pool.ntp.org", "time.nis.gov");

  // Check server connection
  if (client.validateConnection()) 
  {
    Serial.print("Connected to InfluxDB: ");
    Serial.println(client.getServerUrl());
  }
  else
  {
    Serial.print("InfluxDB connection failed: ");
    Serial.println(client.getLastErrorMessage());
  }
}

static void InfluxDB_TaskMng( void )
{
  uint32_t now = millis();
  if( now - influxdb_send_timestamp >= INFLUXDB_SEND_TIME )
  {
    influxdb_send_timestamp = now;
    // Store measured value into point
    sensor.clearFields();
    // Report RSSI of currently connected network
    sensor.addField( "rssi", WiFi.RSSI() );
    // add temperature and humidity values also
    sensor.addField( "temperature", dht11_temperature );
    sensor.addField( "humidity", dht11_humidity );
    
    // Print what are we exactly writing
    Serial.print("Writing: ");
    Serial.println(client.pointToLineProtocol(sensor));
    // If no Wifi signal, try to reconnect it
    if (wifiMulti.run() != WL_CONNECTED)
    {
      Serial.println("Wifi connection lost");
    }
    // Write point
    if (!client.writePoint(sensor))
    {
      Serial.print("InfluxDB write failed: ");
      Serial.println(client.getLastErrorMessage());
    }
  }
}

static void ADC_And_Button_TaskMng(void) {
  // Baca status tombol
  int btn1 = digitalRead(BTN1_PIN);
  int btn2 = digitalRead(BTN2_PIN);
  int btn3 = digitalRead(BTN3_PIN);
  int btn4 = digitalRead(BTN4_PIN);

  // Tambahkan debug untuk memastikan nilai tombol terbaca
  Serial.printf("Raw Button States: BTN1=%d, BTN2=%d, BTN3=%d, BTN4=%d\n", btn1, btn2, btn3, btn4);

  // Tambahkan nilai tombol ke payload
  sensor.addField("Button1", btn1 == LOW ? 1 : 0);
  sensor.addField("Button2", btn2 == LOW ? 1 : 0);
  sensor.addField("Button3", btn3 == LOW ? 1 : 0);
  sensor.addField("Button4", btn4 == LOW ? 1 : 0);

   Serial.println("Data transmitted...");

  // Debug hasil akhir
  // Serial.printf("Processed Buttons: %d, %d, %d, %d\n", (btn1 == LOW ? 1 : 0), (btn2 == LOW ? 1 : 0), (btn3 == LOW ? 1 : 0), (btn4 == LOW ? 1 : 0));
}