#ifndef _FIRMWARE_UPGRADE_H
#define _FIRMWARE_UPGRADE_H

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

#include <Http.h>

class FirmwareUpgrade {
public:
    FirmwareUpgrade(Http& http);
    ~FirmwareUpgrade();

    void SetCheckVersionUrl(std::string check_version_url);
    void SetPostData(const std::string& post_data);
    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:
    Http& http_;
    std::string check_version_url_;
    bool has_new_version_ = false;
    std::string firmware_version_;
    std::string firmware_url_;
    std::string post_data_;
    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);
};

#endif // _FIRMWARE_UPGRADE_H