PagesHandlers.cpp

/*
 * WaterTankController project
 * Copyright (c) 2026 Fedir Vilhota <fredy31415@gmail.com>
 * This software is released under the MIT License.
 * See the LICENSE file in the project root for full license information.
 */
#include "PagesHandlers.h"
#include "TankMonitor.h"
#include "RelayControl.h"
#include "html_templates.h"
#include <WiFi.h>
#include <time.h>
#include <LittleFS.h>

// Helper to format float to string with 2 decimals without using dtostrf/snprintf
static String formatFloat(float val) {
    int intPart = (int)val;
    int fracPart = (int)((val - intPart) * 100);
    if (fracPart < 0) fracPart = -fracPart;
    return String(intPart) + "." + (fracPart < 10 ? "0" : "") + String(fracPart);
}


PagesHandlers::PagesHandlers(TankMonitor& tank_, RelayControl& relay_, Config& configRef)
    : tank(tank_), relay(relay_), _config(configRef)
{}

void PagesHandlers::initPagesHandlers(AsyncWebServer& webServer) {
    setupHomePageHandler(webServer);
    setupVolumesHandler(webServer);
    setupSensorDataHandler(webServer);
    setupRelayStatesHandler(webServer);
    setupSetRelayHandler(webServer);
    setupSetRelayModeHandler(webServer);
    setupLogsListHandler(webServer);
    setupLogChartHandler(webServer);
    setupLogDownloadHandler(webServer);
    setupSetNumZonesHandler(webServer);
    setupStatusHandler(webServer);
}

void PagesHandlers::setupHomePageHandler(AsyncWebServer& webServer) {
    webServer.on("/", HTTP_GET, [this](AsyncWebServerRequest *request){
        String html = String(HTML_HOME_PAGE);
        html.replace("{DATE}", __DATE__);
        html.replace("{TIME}", __TIME__);
        html.replace("{TIME}", __TIME__);
        html.replace("{DISTANCE}", formatFloat(tank.getDistance()));
        html.replace("{VOLUME}", formatFloat(tank.getCurrentVolume()));
        html.replace("{RATE}", formatFloat(tank.getAvgRate()));
        html.replace("{ESTIMATED}", formatFloat(tank.getEstimatedTime()));
        html.replace("{IP}", WiFi.localIP().toString());
        html.replace("%PIN_REL1_VAL%", String(relay.getRelayPin(1)));
        html.replace("%PIN_REL2_VAL%", String(relay.getRelayPin(2)));
        html.replace("%PIN_REL3_VAL%", String(relay.getRelayPin(3)));
        html.replace("%PIN_REL4_VAL%", String(relay.getRelayPin(4)));
        String options = "";
        for (int i = 1; i <= 10; ++i) {
            options += "<option value='" + String(i) + "'";
            if (i == _config.getNumZones()) options += " selected";
            options += ">" + String(i) + "</option>";
        }
        html.replace("__OPTIONS__", options);
        request->send(200, "text/html; charset=UTF-8", html);
    });
}

void PagesHandlers::setupVolumesHandler(AsyncWebServer& webServer) {
    webServer.on("/volumes", HTTP_GET, [this](AsyncWebServerRequest *request){
        String volumesStr = tank.getVolumesCsv();
        request->send(200, "text/plain", volumesStr);
    });
}

void PagesHandlers::setupSensorDataHandler(AsyncWebServer& webServer) {
    webServer.on("/sensor_data", HTTP_GET, [this](AsyncWebServerRequest *request){
        String response = "{";
        response += "\"distance\": " + formatFloat(tank.getDistance());
        response += ",\"volume\": " + formatFloat(tank.getCurrentVolume());
        response += ",\"rate\": " + formatFloat(tank.getAvgRate());
        response += ",\"estimated\": " + formatFloat(tank.getEstimatedTime());
        response += "}";
        request->send(200, "application/json", response);
    });
}

void PagesHandlers::setupRelayStatesHandler(AsyncWebServer& webServer) {
    webServer.on("/relay_states", HTTP_GET, [this](AsyncWebServerRequest *request){
        String json = relay.getStatesJson(2); // реалізуйте цей метод у RelayControl
        request->send(200, "application/json", json);
    });
}

void PagesHandlers::setupSetRelayHandler(AsyncWebServer& webServer) {
    webServer.on("/set_relay", HTTP_GET, [this](AsyncWebServerRequest *request){
        if (request->hasParam("pin") && request->hasParam("state")) {
            int relayPin = request->getParam("pin")->value().toInt();
            bool targetState = (request->getParam("state")->value() == "on");
            bool ok = relay.setRelay(relayPin, targetState); // реалізуйте цей метод
            if (ok) request->send(200, "text/plain", "OK");
            else request->send(403, "text/plain", "Forbidden or invalid pin.");
        } else {
            request->send(400, "text/plain", "Missing 'pin' or 'state' parameter.");
        }
    });
}

