buddylistitemmodel.cpp

/* DUKTO - A simple, fast and multi-platform file transfer tool for LAN users
 * Copyright (C) 2011 Emanuele Colombo
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#include "buddylistitemmodel.h"

#include <QUrl>
#include <QRegularExpression>

#include "platform.h"
#include "peer.h"

BuddyListItemModel::BuddyListItemModel() :
    QStandardItemModel(nullptr)
{
    QHash<int, QByteArray> roleNames;
    roleNames[Ip] = "ip";
    roleNames[Port] = "port";
    roleNames[Username] = "username";
    roleNames[System] = "system";
    roleNames[Platform] = "platform";
    roleNames[GenericAvatar] = "generic";
    roleNames[Avatar] = "avatar";
    roleNames[OsLogo] = "oslogo";
    roleNames[ShowBack] = "showback";

    setItemRoleNames(roleNames);
}

void BuddyListItemModel::addMeElement()
{
    addBuddy("",
             0,
             Platform::getUsername() + " (You)",
             Platform::getHostname(),
             Platform::getPlatformName(),
             QUrl::fromLocalFile(Platform::getAvatarPath()));
}

void BuddyListItemModel::addIpElement()
{
    addBuddy("IP",
             0,
             "IP connection",
             "Send data to a remote device",
             "IP",
             QUrl(""));
}

void BuddyListItemModel::addBuddy(const QString &ip, qint16 port, const QString &username, const QString &system, const QString &platform, const QUrl &avatarPath)
{
    QStandardItem* it = nullptr;
    bool add = true;

    // Check if the same IP is alreay in the buddy list
    if (mItemsMap.contains(ip)) {
        it = mItemsMap[ip];
        add = false;
    }
    else
        it = new QStandardItem();
    it->setData(ip, BuddyListItemModel::Ip);
    it->setData(port, BuddyListItemModel::Port);
    if (add) {
        it->setData(false, BuddyListItemModel::ShowBack);
    }

    // Set (or update) data
    it->setData(username, BuddyListItemModel::Username);
    if (ip != "IP")
        it->setData("at " + system, BuddyListItemModel::System);
    else
        it->setData(system, BuddyListItemModel::System);
    it->setData(platform, BuddyListItemModel::Platform);
    it->setData(avatarPath, BuddyListItemModel::Avatar);

    QString platform_lower = platform.toLower();
    // Update generic avatar
    if ((platform_lower == "symbian") || (platform_lower == "android") || (platform_lower == "ios") || (platform_lower == "blackberry") || (platform_lower == "windowsphone"))
        it->setData("SmartphoneLogo.png", BuddyListItemModel::GenericAvatar);
    else if (platform_lower == "ip")
        it->setData("IpLogo.png", BuddyListItemModel::GenericAvatar);
    else
        it->setData("PcLogo.png", BuddyListItemModel::GenericAvatar);

    // Update logo
    if (platform_lower == "windows")
        it->setData("WindowsLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "macintosh")
        it->setData("AppleLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "linux")
        it->setData("LinuxLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "symbian")
        it->setData("SymbianLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "ios")
        it->setData("IosLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "windowsphone")
        it->setData("WindowsPhoneLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "blackberry")
        it->setData("BlackberryLogo.png", BuddyListItemModel::OsLogo);
    else if (platform_lower == "android")
        it->setData("AndroidLogo.png", BuddyListItemModel::OsLogo);
    else
        it->setData("UnknownLogo.png", BuddyListItemModel::OsLogo);

    // Add elemento to the list
    if (add) {
        appendRow(it);
        if (!ip.isEmpty())
            mItemsMap.insert(ip, it);
        else
            mMeItem = it;
    }
}

void BuddyListItemModel::addBuddy(const Peer &peer)
{
    static int seq = 0;
    static QRegularExpression rx("^(.*)\\sat\\s(.*)\\s\\((.*)\\)$");
    QRegularExpressionMatch match = rx.match(peer.name);

    QString username = match.captured(1);
    QString system = match.captured(2);
    QString platform = match.captured(3);
    QUrl avatarPath = QUrl("http://" + peer.address.toString() + ":" + QString::number(peer.port + 1) + "/dukto/avatar");
    avatarPath.setQuery(QString::number(seq++));

    addBuddy(peer.address.toString(),
             peer.port,
             username,
             system,
             platform,
             avatarPath);
}

void BuddyListItemModel::removeBuddy(const QString &ip)
{
    // Check for element
    if (!mItemsMap.contains(ip)) return;

    // Get element
    QStandardItem* it = mItemsMap[ip];

    // Remove element
    mItemsMap.remove(ip);
    this->removeRow(this->indexFromItem(it).row());
}

void BuddyListItemModel::clearBuddies() {
    // Keep first two rows: Me and Ip
    removeRows(2, rowCount() - 2);

    QMutableHashIterator<QString, QStandardItem*> iter(mItemsMap);
    while (iter.hasNext()) {
        iter.next();
        if (iter.key() != "IP") {
            iter.remove();
        }
    }
}

void BuddyListItemModel::showSingleBack(int idx)
{
    for (int i = 0; i < rowCount(); i++)
        itemFromIndex(index(i, 0))->setData(false, BuddyListItemModel::ShowBack);
    itemFromIndex(index(idx, 0))->setData(true, BuddyListItemModel::ShowBack);
}

QString BuddyListItemModel::buddyNameByIp(const QString &ip)
{
    if (!mItemsMap.contains(ip)) return "";
    return mItemsMap.value(ip)->data(BuddyListItemModel::Username).toString();
}

QStandardItem* BuddyListItemModel::buddyByIp(const QString &ip)
{
    if (!mItemsMap.contains(ip)) return nullptr;
    return mItemsMap.value(ip);
}

QString BuddyListItemModel::fistBuddyIp()
{
    if (this->rowCount() < 3) return "";
    return this->index(2, 0).data(BuddyListItemModel::Ip).toString();
}

void BuddyListItemModel::updateMeElement()
{
    if (mMeItem == nullptr) {
        return;
    }
    static int seq = 0;
    QString path = Platform::getAvatarPath();
    if (path.isEmpty()) {
        mMeItem->setData("", BuddyListItemModel::Avatar);
    } else {
        QUrl url = QUrl::fromLocalFile(path);
        // force update with a different url each time
        url.setQuery("m=" + QString::number(++seq));
        mMeItem->setData(url, BuddyListItemModel::Avatar);
    }
    mMeItem->setData(Platform::getUsername() + " (You)", BuddyListItemModel::Username);
}

QString BuddyListItemModel::getMeGenericAvatar() {
    if (mMeItem == nullptr) {
        return QString();
    }
    return mMeItem->data(BuddyListItemModel::GenericAvatar).toString();
}