//
// Libraries needed:
//  Time.h & TimeLib.h:  https://github.com/PaulStoffregen/Time
//  Timezone.h: https://github.com/JChristensen/Timezone
//  SSD1306.h & SSD1306Wire.h:  https://github.com/squix78/esp8266-oled-ssd1306
//  NTPClient.h: https://github.com/arduino-libraries/NTPClient
//  ESP8266WiFi.h & WifiUDP.h: https://github.com/ekstrand/ESP8266wifi
//

#include <ESP8266WiFi.h>
#include <PubSubClient.h>
#include <String.h>
#include <epd.h>
#include "DHT.h"
#include <Ticker.h>
#include  <stdio.h>
#include "string.h"

// Define NTP properties
#define NTP_OFFSET   60 * 60      // In seconds
#define NTP_INTERVAL 60 * 1000    // In miliseconds
#define NTP_ADDRESS  "ua.pool.ntp.org"  // change this to whatever pool is closest (see ntp.org)

#define DHTPIN 13
#define DHTTYPE DHT22   // DHT 22  (AM2302), AM2321

DHT dht(DHTPIN, DHTTYPE);

Ticker ClockTimer;


#define ssid "DIR-615"              // insert your own ssid
#define pass "HomeMade"         // and password

#define mqtt_server  "hairdresser.cloudmqtt.com"   // Имя сервера MQTT
#define mqtt_port    18608                         // Порт для подключения к серверу MQTT
#define mqtt_user    "user"                    // Логин от сервер
#define mqtt_pass    "pass"                // Пароль от сервера

byte ss = 0;
byte mm = 0;
byte hh = 0;
byte tu = 0;
float Hum = 0.0;
float Temp = 0.0;
char str_temp[6];
float hic = 0.0;
byte d1 = 1;
byte d2 = 1;
byte d3 = 1;
byte d4 = 20;
//const char * days[] = {"Saturday", "Sunday", "Monday", "Tuesday", "Wednesday", "Thursday", "Friday"} ;
//const char * months[] = {"January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"} ;
//const byte monthsCodes[] = {1, 4, 4, 0, 2, 5, 0, 3, 6, 1, 4, 6};
float Voltage = 0.0;
boolean DisplayUpdate = true;
boolean WIFITimeUpdate = true;
boolean WiFiConected = false;
boolean MQTTConected = false;
boolean epdStopmode = false;
byte epdStopmodeCount = 0;

byte m1;
byte h1;
byte d21 = 1;
byte d31 = 1;

float WStationVoltage = 0.0;
char DateString[] = "00.00.0000";
char TimeString[] = "00.00.00";
 

#define wake_up 14

char Buffer[] = "";

void timerIsr()
{
    if (epdStopmode){
        epdStopmodeCount++;
        if (epdStopmodeCount > 1){
            epdStopmodeCount = 0;
            epdStopmode = false;
            epd_enter_stopmode();
        }
    }
    ss++;                           // Seconds
    if (ss > 59){
        ss = 0;
        mm++;                       // Minutes
        DisplayUpdate = true;
        if (mm > 59){
            mm = 0;
            hh++;                   // Hours
            if (hh > 23){
                hh = 0;
                d1++;               // Day of week
                d2++;               // Day of month
                //d3++;               // Month
                if (d1 > 6) d1 = 0;
                if (d2 > 31) d2 = 0;
                if (d3 > 11) d3 = 0;
            }
            tu++;
            if (tu > 20){            // Pause update via WiFi (*hours)
                tu = 0;
                WIFITimeUpdate = true;
            }
        }
    }
}

void callback(const MQTT::Publish& pub)     // Функция получения данных от сервера
{
    Serial.print(pub.topic());                // выводим в сериал порт название топика
    Serial.print(" => ");
    Serial.println(pub.payload_string());     // выводим в сериал порт значение полученных данных
    String payload = pub.payload_string();
   
    if(String(pub.topic()) == "WStation/Battery")    //  получаем данные из топика о напряжении аккумулятора
    {
        WStationVoltage = payload.toFloat();               //  преобразуем полученные данные в тип integer
    }
   
    if(String(pub.topic()) == "WStation/Date")      //  получаем данные из топика Даты
    {
        payload.toCharArray(DateString, 11);
    }
   
    if(String(pub.topic()) == "WStation/Time")      // получаем данные из топика Времени
    {
        //TimeString = payload;
        payload.toCharArray(TimeString, 9);
    }
}

WiFiClient wclient;     
PubSubClient client(wclient, mqtt_server, mqtt_port);

