From df8d85d80aae4af79293f52095ab487a745bef5a Mon Sep 17 00:00:00 2001
From: DukePyrolator <DukePyrolator@anope.org>
Date: Sun, 19 Jan 2014 08:11:19 +0100
Subject: [PATCH] m_chanstats: fixed the config options NSDefChanstats and
 CSDefChanstats

---
 data/chanstats.example.conf         |  6 +++---
 modules/extra/stats/m_chanstats.cpp | 16 +++++++++++++++-
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/data/chanstats.example.conf b/data/chanstats.example.conf
index 1fb408474..ed7a07c8c 100644
--- a/data/chanstats.example.conf
+++ b/data/chanstats.example.conf
@@ -26,10 +26,10 @@ module
 
 	/*
 	 * Enable Chanstats for newly registered nicks / channels.
-	 * Set it to 0 to disable it.
+	 * Set it to No to disable it.
 	 */
-	NSDefChanstats = 1
-	CSDefChanstats = 1
+	NSDefChanstats = Yes
+	CSDefChanstats = Yes
 }
 command { service = "ChanServ"; name = "SET CHANSTATS"; command = "chanserv/set/chanstats"; }
 command { service = "NickServ"; name = "SET CHANSTATS"; command = "nickserv/set/chanstats"; }
diff --git a/modules/extra/stats/m_chanstats.cpp b/modules/extra/stats/m_chanstats.cpp
index c3be9d640..1ca670368 100644
--- a/modules/extra/stats/m_chanstats.cpp
+++ b/modules/extra/stats/m_chanstats.cpp
@@ -168,6 +168,7 @@ class MChanstats : public Module
 	SQL::Query query;
 	Anope::string SmileysHappy, SmileysSad, SmileysOther, prefix;
 	std::vector<Anope::string> TableList, ProcedureList, EventList;
+	bool NSDefChanstats, CSDefChanstats;
 
 	void RunQuery(const SQL::Query &q)
 	{
@@ -490,7 +491,8 @@ class MChanstats : public Module
 		SmileysHappy = block->Get<const Anope::string>("SmileysHappy");
 		SmileysSad = block->Get<const Anope::string>("SmileysSad");
 		SmileysOther = block->Get<const Anope::string>("SmileysOther");
-
+		NSDefChanstats = block->Get<bool>("NSDefChanstats", "no");
+		CSDefChanstats = block->Get<bool>("CSDefChanstats", "no");
 		Anope::string engine = block->Get<const Anope::string>("engine");
 		this->sql = ServiceReference<SQL::Provider>("SQL::Provider", engine);
 		if (sql)
@@ -629,6 +631,18 @@ class MChanstats : public Module
 		query.SetValue("channel", ci->name);
 		this->RunQuery(query);
 	}
+
+	void OnChanRegistered(ChannelInfo *ci)
+	{
+		if (CSDefChanstats)
+			ci->Extend<bool>("CS_STATS");
+	}
+
+	void OnNickRegister(User *user, NickAlias *na)
+	{
+		if (NSDefChanstats)
+			na->nc->Extend<bool>("NS_STATS");
+	}
 };
 
 MODULE_INIT(MChanstats)