#include <WiFi.h>
#include <FirebaseESP32.h>
#include <addons/TokenHelper.h>
#include <addons/RTDBHelper.h>

#define button_pin      34
#define buzzer_pin      19

#define WIFI_SSID "network"
#define WIFI_PASSWORD "123456789"
#define API_KEY "AIzaSyAjIS98IBo4xU9DAqFbo6ZJLPqBwzrJvzk"
#define USER_EMAIL "smartpostcar@gmail.com"
#define USER_PASSWORD "123456789"
#define DATABASE_URL "postcar-57fe1-default-rtdb.firebaseio.com"
#define DATABASE_SECRET "7yJzXNiqlkQZUFb3Ul2KJACDHnztTMrDoFLgPOZM"


FirebaseData fbdo;
FirebaseAuth auth;
FirebaseConfig config;
unsigned long dataMillis = 0;

int button_pressed_flag = 0;
int button_state = 0;
int home1_value_from_DB = 0;
int home2_value_from_DB = 0;
int home3_value_from_DB = 0;

void setup()
{
  Serial.begin(115200);
  setup_inputs_and_outputs();
  connect_to_WiFi_and_Firebase_database();
}

void loop()
{
  read_button_and_send_data_to_database();
  read_from_database();
}


void read_button_and_send_data_to_database() {
  button_state = digitalRead(button_pin);
  if (button_state == 1 && button_pressed_flag == 0) {
    digitalWrite(buzzer_pin, 1);
    button_pressed_flag = 1;
    write_to_database();
    button_pressed_flag = 0;
    digitalWrite(buzzer_pin, 0);
  }
}

void setup_inputs_and_outputs() {
  pinMode(button_pin, INPUT);
  pinMode(buzzer_pin, OUTPUT);
  digitalWrite(buzzer_pin, LOW);
}

void connect_to_WiFi_and_Firebase_database() {
  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();
  Serial.printf("Firebase Client v%s\n\n", FIREBASE_CLIENT_VERSION);
  config.api_key = API_KEY;
  auth.user.email = USER_EMAIL;
  auth.user.password = USER_PASSWORD;
  config.database_url = DATABASE_URL;
  Firebase.reconnectWiFi(true);
  fbdo.setResponseSize(4096);
  String base_path = "/UsersData/";
  config.token_status_callback = tokenStatusCallback; // see addons/TokenHelper.h
  config.max_token_generation_retry = 5;
  Firebase.begin(&config, &auth);
  String var = "$userId";
  String val = "($userId === auth.uid && auth.token.premium_account === true && auth.token.admin === true)";
  Firebase.setReadWriteRules(fbdo, base_path, var, val, val, DATABASE_SECRET);
}

void read_from_database() {

  if (millis() - dataMillis > 5000 && Firebase.ready()) {
    dataMillis = millis();
    //////////////////////////////////////////////////////////
    //home_1
    if (Firebase.RTDB.getInt(&fbdo, "/home1")) {
      if (fbdo.dataType() == "int") {
        home1_value_from_DB = fbdo.intData();
        Serial.print("home1_value_from_DB = ");
        Serial.println(home1_value_from_DB);
      }
    }
    else {
      Serial.println(fbdo.errorReason());
    }
    //////////////////////////////////////////////////////////
    //////////////////////////////////////////////////////////
    //home_2
    if (Firebase.RTDB.getInt(&fbdo, "/home2")) {
      if (fbdo.dataType() == "int") {
        home2_value_from_DB = fbdo.intData();
        Serial.print("home2_value_from_DB = ");
        Serial.println(home2_value_from_DB);
      }
    }
    else {
      Serial.println(fbdo.errorReason());
    }
    //////////////////////////////////////////////////////////
    //////////////////////////////////////////////////////////
    //home_3
    if (Firebase.RTDB.getInt(&fbdo, "/home3")) {
      if (fbdo.dataType() == "int") {
        home3_value_from_DB = fbdo.intData();
        Serial.print("home3_value_from_DB = ");
        Serial.println(home3_value_from_DB);
      }
    }
    else {
      Serial.println(fbdo.errorReason());
    }
    //////////////////////////////////////////////////////////
  }


}

void write_to_database() {
  Serial.printf("Set int... %s\n", Firebase.setInt(fbdo, "/home1", 0) ? "ok" : fbdo.errorReason().c_str());
  Serial.printf("Set int... %s\n", Firebase.setInt(fbdo, "/home2", 0) ? "ok" : fbdo.errorReason().c_str());
  Serial.printf("Set int... %s\n", Firebase.setInt(fbdo, "/home3", 0) ? "ok" : fbdo.errorReason().c_str());
}