giess-o-mat/src/WifiStuff.cpp

/*
 * Copyright (c) 2021 Thomas Buck <thomas@xythobuz.de>
 *
 * This file is part of Giess-o-mat.
 *
 * Giess-o-mat is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Giess-o-mat is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Giess-o-mat.  If not, see <https://www.gnu.org/licenses/>.
 */

#include <Arduino.h>

#ifdef PLATFORM_ESP

#if defined(ARDUINO_ARCH_ESP8266)

#include <ESP8266WiFi.h>
#include <ESP8266WebServer.h>
#include <ESP8266mDNS.h>

#elif defined(ARDUINO_ARCH_ESP32)

#include <WiFi.h>
#include <WebServer.h>
#include <ESPmDNS.h>

#endif

#include <WebSocketsServer.h>

#include "wifi.h"
#include "config.h"
#include "config_pins.h"
#include "Functionality.h"
#include "SimpleUpdater.h"
#include "DebugLog.h"
#include "BoolField.h"
#include "WifiStuff.h"

#ifdef TELEGRAM_TOKEN

#include <WiFiClientSecure.h>
#include <UniversalTelegramBot.h>

//#define TELEGRAM_LOG_TIMINGS

#if defined(ARDUINO_ARCH_ESP8266)
X509List cert(TELEGRAM_CERTIFICATE_ROOT);
#endif

WiFiClientSecure secured_client;
UniversalTelegramBot bot(TELEGRAM_TOKEN, secured_client);
unsigned long last_telegram_time = 0;
String trusted_chat_ids[] = TRUSTED_IDS;

#endif // TELEGRAM_TOKEN

UPDATE_WEB_SERVER server(80);
WebSocketsServer socket = WebSocketsServer(81);
SimpleUpdater updater;
unsigned long last_server_handle_time = 0;
unsigned long last_websocket_update_time = 0;
int wifi_ok = 0;

String message_buffer_a;
String message_buffer_b;
String message_buffer_c;
String message_buffer_d;

#ifdef ENABLE_GPIO_TEST
static bool runningGpioTest = false;
static bool gpioTestState = false;
unsigned long lastGpioTime = 0;
#endif // ENABLE_GPIO_TEST

#ifdef ENABLE_INFLUXDB_LOGGING
#include <InfluxDb.h>
Influxdb influx(INFLUXDB_HOST, INFLUXDB_PORT);
#endif // ENABLE_INFLUXDB_LOGGING

#ifdef ENABLE_GPIO_TEST

void runGpioTest(bool state) {
    lastGpioTime = millis();
    
    for (int i = 0; i < VALVE_COUNT; i++) {
        get_plants()->getValves()->setPin(i, state);
        delay(GPIO_TEST_DELAY);
    }
    
    for (int i = 0; i < PUMP_COUNT; i++) {
        get_plants()->getPumps()->setPin(i, state);
        if (i < (PUMP_COUNT - 1)) {
            delay(GPIO_TEST_DELAY);
        }
    }
}

void handleGpioTest() {
    runningGpioTest = !runningGpioTest;
    gpioTestState = runningGpioTest;
    
    String message = F("GPIOs turned ");
    message += runningGpioTest ? "on" : "off";

    server.send(200, "text/html", message);
    
    runGpioTest(gpioTestState);
}

#endif // ENABLE_GPIO_TEST

void wifi_broadcast_state_change(const char *s) {
#ifdef TELEGRAM_TOKEN
    for (int n = 0; n < (sizeof(trusted_chat_ids) / sizeof(trusted_chat_ids[0])); n++) {
        bot.sendMessage(trusted_chat_ids[n], "New state: " + String(s), "");
    }
#endif // TELEGRAM_TOKEN
}

#ifdef TELEGRAM_TOKEN
enum telegram_state {
    BOT_IDLE,
    BOT_ASKED_FERT,
    BOT_ASKED_PLANTS,
    BOT_ASKED_CONFIRM
};

enum telegram_state bot_state = BOT_IDLE;
String bot_lock = "";
BoolField bot_plants(VALVE_COUNT - 1);
BoolField bot_ferts(PUMP_COUNT);

unsigned long telegram_update_interval() {
    if (bot_state == BOT_IDLE) {
        return TELEGRAM_UPDATE_INTERVAL_SLOW;
    } else {
        return TELEGRAM_UPDATE_INTERVAL_FAST;
    }
}

String telegram_help() {
    String s = "Usage:\n";
    s += "Send /auto and follow prompts.\n";
    s += "Send /abort to cancel menus.\n";
    s += "Send /none to skip menus.\n";
    s += "Send /begin to confirm menus.";
    return s;
}

