#ifndef _FIRMWARE_UPGRADE_H
#define _FIRMWARE_UPGRADE_H

#include <functional>
#include <string>
#include <map>

class FirmwareUpgrade {
public:
    FirmwareUpgrade();
    ~FirmwareUpgrade();

    void SetBoardJson(const std::string& board_json);
    void SetCheckVersionUrl(std::string check_version_url);
    void SetHeader(const std::string& key, const std::string& value);
    void CheckVersion();
    bool HasNewVersion() { return has_new_version_; }
    void StartUpgrade(std::function<void(int progress, size_t speed)> callback);
    void MarkCurrentVersionValid();

private:
    std::string check_version_url_;
    bool has_new_version_ = false;
    std::string firmware_version_;
    std::string firmware_url_;
    std::string board_json_;
    std::map<std::string, std::string> headers_;

    void Upgrade(const std::string& firmware_url);
    std::function<void(int progress, size_t speed)> upgrade_callback_;
    std::vector<int> ParseVersion(const std::string& version);
    bool IsNewVersionAvailable(const std::string& currentVersion, const std::string& newVersion);
    std::string GetPostData();
};

#endif // _FIRMWARE_UPGRADE_H