Shuffle the modules about a bit.

1 files changed, 124 insertions, 0 deletions

diff --git a/modules/log_sql.cpp b/modules/log_sql.cpp
new file mode 100644
index 000000000..498e09957
--- /dev/null
+++ b/modules/log_sql.cpp
@@ -0,0 +1,124 @@
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ *   Copyright (C) 2022-2023 Sadie Powell <sadie@witchery.services>
+ *
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+
+#include "inspircd.h"
+#include "modules/sql.h"
+
+namespace
+{
+	Module* thismod;
+}
+
+class SQLQuery final
+	: public SQL::Query
+{
+public:
+	SQLQuery(Module* mod) ATTR_NOT_NULL(2)
+		: SQL::Query(mod)
+	{
+	}
+
+	void OnResult(SQL::Result& res) override
+	{
+		// Nothing to do here.
+	}
+
+	void OnError(const SQL::Error& error) override
+	{
+		ServerInstance->SNO.WriteGlobalSno('a', "Unable to write to SQL log (query error: {}).", error.ToString());
+	}
+};
+
+class SQLMethod final
+	: public Log::Method
+{
+private:
+	std::string query;
+	dynamic_reference<SQL::Provider> sql;
+	time_t lastwarning = 0;
+
+public:
+	SQLMethod(const dynamic_reference<SQL::Provider>& s, const std::string& q)
+		: query(q)
+		, sql(s)
+	{
+	}
+
+	void OnLog(time_t time, Log::Level level, const std::string& type, const std::string& message) override
+	{
+		if (!sql)
+		{
+			// Only give a warning every 5 minutes to avoid log spam.
+			if (ServerInstance->Time() - lastwarning > 300)
+			{
+				lastwarning = ServerInstance->Time();
+				ServerInstance->SNO.WriteGlobalSno('a', "Unable to write to SQL log (database {} not available).", sql->GetId());
+			}
+			return;
+		}
+
+		SQL::ParamMap params = {
+			{ "level",    ConvToStr(static_cast<uint8_t>(level)) },
+			{ "levelstr", Log::LevelToString(level)              },
+			{ "message",  message                                },
+			{ "time",     ConvToStr(time)                        },
+			{ "type",     type                                   },
+		};
+		sql->Submit(new SQLQuery(thismod), query, params);
+	}
+};
+
+class SQLEngine final
+	: public Log::Engine
+{
+public:
+	SQLEngine(Module* Creator) ATTR_NOT_NULL(2)
+		: Log::Engine(Creator, "sql")
+	{
+	}
+
+	Log::MethodPtr Create(const std::shared_ptr<ConfigTag>& tag) override
+	{
+		dynamic_reference<SQL::Provider> sql(creator, "SQL");
+		const std::string dbid = tag->getString("dbid");
+		if (!dbid.empty())
+			sql.SetProvider("SQL/" + dbid);
+
+		const std::string query = tag->getString("query", "INSERT INTO ircd_log (time, type, message) VALUES (FROM_UNIXTIME($time), '$type', '$message');", 1);
+		return std::make_shared<SQLMethod>(sql, query);
+	}
+};
+
+class ModuleLogSQL final
+	: public Module
+{
+private:
+	SQLEngine engine;
+
+public:
+	ModuleLogSQL()
+		: Module(VF_VENDOR, "Provides the ability to log to a SQL database.")
+		, engine(this)
+	{
+		thismod = this;
+	}
+
+};
+
+MODULE_INIT(ModuleLogSQL)