void telegram_handle_message(int message_id) {
    if (!sm_is_idle()) {
        debug.println("Telegram: message while machine in use");

        if (bot.messages[message_id].text == "/abort") {
            sm_bot_abort();
            bot.sendMessage(bot.messages[message_id].chat_id, "Aborted current cycle!", "");
        } else {
            bot.sendMessage(bot.messages[message_id].chat_id, "Machine is already in use.\nPlease try again later.", "");
        }

        return;
    }

    if ((bot_state == BOT_IDLE) && (bot_lock == "")) {
        bot_lock = bot.messages[message_id].chat_id;
        debug.println("Telegram: locked to " + bot_lock);
    }

    if (bot_lock != bot.messages[message_id].chat_id) {
        debug.println("Telegram: bot locked. abort for chat " + bot.messages[message_id].chat_id);
        bot.sendMessage(bot.messages[message_id].chat_id, "Bot is already in use.\nPlease try again later.", "");
        return;
    }

    if (bot_state == BOT_IDLE) {
        if (bot.messages[message_id].text == "/auto") {
            String s = "Please enter fertilizer numbers.\n";
            s += "Valid numbers: 1 to " + String(PUMP_COUNT) + "\n";
            s += "Send /none to skip.\n";
            s += "Send /abort to cancel.";
            bot_ferts.clear();
            bot_state = BOT_ASKED_FERT;
            bot.sendMessage(bot.messages[message_id].chat_id, s, "");
        } else if (bot.messages[message_id].text == "/abort") {
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Nothing to abort.", "");
        } else {
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, telegram_help(), "");
        }
    } else if (bot_state == BOT_ASKED_FERT) {
        if (bot.messages[message_id].text == "/abort") {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Aborted.", "");
            return;
        } else if (bot.messages[message_id].text != "/none") {
            String buff;
            for (int i = 0; i < bot.messages[message_id].text.length() + 1; i++) {
                if ((i == bot.messages[message_id].text.length()) || (bot.messages[message_id].text[i] == ' ')) {
                    if (buff.length() > 0) {
                        int n = buff.toInt() - 1;
                        buff = "";
                        bot_ferts.set(n);
                    }
                } else if ((bot.messages[message_id].text[i] >= '0') && (bot.messages[message_id].text[i] <= '9')) {
                    buff += bot.messages[message_id].text[i];
                } else {
                    bot_state = BOT_IDLE;
                    bot_lock = "";
                    bot.sendMessage(bot.messages[message_id].chat_id, "Invalid input.\nAborted.", "");
                    return;
                }
            }
        }

        String s = "Please enter plant numbers.\n";
        s += "Valid numbers: 1 to " + String(VALVE_COUNT - 1) + "\n";
        s += "Send /abort to cancel.";
        bot_plants.clear();
        bot_state = BOT_ASKED_PLANTS;
        bot.sendMessage(bot.messages[message_id].chat_id, s, "");
    } else if (bot_state == BOT_ASKED_PLANTS) {
        if (bot.messages[message_id].text == "/abort") {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Aborted.", "");
            return;
        }

        String buff;
        for (int i = 0; i < bot.messages[message_id].text.length() + 1; i++) {
            if ((i == bot.messages[message_id].text.length()) || (bot.messages[message_id].text[i] == ' ')) {
                if (buff.length() > 0) {
                    int n = buff.toInt() - 1;
                    buff = "";
                    bot_plants.set(n);
                }
            } else if ((bot.messages[message_id].text[i] >= '0') && (bot.messages[message_id].text[i] <= '9')) {
                buff += bot.messages[message_id].text[i];
            } else {
                bot_state = BOT_IDLE;
                bot_lock = "";
                bot.sendMessage(bot.messages[message_id].chat_id, "Invalid input.\nAborted.", "");
                return;
            }
        }

        if (bot_plants.countSet() <= 0) {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "No plants selected.\nAborted.", "");
            return;
        }

#ifdef FULLAUTO_MIN_PLANT_COUNT
        if (bot_plants.countSet() < FULLAUTO_MIN_PLANT_COUNT) {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Select at least " + String(FULLAUTO_MIN_PLANT_COUNT) + " plants.\nAborted.", "");
            return;
        }
#endif

        String s = "Input accepted.\nFertilizers:";
        for (int i = 0; i < PUMP_COUNT; i++) {
            if (bot_ferts.isSet(i)) {
                s += " " + String(i + 1);
            }
        }
        s += "\nPlants:";
        for (int i = 0; i < (VALVE_COUNT - 1); i++) {
            if (bot_plants.isSet(i)) {
                s += " " + String(i + 1);
            }
        }
        s += "\nOk? Send /begin to start or /abort to cancel.";
        bot_state = BOT_ASKED_CONFIRM;
        bot.sendMessage(bot.messages[message_id].chat_id, s, "");
    } else if (bot_state == BOT_ASKED_CONFIRM) {
        if (bot.messages[message_id].text == "/abort") {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Aborted.", "");
        } else if (bot.messages[message_id].text == "/begin") {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Auto watering cycle started.", "");
            sm_bot_start_auto(bot_ferts, bot_plants);
        } else {
            bot_state = BOT_IDLE;
            bot_lock = "";
            bot.sendMessage(bot.messages[message_id].chat_id, "Unknown message.\nAborted.", "");
        }
    } else {
        debug.println("Telegram: invalid state");
        bot_state = BOT_IDLE;
        bot_lock = "";
        bot.sendMessage(bot.messages[message_id].chat_id, "Internal error.\nPlease try again.", "");
    }
}

