#ifndef _RIVE_TESTS_RIV_BYTES_HPP_
#define _RIVE_TESTS_RIV_BYTES_HPP_

#include "rive/core/vector_binary_writer.hpp"
#include "rive/file.hpp"
#include "rive/generated/artboard_base.hpp"
#include "rive/generated/backboard_base.hpp"
#include "rive/generated/component_base.hpp"
#include "rive/generated/layout_component_base.hpp"

#include <map>
#include <string>
#include <vector>

// A minimal .riv assembled by hand, so an object or a property can be placed
// where no exporter would put it -- which is how the import paths that only
// a malformed or forward-dated file reaches get exercised.
class RivBytes
{
public:
  RivBytes() : m_writer(&m_objects) {}

  void object(uint16_t typeKey) { m_writer.writeVarUint((uint64_t)typeKey); }
  void end() { m_writer.writeVarUint((uint64_t)0); }

  // ToC field ids: 0 uint (bools ride as uints), 1 string, 2 double.
  void propUint(uint16_t key, uint64_t value)
  {
    m_toc[key] = 0;
    m_writer.writeVarUint((uint64_t)key);
    m_writer.writeVarUint(value);
  }
  void propBool(uint16_t key, bool value)
  {
    m_toc[key] = 0;
    m_writer.writeVarUint((uint64_t)key);
    m_writer.write((uint8_t)(value ? 1 : 0));
  }
  void propString(uint16_t key, const std::string& value)
  {
    m_toc[key] = 1;
    m_writer.writeVarUint((uint64_t)key);
    m_writer.write(value);
  }
  void propFloat(uint16_t key, float value)
  {
    m_toc[key] = 2;
    m_writer.writeVarUint((uint64_t)key);
    m_writer.write(value);
  }
  // A key the file's ToC does not describe.
  void propUintOutsideToc(uint16_t key, uint64_t value)
  {
    m_writer.writeVarUint((uint64_t)key);
    m_writer.writeVarUint(value);
  }

  std::vector<uint8_t> bytes() const
  {
    std::vector<uint8_t> out;
    rive::VectorBinaryWriter writer(&out);
    writer.write((const uint8_t*)"RIVE", 4);
    writer.writeVarUint((uint64_t)rive::File::majorVersion);
    writer.writeVarUint((uint64_t)rive::File::minorVersion);
    writer.writeVarUint((uint64_t)0); // file id
    for (const auto& entry : m_toc)
    {
      writer.writeVarUint((uint64_t)entry.first);
    }
    writer.writeVarUint((uint64_t)0);
    // The reader refreshes its packed int every 8 bits: one uint32 per
    // four keys (RuntimeHeader::read).
    uint32_t packed = 0;
    int bit = 0;
    for (const auto& entry : m_toc)
    {
      packed |= entry.second << bit;
      bit += 2;
      if (bit == 8)
      {
        writer.write(packed);
        packed = 0;
        bit = 0;
      }
    }
    if (bit != 0)
    {
      writer.write(packed);
    }
    out.insert(out.end(), m_objects.begin(), m_objects.end());
    return out;
  }

private:
  std::vector<uint8_t> m_objects;
  rive::VectorBinaryWriter m_writer;
  std::map<uint16_t, uint32_t> m_toc;
};

// One 100x100 artboard named "A", with no Backboard of its own -- for streams
// that need to place something between the two.
inline void writeArtboardObject(RivBytes& riv)
{
  riv.object(rive::ArtboardBase::typeKey);
  riv.propString(rive::ComponentBase::namePropertyKey, "A");
  riv.propFloat(rive::LayoutComponentBase::widthPropertyKey, 100.0f);
  riv.propFloat(rive::LayoutComponentBase::heightPropertyKey, 100.0f);
  riv.end();
}

// The smallest loadable file: a Backboard and one 100x100 artboard named "A".
inline void writeArtboard(RivBytes& riv)
{
  riv.object(rive::BackboardBase::typeKey);
  riv.end();
  writeArtboardObject(riv);
}

#endif