Parcourir la source

[Minor] Add Lua methods to get urls order

tags/3.6
Vsevolod Stakhov il y a 9 mois
Parent
révision
5331d429f1
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 46 ajouts et 0 suppressions
  1. 46
    0
      src/lua/lua_url.c

+ 46
- 0
src/lua/lua_url.c Voir le fichier

@@ -72,6 +72,8 @@ LUA_FUNCTION_DEF(url, init);
LUA_FUNCTION_DEF(url, all);
LUA_FUNCTION_DEF(url, lt);
LUA_FUNCTION_DEF(url, eq);
LUA_FUNCTION_DEF(url, get_order);
LUA_FUNCTION_DEF(url, get_part_order);

static const struct luaL_reg urllib_m[] = {
LUA_INTERFACE_DEF(url, get_length),
@@ -97,6 +99,8 @@ static const struct luaL_reg urllib_m[] = {
LUA_INTERFACE_DEF(url, get_count),
LUA_INTERFACE_DEF(url, get_flags),
LUA_INTERFACE_DEF(url, get_flags_num),
LUA_INTERFACE_DEF(url, get_order),
LUA_INTERFACE_DEF(url, get_part_order),
{"get_redirected", lua_url_get_phished},
LUA_INTERFACE_DEF(url, set_redirected),
{"__tostring", lua_url_tostring},
@@ -942,6 +946,48 @@ lua_url_get_flags_num(lua_State *L)
return 1;
}

static gint
lua_url_get_order(lua_State *L)
{
LUA_TRACE_POINT;
struct rspamd_lua_url *url = lua_check_url(L, 1);

if (url) {
if (url->url->order != (uint16_t) -1) {
lua_pushinteger(L, url->url->order);
}
else {
lua_pushnil(L);
}
}
else {
return luaL_error(L, "invalid arguments");
}

return 1;
}

static gint
lua_url_get_part_order(lua_State *L)
{
LUA_TRACE_POINT;
struct rspamd_lua_url *url = lua_check_url(L, 1);

if (url) {
if (url->url->part_order != (uint16_t) -1) {
lua_pushinteger(L, url->url->part_order);
}
else {
lua_pushnil(L);
}
}
else {
return luaL_error(L, "invalid arguments");
}

return 1;
}

void lua_tree_url_callback(gpointer key, gpointer value, gpointer ud)
{
struct rspamd_lua_url *lua_url;

Chargement…
Annuler
Enregistrer