void telegram_handler(int message_id) {
    debug.println("Telegram: rx " + bot.messages[message_id].chat_id + " \"" + bot.messages[message_id].text + "\"");

    bool found = false;
    for (int n = 0; n < (sizeof(trusted_chat_ids) / sizeof(trusted_chat_ids[0])); n++) {
        if (trusted_chat_ids[n] == bot.messages[message_id].chat_id) {
            found = true;
            break;
        }
    }
    if (!found) {
        bot.sendMessage(bot.messages[message_id].chat_id, "Sorry, not authorized!", "");
        return;
    }

    telegram_handle_message(message_id);
}

void telegram_poll() {
#ifdef TELEGRAM_LOG_TIMINGS
    unsigned long start = millis();
#endif // TELEGRAM_LOG_TIMINGS

    while (int count = bot.getUpdates(bot.last_message_received + 1)) {
        for (int i = 0; i < count; i++) {
            telegram_handler(i);
        }
    }

#ifdef TELEGRAM_LOG_TIMINGS
    unsigned long end = millis();
    debug.println("Telegram: took " + String(end - start) + "ms");
#endif // TELEGRAM_LOG_TIMINGS
}

void telegram_hello() {
    for (int n = 0; n < (sizeof(trusted_chat_ids) / sizeof(trusted_chat_ids[0])); n++) {
        bot.sendMessage(trusted_chat_ids[n], "Giess-o-mat v" FIRMWARE_VERSION " initialized.\nSend /auto to begin.", "");
    }
}
#endif // TELEGRAM_TOKEN

bool wifi_write_database(int duration, const char *type, int id) {
    bool success = false;

#ifdef ENABLE_INFLUXDB_LOGGING
    // we still want to be locally usable / have a snappy ui
    // even when the wifi connection is broken.
    if (WiFi.status() != WL_CONNECTED) {
        debug.println("Won't attempt db write, no WiFi connection.");
        return success;
    }

    InfluxData measurement(type);
    measurement.addTag("version", FIRMWARE_VERSION);
    measurement.addTag("device", WiFi.macAddress());
    measurement.addTag("id", String(id));
    measurement.addValue("duration", duration);
    success = influx.write(measurement);
#endif // ENABLE_INFLUXDB_LOGGING

    if (!success) {
        debug.print("Error writing to InfluxDB ");
        debug.print(INFLUXDB_HOST);
        debug.print(":");
        debug.print(INFLUXDB_PORT);
        debug.print("/");
        debug.print(INFLUXDB_DATABASE);
        debug.print("/");
        debug.println(type);
    }

    return success;
}

void wifi_set_message_buffer(String a, String b, String c, String d) {
    message_buffer_a = a;
    message_buffer_b = b;
    message_buffer_c = c;
    message_buffer_d = d;
}

void wifi_schedule_websocket(void) {
    last_websocket_update_time = 0;
}

void wifi_send_status_broadcast(void) {
    if (WiFi.status() != WL_CONNECTED) {
        return;
    }

    if (socket.connectedClients() <= 0) {
        return;
    }
    
    String a = message_buffer_a ;
    String b = message_buffer_b;
    String c = message_buffer_c;
    String d = message_buffer_d;
    
    a.replace("\"", "'");
    b.replace("\"", "'");
    c.replace("\"", "'");
    d.replace("\"", "'");
    
    String ws = "{\n";
    
    ws += "\"a\": \"" + a + "\",\n";
    ws += "\"b\": \"" + b + "\",\n";
    ws += "\"c\": \"" + c + "\",\n";
    ws += "\"d\": \"" + d + "\",\n";
    
    ws += "\"state\": \"" + String(control_state_name()) + "\",\n";
    
    ws += F("\"valves\": [ ");
    for (int i = 0; i < VALVE_COUNT; i++) {
        ws += "\"";
        ws += get_plants()->getValves()->getPin(i) ? "1" : "0";
        ws += "\"";
        
        if (i < (VALVE_COUNT - 1)) {
            ws += ", ";
        }
    }
    ws += " ],\n";
    
    ws += F("\"pumps\": [ ");
    for (int i = 0; i < PUMP_COUNT; i++) {
        ws += "\"";
        ws += get_plants()->getPumps()->getPin(i) ? "1" : "0";
        ws += "\"";
        
        if (i < (PUMP_COUNT - 1)) {
            ws += ", ";
        }
    }
    ws += " ],\n";
    
    ws += F("\"switches\": [ ");
    for (int i = 0; i < SWITCH_COUNT; i++) {
        bool v = get_plants()->getSwitches()->getPin(i);
        
#ifdef INVERT_SENSOR_BOTTOM
        if (i == 0) {
            v = !v;
        }
#endif // INVERT_SENSOR_BOTTOM
        
#ifdef INVERT_SENSOR_TOP
        if (i == 1) {
            v = !v;
        }
#endif // INVERT_SENSOR_TOP

        ws += "\"";
        ws += v ? "1" : "0";
        ws += "\"";
        
        if (i < (SWITCH_COUNT - 1)) {
            ws += ", ";
        }
    }
    ws += " ],\n";

    ws += F("\"aux\": [ ");
    for (int i = 0; i < AUX_COUNT; i++) {
        ws += "\"";
        ws += get_plants()->getAux()->getPin(i) ? "1" : "0";
        ws += "\"";

        if (i < (AUX_COUNT - 1)) {
            ws += ", ";
        }
    }
    ws += " ],\n";

    ws += F("\"kickstart\": [ ");
    for (int i = 0; i < VALVE_COUNT - 1; i++) {
        ws += "\"";
        ws += get_plants()->getKickstart()->getPin(i) ? "1" : "0";
        ws += "\"";

        if (i < (VALVE_COUNT - 2)) {
            ws += ", ";
        }
    }
    ws += " ],\n";
    
    ws += "\"switchstate\": \"";
    Plants::Waterlevel wl = get_plants()->getWaterlevel();
    if (wl == Plants::empty) {
        ws += F("tank empty");
    } else if (wl == Plants::inbetween) {
        ws += F("tank half-filled");
    } else if (wl == Plants::full) {
        ws += F("tank full");
    } else {
        ws += F("invalid sensor state");
    }
    ws += "\"\n";
    ws += "}";
    
    wifi_send_websocket(ws);
}

