Move modules/stats under extras because it depends on m_mysql, update its config a bit to look similar to all of the other config files

1 files changed, 382 insertions, 0 deletions

diff --git a/modules/extra/stats/irc2sql/tables.cpp b/modules/extra/stats/irc2sql/tables.cpp
new file mode 100644
index 000000000..d80f37535
--- /dev/null
+++ b/modules/extra/stats/irc2sql/tables.cpp
@@ -0,0 +1,382 @@
+#include "irc2sql.h"
+
+void IRC2SQL::CheckTables()
+{
+	Anope::string geoquery("");
+
+	if (firstrun)
+	{
+		/*
+		 * reset some tables to make sure they are really empty
+		 */
+		this->sql->RunQuery("TRUNCATE TABLE " + prefix + "user");
+		this->sql->RunQuery("TRUNCATE TABLE " + prefix + "chan");
+		this->sql->RunQuery("TRUNCATE TABLE " + prefix + "ison");
+		this->sql->RunQuery("UPDATE `" + prefix + "server` SET currentusers=0, online='N'");
+	}
+
+	this->GetTables();
+
+	if (UseGeoIP && GeoIPDB.equals_ci("country") && !this->HasTable(prefix + "geoip_country"))
+	{
+		query = "CREATE TABLE `" + prefix + "geoip_country` ("
+			"`start` INT UNSIGNED NOT NULL,"
+			"`end` INT UNSIGNED NOT NULL,"
+			"`countrycode` varchar(2),"
+			"`countryname` varchar(50),"
+			"PRIMARY KEY `end` (`end`),"
+			"KEY `start` (`start`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (UseGeoIP && GeoIPDB.equals_ci("city") && !this->HasTable(prefix + "geoip_city_blocks"))
+	{
+		query = "CREATE TABLE `" + prefix + "geoip_city_blocks` ("
+			"`start` INT UNSIGNED NOT NULL,"
+			"`end` INT UNSIGNED NOT NULL,"
+			"`locId` INT UNSIGNED NOT NULL,"
+			"PRIMARY KEY `end` (`end`),"
+			"KEY `start` (`start`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+
+	}
+	if (UseGeoIP && GeoIPDB.equals_ci("city") && !this->HasTable(prefix + "geoip_city_location"))
+	{
+		query = "CREATE TABLE `" + prefix + "geoip_city_location` ("
+			"`locId` INT UNSIGNED NOT NULL,"
+			"`country` CHAR(2) NOT NULL,"
+			"`region` CHAR(2) NOT NULL,"
+			"`city` VARCHAR(50),"
+			"`latitude` FLOAT,"
+			"`longitude` FLOAT,"
+			"`areaCode` INT,"
+			"PRIMARY KEY (`locId`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (UseGeoIP && GeoIPDB.equals_ci("city") && !this->HasTable(prefix + "geoip_city_region"))
+	{	query = "CREATE TABLE `" + prefix + "geoip_city_region` ("
+			"`country` CHAR(2) NOT NULL,"
+			"`region` CHAR(2) NOT NULL,"
+			"`regionname` VARCHAR(100) NOT NULL,"
+			"PRIMARY KEY (`country`,`region`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (!this->HasTable(prefix + "server"))
+	{
+		query = "CREATE TABLE `" + prefix + "server` ("
+			"`id` int(11) UNSIGNED NOT NULL AUTO_INCREMENT,"
+			"`name` varchar(64) NOT NULL,"
+			"`hops` tinyint(3) NOT NULL,"
+			"`comment` varchar(255) NOT NULL,"
+			"`link_time` datetime DEFAULT NULL,"
+			"`split_time` datetime DEFAULT NULL,"
+			"`version` varchar(127) NOT NULL,"
+			"`currentusers` int(15) NOT NULL,"
+			"`online` enum('Y','N') NOT NULL DEFAULT 'Y',"
+			"`ulined` enum('Y','N') NOT NULL DEFAULT 'N',"
+			"PRIMARY KEY (`id`),"
+			"UNIQUE KEY `name` (`name`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (!this->HasTable(prefix + "chan"))
+	{
+		query = "CREATE TABLE `" + prefix + "chan` ("
+			"`chanid` int(11) UNSIGNED NOT NULL AUTO_INCREMENT,"
+			"`channel` varchar(255) NOT NULL,"
+			"`currentusers` int(15) NOT NULL DEFAULT 0,"
+			"`topic` varchar(255) DEFAULT NULL,"
+			"`topicauthor` varchar(255) DEFAULT NULL,"
+			"`topictime` datetime DEFAULT NULL,"
+			"`modes` varchar(512) DEFAULT NULL,"
+			"PRIMARY KEY (`chanid`),"
+			"UNIQUE KEY `channel`(`channel`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (!this->HasTable(prefix + "user"))
+	{
+		query = "CREATE TABLE `" + prefix + "user` ("
+			"`nickid` int(11) UNSIGNED NOT NULL AUTO_INCREMENT,"
+			"`nick` varchar(255) NOT NULL DEFAULT '',"
+			"`host` varchar(255) NOT NULL DEFAULT '',"
+			"`vhost` varchar(255) NOT NULL DEFAULT '',"
+			"`chost` varchar(255) NOT NULL DEFAULT '',"
+			"`realname` varchar(255) NOT NULL DEFAULT '',"
+			"`ip` varchar(255) NOT NULL DEFAULT '',"
+			"`ident` varchar(32) NOT NULL DEFAULT '',"
+			"`vident` varchar(32) NOT NULL DEFAULT '',"
+			"`modes` varchar(255) NOT NULL DEFAULT '',"
+			"`account` varchar(255) NOT NULL DEFAULT '',"
+			"`secure` enum('Y','N') NOT NULL DEFAULT 'N',"
+			"`fingerprint` varchar(128) NOT NULL DEFAULT '',"
+			"`signon` datetime DEFAULT NULL,"
+			"`server` varchar(255) NOT NULL DEFAULT '',"
+			"`servid` int(11) UNSIGNED NOT NULL DEFAULT '0',"
+			"`uuid` varchar(32) NOT NULL DEFAULT '',"
+			"`oper` enum('Y','N') NOT NULL DEFAULT 'N',"
+			"`away` enum('Y','N') NOT NULL DEFAULT 'N',"
+			"`awaymsg` varchar(255) NOT NULL DEFAULT '',"
+			"`version` varchar(255) NOT NULL DEFAULT '',"
+			"`geocode` varchar(16) NOT NULL DEFAULT '',"
+			"`geocountry` varchar(64) NOT NULL DEFAULT '',"
+			"`georegion` varchar(100) NOT NULL DEFAULT '',"
+			"`geocity` varchar(128) NOT NULL DEFAULT '',"
+			"`locId` INT UNSIGNED,"
+			"PRIMARY KEY (`nickid`),"
+			"UNIQUE KEY `nick` (`nick`),"
+			"KEY `servid` (`servid`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (!this->HasTable(prefix + "ison"))
+	{
+		query = "CREATE TABLE `" + prefix + "ison` ("
+			"`nickid` int(11) unsigned NOT NULL default '0',"
+			"`chanid` int(11) unsigned NOT NULL default '0',"
+			"`modes` varchar(255) NOT NULL default '',"
+			"PRIMARY KEY  (`nickid`,`chanid`),"
+			"KEY `modes` (`modes`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (!this->HasTable(prefix + "maxusers"))
+	{
+		query = "CREATE TABLE `" + prefix + "maxusers` ("
+			"`name` VARCHAR(255) NOT NULL,"
+			"`maxusers` INT(15) NOT NULL,"
+			"`maxtime` DATETIME NOT NULL,"
+			"`lastused` DATETIME NOT NULL,"
+			"UNIQUE KEY `name` (`name`)"
+			") ENGINE=MyISAM DEFAULT CHARSET=utf8;";
+		this->RunQuery(query);
+	}
+	if (this->HasProcedure(prefix + "UserConnect"))
+		this->RunQuery(SQL::Query("DROP PROCEDURE " + prefix + "UserConnect"));
+
+	if (UseGeoIP)
+	{
+		if (GeoIPDB.equals_ci("country"))
+			geoquery = "UPDATE `" + prefix + "user` AS u "
+					"JOIN ( SELECT `countrycode`, `countryname` "
+						"FROM `" + prefix + "geoip_country` "
+						"WHERE INET_ATON(ip_) <= `end` "
+						"AND `start` <= INET_ATON(ip_) "
+						"ORDER BY `end` ASC LIMIT 1 ) as c "
+					"SET u.geocode = c.countrycode, u.geocountry = c.countryname "
+					"WHERE u.nick = nick_; ";
+		else if (GeoIPDB.equals_ci("city"))
+			geoquery = "UPDATE `" + prefix + "user` as u "
+					"JOIN ( SELECT * FROM `" + prefix + "geoip_city_location` "
+						"WHERE `locID` = ( SELECT `locID` "
+								"FROM `" + prefix + "geoip_city_blocks` "
+								"WHERE INET_ATON(ip_) <= `end` "
+								"AND `start` <= INET_ATON(ip_) "
+								"ORDER BY `end` ASC LIMIT 1 ) "
+						") as l "
+					"SET u.geocode = l.country, "
+					    "u.geocity = l.city, "
+					    "u.locID = l.locID, "
+					    "u.georegion = ( SELECT `regionname` "
+								"FROM `" + prefix + "geoip_city_region` "
+								"WHERE `country` = l.country "
+								"AND `region` = l.region )"
+					"WHERE u.nick = nick_;";
+	}
+	query = "CREATE PROCEDURE `" + prefix + "UserConnect`"
+		"(nick_ varchar(255), host_ varchar(255), vhost_ varchar(255), "
+		"chost_ varchar(255), realname_ varchar(255), ip_ varchar(255), "
+		"ident_ varchar(255), vident_ varchar(255), account_ varchar(255), "
+		"secure_ enum('Y','N'), fingerprint_ varchar(255), signon_ int(15), "
+		"server_ varchar(255), uuid_ varchar(32), modes_ varchar(255), "
+		"oper_ enum('Y','N')) "
+		"BEGIN "
+			"DECLARE cur int(15);"
+			"DECLARE max int(15);"
+			"INSERT INTO `" + prefix + "user` "
+				"(nick, host, vhost, chost, realname, ip, ident, vident, account, "
+				"secure, fingerprint, signon, server, uuid, modes, oper) "
+			"VALUES (nick_, host_, vhost_, chost_, realname_, ip_, ident_, vident_, "
+				"account_, secure_, fingerprint_, FROM_UNIXTIME(signon_), server_, "
+				"uuid_, modes_, oper_) "
+			"ON DUPLICATE KEY UPDATE host=VALUES(host), vhost=VALUES(vhost), "
+				"chost=VALUES(chost), realname=VALUES(realname), ip=VALUES(ip), "
+				"ident=VALUES(ident), vident=VALUES(vident), account=VALUES(account), "
+				"secure=VALUES(secure), fingerprint=VALUES(fingerprint), signon=VALUES(signon), "
+				"server=VALUES(server), uuid=VALUES(uuid), modes=VALUES(modes), "
+				"oper=VALUES(oper);"
+			"UPDATE `" + prefix + "user` as `u`, `" + prefix + "server` as `s`"
+				"SET u.servid = s.id, "
+					"s.currentusers = s.currentusers + 1 "
+				"WHERE s.name = server_ AND u.nick = nick_;"
+			"SELECT `currentusers` INTO cur FROM `" + prefix + "server` WHERE name=server_;"
+			"SELECT `maxusers` INTO max FROM `" + prefix + "maxusers` WHERE name=server_;"
+			"IF found_rows() AND cur <= max THEN "
+				"UPDATE `" + prefix + "maxusers` SET lastused=now() WHERE name=server_;"
+			"ELSE "
+				"INSERT INTO `" + prefix + "maxusers` (name, maxusers, maxtime, lastused) "
+					"VALUES ( server_, cur, now(), now() ) "
+					"ON DUPLICATE KEY UPDATE "
+						"name=VALUES(name), maxusers=VALUES(maxusers),"
+						"maxtime=VALUES(maxtime), lastused=VALUES(lastused);"
+			"END IF;"
+			+ geoquery +
+		"END";
+	this->RunQuery(query);
+
+	if (this->HasProcedure(prefix + "ServerQuit"))
+		this->RunQuery(SQL::Query("DROP PROCEDURE " + prefix + "ServerQuit"));
+	query = "CREATE PROCEDURE " + prefix + "ServerQuit(sname_ varchar(255)) "
+		"BEGIN "
+			/* 1.
+			 * loop through all channels and decrease the user count
+			 * by the number of users that are on this channel AND
+			 * on the splitting server
+			 *
+			 * we dont have to care about channels that get empty, there will be
+			 * an extra OnChannelDelete event triggered from anope.
+			 */
+			"DECLARE no_more_rows BOOLEAN DEFAULT FALSE;"
+			"DECLARE channel_ varchar(255);"
+			"DECLARE ucount_ int;"
+			"DECLARE channel_cursor CURSOR FOR "
+				"SELECT c.channel "
+				"FROM `" + prefix + "chan` as c, `" + prefix + "ison` as i, "
+					"`" + prefix + "user` as u, `" + prefix + "server` as s "
+				"WHERE c.chanid = i.chanid "
+				  "AND i.nickid = u.nickid "
+				  "AND u.servid = s.id "
+				  "AND s.name = sname_;"
+			"DECLARE CONTINUE HANDLER FOR NOT FOUND "
+				"SET no_more_rows = TRUE;"
+			"OPEN channel_cursor;"
+			"the_loop: LOOP "
+				"FETCH channel_cursor INTO channel_;"
+				"IF no_more_rows THEN "
+					"CLOSE channel_cursor;"
+					"LEAVE the_loop;"
+				"END IF;"
+				"SELECT COUNT(*) INTO ucount_ "
+				"FROM `" + prefix + "ison` AS i, `" + prefix + "chan` AS c,"
+					"`" + prefix + "user` AS u, `" + prefix + "server` AS s "
+				"WHERE i.nickid = u.nickid "
+				  "AND u.servid = s.id "
+				  "AND i.chanid = c.chanid "
+				  "AND c.channel = channel_ "
+				  "AND s.name = sname_; "
+				"UPDATE `" + prefix + "chan` "
+				"SET currentusers = currentusers - ucount_ "
+				"WHERE channel = channel_;"
+			"END LOOP;"
+
+			/* 2.
+			 * remove all users on the splitting server from the ison table
+			 */
+			"DELETE i FROM `" + prefix + "ison` AS i "
+				"INNER JOIN `" + prefix + "server` AS s "
+				"INNER JOIN `" + prefix + "user` AS u "
+				"WHERE i.nickid = u.nickid "
+				  "AND u.servid = s.id "
+				  "AND s.name = sname_;"
+
+			/* 3.
+			 * remove all users on the splitting server from the user table
+			 */
+			"DELETE u FROM `" + prefix + "user` AS u "
+				"INNER JOIN `" + prefix + "server` AS s "
+				"WHERE s.id = u.servid "
+				  "AND s.name = sname_;"
+
+			/* 4.
+			 * on the splitting server, set usercount = 0, split_time = now(), online = 'N'
+			 */
+			"UPDATE `" + prefix + "server` SET currentusers = 0, split_time = now(), online = 'N' "
+				"WHERE name = sname_;"
+		"END;"; // end of the procedure
+	this->RunQuery(query);
+
+
+	if (this->HasProcedure(prefix + "UserQuit"))
+		this->RunQuery(SQL::Query("DROP PROCEDURE " + prefix + "UserQuit"));
+	query = "CREATE PROCEDURE `" + prefix + "UserQuit`"
+		"(nick_ varchar(255)) "
+		"BEGIN "
+			/* decrease usercount on the server where the user was on */
+			"UPDATE `" + prefix + "user` AS `u`, `" + prefix + "server` AS `s` "
+				"SET s.currentusers = s.currentusers - 1 "
+				"WHERE u.nick=nick_ AND u.servid = s.id; "
+			/* decrease the usercount on all channels where the user was on */
+			"UPDATE `" + prefix + "user` AS u, `" + prefix + "ison` AS i, "
+				"`" + prefix + "chan` AS c "
+				"SET c.currentusers = c.currentusers - 1 "
+				"WHERE u.nick=nick_ AND u.nickid = i.nickid "
+				"AND i.chanid = c.chanid; "
+			/* remove from all channels where the user was on */
+			"DELETE i FROM `" + prefix + "ison` AS i "
+				"INNER JOIN `" + prefix + "user` as u "
+			"WHERE u.nick = nick_ "
+				"AND i.nickid = u.nickid;"
+			/* remove the user from the user table */
+			"DELETE FROM `" + prefix + "user` WHERE nick = nick_; "
+		"END";
+	this->RunQuery(query);
+
+	if (this->HasProcedure(prefix + "ShutDown"))
+		this->RunQuery(SQL::Query("DROP PROCEDURE " + prefix + "ShutDown"));
+	query = "CREATE PROCEDURE `" + prefix + "ShutDown`()"
+		"BEGIN "
+			"UPDATE `" +  prefix + "server` "
+				"SET currentusers=0, online='N', split_time=now();"
+			"TRUNCATE TABLE `" + prefix + "user`;"
+			"TRUNCATE TABLE `" + prefix + "chan`;"
+			"TRUNCATE TABLE `" + prefix + "ison`;"
+		"END";
+	this->RunQuery(query);
+
+	if (this->HasProcedure(prefix + "JoinUser"))
+		this->RunQuery(SQL::Query("DROP PROCEDURE " + prefix + "JoinUser"));
+	query = "CREATE PROCEDURE `"+ prefix + "JoinUser`"
+		"(nick_ varchar(255), channel_ varchar(255), modes_ varchar(255)) "
+		"BEGIN "
+			"DECLARE cur int(15);"
+			"DECLARE max int(15);"
+			"INSERT INTO `" + prefix + "ison` (nickid, chanid, modes) "
+				"SELECT u.nickid, c.chanid, modes_ "
+				"FROM " + prefix + "user AS u, " + prefix + "chan AS c "
+				"WHERE u.nick=nick_ AND c.channel=channel_;"
+			"UPDATE `" + prefix + "chan` SET currentusers=currentusers+1 "
+				"WHERE channel=channel_;"
+			"SELECT `currentusers` INTO cur FROM `" + prefix + "chan` WHERE channel=channel_;"
+			"SELECT `maxusers` INTO max FROM `" + prefix + "maxusers` WHERE name=channel_;"
+			"IF found_rows() AND cur <= max THEN "
+				"UPDATE `" + prefix + "maxusers` SET lastused=now() WHERE name=channel_;"
+			"ELSE "
+				"INSERT INTO `" + prefix + "maxusers` (name, maxusers, maxtime, lastused) "
+					"VALUES ( channel_, cur, now(), now() ) "
+					"ON DUPLICATE KEY UPDATE "
+						"name=VALUES(name), maxusers=VALUES(maxusers),"
+						"maxtime=VALUES(maxtime), lastused=VALUES(lastused);"
+			"END IF;"
+		"END";
+	this->RunQuery(query);
+
+	if (this->HasProcedure(prefix + "PartUser"))
+		this->RunQuery(SQL::Query("DROP PROCEDURE " + prefix + "PartUser"));
+	query = "CREATE PROCEDURE `" + prefix + "PartUser`"
+		"(nick_ varchar(255), channel_ varchar(255)) "
+		"BEGIN "
+			"DELETE i FROM `" + prefix + "ison` AS i "
+				"INNER JOIN `" + prefix + "user` AS u "
+				"INNER JOIN `" + prefix + "chan` AS c "
+				"WHERE i.nickid = u.nickid "
+					"AND u.nick = nick_ "
+					"AND i.chanid = c.chanid "
+					"AND c.channel = channel_;"
+			"UPDATE `" + prefix + "chan` SET currentusers=currentusers-1 "
+				"WHERE channel=channel_;"
+		"END";
+	this->RunQuery(query);
+}