From 3bff1316d1fc22f81b542d614c4bc90953bfa322 Mon Sep 17 00:00:00 2001 From: frosty Date: Sat, 19 Sep 2026 22:24:20 -0400 Subject: feat: add html.element lua helper --- src/Dom/Dom.c | 44 ++++++++++++++++++++++++++++++++++++++++++++ src/Dom/Dom.h | 1 + src/LuaEngine/LuaEngine.c | 1 + 3 files changed, 46 insertions(+) (limited to 'src') diff --git a/src/Dom/Dom.c b/src/Dom/Dom.c index ede7ddd..6dc6760 100644 --- a/src/Dom/Dom.c +++ b/src/Dom/Dom.c @@ -6,6 +6,50 @@ #include #include +static int html_element(lua_State *lua) { + const char *tag; + + luaL_checktype(lua, 1, LUA_TSTRING); + tag = lua_tostring(lua, 1); + if (tag[0] == '\0') { + return luaL_argerror(lua, 1, "tag cannot be empty"); + } + if (!lua_isnoneornil(lua, 2)) { + luaL_checktype(lua, 2, LUA_TTABLE); + } + if (!lua_isnoneornil(lua, 3)) { + luaL_checktype(lua, 3, LUA_TTABLE); + } + + lua_newtable(lua); + + lua_pushvalue(lua, 1); + lua_setfield(lua, -2, "tag"); + + if (lua_istable(lua, 2)) { + lua_pushvalue(lua, 2); + } else { + lua_newtable(lua); + } + lua_setfield(lua, -2, "attributes"); + + if (lua_istable(lua, 3)) { + lua_pushvalue(lua, 3); + } else { + lua_newtable(lua); + } + lua_setfield(lua, -2, "children"); + + return 1; +} + +void dom_register_html(lua_State *lua) { + lua_newtable(lua); + lua_pushcfunction(lua, html_element); + lua_setfield(lua, -2, "element"); + lua_setglobal(lua, "html"); +} + static int absolute_index(lua_State *lua, int index) { return index < 0 ? lua_gettop(lua) + index + 1 : index; } diff --git a/src/Dom/Dom.h b/src/Dom/Dom.h index 06bd1ff..bee6cd1 100644 --- a/src/Dom/Dom.h +++ b/src/Dom/Dom.h @@ -6,5 +6,6 @@ void dom_push_page(lua_State *lua, const xmlNode *root); xmlNodePtr dom_page_from_lua(lua_State *lua, int table_index); +void dom_register_html(lua_State *lua); #endif diff --git a/src/LuaEngine/LuaEngine.c b/src/LuaEngine/LuaEngine.c index 319f408..c34fd68 100644 --- a/src/LuaEngine/LuaEngine.c +++ b/src/LuaEngine/LuaEngine.c @@ -257,6 +257,7 @@ int process_html_document(htmlDocPtr document, const char *input_path) { luaL_openlibs(lua); register_yapssg(lua); + dom_register_html(lua); lua_logger_register(lua); directory_walker_register(lua, input_path); dom_push_page(lua, root); -- cgit v1.3