#include <DHT.h>
#include "ota.h"
#include <WiFiClient.h>
#include "CCS811.h"

#define TEMP_SENSOR_PIN 12
#define HUMIDITY_SENSOR_PIN 12
#define FAN_PIN 15
#define CO2_PIN 16
#define ULTRASONIC_PIN 12

#define PHOTORESISTOR 0        // pin 0 is analog

#define CHECK_FREQUENCY 2      // check sensors every 2 seconds
#define HUMIDITY_DESIRED 75
#define HUMIDITY_VARIATION 3   // ultrasonic turns on at (75 - 3 = 72) and off
                               // at (75 + 3 = 78)
WiFiClient *wific = 0;

// Tempature + Humidity Sensor
#define DHTPIN 12
#define DHTTYPE DHT11
DHT dht(DHTPIN, DHTTYPE);

/*
 * IIC address default 0x5A, the address becomes 0x5B if the ADDR_SEL is soldered.
 */
//CCS811 sensor(&Wire, /*IIC_ADDRESS=*/0x5A);
CCS811 sensor;

void setup(void)
{
  setupWifi((char *) hostname);
  setupOTA((char *) hostname);

  dht.begin();                    // temp+humidity sensor
  pinMode(PHOTORESISTOR, INPUT);  // photo resistor
  wific = new WiFiClient();

  Serial.begin(115200);

  /*Wait for the chip to be initialized completely, and then exit*/
  while (sensor.begin()) {
    Serial.println("failed to init chip, please check if the chip connection is fine");
    delay(1000);
  }
  sensor.setMeasCycle(sensor.eCycle_250ms);
}

void writeBoth(char *buf)
{
  Serial.print(buf);
  if (wific->connected()) {
    wific->write(buf);
  }
}

struct SensorState
{
  float last_reading;
  virtual void sense() = 0;
};

struct TemperatureSensor : public SensorState { virtual void sense() { last_reading = dht.readTemperature(); } };
struct HumiditySensor : public SensorState { virtual void sense() { last_reading = dht.readHumidity(); } };
struct PhotoSensor : public SensorState { virtual void sense() { last_reading = analogRead(PHOTORESISTOR); } };
struct CO2Sensor : public SensorState { virtual void sense() { last_reading = sensor.getCO2PPM(); } };
struct TVOCSensor : public SensorState { virtual void sense() { last_reading = sensor.getTVOCPPB(); } };

TemperatureSensor  temperatureSensor;
HumiditySensor     humiditySensor;
PhotoSensor        photoSensor;
CO2Sensor          cO2Sensor;
TVOCSensor         tVOCSensor;

void log_reading()
{
  float    temp_c       = temperatureSensor.last_reading;
  float    humidity     = humiditySensor.last_reading;
  float    heat_index_c = dht.computeHeatIndex(temp_c, humidity, false);
  int      photons      = photoSensor.last_reading;
  uint32_t co2          = cO2Sensor.last_reading;
  uint32_t tvoc         = tVOCSensor.last_reading;

  auto fmt = "T: %.2f H: %.2f HI: %.2f Light: %d CO2: %u TVOC: %u\r\n";
  char buf[500];


  snprintf(buf, sizeof(buf), fmt, temp_c, humidity, heat_index_c, photons, co2, tvoc);
  writeBoth(buf);
}

void sensor_loop()
{
  SensorState *sensors[] = { &temperatureSensor, &humiditySensor, &photoSensor, &cO2Sensor, &tVOCSensor };
  for (size_t i=0; i<sizeof(sensors)/sizeof(sensors[0]); ++i)
  {
    sensors[i]->sense();
  }
  sensor.writeBaseLine(0x847B);
}

void loop()
{
  ArduinoOTA.handle();

  auto ip = IPAddress(192,168,1,2);
  auto port = 3141;
  if (!wific->connected()) {
    Serial.println("Attempting to connect");
    wific->connect(ip, port);
  }

  sensor_loop();

  log_reading();

  delay(1000);
}