cs_fantasy_top.cpp

Go to the documentation of this file.

/* Chanstats core functions
 *
 * (C) 2003-2013 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 "../extra/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 CommandCSTop : public Command
{
 public:
        CommandCSTop(Module *creator) : Command (creator, "chanserv/top", 0, 2)
        {
                this->SetDesc(_("Displays the top 3 users of a channel"));
                this->SetSyntax(_("\037channel\037"));
        }

        void Execute(CommandSource &source, const std::vector<Anope::string> &params);
};

class CommandCSTop10 : public Command
{
 public:
        CommandCSTop10(Module *creator) : Command (creator, "chanserv/top10", 0, 2)
        {
                this->SetDesc(_("Displays the top 10 users of a channel"));
                this->SetSyntax(_("\037channel\037"));
        }

        void Execute(CommandSource &source, const std::vector<Anope::string> &params);
};

class CommandCSGTop : public Command
{
 public:
        CommandCSGTop(Module *creator) : Command (creator, "chanserv/gtop", 0, 1)
        {
                this->SetDesc(_("Displays the top 3 users of the network"));
                this->SetSyntax("");
        }

        void Execute(CommandSource &source, const std::vector<Anope::string> &params);
};

class CommandCSGTop10 : public Command
{
 public:
        CommandCSGTop10(Module *creator) : Command (creator, "chanserv/gtop10", 0, 1)
        {
                this->SetDesc(_("Displays the top 10 users of the network"));
                this->SetSyntax("");
        }

        void Execute(CommandSource &source, const std::vector<Anope::string> &params);
};


class CSTop;
static CSTop *me;
class CSTop : public Module
{
        CommandCSTop commandcstop;
        CommandCSGTop commandcsgtop;
        CommandCSTop10 commandcstop10;
        CommandCSGTop10 commandcsgtop10;
        ServiceReference<SQL::Provider> sql;
        MySQLInterface sqlinterface;
        Anope::string prefix;

 public:
        CSTop(const Anope::string &modname, const Anope::string &creator) : Module(modname, creator, CORE),
                commandcstop(this), commandcsgtop(this), commandcstop10(this), commandcsgtop10(this), sql("", ""),
                sqlinterface(this)
        {
                me = this;
                this->SetAuthor("Anope");

                Implementation i[] = { I_OnReload };
                ModuleManager::Attach(i, this, sizeof(i) / sizeof(Implementation));
                this->OnReload();
        }

        void OnReload() anope_override
        {
                ConfigReader config;
                prefix = config.ReadValue("chanstats", "prefix", "anope_", 0);
                Anope::string engine = config.ReadValue("chanstats", "engine", "", 0);
                this->sql = ServiceReference<SQL::Provider>("SQL::Provider", 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 = sql->RunQuery(query);
                if (!res.GetError().empty())
                        throw SQL::Exception(res.GetError());
                return res;
        }

        void DoTop(CommandSource &source, const std::vector<Anope::string> &params, bool is_global, int limit = 1)
        {
                if (!source.c || !source.c->ci)
                        return;

                Anope::string channel;
                if (is_global || params.empty())
                        channel = source.c->ci->name;
                else if (!params.empty())
                        channel = params[0];
                else
                {
                        source.Reply(_("%s not found."), params[0].c_str());
                        return;
                }

                try
                {
                        SQL::Query query;
                        query = "SELECT nick, letters, words, line, actions,"
                                "smileys_happy+smileys_sad+smileys_other as smileys "
                                "FROM `" + prefix + "chanstats` "
                                "WHERE `nick` != '' AND `chan` = @channel@ AND `type` = 'total' "
                                "ORDER BY `letters` DESC LIMIT @limit@;";
                        query.SetValue("limit", limit, false);

                        if (is_global)
                                query.SetValue("channel", "");
                        else
                                query.SetValue("channel", channel.c_str());

                        SQL::Result res = this->RunQuery(query);

                        if (res.Rows() > 0)
                        {
                                source.Reply(_("Top %i of %s"), limit, (is_global ? "Network" : channel.c_str()));
                                for (int i = 0; i < res.Rows(); ++i)
                                {
                                        source.Reply(_("%2lu \002%-16s\002 letters: %s, words: %s, lines: %s, smileys %s, actions: %s"),
                                                i+1, res.Get(i, "nick").c_str(), res.Get(i, "letters").c_str(),
                                                res.Get(i, "words").c_str(), res.Get(i, "line").c_str(), 
                                                res.Get(0, "smileys").c_str(), res.Get(0, "actions").c_str());
                                }
                        }
                        else
                                source.Reply(_("No stats for %s."), is_global ? "Network" : channel.c_str());
                }
                catch (const SQL::Exception &ex)
                {
                        Log(LOG_DEBUG) << ex.GetReason();
                }
        }
};

void CommandCSTop::Execute(CommandSource &source, const std::vector<Anope::string> &params)
{
        me->DoTop(source, params, false, 3);
}

void CommandCSTop10::Execute(CommandSource &source, const std::vector<Anope::string> &params)
{
        me->DoTop(source, params, false, 10);
}

void CommandCSGTop::Execute(CommandSource &source, const std::vector<Anope::string> &params)
{
        me->DoTop(source, params, true, 3);
}

void CommandCSGTop10::Execute(CommandSource &source, const std::vector<Anope::string> &params)
{
        me->DoTop(source, params, true, 10);
}


MODULE_INIT(CSTop)