#include <ESP8266WiFi.h>
#include <ESP8266WiFiMulti.h>
#include <ESP8266WiFi.h>
#include <ESP8266mDNS.h>
#include <WiFiUdp.h>
#include <ArduinoOTA.h>

#include <PubSubClient.h>
const char* debug = "debug";
const char* server = "172.19.1.10";

WiFiClient espClient;
PubSubClient client(espClient);

ESP8266WiFiMulti wifiMulti;

const char* ssid = "iot";
const char* password = "longtimeago";


uint8_t MAC_array[6];
char MAC_char[21];
char sta[23];

#include <Wire.h>
#include <LiquidCrystal_I2C.h>

LiquidCrystal_I2C lcd(0x3F, 20, 4); // 103
//LiquidCrystal_I2C lcd(0x27, 20, 4);

int val;
char str[10];

bool backlight;
bool oldbacklight;

unsigned long tick = 0;
char outside[10] = "";
int co = 0;
int sec = 10;

void numprint(int num);
int level(byte* payload, unsigned int length) {
  if((char)payload[0]=='o' && (char)payload[1]=='n'){
    return 1024;
  }
  if((char)payload[0]=='o' && (char)payload[1]=='f'){
    return 0;
  }

  unsigned int i;
  str[0] = '0';
  str[1] = '0';
  str[2] = '0';
  str[3] = '0';
  str[4] = '\0';

  for(i = 0; i < length; i++) {
    str[i] = (char)payload[i];
  }

  for(i = length; i < 10; i++) {
    str[i] = '\0';
  }

  return atoi(str);
}

void callback(char* topic, byte* payload, unsigned int length) {
  if(topic[1]=='l') {
    if(payload[0]=='0' && length==1) {
      backlight = false;
    } else {
      backlight = true;
    }

    lcd.setCursor(0,2);
    for(unsigned int a=length; a<7; a++) {
      lcd.print(' ');
    }
    for(unsigned int a=0; a<length; a++) {
      lcd.print((char)payload[a]);
    }
    lcd.print(" lux ");
    return ;
  }

  if(topic[14]=='c' && topic[15]=='o' && topic[16]=='2') {
    co = level(payload, length);
    //numprint(level(payload, length));
    return ;
  }

  if(topic[14]=='t') {
    lcd.setCursor(12,2);
    for(unsigned int a=0; a<length; a++) {
      lcd.print((char)payload[a]);
    }
    lcd.print(" C");
    return ;
  }

  if(topic[10]=='t') {
    for(unsigned int a=0; a<10; a++) {
      outside[a] = ' ';
    }
    for(unsigned int a=0; a<length; a++) {
      outside[a] = payload[a];
    }
    return ;
  }

  if(topic[14]=='p') {
    lcd.setCursor(0,3);
    for(unsigned int a=length; a<7; a++) {
      lcd.print(' ');
    }
    for(unsigned int a=0; a<length; a++) {
      lcd.print((char)payload[a]);
    }
    lcd.print(" hPa");
    return ;
  }

  if(topic[14]=='h') {
    if(length > 3) return;
    lcd.setCursor(12,3);
    for(unsigned int a=length; a<5; a++) {
      lcd.print(' ');
    }
    for(unsigned int a=0; a<length; a++) {
      lcd.print((char)payload[a]);
    }
    lcd.print(" %");
    return ;
  }

}

byte user_value1[8]={
    B11111,
    B11111,
    B00000,
    B00000,
    B00000,
    B00000,
    B00000,
    B00000
};
byte user_value2[8]={
    B00000,
    B00000,
    B00000,
    B00000,
    B00000,
    B00000,
    B11111,
    B11111
};
byte user_value3[8]={
    B11111,
    B11111,
    B00000,
    B00000,
    B00000,
    B00000,
    B11111,
    B11111
};
byte user_value4[8]={
    B11111,
    B11111,
    B11111,
    B11111,
    B11111,
    B11111,
    B11111,
    B11111
};
byte user_value5[8]={
    B00000,
    B00000,
    B00011,
    B00011,
    B00011,
    B00011,
    B00000,
    B00000
};
byte user_value6[8]={
    B00000,
    B00000,
    B11000,
    B11000,
    B11000,
    B11000,
    B00000,
    B00000
};

