src/data/contactsmodel.cpp - develop - lmaxyz/telegraf

src/data/contactsmodel.cpp
140 строк5 KB
mbarashkov
Больше рефакторинга
24 фев 2025, 20:06
24 фев 2025, 20:0650096d2
#include "contactsmodel.h"
#include <QListIterator>

namespace {
    const char ID[] = "id";
    const char LAST_NAME[] = "last_name";
    const char FIRST_NAME[] = "first_name";
    const char USERNAME[] = "username";
}

ContactsModel::ContactsModel(QObject *parent)
    : QAbstractListModel(parent)
{    
    connect(TDLibResponseHandler::getInstance(), SIGNAL(usersReceived(QString,QVariantList,int)), this, SLOT(handleUsersReceived(QString,QVariantList,int)));
}

QHash<int, QByteArray> ContactsModel::roleNames() const
{
    QHash<int,QByteArray> roles;
    roles.insert(ContactRole::RoleDisplay, "display");
    roles.insert(ContactRole::RoleTitle, "title");
    roles.insert(ContactRole::RoleUserId, "user_id");
    roles.insert(ContactRole::RoleUsername, "username");
    roles.insert(ContactRole::RolePhotoSmall, "photo_small");
    roles.insert(ContactRole::RoleUserStatus, "user_status");
    roles.insert(ContactRole::RoleUserLastOnline, "user_last_online");
    roles.insert(ContactRole::RoleFilter, "filter");
    return roles;
}

int ContactsModel::rowCount(const QModelIndex &) const
{
    return this->contacts.size();
}

QVariant ContactsModel::data(const QModelIndex &index, int role) const
{
    if (index.isValid()) {
        QVariantMap requestedContact = contacts.value(index.row()).toMap();
        switch (static_cast<ContactRole>(role)) {
            case ContactRole::RoleDisplay: return requestedContact;
            case ContactRole::RoleTitle: return QString(requestedContact.value("first_name").toString() + " " + requestedContact.value("last_name").toString()).trimmed();
            case ContactRole::RoleUserId: return requestedContact.value("id");
            case ContactRole::RoleUsername: return requestedContact.value("usernames").toMap().value("editable_username").toString();
            case ContactRole::RolePhotoSmall: return requestedContact.value("profile_photo").toMap().value("small");
            case ContactRole::RoleUserStatus: return requestedContact.value("status").toMap().value("@type");
            case ContactRole::RoleUserLastOnline: return requestedContact.value("status").toMap().value("was_online");
            case ContactRole::RoleFilter: return QString(requestedContact.value("first_name").toString() + " " + requestedContact.value("last_name").toString() + " " + requestedContact.value("username").toString()).trimmed();
        }
    }
    return QVariant();
}

void ContactsModel::handleUsersReceived(const QString &extra, const QVariantList &userIds, int)
{
    if (extra == "contactsRequested") {
        this->contactIds.clear();
        QListIterator<QVariant> userIdIterator(userIds);
        while (userIdIterator.hasNext()) {
            QVariant nextUserId = userIdIterator.next();
            QString nextUserIdString = nextUserId.toString();

            if (!TDLibDataRepository::getInstance()->containsUserById(nextUserId.toLongLong())) {
                TDLibRequestSender::getInstance()->callGetUserFullInfo(nextUserIdString);
            }

            this->contactIds.append(nextUserIdString);
        }
    }
}

static bool compareUsers(const QVariant &user1, const QVariant &user2)
{
    const QVariantMap userMap1 = user1.toMap();
    const QVariantMap userMap2 = user2.toMap();

    const QString lastName1 = userMap1.value(LAST_NAME).toString();
    const QString lastName2 = userMap2.value(LAST_NAME).toString();
    if (!lastName1.isEmpty()) {
        if (lastName1 < lastName2) {
            return true;
        } else if (lastName1 > lastName2) {
            return false;
        }
    }

    const QString firstName1 = userMap1.value(FIRST_NAME).toString();
    const QString firstName2 = userMap2.value(FIRST_NAME).toString();
    if (firstName1 < firstName2) {
        return true;
    } else if (firstName1 > firstName2) {
        return false;
    }
    const QString userName1 = userMap1.value(USERNAME).toString();
    const QString userName2 = userMap2.value(USERNAME).toString();
    if (userName1 < userName2) {
        return true;
    } else if (userName1 > userName2) {
        return false;
    }
    return userMap1.value(ID).toLongLong() < userMap2.value(ID).toLongLong();
}

void ContactsModel::hydrateContacts()
{
    this->contacts.clear();
    QListIterator<QString> userIdIterator(contactIds);
    while (userIdIterator.hasNext()) {
        QString nextUserId = userIdIterator.next();
        this->contacts.append(TDLibDataRepository::getInstance()->getUserById(nextUserId.toLongLong()));
    }
    std::sort(this->contacts.begin(), this->contacts.end(), compareUsers);
}

void ContactsModel::startImportingContacts()
{
    this->deviceContacts.clear();
}

void ContactsModel::stopImportingContacts()
{
    if (!deviceContacts.isEmpty()) {
        TDLibRequestSender::getInstance()->callImportContacts(deviceContacts);
    }
}

void ContactsModel::importContact(const QVariantMap &singlePerson)
{
    QString firstName = singlePerson.value("firstName").toString();
    QVariantList phoneNumbers = singlePerson.value("phoneNumbers").toList();
    if (!firstName.isEmpty() && !phoneNumbers.isEmpty()) {
        for (QVariant& phoneNumber : phoneNumbers) {
            QVariantMap singleContact;
            singleContact.insert("first_name", firstName);
            singleContact.insert("last_name", singlePerson.value("lastName").toString());
            singleContact.insert("phone_number", phoneNumber.toString());
            deviceContacts.append(singleContact);
        }
    }
}