diff options
Diffstat (limited to 'src/lua/lua_task.c')
-rw-r--r-- | src/lua/lua_task.c | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c index 0dbf8dc0d..2b0e1cfbb 100644 --- a/src/lua/lua_task.c +++ b/src/lua/lua_task.c @@ -279,6 +279,14 @@ LUA_FUNCTION_DEF (task, get_resolver); LUA_FUNCTION_DEF (task, inc_dns_req); /*** + * @method task:has_recipients([type]) + * Return true if there are SMTP or MIME recipients for a task. + * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP recipients and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP recipients and `2` or `mime` means MIME recipients only + * @return {bool} `true` if there are recipients of the following type + */ +LUA_FUNCTION_DEF (task, has_recipients); + +/*** * @method task:get_recipients([type]) * Return SMTP or MIME recipients for a task. This function returns list of internet addresses each one is a table with the following structure: * @@ -290,6 +298,15 @@ LUA_FUNCTION_DEF (task, inc_dns_req); * @return {list of addresses} list of recipients or `nil` */ LUA_FUNCTION_DEF (task, get_recipients); + +/*** + * @method task:has_from([type]) + * Return true if there is SMTP or MIME sender for a task. + * @param {integer|string} type if specified has the following meaning: `0` or `any` means try SMTP recipients and fallback to MIME if failed, `1` or `smtp` means checking merely SMTP recipients and `2` or `mime` means MIME recipients only + * @return {bool} `true` if there is sender of the following type + */ +LUA_FUNCTION_DEF (task, has_from); + /*** * @method task:get_from([type]) * Return SMTP or MIME sender for a task. This function returns list of internet addresses each one is a table with the following structure: @@ -541,7 +558,9 @@ static const struct luaL_reg tasklib_m[] = { LUA_INTERFACE_DEF (task, get_queue_id), LUA_INTERFACE_DEF (task, get_resolver), LUA_INTERFACE_DEF (task, inc_dns_req), + LUA_INTERFACE_DEF (task, has_recipients), LUA_INTERFACE_DEF (task, get_recipients), + LUA_INTERFACE_DEF (task, has_from), LUA_INTERFACE_DEF (task, get_from), LUA_INTERFACE_DEF (task, get_user), LUA_INTERFACE_DEF (task, set_user), @@ -1330,6 +1349,91 @@ lua_task_get_recipients (lua_State *L) return 1; } +#define CHECK_ADDR(addr) do { \ + if (addr == NULL) { \ + ret = 0; \ + } \ + else { \ + ret = internet_address_list_length (addr) > 0 ? 1 : 0; \ + } \ +} while (0) + +static gint +lua_task_has_from (lua_State *L) +{ + struct rspamd_task *task = lua_check_task (L, 1); + gint what = 0; + gboolean ret = FALSE; + + if (task) { + if (lua_gettop (L) == 2) { + /* Get what value */ + what = lua_task_str_to_get_type (L, 2); + } + + switch (what) { + case 1: + /* Here we check merely envelope rcpt */ + CHECK_ADDR (task->from_envelope); + break; + case 2: + /* Here we check merely mime rcpt */ + CHECK_ADDR (task->from_mime); + break; + case 0: + default: + CHECK_ADDR (task->from_envelope); + + if (!ret) { + CHECK_ADDR (task->from_mime); + } + break; + } + } + + lua_pushboolean (L, ret); + + return 1; +} + +static gint +lua_task_has_recipients (lua_State *L) +{ + struct rspamd_task *task = lua_check_task (L, 1); + gint what = 0; + gboolean ret = FALSE; + + if (task) { + if (lua_gettop (L) == 2) { + /* Get what value */ + what = lua_task_str_to_get_type (L, 2); + } + + switch (what) { + case 1: + /* Here we check merely envelope rcpt */ + CHECK_ADDR (task->rcpt_envelope); + break; + case 2: + /* Here we check merely mime rcpt */ + CHECK_ADDR (task->rcpt_mime); + break; + case 0: + default: + CHECK_ADDR (task->rcpt_envelope); + + if (!ret) { + CHECK_ADDR (task->rcpt_mime); + } + break; + } + } + + lua_pushboolean (L, ret); + + return 1; +} + static gint lua_task_get_from (lua_State *L) { |