diff options
author | Adam <Adam@anope.org> | 2014-01-03 19:54:14 -0500 |
---|---|---|
committer | Adam <Adam@anope.org> | 2014-01-03 19:54:14 -0500 |
commit | e1ce6174cee90b71dc461be39d2bf20965ff69e0 (patch) | |
tree | 6f8788ab40a533f3ae791f30cfe4a1f94fadf193 /modules/extra/stats/cs_fantasy_stats.cpp | |
parent | 2781b6946daa82b88b08cdd6ab3c6478f7c70aa1 (diff) |
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
Diffstat (limited to 'modules/extra/stats/cs_fantasy_stats.cpp')
-rw-r--r-- | modules/extra/stats/cs_fantasy_stats.cpp | 179 |
1 files changed, 179 insertions, 0 deletions
diff --git a/modules/extra/stats/cs_fantasy_stats.cpp b/modules/extra/stats/cs_fantasy_stats.cpp new file mode 100644 index 000000000..43f7eb04a --- /dev/null +++ b/modules/extra/stats/cs_fantasy_stats.cpp @@ -0,0 +1,179 @@ +/* Chanstats core functions + * + * (C) 2003-2014 Anope Team + * Contact us at team@anope.org + * + * Please read COPYING and README for further details. + * + * Based on the original code of Epona by Lara. + * Based on the original code of Services by Andy Church. + */ + +#include "module.h" +#include "modules/sql.h" + +class MySQLInterface : public SQL::Interface +{ + public: + MySQLInterface(Module *o) : SQL::Interface(o) { } + + void OnResult(const SQL::Result &r) anope_override + { + } + + void OnError(const SQL::Result &r) anope_override + { + if (!r.GetQuery().query.empty()) + Log(LOG_DEBUG) << "Chanstats: Error executing query " << r.finished_query << ": " << r.GetError(); + else + Log(LOG_DEBUG) << "Chanstats: Error executing query: " << r.GetError(); + } +}; + + +class CommandCSStats : public Command +{ + public: + CommandCSStats(Module *creator) : Command (creator, "chanserv/stats", 0, 2) + { + this->SetDesc(_("Displays your Channel Stats")); + this->SetSyntax(_("[\037channel\037] [\037nick\037]")); + } + + void Execute(CommandSource &source, const std::vector<Anope::string> ¶ms); +}; + +class CommandCSGStats : public Command +{ + public: + CommandCSGStats(Module *creator) : Command (creator, "chanserv/gstats", 0, 2) + { + this->SetDesc(_("Displays your Global Stats")); + this->SetSyntax(_("\037nick\037")); + } + + void Execute(CommandSource &source, const std::vector<Anope::string> ¶ms); +}; + + +class CSStats; +static CSStats *me; +class CSStats : public Module +{ + CommandCSStats commandcsstats; + CommandCSGStats commandcsgstats; + ServiceReference<SQL::Provider> sql; + MySQLInterface sqlinterface; + Anope::string prefix; + public: + CSStats(const Anope::string &modname, const Anope::string &creator) : Module(modname, creator, VENDOR), + commandcsstats(this), commandcsgstats(this), sql("", ""), sqlinterface(this) + { + me = this; + + } + + void OnReload(Configuration::Conf *conf) anope_override + { + prefix = conf->GetModule("m_chanstats")->Get<const Anope::string>("prefix"); + if (prefix.empty()) + prefix = "anope_"; + this->sql = ServiceReference<SQL::Provider>("SQL::Provider", conf->GetModule("m_chanstats")->Get<const Anope::string>("engine")); + } + + SQL::Result RunQuery(const SQL::Query &query) + { + if (!this->sql) + throw SQL::Exception("Unable to locate SQL reference, is m_mysql loaded and configured correctly?"); + + SQL::Result res = this->sql->RunQuery(query); + if (!res.GetError().empty()) + throw SQL::Exception(res.GetError()); + return res; + } + + void DoStats(CommandSource &source, const bool is_global, const std::vector<Anope::string> ¶ms) + { + Anope::string display, channel; + + /* + * possible parameters are: + * stats [channel] [nick] + * stats [channel] + * stats [nick] + * stats + */ + + switch (params.size()) + { + case 2: + channel = params[0]; + display = params[1]; + break; + case 1: + if (params[0][0] == '#') + channel = params[0]; + else + { + if (NickAlias *na = NickAlias::Find(params[0])) + display = na->nc->display; + else + { + source.Reply(_("%s not found."), params[0].c_str()); + return; + } + } + break; + } + + if (display.empty()) + display = source.nc->display; + + try + { + SQL::Query query; + query = "SELECT letters, words, line, smileys_happy+smileys_sad+smileys_other as smileys," + "actions FROM `" + prefix + "chanstats` " + "WHERE `nick` = @nick@ AND `chan` = @channel@ AND `type` = 'total';"; + if (is_global || channel.empty()) + query.SetValue("channel", ""); + else + query.SetValue("channel", channel); + query.SetValue("nick", display); + SQL::Result res = this->RunQuery(query); + + if (res.Rows() > 0) + { + if (is_global) + source.Reply(_("Network stats for %s:"), display.c_str()); + else + source.Reply(_("Channel stats for %s on %s:"), display.c_str(), channel.c_str()); + + source.Reply(_("letters: %s, words: %s, lines: %s, smileys %s, actions: %s"), + res.Get(0, "letters").c_str(), res.Get(0, "words").c_str(), + res.Get(0, "line").c_str(), res.Get(0, "smileys").c_str(), + res.Get(0, "actions").c_str()); + } + else + source.Reply(_("No stats for %s."), display.c_str()); + } + catch (const SQL::Exception &ex) + { + Log(LOG_DEBUG) << ex.GetReason(); + } + + } +}; + +void CommandCSStats::Execute(CommandSource &source, const std::vector<Anope::string> ¶ms) +{ + me->DoStats(source, false, params); +} + +void CommandCSGStats::Execute(CommandSource &source, const std::vector<Anope::string> ¶ms) +{ + me->DoStats(source, true, params); +} + +MODULE_INIT(CSStats) + |