diff options
Diffstat (limited to 'src/lua')
| -rw-r--r-- | src/lua/lua_url.c | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/src/lua/lua_url.c b/src/lua/lua_url.c index 0a301e96d..58c6a83be 100644 --- a/src/lua/lua_url.c +++ b/src/lua/lua_url.c @@ -63,6 +63,7 @@ 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, get_visible); LUA_FUNCTION_DEF (url, create); LUA_FUNCTION_DEF (url, init); LUA_FUNCTION_DEF (url, all); @@ -89,6 +90,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_visible), LUA_INTERFACE_DEF (url, get_count), LUA_INTERFACE_DEF (url, get_flags), {"get_redirected", lua_url_get_phished}, @@ -650,6 +652,27 @@ lua_url_get_count (lua_State *L) return 1; } + /*** +* @method url:get_visible() +* Get visible part of the url with html tags stripped +* @return {string} url string +*/ +static gint +lua_url_get_visible (lua_State *L) +{ + LUA_TRACE_POINT; + struct rspamd_lua_url *url = lua_check_url (L, 1); + + if (url != NULL) { + lua_pushlstring (L, url->url->visible_part, url->url->visible_partlen); + } + else { + lua_pushnil (L); + } + +return 1; +} + /*** * @method url:to_table() * Return url as a table with the following fields: @@ -878,6 +901,7 @@ lua_url_all (lua_State *L) * - `schemaless`: URL has no schema * - `unnormalised`: URL has some unicode unnormalities * - `zw_spaces`: URL has some zero width spaces + * - `url_displayed`: URL has some other url-like string in visible part * @return {table} URL flags */ #define PUSH_FLAG(fl, name) do { \ @@ -918,6 +942,7 @@ lua_url_get_flags (lua_State *L) PUSH_FLAG (RSPAMD_URL_FLAG_SCHEMALESS, "schemaless"); PUSH_FLAG (RSPAMD_URL_FLAG_UNNORMALISED, "unnormalised"); PUSH_FLAG (RSPAMD_URL_FLAG_ZW_SPACES, "zw_spaces"); + PUSH_FLAG (RSPAMD_URL_FLAG_DISPLAY_URL, "url_displayed"); } else { return luaL_error (L, "invalid arguments"); |