WeatherModule.hpp

#pragma once

#include "DrawableModule.hpp"
#include <U8g2_for_Adafruit_GFX.h>
#include "PsramUtils.hpp"
#include "GeneralTimeConverter.hpp"
#include "WeatherIcons_Main.hpp"
#include "WeatherIcons_Special.hpp"
#include "WeatherIconCache.hpp" // NEU: für globalWeatherIconCache
#include <set>

class WebClientModule;
struct DeviceConfig;

struct WeatherCurrentData {
    float temp, feels_like, dew_point, uvi, wind_speed, wind_gust, pop;
    int humidity, clouds;
    time_t sunrise, sunset;
    PsramString icon_name;
};

struct WeatherHourlyData {
    time_t dt;
    float temp, feels_like, pop, rain_1h, snow_1h;
    PsramString icon_name;
};

struct WeatherDailyData {
    time_t dt;
    float temp_min, temp_max, temp_mean, pop, rain, snow, wind_speed, sunshine_duration;
    int cloud_cover;
    time_t sunrise, sunset; 
    PsramString icon_name;
};

struct WeatherAlertData {
    PsramString event;
    time_t start, end;
};

enum class WeatherPageType {
    CURRENT_WEATHER,             // Page 1: NOW weather with icon, temp, feels, humidity, clouds, rain forecast
    TODAY_PART1,                 // Page 2: Today min/max temp, sunrise, sunset
    TODAY_PART2,                 // Page 3: Today cloud coverage, precipitation, wind, mean temp
    PRECIPITATION_CHART,         // Page 4: Area chart for rain/snow (next 24 hours)
    TEMPERATURE_CHART,           // Page 5: Area chart for temperature (next 24 hours)
    HOURLY_FORECAST,             // Page 6+: Hourly forecasts
    DAILY_FORECAST,              // Page 7+: Daily forecasts
    DAILY_TEMPERATURE_CHART,     // Daily temperature chart (area for max, line for min)
    DAILY_PRECIPITATION_CHART,   // Daily precipitation chart (area with POP coloring)
    ALERT                        // Alert page
};

struct WeatherPage {
    WeatherPageType type;
    int index; 
};

class WeatherModule : public DrawableModule {
public:
    WeatherModule(U8G2_FOR_ADAFRUIT_GFX& u8g2, GFXcanvas16& canvas, const GeneralTimeConverter& timeConverter, WebClientModule* webClient);
    ~WeatherModule();

    void begin();
    void logicTick() override;
    void draw() override;
    void onUpdate(std::function<void()> callback);
    bool isEnabled() override;
    unsigned long getDisplayDuration() override;
    void resetPaging() override;
    void activateModule(uint32_t uid) override;

    void setConfig(const DeviceConfig* config);
    void queueData();
    void processData();
    void periodicTick();

private:
    U8G2_FOR_ADAFRUIT_GFX& _u8g2;
    GFXcanvas16& _canvas;
    const GeneralTimeConverter& _timeConverter;
    WebClientModule* _webClient = nullptr;
    const DeviceConfig* _config = nullptr;

    SemaphoreHandle_t _dataMutex = nullptr;

    PsramString _forecastApiUrl;
    PsramString _climateApiUrl;

    char* _pendingForecastBuffer = nullptr;
    size_t _forecastBufferSize = 0;
    bool _forecastDataPending = false;
    
    char* _pendingClimateBuffer = nullptr;
    size_t _climateBufferSize = 0;
    bool _climateDataPending = false;
    
    float _historicalMonthlyAvgTemp = 10.0;
    time_t _lastForecastUpdate = 0;
    time_t _lastClimateUpdate = 0;
    time_t _lastUrlBuildTime = 0;
    bool _dataAvailable = false;

    WeatherCurrentData _currentWeather;
    PsramVector<WeatherHourlyData> _hourlyForecast;
    PsramVector<WeatherDailyData> _dailyForecast;
    PsramVector<WeatherAlertData> _alerts;

    PsramVector<WeatherPage> _pages;
    int _currentPageIndex = 0;
    unsigned int _pageTicks = 0;

    bool _isUrgentViewActive = false;
    uint32_t _currentUrgentUID = 0;
    unsigned long _lastUrgentDisplayTime = 0;
    unsigned long _lastPeriodicCheck = 0;

    std::function<void()> _onUpdateCallback = nullptr;
    
    // Track missing icons to log only once (using PsramAllocator to reduce heap usage)
    std::set<PsramString, std::less<PsramString>, PsramAllocator<PsramString>> _loggedMissingIcons;

    void buildApiUrls();
    void parseForecastData(char* jsonBuffer, size_t size);
    void parseClimateData(char* jsonBuffer, size_t size);

    uint16_t getClimateColorSmooth(float temp);
    PsramString mapWeatherCodeToIcon(int code, bool is_day);
    bool isNightTime(time_t timestamp) const;

    void buildPages();
    void getDayName(char* buf, size_t buf_len, time_t epoch);
    void formatTime(char* buf, size_t buf_len, time_t epoch);
    uint16_t dimColor(uint16_t color, float brightness);

    // NEU: drawWeatherIcon mit Registry/Cache, und PSRAM everywhere:
    void drawWeatherIcon(int x, int y, int size, const PsramString& name, bool isNight);

    void drawCurrentWeatherPage();        // Page 1: NOW weather
    void drawTodayPart1Page();           // Page 2: Today summary part 1
    void drawTodayPart2Page();           // Page 3: Today summary part 2
    void drawPrecipitationChartPage();   // Page 4: Area chart for next 24h precipitation
    void drawTemperatureChartPage();     // Page 5: Area chart for next 24h temperature
    void drawHourlyForecastPage(int pageIndex);  // Page 6+: Hourly forecasts (multiple pages)
    void drawDailyForecastPage(int pageIndex);   // Page 7+: Daily forecasts (multiple pages)
    void drawDailyTemperatureChartPage();        // Daily temperature chart (7 days, area for max, line for min)
    void drawDailyPrecipitationChartPage();      // Daily precipitation chart (7 days, area with POP coloring)
    void drawAlertPage(int index);       // Alert page
    void drawNoDataPage();
};