#pragma once

#include <stdint.h>

#include "crc32.h"

namespace serial_load {

consteval uint32_t gen_cmd(const char * s)
{
  uint32_t x = 0
    | s[0] << 0
    | s[1] << 8
    | s[2] << 16
    | s[3] << 24;

  x ^= x << 13;
  x ^= x >> 17;
  x ^= x << 5;

  return x;
}

union command_reply {
  uint8_t u8[16];
  uint32_t u32[4];
  struct {
    uint32_t cmd;
    uint32_t arg[2];
    uint32_t crc;
  };
};

static inline uint32_t le_bswap(const uint32_t n)
{
  if constexpr (__BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__)
    return n;
  else
    return __builtin_bswap32(n);
}

constexpr union command_reply command_reply(uint32_t cmd, uint32_t arg0, uint32_t arg1)
{
  union command_reply command = {
    .cmd = le_bswap(cmd),
    .arg = { le_bswap(arg0), le_bswap(arg1) },
    .crc = 0,
  };
  command.crc = le_bswap(crc32(command.u8, 12));
  return command;
}

namespace command {
  constexpr uint32_t _write = gen_cmd("WRTE");
  constexpr uint32_t _read = gen_cmd("READ");
  constexpr uint32_t _jump = gen_cmd("JUMP");
  constexpr uint32_t _speed = gen_cmd("SPED");

  constexpr uint32_t _maple_list = gen_cmd("MPLS");

  static_assert(_write == 0x2cc46ed8);
  static_assert(_read == 0xf18d57c7);
  static_assert(_jump == 0xa6696f38);
  static_assert(_speed == 0x27a7a9f4);

  constexpr union command_reply write(uint32_t dest, uint32_t size)
  {
    return command_reply(_write, dest, size);
  }

  constexpr union command_reply read(uint32_t dest, uint32_t size)
  {
    return command_reply(_read, dest, size);
  }

  constexpr union command_reply jump(uint32_t dest)
  {
    return command_reply(_jump, dest, 0);
  }

  constexpr union command_reply speed(uint32_t speed)
  {
    return command_reply(_speed, speed, 0);
  }

  constexpr union command_reply maple_list(uint32_t function_type)
  {
    return command_reply(_maple_list, function_type, 0);
  }
}

namespace reply {
  constexpr uint32_t _write = gen_cmd("wrte");
  constexpr uint32_t _read = gen_cmd("read");
  constexpr uint32_t _jump = gen_cmd("jump");
  constexpr uint32_t _speed = gen_cmd("sped");
  constexpr uint32_t _write_crc = gen_cmd("crcw");
  constexpr uint32_t _read_crc = gen_cmd("crcr");

  constexpr uint32_t _maple_list = gen_cmd("mpls");
  constexpr uint32_t _maple_list_crc = gen_cmd("mlcs");

  static_assert(_write == 0x8c661aaa);
  static_assert(_read  == 0x512f23b5);
  static_assert(_jump  == 0x06cb1b4a);
  static_assert(_speed == 0x8705dd86);
  static_assert(_write_crc == 0x3cccc074);
  static_assert(_read_crc == 0x99cc92f4);

  constexpr union command_reply write(uint32_t dest, uint32_t size)
  {
    return command_reply(_write, dest, size);
  }

  constexpr union command_reply read(uint32_t dest, uint32_t size)
  {
    return command_reply(_read, dest, size);
  }

  constexpr union command_reply jump(uint32_t dest)
  {
    return command_reply(_jump, dest, 0);
  }

  constexpr union command_reply speed(uint32_t speed)
  {
    return command_reply(_speed, speed, 0);
  }

  constexpr union command_reply write_crc(uint32_t crc)
  {
    return command_reply(_write_crc, crc, 0);
  }

  constexpr union command_reply read_crc(uint32_t crc)
  {
    return command_reply(_read_crc, crc, 0);
  }
}

}