#include "ws2812b_writer.h"

#include "esp_log.h"
#include "nvs_flash.h"
#include "state_version.h"

static const char* TAG = "state_store";
static const char* NAMESPACE = "ws";
static const char* KEY = "state";

typedef struct {
  uint32_t version;
  ws_state_t state;
} ws_state_blob_t;

esp_err_t ws_state_save(const ws_state_t* state)
{
  nvs_handle_t h;
  esp_err_t err = nvs_open(NAMESPACE, NVS_READWRITE, &h);
  if (err != ESP_OK) {
    ESP_LOGE(TAG, "nvs_open failed: %d", err);
    return err;
  }

  ws_state_blob_t blob = {.version = WS_STATE_SCHEMA_HASH, .state = *state};
  err = nvs_set_blob(h, KEY, &blob, sizeof(blob));
  if (err == ESP_OK) {
    err = nvs_commit(h);
  }
  nvs_close(h);

  if (err != ESP_OK) {
    ESP_LOGE(TAG, "nvs_set_blob/commit failed: %d", err);
  }
  return err;
}

esp_err_t ws_state_load(ws_state_t* out_state)
{
  nvs_handle_t h;
  esp_err_t err = nvs_open(NAMESPACE, NVS_READONLY, &h);
  if (err != ESP_OK) {
    return err;
  }

  ws_state_blob_t blob;
  size_t len = sizeof(blob);
  err = nvs_get_blob(h, KEY, &blob, &len);
  nvs_close(h);
  if (err != ESP_OK) {
    return err;
  }
  if (len != sizeof(blob) || blob.version != WS_STATE_SCHEMA_HASH) {
    return ESP_ERR_INVALID_VERSION;
  }

  *out_state = blob.state;
  return ESP_OK;
}