/***************************************************************************
 *   This file is part of netctl-plasmoid                                  *
 *                                                                         *
 *   netctl-plasmoid 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.                       *
 *                                                                         *
 *   netctl-plasmoid 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 netctl-plasmoid. If not, see http://www.gnu.org/licenses/  *
 ***************************************************************************/

#include "wpasupinteract.h"

#include <QProcess>

#include "mainwindow.h"
#include "netctlinteract.h"
#include "sleepthread.h"
#include <cstdio>


WpaSup::WpaSup(MainWindow *wid, QStringList wpaConfig, QString sudoPath, QString ifaceDir, QString preferedInterface)
    : parent(wid),
      wpaConf(wpaConfig),
      sudoCommand(sudoPath),
      ifaceDirectory(new QDir(ifaceDir)),
      mainInterface(preferedInterface)
{
    // terminate old loaded profile
    if (QFile(wpaConf[2]).exists() || QDir(wpaConf[4]).exists())
        stopWpaSupplicant();
}


WpaSup::~WpaSup()
{
    delete ifaceDirectory;
}


// general information
QStringList WpaSup::getInterfaceList()
{
    QStringList interfaces;

    interfaces.append(mainInterface);
    QStringList allInterfaces = ifaceDirectory->entryList(QDir::Dirs | QDir::NoDotAndDotDot);
    for (int i=0; i<allInterfaces.count(); i++)
        if (QDir(ifaceDirectory->path() + QDir::separator() + allInterfaces[i] +
                 QDir::separator() + QString("wireless")).exists())
            interfaces.append(allInterfaces[i]);


    return interfaces;
}


// functions
bool WpaSup::wpaCliCall(QString commandLine)
{
    QString interface = getInterfaceList()[0];
    QProcess command;
    command.start(wpaConf[0] + QString(" -i ") + interface + QString(" -p ") + wpaConf[4] +
            QString(" -P ") + wpaConf[2] + QString(" ") + commandLine);
    command.waitForFinished(-1);
    SleepThread::sleep(1);
    if (command.exitCode() == 0)
        return true;
    else
        return false;
}


QString WpaSup::getWpaCliOutput(QString commandLine)
{
    QString interface = getInterfaceList()[0];
    QProcess command;
    command.start(wpaConf[0] + QString(" -i ") + interface + QString(" -p ") + wpaConf[4] +
            QString(" -P ") + wpaConf[2] + QString(" ") + commandLine);
    command.waitForFinished(-1);
    return command.readAllStandardOutput();
}


bool WpaSup::isProfileActive(QString profile)
{
    QString profileFile;
    QList<QStringList> profileList = parent->netctlCommand->getProfileList();
    for (int i=0; i<profileList.count(); i++)
        if (profile == parent->netctlCommand->getSsidFromProfile(profileList[i][0]))
            profileFile = profileList[i][0];
    return parent->netctlCommand->isProfileActive(profileFile);
}


bool WpaSup::isProfileExists(QString profile)
{
    bool exists = false;
    QList<QStringList> profileList = parent->netctlCommand->getProfileList();
    for (int i=0; i<profileList.count(); i++)
        if (profile == parent->netctlCommand->getSsidFromProfile(profileList[i][0]))
            exists = true;
    return exists;
}


bool WpaSup::startWpaSupplicant()
{
    if (!QFile(wpaConf[2]).exists()) {
        QString interface = getInterfaceList()[0];
        QProcess command;
        command.start(sudoCommand + QString(" ") + wpaConf[1] + QString(" -B -P ") + wpaConf[2] +
                QString(" -i ") + interface + QString(" -D ") + wpaConf[3] +
                QString(" -C \"DIR=") + wpaConf[4] + QString(" GROUP=") + wpaConf[5]);
        command.waitForFinished(-1);
        SleepThread::sleep(1);
        if (command.exitCode() != 0)
            return false;
    }
    return true;
}


bool WpaSup::stopWpaSupplicant()
{
    return wpaCliCall(QString("terminate"));
}


QList<QStringList> WpaSup::scanWifi()
{
    QList<QStringList> scanResults;
    startWpaSupplicant();
    if (!wpaCliCall(QString("scan")))
        return scanResults;
    SleepThread::sleep(3);

    QStringList rawOutput = getWpaCliOutput(QString("scan_results")).split(QString("\n"), QString::SkipEmptyParts);
    // remove table header
    rawOutput.removeFirst();
    // remove duplicates
    QStringList rawList;
    for (int i=0; i<rawOutput.count(); i++) {
        bool exist = false;
        if (rawOutput[i].split(QString("\t"), QString::SkipEmptyParts).count() > 4)
            for (int j=0; j<rawList.count(); j++)
                if (rawList[j].split(QString("\t"), QString::SkipEmptyParts).count() > 4)
                    if (rawOutput[i].split(QString("\t"), QString::SkipEmptyParts)[4] ==
                            rawList[j].split(QString("\t"), QString::SkipEmptyParts)[4])
                        exist = true;
        if (!exist)
            rawList.append(rawOutput[i]);
    }

    for (int i=0; i<rawList.count(); i++) {
        QStringList wifiPoint;

        // point name
        if (rawList[i].split(QString("\t"), QString::SkipEmptyParts).count() > 4)
            wifiPoint.append(rawList[i].split(QString("\t"), QString::SkipEmptyParts)[4]);
        else
            wifiPoint.append(QString("<hidden>"));
        // profile status
        QString status;
        if (isProfileExists(wifiPoint[0])) {
            status = QString("exists");
            if (isProfileActive(wifiPoint[0]))
                status = status + QString(" (active)");
            else
                status = status + QString(" (inactive)");
        }
        else
            status = QString("new");
        wifiPoint.append(status);
        // point signal
        wifiPoint.append(rawList[i].split(QString("\t"), QString::SkipEmptyParts)[2]);
        // point security
        QString security = rawList[i].split(QString("\t"), QString::SkipEmptyParts)[3];
        if (security.indexOf(QString("WPA2")) > -1)
            security = QString("WPA2");
        else if (security.indexOf(QString("WPA")) > -1)
            security = QString("WPA");
        else if (security.indexOf(QString("WEP")) > -1)
            security = QString("WEP");
        else
            security = QString("none");
        wifiPoint.append(security);

        scanResults.append(wifiPoint);
    }

    stopWpaSupplicant();
    return scanResults;
}