Move logger code to a separate module.

1 files changed, 146 insertions, 0 deletions

diff --git a/src/lua/lua_logger.c b/src/lua/lua_logger.c
new file mode 100644
index 000000000..0047190bf
--- /dev/null
+++ b/src/lua/lua_logger.c
@@ -0,0 +1,146 @@
+/*
+ * Copyright (c) 2015, Vsevolod Stakhov
+ *
+ * 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 AUTHOR ''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 AUTHOR 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"
+
+
+/* Logger methods */
+LUA_FUNCTION_DEF (logger, err);
+LUA_FUNCTION_DEF (logger, warn);
+LUA_FUNCTION_DEF (logger, info);
+LUA_FUNCTION_DEF (logger, debug);
+
+static const struct luaL_reg loggerlib_f[] = {
+	LUA_INTERFACE_DEF (logger, err),
+	LUA_INTERFACE_DEF (logger, warn),
+	LUA_INTERFACE_DEF (logger, info),
+	LUA_INTERFACE_DEF (logger, debug),
+	{"__tostring", rspamd_lua_class_tostring},
+	{NULL, NULL}
+};
+
+static void
+lua_common_log (GLogLevelFlags level, lua_State *L, const gchar *msg)
+{
+	lua_Debug d;
+	gchar func_buf[128], *p;
+
+	if (lua_getstack (L, 1, &d) == 1) {
+		(void)lua_getinfo (L, "Sl", &d);
+		if ((p = strrchr (d.short_src, '/')) == NULL) {
+			p = d.short_src;
+		}
+		else {
+			p++;
+		}
+		rspamd_snprintf (func_buf, sizeof (func_buf), "%s:%d", p,
+			d.currentline);
+		if (level == G_LOG_LEVEL_DEBUG) {
+			rspamd_conditional_debug (rspamd_main->logger,
+				NULL,
+				func_buf,
+				"%s",
+				msg);
+		}
+		else {
+			rspamd_common_log_function (rspamd_main->logger,
+				level,
+				func_buf,
+				"%s",
+				msg);
+		}
+	}
+	else {
+		if (level == G_LOG_LEVEL_DEBUG) {
+			rspamd_conditional_debug (rspamd_main->logger,
+				NULL,
+				__FUNCTION__,
+				"%s",
+				msg);
+		}
+		else {
+			rspamd_common_log_function (rspamd_main->logger,
+				level,
+				__FUNCTION__,
+				"%s",
+				msg);
+		}
+	}
+}
+
+/*** Logger interface ***/
+static gint
+lua_logger_err (lua_State * L)
+{
+	const gchar *msg;
+	msg = luaL_checkstring (L, 1);
+	lua_common_log (G_LOG_LEVEL_CRITICAL, L, msg);
+	return 1;
+}
+
+static gint
+lua_logger_warn (lua_State * L)
+{
+	const gchar *msg;
+	msg = luaL_checkstring (L, 1);
+	lua_common_log (G_LOG_LEVEL_WARNING, L, msg);
+	return 1;
+}
+
+static gint
+lua_logger_info (lua_State * L)
+{
+	const gchar *msg;
+	msg = luaL_checkstring (L, 1);
+	lua_common_log (G_LOG_LEVEL_INFO, L, msg);
+	return 1;
+}
+
+static gint
+lua_logger_debug (lua_State * L)
+{
+	const gchar *msg;
+	msg = luaL_checkstring (L, 1);
+	lua_common_log (G_LOG_LEVEL_DEBUG, L, msg);
+	return 1;
+}
+
+
+/*** Init functions ***/
+
+static gint
+lua_load_logger (lua_State *L)
+{
+	lua_newtable (L);
+	luaL_register (L, NULL, loggerlib_f);
+
+	return 1;
+}
+
+void
+luaopen_logger (lua_State * L)
+{
+	rspamd_lua_add_preload (L, "rspamd_logger", lua_load_logger);
+}