#ifdef WITH_RIVE_SCRIPTING
#include "lua.h"
#include "lualib.h"
#include "rive/lua/rive_lua_libs.hpp"
#include "rive/assets/blob_asset.hpp"

#include <string.h>

using namespace rive;

static void blob_direct_size(void* udata, void* result)
{
  auto* self = (ScriptedBlob*)udata;
  BlobAsset* asset = self->asset ? self->asset->as<BlobAsset>() : nullptr;
  if (!asset)
  {
    lua_userdatadirectfield_setnil(result);
    return;
  }
  lua_userdatadirectfield_setnumber(result, (double)asset->bytes().size());
}

static int blob_index(lua_State* L)
{
  auto blob = lua_torive<ScriptedBlob>(L, 1);

  int atom;
  const char* name = lua_tostringatom(L, 2, &atom);
  if (!name)
  {
    luaL_typeerrorL(L, 2, lua_typename(L, LUA_TSTRING));
  }

  BlobAsset* blobAsset = blob->asset ? blob->asset->as<BlobAsset>() : nullptr;
  if (blobAsset == nullptr)
  {
    lua_pushnil(L);
    return 1;
  }

  switch (atom)
  {
    case (int)LuaAtoms::size:
      lua_pushnumber(L, static_cast<double>(blobAsset->bytes().size()));
      return 1;
    case (int)LuaAtoms::name:
      lua_pushstring(L, blobAsset->name().c_str());
      return 1;
    case (int)LuaAtoms::data:
    {
      auto bytes = blobAsset->bytes();
      if (bytes.size() > 0)
      {
        void* bufferData = lua_newbuffer(L, bytes.size());
        memcpy(bufferData, bytes.data(), bytes.size());
        return 1;
      }
      lua_pushnil(L);
      return 1;
    }
  }

  luaL_error(L, "'%s' is not a valid index of %s", name, ScriptedBlob::luaName);
  return 0;
}

#ifdef WITH_RIVE_TOOLS
int lua_push_blob(
    lua_State* L,
    const char* name,
    const uint8_t* data,
    size_t size)
{
  auto scriptedBlob = lua_newrive<ScriptedBlob>(L);
  if (data != nullptr && size > 0)
  {
    auto blobAsset = make_rcp<BlobAsset>();
    if (name != nullptr)
    {
      blobAsset->name(name);
    }
    SimpleArray<uint8_t> bytes(data, size);
    blobAsset->decode(bytes, nullptr);
    scriptedBlob->asset = blobAsset;
  }
  return 1;
}
#endif

int luaopen_rive_blob(lua_State* L)
{
  lua_register_rive<ScriptedBlob>(L);

  lua_pushcfunction(L, blob_index, nullptr);
  lua_setfield(L, -2, "__index");
  lua_setreadonly(L, -1, true);
  lua_pop(L, 1);

  lua_registeruserdatadirectfieldget(
      L,
      ScriptedBlob::luaTag,
      "size",
      blob_direct_size);

  return 0;
}

#endif