diff options
| author | Andrew Lewis <nerf@judo.za.org> | 2017-02-15 16:06:05 +0200 |
|---|---|---|
| committer | Andrew Lewis <nerf@judo.za.org> | 2017-02-15 16:43:18 +0200 |
| commit | 549961da0a10caf819290b8ff66cdaeafe8d7510 (patch) | |
| tree | b20ee39b9415a3dd15eb36ebbd4ee3bf87593f53 | |
| parent | d2a697af53239b088e8931e259421b07a77600ed (diff) | |
| download | rspamd-549961da0a10caf819290b8ff66cdaeafe8d7510.tar.gz rspamd-549961da0a10caf819290b8ff66cdaeafe8d7510.zip | |
[Feature] Support setting metric subject from Lua
| -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 9b15eb2c0..af85282f2 100644 --- a/src/lua/lua_task.c +++ b/src/lua/lua_task.c @@ -525,6 +525,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) @@ -764,6 +770,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), @@ -3407,6 +3414,27 @@ lua_task_set_metric_action (lua_State *L) return 1; } +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 = subject; + lua_pushboolean (L, true); + } + else { + return luaL_error (L, "invalid arguments"); + } + + return 1; +} + /* Image functions */ static gint lua_image_get_width (lua_State *L) |