#include "bans.h"

//    This file is part of MRCI.

//    MRCI is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.

//    MRCI is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.

//    You should have received a copy of the GNU General Public License
//    along with MRCI under the LICENSE.md file. If not, see
//    <http://www.gnu.org/licenses/>.

ListBans::ListBans(QObject *parent) : TableViewer(parent)
{
    setParams(TABLE_IPBANS, true);
    addTableColumn(TABLE_IPBANS, COLUMN_TIME);
    addTableColumn(TABLE_IPBANS, COLUMN_IPADDR);
}

BanIP::BanIP(QObject *parent)     : CmdObject(parent) {}
UnBanIP::UnBanIP(QObject *parent) : CmdObject(parent) {}

QString ListBans::cmdName() {return "ls_bans";}
QString BanIP::cmdName()    {return "add_ban";}
QString UnBanIP::cmdName()  {return "rm_ban";}

void ListBans::onDel()
{
    async(ASYNC_UPDATE_BANS, PRIV_IPC);
}

void BanIP::procIn(const QByteArray &binIn, quint8 dType)
{
    if (dType == TEXT)
    {
        QStringList args = parseArgs(binIn, 2);
        QString     ip   = getParam("-ip", args);

        if (ip.isEmpty())
        {
            errTxt("err: The ip address argument (-ip) was not found or is empty.\n");
        }
        else if (!QHostAddress().setAddress(ip))
        {
            errTxt("err: '" + ip + "' is not a valid ip address.\n");
        }
        else
        {
            QHostAddress addr(ip);

            Query db(this);

            db.setType(Query::PUSH, TABLE_IPBANS);
            db.addColumn(COLUMN_IPADDR, addr.toString());
            db.exec();

            async(ASYNC_UPDATE_BANS, PRIV_IPC);
        }
    }
}

void UnBanIP::procIn(const QByteArray &binIn, quint8 dType)
{
    if (dType == TEXT)
    {
        QStringList args = parseArgs(binIn, 2);
        QString     ip   = getParam("-ip", args);

        if (ip.isEmpty())
        {
            errTxt("err: The ip address argument (-ip) was not found or is empty.\n");
        }
        else if (!QHostAddress().setAddress(ip))
        {
            errTxt("err: '" + ip + "' is not a valid ip address.\n");
        }
        else
        {
            QHostAddress addr(ip);

            Query db;

            db.setType(Query::DEL, TABLE_IPBANS);
            db.addCondition(COLUMN_IPADDR, addr.toString());
            db.exec();

            async(ASYNC_UPDATE_BANS, PRIV_IPC);
        }
    }
}