// raw flash storage interface
#include "storage.hpp"

#include <cstring>

#include "hardware/flash.h"
#include "hardware/sync.h"
#include "pico/binary_info.h"
#include "pico/multicore.h"

#include "binary_info.hpp"
#include "config.h"

extern bool core1_started;

bool storage_init() {
  return true;
}

bool is_storage_available() {
  return true;
}

void get_storage_size(uint16_t &block_size, uint32_t &num_blocks) {
  block_size = FLASH_SECTOR_SIZE;
  num_blocks = FLASH_STORAGE_SIZE / FLASH_SECTOR_SIZE;

  bi_decl(bi_block_device(
    BINARY_INFO_TAG_32BLIT,
    "32Blit",
    XIP_BASE + FLASH_STORAGE_OFFSET,
    FLASH_STORAGE_SIZE,
    nullptr,
    BINARY_INFO_BLOCK_DEV_FLAG_READ | BINARY_INFO_BLOCK_DEV_FLAG_WRITE | BINARY_INFO_BLOCK_DEV_FLAG_PT_NONE
  ))
}

int32_t storage_read(uint32_t sector, uint32_t offset, void *buffer, uint32_t size_bytes) {
  memcpy(buffer, (uint8_t *)XIP_NOCACHE_NOALLOC_BASE + FLASH_STORAGE_OFFSET + sector * FLASH_SECTOR_SIZE + offset, size_bytes);

  return size_bytes;
}

int32_t storage_write(uint32_t sector, uint32_t offset, const uint8_t *buffer, uint32_t size_bytes) {
  auto status = save_and_disable_interrupts();

  if(core1_started)
    multicore_lockout_start_blocking(); // pause core1

  if(offset == 0)
    flash_range_erase(FLASH_STORAGE_OFFSET + sector * FLASH_SECTOR_SIZE, FLASH_SECTOR_SIZE);

  flash_range_program(FLASH_STORAGE_OFFSET + sector * FLASH_SECTOR_SIZE + offset, buffer, size_bytes);

  if(core1_started)
    multicore_lockout_end_blocking(); // resume core1

  restore_interrupts(status);

  return size_bytes;
}