awtk/lua/custom.c

#include "base/enums.h"
#include "ui_loader/ui_builder_default.h"

typedef struct _userdata_info_t {
  const char* info;
  void* data;
} userdata_info_t;

static lua_State* s_current_L = NULL;
extern void luaL_openlib(lua_State* L, const char* libname, const luaL_Reg* l, int nup);

static int lftk_newuserdata(lua_State* L, void* data, const char* info, const char* metatable) {
  char str[48];
  userdata_info_t* udata = NULL;
  return_value_if_fail(data != NULL, 0);

  udata = (userdata_info_t*)lua_newuserdata(L, sizeof(userdata_info_t));
  return_value_if_fail(data != NULL, 0);

  udata->data = data;
  udata->info = info;

  if (strstr(info, "/widget_t") != NULL && strcmp(metatable, "lftk.widget_t") == 0) {
    widget_t* widget = (widget_t*)data;
    const key_type_value_t* kv = widget_type_find_by_value(widget->type);
    if (kv != NULL) {
      snprintf(str, sizeof(str), "lftk.%s_t", kv->name);
      metatable = str;
    }
  }

  if (metatable != NULL) {
    int ret = luaL_getmetatable(L, metatable);
    if (ret == 0 && strstr(info, "/widget_t") != NULL) {
      lua_pop(L, 1);
      ret = luaL_getmetatable(L, "lftk.widget_t");
    }
    lua_setmetatable(L, -2);
  }

  return 1;
}

static const luaL_Reg* find_member(const luaL_Reg* funcs, const char* name) {
  const luaL_Reg* iter = funcs;

  while (iter->name) {
    if (*iter->name == *name && strcmp(iter->name, name) == 0) {
      return iter;
    }
    iter++;
  }

  return NULL;
}

static void* lftk_checkudata(lua_State* L, int idx, const char* name) {
  userdata_info_t* udata = (userdata_info_t*)lua_touserdata(L, idx);
  if (udata) {
    // assert(strstr(udata->info, name) != NULL);
    return udata->data;
  } else {
    return NULL;
  }
}

static ret_t call_on_event(void* ctx, event_t* e) {
  lua_State* L = (lua_State*)s_current_L;
  int func_id = (char*)ctx - (char*)NULL;

  lua_settop(L, 0);
  lua_rawgeti(L, LUA_REGISTRYINDEX, func_id);
  lftk_newuserdata(L, e, "event_t", NULL);

  lua_pcall(L, 1, 1, 0);

  return 1;
}

static int wrap_widget_on(lua_State* L) {
  ret_t ret = 0;
  widget_t* widget = (widget_t*)lftk_checkudata(L, 1, "widget_t");
  event_type_t type = (event_type_t)luaL_checkinteger(L, 2);

  if (lua_isfunction(L, 3)) {
    int func_id = 0;
    lua_pushvalue(L, 3);
    func_id = luaL_ref(L, LUA_REGISTRYINDEX);
    ret = (ret_t)widget_on(widget, type, call_on_event, (char*)NULL + func_id);
    lua_pushnumber(L, (lua_Number)ret);

    return 1;
  } else {
    return 0;
  }
}

static int wrap_widget_off(lua_State* L) {
  ret_t ret = 0;
  widget_t* widget = (widget_t*)lftk_checkudata(L, 1, "widget_t");
  uint32_t id = (uint32_t)luaL_checkinteger(L, 2);
  emitter_item_t* item = emitter_find(widget->emitter, id);

  if (item) {
    uint32_t func_id = (char*)(item->ctx) - (char*)NULL;
    luaL_unref(L, LUA_REGISTRYINDEX, func_id);
    ret = (ret_t)widget_off(widget, id);
  }

  lua_pushnumber(L, (lua_Number)(ret));

  return 1;
}

static int to_wstr(lua_State* L) {
  const char* str = (const char*)luaL_checkstring(L, 1);
  uint32_t size = (strlen(str) + 1) * sizeof(wchar_t);
  wchar_t* p = (wchar_t*)lua_newuserdata(L, size);

  utf8_to_utf16(str, p, size);
  lua_pushlightuserdata(L, p);

  return 1;
}

static int to_str(lua_State* L) {
  const wchar_t* str = (const wchar_t*)lua_touserdata(L, 1);
  uint32_t size = (wcslen(str) + 1) * 3;
  char* p = (char*)lua_newuserdata(L, size);

  utf8_from_utf16(str, p, size);
  lua_pushstring(L, p);

  return 1;
}

static ret_t call_on_timer(const timer_info_t* timer) {
  ret_t ret = RET_REMOVE;
  lua_State* L = (lua_State*)s_current_L;
  int func_id = (char*)(timer->ctx) - (char*)NULL;

  lua_settop(L, 0);
  lua_rawgeti(L, LUA_REGISTRYINDEX, func_id);

  lua_pcall(L, 0, 1, 0);

  ret = (ret_t)lua_tonumber(L, -1);

  return ret;
}

static int wrap_timer_add(lua_State* L) {
  uint32_t id = 0;
  if (lua_isfunction(L, 1)) {
    int func_id = 0;
    uint32_t duration_ms = (uint32_t)luaL_checkinteger(L, 2);
    lua_pushvalue(L, 1);
    func_id = luaL_ref(L, LUA_REGISTRYINDEX);

    id = timer_add(call_on_timer, (char*)NULL + func_id, duration_ms);
    lua_pushnumber(L, (lua_Number)id);

    return 1;
  } else {
    return 0;
  }
}

static int wrap_timer_remove(lua_State* L) {
  ret_t ret = 0;
  uint32_t id = (uint32_t)luaL_checkinteger(L, 1);
  const timer_info_t* timer = timer_find(id);

  if (timer) {
    uint32_t func_id = (char*)(timer->ctx) - (char*)NULL;
    luaL_unref(L, LUA_REGISTRYINDEX, func_id);
    ret = (ret_t)timer_remove(id);
  }

  lua_pushnumber(L, (lua_Number)(ret));

  return 1;
}