void wifi_send_websocket(String s) {
    socket.broadcastTXT(s);
}

void handleRoot() {
    String message = F("<!DOCTYPE html>\n");
    message += F("<html><head>\n");
    message += F("<meta charset='utf-8'/>\n");
    message += F("<meta name='viewport' content='width=device-width, initial-scale=1'/>\n");
    message += F("<title>Gieß-o-mat</title>\n");
    
    message += F("<style type='text/css'>\n");
    message += F(".head {\n");
    message += F("text-align: center;\n");
    message += F("}\n");

    message += F(".container {\n");
    message += F("display: flex;\n");
    message += F("max-width: 1200px;\n");
    message += F("margin: auto;\n");
    message += F("}\n");
    
    message += F(".ui {\n");
    message += F("width: max-content;\n");
    message += F("height: max-content;\n");
    message += F("margin-right: 1em;\n");
    message += F("padding: 0 1.0em;\n");
    message += F("border: 1px dashed black;\n");
    message += F("}\n");
    
    message += F(".io {\n");
    message += F("width: max-content;\n");
    message += F("height: max-content;\n");
    message += F("margin-right: 1em;\n");
    message += F("padding: 0.8em 1.0em;\n");
    message += F("border: 1px dashed black;\n");
    message += F("font-family: monospace;\n");
    message += F("}\n");
    
    message += F(".ioelem {\n");
    message += F("width: max-content;\n");
    message += F("border: 1px solid black;\n");
    message += F("border-radius: 50%;\n");
    message += F("padding: 1em;\n");
    message += F("margin: 0.5em;\n");
    message += F("}\n");
    
    message += F(".info {\n");
    message += F("width: max-content;\n");
    message += F("height: max-content;\n");
    message += F("padding: 0 1.0em;\n");
    message += F("border: 1px dashed black;\n");
    message += F("font-family: monospace;\n");
    message += F("}\n");
    
    message += F(".log {\n");
    message += F("max-height: 300px;\n");
    message += F("padding: 0 1.0em;\n");
    message += F("max-width: 1200px;\n");
    message += F("margin: auto;\n");
    message += F("margin-top: 1.5em;\n");
    message += F("border: 1px dashed black;\n");
    message += F("font-family: monospace;\n");
    message += F("overflow-y: scroll;\n");
    message += F("word-break: break-all;\n");
    message += F("}\n");
    
    message += F("#logbuf {\n");
    message += F("white-space: break-spaces;\n");
    message += F("}\n");
    
    message += F(".pad {\n");
    message += F("background: black;\n");
    message += F("border: 3px solid black;\n");
    message += F("border-radius: 20px;\n");
    message += F("width: max-content;\n");
    message += F("padding: 1.5em;\n");
    message += F("margin-left: auto;\n");
    message += F("margin-right: auto;\n");
    message += F("}\n");
    
    message += F(".pad input {\n");
    message += F("background: #fff0cf;\n");
    message += F("border-radius: 6px;\n");
    message += F("font-weight: bold;\n");
    message += F("font-family: monospace;\n");
    message += F("font-size: 1.2em;\n");
    message += F("padding: 0.5em 1em;\n");
    message += F("margin: 0.5em;\n");
    message += F("}\n");
    
    // https://codepen.io/hawkz/pres/RpPaGK
    message += F(".lcd {\n");
    //message += F("background: #9ea18c;\n");
    message += F("background: #9ed18c;\n");
    message += F("border: 5px solid black;\n");
    message += F("border-radius: 10px;\n");
    message += F("width: max-content;\n");
    message += F("padding: 0.65em 1em;\n");
    message += F("box-shadow: inset 0 0 5px 5px rgba(0,0,0,.1);\n");
    message += F("font-weight: bold;\n");
    message += F("font-family: monospace;\n");
    message += F("letter-spacing: 0.1em;\n");
    message += F("font-size: 1.2em;\n");
    message += F("line-height: 160%;\n");
    message += F("color: #21230e;\n");
    message += F("text-shadow: -1px 2px 1px rgba(0,0,0,.1);\n");
    message += F("margin-left: auto;\n");
    message += F("margin-right: auto;\n");
    message += F("}\n");
    
    message += F("#state {\n");
    message += F("text-align: center;\n");
    message += F("}\n");

    message += F("@media (prefers-color-scheme: dark) {\n");
    message += F("body {\n");
    message += F("background-color: black;\n");
    message += F("color: white;\n");
    message += F("}\n");
    message += F(".ui {\n");
    message += F("border: 1px dashed white;\n");
    message += F("}\n");
    message += F(".io {\n");
    message += F("border: 1px dashed white;\n");
    message += F("}\n");
    message += F(".ioelem {\n");
    message += F("border: 1px solid white;\n");
    message += F("}\n");
    message += F(".info {\n");
    message += F("border: 1px dashed white;\n");
    message += F("}\n");
    message += F(".log {\n");
    message += F("border: 1px dashed white;\n");
    message += F("}\n");
    message += F(".pad {\n");
    message += F("background: #222222;\n");
    message += F("border: 3px solid white;\n");
    message += F("}\n");
    message += F(".lcd {\n");
    message += F("border: 5px solid white;\n");
    message += F("}\n");
    message += F("}\n");

    message += F("</style>\n");
    message += F("</head><body>\n");
    message += F("<h1 class='head'>Gieß-o-mat</h1>\n");

    message += F("<div class='container'>\n");
    message += F("<div class='ui'>\n");
    message += F("<pre class='lcd'>\n");
    message += message_buffer_a + '\n';
    message += message_buffer_b + '\n';
    message += message_buffer_c + '\n';
    message += message_buffer_d + '\n';
    message += F("</pre>\n");
    
    message += F("<form class='pad'>\n");
    message += F("<input type='button' value='1'>");
    message += F("<input type='button' value='2'>");
    message += F("<input type='button' value='3'>");
    message += F("<br>\n");
    message += F("<input type='button' value='4'>");
    message += F("<input type='button' value='5'>");
    message += F("<input type='button' value='6'>");
    message += F("<br>\n");
    message += F("<input type='button' value='7'>");
    message += F("<input type='button' value='8'>");
    message += F("<input type='button' value='9'>");
    message += F("<br>\n");
    message += F("<input type='button' value='*'>");
    message += F("<input type='button' value='0'>");
    message += F("<input type='button' value='#'>");
    message += F("</form>\n");
    
    message += F("<p id='state'>\n");
    message += F("State: ");
    message += control_state_name();
    message += F("</p></div>\n");
    
    message += F("<div class='io'>\n");
    message += F("Switches: <span id='switchstate'>");
    
    Plants::Waterlevel wl = get_plants()->getWaterlevel();
    if (wl == Plants::empty) {
        message += F("tank empty");
    } else if (wl == Plants::inbetween) {
        message += F("tank half-filled");
    } else if (wl == Plants::full) {
        message += F("tank full");
    } else {
        message += F("invalid sensor state");
    }
    message += F("</span>");
    
    message += F("<div class='container'>\n");
    for (int i = 0; i < SWITCH_COUNT; i++) {
        message += F("<div class='ioelem switch' style='background-color: ");
        bool v = get_plants()->getSwitches()->getPin(i);
        
#ifdef INVERT_SENSOR_BOTTOM
        if (i == 0) {
            v = !v;
        }
#endif // INVERT_SENSOR_BOTTOM
        
#ifdef INVERT_SENSOR_TOP
        if (i == 1) {
            v = !v;
        }
#endif // INVERT_SENSOR_TOP
        
        if (v) {
            message += F("red");
        } else {
            message += F("green");
        }
        message += F(";'>S");
        message += String(i + 1);
        message += F("</div>");
    }
    message += F("</div><hr>\n");

    message += F("Kickstart:\n");
    message += F("<div class='container'>\n");
    for (int i = 0; i < VALVE_COUNT - 1; i++) {
        message += F("<div class='ioelem kickstart' style='background-color: ");
        if (get_plants()->getKickstart()->getPin(i)) {
            message += F("red");
        } else {
            message += F("green");
        }
        message += F(";'>A");
        message += String(i + 1);
        message += F("</div>");
    }
    message += F("</div><hr>\n");
    
    message += F("Valves:\n");
    message += F("<div class='container'>\n");
    for (int i = 0; i < VALVE_COUNT; i++) {
        message += F("<div class='ioelem valve' style='background-color: ");
        if (get_plants()->getValves()->getPin(i)) {
            message += F("red");
        } else {
            message += F("green");
        }
        message += F(";'>V");
        message += String(i + 1);
        message += F("</div>");
    }
    message += F("</div><hr>\n");
    
    message += F("Pumps:\n");
    message += F("<div class='container'>\n");
    for (int i = 0; i < PUMP_COUNT; i++) {
        message += F("<div class='ioelem pump' style='background-color: ");
        if (get_plants()->getPumps()->getPin(i)) {
            message += F("red");
        } else {
            message += F("green");
        }
        message += F(";'>P");
        message += String(i + 1);
        message += F("</div>");
    }
    message += F("</div><hr>\n");

    message += F("Aux:\n");
    message += F("<div class='container'>\n");
    for (int i = 0; i < AUX_COUNT; i++) {
        message += F("<div class='ioelem aux' style='background-color: ");
        if (get_plants()->getAux()->getPin(i)) {
            message += F("red");
        } else {
            message += F("green");
        }
        message += F(";'>A");
        message += String(i + 1);
        message += F("</div>");
    }
    message += F("</div><hr>\n");

    message += F("Green means valve is closed / pump is off / switch is not submersed.\n");
    message += F("<br>\n");
    message += F("Red means valve is open / pump is running / switch is submersed.</div>\n");
    
    message += F("<div class='info'><p>\n");
    message += F("Version: ");
    message += FIRMWARE_VERSION;
    message += F("\n<br>\n");
    message += F("Build Date: ");
    message += __DATE__;
    message += F("\n<br>\n");
    message += F("Build Time: ");
    message += __TIME__;
    message += F("\n</p><p>\n");
    message += F("MAC: ");
    message += WiFi.macAddress();
    message += F("\n<br>\n");
    message += F("IPv4: ");
    message += WiFi.localIP().toString();
    message += F("\n</p>\n");

#if defined(ARDUINO_ARCH_ESP8266)
    
    message += F("\n<p>\n");
    message += F("Reset reason: ");
    message += ESP.getResetReason();
    message += F("\n<br>\n");
    message += F("Free heap: ");
    message += String(ESP.getFreeHeap());
    message += F(" (");
    message += String(ESP.getHeapFragmentation());
    message += F("% fragmentation)");
    message += F("\n<br>\n");
    message += F("Free sketch space: ");
    message += String(ESP.getFreeSketchSpace());
    message += F("\n<br>\n");
    message += F("Flash chip real size: ");
    message += String(ESP.getFlashChipRealSize());

    if (ESP.getFlashChipSize() != ESP.getFlashChipRealSize()) {
        message += F("\n<br>\n");
        message += F("WARNING: sdk chip size (");
        message += (ESP.getFlashChipSize());
        message += F(") does not match!");
    }
    
    message += F("\n</p>\n");
    
#elif defined(ARDUINO_ARCH_ESP32)

    message += F("\n<p>\n");
    message += F("Free heap: ");
    message += String(ESP.getFreeHeap() / 1024.0);
    message += F("k\n<br>\n");
    message += F("Free sketch space: ");
    message += String(ESP.getFreeSketchSpace() / 1024.0);
    message += F("k\n<br>\n");
    message += F("Flash chip size: ");
    message += String(ESP.getFlashChipSize() / 1024.0);
    message += F("k\n</p>\n");
    
#endif

    message += F("<p>\n");
#ifdef TELEGRAM_TOKEN
    message += F("Telegram: ");
    message += TELEGRAM_UPDATE_INTERVAL_SLOW;
    message += F("ms / ");
    message += TELEGRAM_UPDATE_INTERVAL_FAST;
    message += F("ms\n");
#else
    message += F("Telegram bot not enabled!\n");
#endif
    message += F("</p>\n");

    message += F("<p>\n");
#ifdef ENABLE_INFLUXDB_LOGGING
    message += F("InfluxDB: ");
    message += INFLUXDB_DATABASE;
    message += F(" @ ");
    message += INFLUXDB_HOST;
    message += F(":");
    message += String(INFLUXDB_PORT);
    message += F("\n");
#else
    message += F("InfluxDB logging not enabled!\n");
#endif
    message += F("</p>\n");

    message += F("<p>Try <a href='/update'>/update</a> for OTA firmware updates!</p>\n");
    message += F("<p>Made by <a href='https://xythobuz.de'>xythobuz</a></p>\n");
    message += F("<p><a href='https://git.xythobuz.de/thomas/giess-o-mat'>Project Repository</a></p>\n");
    
#ifdef ENABLE_GPIO_TEST
    message += F("<p><a href='/gpiotest'>GPIO Test</a></p>\n");
#endif // ENABLE_GPIO_TEST
    
    message += F("</div></div>\n");
    
    message += F("<div class='log'><pre id='logbuf'>\n");
    message += debug.getBuffer();
    message += F("</pre></div>\n");
    message += F("</body>\n");
    
    message += F("<script type='text/javascript'>\n");
    message += F("var socket = new WebSocket('ws://' + window.location.hostname + ':81');\n");
    message += F("socket.onmessage = function(e) {\n");
    message += F(    "if (e.data.startsWith('log:')) {\n");
    message += F(        "var log = document.getElementById('logbuf');\n");
    message += F(        "var div = document.getElementsByClassName('log')[0];\n");
    message += F(        "log.innerHTML += e.data.substring(4);\n");
    message += F(        "if (log.innerHTML.length > (1024 * 1024)) {\n");
    message += F(            "log.innerHTML = log.innerHTML.substring(1024 * 1024);\n");
    message += F(        "}\n");
    message += F(        "div.scrollTop = div.scrollHeight;\n");
    message += F(        "return;\n");
    message += F(    "}\n");
    message += F(    "var msg = JSON.parse(e.data);\n");
    message += F(    "var str = msg.a + '\\n' + msg.b + '\\n' + msg.c + '\\n' + msg.d;\n");
    //message += F(    "console.log(str);\n");
    message += F(    "var lcd = document.getElementsByClassName('lcd');\n");
    message += F(    "lcd[0].innerHTML = str;\n");
    message += F(    "var state = document.getElementById('state');\n");
    message += F(    "state.innerHTML = \"State: \" + msg.state;\n");
    
    message += F(    "for (let i = 0; i < ");
    message += String(VALVE_COUNT);
    message += F("; i++) {\n");
    message += F(       "var valves = document.getElementsByClassName('valve');\n");
    message += F(       "if (msg.valves[i] == '0') {\n");
    message += F(           "valves[i].style = 'background-color: green;';\n");
    message += F(       "} else {\n");
    message += F(           "valves[i].style = 'background-color: red;';\n");
    message += F(       "}\n");
    message += F(    "}\n");
    
    message += F(    "for (let i = 0; i < ");
    message += String(PUMP_COUNT);
    message += F("; i++) {\n");
    message += F(       "var pumps = document.getElementsByClassName('pump');\n");
    message += F(       "if (msg.pumps[i] == '0') {\n");
    message += F(           "pumps[i].style = 'background-color: green;';\n");
    message += F(       "} else {\n");
    message += F(           "pumps[i].style = 'background-color: red;';\n");
    message += F(       "}\n");
    message += F(    "}\n");
    
    message += F(    "for (let i = 0; i < ");
    message += String(SWITCH_COUNT);
    message += F("; i++) {\n");
    message += F(       "var switches = document.getElementsByClassName('switch');\n");
    message += F(       "if (msg.switches[i] == '0') {\n");
    message += F(           "switches[i].style = 'background-color: green;';\n");
    message += F(       "} else {\n");
    message += F(           "switches[i].style = 'background-color: red;';\n");
    message += F(       "}\n");
    message += F(    "}\n");

    message += F(    "for (let i = 0; i < ");
    message += String(AUX_COUNT);
    message += F("; i++) {\n");
    message += F(       "var aux = document.getElementsByClassName('aux');\n");
    message += F(       "if (msg.aux[i] == '0') {\n");
    message += F(           "aux[i].style = 'background-color: green;';\n");
    message += F(       "} else {\n");
    message += F(           "aux[i].style = 'background-color: red;';\n");
    message += F(       "}\n");
    message += F(    "}\n");

    message += F(    "for (let i = 0; i < ");
    message += String(VALVE_COUNT - 1);
    message += F("; i++) {\n");
    message += F(       "var kickstart = document.getElementsByClassName('kickstart');\n");
    message += F(       "if (msg.kickstart[i] == '0') {\n");
    message += F(           "kickstart[i].style = 'background-color: green;';\n");
    message += F(       "} else {\n");
    message += F(           "kickstart[i].style = 'background-color: red;';\n");
    message += F(       "}\n");
    message += F(    "}\n");
    
    message += F(    "var switchstate = document.getElementById('switchstate');\n");
    message += F(    "switchstate.innerHTML = msg.switchstate;\n");
    message += F("};\n");
    message += F("var hist = document.getElementsByClassName('log')[0];\n");
    message += F("hist.scrollTop = hist.scrollHeight;\n");
    
    message += F("var buttons = document.getElementsByTagName('input');\n");
    message += F("for (let i = 0; i < buttons.length; i++) {\n");
    message += F(    "buttons[i].addEventListener('click', updateButton);\n");
    message += F("}\n");
    message += F("function updateButton() {\n");
    message += F(    "socket.send(this.value);\n");
    message += F("}\n");
    message += F("</script>\n");
    message += F("</html>\n");

    server.send(200, "text/html", message);
}

