#include <ESP8266WiFi.h>
#include <PubSubClient.h>
#include <SoftwareSerial.h>
#include <SimpleTimer.h>

#include "settings.h"
#include "wifi.h"

WiFiClient wifiClient;
PubSubClient mqttClient;
SoftwareSerial geigerCounterSerial(PIN_UART_RX, PIN_UART_TX);
SimpleTimer timer;

String serialInput = "";
char serialInputHelper[RECV_LINE_SIZE];

int lastCPM = 0, currentCPM = 0;
float lastuSv = 0, currentuSv = 0;

void setup() {
  
  Serial.begin(115200);
  geigerCounterSerial.begin(BAUD_GEIGERCOUNTER);
  
  delay(10);

  Serial.print("Connecting to ");
  Serial.println(ssid);

  WiFi.begin(ssid, password);

  while (WiFi.status() != WL_CONNECTED) {
    Serial.print(".");
    delay(500);
  }

  Serial.println("");
  Serial.println("WiFi connected");
  Serial.println("IP address: ");
  Serial.println(WiFi.localIP());

  timer.setInterval(UPDATE_INTERVAL_SECONDS * 1000L, updateRadiationValues);
}

void updateRadiationValues() {

  
    char tmp[8];
    
    if(currentCPM != lastCPM) {
      String(currentCPM).toCharArray(tmp, 8);
      Serial.print("Sending CPM");
      Serial.println(tmp);
      mqttClient.publish(MQTT_TOPIC_CPM, tmp, true);  
    }

    if(currentuSv != lastuSv) {
      String(currentuSv).toCharArray(tmp, 8);
      Serial.print("Sending uSv");
      Serial.println(tmp);
      mqttClient.publish(MQTT_TOPIC_USV, tmp, true);
    }

    lastCPM = currentCPM;
    lastuSv = currentuSv;
}

void connectMqtt() {

  bool newConnection = false; 

  while (!mqttClient.connected()) {
    mqttClient.setClient(wifiClient);
    mqttClient.setServer(mqttHost, 1883);
    mqttClient.connect("geigercounter", MQTT_TOPIC_LAST_WILL, 1, true, "disconnected");
    
    delay(1000);
    newConnection = true;
  }

  if(newConnection) {
    mqttClient.publish(MQTT_TOPIC_LAST_WILL, "connected", true);
  }

}

void parseReceivedLine(char* input) {
  
    char segment = 0;
    char *token;

    float uSv = 0;
    float cpm = 0;

    token = strtok(input, delimiter);

    while (token != NULL) {

        switch(segment) {

            // This is just for validation
            case IDX_CPS_KEY: if (strcmp(token, "CPS") != 0) return;  break;
            case IDX_CPM_KEY: if (strcmp(token, "CPM") != 0) return;  break;
            case IDX_uSv_KEY: if (strcmp(token, "uSv/hr") != 0) return;  break;
                          
            case IDX_CPM:
                Serial.printf("Current CPM: %s\n", token);
                cpm = String(token).toInt();                
                break;

            case IDX_uSv:
                Serial.printf("Current uSv/hr: %s\n", token);
                uSv = String(token).toFloat();
                break;
        }

        if(segment > 7) {
          // Invalid! There should be no more than 7 segments
          return;
        }

        token = strtok(NULL, delimiter);
        segment++;
    }

    currentuSv = uSv;
    currentCPM = cpm;
}

void loop() {
  
  connectMqtt();
  timer.run();
  mqttClient.loop();

  if (geigerCounterSerial.available()) {
    char in = (char) geigerCounterSerial.read();

    serialInput += in;

    if (in == '\n') {
      serialInput.toCharArray(serialInputHelper, RECV_LINE_SIZE);
      parseReceivedLine(serialInputHelper);

      serialInput = "";
    }

    // Just in case the buffer gets to big, start from scratch
    if(serialInput.length() > RECV_LINE_SIZE + 10) {
      serialInput = "";
    }

    Serial.write(in);
  }  
}