// LIBRARIES
#include <ArduinoJson.h>
#include <EEPROM.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h> // can be removed as it isn't compatible with Arduino

// Constants
const int LIGHT1_PIN = 4;
const int LIGHT2_PIN = 5;
const int PUMP_PIN = 8;
const int DHTPIN = 7;     
#define DHTTYPE DHT22   
const int THERMISTER_PIN = A2;
const int EC_PIN = A1;
const int PH_PIN = A8;
const int WATER_LEVEL_PIN = A3;  
const int PH_SAMPLING_INTERVAL = 20; 
const int PH_SAMPLING_NUMBER = 40;
const int PRINT_INTERVAL = 800;

bool found = false;

// Simulator constants
const int lux = 300;

// Control for pump run length (in milliseconds)
int PUMP_TIME = 1000;

// Message flow variables
String sending; 
String sensor; 
int state = 0;

// Sensor variables
float hum;  
float temp; 
float fahrenheit; 
int water_level_reading; 
int pH_array[PH_SAMPLING_NUMBER]; 
int pH_index = 0;
float voltage, ecValue, temperature = 25;

// Light variables
bool lights_status = false; 
char lights_color[10] = "blue"; 

// Wavelength structure for light colors
struct LightWavelength {
  char color[10];
  int wavelength;
};

struct LightWavelength light_wavelengths[7] = {
  {"red", 650},
  {"orange", 600},
  {"yellow", 580},
  {"green", 550},
  {"cyan", 500},
  {"blue", 450},
  {"violet", 400}
};

// Light functions
void lights_on(bool status, const char* color) {
  lights_status = status;
  bool light_time = true;
  strncpy(lights_color, color, sizeof(lights_color) - 1);
  // Timer for light cycle (every 3 hours)
  while (light_time) {
    int lux = 300;
    delay(10800000);  // sleep(10800000);
    // Simulate lux value changes
    lux = 0;
    delay(10800000);
  }
}

int read_wavelength() {
  int wavelength = 0;
  for (int i = 0; i < sizeof(light_wavelengths) / sizeof(light_wavelengths[0]); i++) {
    if (strcmp(light_wavelengths[i].color, lights_color) == 0) {
      wavelength = light_wavelengths[i].wavelength;
      break;
    }
  }
  return wavelength + (rand() % (5 + (-5) + 1) + (-5)); 
}

// Sensor reading functions
double readAirTemperature() {
  temp = 25.0; // dht.readTemperature();
  while ((lux == 300) && (temp <= 30.0)) {
    temp = temp + 2;
    //delay(300000);  // sleep(300000);
  }
  while ((lux == 0) && (temp >= 25.0)) {
    temp = temp - 2;
    //delay(300000);  // sleep(300000);
  }
  return temp;
}

double readWaterLevel() {
  water_level_reading = 600;
  //double new_water_level_reading;
  while ((lux == 300) && (water_level_reading >= 316)) {
    water_level_reading -= (0.5 * water_level_reading);
    //delay(300000);  // sleep(300000);
  }
  return (double) water_level_reading;
}
double Conductivity(double initialEC, double initialWaterLevel, double currentWaterLevel) {
   
    if (currentWaterLevel <= 0) {// Prevent division by zero
        return -1; 
    }
     
    double concentration = initialWaterLevel / currentWaterLevel;// Calculate the concentration

    
    double adjustedEC = initialEC * concentration;// Adjust EC based on the concentration 

    return adjustedEC;
}
double readElectricConductivity() {
    double initialEC = 300.0;        // Initial EC of water in µS/cm
    double initialWaterLevel = 600.0;    // Initial water level in mL
    //double currentLevel = readWaterLevel();  //Function to get the current water level
    double currentLevel = 200.0; //simulation input for different water levels

    return Conductivity(initialEC, initialWaterLevel, currentLevel);
}


double readVisibleLight() {
  return 300; // Simulated UV sensor reading
}

double readPH() {
  float pHValue = 0.0;
  while ((lux == 300) && pHValue <= 414.12) {
    pHValue += 50.00;  
  }
  return (double) pHValue;
  delay(300000);// sleep(300000);

}

double sim_readAirHumidity() {
  hum = 45;
  while ((lux == 300) && (hum <= 100)){
    hum += 1;
    
  }
  while ((lux == 0) && (hum >= 45)) {
    hum -= 1;
    
  }
  return hum;
  delay(300000); //sleep(300000)
}

double readWaterTemperature() { 
  int reading = 21;
  while ((lux == 300) && (reading <= 25)) {
    reading += 1;
    delay(300000);  // sleep(300000);
  }
  while ((lux == 0) && (reading >= 21)) {
    reading -= 1;
    delay(300000);  // sleep(300000);
  }
  return (double) reading;
}

double averageArray(int* arr, int number){
  int amount = 0;
  for(int i = 0; i < number; i++) {
    amount += arr[i];
  }
  return (double) amount / number;
}

void setup() {
  Serial.begin(9600);
  Serial.println("Welcome to the Serial Monitor!");
  
  pinMode(LIGHT1_PIN, OUTPUT);
  pinMode(LIGHT2_PIN, OUTPUT);
  pinMode(PUMP_PIN, OUTPUT);
}

void loop() {
  while (!found) {
    if (Serial.available() > 0) {
      String sync_message = Serial.readStringUntil('\n');
      if (sync_message == "Old MacDonald") {
        found = true;
        Serial.println("had a farm");
        delay(250);
      }
    }
  }

  if (Serial.available() > 0) {
    String request = Serial.readStringUntil('\n');
    DynamicJsonDocument serialDoc(2048);
    DeserializationError err = deserializeJson(serialDoc, request);

    if (err) {
      Serial.print("Error Parsing JSON");
      return;
    }

    JsonObject toSerial = serialDoc.as<JsonObject>();
    sensor = serialDoc["sensor"].as<String>();

    if (sensor == "airTemp") {
      toSerial["value"] = readAirTemperature();
    } else if (sensor == "humidity") {
      toSerial["value"] = sim_readAirHumidity();
    } else if (sensor == "lightCheck") {
      toSerial["value"] = readVisibleLight();
    } else if (sensor == "waterTemp") {
      toSerial["value"] = readWaterTemperature();
    } else if (sensor == "PH") {
      toSerial["value"] = readPH();
    } else if (sensor == "EC") {
      toSerial["value"] = readElectricConductivity();
    } else if (sensor == "waterLevel") {
      toSerial["value"] = readWaterLevel();
    } else if (sensor == "light1") {
      state = int(serialDoc["value"]);
      digitalWrite(LIGHT1_PIN, state);
    } else if (sensor == "light2") {
      state = int(serialDoc["value"]);
      digitalWrite(LIGHT2_PIN, state);
    } else if (sensor == "pump") {
      PUMP_TIME = int(serialDoc["value"]);
      digitalWrite(PUMP_PIN, HIGH);
      delay(PUMP_TIME);
      digitalWrite(PUMP_PIN, LOW);
    }
    
    serializeJson(serialDoc, sending);
    Serial.println(sending);
    sending = "";
    
  }
}