void webSocketEvent(uint8_t num, WStype_t type, uint8_t *payload, size_t length) {
    if ((type != WStype_TEXT) || (length != 1)) {
        debug.println("Websocket: invalid type=" + String(type) + " len=" + String(length) + " data=" + String((char *)payload));
        return;
    }
    
    char c = payload[0];
    if ((c >= '0') && (c <= '9')) {
        control_act_input(c - '0');
    } else if (c == '*') {
        control_act_input(-1);
    } else if (c == '#') {
        control_act_input(-2);
    }
}

void wifi_setup() {
    // Build hostname string
    String hostname = "giess-o-mat";

    int ws = 0, connect_attempts = 0;

#if defined(ARDUINO_ARCH_ESP8266)

    // Connect to WiFi AP
    debug.println("WiFi: initializing");
    WiFi.hostname(hostname);
    WiFi.mode(WIFI_STA);
    
    debug.print("WiFi: connecting");
    WiFi.begin(WIFI_SSID, WIFI_PW);

#ifdef TELEGRAM_TOKEN
    secured_client.setTrustAnchors(&cert);
#endif // TELEGRAM_TOKEN

    while (((ws = WiFi.status()) != WL_CONNECTED) && (connect_attempts < MAX_WIFI_CONNECT_ATTEMPTS)) {
        connect_attempts++;
        debug.print(String(" ") + String(ws));
        delay(LED_CONNECT_BLINK_INTERVAL);
        digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));
    }
    debug.println();
    debug.println(String("WiFi: status=") + String(WiFi.status()));
    