void setup()
{
  Wire.begin();
  lcd.init();
  lcd.noBacklight();
  lcd.clear();
  lcd.backlight();
  backlight = true;
  oldbacklight = true;
  WiFi.mode(WIFI_STA);
  wifiMulti.addAP(ssid,password);
  while (wifiMulti.run() != WL_CONNECTED) {
    delay(10);
    yield();
  }

  unsigned int i;
  WiFi.macAddress(MAC_array);
  for (i = 0; i < sizeof(MAC_array); ++i){
    sprintf(MAC_char,"%s%02x",MAC_char,MAC_array[i]);
  }
  sprintf(sta, "/%s/status", MAC_char);

  lcd.createChar(0, user_value1);
  lcd.createChar(1, user_value2);
  lcd.createChar(2, user_value3);
  lcd.createChar(3, user_value4);
  lcd.createChar(4, user_value5);
  lcd.createChar(5, user_value6);

  client.setServer(server, 1883);
  client.setCallback(callback);
  client.connect(MAC_char);
  client.publish(debug,MAC_char);
  client.publish(sta, PROJECT);
  client.subscribe("/light");
  client.subscribe("/dc4f22660be4/temperature");
  client.subscribe("/dc4f22660be4/humidity");
  client.subscribe("/dc4f22660be4/pressure");
  client.subscribe("/sensor/0/temperature");
  client.subscribe("/b4e62d675b8d/co2");

  ArduinoOTA.setHostname(MAC_char);
  ArduinoOTA.begin();
  ESP.wdtEnable(WDTO_8S);
  tick = millis();
}

void myprint(int x, const uint8_t num[6]) {
  yield();
  lcd.setCursor(x,0);
  yield();
  lcd.write(num[0]); yield(); lcd.write(num[1]); yield(); lcd.write(num[2]);
  yield();
  lcd.setCursor(x,1);
  yield();
  lcd.write(num[3]); yield(); lcd.write(num[4]); yield(); lcd.write(num[5]);
}

const uint8_t numbers[][6] = {
  {
    3, 0, 3,
    3, 1, 3
  },
  {
    0, 3, 160,
    1, 3, 1
  },
  {
    2, 2, 3,
    3, 1, 1
  },
  {
    2, 2, 3,
    1, 1, 3
  },
  {
    3, 1, 3,
    160, 160, 3
  },
  {
    3, 2, 2,
    1, 1, 3
  },
  {
    3, 2, 2,
    3, 1, 3
  },
  {
    0, 0, 3,
    160, 160, 3
  },
  {
    3, 2, 3,
    3, 1, 3
  },
  {
    3, 2, 3,
    1, 1, 3
  },
  {
    160, 160, 160,
    160, 160, 160
  },
  {
    3, 0, 0,
    3, 1, 1
  },
  {
    160, 1, 160,
    160, 160, 160
  },
  {
    3, 2, 3,
    160, 160, 160
  }
};

void numprint(int num) {
  if(num >= 10000) {
    myprint(0, numbers[num / 10000 % 10]);
  } else {
    myprint(0, numbers[10]);
  }

  if(num >= 1000) {
    myprint(4, numbers[num / 1000 % 10]);
  } else {
    myprint(4, numbers[10]);
  }

  if(num >= 100) {
    myprint(8, numbers[num / 100 % 10]);
  } else {
    myprint(8, numbers[10]);
  }

  if(num >= 10) {
    myprint(12, numbers[num / 10 % 10]);
  } else {
    myprint(12, numbers[10]);
  }

  if(num >= 0) {
    myprint(16, numbers[num / 1 % 10]);
  } else {
    myprint(16, numbers[10]);
  }
}


void loop()
{
  ESP.wdtFeed();

  if (wifiMulti.run() == WL_CONNECTED) {
    yield();
  }

  if(oldbacklight != backlight) {
    oldbacklight = backlight;
    if(backlight) {
      lcd.backlight();
    } else {
      lcd.noBacklight();
    }
  }

  if(client.connected()) {
    client.loop();
  } else {
    client.connect(MAC_char);
    client.subscribe("/light");
    client.subscribe("/dc4f22660be4/temperature");
    client.subscribe("/dc4f22660be4/humidity");
    client.subscribe("/dc4f22660be4/pressure");
    client.subscribe("/sensor/0/temperature");
    client.subscribe("/b4e62d675b8d/co2");
  }


  if(millis() - tick >= 1000) {
    tick = millis();
    sec--; if(sec<0) sec = 10;

    if(sec == 5) {
      for(int a=0; a<5; a++) {
        if('-' == outside[a]) { myprint(a*4, numbers[12]); }
        if('0' == outside[a]) { myprint(a*4, numbers[0]); }
        if('1' == outside[a]) { myprint(a*4, numbers[1]); }
        if('2' == outside[a]) { myprint(a*4, numbers[2]); }
        if('3' == outside[a]) { myprint(a*4, numbers[3]); }
        if('4' == outside[a]) { myprint(a*4, numbers[4]); }
        if('5' == outside[a]) { myprint(a*4, numbers[5]); }
        if('6' == outside[a]) { myprint(a*4, numbers[6]); }
        if('7' == outside[a]) { myprint(a*4, numbers[7]); }
        if('8' == outside[a]) { myprint(a*4, numbers[8]); }
        if('9' == outside[a]) { myprint(a*4, numbers[9]); }
        if('.' == outside[a]) { 
          myprint(a*4, numbers[13]);
          a++;
          myprint(a*4, numbers[11]);
          for(a = a + 1; a<5; a++) {
            myprint(a*4, numbers[10]);
          }
        }
      }
    }

    if(sec == 0) {
      numprint(co);
    }
  }

  delay(10);
  ESP.wdtFeed();
  ArduinoOTA.handle();
  ESP.wdtFeed();
}