From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sun, 9 Oct 2016 19:53:52 +0000 (+0100)
Subject: [Minor] Add method to get classifier options from Lua
X-Git-Tag: 1.4.0~295
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3d0a1252f12c8d18b959e6cd3367733160d976a4;p=thirdparty%2Frspamd.git

[Minor] Add method to get classifier options from Lua
---

diff --git a/src/lua/lua_classifier.c b/src/lua/lua_classifier.c
index 488a9d4776..871433124b 100644
--- a/src/lua/lua_classifier.c
+++ b/src/lua/lua_classifier.c
@@ -19,9 +19,11 @@
 /* Classifier methods */
 LUA_FUNCTION_DEF (classifier, get_statfiles);
 LUA_FUNCTION_DEF (classifier, get_statfile_by_label);
+LUA_FUNCTION_DEF (classifier, get_param);
 
 static const struct luaL_reg classifierlib_m[] = {
 	LUA_INTERFACE_DEF (classifier, get_statfiles),
+	LUA_INTERFACE_DEF (classifier, get_param),
 	LUA_INTERFACE_DEF (classifier, get_statfile_by_label),
 	{"__tostring", rspamd_lua_class_tostring},
 	{NULL, NULL}
@@ -84,6 +86,29 @@ lua_classifier_get_statfiles (lua_State *L)
 	return 1;
 }
 
+static gint
+lua_classifier_get_param (lua_State *L)
+{
+	struct rspamd_classifier_config *ccf = lua_check_classifier (L);
+	const gchar *param;
+	const ucl_object_t *value;
+
+	param = luaL_checkstring (L, 2);
+
+	if (ccf != NULL && param != NULL) {
+		value = ucl_object_lookup (ccf->opts, param);
+
+		if (value != NULL) {
+			ucl_object_push_lua (L, value, true);
+			return 1;
+		}
+	}
+
+	lua_pushnil (L);
+
+	return 1;
+}
+
 /* Get statfile with specified label */
 static gint
 lua_classifier_get_statfile_by_label (lua_State *L)