* Reorganze lua support in rspamd

1 files changed, 172 insertions, 0 deletions

diff --git a/src/lua/lua_message.c b/src/lua/lua_message.c
new file mode 100644
index 000000000..499945ccb
--- /dev/null
+++ b/src/lua/lua_message.c
@@ -0,0 +1,172 @@
+/*
+ * Copyright (c) 2009, Rambler media
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *     * Redistributions of source code must retain the above copyright
+ *       notice, this list of conditions and the following disclaimer.
+ *     * Redistributions in binary form must reproduce the above copyright
+ *       notice, this list of conditions and the following disclaimer in the
+ *       documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY Rambler media ''AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL Rambler BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+
+#include "lua_common.h"
+#include "../message.h"
+
+#define LUA_GMIME_BRIDGE_GET(class, name, mime_class)									\
+static int																				\
+lua_##class##_##name(lua_State *L)														\
+{																						\
+	GMime##mime_class *obj = lua_check_##class(L);										\
+	if (obj != NULL) {																	\
+		lua_pushstring (L, g_mime_##class##_##name(obj));								\
+	}																					\
+	else {																				\
+		lua_pushnil (L);																\
+	}																					\
+	return 1;																			\
+}																			
+
+#define LUA_GMIME_BRIDGE_SET(class, name, mime_class)									\
+static int																				\
+lua_##class##_##name(lua_State *L)														\
+{																						\
+	const char *str;																	\
+	GMime##mime_class *obj = lua_check_##class(L);										\
+	if (obj != NULL) {																	\
+		str = luaL_checkstring (L, 2);													\
+		g_mime_##class##_##name(obj, str);												\
+	}																					\
+	else {																				\
+		lua_pushnil (L);																\
+	}																					\
+	return 1;																			\
+}																			
+
+/*  Message methods */
+LUA_FUNCTION_DEF(message, get_subject);
+LUA_FUNCTION_DEF(message, set_subject);
+LUA_FUNCTION_DEF(message, get_message_id);
+LUA_FUNCTION_DEF(message, set_message_id);
+LUA_FUNCTION_DEF(message, get_sender);
+LUA_FUNCTION_DEF(message, set_sender);
+LUA_FUNCTION_DEF(message, get_reply_to);
+LUA_FUNCTION_DEF(message, set_reply_to);
+LUA_FUNCTION_DEF(message, get_header);
+LUA_FUNCTION_DEF(message, set_header);
+
+static const struct luaL_reg msglib_m[] = {
+	LUA_INTERFACE_DEF(message, get_subject),
+	LUA_INTERFACE_DEF(message, set_subject),
+	LUA_INTERFACE_DEF(message, get_message_id),
+	LUA_INTERFACE_DEF(message, set_message_id),
+	LUA_INTERFACE_DEF(message, get_sender),
+	LUA_INTERFACE_DEF(message, set_sender),
+	LUA_INTERFACE_DEF(message, get_reply_to),
+	LUA_INTERFACE_DEF(message, set_reply_to),
+	LUA_INTERFACE_DEF(message, get_header),
+	LUA_INTERFACE_DEF(message, set_header),
+	{NULL, NULL}
+};
+
+
+
+static GMimeMessage *
+lua_check_message (lua_State *L)
+{
+	void *ud = luaL_checkudata (L, 1, "Rspamd.message");
+	luaL_argcheck (L, ud != NULL, 1, "'message' expected");
+	return (GMimeMessage *)ud;
+}
+
+
+/*** Message interface	***/
+
+LUA_GMIME_BRIDGE_GET(message, get_subject, Message)
+LUA_GMIME_BRIDGE_SET(message, set_subject, Message)
+LUA_GMIME_BRIDGE_GET(message, get_message_id, Message)
+LUA_GMIME_BRIDGE_SET(message, set_message_id, Message)
+LUA_GMIME_BRIDGE_GET(message, get_sender, Message)
+LUA_GMIME_BRIDGE_SET(message, set_sender, Message)
+LUA_GMIME_BRIDGE_GET(message, get_reply_to, Message)
+LUA_GMIME_BRIDGE_SET(message, set_reply_to, Message)
+
+static int
+lua_message_get_header (lua_State *L)
+{
+	const char *headern;
+	GMimeMessage *obj = lua_check_message (L);
+	GList *res = NULL, *cur;
+
+	if (obj != NULL) {
+		headern = luaL_checkstring (L, 2);
+		if (headern) {
+			res = message_get_header (NULL, obj, headern);
+			if (res) {
+				cur = res;
+				while (cur) {
+					lua_pushstring (L, (const char *)cur->data);
+					g_free (cur->data);
+					cur = g_list_next (cur);
+				}
+				g_free (res);
+			}
+			else {
+				lua_pushnil (L);
+			}
+		}
+		else {
+			lua_pushnil (L);
+		}
+	}
+	else {
+		lua_pushnil (L);
+	}
+
+	return 1;
+}
+
+static int
+lua_message_set_header (lua_State *L)
+{
+	const char *headern, *headerv;
+	GMimeMessage *obj = lua_check_message (L);
+
+	if (obj != NULL) {
+		headern = luaL_checkstring (L, 2);
+		headerv = luaL_checkstring (L, 3);
+		if (headern && headerv) {
+			message_set_header (obj, headern, headerv);
+		}
+		else {
+			lua_pushnil (L);
+		}
+	}
+	else {
+		lua_pushnil (L);
+	}
+	
+	return 1;
+}
+
+int
+luaopen_message (lua_State *L)
+{
+	lua_newclass (L, "Rspamd.message", msglib_m);
+	luaL_openlib (L, "message", msglib_m, 0);
+    
+	return 1;
+}