void setup ()
{
    //Serial.begin(115200); // most ESP-01&#39;s use 115200 but this could vary
                            // Инициализация Serial есть в библиотеке дисплея - epd.h

    pinMode(2, OUTPUT);
   
   
    epd_init();
    pinMode(wake_up, OUTPUT);
    wakeup();
    epd_set_memory(MEM_TF);
    delay(1000);
   
    epd_clear();
    epd_set_color(BLACK, WHITE);
    epd_set_en_font(ASCII64);

    sprintf(Buffer,"Connecting to:  %s . . .", ssid);
    epd_disp_string(Buffer, 0, 0);
    epd_udpate();
   
    WiFi.begin(ssid, pass);
    while (WiFi.status() != WL_CONNECTED)
    {
        delay(500);
        Serial.print(".");
        d31++;
        if (d31 > 100){
            break;
        }
       
    }
    d31=1;
   
    epd_disp_string("Connected.", 0, 80);
    epd_disp_string("Connecting to: MQTT...", 0, 160);
    epd_udpate();
    Serial.println("");
    Serial.print("Connected to WiFi at ");
    Serial.print(WiFi.localIP());
    Serial.println("");
   
    ClockTimer.attach(1, timerIsr);
    dht.begin();
    delay(2000);
    epd_clear();

    digitalWrite(2, HIGH);
}

void loop()
{
    WiFiupd();
   
    if (DisplayUpdate){
        DisplayUpdate = false;
        wakeup();
       
        epd_disp_bitmap("FON1.BMP", 0, 0);
       
        if (WiFiConected){
            epd_disp_bitmap("WON.BMP", 582, 0);     // выводим картинку подключенного WiFi
            if (MQTTConected){
                epd_disp_bitmap("MQON.BMP", 0, 136);     // выводим картинку подключенного WiFi
            }else{
                epd_disp_bitmap("MQOF.BMP", 0, 136);     // выводим картинку подключенного WiFi
            }
        }else{
            epd_disp_bitmap("WOFF.BMP", 582, 0);        // Выводим кортинку отключенного WiFi с переподключением
            epd_disp_bitmap("MQOF.BMP", 0, 136);     // выводим картинку подключенного WiFi
        }
       
        drawWStationData();
        yield();
        drawTime();
        yield();
        sonsorDHT();
        yield();
        drawBattery();

        yield();
        epd_set_color(DARK_GRAY, WHITE);
        epd_set_en_font(ASCII32);
        sprintf(Buffer,"%02d.%02d  %02d:%02d", d21, d31+1, h1, m1);
        epd_disp_string(Buffer, 25, 320);
        //sprintf(Buffer,"%02d.%02d", d21, d31);
        //epd_disp_string(Buffer, 30, 320);
        epd_set_color(BLACK, WHITE);
        epd_set_en_font(ASCII64);

        epd_udpate();
        epdStopmode = true;
       
    }
}

void drawWStationData(){
    yield();
    Serial.println("-1-");
    dtostrf(WStationVoltage, 4, 2, str_temp);
    sprintf(Buffer,"%s V", str_temp);       // напряжение на аккумуляторе погодной станции
    epd_disp_string(Buffer, 62, 35);
}

void wakeup()
{
    yield();
    digitalWrite(wake_up, LOW);
    delayMicroseconds(10);
    digitalWrite(wake_up, HIGH);
    delayMicroseconds(500);
    digitalWrite(wake_up, LOW);
    delay(10);
}

void sonsorDHT(){
    yield();
    Hum = dht.readHumidity();
    Temp = dht.readTemperature();
    //hic = dht.computeHeatIndex(Temp, Hum, false);

    if (isnan(Hum) || isnan(Temp)) {
        epd_disp_string("ERR", 265, 495);
        epd_disp_string("ERR", 565, 495);
    }else{
        dtostrf(Temp, 4, 1, str_temp);
        sprintf(Buffer,"%s", str_temp);
        epd_disp_string(Buffer, 265, 495);
       
        dtostrf(Hum, 4, 1, str_temp);
        sprintf(Buffer,"%s", str_temp);
        epd_disp_string(Buffer, 565, 495);
    }
}

void drawTime(){
    Date_Time();
   
    yield();
    m1 = mm;
    h1 = hh;
    d21 = d2;
    d31 = d3;

    /*epd_disp_string(days[d1-1], 317, 300);
    sprintf(Buffer,"%d", d2);
    epd_disp_string(Buffer, 649, 273);
    epd_set_en_font(ASCII32);
    epd_disp_string(months[d3-1], 622, 342);
    epd_set_en_font(ASCII64);
    */
    epd_disp_string(DateString, 360, 300);
   
    byte NunbH = hh / 10;
    byte NunbL = hh % 10;
    sprintf(Buffer,"%d.BMP", NunbH);
    epd_disp_bitmap(Buffer, 212, 84);
    sprintf(Buffer,"%d.BMP", NunbL);
    epd_disp_bitmap(Buffer, 345, 84);
   
    NunbH = mm / 10;
    NunbL = mm % 10;
    sprintf(Buffer,"%d.BMP", NunbH);
    epd_disp_bitmap(Buffer, 545, 84);
    sprintf(Buffer,"%d.BMP", NunbL);
    epd_disp_bitmap(Buffer, 679, 84);
}

