diff options
| author | Andrew Lewis <nerf@judo.za.org> | 2017-09-07 21:49:48 +0200 |
|---|---|---|
| committer | Andrew Lewis <nerf@judo.za.org> | 2017-09-07 21:49:48 +0200 |
| commit | 52c0789dad8a6693eb3bd85acf15e1917b1d8e31 (patch) | |
| tree | 6334c029f54bd1f19e0e490da8930622224b0e76 /src/lua/lua_task.c | |
| parent | e2cc675c46b3c64652d06243ca8448c47702210c (diff) | |
| download | rspamd-52c0789dad8a6693eb3bd85acf15e1917b1d8e31.tar.gz rspamd-52c0789dad8a6693eb3bd85acf15e1917b1d8e31.zip | |
Revert "[Fix] Remove incorrect method `task:set_metric_subject`"
This reverts commit e809039a9e84cb6235a6447d1c9c8fc34ddcf8c6.
Diffstat (limited to 'src/lua/lua_task.c')
| -rw-r--r-- | src/lua/lua_task.c | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/src/lua/lua_task.c b/src/lua/lua_task.c index 2f5883563..9d139044f 100644 --- a/src/lua/lua_task.c +++ b/src/lua/lua_task.c @@ -586,6 +586,12 @@ LUA_FUNCTION_DEF (task, set_metric_score); * @param {string} action name to set */ LUA_FUNCTION_DEF (task, set_metric_action); +/*** + * @method task:set_metric_subject(subject) + * Set the subject in the default metric + * @param {string} subject subject to set + */ +LUA_FUNCTION_DEF (task, set_metric_subject); /*** * @method task:learn(is_spam[, classifier) @@ -880,6 +886,7 @@ static const struct luaL_reg tasklib_m[] = { LUA_INTERFACE_DEF (task, get_metric_action), LUA_INTERFACE_DEF (task, set_metric_score), LUA_INTERFACE_DEF (task, set_metric_action), + LUA_INTERFACE_DEF (task, set_metric_subject), LUA_INTERFACE_DEF (task, learn), LUA_INTERFACE_DEF (task, set_settings), LUA_INTERFACE_DEF (task, get_settings), @@ -4069,6 +4076,27 @@ lua_task_set_metric_action (lua_State *L) } static gint +lua_task_set_metric_subject (lua_State *L) +{ + struct rspamd_task *task = lua_check_task (L, 1); + const gchar *subject; + struct rspamd_metric *metric; + + metric = task->cfg->default_metric; + subject = luaL_checkstring (L, 2); + + if (task && metric && subject) { + metric->subject = rspamd_mempool_strdup (task->task_pool, subject); + lua_pushboolean (L, true); + } + else { + return luaL_error (L, "invalid arguments"); + } + + return 1; +} + +static gint lua_task_get_protocol_reply (lua_State *L) { struct rspamd_task *task = lua_check_task (L, 1); |