diff options
Diffstat (limited to 'src/lua/lua_url.c')
| -rw-r--r-- | src/lua/lua_url.c | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/src/lua/lua_url.c b/src/lua/lua_url.c index 8826bb41c..15966bfa3 100644 --- a/src/lua/lua_url.c +++ b/src/lua/lua_url.c @@ -56,6 +56,7 @@ LUA_FUNCTION_DEF (url, is_html_displayed); LUA_FUNCTION_DEF (url, is_subject); LUA_FUNCTION_DEF (url, get_phished); LUA_FUNCTION_DEF (url, get_tag); +LUA_FUNCTION_DEF (url, get_count); LUA_FUNCTION_DEF (url, get_tags); LUA_FUNCTION_DEF (url, add_tag); LUA_FUNCTION_DEF (url, create); @@ -82,6 +83,7 @@ static const struct luaL_reg urllib_m[] = { LUA_INTERFACE_DEF (url, get_tag), LUA_INTERFACE_DEF (url, get_tags), LUA_INTERFACE_DEF (url, add_tag), + LUA_INTERFACE_DEF (url, get_count), {"get_redirected", lua_url_get_phished}, {"__tostring", lua_url_get_text}, {NULL, NULL} @@ -529,6 +531,26 @@ lua_url_get_tld (lua_State *L) } /*** + * @method url:get_count() + * Return number of occurrencies for this particular URL + * @return {number} number of occurrencies + */ +static gint +lua_url_get_count (lua_State *L) +{ + struct rspamd_lua_url *url = lua_check_url (L, 1); + + if (url != NULL && url->url != NULL) { + lua_pushnumber (L, url->url->count); + } + else { + lua_pushnil (L); + } + + return 1; +} + +/*** * @method url:to_table() * Return url as a table with the following fields: * |