/**
 * @copyright Utilitech AS 2023
 * License: Fair Source
 * 
 */

#ifndef _ENERGYACCOUNTING_H
#define _ENERGYACCOUNTING_H

#include "Arduino.h"
#include "AmsData.h"
#include "AmsDataStorage.h"
#include "PriceService.h"

struct EnergyAccountingPeak {
    uint8_t day;
    uint8_t hour;
    uint16_t value;
};

struct EnergyAccountingPeak6 {
    uint8_t day;
    uint16_t value;
};

struct EnergyAccountingData {
    uint8_t version;
    uint8_t month;
    int32_t costToday;
    int32_t costYesterday;
    int32_t costThisMonth;
    int32_t costLastMonth;
    int32_t incomeToday;
    int32_t incomeYesterday;
    int32_t incomeThisMonth;
    int32_t incomeLastMonth;
    uint32_t lastMonthImport;
    uint32_t lastMonthExport;
    uint8_t lastMonthAccuracy;
    EnergyAccountingPeak peaks[5];
    time_t lastUpdated;
};

struct EnergyAccountingData6 {
    uint8_t version;
    uint8_t month;
    int32_t costYesterday;
    int32_t costThisMonth;
    int32_t costLastMonth;
    int32_t incomeYesterday;
    int32_t incomeThisMonth;
    int32_t incomeLastMonth;
    uint32_t lastMonthImport;
    uint32_t lastMonthExport;
    uint8_t lastMonthAccuracy;
    EnergyAccountingPeak6 peaks[5];
};

struct EnergyAccountingRealtimeData {
    uint8_t magic;
    uint8_t currentHour;
    uint8_t currentDay;
    uint8_t currentThresholdIdx;
    float use;
    float costHour;
    float costDay;
    float produce;
    float incomeHour;
    float incomeDay;
    unsigned long lastImportUpdateMillis;
    unsigned long lastExportUpdateMillis;
};


class EnergyAccounting {
public:
    #if defined(AMS_REMOTE_DEBUG)
    EnergyAccounting(RemoteDebug*, EnergyAccountingRealtimeData*);
    #else
    EnergyAccounting(Stream*, EnergyAccountingRealtimeData*);
    #endif
    void setup(AmsDataStorage *ds, EnergyAccountingConfig *config);
    void setPriceService(PriceService *ps);
    void setTimezone(Timezone*);
    EnergyAccountingConfig* getConfig();
    bool update(AmsData* amsData);
    bool load();
    bool save();
    bool isInitialized();

    float getUseThisHour();
    float getUseToday();
    float getUseThisMonth();
    float getUseLastMonth();

    float getProducedThisHour();
    float getProducedToday();
    float getProducedThisMonth();
    float getProducedLastMonth();

    float getCostThisHour();
    float getCostToday();
    float getCostYesterday();
    float getCostThisMonth();
    float getCostLastMonth();

    float getIncomeThisHour();
    float getIncomeToday();
    float getIncomeYesterday();
    float getIncomeThisMonth();
    float getIncomeLastMonth();

    float getMonthMax();
    uint8_t getCurrentThreshold();
    EnergyAccountingPeak getPeak(uint8_t);

    EnergyAccountingData getData();
    void setData(EnergyAccountingData&);

    void setCurrency(String currency);

private:
    #if defined(AMS_REMOTE_DEBUG)
    RemoteDebug* debugger = NULL;
    #else
    Stream* debugger = NULL;
    #endif
    bool init = false, initPrice = false;
    AmsDataStorage *ds = NULL;
    PriceService *ps = NULL;
    EnergyAccountingConfig *config = NULL;
    Timezone *tz = NULL;
    EnergyAccountingData data = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
    EnergyAccountingRealtimeData* realtimeData = NULL;
    String currency = "";

    void calcDayCost();
    bool updateMax(uint16_t val, uint8_t day, uint8_t hour);
};

#endif