void PagesHandlers::setupSetRelayModeHandler(AsyncWebServer& webServer) {
    webServer.on("/set_relay_mode", HTTP_GET, [this](AsyncWebServerRequest *request){
        if (request->hasParam("pin") && request->hasParam("mode")) {
            int relayPin = request->getParam("pin")->value().toInt();
            String modeStr = request->getParam("mode")->value();
            int mode = 1;
            if (modeStr == "manual") mode = 0;
            else if (modeStr == "auto") mode = 1;
            //else if (modeStr == "prog") mode = 2;
            bool ok = relay.setControlMode(relayPin, mode); // реалізуйте цей метод
            if (ok) request->send(200, "text/plain", "Mode updated.");
            else request->send(400, "text/plain", "Invalid pin or mode.");
        } else {
            request->send(400, "text/plain", "Missing 'pin' or 'mode' parameter.");
        }
    });
}

void PagesHandlers::setupLogsListHandler(AsyncWebServer& webServer) {
    webServer.on("/logs", HTTP_GET, [](AsyncWebServerRequest *request){
        String html = String(HTML_LOGS_LIST_HEAD);
        File root = LittleFS.open("/");
        File file = root.openNextFile();
        while (file) {
            String fname = file.name();
            if(fname.startsWith("log_") && fname.endsWith(".csv")) {
                String date = fname.substring(4, 14); // YYYY-MM-DD
                html += "<li><a href=\"/log?date=" + date + "\">" + date + "</a> | ";
                html += "<a href=\"/download_log?date=" + date + "\">Завантажити CSV</a></li>";
            }
            file = root.openNextFile();
        }
        html += "</ul></body></html>";
        request->send(200, "text/html; charset=utf-8", html);
    });
}

void PagesHandlers::setupLogChartHandler(AsyncWebServer& webServer) {
    webServer.on("/log", HTTP_GET, [](AsyncWebServerRequest *request){
        if (!request->hasParam("date")) {
            request->send(400, "text/plain", "Параметр date відсутній");
            return;
        }
        String date = request->getParam("date")->value();
        String filename = "/log_" + date + ".csv";
        if (!LittleFS.exists(filename)) {
            request->send(404, "text/plain", "Файл не знайдено");
            return;
        }
        // Зчитуємо дані для JS
        File file = LittleFS.open(filename, "r");
        String data = "[";
        bool first = true;
        int lastTimeInMinutes = -1; // Час попереднього запису в хвилинах від початку дня
        while (file.available()) {
            String line = file.readStringUntil('\n');
            line.trim();
            String t = line.substring(0, line.indexOf(','));
            String v = line.substring(line.indexOf(',')+1);
            int vi = v.toInt(); // Optimized: remove toFloat/roundf since log is integer
            int H, M, S = 0;
            if (t.length() == 5) { // HH:MM
                H = t.substring(0,2).toInt();
                M = t.substring(3,5).toInt();
            } else if (t.length() == 8) { // HH:MM:SS
                H = t.substring(0,2).toInt();
                M = t.substring(3,5).toInt();
                S = t.substring(6,8).toInt();
            } else if (t.length() == 19) {
                H = t.substring(11,13).toInt();
                M = t.substring(14,16).toInt();
                S = t.substring(17,19).toInt();
            } else {
                continue;
            }
            
            // Перевірка: пропускаємо записи з часом, меншим за попередній
            // (це означає, що запис був до синхронізації NTP)
            int currentTimeInMinutes = H * 60 + M;
            if (lastTimeInMinutes != -1 && currentTimeInMinutes < lastTimeInMinutes) {
                // Пропускаємо цей запис
                continue;
            }
            lastTimeInMinutes = currentTimeInMinutes;
            
            int y = date.substring(0,4).toInt();
            int m = date.substring(5,7).toInt() - 1; // JS: 0=січень!
            int d = date.substring(8,10).toInt();
            if (!first) data += ",";
            data += "[Date.UTC(" + String(y) + "," + String(m) + "," + String(d) + "," + String(H) + "," + String(M) + "," + String(S) + ")," + String(vi) + "]";
            first = false;
        }
        data += "]";
        file.close();
        String html = String(HTML_LOG_CHART_TEMPLATE);
        html.replace("{DATE}", date);
        html.replace("{DATA}", data);
        request->send(200, "text/html; charset=utf-8", html);
    });
}

void PagesHandlers::setupSetNumZonesHandler(AsyncWebServer& webServer) {
    webServer.on("/setNumZones", HTTP_GET, [this](AsyncWebServerRequest *request){ // <--- Додаємо [this]
        if (request->hasParam("value")) {
            int nz = request->getParam("value")->value().toInt();
            _config.setNumZones(nz);    // збережи в об’єкті
            _config.save();             // збережи у файл
        }
        request->send(200, "text/plain", "OK");
    });
}

