[VM] Improve token API

gideros · Jul 4, 2023 · 9a87cc9 · 9a87cc9
1 parent 985f21b
commit 9a87cc9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/VM/include/lua.h b/VM/include/lua.h
@@ -445,6 +445,7 @@ LUA_API lua_Callbacks* lua_callbacks(lua_State* L);
 */
 LUA_API size_t lua_newtoken(lua_State* L, const char *str);
 LUA_API void lua_pushtoken(lua_State* L, int token);
+LUA_API int lua_gettoken(lua_State* L, int idx, int token);
 LUA_API int lua_rawgettoken(lua_State* L, int idx, int token);
 LUA_API void lua_rawsettoken(lua_State* L, int idx, int token);
 

diff --git a/VM/src/lapi.cpp b/VM/src/lapi.cpp
@@ -1536,6 +1536,19 @@ void lua_pushtoken(lua_State* L, int token)
     return;
 }
 
+int lua_gettoken(lua_State* L, int idx, int token)
+{
+    api_check(L, token<L->global->ttoken.size());
+    luaC_threadbarrier(L);
+    StkId t = index2addr(L, idx);
+    api_check(L, ttistable(t));
+    TValue key;
+    setsvalue(L, &key, L->global->ttoken[token]);
+    luaV_gettable(L, t, &key, L->top);
+    api_incr_top(L);
+    return ttype(L->top - 1);
+}
+
 int lua_rawgettoken(lua_State* L, int idx, int token)
 {
 	api_check(L, token<L->global->ttoken.size());