#elif defined(ARDUINO_ARCH_ESP32)

    // Set hostname workaround
    debug.println("WiFi: set hostname");
    WiFi.config(INADDR_NONE, INADDR_NONE, INADDR_NONE);
    WiFi.setHostname(hostname.c_str());
    
    // Workaround for WiFi connecting only every 2nd reset
    // https://github.com/espressif/arduino-esp32/issues/2501#issuecomment-513602522
    debug.println("WiFi: connection work-around");
    WiFi.onEvent([](WiFiEvent_t event, WiFiEventInfo_t info) {
        if (info.disconnected.reason == 202) {
            esp_sleep_enable_timer_wakeup(10);
            esp_deep_sleep_start();
            delay(100);
        }
    }, WiFiEvent_t::SYSTEM_EVENT_STA_DISCONNECTED);

    // Connect to WiFi AP
    debug.println("WiFi: SSID=" WIFI_SSID);
    debug.print("WiFi: connecting");
    WiFi.mode(WIFI_STA);
    WiFi.begin(WIFI_SSID, WIFI_PW);

#ifdef TELEGRAM_TOKEN
    secured_client.setCACert(TELEGRAM_CERTIFICATE_ROOT);
#endif // TELEGRAM_TOKEN

    while (((ws = WiFi.status()) != WL_CONNECTED) && (connect_attempts < MAX_WIFI_CONNECT_ATTEMPTS)) {
        connect_attempts++;
        debug.print(String(" ") + String(ws));
        delay(LED_CONNECT_BLINK_INTERVAL);
        digitalWrite(BUILTIN_LED_PIN, !digitalRead(BUILTIN_LED_PIN));
    }
    debug.println();
    debug.println(String("WiFi: status=") + String(WiFi.status()));
    
    // Set hostname workaround
    debug.println("WiFi: set hostname work-around");
    WiFi.setHostname(hostname.c_str());

