#include "module.h"
#include "lauxlib.h"
#include "common.h"
#include "lmem.h"

#include "qrcodegen.h"

// qrcodegen.encodeText(text, [{minver=int, maxver=int, ecl=int, mask=int, boostecl=bool}])
// all params except text are optional
static int encodeText(lua_State *L)
{
  lua_settop(L, 2);
  const char *text = luaL_checkstring(L, 1);

  int min_ver = opt_checkint_range(L, "minver", qrcodegen_VERSION_MIN, 
                                   qrcodegen_VERSION_MIN, qrcodegen_VERSION_MAX);

  int max_ver = opt_checkint_range(L, "maxver", qrcodegen_VERSION_MAX,
                                   min_ver, qrcodegen_VERSION_MAX);

  int ecl = opt_checkint_range(L, "ecl", qrcodegen_Ecc_LOW,
                               qrcodegen_Ecc_LOW, qrcodegen_Ecc_HIGH);

  int mask = opt_checkint_range(L, "mask", qrcodegen_Mask_AUTO,
                                qrcodegen_Mask_AUTO, qrcodegen_Mask_7);

  bool boost = opt_checkbool(L, "boostecl", false);

  const size_t buf_len = qrcodegen_BUFFER_LEN_FOR_VERSION(max_ver);
  uint8_t *result = (uint8_t *)luaM_malloc(L, buf_len * 2); 
  uint8_t *tempbuf = result + buf_len;

  bool ok = qrcodegen_encodeText(text, tempbuf, result, ecl, min_ver, max_ver, mask, boost);
  if (ok) {
    lua_pushlstring(L, (const char *)result, buf_len);
  } else {
    lua_pushnil(L);
  }
  luaM_freemem(L, result, buf_len * 2);
  return 1;
}

static int getSize(lua_State *L)
{
  const uint8_t* data = (const uint8_t *)luaL_checkstring(L, 1);
  lua_pushinteger(L, qrcodegen_getSize(data));
  return 1;
}

static int getPixel(lua_State *L)
{
  const uint8_t* data = (const uint8_t *)luaL_checkstring(L, 1);
  int x = luaL_checkint(L, 2);
  int y = luaL_checkint(L, 3);
  lua_pushboolean(L, qrcodegen_getModule(data, x, y));
  return 1;
}

static const LUA_REG_TYPE qrcodegen_map[] = {
  { LSTRKEY("encodeText"), LFUNCVAL(encodeText) },
  { LSTRKEY("getSize"), LFUNCVAL(getSize) },
  { LSTRKEY("getPixel"), LFUNCVAL(getPixel) },
  { LSTRKEY("LOW"), LNUMVAL(qrcodegen_Ecc_LOW) },
  { LSTRKEY("MEDIUM"), LNUMVAL(qrcodegen_Ecc_MEDIUM) },
  { LSTRKEY("QUARTILE"), LNUMVAL(qrcodegen_Ecc_QUARTILE) },
  { LSTRKEY("HIGH"), LNUMVAL(qrcodegen_Ecc_HIGH) },
  { LSTRKEY("AUTO"), LNUMVAL(qrcodegen_Mask_AUTO) },
  { LNILKEY, LNILVAL }
};

NODEMCU_MODULE(QRCODEGEN, "qrcodegen", qrcodegen_map, NULL);