#include "qwifinetworklist.h"

#include <QtCore>

const int ID_BSSID = (Qt::ItemDataRole) (Qt::UserRole + 1);
const int ID_SSID = (Qt::ItemDataRole) (Qt::UserRole + 2);
const int ID_SIGNAL = (Qt::ItemDataRole) (Qt::UserRole + 3);
const int ID_WPA2 = (Qt::ItemDataRole) (Qt::UserRole + 4);
const int ID_WPA = (Qt::ItemDataRole) (Qt::UserRole + 5);
const int ID_NETWORK = (Qt::ItemDataRole) (Qt::UserRole + 6);

QWifiNetworkList::QWifiNetworkList(QWifiManager *manager)
    : m_manager(manager)
{
}


QHash<int, QByteArray> QWifiNetworkList::roleNames() const
{
    QHash<int, QByteArray> names;
    names.insert(ID_BSSID, "bssid");
    names.insert(ID_SSID, "ssid");
    names.insert(ID_SIGNAL, "strength");
    names.insert(ID_WPA2, "wpa2");
    names.insert(ID_WPA, "wpa");
    names.insert(ID_NETWORK, "network");
    return names;
}



QVariant QWifiNetworkList::data(const QModelIndex &index, int role) const
{
    QWifiNetwork *n = m_networks.at(index.row());

    switch (role) {
    case ID_BSSID: return n->bssid();
    case ID_SSID: return n->ssid();
    case ID_SIGNAL: return n->signalStrength();
    case ID_WPA2: return n->supportsWPA2();
    case ID_WPA: return n->supportsWPA();
    case ID_NETWORK: return QVariant::fromValue((QObject *) n);
    }

    qDebug("QWifiNetworkList::data(), undefined role: %d\n", role);

    return QVariant();
}

QWifiNetwork *QWifiNetworkList::networkForBSSID(const QByteArray &bssid, int *pos)
{
    for (int i=0; i<m_networks.size(); ++i) {
        if (m_networks.at(i)->bssid() == bssid) {
            if (pos)
                *pos = i;
            return m_networks.at(i);
        }
    }
    return 0;
}


void QWifiNetworkList::parseScanResults(const QByteArray &results)
{
    QList<QByteArray> lines = results.split('\n');

    QSet<QByteArray> bssids;
    for (int i=1; i<lines.size(); ++i) {
        QList<QByteArray> info = lines.at(i).split('\t');
        if (info.size() < 5 || info.at(4).isEmpty() || info.at(0).isEmpty())
            continue;
        bssids.insert(info.at(0));
        int pos = 0;
        QWifiNetwork *existing = networkForBSSID(info.at(0), &pos);
        if (!existing) {
            QWifiNetwork *network = new QWifiNetwork();
            network->setBssid(info.at(0));
            network->setFlags(info.at(3));
            network->setSignalStrength(info.at(2).toInt());
            network->setSsid(info.at(4));
            beginInsertRows(QModelIndex(), m_networks.size(), m_networks.size());
            m_networks << network;
            endInsertRows();

        } else {
            existing->setSignalStrength(info.at(2).toInt());
            dataChanged(createIndex(pos, 0), createIndex(pos, 0));
        }
    }

    for (int i=0; i<m_networks.size(); ) {
        if (!bssids.contains(m_networks.at(i)->bssid())) {
            beginRemoveRows(QModelIndex(), i, i);
            delete m_networks.takeAt(i);
            endRemoveRows();
        } else {
            ++i;
        }
    }

//    for (int i=0; i<m_networks.size(); ++i) {
//        qDebug() << " - network:" << m_networks.at(i)->bssid() << m_networks.at(i)->ssid() << m_networks.at(i)->flags() << m_networks.at(i)->signalStrength();
//    }
}