SparrowBot/app/punishermodule.cpp

84 lines
2.3 KiB
C++

#include "punishermodule.h"
#include "message.h"
#include "user.h"
#include <QStringList>
PunisherModule::PunisherModule()
{
}
bool PunisherModule::messageHandler(Message msg)
{
for(const QString &w : forbidden_words)
{
if(msg.args.contains(w, Qt::CaseInsensitive))
{
answer = QString("KICK #%1 %2\r\n").arg(getChan()).arg(msg.nick);
return true;
}
}
if(msg.args.startsWith("!forbidden add ") && getUsers()->getFromNick(msg.nick)->isOp())
{
QStringList str = msg.args.split(' ');
int i=2;
answer = QString("");
for(; i<str.size(); ++i)
{
forbidden_words.push_back(str[i]);
answer += say(QString("added %1 to the list of forbidden words").arg(str[i]));
}
if(i==2)
answer = say(QString("missing argument"));
return true;
}
if(msg.args.startsWith("!forbidden remove ") && getUsers()->getFromNick(msg.nick)->isOp())
{
QStringList str = msg.args.split(' ');
int i=2;
answer = QString("");
for(; i<str.size(); ++i)
{
for(auto it = forbidden_words.begin(); it != forbidden_words.end(); it++)
{
if(str[i].compare(*it, Qt::CaseInsensitive) == 0)
{
forbidden_words.erase(it);
answer += say(QString("removed %1 to the list of forbidden words").arg(str[i]));
}
else
answer += say(QString("%1 was not part of the list of forbidden words").arg(str[i]));
}
}
if(i==2)
answer = say(QString("missing argument"));
return true;
}
if(msg.args.compare("!forbidden reset") == 0 && getUsers()->getFromNick(msg.nick)->isOp())
{
forbidden_words.clear();
answer = say(QString("removed all forbidden words"));
return true;
}
if(msg.args.compare("!forbidden list") == 0)
{
QString str = QString("Forbidden words are : ");
for(const QString &w : forbidden_words)
str.append(w).append(" ");
answer = say(str);
return true;
}
return false;
}
QString PunisherModule::getName()
{
return QString("punisher");
}