#endif

    if ((connect_attempts >= MAX_WIFI_CONNECT_ATTEMPTS)
            || (WiFi.status() != WL_CONNECTED)) {
        debug.println("WiFi: init failed!");
        wifi_ok = 0;
        return;
    }
    wifi_ok = 1;

    debug.print("WiFi: got IPv4: ");
    debug.println(WiFi.localIP().toString());

#ifdef ENABLE_INFLUXDB_LOGGING
    // Setup InfluxDB Client
    debug.println("WiFi: set InfluxDB database");
    influx.setDb(INFLUXDB_DATABASE);
#endif // ENABLE_INFLUXDB_LOGGING

    // Setup HTTP Server
    debug.println("WiFi: initializing HTTP server");
    MDNS.begin(hostname.c_str());
    updater.setup(&server);
    server.on("/", handleRoot);
    
#ifdef ENABLE_GPIO_TEST
    server.on("/gpiotest", handleGpioTest);
#endif // ENABLE_GPIO_TEST

#ifdef TELEGRAM_TOKEN
    debug.print("WiFi: getting NTP time");
    configTime(0, 0, "pool.ntp.org");
    time_t now = time(nullptr);
    while (now < 24 * 60 * 60) {
        debug.print(".");
        delay(100);
        now = time(nullptr);
    }
    debug.println(" done!");
    debug.println("WiFi: time is " + String(now));

    debug.println("WiFi: initializing Telegram");
    const String commands = F("["
        "{\"command\":\"auto\", \"description\":\"Start automatic watering cycle\"},"
        "{\"command\":\"confirm\", \"description\":\"Proceed with any menu inputs\"},"
        "{\"command\":\"none\", \"description\":\"Proceed without menu input\"},"
        "{\"command\":\"abort\", \"description\":\"Cancel any menu inputs\"}"
    "]");
    bot.setMyCommands(commands);
    telegram_hello();
