#pragma once
#ifdef USE_HOST

#include "esphome/core/preference_backend.h"
#include <cstring>
#include <map>
#include <string>
#include <vector>

namespace esphome::host {

class HostPreferences final : public PreferencesMixin<HostPreferences> {
 public:
  using PreferencesMixin<HostPreferences>::make_preference;
  bool sync();
  bool reset();

  ESPPreferenceObject make_preference(size_t length, uint32_t type, bool in_flash);
  ESPPreferenceObject make_preference(size_t length, uint32_t type) { return make_preference(length, type, false); }

  bool save(uint32_t key, const uint8_t *data, size_t len) {
    if (len > 255)
      return false;
    this->setup_();
    std::vector vec(data, data + len);
    this->data[key] = vec;
    return true;
  }

  bool load(uint32_t key, uint8_t *data, size_t len) {
    if (len > 255)
      return false;
    this->setup_();
    auto it = this->data.find(key);
    if (it == this->data.end())
      return false;
    const auto &vec = it->second;
    if (vec.size() != len)
      return false;
    memcpy(data, vec.data(), len);
    return true;
  }

 protected:
  void setup_();
  bool setup_complete_{};
  std::string filename_{};
  std::map<uint32_t, std::vector<uint8_t>> data{};
};

void setup_preferences();
extern HostPreferences *host_preferences;  // NOLINT(cppcoreguidelines-avoid-non-const-global-variables)

}  // namespace esphome::host

DECLARE_PREFERENCE_ALIASES(esphome::host::HostPreferences)

#endif  // USE_HOST