void drawBattery(){
    yield();
    float sample = 0.0;
   
    // 3.3 = 1024

    for(int i=0;i< 50;i++)
    {
        sample = sample + analogRead(A0); //read the voltage from the divider circuit
        delay (2);
    }
    sample = sample / 50;
    Voltage = 1.5 * (sample * 3.3 / 1024.0);

   
    if (Voltage > 4.6){
        epd_disp_bitmap("BAT1.BMP", 681, 0);
    }else if (Voltage > 4.1){
        epd_disp_bitmap("BAT8.BMP", 681, 0);
    }else if (Voltage > 3.98){
        epd_disp_bitmap("BAT7.BMP", 681, 0);
    }else if (Voltage > 3.87){
        epd_disp_bitmap("BAT6.BMP", 681, 0);
    }else if (Voltage > 3.75){
        epd_disp_bitmap("BAT5.BMP", 681, 0);
    }else if (Voltage > 3.64){
        epd_disp_bitmap("BAT4.BMP", 681, 0);
    }else if (Voltage > 3.52){
        epd_disp_bitmap("BAT3.BMP", 681, 0);
    }else if (Voltage > 3.41){
        epd_disp_bitmap("BAT2.BMP", 681, 0);
    }else{
       
        epd_clear();                            // Выключение
        epd_udpate();
        delay(2000);
        epd_disp_bitmap("BATLOW.BMP", 0, 0);
        epd_set_color(GRAY, WHITE);
        epd_set_en_font(ASCII32);
        sprintf(Buffer,"Shutdown time: %02d.%02d  %02d:%02d", d2, d3+1, hh, mm);
        epd_disp_string(Buffer, 0, 0);
        sprintf(Buffer,"%02d.%02d  %02d:%02d", d21, d31+1, h1, m1);
        epd_disp_string(Buffer, 0, 40);
        epd_udpate();
        delay(2000);
        epd_enter_stopmode();
        delay(1000);
        ESP.deepSleep(0);
    }
}

void Date_Time(){
    yield();
    String myString;
    //char myChar;    // = DateString.charAt(0);
    //int myNum;      // = String(myChar).toInt();

    /*if (myNum > 0){
        myString = DateString.substring(0,2);
        d2 = myString.toInt();
    }else{
        myChar = DateString.charAt(1);
        d2 = String(myChar).toInt();
    }

    myChar = DateString.charAt(3);
    myNum = String(myChar).toInt();
    if (myNum > 0){
        myString = DateString.substring(3,5);
        d3 = myString.toInt();
    }else{
        myChar = DateString.charAt(4);
        d3 = String(myChar).toInt();
    }

    myString = DateString.substring(8);
    d4 = myString.toInt();

    d1 = ((d2 + monthsCodes[d3-1] + ((6 + d4 + (d4/4))%7)) % 7)-1;   // https://lifehacker.ru/kakoj-den-nedeli/
    */
   
    char _el[3] ="";
    int myNum = String(TimeString[0]).toInt();
    if (myNum > 0){
        strncpy(_el,&TimeString[0],2);
        printf("%s",_el);
        hh = String(_el).toInt();
    }else{
        hh = String(TimeString[0]).toInt();
    }
    yield();

    myNum = String(TimeString[3]).toInt();
    if (myNum > 0){
        //_el[3] ="";
        strncpy(_el,&TimeString[3],2);
        printf("%s",_el);
        mm = String(_el).toInt();
    }else{
        mm = String(TimeString[4]).toInt();
    }
}

void WiFiupd(){
    if (WiFi.status() != WL_CONNECTED) // attempt to connect to wifi again if disconnected
    {
        Serial.print("Connecting to ");
        Serial.print(ssid);
        Serial.println("...");
        WiFi.begin(ssid, pass);
        WiFiConected = false;
    }
    if (WiFi.status() == WL_CONNECTED) //Check WiFi connection status
    {
        WiFiConected = true;
        if (!client.connected()) {
            yield();
            Serial.println("Connecting to MQTT server");
            if (client.connect(MQTT::Connect("WStationDisplay")
                                 .set_auth(mqtt_user, mqtt_pass))) {
                Serial.println("Connected to MQTT server");
                client.set_callback(callback);
                client.subscribe("WStation/Time");                  // подписывааемся по топик времени
                client.subscribe("WStation/Battery");               // подписывааемся по топик с данными аккумулятора
                client.subscribe("WStation/Date");                  // подписывааемся по топик даты
               
                MQTTConected = true;
            } else {
                Serial.println("Could not connect to MQTT server");   
                MQTTConected = false;
            }
        }
        if (client.connected()){
            MQTTConected = true;
            yield();
            client.loop();
        }
    }
}