#endif // TELEGRAM_TOKEN

    server.begin();
    MDNS.addService("http", "tcp", 80);
    MDNS.addService("http", "tcp", 81);

    socket.begin();
    socket.onEvent(webSocketEvent);
    
    debug.println("WiFi: setup done");
}

void wifi_run() {
    // reset ESP every 6h to be safe
    if ((millis() >= (6UL * 60UL * 60UL * 1000UL)) && (sm_is_idle())) {
        ESP.restart();
    }

    if (!wifi_ok) {
        // nothing to handle
        return;
    }

    if ((millis() - last_server_handle_time) >= SERVER_HANDLE_INTERVAL) {
        last_server_handle_time = millis();
        server.handleClient();
        socket.loop();
        
#ifdef ARDUINO_ARCH_ESP8266
        MDNS.update();
#endif // ARDUINO_ARCH_ESP8266
    }
    
    if ((millis() - last_websocket_update_time) >= WEBSOCKET_UPDATE_INTERVAL) {
        last_websocket_update_time = millis();
        wifi_send_status_broadcast();
    }
    
#ifdef ENABLE_GPIO_TEST
    if (runningGpioTest && ((millis() - lastGpioTime) >= GPIO_TEST_INTERVAL)) {
        gpioTestState = !gpioTestState;
        runGpioTest(gpioTestState);
    }
#endif // ENABLE_GPIO_TEST

#ifdef TELEGRAM_TOKEN
    if ((millis() - last_telegram_time) >= telegram_update_interval()) {
        telegram_poll();
        last_telegram_time = millis();
    }
#endif // TELEGRAM_TOKEN
}

#endif // PLATFORM_ESP