// #include <arduino.h>
#include "nodeCodeHeader.h"
// #include <DHT.h>
// #include <Adafruit_SH110X.h>


nodeReadings::nodeReadings() {

  dht = new DHT(DHTPIN, DHTTYPE);
  display = new Adafruit_SH1106G(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);
  
  // webSocket = new WebSocketsClient();
  webSocket = new websockets(); //nu naar eigen class 
  sgp = new Adafruit_SGP30();

  resetValues();  

}

void nodeReadings::resetValues() {
    counter = 0;
    eCO2 = 0;
    TVOC = 0;
    temperature = 0;
    humidity = 0;
    currentMillis = 0;
    lastMillis = 0;
    errorSGP30 = false;
    errorDHT11 = false;
    noise = false;
}

void nodeReadings::setup(){
    // DHT dht(DHTPIN, DHTTYPE);


    // make serial connection at 115200 baud
    Serial.begin(115200);

    // tell display what settings to use
    display->begin(i2c_adress, true);
    display->clearDisplay();

    // tell sensors to start reading
    dht->begin();
    sgp->begin();

    pinMode(MICPIN, INPUT);
    pinMode(DHTPIN, INPUT);
    
    webSocket->websocketSetup();

    // webSocketClass.websocketSetup();


}

void nodeReadings::loop() {
    // update when interval is met
    if (currentMillis - lastMillis >= interval){
        lastMillis = millis();
        update();
    }

    // update the counter
    currentMillis = millis();

    // loop the websocket connection so it stays alive
    webSocket->loop();
}

void nodeReadings::update(){

  // display sensordata on oled screen
  displayData();
  
  // webSocket->sendTXT("{\"Temp\":\"" + String(temperature) + "\",\"Humi\":\"" + String(humidity) + "\",\"eCO2\":\"" + String(sgp->eCO2) + "\",\"TVOC\":\"" + String(sgp->TVOC) + "\"}");
  webSocket->sendMyText("{\"node\": \"" + String(WiFi.macAddress()) + "\", \"Temp\":\"" + String(temperature) + "\",\"Humi\":\"" + String(humidity) + "\",\"eCO2\":\"" + String(sgp->eCO2) + "\",\"TVOC\":\"" + String(sgp->TVOC) + "\"}");

  sgp->getIAQBaseline(&eCO2_base, &TVOC_base);

  // read dht11 sensor
  temperature = float(dht->readTemperature());
  humidity = float(dht->readHumidity());

  // check if any errors occured when reading sensors
  checkForError();
}

void nodeReadings::displayData() {
  // clear display for new info
  display->clearDisplay();

  // display the data on the oled screen
  display->setTextSize(1);
  display->setTextColor(SH110X_WHITE);
  display->setCursor(0,0);
  display->println("Temperature: " + String(temperature) + " C");
  display->println("Humidity: " + String(humidity) + " %");
  display->println("eCO2: " + String(sgp->eCO2) + " ppm");
  display->println("TVOC: " + String(sgp->TVOC) + " ppb");
  display->display();
}

void nodeReadings::checkForError(){
  if (!sgp->IAQmeasure()) {
    Serial.println("SGP30: BAD");
    errorSGP30 = true;
  } else {
    Serial.println("SGP30: OK");
    errorSGP30 = false;
  }

  if (isnan(temperature) || isnan(humidity)){
    Serial.println("DHT11: BAD");
    errorDHT11 = true;
  } else {
    Serial.println("DHT11: OK");
    errorDHT11 = false;
  }
}