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

#define RelayPin 5    //D1
#define BUILTIN_LED 2 //D0

const char *ssid = "WIFI-SSID";
const char *password = "WIFI-PASSORD";
const char *mqtt_server = "IP_TIL_BROKER";

String clientId = "bartender1";
const char *topic = "bartender/1";
const char *motor_topic = "bartender/1/motor/set";
const char *motor_status_topic = "bartender/1/motor/get";
const char *runtime_topic = "bartender/1/runtime";
const char *runtime_status_topic = "bartender/1/runtimestatus";

unsigned long lastMotorSet;
unsigned long lastRuntimeSet;
unsigned long lastRuntimeUpdate;
float runtimeOriginal;
float runtime;
int motorStatus;
bool ignoreNextMotorStatusMessage;

WiFiClient espClient;
PubSubClient client(espClient);

void setup()
{
  //Set pins at output
  pinMode(BUILTIN_LED, OUTPUT);
  pinMode(RelayPin, OUTPUT);

  //Turn off motor
  digitalWrite(BUILTIN_LED, HIGH);
  digitalWrite(RelayPin, HIGH);

  //Connect to wifi and MQTT broker
  Serial.begin(115200);
  setup_wifi();
  client.setServer(mqtt_server, 1883);
  client.setCallback(callback);
}

void setup_wifi()
{
  delay(10);

  // Connect to wifi
  Serial.println();
  Serial.print("Connecting to ");
  Serial.println(ssid);

  WiFi.mode(WIFI_STA);
  WiFi.begin(ssid, password);

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

  randomSeed(micros());

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

//Handle reconnect if necessary
void reconnect()
{
  // Loop until we're reconnected
  while (!client.connected())
  {
    Serial.print("Attempting MQTT connection...");

    if (client.connect(clientId.c_str()))
    {
      Serial.println("connected");
      client.subscribe(motor_topic);
      client.subscribe(runtime_topic);
    }
    else
    {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 2 seconds");
      // Wait 5 seconds before retrying
      delay(2000);
    }
  }
}

//Handle incomming MQTT callbacks
void callback(char *topic, byte *payload, unsigned int length)
{
  Serial.print("Message arrived [");
  Serial.print(topic);
  Serial.print("] ");
  String stringPayload = "";
  for (int i = 0; i < length; i++)
  {
    //Serial.print((char)payload[i]);
    stringPayload = stringPayload + (char)payload[i];
  }
  //Serial.println();
  Serial.println(stringPayload);

  if (strcmp(runtime_topic, topic) == 0)
  {
    Serial.println("Detected runtime_topic");
    lastRuntimeSet = millis();
    lastRuntimeUpdate = millis();
    runtime = stringPayload.toFloat() * 1000;
    runtimeOriginal = runtime;
  }

  if (strcmp(motor_topic, topic) == 0)
  {
    // if (ignoreNextMotorStatusMessage == false)
    // {
      Serial.println("Detected motor_topic");
      lastMotorSet = millis();
      if (stringPayload == "1")
        if (motorStatus == 0)
          MotorOn();
      if (stringPayload == "0")
        if (motorStatus == 1)
          MotorOff();
  }
}

unsigned long lastReadTime;
void loop()
{

  if (!client.connected())
  {
    reconnect();
  }
  client.loop();

  delay(1);

  if (runtime > 0)
  {
    runtime = runtime - (millis() - lastRuntimeUpdate);
    lastRuntimeUpdate = millis();
  }
  if (runtime < 0)
    runtime = 0;

  if (lastRuntimeSet > lastMotorSet)
  {
    if (runtime > 0)
    {
      if (motorStatus == 0)
      {
        //ignoreNextMotorStatusMessage = true;
        MotorOn();
      }
    }
    else
    {
      if (motorStatus == 1)
      {
        //ignoreNextMotorStatusMessage = true;
        MotorOff();
      }
    }
  }

  if (millis() - lastReadTime > 250)
  {
    if (motorStatus == 1 && runtime > 0)
    {
      float percent = (runtimeOriginal - runtime) / runtimeOriginal * 100;
      String percenString = String(percent, 0);
      client.publish(runtime_status_topic, percenString.c_str());
      Serial.print(percent);
      Serial.print("% -- ");
    }

    Serial.println(runtime);
    lastReadTime = millis();
  }
}

void MotorOn()
{
  motorStatus = 1;

  digitalWrite(RelayPin, LOW);
  digitalWrite(BUILTIN_LED, LOW);

  client.publish(motor_status_topic, "1");
  client.publish(runtime_status_topic, "0");
  Serial.println("Turning on...");
}

void MotorOff()
{
  motorStatus = 0;

  digitalWrite(BUILTIN_LED, HIGH);
  digitalWrite(RelayPin, HIGH);

  client.publish(motor_status_topic, "0");
  client.publish(runtime_status_topic, "100");
  Serial.println("Turning off...");
}