void PagesHandlers::setupLogDownloadHandler(AsyncWebServer& webServer) {
    webServer.on("/download_log", HTTP_GET, [](AsyncWebServerRequest *request){
        if (!request->hasParam("date")) {
            request->send(400, "text/plain", "Параметр date відсутній");
            return;
        }
        String date = request->getParam("date")->value();
        String filename = "/log_" + date + ".csv";
        
        if (!LittleFS.exists(filename)) {
            request->send(404, "text/plain", "Файл не знайдено");
            return;
        }
        
        // Відправляємо файл для завантаження
        // Відправляємо файл для завантаження
        request->send(LittleFS, filename, "text/csv; charset=UTF-8", true);
    });
}

void PagesHandlers::setupStatusHandler(AsyncWebServer& webServer) {
    webServer.on("/api/status", HTTP_GET, [this](AsyncWebServerRequest *request){
        // 1. Timestamp
        time_t now = time(nullptr);
        struct tm *tm_struct = localtime(&now);
        
        // Manual ISO 8601 formatting to ensure correct format
        char timeStr[32];
        snprintf(timeStr, sizeof(timeStr), "%04d-%02d-%02dT%02d:%02d:%02d+02:00", 
                 tm_struct->tm_year + 1900, 
                 tm_struct->tm_mon + 1, 
                 tm_struct->tm_mday, 
                 tm_struct->tm_hour, 
                 tm_struct->tm_min, 
                 tm_struct->tm_sec);
        
        // 2. Sensors
        String sensors = "[";
        sensors += "{\"name\":\"distance\",\"value\":\"" + formatFloat(tank.getDistance()) + "\"},";
        sensors += "{\"name\":\"volume\",\"value\":\"" + formatFloat(tank.getCurrentVolume()) + "\"},";
        sensors += "{\"name\":\"rate\",\"value\":\"" + formatFloat(tank.getAvgRate()) + "\"},";
        sensors += "{\"name\":\"estimated\",\"value\":\"" + formatFloat(tank.getEstimatedTime()) + "\"}";
        sensors += "]";

        // 3. Relays
        String relays = "[";
        for(int i=1; i<=RelayControl::RELAY_COUNT; i++) {
             if(i > 1) relays += ",";
             int pin = relay.getRelayPin(i);
             String name = pin == 8 ? "Будинок" : ( pin == 10 ? "Свердловина" : String(pin));
             relays += "{";
             relays += "\"name\":\"" + name + "\",";
             relays += "\"pin\":" + String(pin) + ",";
             relays += "\"num\":" + String(i) + ",";
             relays += "\"state\":" + String(relay.getRelay(i) ? "true" : "false") + ",";
             relays += "\"mode\":" + String(relay.getControlMode(i));
             relays += "}";
        }
        relays += "]";

        // 4. Logs (Hourly Averages)
        String logs = "[";
        // Calculate date string for filename: log_YYYY-MM-DD.csv
        String dateStr = String(tm_struct->tm_year + 1900) + "-" + 
                         ((tm_struct->tm_mon + 1 < 10) ? "0" : "") + String(tm_struct->tm_mon + 1) + "-" +
                         ((tm_struct->tm_mday < 10) ? "0" : "") + String(tm_struct->tm_mday);
        String filename = "/log_" + dateStr + ".csv";
        
        if (LittleFS.exists(filename)) {
            File file = LittleFS.open(filename, "r");
            long hourlySum[24] = {0};
            int hourlyCount[24] = {0};
            
            while(file.available()) {
                String line = file.readStringUntil('\n');
                line.trim();
                if(line.length() == 0) continue;
                
                // Format: HH:MM,VAL or HH:MM:SS,VAL
                int commaIdx = line.indexOf(',');
                if(commaIdx > 0) {
                    String timePart = line.substring(0, commaIdx);
                    String valPart = line.substring(commaIdx+1);
                    
                    // Extract hour
                    // HH:MM ot HH:MM:SS, so first 2 chars are hour
                    // Assuming standard format from Log.cpp
                    if(timePart.length() >= 2) {
                        int h = timePart.substring(0, 2).toInt();
                        if(h >= 0 && h < 24) {
                            hourlySum[h] += valPart.toInt();
                            hourlyCount[h]++;
                        }
                    }
                }
            }
            file.close();
            
            bool first = true;
            for(int i=0; i<24; i++) {
                if(hourlyCount[i] > 0) {
                    if(!first) logs += ",";
                    logs += "{\"Time\":" + String(i) + ",\"Volume\":" + String(hourlySum[i] / hourlyCount[i]) + "}";
                    first = false;
                }
            }
        }
        logs += "]";

        // Construct final JSON
        String json = "{";
        json += "\"status\":\"ok\",";
        json += "\"message\":\"Service is running smoothly\",";
        json += "\"timestamp\":\"" + String(timeStr) + "\",";
        json += "\"sensors\":" + sensors + ",";
        json += "\"relays\":" + relays + ",";
        json += "\"logs\":" + logs;
        json += "}";

        request->send(200, "application/json", json);
    });
}